/**************************************************************************
 * 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$ */

#include <Riostream.h>

#include <TDirectory.h>

#include "AliRunLoader.h"
#include "AliRun.h"
#include <AliESD.h>
#include "AliLog.h"
#include <TClonesArray.h>
#include "AliT0RecPoint.h"
#include "AliT0.h"
#include "AliRawReader.h"
#include "AliT0RawReader.h"
#include "AliT0Loader.h"
#include "AliT0digit.h"
#include "AliT0Reconstructor.h"
#include "AliT0Parameters.h"
#include "AliCDBLocal.h"
#include "AliCDBStorage.h"
#include "AliCDBManager.h"
#include "AliCDBEntry.h"

#include <TArrayI.h>
#include <TGraph.h>

ClassImp(AliT0Reconstructor)

//____________________________________________________________________
void
AliT0Reconstructor::Init(AliRunLoader* runLoader,TTree* digitsTree ) const
{

 // Initialize the reconstructor
  AliDebug(2, Form("Init called with runloader 0x%x", runLoader));
 
  // Initialize the parameters
   AliT0Loader* pStartLoader = (AliT0Loader*) runLoader->GetLoader("T0Loader");
 
  pStartLoader->LoadDigits("READ");

  digitsTree = pStartLoader->TreeD();

}

//____________________________________________________________________
  void  AliT0Reconstructor::ConvertDigits(AliRawReader* rawReader, TTree* digitsTree) const
{
  //T0 raw data-> digits conversion
 // reconstruct time information from raw data
    AliT0 *baseT0;
    baseT0->Raw2Digits(rawReader, digitsTree);
  

}
void AliT0Reconstructor::Reconstruct(TTree*digitsTree, TTree*clustersTree) const
  //void AliT0Reconstructor::Reconstruct(AliRunLoader *runLoader) const
{
// T0 digits reconstruction
// T0RecPoint writing 

    //Q->T-> coefficients !!!! should be asked!!!
  //  Float_t ph2MIP=500;
  Float_t gain[24], timeDelayCFD[24], timeDelayLED[24];
  Float_t zdetA,zdetC;
  TObjArray slewingLED;
    
  TArrayI * fADC = new TArrayI(24); 
  TArrayI * fTimeCFD = new TArrayI(24); 
  TArrayI * fADCLED = new TArrayI(24); 
  TArrayI * fTimeLED = new TArrayI(24); 

  AliT0Parameters* param = AliT0Parameters::Instance();
  param->Init();

  Int_t mV2Mip = param->GetmV2Mip();     
  //mV2Mip = param->GetmV2Mip();     
  Int_t channelWidth = param->GetChannelWidth() ;  
  
  for (Int_t i=0; i<24; i++){
    timeDelayCFD[i] = param->GetTimeDelayCFD(i);
    timeDelayLED[i] = param->GetTimeDelayLED(i);
    gain[i] = param->GetGain(i);
    //gain[i] = 1;
    slewingLED.AddAtAndExpand(param->GetSlew(i),i);
  }
  zdetC = param->GetZposition(0);
  zdetA  = param->GetZposition(1);
    
  AliDebug(1,Form("Start DIGITS reconstruction "));

  TBranch *brDigits=digitsTree->GetBranch("T0");
  AliT0digit *fDigits = new AliT0digit() ;
  if (brDigits) {
    brDigits->SetAddress(&fDigits);
  }else{
    cerr<<"EXEC Branch T0 digits not found"<<endl;
    return;
  }
  
   digitsTree->GetEvent(0);
   digitsTree->GetEntry(0);
   brDigits->GetEntry(0);
  fDigits->GetTime(*fTimeCFD);
  fDigits->GetADC(*fADC);
  fDigits->GetTimeAmp(*fTimeLED);
  fDigits->GetADCAmp(*fADCLED);


  Float_t besttimeright=999999;
  Float_t besttimeleft=999999;
  Int_t pmtBestRight=99999;
  Int_t pmtBestLeft=99999;
  Float_t timeDiff=999999, meanTime=0;
  


   AliT0RecPoint* frecpoints= new AliT0RecPoint ();
   clustersTree->Branch( "T0", "AliT0RecPoint" ,&frecpoints, 405,1);

  Float_t time[24], adc[24];
  for (Int_t ipmt=0; ipmt<24; ipmt++) {
    if(fTimeCFD->At(ipmt)>0 ){
      time[ipmt] = channelWidth *( fTimeCFD->At(ipmt)) - 1000*timeDelayCFD[ipmt];
      //   Float_t adc_digPs = channelWidth * Float_t (fADCLED->At(ipmt)) ;
      //      adc[ipmt] = TMath::Exp(adc_digPs/1000) /gain[ipmt];
      adc[ipmt]=1;
      AliDebug(1,Form(" time %f ps,  adc %f mv in MIP %i\n ",
		      time[ipmt], adc[ipmt], Int_t (adc[ipmt]/mV2Mip +0.5)));
      frecpoints->SetTime(ipmt,time[ipmt]);
      frecpoints->SetAmp(ipmt,adc[ipmt]);
    }
    else {
      time[ipmt] = 0;
      adc[ipmt] = 0;
    }
  }

  for (Int_t ipmt=0; ipmt<12; ipmt++){
    if(time[ipmt] > 1 ) {
      if(time[ipmt]<besttimeleft){
	besttimeleft=time[ipmt]; //timeleft
	pmtBestLeft=ipmt;
      }
    }
  }
  for ( Int_t ipmt=12; ipmt<24; ipmt++){
    if(time[ipmt] > 1) {
      if(time[ipmt]<besttimeright) {
	besttimeright=time[ipmt]; //timeright
        pmtBestRight=ipmt;}
    }
  }
  if(besttimeright !=999999)  frecpoints->SetTimeBestRight(Int_t(besttimeright));
  if( besttimeleft != 999999 ) frecpoints->SetTimeBestLeft(Int_t(besttimeleft));
  AliDebug(1,Form(" besttimeA %f ps,  besttimeC %f ps",besttimeright, besttimeleft));
  Float_t c = 0.0299792; // cm/ps
  Float_t vertex = 0;
  if(besttimeright !=999999 && besttimeleft != 999999 ){
    timeDiff = besttimeleft - besttimeright;
    meanTime = (besttimeright + besttimeleft)/2.;
    vertex = c*(timeDiff)/2.; //-(lenr-lenl))/2;
    AliDebug(1,Form("  timeDiff %f ps,  meanTime %f ps, vertex %f cm",timeDiff, meanTime,vertex ));
    frecpoints->SetVertex(vertex);
    frecpoints->SetMeanTime(Int_t(meanTime));
    
  }
  clustersTree->Fill();
}


