// /////////////////////////////////////////////////////////////
// // TpcDigiPar
// //
// // Container class for Digitisation parameters
// // class is inherited from FairParGenericSet
// //
// /////////////////////////////////////////////////////////////

#include "TpcDigiPar.h"
#include <iostream>
#include <fstream>
#include <exception>
#include "TSystem.h"
#include "TpcDigiMapper.h"


ClassImp(TpcDigiPar)

TpcDigiPar::TpcDigiPar(const char* name,const char* title,
			     const char* context)
  : FairParGenericSet(name,title,context),initialized(false),
    fgas(NULL), fgem(NULL), fpadplane(NULL), ffrontend(NULL), fpadshapepool(NULL),
    fpadResponseFile(77),fpadResponseFileName("None"),fClusterErrorFileName("")
{
  std::cout<<"TpcDigiPar::TpcDigiPar"<<std::endl;
  clear();
}

TpcDigiPar::~TpcDigiPar()
{
  if(fgas!=NULL)delete fgas;
  if(fgem!=NULL)delete fgem;
  if(fpadplane!=NULL)delete fpadplane;
  if(fpadshapepool!=NULL)delete fpadshapepool;
  if(ffrontend!=NULL)delete ffrontend;

}


void TpcDigiPar::putParams(FairParamList* list)
{
  std::cout<<"------------------------------------------------"<<std::endl;
  std::cout<<"TpcDigiPar::putParams "<<initialized<<" "<<changed<<std::endl;
  std::cout<<"------------------------------------------------"<<std::endl;
	if(!list) return;
	list->add("TpcGasFile",ftpcGasFile);
	list->add("EField",fEField);
	list->add("AttachFlag",fattach);
	list->add("LogitudinaldiffusionFlag",fdiffuseL);
	list->add("TransversediffusionFlag",fdiffuseT);
	list->add("DriftDistortionFlag",fdistort);
	list->add("zGem",fzGem);
	list->add("zMax",fzMax);
	list->add("Gain",fgain);
	list->add("Supression", fsupression);
	list->add("Spread",fspread);
	list->add("MinSignalAmp",fminSignalAmp);
	list->add("rMin",frmin);
	list->add("rMax",frmax);
	list->add("PadPlaneFile",fpadPlaneFile);
	list->add("PadShapeFile",fpadShapes);
	list->add("PadShapeRange",fshaperange);
	list->add("PadShapeStep",fshapestep);
	list->add("PadShapeIntStep",fintstep);
	list->add("ADCThreshold",fadcthres);
	list->add("ADCMax",fadcmax);
	list->add("ADCBits",fadcbits);
	list->add("SamplingFreq",fsamplefreq);
	list->add("T0_wallclock",ft0);
	list->add("TimeBits",ftimebits);
	list->add("PSAThreshold",fpsathres);
	list->add("Shaper_tDiff",ftdiff);
	list->add("Shaper_tInt",ftint);
	list->add("Shaper_tSig",ftsig);
	list->add("TOTPSA",fpsa);
	list->add("GAUSSIANNOISE",fgaussianNoise);
	list->add("GAUSSIANNOISEAMP",fgaussianNoiseAmp);
	list->add("Rate",frate);
	list->add("ManDriftVel",fManDriftVel);
	list->add("AlignmentFile",fAlignmentFile);
	list->add("SampleThresh",fsamthr);
	list->add("GainCorrFile",fGainCorrFile);
	list->add("ClusterCorrFile",fClustCorrFile);
	list->add("PadResponseFile",fpadResponseFile);
	list->add("ClusterErrorFile",fClusterErrorFile);

}

