// SPDX-License-Identifier: LGPL-3.0-or-other
// Copyright (C) 2021 Contributors to the SLS Detector Package
#include "TCanvas.h"
#include "TH2F.h"
#include "TROOT.h"
#include "interpolation_EtaVEL.h"
//#include "EtaVEL.h"
#include "EtaVEL.cpp"
/*
Zum erstellen der correction map ist createGainAndEtaFile(...) in EVELAlg.C der
entry point. Zum erstellen des HR images ist createImage(...) der entry point.
*/
interpolation_EtaVEL::interpolation_EtaVEL(int nx, int ny, int ns,
                                           double etamin, double etamax, int p)
    : slsInterpolation(nx, ny, ns), newEta(NULL), heta(NULL), plot(p) {
    newEta = new EtaVEL(nSubPixels, etamin, etamax, nPixelsX, nPixelsY);
    heta = new TH2F("heta", "heta", 50 * nSubPixels, etamin, etamax,
                    50 * nSubPixels, etamin, etamax);
    heta->SetStats(kFALSE);
}

interpolation_EtaVEL::~interpolation_EtaVEL() {
    delete newEta;
    delete heta;
}

void interpolation_EtaVEL::prepareInterpolation(int &ok, int maxit) {
    int nit = 0;
    while ((newEta->converged != 1) && nit++ < maxit) {
        cout << " -------------- new step " << nit << endl;
        iterate();
    }
    if (plot) {
        Draw();
        gPad->Modified();
        gPad->Update();
    }
    if (newEta->converged == 1)
        ok = 1;
    else
        ok = 0;
}

int interpolation_EtaVEL::addToFlatField(Double_t *cluster, Double_t &etax,
                                         Double_t &etay) {
    Double_t sum, totquad, sDum[2][2];
    int corner = calcEta(cluster, etax, etay, sum, totquad, sDum);
    // check if it's OK...should redo it every time?
    // or should we fill a finer histogram and afterwards re-fill the newEta?
    addToFlatField(etax, etay);
    return corner;
}

int interpolation_EtaVEL::addToFlatField(Double_t etax, Double_t etay) {
    // newEta->fill(etaX,etaY);
    heta->Fill(etax, etay);
    return 0;
}

void interpolation_EtaVEL::iterate() {
    cout << " -------------- newEta refilled" << endl;
    for (int ibx = 0; ibx < heta->GetNbinsX(); ibx++) {
        for (int iby = 0; iby < heta->GetNbinsY(); iby++) {
            newEta->fill(heta->GetXaxis()->GetBinCenter(ibx + 1),
                         heta->GetYaxis()->GetBinCenter(iby + 1),
                         heta->GetBinContent(ibx + 1, iby + 1));
        }
    }
    newEta->updatePixelPos();
    cout << " -------------- pixelPosition updated" << endl;
}

void interpolation_EtaVEL::DrawH() {
    heta->Draw("col");
    (newEta->plotPixelBorder())->Draw();
}

void interpolation_EtaVEL::getInterpolatedPosition(Int_t x, Int_t y,
                                                   Double_t *cluster,
                                                   Double_t &int_x,
                                                   Double_t &int_y) {

    Double_t etax, etay, sum, totquad, sDum[2][2];

    int corner = calcEta(cluster, etax, etay, sum, totquad, sDum);

    int bin = newEta->findBin(etax, etay);
    if (bin <= 0) {
        int_x = -1;
        int_y = -1;
        return;
    }
    double subX =
        ((double)(newEta->getXBin(bin)) + .5) / ((double)newEta->getNPixels());
    double subY =
        ((double)(newEta->getYBin(bin)) + .5) / ((double)newEta->getNPixels());

    double dX, dY;
    switch (corner) {
    case TOP_LEFT:
        dX = -1.;
        dY = +1.;
        break;
    case TOP_RIGHT:
        dX = +1.;
        dY = +1.;
        break;
    case BOTTOM_LEFT:
        dX = -1.;
        dY = -1.;
        break;
    case BOTTOM_RIGHT:
        dX = +1.;
        dY = -1.;
        break;
    default:
        dX = 0;
        dY = 0;
    }

    int_x = ((double)x) + subX + 0.5 * dX;
    int_y = ((double)y) + subY + 0.5 * dY;

    // cout << corner << " " << subX<< " " << subY << " " << dX << " " << dY <<
    // " " << int_x << " " << int_y << endl;
};

// void interpolation_EtaVEL::Streamer(TBuffer &b){newEta->Streamer(b);};
void interpolation_EtaVEL::getInterpolatedBin(Double_t *cluster, Int_t &int_x,
                                              Int_t &int_y) {

    Double_t etax, etay, sum, totquad, sDum[2][2];

    int corner = calcEta(cluster, etax, etay, sum, totquad, sDum);

    int bin = newEta->findBin(etax, etay);
    if (bin < 0) {
        int_x = -1;
        int_y = -1;
        return;
    }
    int_x = newEta->getXBin(bin);
    int_y = newEta->getYBin(bin);
};