#include "TChain.h"
#include "TTree.h"
#include "TList.h"
#include "TH1F.h"
#include "TH2F.h"
#include "TCanvas.h"
#include "TMath.h"
#include "TObjString.h"

#include "AliAnalysisTask.h"
#include "AliPhysicsSelection.h"
#include "AliAnalysisManager.h"

#include "AliESDEvent.h"
#include "AliESDfriend.h"
#include "AliESDInputHandler.h"
#include "AliESDTZEROfriend.h"
#include "AliT0HIanalysisTask.h"
#include "AliESDVZERO.h"
#include "AliMultiplicity.h"
#include "AliTriggerAnalysis.h"
#include "AliESDpid.h"
#include "AliESDtrackCuts.h"
#include "AliCentrality.h"
#include "AliVEvent.h"

ClassImp(AliT0HIanalysisTask)

//________________________________________________________________________
AliT0HIanalysisTask::AliT0HIanalysisTask(const char *name) 
  : AliAnalysisTaskSE(name), fESD(0), fOutputList(0), 
  fT0OutTree(0x0)
					//,    fESDpid(new AliESDpid())
{
  // Constructor

  // Define input and output slots here
  // Input slot #0 works with a TChain
  DefineInput(0, TChain::Class());
  // Output slot #0 id reserved by the base class for AOD
  // Output slot #1 writes into a TH1 container
  DefineOutput(1, TList::Class());
}

//________________________________________________________________________
AliT0HIanalysisTask::~AliT0HIanalysisTask() 
{
// Destructor
   if (fT0OutTree) delete fT0OutTree ;
}

//_____________________________________________________________________________
Bool_t AliT0HIanalysisTask::UserNotify()
{
//
// Calls the mother class Notify()
//

//  AliDebug(AliLog::kDebug,"<-");
  // AliDebug(AliLog::kDebug,"->");

  return AliAnalysisTaskSE::UserNotify();
}

//________________________________________________________________________
void AliT0HIanalysisTask::UserCreateOutputObjects()
{
  // Create histograms
  // Called once
   fOutputList = new TList();
  fOutputList->SetOwner(); // Will delete the histos on cleanup

  fT0OutTree=new TTree("t0tree","None here");
  fT0OutTree->Branch("fNevent", &fEvent);
  fT0OutTree->Branch("fOrbit", &fOrbit);
  fT0OutTree->Branch("fBC", &fBC);
  fT0OutTree->Branch("fNcont", &fNcont, "fNcont/I");
  fT0OutTree->Branch("fNcontTPC", &fNcontTPC);
  fT0OutTree->Branch("fVertexTPC", &fVertexTPC);
  fT0OutTree->Branch("fVertexPrim", &fVertexPrim);
  fT0OutTree->Branch("vertexT0", &fVertex, "fVertex/F");
  fT0OutTree->Branch("vertexSPD", &fVertexSPD, "vertexSPD/F");
  fT0OutTree->Branch("meanAC", &fMeanAC,"meanAC/F");
  fT0OutTree->Branch("meanA", &fMeanA,"meanA/F");
  fT0OutTree->Branch("meanC", &fMeanC,"meanC/F");
  fT0OutTree->Branch("trackletSPD", &fTrackletSPD,"trackletSPD/I");
  fT0OutTree->Branch("clustersSPD", &fClustersSPD,"clustersSPD/I");
  fT0OutTree->Branch("TOFtracks", &fTOFtracks);
  fT0OutTree->Branch("ESDtracks", &fESDtracks);
  fT0OutTree->Branch("multV0A", &fMultV0A,"multV0A/F");
  fT0OutTree->Branch("multV0C", &fMultV0C,"multV0C/F");
  fT0OutTree->Branch("timeV0A", &fTimeV0A,"timeV0A/F");
  fT0OutTree->Branch("timeV0C", &fTimeV0C,"timeV0C/F");
  fT0OutTree->Branch("sumampA", &fSumampA);
  fT0OutTree->Branch("sumampC", &fSumampC);
  fT0OutTree->Branch("pileup", &fpileup);
  fT0OutTree->Branch("trigger", &fTrigger);
  fT0OutTree->Branch("triggerinput", &fTriggerinput);
  fT0OutTree->Branch("T0trigger", &fT0Trigger);
  fT0OutTree->Branch("t0tofTrack",  &t0tofTrack);
  fT0OutTree->Branch("ZDCcut", &fZDCcut);
  fT0OutTree->Branch("ftimestamp", &ftimestamp);
  fT0OutTree->Branch("centralityV0M", &fcentralityV0M);
  fT0OutTree->Branch("centralityZDC", &fcentralityZDC);
  fT0OutTree->Branch("centralityTRK", &fcentralityTRK);
  fT0OutTree->Branch("centralityCLA", &fcentralityCLA);
  for ( Int_t i=0; i<24; i++) {
    fT0OutTree->Branch(Form("amp%i", i+1), &famp[i]);
    fT0OutTree->Branch(Form("time%i", i+1), &ftime[i]);
  } 
  for ( Int_t i=0; i<5; i++) {
    fT0OutTree->Branch(Form("fOrA%i", i+1), &fOrA[i]);
    fT0OutTree->Branch(Form("fOrC%i", i+1), &fOrC[i]);
    fT0OutTree->Branch(Form("fTVDC%i", i+1), &fTVDC[i]);  
    for ( Int_t ii=0; ii<24; ii++) 
      fT0OutTree->Branch(Form("fRawTime%i_%i", ii+1, i+1), &fRawTime[ii][i]);
  }
  TString pilename[3] = {"T0pileup", "T0background", "T0satellite"};
  for ( Int_t i=0; i<3; i++) 
    fT0OutTree->Branch(pilename[i].Data(), &fT0pileup[i]);
  fT0OutTree->Branch("meanACbest", &fMeanACcalc,"meanACcalc/F");
  fT0OutTree->Branch("meanAbest", &fMeanAcalc,"meanAcalc/F");
  fT0OutTree->Branch("meanCbest", &fMeanCcalc,"meanCcalc/F");
  fT0OutTree->Branch("multEstimator", &fMultiplicity);


  fOutputList->Add(fT0OutTree);
 
   PostData(1, fOutputList);
  
}

