// SPDX-License-Identifier: LGPL-3.0-or-other
// Copyright (C) 2021 Contributors to the SLS Detector Package
#ifndef ETA_INTERPOLATION_POSXY_H
#define ETA_INTERPOLATION_POSXY_H

//#include "sls/tiffIO.h"
#include "eta2InterpolationBase.h"
#include "eta3InterpolationBase.h"
#include "etaInterpolationBase.h"

class etaInterpolationPosXY : public virtual etaInterpolationBase {
  public:
    etaInterpolationPosXY(int nx = 400, int ny = 400, int ns = 25, int nsy = 25,
                          int nb = -1, int nby = -1, double emin = 1,
                          double emax = 0)
        : etaInterpolationBase(nx, ny, ns, nsy, nb, nby, emin, emax){
              //  cout << "epxy " << nb << " " << emin << " " << emax << endl;
              //  cout << nbeta << " " << etamin << " " << etamax << endl;
          };

    etaInterpolationPosXY(etaInterpolationPosXY *orig)
        : etaInterpolationBase(orig){};

    virtual etaInterpolationPosXY *Clone() = 0; /* { */

    /* return new etaInterpolationPosXY(this); */

    /* }; */

    virtual void prepareInterpolation(int &ok) {
        ok = 1;

        ///*Eta Distribution Rebinning*///
        // double bsize=1./nSubPixels; //precision
        // cout<<"nPixelsX = "<<nPixelsX<<" nPixelsY = "<<nPixelsY<<" nSubPixels
        // = "<<nSubPixels<<endl;
        double tot_eta = 0;
        double tot_eta_x = 0;
        double tot_eta_y = 0;
        for (int ip = 0; ip < nbetaX * nbetaY; ip++)
            tot_eta += heta[ip];
        cout << "total eta entries is :" << tot_eta << endl;
        if (tot_eta <= 0) {
            ok = 0;
            return;
        };

        double *hx = new double[nbetaX];  // profile x
        double *hy = new double[nbetaY];  // profile y
        double *hix = new double[nbetaX]; // integral of projection x
        double *hiy = new double[nbetaY]; // integral of projection y
        //  int ii=0;
        double etax, etay;
        for (int ib = 0; ib < nbetaX; ib++) {

            // tot_eta_y=0;

            for (int iby = 0; iby < nbetaY; iby++) {
                etay = etamin + iby * etastepY;
                // cout << etax << endl;

                // tot_eta_x+=hx[iby];
                if (etay >= 0 && etay <= 1)
                    hy[iby] = heta[ib + iby * nbetaX];
                else
                    hy[iby] = 0;
                // tot_eta_y+=hy[iby];
            }

            hiy[0] = hy[0];

            for (int iby = 1; iby < nbetaY; iby++) {
                hiy[iby] = hiy[iby - 1] + hy[iby];
            }

            tot_eta_y = hiy[nbetaY - 1] + 1;

            for (int iby = 0; iby < nbetaY; iby++) {
                if (tot_eta_y <= 0) {
                    hhy[ib + iby * nbetaX] = -1;
                    // ii=(ibx*nSubPixels)/nbeta;
                } else {
                    // if (hiy[ibx]>tot_eta_y*(ii+1)/nSubPixels) ii++;
                    hhy[ib + iby * nbetaX] = hiy[iby] / tot_eta_y;
                }
            }
        }

        for (int ib = 0; ib < nbetaY; ib++) {

            for (int ibx = 0; ibx < nbetaX; ibx++) {
                etax = etamin + ibx * etastepX;
                // cout << etax << endl;
                if (etax >= 0 && etax <= 1)
                    hx[ibx] = heta[ibx + ib * nbetaX];
                else {
                    hx[ibx] = 0;
                }
            }
            hix[0] = hx[0];

            for (int ibx = 1; ibx < nbetaX; ibx++) {
                hix[ibx] = hix[ibx - 1] + hx[ibx];
            }

            tot_eta_x = hix[nbetaX - 1] + 1;

            for (int ibx = 0; ibx < nbetaX; ibx++) {
                if (tot_eta_x <= 0) {
                    hhx[ibx + ib * nbetaX] = -1;
                } else {
                    hhx[ibx + ib * nbetaX] = hix[ibx] / tot_eta_x;
                }
            }

            for (int ibx = 0; ibx < nbetaX; ibx++) {
                if (tot_eta_x <= 0) {
                    hhx[ibx + ib * nbetaX] = -1;
                } else {
                    // if (hix[ibx]>tot_eta_x*(ii+1)/nSubPixels) ii++;
                    hhx[ibx + ib * nbetaX] = hix[ibx] / tot_eta_x;
                }
            }
        }

        int ibx, iby, ib;

        iby = 0;
        while (hhx[iby * nbetaY + nbetaY / 2] < 0)
            iby++;
        for (ib = 0; ib < iby; ib++) {
            for (ibx = 0; ibx < nbetaX; ibx++)
                hhx[ibx + nbetaX * ib] = hhx[ibx + nbetaX * iby];
        }
        iby = nbetaY - 1;

        while (hhx[iby * nbetaY + nbetaY / 2] < 0)
            iby--;
        for (ib = iby + 1; ib < nbetaY; ib++) {
            for (ibx = 0; ibx < nbetaX; ibx++)
                hhx[ibx + nbetaX * ib] = hhx[ibx + nbetaX * iby];
        }

        iby = 0;
        while (hhy[nbetaX / 2 * nbetaX + iby] < 0)
            iby++;
        for (ib = 0; ib < iby; ib++) {
            for (ibx = 0; ibx < nbetaY; ibx++)
                hhy[ib + nbetaX * ibx] = hhy[iby + nbetaX * ibx];
        }
        iby = nbetaX - 1;

        while (hhy[nbetaX / 2 * nbetaX + iby] < 0)
            iby--;
        for (ib = iby + 1; ib < nbetaX; ib++) {
            for (ibx = 0; ibx < nbetaY; ibx++)
                hhy[ib + nbetaX * ibx] = hhy[iby + nbetaX * ibx];
        }

#ifdef SAVE_ALL
        debugSaveAll();
#endif
        delete[] hx;
        delete[] hy;
        delete[] hix;
        delete[] hiy;

        return;
    }
};

