move cluster energy cut after cluster recalculation is done

[u/mrichter/AliRoot.git] / PWG4 / PartCorrBase / AliCaloTrackReader.cxx

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

//_________________________________________________________________________
// Base class for reading data: MonteCarlo, ESD or AOD, of PHOS EMCAL and 
// Central Barrel Tracking detectors (CTS).
// Not all MC particles/tracks/clusters are kept, some kinematical/fiducial restrictions are done.
// Mother class of : AliCaloTrackESDReader: Fills ESD data in 3 TObjArrays (PHOS, EMCAL, CTS)
//                 : AliCaloTrackMCReader : Fills Kinematics data in 3 TObjArrays (PHOS, EMCAL, CTS)
//                 : AliCaloTrackAODReader: Fills AOD data in 3 TObjArrays (PHOS, EMCAL, CTS) 
//-- Author: Gustavo Conesa (LNF-INFN) 
//////////////////////////////////////////////////////////////////////////////


// --- ROOT system ---
#include <TFile.h>

//---- ANALYSIS system ----
#include "AliMCEvent.h"
#include "AliAODMCHeader.h"
#include "AliGenPythiaEventHeader.h"
#include "AliESDEvent.h"
#include "AliAODEvent.h"
#include "AliVTrack.h"
#include "AliVParticle.h"
#include "AliMixedEvent.h"
#include "AliESDtrack.h"
#include "AliESDtrackCuts.h"
#include "AliTriggerAnalysis.h"
#include "AliESDVZERO.h"

//---- PartCorr/EMCAL ---
#include "AliEMCALRecoUtils.h"
#include "AliCaloTrackReader.h"

ClassImp(AliCaloTrackReader)
  
  
//____________________________________________________________________________
  AliCaloTrackReader::AliCaloTrackReader() : 
    TObject(),                   fEventNumber(-1), //fCurrentFileName(""),
    fDataType(0),                fDebug(0), 
    fFiducialCut(0x0),           fCheckFidCut(kFALSE), 
    fComparePtHardAndJetPt(0),   fPtHardAndJetPtFactor(7),
    fCTSPtMin(0),                fEMCALPtMin(0),                  fPHOSPtMin(0), 
    fCTSPtMax(1000),             fEMCALPtMax(1000),               fPHOSPtMax(1000), 
    fAODBranchList(new TList ),
    fCTSTracks(new TObjArray()), fEMCALClusters(new TObjArray()), fPHOSClusters(new TObjArray()),
    fEMCALCells(0x0),            fPHOSCells(0x0),
    fInputEvent(0x0),            fOutputEvent(0x0),fMC(0x0),
    fFillCTS(0),                 fFillEMCAL(0),                   fFillPHOS(0),
    fFillEMCALCells(0),          fFillPHOSCells(0), 
    fRecalculateClusters(kFALSE),fSelectEmbeddedClusters(kFALSE),
    fTrackStatus(0),             fTrackFilterMask(0),             fESDtrackCuts(0), 
    fTrackMult(0),               fTrackMultEtaCut(0.8),
    fReadStack(kFALSE),          fReadAODMCParticles(kFALSE), 
    fDeltaAODFileName("deltaAODPartCorr.root"),
    fFiredTriggerClassName(""),  fAnaLED(kFALSE),
    fTaskName(""),               fCaloUtils(0x0), 
    fMixedEvent(NULL),           fNMixedEvent(1),                 fVertex(NULL), 
    fWriteOutputDeltaAOD(kFALSE),fOldAOD(kFALSE),                 fCaloFilterPatch(kFALSE),
    fEMCALClustersListName(""),  fZvtxCut(0.),                    
    fAcceptFastCluster(kTRUE),   fRemoveLEDEvents(kFALSE), 
    fDoEventSelection(kFALSE),   fDoV0ANDEventSelection(kFALSE),  fUseEventsWithPrimaryVertex(kFALSE),
    fTriggerAnalysis (new AliTriggerAnalysis), 
    fCentralityClass("V0M"),     fCentralityOpt(10),
    fEventPlaneMethod("Q")
   
{
  //Ctor
  
  //Initialize parameters
  InitParameters();
}

//_________________________________
AliCaloTrackReader::~AliCaloTrackReader() {
  //Dtor
  
  delete fFiducialCut ;
	
  if(fAODBranchList){
    fAODBranchList->Delete();
    delete fAODBranchList ;
  }  
  
  if(fCTSTracks){
    if(fDataType!=kMC)fCTSTracks->Clear() ; 
    else              fCTSTracks->Delete() ; 
    delete fCTSTracks ;
  }
  
  if(fEMCALClusters){
    if(fDataType!=kMC)fEMCALClusters->Clear("C") ; 
    else              fEMCALClusters->Delete() ; 
    delete fEMCALClusters ;
  }
  
  if(fPHOSClusters){
    if(fDataType!=kMC)fPHOSClusters->Clear("C") ; 
    else              fPHOSClusters->Delete() ; 
    delete fPHOSClusters ;
  }
  
  if(fVertex){
    for (Int_t i = 0; i < fNMixedEvent; i++) {
      delete [] fVertex[i] ;

    }
    delete [] fVertex ;
	}

  delete fESDtrackCuts;
  delete fTriggerAnalysis;
  
//  Pointers not owned, done by the analysis frame
//  if(fInputEvent)  delete fInputEvent ;
//  if(fOutputEvent) delete fOutputEvent ;
//  if(fMC)          delete fMC ;  
//  Pointer not owned, deleted by maker
//  if (fCaloUtils) delete fCaloUtils ;

}

