#ifndef INTERPOLATINGDETECTOR_H
#define  INTERPOLATINGDETECTOR_H


#include "singlePhotonDetector.h"

#include "slsInterpolation.h"


#ifdef MYROOT1
#include <TTree.h>

#endif


#include <iostream>

using namespace std;


class interpolatingDetector : public singlePhotonDetector {

  /** @short class to perform pedestal subtraction etc. and find single photon clusters for an analog detector */

 public:


  /**

     Constructor (no error checking if datasize and offsets are compatible!)
     \param d detector data structure to be used
     \param csize cluster size (should be an odd number). Defaults to 3
     \param nsigma number of rms to discriminate from the noise. Defaults to 5
     \param sign 1 if photons are positive, -1 if  negative
     \param cm common mode subtraction algorithm, if any. Defaults to NULL i.e. none
     \param nped number of samples for pedestal averaging
     \param nd number of dark frames to average as pedestals without photon discrimination at the beginning of the measurement


  */
  
  
 interpolatingDetector(slsDetectorData<uint16_t> *d, slsInterpolation *inte,
		       double nsigma=5,  
		       int sign=1, 
		       commonModeSubtraction *cm=NULL,
		       int nped=1000, 
		       int nd=100, int nnx=-1, int nny=-1) : 
  singlePhotonDetector(d, 3,nsigma,sign, cm, nped, nd, nnx, nny) , interp(inte)  {};
  
  
  
#ifdef MYROOT1
  virtual TH2F *addToInterpolatedImage(char *data, single_photon_hit *clusters, int &nph)
#endif
#ifndef MYROOT1
  virtual int *addToInterpolatedImage(char *data, single_photon_hit *clusters, int &nph)
#endif
  {
    nph=addFrame(data,clusters,0); 
    if (interp)
      return interp->getInterpolatedImage();
    else
      return NULL;
  };
  

#ifdef MYROOT1
  virtual TH2F *addToFlatField(char *data, single_photon_hit *clusters, int &nph)
#endif
#ifndef MYROOT1
    virtual int *addToFlatField(char *data, single_photon_hit *clusters, int &nph)
#endif
  {
    nph=addFrame(data,clusters,1); 
    if (interp)
      return interp->getFlatField();
    else
      return NULL;
  };
  
  
  
  
  int addFrame(char *data, single_photon_hit *clusters, int ff=0) {
    
    
    
    double int_x,int_y, eta_x, eta_y;
    int nph=0;
    double val[ny][nx];
    int cy=(clusterSizeY+1)/2;
    int cs=(clusterSize+1)/2;
    int ir, ic;
    double cc[2][2];
    double max=0, tl=0, tr=0, bl=0,br=0, *v, vv;
    int xoff,yoff;
    int skip=0;
    if (iframe<nDark) {
      //cout << iframe << "+"<< nDark <<endl;
      addToPedestal(data);
      return 0;
    }

    newFrame();
    for (int ix=1; ix<nx-1; ix++) {
      for (int iy=1; iy<ny-1; iy++) {
	skip=0;
	max=0;
	tl=0;
	tr=0;
	bl=0;
	br=0;
	tot=0;
	quadTot=0;
	quad=UNDEFINED_QUADRANT;
      
      
      
      
      
	(clusters+nph)->rms=getPedestalRMS(ix,iy);

	//	cout << iframe << " " << nph << " " << ix << " " << iy << endl;
	
      for (int ir=-(clusterSizeY/2); ir<(clusterSizeY/2)+1; ir++) {
	for (int ic=-(clusterSize/2); ic<(clusterSize/2)+1; ic++) {
	  

	  //  if ((iy+ir)>=0 && (iy+ir)<ny && (ix+ic)>=0 && (ix+ic)<nx) {
	    
	    if ((iy>1 && ir<=0)) {
	      ;
	    } else if (ix>1 && ic<=0) {
	      ;
	    } else {
	      val[iy+ir][ix+ic]=subtractPedestal(data,ix+ic,iy+ir);
	      eventMask[iy+ir][ix+ic]=PEDESTAL; 
	    }

	    /* if ((iy+ir)>=iy  && (ix+ic)>=ix) { */
	    /*   val[iy+ir][ix+ic]=subtractPedestal(data,ix+ic,iy+ir); */
	    /*   eventMask[iy+ir][ix+ic]=PEDESTAL;  */
	    /* }  */
	  
	  //   cout << ir << " " << ic << " " << val[iy+ir][ix+ic] << endl;
	    v=&(val[iy+ir][ix+ic]);
	
	    if (ir==0 && ic==0) {
	      if (*v<-nSigma*(clusters+nph)->rms) {
		eventMask[iy][ix]=NEGATIVE_PEDESTAL;
		//	cout << "neg ped" << endl;
	      }
	    } 

	    // if (skip==0) {
	      tot+=*v;
	      if (ir<=0 && ic<=0)
		bl+=*v;
	      if (ir<=0 && ic>=0)
		br+=*v;
	      if (ir>=0 && ic<=0)
		tl+=*v;
	      if (ir>=0 && ic>=0)
		tr+=*v;
	      if (*v>max) {
		max=*v;
	      }
	      // }
	    
	    //  } else skip=1;
	}
      }
	
	if (bl>=br && bl>=tl && bl>=tr) {
	  (clusters+nph)->quad=BOTTOM_LEFT;
	  (clusters+nph)->quadTot=bl;
	  xoff=0;
	  yoff=0;
	} else if (br>=bl && br>=tl && br>=tr) {
	  (clusters+nph)->quad=BOTTOM_RIGHT;
	  (clusters+nph)->quadTot=br;
	  xoff=1;
	  yoff=0;
	} else if (tl>=br && tl>=bl && tl>=tr) {
	  (clusters+nph)->quad=TOP_LEFT;
	  (clusters+nph)->quadTot=tl;
	  xoff=0;
	  yoff=1;
	} else if   (tr>=bl && tr>=tl && tr>=br) {
	  (clusters+nph)->quad=TOP_RIGHT;
	  (clusters+nph)->quadTot=tr;
	  xoff=1;
	  yoff=1;
	} 
	
	if (max>nSigma*(clusters+nph)->rms || tot>sqrt(clusterSizeY*clusterSize)*nSigma*(clusters+nph)->rms || ((clusters+nph)->quadTot)>sqrt(cy*cs)*nSigma*(clusters+nph)->rms) {
	  if (val[iy][ix]>=max) {
	    // cout << "max" << endl;
	    eventMask[iy][ix]=PHOTON_MAX;
	    (clusters+nph)->tot=tot;
	    (clusters+nph)->x=ix;
	    (clusters+nph)->y=iy;
	    (clusters+nph)->ped=getPedestal(ix,iy, 0);
	    //	  cout << iframe << " " << ix  << " " << iy << " "<< (clusters+nph)->tot << " " << (clusters+nph)->quadTot << " " << (clusters+nph)->ped<< " " << (clusters+nph)->rms << endl; 
	    for (int ir=-(clusterSizeY/2); ir<(clusterSizeY/2)+1; ir++) {
	      for (int ic=-(clusterSize/2); ic<(clusterSize/2)+1; ic++) {
		if ((iy+ir)>=0 && (iy+ir)<ny && (ix+ic)>=0 && (ix+ic)<nx) {
		  (clusters+nph)->set_data(val[iy+ir][ix+ic],ic,ir);
		  cout << val[iy+ir][ix+ic] << " " ;
		}
	      }
	      cout << endl << " " ;
	    }
	    
	     cout << endl << " " ;
	    
	  cc[0][0]=(clusters+nph)->get_data(-1+xoff,-1+yoff);
	  cc[1][0]=(clusters+nph)->get_data(-1+xoff,0+yoff);
	  cc[0][1]=(clusters+nph)->get_data(0+xoff,-1+yoff);
	  cc[1][1]=(clusters+nph)->get_data(0+xoff,0+yoff);
	  
	  cout << cc[0][0] << " " << cc[0][1] << endl;
	  cout << cc[1][0] << " " << cc[1][1] << endl;
	  

	  cout << endl << " " ;
	  
	if (interp) {
	  interp->calcEta((clusters+nph)->quadTot,cc,eta_x, eta_y);
	  cout << eta_x << " " << eta_y << endl;
	  cout << "**************************************************************************"<< endl;
	  // cout << "eta" << endl;
	  if (ff) {
	    interp->addToFlatField(eta_x,eta_y);
	  } else {
	    //   cout << "interp" << endl;
	    interp->getInterpolatedPosition(ix,iy,eta_x,eta_y,(clusters+nph)->quad,int_x,int_y);
	    // cout << "add" << endl;
	    interp->addToImage(int_x, int_y);
	  }
	  // cout << "done" << endl;
	}	      
	nph++;
	  } else {
	    //   cout << "ph" << endl;
	    eventMask[iy][ix]=PHOTON;
	  }
      } else if (eventMask[iy][ix]==PEDESTAL) {
	  //  cout << "ped" << endl;
	addToPedestal(data,ix,iy);
      }
	
      }
    }
    
    return  nph;
    
  };

  
  
 protected:
  
  
  slsInterpolation *interp;

};





#endif