/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

/* $Id$ */
/* History of cvs commits:
 *
 * $Log$ */

//*-- Author: Marco van Leeuwen (LBL)
#include "AliEMCALRawUtils.h"

#include "TF1.h"
#include "TGraph.h"
#include "TSystem.h"

#include "AliLog.h"
#include "AliRunLoader.h"
#include "AliCaloAltroMapping.h"
#include "AliAltroBuffer.h"
#include "AliRawReader.h"
#include "AliCaloRawStream.h"
#include "AliDAQ.h"

#include "AliEMCALLoader.h"
#include "AliEMCALGeometry.h"
#include "AliEMCALDigitizer.h"
#include "AliEMCALDigit.h"


ClassImp(AliEMCALRawUtils)

// Signal shape parameters
Int_t    AliEMCALRawUtils::fgOrder       = 2 ;      // Order of gamma function 
Double_t AliEMCALRawUtils::fgTimeMax     = 2.56E-5 ;  // each sample is over 100 ns fTimeMax/fTimeBins
Double_t AliEMCALRawUtils::fgTau         = 165E-9 ;   // 165 ns (from testbeam; not very accurate)
Double_t AliEMCALRawUtils::fgTimeTrigger = 1.5E-6 ;   // 15 time bins ~ 1.5 musec

// some digitization constants
Int_t    AliEMCALRawUtils::fgThreshold = 1;
Int_t    AliEMCALRawUtils::fgDDLPerSuperModule = 2;  // 2 ddls per SuperModule

AliEMCALRawUtils::AliEMCALRawUtils(): fHighLowGainFactor(0.) {
  fHighLowGainFactor = 16. ;          // adjusted for a low gain range of 82 GeV (10 bits) 
}
//____________________________________________________________________________
AliEMCALRawUtils::~AliEMCALRawUtils() {
}
//____________________________________________________________________________
void AliEMCALRawUtils::Digits2Raw()
{
  // convert digits of the current event to raw data
  
  AliRunLoader *rl = AliRunLoader::GetRunLoader();
  AliEMCALLoader *loader = dynamic_cast<AliEMCALLoader*>(rl->GetDetectorLoader("EMCAL"));

  // get the digits
  loader->LoadDigits("EMCAL");
  loader->GetEvent();
  TClonesArray* digits = loader->Digits() ;
  
  if (!digits) {
    Warning("Digits2Raw", "no digits found !");
    return;
  }
    
  // get the geometry
  AliEMCALGeometry* geom = AliEMCALGeometry::GetInstance();
  if (!geom) {
    AliError(Form("No geometry found !"));
    return;
  }
  
  static const Int_t nDDL = 12*2; // 12 SM hardcoded for now. Buffers allocated dynamically, when needed, so just need an upper limit here
  AliAltroBuffer* buffers[nDDL];
  for (Int_t i=0; i < nDDL; i++)
    buffers[i] = 0;

  Int_t adcValuesLow[fgkTimeBins];
  Int_t adcValuesHigh[fgkTimeBins];

  //Load Mapping RCU files once
  TString path = gSystem->Getenv("ALICE_ROOT");
  path += "/EMCAL/mapping/RCU";
  TString path0 = path+"0.data";//This file will change in future
  TString path1 = path+"1.data";//This file will change in future
  AliAltroMapping * mapping[2] ; // For the moment only 2
  mapping[0] = new AliCaloAltroMapping(path0.Data());
  mapping[1] = new AliCaloAltroMapping(path1.Data());

  // loop over digits (assume ordered digits)
  for (Int_t iDigit = 0; iDigit < digits->GetEntries(); iDigit++) {
    AliEMCALDigit* digit = dynamic_cast<AliEMCALDigit *>(digits->At(iDigit)) ;
    if (digit->GetAmp() < fgThreshold) 
      continue;

    //get cell indices
    Int_t nSM = 0;
    Int_t nIphi = 0;
    Int_t nIeta = 0;
    Int_t iphi = 0;
    Int_t ieta = 0;
    Int_t nModule = 0;
    geom->GetCellIndex(digit->GetId(), nSM, nModule, nIphi, nIeta);
    geom->GetCellPhiEtaIndexInSModule(nSM, nModule, nIphi, nIeta,iphi, ieta) ;
    
    //Check which is the RCU of the cell.
    Int_t iRCU = -111;
    //RCU0
    if (0<=iphi&&iphi<8) iRCU=0; // first cable row
    else if (8<=iphi&&iphi<16 && 0<=ieta&&ieta<24) iRCU=0; // first half; 
    //second cable row
    //RCU1
    else if(8<=iphi&&iphi<16 && 24<=ieta&&ieta<48) iRCU=1; // second half; 
    //second cable row
    else if(16<=iphi&&iphi<24) iRCU=1; // third cable row
    
    //Which DDL?
    Int_t iDDL = fgDDLPerSuperModule* nSM + iRCU;
    if (iDDL >= nDDL)
      Fatal("Digits2Raw()","Non-existent DDL board number: %d", iDDL);

    if (buffers[iDDL] == 0) {      
      // open new file and write dummy header
      TString fileName = AliDAQ::DdlFileName("EMCAL",iDDL);
      buffers[iDDL] = new AliAltroBuffer(fileName.Data(),mapping[iRCU]);
      buffers[iDDL]->WriteDataHeader(kTRUE, kFALSE);  //Dummy;
    }
    
    // out of time range signal (?)
    if (digit->GetTimeR() > GetRawFormatTimeMax() ) {
      AliInfo("Signal is out of time range.\n");
      buffers[iDDL]->FillBuffer((Int_t)digit->GetAmp());
      buffers[iDDL]->FillBuffer(GetRawFormatTimeBins() );  // time bin
      buffers[iDDL]->FillBuffer(3);          // bunch length      
      buffers[iDDL]->WriteTrailer(3, ieta, iphi, nSM);  // trailer
      // calculate the time response function
    } else {
      Bool_t lowgain = RawSampledResponse(digit->GetTimeR(), digit->GetAmp(), adcValuesHigh, adcValuesLow) ; 
      if (lowgain) 
	buffers[iDDL]->WriteChannel(ieta, iphi, 0, GetRawFormatTimeBins(), adcValuesLow, fgThreshold);
      else 
	buffers[iDDL]->WriteChannel(ieta,iphi, 1, GetRawFormatTimeBins(), adcValuesHigh, fgThreshold);
    }
  }
  
  // write headers and close files
  for (Int_t i=0; i < nDDL; i++) {
    if (buffers[i]) {
      buffers[i]->Flush();
      buffers[i]->WriteDataHeader(kFALSE, kFALSE);
      delete buffers[i];
    }
  }
  mapping[0]->Delete();
  mapping[1]->Delete();
  loader->UnloadDigits();
}