//_________________________________________________________________________
Bool_t AliCaloTrackReader::ComparePtHardAndJetPt(){
  // Check the event, if the requested ptHard is much larger than the jet pT, then there is a problem.
  // Only for PYTHIA.
  if(!fReadStack) return kTRUE; //Information not filtered to AOD
  
  if(!strcmp(GetGenEventHeader()->ClassName(), "AliGenPythiaEventHeader")){
    TParticle * jet =  0;
    AliGenPythiaEventHeader* pygeh= (AliGenPythiaEventHeader*) GetGenEventHeader();
    Int_t nTriggerJets =  pygeh->NTriggerJets();
    Float_t ptHard = pygeh->GetPtHard();
    
    //if(fDebug > 1) printf("AliMCAnalysisUtils::PythiaEventHeader: Njets: %d, pT Hard %f\n",nTriggerJets, ptHard);
    Float_t tmpjet[]={0,0,0,0};
    for(Int_t ijet = 0; ijet< nTriggerJets; ijet++){
      pygeh->TriggerJet(ijet, tmpjet);
      jet = new TParticle(94, 21, -1, -1, -1, -1, tmpjet[0],tmpjet[1],tmpjet[2],tmpjet[3], 0,0,0,0);
      //Compare jet pT and pt Hard
      //if(fDebug > 1) printf("AliMCAnalysisUtils:: %d pycell jet pT %f\n",ijet, jet->Pt());
      if(jet->Pt() > fPtHardAndJetPtFactor * ptHard) {
        printf("AliMCAnalysisUtils::PythiaEventHeader: Njets: %d, pT Hard %2.2f, pycell jet pT %2.2f, rejection factor %1.1f\n",
               nTriggerJets, ptHard, jet->Pt(), fPtHardAndJetPtFactor);
        return kFALSE;
      }
    }
    if(jet) delete jet; 
  }
  
  return kTRUE ;
  
}

//____________________________________________________________________________
AliStack* AliCaloTrackReader::GetStack() const {
  //Return pointer to stack
  if(fMC)
    return fMC->Stack();
  else{
    if(fDebug > 1) printf("AliCaloTrackReader::GetStack() - Stack is not available\n"); 
    return 0x0 ;
  }
}

//____________________________________________________________________________
AliHeader* AliCaloTrackReader::GetHeader() const {
  //Return pointer to header
  if(fMC)
    return fMC->Header();
  else{
    printf("AliCaloTrackReader::Header is not available\n"); 
    return 0x0 ;
  }
}
//____________________________________________________________________________
AliGenEventHeader* AliCaloTrackReader::GetGenEventHeader() const {
  //Return pointer to Generated event header
  if(fMC)
    return fMC->GenEventHeader();
  else{
    printf("AliCaloTrackReader::GenEventHeader is not available\n"); 
    return 0x0 ;
  }
}

//____________________________________________________________________________
TClonesArray* AliCaloTrackReader::GetAODMCParticles(Int_t input) const {
  //Return list of particles in AOD. Do it for the corresponding input event.
  
  TClonesArray * rv = NULL ; 
  if(fDataType == kAOD){
 
    if(input == 0){
      //Normal input AOD
      AliAODEvent * evt = dynamic_cast<AliAODEvent*> (fInputEvent) ;
      if(evt)
        rv = (TClonesArray*)evt->FindListObject("mcparticles");
      else  
        printf("AliCaloTrackReader::GetAODMCParticles() - wrong AOD input index? %d, or non existing tree? \n",input); 
      
    }  
    
  } else {
      printf("AliCaloTrackReader::GetAODMCParticles() - Input are not AODs\n"); 
  }
  
  return rv ; 
}

//____________________________________________________________________________
AliAODMCHeader* AliCaloTrackReader::GetAODMCHeader(Int_t input) const {
  //Return MC header in AOD. Do it for the corresponding input event.
  AliAODMCHeader *mch = NULL;
  if(fDataType == kAOD){
    //Normal input AOD
    if(input == 0) {
      mch = (AliAODMCHeader*)((AliAODEvent*)fInputEvent)->FindListObject("mcheader");
    }
    //		//Second input AOD
    //		else if(input == 1){ 
    //       mch = (AliAODMCHeader*) fSecondInputAODEvent->FindListObject("mcheader");
    //  }
    else {
      printf("AliCaloTrackReader::GetAODMCHeader() - wrong AOD input index, %d\n",input);
    }
  }
  else {
    printf("AliCaloTrackReader::GetAODMCHeader() - Input are not AODs\n");
  }
  
  return mch;
}

//_______________________________________________________________
void AliCaloTrackReader::Init()
{
  //Init reader. Method to be called in AliAnaPartCorrMaker
    
  if(fReadStack && fReadAODMCParticles){
    printf("AliCaloTrackReader::Init() - Cannot access stack and mcparticles at the same time, change them \n");
    fReadStack = kFALSE;
    fReadAODMCParticles = kFALSE;
  }
  
}

//_______________________________________________________________
void AliCaloTrackReader::InitParameters()
{
  //Initialize the parameters of the analysis.
  fDataType   = kESD ;
  fCTSPtMin   = 0.1 ;
  fEMCALPtMin = 0.1 ;
  fPHOSPtMin  = 0.1 ;
  fCTSPtMax   = 1000. ;
  fEMCALPtMax = 1000. ;
  fPHOSPtMax  = 1000. ;
  
  //Do not filter the detectors input by default.
  fFillEMCAL      = kFALSE;
  fFillPHOS       = kFALSE;
  fFillCTS        = kFALSE;
  fFillEMCALCells = kFALSE;
  fFillPHOSCells  = kFALSE;

  fReadStack             = kFALSE; // Check in the constructor of the other readers if it was set or in the configuration file
  fReadAODMCParticles    = kFALSE; // Check in the constructor of the other readers if it was set or in the configuration file
  fDeltaAODFileName      = "deltaAODPartCorr.root";
  fFiredTriggerClassName = "";
	 	
  fAnaLED = kFALSE;
  
  //We want tracks fitted in the detectors:
  //fTrackStatus=AliESDtrack::kTPCrefit;
  //fTrackStatus|=AliESDtrack::kITSrefit;  
  fTrackFilterMask = 128; //For AODs, but what is the difference between fTrackStatus and fTrackFilterMask?
  
  fESDtrackCuts = AliESDtrackCuts::GetStandardTPCOnlyTrackCuts(); //initialize with TPC only tracks 

  fV0ADC[0] = 0;   fV0ADC[1] = 0; 
  fV0Mul[0] = 0;   fV0Mul[1] = 0; 

  fZvtxCut   = 10.;
  
  //Centrality
  fCentralityBin[0]=fCentralityBin[1]=-1;
    
}

