#include "angularConversionStatic.h"

#include <iostream>
#include <fstream>
#include <sstream>
#include <math.h>
#include "angleConversionConstant.h"

#include "sls_detector_defs.h"
#include "angleFunction.h"
using namespace std;

angularConversionStatic::angularConversionStatic()
{
  //angleFunctionPointer=0;
  registerAngleFunctionCallback(&defaultAngleFunction);

}

angularConversionStatic::~angularConversionStatic(){

}



double* angularConversionStatic::convertAngles(double pos, int nch, int *chansPerMod, angleConversionConstant **angOff, int *mF, double fo, double go, int angdir) {
  int imod=0;
  double    *ang=new double[nch];
  double enc=pos;
  angleConversionConstant *p=NULL;
  
  int ch0=0;
  int chlast=chansPerMod[0]-1;
  int nchmod=chansPerMod[0];
  p=angOff[imod];      
  if (mF[imod]==0)
    enc=0;
  else
    enc=pos;

  for (int ip=0; ip<nch; ip++) {
#ifdef VERBOSE
    //  cout << "ip " << ip << " ch0 " << ch0 << " chlast " << chlast << " imod " << imod << endl;
#endif
    if (ip>chlast) {
      imod++; 
      p=angOff[imod];      
      if (mF[imod]==0)
	enc=0;
      else
	enc=pos;
      
      ch0=chlast+1;
      nchmod=chansPerMod[imod];
      if (nchmod>0)
	chlast=ch0+nchmod-1;
    }
    
    if (p)
      ang[ip]=angle(ip-ch0,				\
		    enc,				\
		    fo+go,				\
		    p->r_conversion,			\
		    p->center,				\
		    p->offset,				\
		    p->tilt,				\
		    angdir		  );
  }
  return ang;
}

double angularConversionStatic::convertAngle(double pos, int ich, int *chansPerMod, angleConversionConstant **angOff, int *mF, double fo, double go, int angdir) {

  int imod=0;
  double    ang;
  double enc=0, trans=0;
  angleConversionConstant *p=NULL;
  
  int ch0=0;
  int chlast=chansPerMod[0]-1;
  int nchmod=chansPerMod[0];

  

  while (ich>chlast) {
    imod++;
    ch0=chlast+1;
    nchmod=chansPerMod[imod];
    chlast=ch0+nchmod-1;
  }
  
  p=angOff[imod];
  


  switch (mF[imod]) {
  case 0:
    enc=0;
    trans=0;
    break;
  case 1:
    enc=pos;
    trans=0;
    break;
  case -1:
    enc=-pos;
    trans=0;
    break;
  case 2:
    enc=0;
    trans=pos;
    break;
  case -2:
    enc=0;
    trans=-pos;
    break;
  default:
    enc=0;
    trans=0;
  }
   

    
  if (p)
    ang=angle(ich-ch0,				\
	      enc,					\
	      fo+go,					\
	      p->r_conversion,				\
	      p->center,				\
	      p->offset,				\
	      trans,					\
	      angdir		  );
  
  return ang;
  
}



//static!
int angularConversionStatic::readAngularConversion(string fname, int nmod, angleConversionConstant *angOff) {

  ifstream infile;
  string ss;

#ifdef VERBOSE
  std::cout<< "Opening file "<< fname << std::endl;
#endif
  infile.open(fname.c_str(), ios_base::in);
  if (infile.is_open()) {
    readAngularConversion(infile, nmod, angOff);
    infile.close();
  } else {
    std::cout<< "Could not open calibration file "<< fname << std::endl;
    return -1;
  }
  return 0;
}


//static
int angularConversionStatic::readAngularConversion( ifstream& infile, int nmod, angleConversionConstant *angOff) {
  string str;
  int mod;
  double center, ecenter;
  double r_conv, er_conv;
  double off, eoff;
  string ss;
  int interrupt=0;
  int nm=0;
  //" module %i center %E +- %E conversion %E +- %E offset %f +- %f \n"
  while (infile.good() and interrupt==0) {
    getline(infile,str);
#ifdef VERBOSE
    cout << "** mod " << nm << " " ;
    std::cout<< str << std::endl;
#endif
    istringstream ssstr(str);
    ssstr >> ss >> mod;
    ssstr >> ss >> center;
    ssstr >> ss >> ecenter;
    ssstr >> ss >> r_conv;
    ssstr >> ss >> er_conv;
    ssstr >> ss >> off;
    ssstr >> ss >> eoff;
    if (nm<nmod && nm>=0 ) {
      angOff[nm].center=center;
      angOff[nm].r_conversion=r_conv;
      angOff[nm].offset=off;
      angOff[nm].ecenter=ecenter;
      angOff[nm].er_conversion=er_conv;
      angOff[nm].eoffset=eoff;
    } else
      break;
    //cout << nm<<"  " << angOff[nm].offset << endl;
    nm++;
    if (nm>=nmod)
      break;

    


  }
  return nm;
 }