//________________________________________________________________________
void AliT0HIanalysisTask::UserExec(Option_t *) 
{
  // Main loop
  // Called for each event
  /*   
  AliAnalysisManager* anMan = AliAnalysisManager::GetAnalysisManager();
  AliESDInputHandler *handler =
    (AliESDInputHandler*)anMan->GetInputEventHandler();
  UInt_t selFlag = handler->IsEventSelected();
  Bool_t fIsSelected = selFlag & AliVEvent::kCINT7 ;  
  cout<< selFlag << " SelFlag "
      << (selFlag&AliVEvent::kCINT5) <<" CINT5 "
      << (selFlag&AliVEvent::kINT7)  << " kCINT7 "
      << (selFlag&AliVEvent::kINT8)  << " kINT8"<<endl;
  */
  // printf (" @@@@@@@@@@@@@@@AliT0HIanalysisTask::UserExec() \n");
   fVertex= fVertexSPD =  fMeanAC = fMeanA =  fMeanC =  fTrackletSPD=-99999;
  fMeanACcalc = fMeanAcalc =  fMeanCcalc = fNcont = fNcontTPC = -99999;
  fMultV0A=fMultV0C=fTimeV0A=fTimeV0C=fMultiplicity = -99999;
  
  for (Int_t i=0; i<24; i++) {
    famp[i] = ftime[i] = -9999;
    for ( Int_t i0=0; i0<5; i0++) {
      fRawTime[i][i0] = -9999;
      if(i0==0) { 
	fTVDC[i0]=-9999;
	fOrC[i0]=-9999;
	fOrA[i0]=-9999;
      }
    }
  }
  Float_t orA , orC, tvdc;
  orA = orC = tvdc = -9999;
  fBC = -9999;
  // Post output data.
  fESD = dynamic_cast<AliESDEvent*>(InputEvent());
  if (!fESD) {
    printf("ERROR: fESD not available\n");
    return;
  }

  if(fESD)
    {
      //    if(fIsSelected) 
	{
	  //  AliESDHeader* header = fESD-> GetHeader();
	AliESDVZERO* esdV0 = fESD->GetVZEROData();
	const AliMultiplicity *mult = fESD->GetMultiplicity();
	// printf(" VZERO MULT \n");
	AliESDTZERO* tz= (AliESDTZERO*) fESD->GetESDTZERO();
	//   printf(" TZERO \n");
	const Double32_t *amp, *time, *mean ;
	const Double32_t *meanbest;
	fVertex= fVertexSPD =  fMeanAC = fMeanA =  fMeanC =  fTrackletSPD=-99999;
	fMeanACcalc = fMeanAcalc =  fMeanCcalc = fNcont = fNcontTPC -99999;
	fMultV0A=fMultV0C=fTimeV0A=fTimeV0C=fMultiplicity = -99999;
	
	fSumampA=0;
	fSumampC=0;
	fNcont=0;
	  Float_t shift=0;
	  
	  TString triggers = fESD->GetFiredTriggerClasses();
	  fTrigger.SetString(triggers.Data());
	  //	  if ( !triggers.Contains("CINT7-B") ) 
	  //  cout<<triggers<<endl;
	  TString inputtriggers =fESD-> GetHeader()->GetFiredTriggerInputs();
	  fTriggerinput.SetString(inputtriggers.Data());
	  //	  cout<<inputtriggers<<endl;
	  
	  fEvent=fESD->GetEventNumberInFile();
	  fT0Trigger=fESD->GetT0Trig();

	  fESDtracks=fESD->GetNumberOfTracks();
	  AliTriggerAnalysis *trigAna = new AliTriggerAnalysis;
	  ftimestamp=fESD->GetTimeStamp(); // - 1301217882; 
	  //	  printf (" timestamp %d \n", ftimestamp);
	  
	  //  fMultiplicity =  AliESDtrackCuts::GetReferenceMultiplicity(fESD);
	  // cout<<" fMultiplicity "<<fMultiplicity<<endl;
	  fOrbit=fESD->GetOrbitNumber();
	  fBC=fESD->GetBunchCrossNumber();
	  fNcont = fESD->GetPrimaryVertexSPD()->GetNContributors();
	  fNcontTPC = fESD->GetPrimaryVertexTPC()->GetNContributors();
	  const AliESDVertex *vtxESD = fESD->GetPrimaryVertex();
	  TString vtxTyp = vtxESD->GetTitle();
	  //	  printf(" %s \n", vtxTyp.Data() );
	  Bool_t fVtxOK = kFALSE;
	  if ( !vtxTyp.Contains("vertexer: Z") ||
	       (vtxESD->GetDispersion()<0.04 && vtxESD->GetZRes()<0.25)) {
	    fVtxOK = kTRUE;
	    fVertexPrim = vtxESD->GetZ();
	    //   printf(" GetPrimaryVertex()  %f \n", fVertexPrim);
	  }   


	  if(fNcontTPC>=2 && fVtxOK )
	    fVertexTPC = fESD->GetPrimaryVertexTPC() ->GetZ();
	  if(fNcont>=0 && fVtxOK) {
	    fVertexSPD = fESD->GetPrimaryVertex()->GetZ();
	  }
	  //	  printf(" SPD Vertex()  %f  TPC %f \n", fVertexSPD, fVertexTPC);
	  fTrackletSPD = mult->GetNumberOfTracklets();
	  fClustersSPD = mult->GetNumberOfITSClusters(0); 
	  fMultV0A =  esdV0->GetMTotV0A();
	  fMultV0C =  esdV0->GetMTotV0C();
	  //TOF hit  
	  Int_t ntracksMatchedToTOF = 0; 
	  Int_t ntracks = fESD->GetNumberOfTracks();
	  for(Int_t itrk=0;itrk<ntracks;itrk++){
	    AliESDtrack* track = fESD->GetTrack(itrk);
	    if (!track) {
	      Printf("ERROR: Could not receive track %d", itrk);
	      continue;
	    }
	    //no track selection just TOF hit
	    if (track->IsOn(AliESDtrack::kTOFout)) ntracksMatchedToTOF++;
	  }
	  fTOFtracks = ntracksMatchedToTOF;
	     if(fESDpid){ //get T0_TOF 
	     fESDpid->SetTOFResponse(fESD,AliESDpid::kTOF_T0);
	     t0tofTrack =(Float_t) (fESDpid->GetTOFResponse().GetStartTime(10.0)); //Get start time from all tracks 
	     }	
	  
	  
	  fpileup = fESD->IsPileupFromSPD();
	  fTimeV0C = esdV0->GetV0CTime();
	  fTimeV0A = esdV0->GetV0ATime();
	  //	  printf(" Vo info \n");      
	  
	  
	  mean = fESD->GetT0TOF();
	  fVertex=fESD->GetT0zVertex();
	  fMeanA = mean[1];
	  fMeanC = mean[2];
	  if (TMath::Abs(fMeanA)<9999 && TMath::Abs(fMeanC)<9999) 
	    fMeanAC = mean[0] ;
	  fSumampC = tz->GetMultC();
	  fSumampA = tz->GetMultA();
	  //    printf("!!!!!!!!fSumampC %f fSumampA %f \n",fSumampC, fSumampA);
	  
	  amp=fESD->GetT0amplitude();
	  time=fESD->GetT0time();
	  for (Int_t i=0; i<24; i++){ 
	    if( time[i]>100 && amp[i]>0.1){
	  //  if(i<12) fSumampC += Int_t (amp[i]+0.5);
	  // if(i>11) fSumampA += Int_t (amp[i]+0.5);
	      famp[i] = amp[i];
	      ftime[i] = time[i];
	    }
      }
	  
	  //new OrA OrC TVDC
      // AliESDTZERO* tz= (AliESDTZERO*) fESD->GetESDTZERO();
      meanbest = tz->GetT0TOFbest();
      fMeanAcalc = meanbest[1];
      fMeanCcalc = meanbest[2];
      if (TMath::Abs(fMeanAcalc)<9999 && TMath::Abs(fMeanCcalc)<9999) 
	fMeanACcalc = meanbest[0] ;
      for (Int_t ii=0; ii<5; ii++){ 
	orA   = tz->GetOrA(ii);
	orC   = tz->GetOrC(ii);
	tvdc  = tz->GetTVDC(ii);
	//	printf("  new signasl %d  %f %f %f \n",ii, orA, orC, tvdc);
	if(ii==0) {
	  fOrA[ii] = orA;
	  fOrC[ii] = orC;
	  fTVDC[ii] = tvdc;
	  //	printf("  new signasl %d  %f %f %f \n",ii, orA, orC, tvdc);
	  //  cout<<tvdc<<" "<<fT0Trigger<<endl;
	  if (tvdc>-5 && tvdc<5 && tvdc!=0) cout<<tvdc<<" "<<fT0Trigger<<endl;
	} else {
	  if ( ii>0 && fOrA[ii-1] !=orA)	fOrA[ii] = orA;
	  else fOrA[ii]=-9999;
	  if ( ii>0 && fOrC[ii-1] !=orC)	fOrC[ii] = orC;
	  else fOrC[ii]=-9999;
	  if ( ii>0 && fTVDC[ii-1] != tvdc)	fTVDC[ii] = tvdc;
	  else fTVDC[ii]=-9999;
	}
	for (Int_t iii=0; iii<24; iii++)
	  fRawTime[iii][ii] =  tz->GetTimeFull(iii,ii);
      }
      fT0pileup[0] = tz->GetPileupFlag();
      fT0pileup[1] = tz->GetBackgroundFlag();
      fT0pileup[2] = tz->GetSatellite();
   
       AliCentrality *centrality = fESD->GetCentrality();
      
      //The centrality function you can use
      
      fcentralityV0M = centrality->GetCentralityPercentile("V0M"); // returns the centrality 
      fcentralityTRK = centrality->GetCentralityPercentile("TRK"); // returns the centrality 
      fcentralityZDC = centrality->GetCentralityPercentile("ZEMvsZDC"); // returns the 
//      fcentralityTRK = centrality->GetCentralityClass10("TRK"); // returns centrality class
       	
  //    fcentralityZDC = centrality->GetCentralityClass10("ZEMvsZDC"); // returns centrality 
      
	} //selected
    } //ESD

  fT0OutTree->Fill();
    
  PostData(1, fOutputList);
  
}      

//________________________________________________________________________
void AliT0HIanalysisTask::Terminate(Option_t *) 
{
  // Draw result to the screen
  // Called once at the end of the query

  /* fOutputList = dynamic_cast<TList*> (GetOutputData(1));
  if (!fOutputList) {
    printf("ERROR: Output list not available\n");
    return;
  }
    */  

}