//________________________________________________________________
void AliCaloTrackReader::Print(const Option_t * opt) const
{

  //Print some relevant parameters set for the analysis
  if(! opt)
    return;

  printf("***** Print: %s %s ******\n", GetName(), GetTitle() ) ;
  printf("Task name      : %s\n", fTaskName.Data()) ;
  printf("Data type      : %d\n", fDataType) ;
  printf("CTS Min pT     : %2.1f GeV/c\n", fCTSPtMin) ;
  printf("EMCAL Min pT   : %2.1f GeV/c\n", fEMCALPtMin) ;
  printf("PHOS Min pT    : %2.1f GeV/c\n", fPHOSPtMin) ;
  printf("CTS Max pT     : %2.1f GeV/c\n", fCTSPtMax) ;
  printf("EMCAL Max pT   : %2.1f GeV/c\n", fEMCALPtMax) ;
  printf("PHOS Max pT    : %2.1f GeV/c\n", fPHOSPtMax) ;
  printf("Use CTS         =     %d\n", fFillCTS) ;
  printf("Use EMCAL       =     %d\n", fFillEMCAL) ;
  printf("Use PHOS        =     %d\n", fFillPHOS) ;
  printf("Use EMCAL Cells =     %d\n", fFillEMCALCells) ;
  printf("Use PHOS  Cells =     %d\n", fFillPHOSCells) ;
  printf("Track status    =     %d\n", (Int_t) fTrackStatus) ;
  printf("Track filter mask (AODs) =  %d\n", (Int_t) fTrackFilterMask) ;
  printf("Track Mult Eta Cut =  %d\n", (Int_t) fTrackMultEtaCut) ;
  printf("Write delta AOD =     %d\n", fWriteOutputDeltaAOD) ;
  printf("Recalculate Clusters = %d\n", fRecalculateClusters) ;
  
  if(fComparePtHardAndJetPt)
	  printf("Compare jet pt and pt hard to accept event, factor = %2.2f",fPtHardAndJetPtFactor);
		
  printf("Read Kine from, stack? %d, AOD ? %d \n", fReadStack, fReadAODMCParticles) ;
  printf("Delta AOD File Name =     %s\n", fDeltaAODFileName.Data()) ;
  printf("Centrality: Class %s, Option %d, Bin [%d,%d] \n", fCentralityClass.Data(),fCentralityOpt,fCentralityBin[0], fCentralityBin[1]) ;

  printf("    \n") ;
  
} 

