[u/mrichter/AliRoot.git] / PWG / CaloTrackCorrBase / AliAnaCaloTrackCorrBaseClass.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.                  *
 **************************************************************************/

//_________________________________________________________________________
// Base class for CaloTrackCorr analysis algorithms
//-- Author: Gustavo Conesa (LNF-INFN, LPSC-Grenoble) 
//
//
//_________________________________________________________________________


// --- ROOT system ---
#include <TClonesArray.h>
//#include <Riostream.h>

//---- AliRoot system ----
#include "AliAnaCaloTrackCorrBaseClass.h"
#include "AliCaloTrackReader.h"
#include "AliCalorimeterUtils.h"
#include "AliCaloPID.h"
#include "AliFiducialCut.h"
#include "AliIsolationCut.h"
#include "AliMCAnalysisUtils.h"
#include "AliNeutralMesonSelection.h"
#include "AliVCaloCells.h" 
#include "AliAODEvent.h"
#include "AliAODHandler.h"
#include "AliAnalysisManager.h"
#include "AliAODPWG4Particle.h"

ClassImp(AliAnaCaloTrackCorrBaseClass)


//__________________________________________________________
AliAnaCaloTrackCorrBaseClass::AliAnaCaloTrackCorrBaseClass() : 
TObject(), 
fDataMC(0),                   fDebug(0),                   fCheckFidCut(0),
fCheckCaloPID(0),             fRecalculateCaloPID(0), 
fMinPt(0),                    fMaxPt(0),                   fPairTimeCut(200), 
fMultiBin(0),                 fNZvertBin(0),
fNrpBin(0),                   fNCentrBin(0),
fNmaxMixEv(0),                fMaxMulti(0),                fMinMulti(0),
fUseSelectEvent(kFALSE),      fMakePlots(kFALSE),
fInputAODBranch(0x0),         fInputAODName(""),
fOutputAODBranch(0x0),        fNewAOD(kFALSE),
fOutputAODName(""),           fOutputAODClassName(""),
fAODObjArrayName(""),         fAddToHistogramsName(""),
fCaloPID(0x0),                fCaloUtils(0x0),
fFidCut(0x0),                 fHisto(0x0),
fIC(0x0),                     fMCUtils(0x0),                
fNMS(0x0),                    fReader(0x0)
{
  //Default Ctor
  
  //Initialize parameters
  InitParameters();
}

//___________________________________________________________
AliAnaCaloTrackCorrBaseClass::~AliAnaCaloTrackCorrBaseClass() 
{
  // Remove all pointers except analysis output pointers.
  
  //delete fCaloUtils ; //Already deleted in maker
  //delete fReader ;    //Already deleted in maker
	
  delete fCaloPID ; 
  delete fFidCut  ;  
  delete fIC      ;      
  delete fMCUtils ; 
  delete fNMS     ;     
  delete fHisto   ;    
}

//______________________________________________________________________
void AliAnaCaloTrackCorrBaseClass::AddAODParticle(AliAODPWG4Particle pc)
{
  //Put AOD calo cluster in the AODParticleCorrelation array
  
  if(fOutputAODBranch){
    
    Int_t i = fOutputAODBranch->GetEntriesFast();
    //new((*fOutputAODBranch)[i])  AliAODPWG4Particle(pc);
    if(strcmp(fOutputAODBranch->GetClass()->GetName(),"AliAODPWG4Particle")==0)
      new((*fOutputAODBranch)[i])  AliAODPWG4Particle(pc);
    else   if(strcmp(fOutputAODBranch->GetClass()->GetName(),"AliAODPWG4ParticleCorrelation")==0)
      new((*fOutputAODBranch)[i])  AliAODPWG4ParticleCorrelation(pc);
    else {
      printf("AliAnaCaloTrackCorrBaseClass::AddAODParticle() - Cannot add an object of type < %s >, to the AOD TClonesArray \n",  
             fOutputAODBranch->GetClass()->GetName());
      abort();    
    }
  }
  else {
    printf(" AliAnaCaloTrackCorrBaseClass::AddAODParticle() - No AOD branch available!!!\n");
    abort();
  }
  
}	

//_______________________________________________________________________________
Int_t AliAnaCaloTrackCorrBaseClass::CheckMixedEventVertex(const Int_t caloLabel, 
                                                          const Int_t trackLabel)
{
  // Check vertex in mixed events
  
  if (!GetMixedEvent()) return 1; // Not mixed event continue normal processing
  
  Int_t evt = -1;
  
  if     (caloLabel  >= 0 )
  {
    evt = GetMixedEvent()->EventIndexForCaloCluster(caloLabel) ;
  }
  else if(trackLabel >= 0 )
  {
    evt = GetMixedEvent()->EventIndex(trackLabel) ;
  }
  else  
    return 0; // go to next entry in the particle list
  
  if(evt == -1) 
    return 0 ; // to content coverity
  
  if (TMath::Abs(GetVertex(evt)[2]) > GetZvertexCut())  return -1; // Vertex out of range process next event
  
  return 1 ; // continue processing normally
  
}