Bool_t TpcDigiPar::getParams(FairParamList* list)
{
  std::cout<<"------------------------------------------------"<<std::endl;
  std::cout<<"TpcDigiPar::getParams "<<initialized<<std::endl;
  std::cout<<"------------------------------------------------"<<std::endl;
  if(!initialized){
    std::cout<<"TpcDigiPar::getParams"<<std::endl;
    if (!list) {std::cout<<"list not found"<<std::endl; return kFALSE;}
    if(!list->fill("EField",&fEField))
	  {std::cout<<"par: EField not found"<<std::endl; return kFALSE;}
    if(!list->fill("TpcGasFile",&ftpcGasFile))
	  {std::cout<<"par: TpcGasFile not found"<<std::endl; return kFALSE;}
    if(!list->fill("AttachFlag",&fattach))
	  {std::cout<<"par: AttachFlag not found"<<std::endl; return kFALSE;}
    if(!list->fill("LogitudinaldiffusionFlag",&fdiffuseL))
	  {std::cout<<"par: LogitudinaldiffusionFlag not found"<<std::endl; return kFALSE;}
    if(!list->fill("TransversediffusionFlag",&fdiffuseT))
	  {std::cout<<"par: TransversediffusionFlag not found"<<std::endl; return kFALSE;}
    if(!list->fill("DriftDistortionFlag",&fdistort))
	  {std::cout<<"par: DriftDistortionFlag not found"<<std::endl; return kFALSE;}
    if(!list->fill("zGem",&fzGem)){
       std::cout<<"par: zGem not found, also not used"<<std::endl;
    }else{
      std::cerr<<"par: zGem found in parameter file, this parameter is outdated and not used anymore, please remove"
          <<std::endl;
    }
    if(!list->fill("zMax",&fzMax))
	  {std::cout<<"par: zMax not found"<<std::endl; return kFALSE;}
    if(!list->fill("Gain",&fgain))
	  {std::cout<<"par: Gain not found"<<std::endl; return kFALSE;}
    if(!list->fill("Supression",&fsupression))
	  {std::cout<<"par: Supression not found"<<std::endl; return kFALSE;}
    if(!list->fill("Spread",&fspread))
	  {std::cout<<"par: Spread not found"<<std::endl; return kFALSE;}
    if(!list->fill("MinSignalAmp",&fminSignalAmp))
	  {std::cout<<"par: MinSignalAmp not found"<<std::endl; return kFALSE;}
    if(!list->fill("rMin",&frmin))
	  {std::cout<<"par: rMin not found"<<std::endl; return kFALSE;}
    if(!list->fill("rMax",&frmax))
	  {std::cout<<"par: rMax not found"<<std::endl; return kFALSE;}
    if(!list->fill("PadPlaneFile",&fpadPlaneFile))
	  {std::cout<<"par: PadPlaneFile not found"<<std::endl; return kFALSE;}
    if(!list->fill("PadShapeFile",&fpadShapes))
	  {std::cout<<"par: PadShapeFile not found"<<std::endl; return kFALSE;}
    if(!list->fill("PadShapeRange",&fshaperange))
	  {std::cout<<"par: PadShapeRange not found"<<std::endl; return kFALSE;}
    if(!list->fill("PadShapeStep",&fshapestep))
	  {std::cout<<"par: PadShapeStep not found"<<std::endl; return kFALSE;}
    if(!list->fill("PadShapeIntStep",&fintstep))
	  {std::cout<<"par: PadShapeIntStep not found"<<std::endl; return kFALSE;}
    if(!list->fill("ADCThreshold",&fadcthres))
	  {std::cout<<"par: ADCThreshold not found"<<std::endl; return kFALSE;}
    if(!list->fill("ADCMax",&fadcmax))
	  {std::cout<<"par: ADCMax not found"<<std::endl; return kFALSE;}
    if(!list->fill("ADCBits",&fadcbits))
	  {std::cout<<"par: ADCBits not found"<<std::endl; return kFALSE;}
    if(!list->fill("SamplingFreq",&fsamplefreq))
	  {std::cout<<"par: SamplingFreq not found"<<std::endl; return kFALSE;}
    if(!list->fill("T0_wallclock",&ft0))
	  {std::cout<<"par: T0_wallclock not found"<<std::endl; return kFALSE;}
    if(!list->fill("TimeBits",&ftimebits))
	  {std::cout<<"par: TimeBits not found"<<std::endl; return kFALSE;}
    if(!list->fill("PSAThreshold",&fpsathres))
	  {std::cout<<"par: PSAThreshold not found"<<std::endl; return kFALSE;}
    if(!list->fill("Shaper_tDiff",&ftdiff))
	  {std::cout<<"par: Shaper_tDiff not found"<<std::endl; return kFALSE;}
    if(!list->fill("Shaper_tInt",&ftint))
	  {std::cout<<"par: Shaper_tInt not found"<<std::endl; return kFALSE;}
    if(!list->fill("Shaper_tSig",&ftsig))
	  {std::cout<<"par: Shaper_tSig not found"<<std::endl; return kFALSE;}
    if(!list->fill("TOTPSA",&fpsa))
	  {std::cout<<"par: TOTPSA not found"<<std::endl; return kFALSE;}
    if(!list->fill("GAUSSIANNOISE",&fgaussianNoise))
	  {std::cout<<"par: GAUSSIANNOISE not found"<<std::endl; return kFALSE;}
    if(!list->fill("GAUSSIANNOISEAMP",&fgaussianNoiseAmp))
	  {std::cout<<"par: GAUSSIANNOISEAMP not found"<<std::endl; return kFALSE;}
    if(!list->fill("Rate",&frate))
	  {std::cout<<"par: Rate not found"<<std::endl; return kFALSE;}  
    if(!list->fill("ManDriftVel",&fManDriftVel)) {
      std::cout<<std::endl;
      std::cout<<"**************** WARNING IN TPCDIGIPAR **********************"<<std::endl;
      std::cout<<"* WARNING: 'ManDriftVel' not found in parfile               *"<<std::endl;
      std::cout<<"*           SETTING PARAMETER TO MINUS 1!                      *"<<std::endl;
      std::cout<<"*************************************************************"<<std::endl;
      fManDriftVel=-1.;
    }else{
      if(fManDriftVel<=0){
        std::cout<<std::endl;
        std::cout<<"**************** INFO      TPCDIGIPAR **********************"<<std::endl;
        std::cout<<"*          Using drift velocity from GAS file               *"<<std::endl;
        std::cout<<"*                                                           *"<<std::endl;
        std::cout<<"*************************************************************"<<std::endl;
      }else{
        std::cout<<std::endl;
        std::cout<<"**************** INFO      TPCDIGIPAR **********************"<<std::endl;
        std::cout<<"*          Using drift velocity from PAR FILE               *"<<std::endl;
        std::cout<<"*                          "<<fManDriftVel<<"                              *"<<std::endl;
        std::cout<<"*************************************************************"<<std::endl;
      }
    }
    if(!list->fill("SampleThresh",&fsamthr))
	  {
	    std::cout<<"par: SampleThresh not found, setting it to 0"<<std::endl;
	    fsamthr=0.;
	  }
    if(!list->fill("AlignmentFile",&fAlignmentFile))
    {std::cout<<"par: AlignmentFile not found"<<std::endl; return kFALSE;}
    if(!list->fill("GainCorrFile",&fGainCorrFile))
	  {std::cout<<"par: GainCorrFile not found"<<std::endl;
	    fGainCorrFileName="";}
    else
      fGainCorrFileName=readString(fGainCorrFile);
    if(!list->fill("ClusterCorrFile",&fClustCorrFile))
	  {std::cout<<"par: ClusterPosCorrFile not found"<<std::endl;
	    fClustCorrFileName="";}
    else
      fClustCorrFileName=readString(fClustCorrFile);
    //read strings
    ftpcGasFileName=readString(ftpcGasFile);
    fpadPlaneFileName=readString(fpadPlaneFile);
    fpadShapesName=readString(fpadShapes);
    fAlignmentFileName=readString(fAlignmentFile);

	if(!list->fill("PadResponseFile",&fpadResponseFile))
	{
		std::cout<<"par: PadResponseFile not found, using standard padresponse"<<std::endl;
		fpadResponseFileName="None";
	}
	else
	{
		fpadResponseFileName=readString(fpadResponseFile);
	}

	if(!list->fill("ClusterErrorFile",&fClusterErrorFile))
	{
		std::cout<<"par: ClusterErrorFile not found!"<<std::endl;
		return kFALSE;
	}
	else
	{
		fClusterErrorFileName=readString(fClusterErrorFile);
	}

    // update interfaces!
    // TODO: implement caching!
    try{
      if(fgas!=NULL){
        std::cout<<"fgas!=NULL , getting new gas"<<std::endl;
        delete fgas;
        getGas();
      }
      if(fgem!=NULL){
        delete fgem;
        getGem();
      }

      if(fpadplane!=NULL){
        delete fpadplane;
        if(fpadshapepool!=NULL){
          delete fpadshapepool;
          // padshapes should be reloaded by padplane
        }
        getPadPlane();
      }

      if(ffrontend!=NULL){
        delete ffrontend;
        getFrontend();
      }
    }
    catch (std::exception& ex){
      std::cout<<ex.what()<<std::endl;
      throw ex;
    }
    initialized=true;
  }
	return kTRUE;
}