//___________________________________________________
Bool_t AliCaloTrackReader::FillInputEvent(const Int_t iEntry, const char * /*currentFileName*/) {
  //Fill the event counter and input lists that are needed, called by the analysis maker.

  fEventNumber = iEntry;
  //fCurrentFileName = TString(currentFileName);
  if(!fInputEvent) {
	  if(fDebug >= 0) printf("AliCaloTrackReader::FillInputEvent() - Input event not available, skip event analysis\n");
	  return kFALSE;
  }
  //Select events only fired by a certain trigger configuration if it is provided
  Int_t eventType = 0;
  if(fInputEvent->GetHeader())
	  eventType = ((AliVHeader*)fInputEvent->GetHeader())->GetEventType();

  if (GetFiredTriggerClasses().Contains("FAST")  && !GetFiredTriggerClasses().Contains("ALL") && !fAcceptFastCluster) {
     if(fDebug > 0)  printf("AliCaloTrackReader::FillInputEvent - Do not count events from fast cluster, trigger name %s\n",fFiredTriggerClassName.Data());
    return kFALSE;
  }
  
  //-------------------------------------------------------------------------------------
  // Reject event if large clusters with large energy
  // Use only for LHC11a data for the moment, and if input is clusterizer V1 or V1+unfolding
  // If clusterzer NxN or V2 it does not help
  //-------------------------------------------------------------------------------------
  if(fRemoveLEDEvents){
    

    //printf("Event %d\n",GetEventNumber());
    for (Int_t i = 0; i < fInputEvent->GetNumberOfCaloClusters(); i++)
    {
      AliVCluster *clus = fInputEvent->GetCaloCluster(i);
      if(clus->IsEMCAL()){               
        if ((clus->E() > 500 && clus->GetNCells() > 200 ) || clus->GetNCells() > 200) {
          Int_t absID = clus->GetCellsAbsId()[0];
          Int_t sm = GetCaloUtils()->GetEMCALGeometry()->GetSuperModuleNumber(absID);
           if(fDebug > 0) printf("AliCaloTrackReader::FillInputEvent - reject event %d with cluster : E %f, ncells %d, absId(0) %d, SM %d\n",GetEventNumber(),clus->E(),  clus->GetNCells(),absID, sm);
          return kFALSE;
        }
      }
    }
    
    // Count number of cells with energy larger than 0.1 in SM3, cut on this number
    Int_t ncellsSM3 = 0;
    Int_t ncellsSM4 = 0;
    for(Int_t icell = 0; icell < fInputEvent->GetEMCALCells()->GetNumberOfCells(); icell++){
      Int_t absID = fInputEvent->GetEMCALCells()->GetCellNumber(icell);
      Int_t sm = GetCaloUtils()->GetEMCALGeometry()->GetSuperModuleNumber(absID);
      if(fInputEvent->GetEMCALCells()->GetAmplitude(icell) > 0.1 && sm==3) ncellsSM3++;
      if(fInputEvent->GetEMCALCells()->GetAmplitude(icell) > 0.1 && sm==4) ncellsSM4++;
    }
    
    Int_t ncellcut = 21;
    if(fFiredTriggerClassName.Contains("EMC")) ncellcut = 35;
    
    if(ncellsSM3 >= ncellcut || ncellsSM4 >= 100) {
       if(fDebug > 0) printf(" AliCaloTrackReader::FillInputEvent() - reject event with ncells in SM3 %d and SM4 %d\n",ncellsSM3, ncellsSM4);
      return kFALSE;
    }
  }// Remove LED events
  
  // Reject pure LED events?
  if( fFiredTriggerClassName  !="" && !fAnaLED){
    if(eventType!=7)
      return kFALSE; //Only physics event, do not use for simulated events!!!
    if(fDebug > 0) 
      printf("AliCaloTrackReader::FillInputEvent() - FiredTriggerClass <%s>, selected class <%s>, compare name %d\n",
	     GetFiredTriggerClasses().Data(),fFiredTriggerClassName.Data(), GetFiredTriggerClasses().Contains(fFiredTriggerClassName));
    if( !GetFiredTriggerClasses().Contains(fFiredTriggerClassName) ) return kFALSE;
    else if(fDebug > 0) printf("AliCaloTrackReader::FillInputEvent() - Accepted triggered event\n");
  }
  else if(fAnaLED){
//	  kStartOfRun =       1,    // START_OF_RUN
//	  kEndOfRun =         2,    // END_OF_RUN
//	  kStartOfRunFiles =  3,    // START_OF_RUN_FILES
//	  kEndOfRunFiles =    4,    // END_OF_RUN_FILES
//	  kStartOfBurst =     5,    // START_OF_BURST
//	  kEndOfBurst =       6,    // END_OF_BURST
//	  kPhysicsEvent =     7,    // PHYSICS_EVENT
//	  kCalibrationEvent = 8,    // CALIBRATION_EVENT
//	  kFormatError =      9,    // EVENT_FORMAT_ERROR
//	  kStartOfData =      10,   // START_OF_DATA
//	  kEndOfData =        11,   // END_OF_DATA
//	  kSystemSoftwareTriggerEvent   = 12, // SYSTEM_SOFTWARE_TRIGGER_EVENT
//	  kDetectorSoftwareTriggerEvent = 13  // DETECTOR_SOFTWARE_TRIGGER_EVENT
	 
	  if(eventType!=7 && fDebug > 1 )printf("AliCaloTrackReader::FillInputEvent() - DO LED, Event Type <%d>, 8 Calibration \n",  eventType);
	  if(eventType!=8)return kFALSE;
  }
  
  //In case of analysis of events with jets, skip those with jet pt > 5 pt hard	
  if(fComparePtHardAndJetPt && GetStack()) {
    if(!ComparePtHardAndJetPt()) return kFALSE ;
  }

  //Fill Vertex array
  FillVertexArray();
  //Reject events with Z vertex too large, only for SE analysis, if not, cut on the analysis code
  if(!GetMixedEvent() && TMath::Abs(fVertex[0][2]) > fZvtxCut) return kFALSE;  
  
  //------------------------------------------------------
  //Event rejection depending on vertex, pileup, v0and
  //------------------------------------------------------
  if(fDoEventSelection){
    if(!fCaloFilterPatch){
      //Do not analyze events with pileup
      Bool_t bPileup = fInputEvent->IsPileupFromSPD(3, 0.8, 3., 2., 5.); //Default values, if not it does not compile
      //Bool_t bPileup = event->IsPileupFromSPD(); 
      if(bPileup) return kFALSE;
      
      if(fDoV0ANDEventSelection){
        Bool_t bV0AND = kTRUE; 
        AliESDEvent* esd = dynamic_cast<AliESDEvent*> (fInputEvent);
        if(esd) 
          bV0AND = fTriggerAnalysis->IsOfflineTriggerFired(esd, AliTriggerAnalysis::kV0AND);
        //else bV0AND = //FIXME FOR AODs
        if(!bV0AND) return kFALSE;
      }
      
      if(fUseEventsWithPrimaryVertex && !CheckForPrimaryVertex()) return kFALSE;
      
    }//CaloFilter patch
    else{ 
      if(fInputEvent->GetNumberOfCaloClusters() > 0) {
        AliVCluster * calo = fInputEvent->GetCaloCluster(0);
        if(calo->GetNLabels() == 4){
          Int_t * selection = calo->GetLabels();
          Bool_t bPileup = selection[0];
          if(bPileup) return kFALSE;
          
          Bool_t bGoodV = selection[1]; 
          if(fUseEventsWithPrimaryVertex && !bGoodV) return kFALSE;
          
          if(fDoV0ANDEventSelection){
            Bool_t bV0AND = selection[2]; 
            if(!bV0AND) return kFALSE;
          }
          
          fTrackMult = selection[3];
          if(fTrackMult == 0) return kFALSE;
        } else {
          //First filtered AODs, track multiplicity stored there.  
          fTrackMult = (Int_t) ((AliAODHeader*)fInputEvent->GetHeader())->GetCentrality();
          if(fTrackMult == 0) return kFALSE;          
        }
      }//at least one cluster
      else {
        //printf("AliCaloTrackReader::FillInputEvent() - No clusters in event\n");
        //Remove events with  vertex (0,0,0), bad vertex reconstruction
        if(fUseEventsWithPrimaryVertex && TMath::Abs(fVertex[0][0]) < 1.e-6 && TMath::Abs(fVertex[0][1]) < 1.e-6 && TMath::Abs(fVertex[0][2]) < 1.e-6) return kFALSE;
        
        //First filtered AODs, track multiplicity stored there.  
        fTrackMult = (Int_t) ((AliAODHeader*)fInputEvent->GetHeader())->GetCentrality();
        if(fTrackMult == 0) return kFALSE;
      }// no cluster
    }// CaloFileter patch
  }// Event selection
  //------------------------------------------------------

  //Check if there is a centrality value, PbPb analysis, and if a centrality bin selection is requested
  //If we need a centrality bin, we select only those events in the corresponding bin.
  if(GetCentrality() && fCentralityBin[0]>=0 && fCentralityBin[1]>=0 && fCentralityOpt==100){
    Int_t cen = GetEventCentrality();
    if(cen > fCentralityBin[1] || cen < fCentralityBin[0]) return kFALSE; //reject events out of bin.
  }
  
  //Fill the arrays with cluster/tracks/cells data
   if(fFillEMCALCells) 
    FillInputEMCALCells();
  if(fFillPHOSCells)  
    FillInputPHOSCells();
	
  if(fFillCTS){   
    FillInputCTS();
    //Accept events with at least one track
    if(fTrackMult == 0 && fDoEventSelection) return kFALSE;
  }
  
  if(fFillEMCAL) 
    FillInputEMCAL();
  if(fFillPHOS)  
    FillInputPHOS();

  FillInputVZERO();
	
  return kTRUE ;
}

