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

#include <fstream>
#include <iostream>

template <class dataType> class slsDetectorData {

  protected:
    const int nx;  /**< Number of pixels in the x direction */
    const int ny;  /**< Number of pixels in the y direction */
    int dataSize;  /**<size of the data constituting one frame */
    int **dataMap; /**< Array of size nx*ny storing the pointers to the data in
                      the dataset (as offset)*/
    dataType **dataMask; /**< Array of size nx*ny storing the polarity of the
                            data in the dataset (should be 0 if no inversion is
                            required, 0xffffffff is inversion is required) */
    int **dataROIMask; /**< Array of size nx*ny 1 if channel is good (or in the
                          ROI), 0 if bad channel (or out of ROI)   */
    int *xmap;
    int *ymap;
    dataType **orderedData;
    int isOrdered;

  public:
    /**


       General slsDetectors data structure. Works for data acquired using the
    slsDetectorReceiver. Can be generalized to other detectors (many virtual
    funcs).

       Constructor (no error checking if datasize and offsets are compatible!)
    \param npx number of pixels in the x direction
    \param npy number of pixels in the y direction (1 for strips)
    \param dsize size of the data
    \param dMap array of size nx*ny storing the pointers to the data in the
    dataset (as offset) \param dMask Array of size nx*ny storing the polarity of
    the data in the dataset (should be 0 if no inversion is required, 0xffffffff
    is inversion is required) \param dROI Array of size nx*ny. The elements are
    1s if the channel is good or in the ROI, 0 is bad or out of the ROI. NULL
    (default) means all 1s.

    */
    slsDetectorData(int npx, int npy, int dsize, int **dMap = NULL,
                    dataType **dMask = NULL, int **dROI = NULL)
        : nx(npx), ny(npy), dataSize(dsize), orderedData(NULL), isOrdered(0) {

        int el = dsize / sizeof(dataType);
        xmap = new int[el];
        ymap = new int[el];

        orderedData = new dataType *[ny];
        dataMap = new int *[ny];
        dataMask = new dataType *[ny];
        dataROIMask = new int *[ny];
        for (int i = 0; i < ny; i++) {
            dataMap[i] = new int[nx];
            orderedData[i] = new dataType[nx];
            dataMask[i] = new dataType[nx];
            dataROIMask[i] = new int[nx];
            for (int j = 0; j < nx; j++)
                dataROIMask[i][j] = 1;
        }

        for (int ip = 0; ip < el; ip++) {
            xmap[ip] = -1;
            ymap[ip] = -1;
        }

        setDataMap(dMap);
        setDataMask(dMask);
        setDataROIMask(dROI);
    }

    virtual ~slsDetectorData() {
        for (int i = 0; i < ny; i++) {
            delete[] dataMap[i];
            delete[] dataMask[i];
            delete[] dataROIMask[i];
            delete[] orderedData[i];
        }
        delete[] dataMap;
        delete[] dataMask;
        delete[] dataROIMask;
        delete[] orderedData;
        delete[] xmap;
        delete[] ymap;
    };

    virtual int getPointer(int ix, int iy) { return dataMap[iy][ix]; };
    /**
       defines the data map (as offset) - no error checking if datasize and
       offsets are compatible! \param dMap array of size nx*ny storing the
       pointers to the data in the dataset (as offset). If NULL (default),the
       data are arranged as if read out row by row
       (dataMap[iy][ix]=(iy*nx+ix)*sizeof(dataType);)
    */
    void setDataMap(int **dMap = NULL) {

        int ip = 0;
        int ix, iy;
        if (dMap == NULL) {
            for (iy = 0; iy < ny; iy++) {
                for (ix = 0; ix < nx; ix++) {
                    dataMap[iy][ix] = (iy * nx + ix) * sizeof(dataType);
                }
            }
        } else {
            // cout << "set dmap "<< dataMap << " " << dMap << endl;
            for (iy = 0; iy < ny; iy++) {
                // cout << iy << endl;
                for (ix = 0; ix < nx; ix++) {
                    dataMap[iy][ix] = dMap[iy][ix];
                    // cout << ix << " " << iy << endl;
                    /*ip=dataMap[ix][iy]/sizeof(dataType);
                      xmap[ip]=ix;
                      ymap[ip]=iy;Annaa*/
                }
            }
        }
        for (iy = 0; iy < ny; iy++) {
            for (ix = 0; ix < nx; ix++) {
                ip = dataMap[iy][ix] / sizeof(dataType);
                xmap[ip] = ix;
                ymap[ip] = iy;
            }
        }

        // cout << "nx:" <<nx << " ny:" << ny << endl;
    };

    /**
       defines the data mask i.e. the polarity of the data
       \param dMask Array of size nx*ny storing the polarity of the data in the
       dataset (should be 0 if no inversion is required, 0xffffffff is inversion
       is required)

    */
    void setDataMask(dataType **dMask = NULL) {

        if (dMask != NULL) {

            for (int iy = 0; iy < ny; iy++)
                for (int ix = 0; ix < nx; ix++)
                    dataMask[iy][ix] = dMask[iy][ix];
        } else {
            for (int iy = 0; iy < ny; iy++)
                for (int ix = 0; ix < nx; ix++)
                    dataMask[iy][ix] = 0;
        }
    };
    /**
       defines the region of interest and/or the bad channels mask
       \param dROI Array of size nx*ny. The lements are 1s if the channel is
       good or in the ROI, 0 is bad or out of the ROI. NULL (default) means all
       1s.

    */
    void setDataROIMask(int **dROI = NULL) {

        if (dROI != NULL) {

            for (int iy = 0; iy < ny; iy++)
                for (int ix = 0; ix < nx; ix++)
                    dataROIMask[iy][ix] = dROI[iy][ix];
        } else {

            for (int iy = 0; iy < ny; iy++)
                for (int ix = 0; ix < nx; ix++)
                    dataROIMask[iy][ix] = 1;
        }
    };

    /**
       Define bad channel or roi mask for a single channel
       \param ix channel x coordinate
       \param iy channel y coordinate (1 for strips)
       \param i 1 if pixel is good (or in the roi), 0 if bad
       \returns 1 if pixel is good, 0 if it's bad, -1 if pixel is out of range
    */
    int setGood(int ix, int iy, int i = 1) {
        if (ix >= 0 && ix < nx && iy >= 0 && iy < ny)
            dataROIMask[iy][ix] = i;
        return isGood(ix, iy);
    };
    /**
       Define bad channel or roi mask for a single channel
       \param ix channel x coordinate
       \param iy channel y coordinate (1 for strips)
       \returns 1 if pixel is good, 0 if it's bad, -1 if pixel is out of range
    */
    int isGood(int ix, int iy) {
        if (ix >= 0 && ix < nx && iy >= 0 && iy < ny)
            return dataROIMask[iy][ix];
        else
            return -1;
    };

    /**
       Returns detector size in x,y
       \param npx reference to number of channels in x
       \param npy reference to number of channels in y (will be 1 for strips)
       \returns total number of channels
    */
    int getDetectorSize(int &npx, int &npy) {
        npx = nx;
        npy = ny;
        return nx * ny;
    };

    /** Returns the size of the data frame */
    int getDataSize() { return dataSize; };

    /** changes the size of the data frame */
    int setDataSize(int d) {
        dataSize = d;
        return dataSize;
    };

    virtual void getPixel(int ip, int &x, int &y) {
        x = xmap[ip];
        y = ymap[ip];
    };

    virtual dataType **getData(char *ptr, int dsize = -1) {
        int el = dsize / sizeof(dataType);
        // dataType **data;
        int ix, iy;
        // data=new dataType*[ny];
        // for(int i = 0; i < ny; i++) {
        //   data[i]=new dataType[nx];
        // }
        isOrdered = 0;
        if (dsize <= 0 || dsize > dataSize)
            dsize = dataSize;

        for (int ip = 0; ip < (el); ip++) {
            getPixel(ip, ix, iy);
            if (ix >= 0 && ix < nx && iy >= 0 && iy < ny) {
                // data[iy][ix]=getChannel(ptr,ix,iy);
                orderedData[iy][ix] = *(ptr + ip); // getChannel(ptr,ix,iy);
            }
        }
        isOrdered = 1;
        return orderedData;
    }

    void newFrame() { isOrdered = 0; };

    virtual double **getImage(char *ptr, int dsize = -1) {

        double **data;
        int ix, iy;
        data = new double *[ny];
        for (int i = 0; i < ny; i++) {
            data[i] = new double[nx];
        }
        int el = dsize / sizeof(dataType);
        if (dsize <= 0 || dsize > dataSize)
            dsize = dataSize;
        for (int ip = 0; ip < el; ip++) {
            getPixel(ip, ix, iy);
            if (ix >= 0 && ix < nx && iy >= 0 && iy < ny) {
                data[iy][ix] = getValue(ptr, ix, iy);
            }
        }
        return data;
    };

    /**
       Returns the value of the selected channel for the given dataset. Virtual
       function, can be overloaded. \param data pointer to the dataset
       (including headers etc) \param ix pixel number in the x direction \param
       iy pixel number in the y direction \returns data for the selected
       channel, with inversion if required

    */
    virtual dataType getChannel(char *data, int ix, int iy = 0) {
        dataType m = 0, d = 0;
        if (ix >= 0 && ix < nx && iy >= 0 && iy < ny && dataMap[iy][ix] >= 0 &&
            dataMap[iy][ix] < dataSize) {
            // cout << ix << " " << iy << " " ;
            // cout << dataMap[ix][iy] << " " << (void*)data << " " <<
            // dataSize<< endl;
            m = dataMask[iy][ix];
            if (isOrdered == 0)
                d = *((dataType *)(data + getPointer(ix, iy)));
            else
                d = orderedData[iy][ix];
        }
        return d ^ m;
    };

    virtual int getGain(char *data, int ix, int iy = 0) { return 0; };

    /**

       Returns the value of the selected channel for the given dataset. Virtual
       function, can be overloaded. \param data pointer to the dataset
       (including headers etc) \param ix pixel number in the x direction \param
       iy pixel number in the y direction \returns data for the selected
       channel, with inversion if required or -1 if its a missing packet

    */

    virtual int getChannelwithMissingPackets(char *data, int ix, int iy) {
        return 0;
    };

    /**
       Returns the value of the selected channel for the given dataset as
       double. \param data pointer to the dataset (including headers etc) \param
       ix pixel number in the x direction \param iy pixel number in the y
       direction \returns data for the selected channel, with inversion if
       required as double

    */
    virtual double getValue(char *data, int ix, int iy = 0) {
        /* cout << " x "<< ix << " y"<< iy << " val " << getChannel(data, ix,
         * iy)<< endl;*/
        return (double)getChannel(data, ix, iy);
    };

    /**
       Returns the frame number for the given dataset. Purely virtual func.
       \param buff pointer to the dataset
       \returns frame number

    */
    virtual int getFrameNumber(char *buff) = 0;

    /**

       Returns the packet number for the given dataset. purely virtual func
       \param buff pointer to the dataset
       \returns packet number number


       virtual int getPacketNumber(char *buff)=0;

    */

    /**

       Loops over a memory slot until a complete frame is found (i.e. all
       packets 0 to nPackets, same frame number). purely virtual func \param
       data pointer to the memory to be analyzed \param ndata reference to the
       amount of data found for the frame, in case the frame is incomplete at
       the end of the memory slot \param dsize size of the memory slot to be
       analyzed \returns pointer to the beginning of the last good frame (might
       be incomplete if ndata smaller than dataSize), or NULL if no frame is
       found

    */
    virtual char *findNextFrame(char *data, int &ndata, int dsize) = 0;

    /**

      Loops over a file stream until a complete frame is found (i.e. all packets
      0 to nPackets, same frame number). Can be overloaded for different kind of
      detectors! \param filebin input file stream (binary) \returns pointer to
      the begin of the last good frame, NULL if no frame is found or last frame
      is incomplete

    */
    virtual char *readNextFrame(std::ifstream &filebin) = 0;
};

#endif