//static
int angularConversionStatic:: writeAngularConversion(string fname, int nmod, angleConversionConstant *angOff) {

  ofstream outfile;
  outfile.open (fname.c_str(),ios_base::out);
  if (outfile.is_open())
  {  
    writeAngularConversion(outfile, nmod, angOff);
    outfile.close();
  } else {
    std::cout<< "Could not open file " << fname << "for writing"<< std::endl;
    return -1;
  }
  //" module %i center %E +- %E conversion %E +- %E offset %f +- %f \n"
  return 0;
}



//static 
int angularConversionStatic:: writeAngularConversion(ofstream& outfile, int nmod, angleConversionConstant *angOff) {

  for (int imod=0; imod<nmod; imod++) {
      outfile << " module " << imod << " center "<< angOff[imod].center<<"  +- "<< angOff[imod].ecenter<<" conversion "<< angOff[imod].r_conversion << " +- "<< angOff[imod].er_conversion <<  " offset "<< angOff[imod].offset << " +- "<< angOff[imod].eoffset << std::endl;
  }
  return 0;
}


//static
int angularConversionStatic::resetMerging(double *mp, double *mv, double *me, int *mm, int nb) {
  
		
#ifdef VERBOSE
  cout << "creating merging arrays "<<  nb << endl;
#endif
  

  for (int ibin=0; ibin<nb; ibin++) {
    mp[ibin]=0;
    mv[ibin]=0;
    me[ibin]=0;
    mm[ibin]=0;
  }
  return slsDetectorDefs::OK;
}


//static
int angularConversionStatic::finalizeMerging(double *mp, double *mv, double *me, int *mm,int nb) {
   int np=0;
   for (int ibin=0; ibin<nb; ibin++) {
     if (mm[ibin]>0) {
      	
#ifdef VERBOSE 
       cout << "finalize " << ibin << "  "<< mm[ibin] << " " << mp[ibin]<< mv[ibin] << me[ibin] << endl;
#endif
       mp[np]=mp[ibin]/mm[ibin];
       mv[np]=mv[ibin]/mm[ibin];
       me[np]=me[ibin]/mm[ibin];
       me[np]=sqrt(me[ibin]);
       mm[np]=mm[ibin];
       np++;
     }
  }
  return np;
}

//static
int  angularConversionStatic::addToMerging(double *p1, double *v1, double *e1, double *mp, double *mv,double *me, int *mm, int nchans, double binsize,int nbins, int *badChanMask ) {


  double binmi=-180.;
  int ibin=0;

  if (p1==NULL)
    return 0;
  if (v1==NULL)
    return slsDetectorDefs::FAIL;

  if (mp==NULL) //can be changed if we want to use a fixed bin algorithm!
    return slsDetectorDefs::FAIL;

  if (mv==NULL)
    return slsDetectorDefs::FAIL;
  if (me==NULL)
    return slsDetectorDefs::FAIL;
  if (mm==NULL)
    return slsDetectorDefs::FAIL;
  if (nchans==0)
    return slsDetectorDefs::FAIL;
  
  if (binsize<=0)
    return slsDetectorDefs::FAIL;

  if (nbins<=0)
    return slsDetectorDefs::FAIL;
  
  for (int ip=0; ip<nchans; ip++) {
    if (badChanMask) {
      if (badChanMask[ip]) {
#ifdef VERBOSE
	cout << "channel " << ip << " is bad " << endl;
#endif
	  continue;
      }
    }
    ibin=(int)((p1[ip]-binmi)/binsize);
   
 
    if (ibin<nbins && ibin>=0) {
      mp[ibin]+=p1[ip];
      mv[ibin]+=v1[ip];
      if (e1)
	me[ibin]+=(e1[ip]*e1[ip]);
      else
	me[ibin]+=v1[ip];
      mm[ibin]++;

#ifdef VERBOSE
      cout << "add " << ibin << "  "<< mm[ibin] << " " << mp[ibin]<< mv[ibin] << me[ibin] << endl;
#endif
    } else
      return slsDetectorDefs::FAIL;
  }
  

  return slsDetectorDefs::OK;
  
}

int  angularConversionStatic::addPointToMerging(double p1, double v1, double e1, double *mp, double *mv,double *me, int *mm,  double binsize,int nbins) {


  double binmi=-180.;
  int ibin=0;


  if (mp==NULL) //can be changed if we want to use a fixed bin algorithm!
    return slsDetectorDefs::FAIL;

  if (mv==NULL)
    return slsDetectorDefs::FAIL;
  if (me==NULL)
    return slsDetectorDefs::FAIL;
  if (mm==NULL)
    return slsDetectorDefs::FAIL;
  
  if (binsize<=0)
    return slsDetectorDefs::FAIL;

  if (nbins<=0)
    return slsDetectorDefs::FAIL;
  

    ibin=(int)((p1-binmi)/binsize);
   
 
    if (ibin<nbins && ibin>=0) {
      mp[ibin]+=p1;
      mv[ibin]+=v1;
      if (e1)
	me[ibin]+=(e1*e1);
      else
	me[ibin]+=v1;
      mm[ibin]++;

#ifdef VERBOSE
      cout << "add " << ibin << "  "<< mm[ibin] << " " << mp[ibin]<< mv[ibin] << me[ibin] << endl;
#endif
    }  else
      return slsDetectorDefs::FAIL;
  

  return slsDetectorDefs::OK;
  
}