//__________________________________________________
void AliCaloTrackReader::ResetLists() {
  //  Reset lists, called by the analysis maker 

  if(fCTSTracks)       fCTSTracks     -> Clear();
  if(fEMCALClusters)   fEMCALClusters   -> Clear("C");
  if(fPHOSClusters)    fPHOSClusters    -> Clear("C");
//  if(fEMCALCells) fEMCALCells -> Clear("");
//  if(fPHOSCells)  fPHOSCells  -> Clear("");

  fV0ADC[0] = 0;   fV0ADC[1] = 0; 
  fV0Mul[0] = 0;   fV0Mul[1] = 0; 

}

//____________________________________________________________________________
void AliCaloTrackReader::SetInputEvent(AliVEvent* const input)  
{
  fInputEvent  = input;
  fMixedEvent = dynamic_cast<AliMixedEvent*>(GetInputEvent()) ; 
  if (fMixedEvent) {
    fNMixedEvent = fMixedEvent->GetNumberOfEvents() ; 
  }

  //Delete previous vertex
  if(fVertex){
    for (Int_t i = 0; i < fNMixedEvent; i++) {
      delete [] fVertex[i] ; 
    }
    delete [] fVertex ;
  }
  
  fVertex = new Double_t*[fNMixedEvent] ; 
  for (Int_t i = 0; i < fNMixedEvent; i++) {
    fVertex[i] = new Double_t[3] ; 
    fVertex[i][0] = 0.0 ; 
    fVertex[i][1] = 0.0 ; 
    fVertex[i][2] = 0.0 ; 
  }
}

//__________________________________________________
Int_t AliCaloTrackReader::GetEventCentrality() const {
  //Return current event centrality
  
  if(GetCentrality()){
    if(fCentralityOpt==100)     return (Int_t) GetCentrality()->GetCentralityPercentile(fCentralityClass); // 100 bins max
    else if(fCentralityOpt==10) return GetCentrality()->GetCentralityClass10(fCentralityClass);// 10 bins max
    else if(fCentralityOpt==20) return GetCentrality()->GetCentralityClass5(fCentralityClass); // 20 bins max
    else {
      printf("AliAnaPartCorrBaseClass::Unknown centrality option %d, use 10, 20 or 100\n",fCentralityOpt);
      return 0;
    } 
  }
  else return 0;
  
}

//____________________________________________________________________________
void AliCaloTrackReader::GetVertex(Double_t vertex[3]) const {
  //Return vertex position to be used for single event analysis
  vertex[0]=fVertex[0][0];  
  vertex[1]=fVertex[0][1];  
  vertex[2]=fVertex[0][2];
}

//____________________________________________________________________________
void AliCaloTrackReader::GetVertex(Double_t vertex[3], const Int_t evtIndex) const {
  //Return vertex position for mixed event, recover the vertex in a particular event.
  
  vertex[0]=fVertex[evtIndex][0];  vertex[1]=fVertex[evtIndex][1];  vertex[2]=fVertex[evtIndex][2];
  
}

//____________________________________________________________________________
void AliCaloTrackReader::FillVertexArray() {
  
  //Fill data member with vertex
  //In case of Mixed event, multiple vertices
  
  //Delete previous vertex
  if(fVertex){
    for (Int_t i = 0; i < fNMixedEvent; i++) {
      delete [] fVertex[i] ; 
    }
    delete [] fVertex ;  
  }
  
  fVertex = new Double_t*[fNMixedEvent] ; 
  for (Int_t i = 0; i < fNMixedEvent; i++) {
    fVertex[i] = new Double_t[3] ; 
    fVertex[i][0] = 0.0 ; 
    fVertex[i][1] = 0.0 ; 
    fVertex[i][2] = 0.0 ; 
  }          
  
  if (!fMixedEvent) { //Single event analysis
    if(fDataType!=kMC){

      if(fInputEvent->GetPrimaryVertex()){
        fInputEvent->GetPrimaryVertex()->GetXYZ(fVertex[0]); 
      }
      else {
        printf("AliCaloTrackReader::FillVertexArray() - NULL primary vertex\n");
        fVertex[0][0]=0.;   fVertex[0][1]=0.;   fVertex[0][2]=0.;
      }//Primary vertex pointer do not exist
      
    } else {//MC read event 
      fVertex[0][0]=0.;   fVertex[0][1]=0.;   fVertex[0][2]=0.;
    }
      
    if(fDebug > 1)
      printf("AliCaloTrackReader::FillVertexArray() - Single Event Vertex : %f,%f,%f\n",fVertex[0][0],fVertex[0][1],fVertex[0][2]);

  } else { // MultiEvent analysis
    for (Int_t iev = 0; iev < fNMixedEvent; iev++) {
      if (fMixedEvent->GetVertexOfEvent(iev))
        fMixedEvent->GetVertexOfEvent(iev)->GetXYZ(fVertex[iev]);
      else { // no vertex found !!!!
        AliWarning("No vertex found");
      }

      if(fDebug > 1)
        printf("AliCaloTrackReader::FillVertexArray() - Multi Event %d Vertex : %f,%f,%f\n",iev,fVertex[iev][0],fVertex[iev][1],fVertex[iev][2]);

    }
  }
  
}