//____________________________________________________________________________
void AliEMCALRawUtils::Raw2Digits(AliRawReader* reader)
{
  // convert raw data of the current event to digits
  AliEMCALGeometry * geom = AliEMCALGeometry::GetInstance();
  if (!geom) {
    AliError(Form("No geometry found !"));
    return;
  }

  AliRunLoader *rl = AliRunLoader::GetRunLoader();
  AliEMCALLoader *loader = dynamic_cast<AliEMCALLoader*>(rl->GetDetectorLoader("EMCAL"));

  // get the digits
  TClonesArray* digits = loader->Digits() ;
  digits->Clear(); 

  if (!digits) {
    Error("Raw2Digits", "no digits found !");
    return;
  }
  if (!reader) {
    Error("Raw2Digits", "no raw reader found !");
    return;
  }

  // Use AliAltroRawStream to read the ALTRO format.  No need to
  // reinvent the wheel :-) 
  AliCaloRawStream in(reader,"EMCAL");
  // Select EMCAL DDL's;
  reader->Select("EMCAL");

  // reading is from previously existing AliEMCALGetter.cxx
  // ReadRaw method
  TF1 * signalF = new TF1("signal", RawResponseFunction, 0, GetRawFormatTimeMax(), 4);
  signalF->SetParNames("Charge", "Gain", "Amplitude", "TimeZero"); 
  
  Int_t id =  -1;
  Int_t idigit = 0 ; 
  Double_t time = 0. ; 
  Double_t amp = 0. ; 

  TGraph * gSig = new TGraph(GetRawFormatTimeBins()) ; 

  Int_t eofReached = 0;
  Int_t lowGain = 0;

  in.Next(); // Go to first digit
  do {
    id =  geom->GetAbsCellIdFromCellIndexes(in.GetModule(), in.GetRow(), in.GetColumn()) ;
    lowGain = in.IsLowGain();
    gSig->SetPoint(in.GetTime(), 
		   in.GetTime()* GetRawFormatTimeMax() / GetRawFormatTimeBins(), 
		   in.GetSignal()) ;

    Int_t iTime = 1;
    do {
      if (!in.Next())
	eofReached = 1;
      else {
	gSig->SetPoint(in.GetTime(), 
		       in.GetTime()* GetRawFormatTimeMax() / GetRawFormatTimeBins(), 
		       in.GetSignal()) ;
      }
      iTime++;
    } while (!eofReached && !in.IsNewRow() && !in.IsNewColumn() && !in.IsNewModule());

    FitRaw(gSig, signalF, amp, time) ; 
    if (lowGain) 
      amp *= fHighLowGainFactor; 
    
    if (amp > 0) {
      AliDebug(2,Form("id %d amp %g", id, amp));
      new((*digits)[idigit]) AliEMCALDigit( -1, -1, id, (Int_t)amp, time, idigit) ;	
	  idigit++ ; 
    }
    Int_t index ; 
	
    // Reset graph
    for (index = 0; index < GetRawFormatTimeBins(); index++) {
      gSig->SetPoint(index, index * GetRawFormatTimeMax() / GetRawFormatTimeBins(), 0) ;  
    } 
  } while (!eofReached); // EMCAL entries loop
  
  delete signalF ; 
  delete gSig;
  
  return ; 
}