class eta2InterpolationPosXY : public virtual eta2InterpolationBase,
                               public virtual etaInterpolationPosXY {
  public:
    eta2InterpolationPosXY(int nx = 400, int ny = 400, int ns = 25,
                           int nsy = 25, int nb = -1, int nby = -1,
                           double emin = 1, double emax = 0)
        : etaInterpolationBase(nx, ny, ns, nsy, nb, nby, emin, emax),
          eta2InterpolationBase(nx, ny, ns, nsy, nb, nby, emin, emax),
          etaInterpolationPosXY(nx, ny, ns, nsy, nb, nby, emin, emax){
              //  cout << "e2pxy " << nb << " " << emin << " " << emax << endl;
          };

    eta2InterpolationPosXY(eta2InterpolationPosXY *orig)
        : etaInterpolationBase(orig), etaInterpolationPosXY(orig){};

    virtual eta2InterpolationPosXY *Clone() {
        return new eta2InterpolationPosXY(this);
    };
};

class eta3InterpolationPosXY : public virtual eta3InterpolationBase,
                               public virtual etaInterpolationPosXY {
  public:
    eta3InterpolationPosXY(int nx = 400, int ny = 400, int ns = 25,
                           int nsy = 25, int nb = -1, int nby = -1,
                           double emin = 1, double emax = 0)
        : etaInterpolationBase(nx, ny, ns, nsy, nb, nby, emin, emax),
          eta3InterpolationBase(nx, ny, ns, nsy, nb, nby, emin, emax),
          etaInterpolationPosXY(nx, ny, ns, nsy, nb, nby, emin, emax){
              //   cout << "e3pxy " << nbeta << " " << etamin << " " << etamax
              //   << " " << nSubPixels<< endl;
          };

    eta3InterpolationPosXY(eta3InterpolationPosXY *orig)
        : etaInterpolationBase(orig), etaInterpolationPosXY(orig){};

    virtual eta3InterpolationPosXY *Clone() {
        return new eta3InterpolationPosXY(this);
    };
};

#endif