//____________________________________________________________________________
void AliCaloTrackReader::FillInputCTS() {
  //Return array with Central Tracking System (CTS) tracks
  
  if(fDebug > 2 ) printf("AliCaloTrackReader::FillInputCTS()\n");
  
  Int_t nTracks   = fInputEvent->GetNumberOfTracks() ;
  Double_t p[3];
  fTrackMult = 0;
  Int_t nstatus = 0;
  for (Int_t itrack =  0; itrack <  nTracks; itrack++) {////////////// track loop
    AliVTrack * track = (AliVTrack*)fInputEvent->GetTrack(itrack) ; // retrieve track from esd

    //Select tracks under certain conditions, TPCrefit, ITSrefit ... check the set bits
    if (fTrackStatus && !((track->GetStatus() & fTrackStatus) == fTrackStatus)) 
      continue ;
    
    nstatus++;
    
    if     (fDataType==kESD && !fESDtrackCuts->AcceptTrack((AliESDtrack*)track))
    {
      continue;
    }
    else if(fDataType==kAOD)
    {
      AliAODTrack *aodtrack = dynamic_cast <AliAODTrack*>(track);
      if(aodtrack){
        if(fDebug > 2 ) 
          printf("AliCaloTrackReader::FillInputCTS():AOD track type: %c \n", aodtrack->GetType());
        if (fDataType!=kMC && aodtrack->TestFilterMask(fTrackFilterMask)==kFALSE) continue;
        if(aodtrack->GetType()!=AliAODTrack::kPrimary) continue;
      }
    }
    
    //Count the tracks in eta < 0.9
    //printf("Eta %f cut  %f\n",TMath::Abs(track->Eta()),fTrackMultEtaCut);
    if(TMath::Abs(track->Eta())< fTrackMultEtaCut) fTrackMult++;
    
    track->GetPxPyPz(p) ;
    TLorentzVector momentum(p[0],p[1],p[2],0);
    
    if(fCTSPtMin < momentum.Pt() && fCTSPtMax > momentum.Pt()){
      
      if(fCheckFidCut && !fFiducialCut->IsInFiducialCut(momentum,"CTS")) 
        continue;
      
      if(fDebug > 2 && momentum.Pt() > 0.1) 
        printf("AliCaloTrackReader::FillInputCTS() - Selected tracks E %3.2f, pt %3.2f, phi %3.2f, eta %3.2f\n",
               momentum.E(),momentum.Pt(),momentum.Phi()*TMath::RadToDeg(),momentum.Eta());
      
      if (fMixedEvent) {
        track->SetID(itrack);
      }
      
      fCTSTracks->Add(track);        
       
    }//Pt and Fiducial cut passed. 
  }// track loop
	
  //fCTSTracksNormalInputEntries = fCTSTracks->GetEntriesFast();
  if(fDebug > 1) 
    printf("AliCaloTrackReader::FillInputCTS()   - aod entries %d, input tracks %d, pass status %d, multipliticy %d\n", fCTSTracks->GetEntriesFast(), nTracks, nstatus, fTrackMult);//fCTSTracksNormalInputEntries);

}

//____________________________________________________________________________
void AliCaloTrackReader::FillInputEMCALAlgorithm(AliVCluster * clus, const Int_t iclus) {
  //Fill the EMCAL data in the array, do it
  
  Int_t vindex = 0 ;  
  if (fMixedEvent) 
    vindex = fMixedEvent->EventIndexForCaloCluster(iclus);
  
  //Reject clusters with bad channels, close to borders and exotic;
  if(!GetCaloUtils()->GetEMCALRecoUtils()->IsGoodCluster(clus,GetCaloUtils()->GetEMCALGeometry(),GetEMCALCells(),fInputEvent->GetBunchCrossNumber())) return;
  
  //Mask all cells in collumns facing ALICE thick material if requested
  if(GetCaloUtils()->GetNMaskCellColumns()){
    Int_t absId   = -1;
    Int_t iSupMod = -1;
    Int_t iphi    = -1;
    Int_t ieta    = -1;
    Bool_t shared = kFALSE;
    GetCaloUtils()->GetEMCALRecoUtils()->GetMaxEnergyCell(GetCaloUtils()->GetEMCALGeometry(), GetEMCALCells(),clus,absId,iSupMod,ieta,iphi,shared);
    if(GetCaloUtils()->MaskFrameCluster(iSupMod, ieta)) return;
  }
  
  if(fSelectEmbeddedClusters){
    if(clus->GetNLabels()==0 || clus->GetLabel() < 0) return;
    //else printf("Embedded cluster,  %d, n label %d label %d  \n",iclus,clus->GetNLabels(),clus->GetLabel());
  }

  //Float_t pos[3];
  //clus->GetPosition(pos);
  //printf("Before Corrections: e %f, x %f, y %f, z %f\n",clus->E(),pos[0],pos[1],pos[2]);
  
  if(fRecalculateClusters){
    //Recalibrate the cluster energy 
    if(GetCaloUtils()->IsRecalibrationOn()) {
      
      Float_t energy = GetCaloUtils()->RecalibrateClusterEnergy(clus, GetEMCALCells());
      
      clus->SetE(energy);
      //printf("Recalibrated Energy %f\n",clus->E());  
      
      GetCaloUtils()->RecalculateClusterShowerShapeParameters(GetEMCALCells(),clus);
      GetCaloUtils()->RecalculateClusterPID(clus);
      
    } // recalculate E
    
    //Recalculate distance to bad channels, if new list of bad channels provided
    GetCaloUtils()->RecalculateClusterDistanceToBadChannel(GetEMCALCells(),clus);
    
    //Recalculate cluster position
    if(GetCaloUtils()->IsRecalculationOfClusterPositionOn()){
      GetCaloUtils()->RecalculateClusterPosition(GetEMCALCells(),clus); 
      //clus->GetPosition(pos);
      //printf("After  Corrections: e %f, x %f, y %f, z %f\n",clus->E(),pos[0],pos[1],pos[2]);
    }
  }
  
  // Recalculate TOF
  if(GetCaloUtils()->GetEMCALRecoUtils()->IsTimeRecalibrationOn()) {
    
    Double_t tof      = clus->GetTOF();
    Float_t  frac     =-1;
    Int_t    absIdMax = GetCaloUtils()->GetMaxEnergyCell(fEMCALCells, clus,frac);
    
    if(fDataType==AliCaloTrackReader::kESD){ 
      tof = fEMCALCells->GetCellTime(absIdMax);
    }
    
    GetCaloUtils()->GetEMCALRecoUtils()->RecalibrateCellTime(absIdMax,fInputEvent->GetBunchCrossNumber(),tof);
    
    clus->SetTOF(tof);
    
  }// Time recalibration    
  
  //Correct non linearity
  if(GetCaloUtils()->IsCorrectionOfClusterEnergyOn()){
    GetCaloUtils()->CorrectClusterEnergy(clus) ;
    //printf("Linearity Corrected Energy %f\n",clus->E());  
    
    //In case of MC analysis, to match resolution/calibration in real data
    Float_t rdmEnergy = GetCaloUtils()->GetEMCALRecoUtils()->SmearClusterEnergy(clus);
    // printf("\t Energy %f, smeared %f\n", clus->E(),rdmEnergy);
    clus->SetE(rdmEnergy);
  }
  
  TLorentzVector momentum ;
  
  clus->GetMomentum(momentum, fVertex[vindex]);      
  
  if(fCheckFidCut && !fFiducialCut->IsInFiducialCut(momentum,"EMCAL")) 
    return;
  
  if(fDebug > 2 && momentum.E() > 0.1) 
    printf("AliCaloTrackReader::FillInputEMCAL() - Selected clusters E %3.2f, pt %3.2f, phi %3.2f, eta %3.2f\n",
           momentum.E(),momentum.Pt(),momentum.Phi()*TMath::RadToDeg(),momentum.Eta());
  
  if(fEMCALPtMin < momentum.E() && fEMCALPtMax > momentum.E()) return;
  
  if (fMixedEvent) 
    clus->SetID(iclus) ; 
  
  fEMCALClusters->Add(clus);	
  
}