void AliT0Reconstructor::FillESD(AliRunLoader* runLoader, AliESD *pESD) const
{

  /***************************************************
  Resonstruct digits to vertex position
  ****************************************************/
  
  
  if (!runLoader) {
    AliError("Reconstruct >> No run loader");
    return;
  }
  
  AliDebug(1,Form("Start FillESD T0"));

  AliT0Loader* pStartLoader = (AliT0Loader*) runLoader->GetLoader("T0Loader");
 
  pStartLoader->LoadRecPoints("READ");

  TTree *treeR = pStartLoader->TreeR();
  
   AliT0RecPoint* frecpoints= new AliT0RecPoint ();
    if (!frecpoints) {
    AliError("Reconstruct Fill ESD >> no recpoints found");
    return;
  }
  
  AliDebug(1,Form("Start FillESD T0"));
  TBranch *brRec = treeR->GetBranch("T0");
  if (brRec) {
    brRec->SetAddress(&frecpoints);
  }else{
    cerr<<"EXEC Branch T0 rec not found"<<endl;
    // exit(111);
    return;
  } 
    
    brRec->GetEntry(0);
    Float_t timeStart, Zposition, amp[24], time[24];
    Int_t i;
    Zposition = frecpoints -> GetVertex();
    timeStart = frecpoints -> GetMeanTime();
    for ( i=0; i<24; i++) {
       time[i] = Float_t (frecpoints -> GetTime(i)) / 1000.; // ps to ns
      amp[i] = frecpoints -> GetAmp(i);
    }
    pESD->SetT0zVertex(Zposition); //vertex Z position 
    pESD->SetT0(timeStart);        // interaction time 
    pESD->SetT0time(time);         // best TOF on each PMT 
    pESD->SetT0amplitude(amp);     // number of particles(MIPs) on each PMT
    cout<<" ESD >> "<<Zposition<<" "<<timeStart<<endl;

    pStartLoader->UnloadRecPoints();
   
} // vertex in 3 sigma