//____________________________________________________________________________ 
void AliEMCALRawUtils::FitRaw(TGraph * gSig, TF1* signalF, Double_t & amp, Double_t & time)
{
  // Fits the raw signal time distribution; from AliEMCALGetter 

  const Int_t kNoiseThreshold = 0 ;
  Double_t timezero1 = 0., timezero2 = 0., timemax = 0. ;
  Double_t signal = 0., signalmax = 0. ;       
  amp = time = 0. ; 

  timezero1 = signalmax = timemax = 0. ;
  timezero2 = GetRawFormatTimeMax();
  Int_t index ; 
  for (index = 0; index < GetRawFormatTimeBins(); index++) {
    gSig->GetPoint(index, time, signal) ; 
    if (signal > kNoiseThreshold && timezero1 == 0.) 
      timezero1 = time ;
    if (signal <= kNoiseThreshold && timezero1 > 0. && timezero2 == 0.)
      timezero2 = time ; 
    if (signal > signalmax) {
      signalmax = signal ; 
      timemax   = time ; 
    }
  }

  if ( timemax < GetRawFormatTimeMax() * 0.4 ) { // else its noise 
    signalF->SetParameter(0, signalmax) ; 
    signalF->SetParameter(1, timemax) ; 
    gSig->Fit(signalF, "QRON", "", 0., timezero2); //, "QRON") ; 
    amp = signalF->GetParameter(0) ; 
    time = signalF->GetParameter(1) - fgTimeTrigger;
  }
  return;
}
//__________________________________________________________________
Double_t AliEMCALRawUtils::RawResponseFunction(Double_t *x, Double_t *par)
{
  // Shape of the electronics raw reponse:
  // It is a semi-gaussian, 2nd order Gamma function of the general form
  //
  // t' = (t - t0 + tau) / tau
  // F = A * t**N * exp( N * ( 1 - t) )   for t >= 0
  // F = 0                                for t < 0 
  //
  // parameters:
  // A:   par[0]   // Amplitude = peak value
  // t0:  par[1]
  // tau: fgTau
  // N:   fgOrder
  //
  Double_t signal ;
  Double_t xx = ( x[0] - par[1] + fgTau ) / fgTau ; 

  if (xx < 0) 
    signal = 0. ;  
  else {  
    signal = par[0] * TMath::Power(xx , fgOrder) * TMath::Exp(fgOrder * (1 - xx )) ; 
  }
  return signal ;  
}

//__________________________________________________________________
Bool_t AliEMCALRawUtils::RawSampledResponse(
const Double_t dtime, const Double_t damp, Int_t * adcH, Int_t * adcL) const 
{
  // for a start time dtime and an amplitude damp given by digit, 
  // calculates the raw sampled response AliEMCAL::RawResponseFunction

  const Int_t kRawSignalOverflow = 0x3FF ; 
  Bool_t lowGain = kFALSE ; 

  TF1 signalF("signal", RawResponseFunction, 0, GetRawFormatTimeMax(), 4);
  signalF.SetParameter(0, damp) ; 
  signalF.SetParameter(1, dtime + fgTimeTrigger) ; 

  for (Int_t iTime = 0; iTime < GetRawFormatTimeBins(); iTime++) {
    Double_t time = iTime * GetRawFormatTimeMax() / GetRawFormatTimeBins() ;
    Double_t signal = signalF.Eval(time) ;     
    adcH[iTime] =  static_cast<Int_t>(signal + 0.5) ;
    if ( adcH[iTime] > kRawSignalOverflow ){  // larger than 10 bits 
      adcH[iTime] = kRawSignalOverflow ;
      lowGain = kTRUE ; 
    }

    signal /= fHighLowGainFactor;

    adcL[iTime] =  static_cast<Int_t>(signal + 0.5) ;
    if ( adcL[iTime] > kRawSignalOverflow)  // larger than 10 bits 
      adcL[iTime] = kRawSignalOverflow ;
  }
  return lowGain ; 
}