//____________________________________________________________________________
void AliCaloTrackReader::FillInputEMCAL() {
  //Return array with EMCAL clusters in aod format
  
  if(fDebug > 2 ) printf("AliCaloTrackReader::FillInputEMCAL()\n");
  
  // First recalibrate cells, time or energy
  //  if(GetCaloUtils()->IsRecalibrationOn())
  //    GetCaloUtils()->GetEMCALRecoUtils()->RecalibrateCells(GetCaloUtils()->GetEMCALGeometry(), 
  //                                                          GetEMCALCells(), 
  //                                                          fInputEvent->GetBunchCrossNumber());
  
  //Loop to select clusters in fiducial cut and fill container with aodClusters
  if(fEMCALClustersListName==""){
    Int_t nclusters = fInputEvent->GetNumberOfCaloClusters();
    for (Int_t iclus =  0; iclus <  nclusters; iclus++) {
      AliVCluster * clus = 0;
      if ( (clus = fInputEvent->GetCaloCluster(iclus)) ) {
        if (IsEMCALCluster(clus)){          
          FillInputEMCALAlgorithm(clus, iclus);
        }//EMCAL cluster
      }// cluster exists
    }// cluster loop
    
    //Recalculate track matching
    GetCaloUtils()->RecalculateClusterTrackMatching(fInputEvent);
    
  }//Get the clusters from the input event
  else {
    TClonesArray * clusterList = 0x0; 
    if(fOutputEvent) clusterList = dynamic_cast<TClonesArray*> (fOutputEvent->FindListObject(fEMCALClustersListName));
    if(!clusterList){
      //printf("AliCaloTrackReader::FillInputEMCAL() - Wrong name of list with clusters? Try input event <%s>\n",fEMCALClustersListName.Data());
      //List not in output event, try input event
      clusterList = dynamic_cast<TClonesArray*> (fInputEvent->FindListObject(fEMCALClustersListName));
      if(!clusterList){
        printf("AliCaloTrackReader::FillInputEMCAL() - Wrong name of list with clusters?  <%s>\n",fEMCALClustersListName.Data());
        return;
      }
    }
    Int_t nclusters = clusterList->GetEntriesFast();
    for (Int_t iclus =  0; iclus <  nclusters; iclus++) {
      AliVCluster * clus = dynamic_cast<AliVCluster*> (clusterList->At(iclus));
      //printf("E %f\n",clus->E());
      if (clus) FillInputEMCALAlgorithm(clus, iclus);
      else printf("AliCaloTrackReader::FillInputEMCAL() - Null cluster in list!\n");
      
    }// cluster loop
    
    //Recalculate track matching, not necessary, already done in the reclusterization task
    //GetCaloUtils()->RecalculateClusterTrackMatching(fInputEvent,clusterList);
    
  }
    
  if(fDebug > 1) printf("AliCaloTrackReader::FillInputEMCAL() - aod entries %d\n",  fEMCALClusters->GetEntriesFast());
  
}