const TpcGas* 
TpcDigiPar::getGas(){
  if(fgas==NULL){
    // load gas file
    std::cerr<<"########################"<<std::endl;
    std::cerr<<"########################"<<std::endl;
    std::cerr<<"########################"<<std::endl;
    std::cerr<<ftpcGasFileName.Data()<<std::endl;
    std::cerr<<"########################"<<std::endl;
    std::cerr<<"########################"<<std::endl;
    std::cerr<<"########################"<<std::endl;
    fgas=new TpcGas(ftpcGasFileName.Data(),fEField);
  }
  return new TpcGas(ftpcGasFileName.Data(),fEField);
}

const TpcGem* 
TpcDigiPar::getGem(){
  if(fgem==NULL){
    fgem=new TpcGem(fgain,fspread);
  }
  return fgem;
}

TpcPadPlane* 
TpcDigiPar::getPadPlane(){
  if(fpadplane==NULL){
    fpadplane=new TpcPadPlane(fpadPlaneFileName.Data(),getPadShapes());
  }
  return fpadplane;
}

const TpcPadShapePool* 
TpcDigiPar::getPadShapes(){
  if(fpadshapepool==NULL){
	  if(fpadResponseFileName!="None")
	  {
		  fpadshapepool=new TpcPadShapePool(fpadShapesName.Data(),fpadResponseFileName.Data());
	  }
	  else
	  {
		  fpadshapepool=new TpcPadShapePool(fpadShapesName.Data(),*getGem(),
				  fshaperange,fshapestep,fintstep);
	  }
  }
  return fpadshapepool;
}