//________________________________________________________________
void AliAnaCaloTrackCorrBaseClass::ConnectInputOutputAODBranches() 
{
  //Recover ouput and input AOD pointers for each event in the maker
	
  //Delta AODs
  if(fDebug > 3) printf("AliAnaCaloTrackCorrBaseClass::ConnectInputOutputAODBranches() - Connect Input with name: <%s>; Connect output with name <%s>\n",fInputAODName.Data(),fOutputAODName.Data());
  
  //Get the AOD handler, if output AOD is created use it, if not get the branches from the input which should be deltaAODs
  AliAODHandler* aodHandler = 0x0;
  Bool_t outAOD = kFALSE;
  if((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler()) outAOD = kTRUE;
  if(outAOD) aodHandler = (AliAODHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler()); 
  else 	     aodHandler = (AliAODHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler());
  
  if(!GetReader()->WriteDeltaAODToFile())
  {
    fOutputAODBranch =  (TClonesArray *) (fReader->GetAODBranchList())->FindObject(fOutputAODName);
    fInputAODBranch  =  (TClonesArray *) (fReader->GetAODBranchList())->FindObject(fInputAODName);	
  }
  else if (aodHandler->GetExtensions()) { 
    
    AliAODExtension *ext = (AliAODExtension*)aodHandler->GetExtensions()->FindObject(GetReader()->GetDeltaAODFileName()); 
    if(ext){
      AliAODEvent *aodEvent = ext->GetAOD(); 
      if(fNewAOD)fOutputAODBranch = (TClonesArray*) aodEvent->FindListObject(fOutputAODName);
      fInputAODBranch = (TClonesArray*) aodEvent->FindListObject(fInputAODName); 	  
      if(!fOutputAODBranch && fNewAOD) fOutputAODBranch =  (TClonesArray *) fReader->GetOutputEvent()->FindListObject(fOutputAODName);
      if(!fInputAODBranch)  fInputAODBranch  =  (TClonesArray *) fReader->GetOutputEvent()->FindListObject(fInputAODName);
    }
    else{//If no Delta AODs, kept in standard branch, to revise. 
      if(fNewAOD && fReader->GetOutputEvent()) {
        fOutputAODBranch =  (TClonesArray *) fReader->GetOutputEvent()->FindListObject(fOutputAODName);
        fInputAODBranch  =  (TClonesArray *) fReader->GetOutputEvent()->FindListObject(fInputAODName);	
      }
      else {
        fInputAODBranch  =  (TClonesArray *) fReader->GetInputEvent()->FindListObject(fInputAODName);	
        if(!fInputAODBranch && fReader->GetOutputEvent() ) 
          fInputAODBranch  =  (TClonesArray *) fReader->GetOutputEvent()->FindListObject(fInputAODName);//Try the output event.
      }
    }
  }
  else{ //If no Delta AODs, kept in standard branch
    if(fNewAOD && fReader->GetOutputEvent()) {
      fOutputAODBranch =  (TClonesArray *) fReader->GetOutputEvent()->FindListObject(fOutputAODName);
      fInputAODBranch  =  (TClonesArray *) fReader->GetOutputEvent()->FindListObject(fInputAODName);	
    }
    else{ 
      fInputAODBranch  =  (TClonesArray *) fReader->GetInputEvent()->FindListObject(fInputAODName);
      if(!fInputAODBranch && fReader->GetOutputEvent())  
        fInputAODBranch  =  (TClonesArray *) fReader->GetOutputEvent()->FindListObject(fInputAODName);//Try the output event.
    }
  }
  
  if(GetDebug() > 1){
    if(fNewAOD && !fOutputAODBranch) 
      printf(" AliAnaCaloTrackCorrBaseClass::ConnectInputOutputAODBranches() - Output Branch <%s>, not found!\n",fOutputAODName.Data());
    if(!fNewAOD && !fInputAODBranch) 
      printf(" AliAnaCaloTrackCorrBaseClass::ConnectInputOutputAODBranches() - Input Branch  <%s>, not found!\n",fInputAODName.Data());
  }
}

//__________________________________________________________________________________________
AliVCluster * AliAnaCaloTrackCorrBaseClass::FindCluster(TObjArray* clusters, const Int_t id, 
                                                        Int_t & iclus, const Int_t first) 
{
  // Given the cluster ID stored in AliAODPWG4Particle, get the originator cluster and its index in the array
  
  if(!clusters) return 0x0;
  
  for(iclus = first; iclus < clusters->GetEntriesFast(); iclus++){
    AliVCluster *cluster= dynamic_cast<AliVCluster*> (clusters->At(iclus));
    if(cluster){
      if     (cluster->GetID()==id) {
        return cluster;
      }
    }      
  }// calorimeter clusters loop
  
  return 0x0;
  
}

//______________________________________________________________________________
TClonesArray * AliAnaCaloTrackCorrBaseClass::GetAODBranch(TString aodName) const 
{
	//Recover ouput and input AOD pointers for each event in the maker
	
	//Delta AODs
	if(fDebug > 3) printf("AliAnaCaloTrackCorrBaseClass::GetAODBranch() - Get Input Branch with name: <%s>; \n",aodName.Data());
	
  //Get the AOD handler, if output AOD is created use it, if not get the branches from the input which should be deltaAODs
  AliAODHandler* aodHandler = 0x0;
  Bool_t outAOD = kFALSE;
  if((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler()) outAOD = kTRUE;
  if(outAOD) aodHandler = (AliAODHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler()); 
  else 	     aodHandler = (AliAODHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler());
  
  if(!GetReader()->WriteDeltaAODToFile())
  {
    return  (TClonesArray *) (fReader->GetAODBranchList())->FindObject(aodName);
  }
  else if (aodHandler->GetExtensions())
  { 
    AliAODExtension *ext = (AliAODExtension*)aodHandler->GetExtensions()->FindObject(GetReader()->GetDeltaAODFileName()); 
    if(ext){
      AliAODEvent *aodEvent = ext->GetAOD(); 
      TClonesArray * aodbranch =  (TClonesArray*) aodEvent->FindListObject(aodName);
      if(aodbranch) return aodbranch;
      else {
        if(outAOD) return  (TClonesArray *) fReader->GetOutputEvent()->FindListObject(aodName);
        else       return  (TClonesArray *) fReader->GetInputEvent() ->FindListObject(aodName);
      }
    }
    else{//If no Delta AODs, kept in standard branch, to revise. 
      if(outAOD) return (TClonesArray *) fReader->GetOutputEvent()->FindListObject(aodName);
      else       return (TClonesArray *) fReader->GetInputEvent() ->FindListObject(aodName);
    }
  }
  else{ //If no Delta AODs, kept in standard branch, to revise. 
    if(outAOD) return (TClonesArray *)  fReader->GetOutputEvent()->FindListObject(aodName);
    else       return  (TClonesArray *) fReader->GetInputEvent() ->FindListObject(aodName);
  }
  
}

//_____________________________________________________________
TObjArray *  AliAnaCaloTrackCorrBaseClass::GetCTSTracks() const 
{
  //Get list of referenced tracks from reader
  
  return fReader->GetCTSTracks(); 
  
}

//________________________________________________________________
TObjArray *  AliAnaCaloTrackCorrBaseClass::GetPHOSClusters() const 
{
  //Get list of PHOS reference caloclusters from reader
  
  return fReader->GetPHOSClusters(); 
  
}

//_________________________________________________________________
TObjArray *  AliAnaCaloTrackCorrBaseClass::GetEMCALClusters() const 
{
  //Get list of emcal referenced caloclusters from reader
  
  return fReader->GetEMCALClusters(); 
  
}

//______________________________________________________________________
TClonesArray *  AliAnaCaloTrackCorrBaseClass::GetAODCaloClusters() const 
{
  //Get list of all caloclusters in AOD output file 
  
  return fReader->GetOutputEvent()->GetCaloClusters(); 
  
}

//________________________________________________________________
TClonesArray *  AliAnaCaloTrackCorrBaseClass::GetAODTracks() const 
{
  //Get list of all tracks in AOD output file 
  
  return fReader->GetOutputEvent()->GetTracks(); 
  
}

//____________________________________________________________
TString  AliAnaCaloTrackCorrBaseClass::GetBaseParametersList()  
{
  //Put data member values in string to keep in output container
  
  TString parList ; //this will be list of parameters used for this analysis.
  const Int_t buffersize = 255;
  char onePar[buffersize] ;
  snprintf(onePar,buffersize,"--- AliAnaCaloTrackCorrBaseClass ---\n") ;
  parList+=onePar ;	
  snprintf(onePar,buffersize,"Minimal P_t: %2.2f ; Max\n", fMinPt) ;
  parList+=onePar ;
  snprintf(onePar,buffersize,"Minimal P_t: %2.2f ; Max\n", fMaxPt) ;
  parList+=onePar ;
  snprintf(onePar,buffersize,"|t_{1}-t_{2}| < %2.2f ; Max\n", fPairTimeCut) ;
  parList+=onePar ;
  snprintf(onePar,buffersize,"fDataMC =%d (Check MC information, on/off) \n",fDataMC) ;
  parList+=onePar ;
  snprintf(onePar,buffersize,"fCheckFidCut=%d (Check Fiducial cut selection on/off) \n",fCheckFidCut) ;
  parList+=onePar ;
  snprintf(onePar,buffersize,"fCheckCaloPID =%d (Use Bayesian PID in calorimetes, on/off) \n",fCheckCaloPID) ;
  parList+=onePar ;
  snprintf(onePar,buffersize,"fRecalculateCaloPID  =%d (Calculate PID from shower/tof/tracking parameters, on/off) \n",fRecalculateCaloPID) ;
  parList+=onePar ;
  snprintf(onePar,buffersize,"fInputAODName  =%s Input AOD name \n",fInputAODName.Data()) ;
  parList+=onePar ;	
  if(fNewAOD){
    snprintf(onePar,buffersize,"fOutputAODName  =%s Output AOD name \n",fOutputAODName.Data()) ;
    parList+=onePar ;	
    snprintf(onePar,buffersize,"fOutputAODClassName  =%s Output AOD class name \n",fOutputAODClassName.Data()) ;
    parList+=onePar ;	
  }
  snprintf(onePar,buffersize,"fAODObjArrayName  =%s Reference arrays in AOD name \n",fAODObjArrayName.Data()) ;
  parList+=onePar ;	
  snprintf(onePar,buffersize,"fAddToHistogramsName  =%s String added to beginning of histograms name \n",fAddToHistogramsName.Data()) ;
  parList+=onePar ;	
	
  return parList; 
  
}

//_____________________________________________________________________
TClonesArray * AliAnaCaloTrackCorrBaseClass::GetCreateOutputAODBranch() 
{
  //Create AOD branch filled in the analysis
  
  printf("Create AOD branch of %s objects and with name < %s >\n",
         fOutputAODClassName.Data(),fOutputAODName.Data()) ;
  
  TClonesArray * aodBranch = new TClonesArray(fOutputAODClassName, 0);
  aodBranch->SetName(fOutputAODName);
  return aodBranch ;
  
}

//________________________________________________________
Int_t AliAnaCaloTrackCorrBaseClass::GetEventNumber() const 
{
  //Get current event number
  
  return fReader->GetEventNumber() ; 
  
}


//__________________________________________________________
AliStack *  AliAnaCaloTrackCorrBaseClass::GetMCStack() const 
{
  //Get stack pointer from reader
  
  return fReader->GetStack(); 
  
}
//____________________________________________________________
AliHeader *  AliAnaCaloTrackCorrBaseClass::GetMCHeader() const
{
  //Get header pointer from reader
  
  return fReader->GetHeader(); 
  
}

//____________________________________________________________________________
AliGenEventHeader *  AliAnaCaloTrackCorrBaseClass::GetMCGenEventHeader() const
{
  //Get GenEventHeader pointer from reader
  
  return fReader->GetGenEventHeader(); 
  
}

//_________________________________________________
void AliAnaCaloTrackCorrBaseClass::InitParameters()
{ 
  //Initialize the parameters of the analysis.
  fDataMC              = kFALSE;
  fDebug               = -1;
  fCheckCaloPID        = kTRUE ;
  fCheckFidCut         = kFALSE ;
  fRecalculateCaloPID  = kFALSE ;
  fMinPt               = 0.1  ; //Min pt in particle analysis
  fMaxPt               = 300. ; //Max pt in particle analysis
  fMultiBin            = 1;
  fNZvertBin           = 1;
  fNrpBin              = 1;
  fMaxMulti            = 1000;
  fMinMulti            = 0;
  fUseSelectEvent      = kFALSE ;
  
  //fReader    = new AliCaloTrackReader(); //Initialized in maker
  //fCaloUtils = new AliCalorimeterUtils();//Initialized in maker
  
  fNewAOD              = kFALSE ;
  fOutputAODName       = "CaloTrackCorr";
  fOutputAODClassName  = "AliAODPWG4Particle";
  fInputAODName        = "CaloTrackCorr";
  fAddToHistogramsName = "";
  fAODObjArrayName     = "Ref";
  
}

//__________________________________________________________________
void AliAnaCaloTrackCorrBaseClass::Print(const Option_t * opt) const
{
  //Print some relevant parameters set for the analysis
  
  if(! opt)
    return;
	
  printf("New AOD:            =     %d\n",fNewAOD);
  printf("Input AOD name:     =     %s\n",fInputAODName.Data());
  printf("Output AOD name:    =     %s\n",fOutputAODName.Data());
  printf("Output AOD Class name: =  %s\n",fOutputAODClassName.Data());
  printf("Min Photon pT       =     %2.2f\n",  fMinPt) ;
  printf("Max Photon pT       =     %3.2f\n",  fMaxPt) ;
  printf("Check PID           =     %d\n",     fCheckCaloPID) ;
  printf("Recalculate PID     =     %d\n",     fRecalculateCaloPID) ;
  printf("Check Fiducial cut  =     %d\n",     fCheckFidCut) ;
  printf("Check MC labels     =     %d\n",     fDataMC);
  printf("Make plots?         =     %d \n",    fMakePlots); 	
  printf("Debug Level         =     %d\n",     fDebug);
  
  printf("Name of reference array      : %s\n", fAODObjArrayName.Data());	
  printf("String added histograms name : %s\n", fAddToHistogramsName.Data());
	
  printf("    \n") ;
  
}
Commit	Line	Data
	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16	//_________________________________________________________________________
	17	// Base class for CaloTrackCorr analysis algorithms
	18	//-- Author: Gustavo Conesa (LNF-INFN, LPSC-Grenoble)
	19	//
	20	//
	21	//_________________________________________________________________________
	22
	23
	24	// --- ROOT system ---
	25	#include <TClonesArray.h>
	26	//#include <Riostream.h>
	27
	28	//---- AliRoot system ----
	29	#include "AliAnaCaloTrackCorrBaseClass.h"
	30	#include "AliCaloTrackReader.h"
	31	#include "AliCalorimeterUtils.h"
	32	#include "AliCaloPID.h"
	33	#include "AliFiducialCut.h"
	34	#include "AliIsolationCut.h"
	35	#include "AliMCAnalysisUtils.h"
	36	#include "AliNeutralMesonSelection.h"
	37	#include "AliVCaloCells.h"
	38	#include "AliAODEvent.h"
	39	#include "AliAODHandler.h"
	40	#include "AliAnalysisManager.h"
	41	#include "AliAODPWG4Particle.h"
	42
	43	ClassImp(AliAnaCaloTrackCorrBaseClass)
	44
	45
	46	//__________________________________________________________
	47	AliAnaCaloTrackCorrBaseClass::AliAnaCaloTrackCorrBaseClass() :
	48	TObject(),
	49	fDataMC(0), fDebug(0), fCheckFidCut(0),
	50	fCheckCaloPID(0), fRecalculateCaloPID(0),
	51	fMinPt(0), fMaxPt(0), fPairTimeCut(200),
	52	fMultiBin(0), fNZvertBin(0),
	53	fNrpBin(0), fNCentrBin(0),
	54	fNmaxMixEv(0), fMaxMulti(0), fMinMulti(0),
	55	fUseSelectEvent(kFALSE), fMakePlots(kFALSE),
	56	fInputAODBranch(0x0), fInputAODName(""),
	57	fOutputAODBranch(0x0), fNewAOD(kFALSE),
	58	fOutputAODName(""), fOutputAODClassName(""),
	59	fAODObjArrayName(""), fAddToHistogramsName(""),
	60	fCaloPID(0x0), fCaloUtils(0x0),
	61	fFidCut(0x0), fHisto(0x0),
	62	fIC(0x0), fMCUtils(0x0),
	63	fNMS(0x0), fReader(0x0)
	64	{
	65	//Default Ctor
	66
	67	//Initialize parameters
	68	InitParameters();
	69	}
	70
	71	//___________________________________________________________
	72	AliAnaCaloTrackCorrBaseClass::~AliAnaCaloTrackCorrBaseClass()
	73	{
	74	// Remove all pointers except analysis output pointers.
	75
	76	//delete fCaloUtils ; //Already deleted in maker
	77	//delete fReader ; //Already deleted in maker
	78
	79	delete fCaloPID ;
	80	delete fFidCut ;
	81	delete fIC ;
	82	delete fMCUtils ;
	83	delete fNMS ;
	84	delete fHisto ;
	85	}
	86
	87	//______________________________________________________________________
	88	void AliAnaCaloTrackCorrBaseClass::AddAODParticle(AliAODPWG4Particle pc)
	89	{
	90	//Put AOD calo cluster in the AODParticleCorrelation array
	91
	92	if(fOutputAODBranch){
	93
	94	Int_t i = fOutputAODBranch->GetEntriesFast();
	95	//new((*fOutputAODBranch)[i]) AliAODPWG4Particle(pc);
	96	if(strcmp(fOutputAODBranch->GetClass()->GetName(),"AliAODPWG4Particle")==0)
	97	new((*fOutputAODBranch)[i]) AliAODPWG4Particle(pc);
	98	else if(strcmp(fOutputAODBranch->GetClass()->GetName(),"AliAODPWG4ParticleCorrelation")==0)
	99	new((*fOutputAODBranch)[i]) AliAODPWG4ParticleCorrelation(pc);
	100	else {
	101	printf("AliAnaCaloTrackCorrBaseClass::AddAODParticle() - Cannot add an object of type < %s >, to the AOD TClonesArray \n",
	102	fOutputAODBranch->GetClass()->GetName());
	103	abort();
	104	}
	105	}
	106	else {
	107	printf(" AliAnaCaloTrackCorrBaseClass::AddAODParticle() - No AOD branch available!!!\n");
	108	abort();
	109	}
	110
	111	}
	112
	113	//_______________________________________________________________________________
	114	Int_t AliAnaCaloTrackCorrBaseClass::CheckMixedEventVertex(const Int_t caloLabel,
	115	const Int_t trackLabel)
	116	{
	117	// Check vertex in mixed events
	118
	119	if (!GetMixedEvent()) return 1; // Not mixed event continue normal processing
	120
	121	Int_t evt = -1;
	122
	123	if (caloLabel >= 0 )
	124	{
	125	evt = GetMixedEvent()->EventIndexForCaloCluster(caloLabel) ;
	126	}
	127	else if(trackLabel >= 0 )
	128	{
	129	evt = GetMixedEvent()->EventIndex(trackLabel) ;
	130	}
	131	else
	132	return 0; // go to next entry in the particle list
	133
	134	if(evt == -1)
	135	return 0 ; // to content coverity
	136
	137	if (TMath::Abs(GetVertex(evt)[2]) > GetZvertexCut()) return -1; // Vertex out of range process next event
	138
	139	return 1 ; // continue processing normally
	140
	141	}
	142
	143	//________________________________________________________________
	144	void AliAnaCaloTrackCorrBaseClass::ConnectInputOutputAODBranches()
	145	{
	146	//Recover ouput and input AOD pointers for each event in the maker
	147
	148	//Delta AODs
	149	if(fDebug > 3) printf("AliAnaCaloTrackCorrBaseClass::ConnectInputOutputAODBranches() - Connect Input with name: <%s>; Connect output with name <%s>\n",fInputAODName.Data(),fOutputAODName.Data());
	150
	151	//Get the AOD handler, if output AOD is created use it, if not get the branches from the input which should be deltaAODs
	152	AliAODHandler* aodHandler = 0x0;
	153	Bool_t outAOD = kFALSE;
	154	if((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler()) outAOD = kTRUE;
	155	if(outAOD) aodHandler = (AliAODHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
	156	else aodHandler = (AliAODHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler());
	157
	158	if(!GetReader()->WriteDeltaAODToFile())
	159	{
	160	fOutputAODBranch = (TClonesArray *) (fReader->GetAODBranchList())->FindObject(fOutputAODName);
	161	fInputAODBranch = (TClonesArray *) (fReader->GetAODBranchList())->FindObject(fInputAODName);
	162	}
	163	else if (aodHandler->GetExtensions()) {
	164
	165	AliAODExtension ext = (AliAODExtension)aodHandler->GetExtensions()->FindObject(GetReader()->GetDeltaAODFileName());
	166	if(ext){
	167	AliAODEvent *aodEvent = ext->GetAOD();
	168	if(fNewAOD)fOutputAODBranch = (TClonesArray*) aodEvent->FindListObject(fOutputAODName);
	169	fInputAODBranch = (TClonesArray*) aodEvent->FindListObject(fInputAODName);
	170	if(!fOutputAODBranch && fNewAOD) fOutputAODBranch = (TClonesArray *) fReader->GetOutputEvent()->FindListObject(fOutputAODName);
	171	if(!fInputAODBranch) fInputAODBranch = (TClonesArray *) fReader->GetOutputEvent()->FindListObject(fInputAODName);
	172	}
	173	else{//If no Delta AODs, kept in standard branch, to revise.
	174	if(fNewAOD && fReader->GetOutputEvent()) {
	175	fOutputAODBranch = (TClonesArray *) fReader->GetOutputEvent()->FindListObject(fOutputAODName);
	176	fInputAODBranch = (TClonesArray *) fReader->GetOutputEvent()->FindListObject(fInputAODName);
	177	}
	178	else {
	179	fInputAODBranch = (TClonesArray *) fReader->GetInputEvent()->FindListObject(fInputAODName);
	180	if(!fInputAODBranch && fReader->GetOutputEvent() )
	181	fInputAODBranch = (TClonesArray *) fReader->GetOutputEvent()->FindListObject(fInputAODName);//Try the output event.
	182	}
	183	}
	184	}
	185	else{ //If no Delta AODs, kept in standard branch
	186	if(fNewAOD && fReader->GetOutputEvent()) {
	187	fOutputAODBranch = (TClonesArray *) fReader->GetOutputEvent()->FindListObject(fOutputAODName);
	188	fInputAODBranch = (TClonesArray *) fReader->GetOutputEvent()->FindListObject(fInputAODName);
	189	}
	190	else{
	191	fInputAODBranch = (TClonesArray *) fReader->GetInputEvent()->FindListObject(fInputAODName);
	192	if(!fInputAODBranch && fReader->GetOutputEvent())
	193	fInputAODBranch = (TClonesArray *) fReader->GetOutputEvent()->FindListObject(fInputAODName);//Try the output event.
	194	}
	195	}
	196
	197	if(GetDebug() > 1){
	198	if(fNewAOD && !fOutputAODBranch)
	199	printf(" AliAnaCaloTrackCorrBaseClass::ConnectInputOutputAODBranches() - Output Branch <%s>, not found!\n",fOutputAODName.Data());
	200	if(!fNewAOD && !fInputAODBranch)
	201	printf(" AliAnaCaloTrackCorrBaseClass::ConnectInputOutputAODBranches() - Input Branch <%s>, not found!\n",fInputAODName.Data());
	202	}
	203	}
	204
	205	//__________________________________________________________________________________________
	206	AliVCluster * AliAnaCaloTrackCorrBaseClass::FindCluster(TObjArray* clusters, const Int_t id,
	207	Int_t & iclus, const Int_t first)
	208	{
	209	// Given the cluster ID stored in AliAODPWG4Particle, get the originator cluster and its index in the array
	210
	211	if(!clusters) return 0x0;
	212
	213	for(iclus = first; iclus < clusters->GetEntriesFast(); iclus++){
	214	AliVCluster cluster= dynamic_cast<AliVCluster> (clusters->At(iclus));
	215	if(cluster){
	216	if (cluster->GetID()==id) {
	217	return cluster;
	218	}
	219	}
	220	}// calorimeter clusters loop
	221
	222	return 0x0;
	223
	224	}
	225
	226	//______________________________________________________________________________
	227	TClonesArray * AliAnaCaloTrackCorrBaseClass::GetAODBranch(TString aodName) const
	228	{
	229	//Recover ouput and input AOD pointers for each event in the maker
	230
	231	//Delta AODs
	232	if(fDebug > 3) printf("AliAnaCaloTrackCorrBaseClass::GetAODBranch() - Get Input Branch with name: <%s>; \n",aodName.Data());
	233
	234	//Get the AOD handler, if output AOD is created use it, if not get the branches from the input which should be deltaAODs
	235	AliAODHandler* aodHandler = 0x0;
	236	Bool_t outAOD = kFALSE;
	237	if((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler()) outAOD = kTRUE;
	238	if(outAOD) aodHandler = (AliAODHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
	239	else aodHandler = (AliAODHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler());
	240
	241	if(!GetReader()->WriteDeltaAODToFile())
	242	{
	243	return (TClonesArray *) (fReader->GetAODBranchList())->FindObject(aodName);
	244	}
	245	else if (aodHandler->GetExtensions())
	246	{
	247	AliAODExtension ext = (AliAODExtension)aodHandler->GetExtensions()->FindObject(GetReader()->GetDeltaAODFileName());
	248	if(ext){
	249	AliAODEvent *aodEvent = ext->GetAOD();
	250	TClonesArray * aodbranch = (TClonesArray*) aodEvent->FindListObject(aodName);
	251	if(aodbranch) return aodbranch;
	252	else {
	253	if(outAOD) return (TClonesArray *) fReader->GetOutputEvent()->FindListObject(aodName);
	254	else return (TClonesArray *) fReader->GetInputEvent() ->FindListObject(aodName);
	255	}
	256	}
	257	else{//If no Delta AODs, kept in standard branch, to revise.
	258	if(outAOD) return (TClonesArray *) fReader->GetOutputEvent()->FindListObject(aodName);
	259	else return (TClonesArray *) fReader->GetInputEvent() ->FindListObject(aodName);
	260	}
	261	}
	262	else{ //If no Delta AODs, kept in standard branch, to revise.
	263	if(outAOD) return (TClonesArray *) fReader->GetOutputEvent()->FindListObject(aodName);
	264	else return (TClonesArray *) fReader->GetInputEvent() ->FindListObject(aodName);
	265	}
	266
	267	}
	268
	269	//_____________________________________________________________
	270	TObjArray * AliAnaCaloTrackCorrBaseClass::GetCTSTracks() const
	271	{
	272	//Get list of referenced tracks from reader
	273
	274	return fReader->GetCTSTracks();
	275
	276	}
	277
	278	//________________________________________________________________
	279	TObjArray * AliAnaCaloTrackCorrBaseClass::GetPHOSClusters() const
	280	{
	281	//Get list of PHOS reference caloclusters from reader
	282
	283	return fReader->GetPHOSClusters();
	284
	285	}
	286
	287	//_________________________________________________________________
	288	TObjArray * AliAnaCaloTrackCorrBaseClass::GetEMCALClusters() const
	289	{
	290	//Get list of emcal referenced caloclusters from reader
	291
	292	return fReader->GetEMCALClusters();
	293
	294	}
	295
	296	//______________________________________________________________________
	297	TClonesArray * AliAnaCaloTrackCorrBaseClass::GetAODCaloClusters() const
	298	{
	299	//Get list of all caloclusters in AOD output file
	300
	301	return fReader->GetOutputEvent()->GetCaloClusters();
	302
	303	}
	304
	305	//________________________________________________________________
	306	TClonesArray * AliAnaCaloTrackCorrBaseClass::GetAODTracks() const
	307	{
	308	//Get list of all tracks in AOD output file
	309
	310	return fReader->GetOutputEvent()->GetTracks();
	311
	312	}
	313
	314	//____________________________________________________________
	315	TString AliAnaCaloTrackCorrBaseClass::GetBaseParametersList()
	316	{
	317	//Put data member values in string to keep in output container
	318
	319	TString parList ; //this will be list of parameters used for this analysis.
	320	const Int_t buffersize = 255;
	321	char onePar[buffersize] ;
	322	snprintf(onePar,buffersize,"--- AliAnaCaloTrackCorrBaseClass ---\n") ;
	323	parList+=onePar ;
	324	snprintf(onePar,buffersize,"Minimal P_t: %2.2f ; Max\n", fMinPt) ;
	325	parList+=onePar ;
	326	snprintf(onePar,buffersize,"Minimal P_t: %2.2f ; Max\n", fMaxPt) ;
	327	parList+=onePar ;
	328	snprintf(onePar,buffersize,"\|t_{1}-t_{2}\| < %2.2f ; Max\n", fPairTimeCut) ;
	329	parList+=onePar ;
	330	snprintf(onePar,buffersize,"fDataMC =%d (Check MC information, on/off) \n",fDataMC) ;
	331	parList+=onePar ;
	332	snprintf(onePar,buffersize,"fCheckFidCut=%d (Check Fiducial cut selection on/off) \n",fCheckFidCut) ;
	333	parList+=onePar ;
	334	snprintf(onePar,buffersize,"fCheckCaloPID =%d (Use Bayesian PID in calorimetes, on/off) \n",fCheckCaloPID) ;
	335	parList+=onePar ;
	336	snprintf(onePar,buffersize,"fRecalculateCaloPID =%d (Calculate PID from shower/tof/tracking parameters, on/off) \n",fRecalculateCaloPID) ;
	337	parList+=onePar ;
	338	snprintf(onePar,buffersize,"fInputAODName =%s Input AOD name \n",fInputAODName.Data()) ;
	339	parList+=onePar ;
	340	if(fNewAOD){
	341	snprintf(onePar,buffersize,"fOutputAODName =%s Output AOD name \n",fOutputAODName.Data()) ;
	342	parList+=onePar ;
	343	snprintf(onePar,buffersize,"fOutputAODClassName =%s Output AOD class name \n",fOutputAODClassName.Data()) ;
	344	parList+=onePar ;
	345	}
	346	snprintf(onePar,buffersize,"fAODObjArrayName =%s Reference arrays in AOD name \n",fAODObjArrayName.Data()) ;
	347	parList+=onePar ;
	348	snprintf(onePar,buffersize,"fAddToHistogramsName =%s String added to beginning of histograms name \n",fAddToHistogramsName.Data()) ;
	349	parList+=onePar ;
	350
	351	return parList;
	352
	353	}
	354
	355	//_____________________________________________________________________
	356	TClonesArray * AliAnaCaloTrackCorrBaseClass::GetCreateOutputAODBranch()
	357	{
	358	//Create AOD branch filled in the analysis
	359
	360	printf("Create AOD branch of %s objects and with name < %s >\n",
	361	fOutputAODClassName.Data(),fOutputAODName.Data()) ;
	362
	363	TClonesArray * aodBranch = new TClonesArray(fOutputAODClassName, 0);
	364	aodBranch->SetName(fOutputAODName);
	365	return aodBranch ;
	366
	367	}
	368
	369	//________________________________________________________
	370	Int_t AliAnaCaloTrackCorrBaseClass::GetEventNumber() const
	371	{
	372	//Get current event number
	373
	374	return fReader->GetEventNumber() ;
	375
	376	}
	377
	378
	379	//__________________________________________________________
	380	AliStack * AliAnaCaloTrackCorrBaseClass::GetMCStack() const
	381	{
	382	//Get stack pointer from reader
	383
	384	return fReader->GetStack();
	385
	386	}
	387	//____________________________________________________________
	388	AliHeader * AliAnaCaloTrackCorrBaseClass::GetMCHeader() const
	389	{
	390	//Get header pointer from reader
	391
	392	return fReader->GetHeader();
	393
	394	}
	395
	396	//____________________________________________________________________________
	397	AliGenEventHeader * AliAnaCaloTrackCorrBaseClass::GetMCGenEventHeader() const
	398	{
	399	//Get GenEventHeader pointer from reader
	400
	401	return fReader->GetGenEventHeader();
	402
	403	}
	404
	405	//_________________________________________________
	406	void AliAnaCaloTrackCorrBaseClass::InitParameters()
	407	{
	408	//Initialize the parameters of the analysis.
	409	fDataMC = kFALSE;
	410	fDebug = -1;
	411	fCheckCaloPID = kTRUE ;
	412	fCheckFidCut = kFALSE ;
	413	fRecalculateCaloPID = kFALSE ;
	414	fMinPt = 0.1 ; //Min pt in particle analysis
	415	fMaxPt = 300. ; //Max pt in particle analysis
	416	fMultiBin = 1;
	417	fNZvertBin = 1;
	418	fNrpBin = 1;
	419	fMaxMulti = 1000;
	420	fMinMulti = 0;
	421	fUseSelectEvent = kFALSE ;
	422
	423	//fReader = new AliCaloTrackReader(); //Initialized in maker
	424	//fCaloUtils = new AliCalorimeterUtils();//Initialized in maker
	425
	426	fNewAOD = kFALSE ;
	427	fOutputAODName = "CaloTrackCorr";
	428	fOutputAODClassName = "AliAODPWG4Particle";
	429	fInputAODName = "CaloTrackCorr";
	430	fAddToHistogramsName = "";
	431	fAODObjArrayName = "Ref";
	432
	433	}
	434
	435	//__________________________________________________________________
	436	void AliAnaCaloTrackCorrBaseClass::Print(const Option_t * opt) const
	437	{
	438	//Print some relevant parameters set for the analysis
	439
	440	if(! opt)
	441	return;
	442
	443	printf("New AOD: = %d\n",fNewAOD);
	444	printf("Input AOD name: = %s\n",fInputAODName.Data());
	445	printf("Output AOD name: = %s\n",fOutputAODName.Data());
	446	printf("Output AOD Class name: = %s\n",fOutputAODClassName.Data());
	447	printf("Min Photon pT = %2.2f\n", fMinPt) ;
	448	printf("Max Photon pT = %3.2f\n", fMaxPt) ;
	449	printf("Check PID = %d\n", fCheckCaloPID) ;
	450	printf("Recalculate PID = %d\n", fRecalculateCaloPID) ;
	451	printf("Check Fiducial cut = %d\n", fCheckFidCut) ;
	452	printf("Check MC labels = %d\n", fDataMC);
	453	printf("Make plots? = %d \n", fMakePlots);
	454	printf("Debug Level = %d\n", fDebug);
	455
	456	printf("Name of reference array : %s\n", fAODObjArrayName.Data());
	457	printf("String added histograms name : %s\n", fAddToHistogramsName.Data());
	458
	459	printf(" \n") ;
	460
	461	}
	462
	463
	464