//____________________________________________________________________________
void AliCaloTrackReader::FillInputPHOS() {
  //Return array with PHOS clusters in aod format
  
  if(fDebug > 2 ) printf("AliCaloTrackReader::FillInputPHOS()\n");
  
  //Loop to select clusters in fiducial cut and fill container with aodClusters
  Int_t nclusters = fInputEvent->GetNumberOfCaloClusters();
  for (Int_t iclus = 0; iclus < nclusters; iclus++) {
    AliVCluster * clus = 0;
    if ( (clus = fInputEvent->GetCaloCluster(iclus)) ) {
      if (IsPHOSCluster(clus)){
        //Check if the cluster contains any bad channel and if close to calorimeter borders
        Int_t vindex = 0 ;  
        if (fMixedEvent) 
          vindex = fMixedEvent->EventIndexForCaloCluster(iclus);
        if( GetCaloUtils()->ClusterContainsBadChannel("PHOS",clus->GetCellsAbsId(), clus->GetNCells())) 
          continue;
        if(!GetCaloUtils()->CheckCellFiducialRegion(clus, fInputEvent->GetPHOSCells(), fInputEvent, vindex)) 
          continue;
                
        if(fRecalculateClusters){
          
          //Recalibrate the cluster energy 
          if(GetCaloUtils()->IsRecalibrationOn()) {
            Float_t energy = GetCaloUtils()->RecalibrateClusterEnergy(clus, (AliAODCaloCells*)GetPHOSCells());
            clus->SetE(energy);
          }
          
        }
        
        TLorentzVector momentum ;
        
        clus->GetMomentum(momentum, fVertex[vindex]);      
        
        if(fCheckFidCut && !fFiducialCut->IsInFiducialCut(momentum,"PHOS")) 
          continue;
        
        if(fDebug > 2 && momentum.E() > 0.1) 
          printf("AliCaloTrackReader::FillInputPHOS() - Selected clusters E %3.2f, pt %3.2f, phi %3.2f, eta %3.2f\n",
                 momentum.E(),momentum.Pt(),momentum.Phi()*TMath::RadToDeg(),momentum.Eta());        
        
        if(fPHOSPtMin < momentum.E() && fPHOSPtMax > momentum.E()) continue;
        
        if (fMixedEvent) {
          clus->SetID(iclus) ; 
        }              
        
        fPHOSClusters->Add(clus);	
        
      }//PHOS cluster
    }//cluster exists
  }//esd cluster loop
  
  if(fDebug > 1) printf("AliCaloTrackReader::FillInputPHOS()  - aod entries %d\n",  fPHOSClusters->GetEntriesFast());
  
}

//____________________________________________________________________________
void AliCaloTrackReader::FillInputEMCALCells() {
  //Return array with EMCAL cells in aod format
  
  fEMCALCells = fInputEvent->GetEMCALCells(); 
  
}

//____________________________________________________________________________
void AliCaloTrackReader::FillInputPHOSCells() {
  //Return array with PHOS cells in aod format
  
  fPHOSCells = fInputEvent->GetPHOSCells(); 
  
}

//____________________________________________________________________________
void AliCaloTrackReader::FillInputVZERO(){
  //Fill VZERO information in data member, add all the channels information.
  AliVVZERO* v0 = fInputEvent->GetVZEROData();
  //printf("Init V0: ADC (%d,%d), Multiplicity (%d,%d) \n",fV0ADC[0],fV0ADC[1],fV0Mul[0],fV0Mul[1]);
  
  if (v0) 
  {
    AliESDVZERO* esdV0 = dynamic_cast<AliESDVZERO*> (v0);
    for (Int_t i = 0; i < 32; i++)
    {
      if(esdV0){//Only available in ESDs
        fV0ADC[0] += (Int_t)esdV0->GetAdcV0C(i);
        fV0ADC[1] += (Int_t)esdV0->GetAdcV0A(i);
      }
      fV0Mul[0] += (Int_t)v0->GetMultiplicityV0C(i);
      fV0Mul[1] += (Int_t)v0->GetMultiplicityV0A(i);
    }
    if(fDebug > 0)
      printf("V0: ADC (%d,%d), Multiplicity (%d,%d) \n",fV0ADC[0],fV0ADC[1],fV0Mul[0],fV0Mul[1]);
  }
  else
  {
    if(fDebug > 0)
      printf("Cannot retrieve V0 ESD! Run w/ null V0 charges\n ");
  }
}


//____________________________________________________________________________
Bool_t AliCaloTrackReader::IsEMCALCluster(AliVCluster* cluster) const {
  // Check if it is a cluster from EMCAL. For old AODs cluster type has
  // different number and need to patch here
    
  if(fDataType==kAOD && fOldAOD)
  {
    if (cluster->GetType() == 2) return kTRUE;
    else                         return kFALSE;
  }
  else 
  {
    return cluster->IsEMCAL();
  }

}

//____________________________________________________________________________
Bool_t AliCaloTrackReader::IsPHOSCluster(AliVCluster * cluster) const {
  //Check if it is a cluster from PHOS.For old AODs cluster type has
  // different number and need to patch here
  
  if(fDataType==kAOD && fOldAOD)
  {
    Int_t type = cluster->GetType();
    if (type == 0 || type == 1) return kTRUE;
    else                        return kFALSE;
  }
  else 
  {
    return cluster->IsPHOS();
  }
  
}

//____________________________________________________________________________
Bool_t AliCaloTrackReader::CheckForPrimaryVertex(){
  //Check if the vertex was well reconstructed, copy from V0Reader of conversion group
  //Only for ESDs ...
  
  AliESDEvent * event = dynamic_cast<AliESDEvent*> (fInputEvent);
  if(!event) return kTRUE;
  
  if(event->GetPrimaryVertexTracks()->GetNContributors() > 0) {
    return kTRUE;
  }
  
  if(event->GetPrimaryVertexTracks()->GetNContributors() < 1) {
    // SPD vertex
    if(event->GetPrimaryVertexSPD()->GetNContributors() > 0) {
      //cout<<"spd vertex type::"<< fESDEvent->GetPrimaryVertex()->GetName() << endl;
      return kTRUE;
      
    }
    if(event->GetPrimaryVertexSPD()->GetNContributors() < 1) {
      //      cout<<"bad vertex type::"<< event->GetPrimaryVertex()->GetName() << endl;
      return kFALSE;
    }
  }
  
  return kFALSE;  
  
}