const TpcFrontend* 
TpcDigiPar::getFrontend(){
  if(ffrontend==NULL){
    ffrontend=new TpcFrontend(ftint,ftdiff,ftsig,
			      fadcthres,fadcmax,fadcbits,
			      fsamplefreq,ft0,ftimebits,
			      fpsathres);
  }
  return ffrontend;
}

TString
TpcDigiPar::readString(int i){
  std::cout<<"TpcDigiPar::readString"<<std::endl;
  ifstream infile;
  TString sysFile = gSystem->Getenv("VMCWORKDIR");
  TString parFile = sysFile+"/tpc/parfiles/tpc.files.par";
  infile.open(parFile.Data());
  //  infile.open("tpc/tpc.files.par");
  if(!infile){
    std::cout<<"File " << parFile.Data() << " not found!"<<std::endl;
    throw;
  }
  char buffer[256];
  for(int j=0;j<i+1;++j){
    if(infile.eof()){
      std::cout<<"Did not find string#"<<i<<"in file " << parFile.Data() <<std::endl;
      throw;
    }
    infile.getline(buffer,256);
  }
  infile.close();
  return TString(sysFile+"/"+buffer);
}

double TpcDigiPar::windowMin() {
  //ft0=this->getFrontend()->t0();
  double v0=fManDriftVel;
  if(v0<=0.){
    v0=this->getGas()->VDrift();
  }
  return ft0*v0;
}

double TpcDigiPar::windowMax() {
  //double end = z_from_tick(fmaxSamples);
  int maxSamples=this->getFrontend()->maxSamples();
  double v0=fManDriftVel;
  if(v0<=0.){
    v0=this->getGas()->VDrift();
  }
  double tbin=1./this->getFrontend()->samplingFrequency()*1000; //1/MHz ->ns
  double end= (maxSamples*tbin+ft0)*v0; //should be same as z_from_tick

  if(end>fzMax){
    return fzMax;
  }
  return end;
}