1 #ifndef ALICALOTRACKREADER_H
2 #define ALICALOTRACKREADER_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
7 //_________________________________________________________________________
8 // Base class for reading data: MonteCarlo, ESD or AOD, of PHOS EMCAL and
9 // Central Barrel Tracking detectors.
10 // Not all MC particles/tracks/clusters are kept, some kinematical restrictions are done.
11 // Mother class of : AliCaloTrackESDReader: Fills ESD data in 3 TObjArrays (PHOS, EMCAL, CTS)
12 // : AliCaloTrackMCReader: Fills Kinematics data in 3 TObjArrays (PHOS, EMCAL, CTS)
13 // : AliCaloTrackAODReader: Fills AOD data in 3 TObjArrays (PHOS, EMCAL, CTS)
15 // This part is commented: Mixing analysis can be done, input AOD with events
16 // is opened in the AliCaloTrackReader::Init()
18 // -- Author: Gustavo Conesa (INFN-LNF)
20 // --- ROOT system ---
27 //--- ANALYSIS system ---
28 #include "AliVCaloCells.h"
31 class AliGenEventHeader ;
32 #include "AliVEvent.h"
37 class AliESDtrackCuts;
39 class AliTriggerAnalysis;
42 #include "AliCalorimeterUtils.h"
43 #include "AliFiducialCut.h"
45 class AliCaloTrackReader : public TObject {
48 AliCaloTrackReader() ; // ctor
49 virtual ~AliCaloTrackReader() ;//virtual dtor
51 AliCaloTrackReader(const AliCaloTrackReader & g) ; // cpy ctor
52 AliCaloTrackReader & operator = (const AliCaloTrackReader & g) ;//cpy assignment
56 //--------------------------------
58 //--------------------------------
61 virtual void InitParameters();
62 virtual void Print(const Option_t * opt) const;
63 virtual void ResetLists();
65 virtual Int_t GetDebug() const { return fDebug ; }
66 virtual void SetDebug(Int_t d) { fDebug = d ; }
68 enum inputDataType {kESD, kAOD, kMC};
69 virtual Int_t GetDataType() const { return fDataType ; }
70 virtual void SetDataType(Int_t data ) { fDataType = data ; }
72 virtual Int_t GetEventNumber() const { return fEventNumber ; }
73 //virtual TString GetCurrentFileName() const { return fCurrentFileName ; }
75 TString GetTaskName() const { return fTaskName ; }
76 void SetTaskName(TString name) { fTaskName = name ; }
78 //---------------------------------------
79 //Input/output event setters and getters
80 //---------------------------------------
81 virtual void SetInputEvent(AliVEvent* const input) ;
82 virtual void SetOutputEvent(AliAODEvent* const aod) { fOutputEvent = aod ; }
83 virtual void SetMC(AliMCEvent* const mc) { fMC = mc ; }
84 virtual void SetInputOutputMCEvent(AliVEvent* /*esd*/, AliAODEvent* /*aod*/, AliMCEvent* /*mc*/) { ; }
87 virtual TList * GetAODBranchList() const { return fAODBranchList ; }
88 void SetDeltaAODFileName(TString name ) { fDeltaAODFileName = name ; }
89 TString GetDeltaAODFileName() const { return fDeltaAODFileName ; }
90 void SwitchOnWriteDeltaAOD() { fWriteOutputDeltaAOD = kTRUE ; }
91 void SwitchOffWriteDeltaAOD() { fWriteOutputDeltaAOD = kFALSE; }
92 Bool_t WriteDeltaAODToFile() const { return fWriteOutputDeltaAOD ; }
94 //------------------------------------------------------------
95 //Clusters/Tracks arrays filtering/filling methods and switchs
96 //------------------------------------------------------------
98 //Minimum pt setters and getters
99 virtual Float_t GetEMCALPtMin() const { return fEMCALPtMin ; }
100 virtual Float_t GetPHOSPtMin() const { return fPHOSPtMin ; }
101 virtual Float_t GetCTSPtMin() const { return fCTSPtMin ; }
103 virtual void SetEMCALPtMin(Float_t pt) { fEMCALPtMin = pt ; }
104 virtual void SetPHOSPtMin(Float_t pt) { fPHOSPtMin = pt ; }
105 virtual void SetCTSPtMin(Float_t pt) { fCTSPtMin = pt ; }
108 virtual AliFiducialCut * GetFiducialCut() { if(!fFiducialCut) fFiducialCut = new AliFiducialCut();
109 return fFiducialCut ;}
110 virtual void SetFiducialCut(AliFiducialCut * const fc) { fFiducialCut = fc ;}
111 virtual Bool_t IsFiducialCutOn() const { return fCheckFidCut ; }
112 virtual void SwitchOnFiducialCut() { fCheckFidCut = kTRUE; fFiducialCut = new AliFiducialCut();}
113 virtual void SwitchOffFiducialCut() { fCheckFidCut = kFALSE ; }
116 Bool_t IsEMCALCluster(AliVCluster *clus) const;
117 Bool_t IsPHOSCluster (AliVCluster *clus) const;
118 //Patch for cluster origin for Old AODs implementation
119 void SwitchOnOldAODs() { fOldAOD = kTRUE ; }
120 void SwitchOffOldAODs() { fOldAOD = kFALSE ; }
122 // Cluster/track/cells switchs
123 Bool_t IsCTSSwitchedOn() const { return fFillCTS ; }
124 void SwitchOnCTS() { fFillCTS = kTRUE ; }
125 void SwitchOffCTS() { fFillCTS = kFALSE ; }
127 Bool_t IsEMCALSwitchedOn() const { return fFillEMCAL ; }
128 void SwitchOnEMCAL() { fFillEMCAL = kTRUE ; }
129 void SwitchOffEMCAL() { fFillEMCAL = kFALSE ; }
131 Bool_t IsPHOSSwitchedOn() const { return fFillPHOS ; }
132 void SwitchOnPHOS() { fFillPHOS = kTRUE ; }
133 void SwitchOffPHOS() { fFillPHOS = kFALSE ; }
135 Bool_t IsEMCALCellsSwitchedOn() const { return fFillEMCALCells ; }
136 void SwitchOnEMCALCells() { fFillEMCALCells = kTRUE ; }
137 void SwitchOffEMCALCells() { fFillEMCALCells = kFALSE ; }
139 Bool_t IsPHOSCellsSwitchedOn() const { return fFillPHOSCells ; }
140 void SwitchOnPHOSCells() { fFillPHOSCells = kTRUE ; }
141 void SwitchOffPHOSCells() { fFillPHOSCells = kFALSE ; }
143 // Filling/ filtering / detector information access methods
144 virtual Bool_t FillInputEvent(const Int_t iEntry, const char *currentFileName) ;
145 virtual void FillInputCTS() ;
146 virtual void FillInputEMCAL() ;
147 virtual void FillInputEMCALAlgorithm(AliVCluster * clus, const Int_t iclus) ;
148 virtual void FillInputPHOS() ;
149 virtual void FillInputEMCALCells() ;
150 virtual void FillInputPHOSCells() ;
151 virtual void FillInputVZERO() ;
153 Int_t GetV0Signal(Int_t i) const { return fV0ADC[i] ; }
154 Int_t GetV0Multiplicity(Int_t i) const { return fV0Mul[i] ; }
156 void SetEMCALClusterListName(TString &name) {fEMCALClustersListName = name ; }
158 // Arrayes with clusters/track/cells access method
159 virtual TObjArray* GetCTSTracks() const { return fCTSTracks ; }
160 virtual TObjArray* GetEMCALClusters() const { return fEMCALClusters ; }
161 virtual TObjArray* GetPHOSClusters() const { return fPHOSClusters ; }
162 virtual AliVCaloCells* GetEMCALCells() const { return fEMCALCells ; }
163 virtual AliVCaloCells* GetPHOSCells() const { return fPHOSCells ; }
165 //Methods for mixing with external input file (AOD)
166 //virtual TTree* GetSecondInputAODTree() const {return fSecondInputAODTree ; }
167 //virtual void SetSecondInputAODTree(TTree * tree) {fSecondInputAODTree = tree ;
168 // fSecondInputAODEvent->ReadFromTree(tree);}//Connect tree and AOD event.
170 //virtual AliAODEvent* GetSecondInputAODEvent() const { return fSecondInputAODEvent ; }
172 //TString GetSecondInputFileName() const {return fSecondInputFileName ; }
173 //void SetSecondInputFileName(TString name) { fSecondInputFileName = name ; }
175 //Int_t GetSecondInputFirstEvent() const {return fSecondInputFirstEvent ; }
176 //void SetSecondInputFirstEvent(Int_t iEvent0) { fSecondInputFirstEvent = iEvent0 ; }
178 // Int_t GetAODCTSNormalInputEntries() {if(!fSecondInputAODTree) { fAODCTSNormalInputEntries = fAODCTS->GetEntriesFast() ;}
179 // return fAODCTSNormalInputEntries ; }
180 // Int_t GetAODEMCALNormalInputEntries() {if(!fSecondInputAODTree) { fAODEMCALNormalInputEntries = fAODEMCAL->GetEntriesFast();}
181 // return fAODEMCALNormalInputEntries ; }
182 // Int_t GetAODPHOSNormalInputEntries() {if(!fSecondInputAODTree) { fAODPHOSNormalInputEntries = fAODPHOS->GetEntriesFast() ;}
183 // return fAODPHOSNormalInputEntries ; }
186 //-------------------------------------
187 // Event/track selection methods
188 //-------------------------------------
190 void SetFiredTriggerClassName(TString name ) { fFiredTriggerClassName = name ; }
191 TString GetFiredTriggerClassName() const { return fFiredTriggerClassName ; }
192 virtual TString GetFiredTriggerClasses() { return "" ; } // look the ESD/AOD reader
194 void SwitchOnEventSelection() { fDoEventSelection = kTRUE ; }
195 void SwitchOffEventSelection() { fDoEventSelection = kFALSE ; }
196 Bool_t IsEventSelectionDone() const { return fDoEventSelection ; }
198 void SwitchOnV0ANDSelection() { fDoV0ANDEventSelection = kTRUE ; }
199 void SwitchOffV0ANDSelection() { fDoV0ANDEventSelection = kFALSE ; }
200 Bool_t IsV0ANDEventSelectionDone() const { return fDoV0ANDEventSelection ; }
203 ULong_t GetTrackStatus() const {return fTrackStatus ; }
204 void SetTrackStatus(ULong_t bit) { fTrackStatus = bit ; }
206 AliESDtrackCuts* GetTrackCuts() const { return fESDtrackCuts ; }
207 void SetTrackCuts(AliESDtrackCuts * cuts) { fESDtrackCuts = cuts ; }
208 Int_t GetTrackMultiplicity() const { return fTrackMult ; }
209 Float_t GetTrackMultiplicityEtaCut() const { return fTrackMultEtaCut ; }
210 void SetTrackMultiplicityEtaCut(Float_t eta) { fTrackMultEtaCut = eta ; }
212 // Calorimeter specific and patches
213 void AnalyzeOnlyLED() { fAnaLED = kTRUE ; }
214 void AnalyzeOnlyPhysics() { fAnaLED = kFALSE ; }
216 void SwitchOnCaloFilterPatch() { fCaloFilterPatch = kTRUE ; fFillCTS = kFALSE ; }
217 void SwitchOffCaloFilterPatch() { fCaloFilterPatch = kFALSE ; }
218 Bool_t IsCaloFilterPatchOn() const { if(fDataType == kAOD) { return fCaloFilterPatch ; }
219 else { return kFALSE ; } }
221 //-------------------------------
223 //-------------------------------
224 virtual void GetVertex(Double_t v[3]) const ;
225 virtual Double_t* GetVertex(const Int_t evtIndex) const { return fVertex[evtIndex] ; }
226 virtual void GetVertex(Double_t vertex[3], const Int_t evtIndex) const ;
227 virtual void FillVertexArray();
228 virtual Bool_t CheckForPrimaryVertex();
229 // virtual void GetSecondInputAODVertex(Double_t *) const {;}
230 virtual Float_t GetZvertexCut() const { return fZvtxCut ; } //cut on vertex position
231 virtual void SetZvertexCut(Float_t zcut=10.) { fZvtxCut=zcut ; } //cut on vertex position
233 //------------------------
235 //------------------------
236 virtual AliCentrality* GetCentrality() const { return fInputEvent->GetCentrality() ; }
237 virtual void SetCentralityClass(TString name) { fCentralityClass = name ; }
238 virtual void SetCentralityOpt(Int_t opt) { fCentralityOpt = opt ; }
239 virtual TString GetCentralityClass() const { return fCentralityClass ; }
240 virtual Int_t GetCentralityOpt() const { return fCentralityOpt ; }
241 virtual Int_t GetEventCentrality() const ;
242 virtual void SetCentralityBin(Int_t min, Int_t max) //Set the centrality bin to select the event. If used, then need to get percentile
243 { fCentralityBin[0]=min; fCentralityBin[1]=max;
244 if(min>=0 && max > 0) fCentralityOpt = 100 ; }
245 virtual Float_t GetCentralityBin(Int_t i) const { if(i < 0 || i > 1) return 0 ;
246 else return fCentralityBin[i] ; }
248 //-------------------------------------
250 //-------------------------------------
251 AliCalorimeterUtils * GetCaloUtils() const { return fCaloUtils ; }
252 void SetCaloUtils(AliCalorimeterUtils * caloutils) { fCaloUtils = caloutils ; }
254 void SwitchOnSuspiciousClustersRemoval() { fRemoveSuspiciousClusters = kTRUE ; }
255 void SwitchOffSuspiciousClustersRemoval() { fRemoveSuspiciousClusters = kFALSE ; }
256 Bool_t IsSuspiciousClustersRemovalOn() const { return fRemoveSuspiciousClusters ; }
259 void SwitchOnClusterEnergySmearing() { fSmearClusterEnergy = kTRUE ; }
260 void SwitchOffClusterEnergySmearing() { fSmearClusterEnergy = kFALSE ; }
261 Bool_t IsClusterEnergySmeared() const { return fSmearClusterEnergy ; }
262 void SetSmearingParameters(Int_t i, Float_t param) { if(i < 3)fSmearClusterParam[i] = param ; }
264 virtual Double_t GetBField() const { return fInputEvent->GetMagneticField() ; }
266 //------------------------------------------------
267 // MC analysis specific methods
268 //-------------------------------------------------
270 //Kinematics and galice.root available
271 virtual AliStack* GetStack() const ;
272 virtual AliHeader* GetHeader() const ;
273 virtual AliGenEventHeader* GetGenEventHeader() const ;
275 //Filtered kinematics in AOD
276 virtual TClonesArray* GetAODMCParticles(Int_t input = 0) const ;
277 virtual AliAODMCHeader* GetAODMCHeader(Int_t input = 0) const ;
279 virtual AliVEvent* GetInputEvent() const { return fInputEvent ; }
280 virtual AliAODEvent* GetOutputEvent() const { return fOutputEvent ; }
281 virtual AliMCEvent* GetMC() const { return fMC ; }
282 virtual AliMixedEvent* GetMixedEvent() const { return fMixedEvent ; }
283 virtual Int_t GetNMixedEvent() const { return fNMixedEvent ; }
285 void SwitchOnStack() { fReadStack = kTRUE ; }
286 void SwitchOffStack() { fReadStack = kFALSE ; }
287 void SwitchOnAODMCParticles() { fReadAODMCParticles = kTRUE ; }
288 void SwitchOffAODMCParticles() { fReadAODMCParticles = kFALSE ; }
289 Bool_t ReadStack() const { return fReadStack ; }
290 Bool_t ReadAODMCParticles() const { return fReadAODMCParticles ; }
292 //Select generated events, depending on comparison of pT hard and jets.
293 virtual Bool_t ComparePtHardAndJetPt() ;
294 virtual Bool_t IsPtHardAndJetPtComparisonSet() const { return fComparePtHardAndJetPt ; }
295 virtual void SetPtHardAndJetPtComparison(Bool_t compare) { fComparePtHardAndJetPt = compare ; }
296 virtual Float_t GetPtHardAndJetFactor() const { return fPtHardAndJetPtFactor ; }
297 virtual void SetPtHardAndJetPtFactor(Float_t factor) { fPtHardAndJetPtFactor = factor ; }
299 //MC reader methods, declared there to allow compilation, they are only used in the MC reader:
301 virtual void AddNeutralParticlesArray(TArrayI & /*array*/) { ; }
302 virtual void AddChargedParticlesArray(TArrayI & /*array*/) { ; }
303 virtual void AddStatusArray(TArrayI & /*array*/) { ; }
305 virtual void SwitchOnPi0Decay() { ; }
306 virtual void SwitchOffPi0Decay() { ; }
307 virtual void SwitchOnStatusSelection() { ; }
308 virtual void SwitchOffStatusSelection() { ; }
309 virtual void SwitchOnOverlapCheck() { ; }
310 virtual void SwitchOffOverlapCheck() { ; }
311 virtual void SwitchOnOnlyGeneratorParticles() { ; }
312 virtual void SwitchOffOnlyGeneratorParticles() { ; }
314 virtual void SetEMCALOverlapAngle(Float_t /*angle*/) { ; }
315 virtual void SetPHOSOverlapAngle(Float_t /*angle*/) { ; }
319 Int_t fEventNumber; // Event number
320 //TString fCurrentFileName;// Current file name under analysis
321 Int_t fDataType ; // Select MC:Kinematics, Data:ESD/AOD, MCData:Both
322 Int_t fDebug; // Debugging level
323 AliFiducialCut * fFiducialCut; //! Acceptance cuts
324 Bool_t fCheckFidCut ; // Do analysis for clusters in defined region
326 Bool_t fComparePtHardAndJetPt; // In MonteCarlo, jet events, reject fake events with wrong jet energy.
327 Float_t fPtHardAndJetPtFactor; // Factor between ptHard and jet pT to reject/accept event.
329 Float_t fCTSPtMin; // pT Threshold on charged particles
330 Float_t fEMCALPtMin; // pT Threshold on emcal clusters
331 Float_t fPHOSPtMin; // pT Threshold on phos clusters
333 TList * fAODBranchList ; //! List with AOD branches created and needed in analysis
334 //FIXME CHANGE NAMES, remove AOD, now they are VClusters, VTracks ...
335 TObjArray * fCTSTracks ; //! temporal array with tracks
336 TObjArray * fEMCALClusters ; //! temporal array with EMCAL CaloClusters
337 TObjArray * fPHOSClusters ; //! temporal array with PHOS CaloClusters
338 AliVCaloCells * fEMCALCells ; //! temporal array with EMCAL CaloCells
339 AliVCaloCells * fPHOSCells ; //! temporal array with PHOS CaloCells
341 AliVEvent * fInputEvent; //! pointer to esd or aod input
342 AliAODEvent * fOutputEvent; //! pointer to aod output
343 AliMCEvent * fMC; //! Monte Carlo Event Handler
345 Bool_t fFillCTS; // use data from CTS
346 Bool_t fFillEMCAL; // use data from EMCAL
347 Bool_t fFillPHOS; // use data from PHOS
348 Bool_t fFillEMCALCells; // use data from EMCAL
349 Bool_t fFillPHOSCells; // use data from PHOS
350 Bool_t fRemoveSuspiciousClusters; // Remove high energy clusters with low number of cells
351 Bool_t fSmearClusterEnergy; // Smear cluster energy, to be done only for simulated data to match real data
352 Float_t fSmearClusterParam[3]; // Smearing parameters
353 TRandom3 fRandom; // Random generator
355 // TTree * fSecondInputAODTree; // Tree with second input AOD, for mixing analysis.
356 // AliAODEvent* fSecondInputAODEvent; //! pointer to second input AOD event.
357 // TString fSecondInputFileName; // File with AOD data to mix with normal stream of data.
358 // Int_t fSecondInputFirstEvent; // First event to be considered in the mixing.
360 // Int_t fAODCTSNormalInputEntries; // Number of entries in CTS in case of standard input, larger with mixing.
361 // Int_t fAODEMCALNormalInputEntries; // Number of entries in EMCAL in case of standard input, larger with mixing.
362 // Int_t fAODPHOSNormalInputEntries; // Number of entries in PHOS in case of standard input, larger with mixing.
364 ULong_t fTrackStatus ; // Track selection bit, select tracks refitted in TPC, ITS ...
365 AliESDtrackCuts *fESDtrackCuts ; // Track cut
366 Int_t fTrackMult ; // Track multiplicity
367 Float_t fTrackMultEtaCut ; // Track multiplicity eta cut
368 Bool_t fReadStack ; // Access kine information from stack
369 Bool_t fReadAODMCParticles ; // Access kine information from filtered AOD MC particles
371 TString fDeltaAODFileName ; // Delta AOD file name
372 TString fFiredTriggerClassName; // Name of trigger event type used to do the analysis
374 Bool_t fAnaLED; // Analyze LED data only.
376 TString fTaskName; // Name of task that executes the analysis
378 AliCalorimeterUtils * fCaloUtils ; // Pointer to CalorimeterUtils
380 AliMixedEvent * fMixedEvent ; //! mixed event object. This class is not the owner
381 Int_t fNMixedEvent ; // number of events in mixed event buffer
382 Double_t ** fVertex ; //! vertex array 3 dim for each mixed event buffer
384 Bool_t fWriteOutputDeltaAOD;// Write the created delta AOD objects into file
385 Bool_t fOldAOD; // Old AODs, before revision 4.20
387 Int_t fV0ADC[2] ; // Integrated V0 signal
388 Int_t fV0Mul[2] ; // Integrated V0 Multiplicity
390 Bool_t fCaloFilterPatch; // CaloFilter patch
391 TString fEMCALClustersListName; //Alternative list of clusters produced elsewhere and not from InputEvent
392 Float_t fZvtxCut ; // Cut on vertex position
393 Bool_t fDoEventSelection; // Select events depending on V0, pileup, vertex well reconstructed, at least 1 track ...
394 Bool_t fDoV0ANDEventSelection; // Select events depending on V0, fDoEventSelection should be on
395 AliTriggerAnalysis* fTriggerAnalysis; // Access to trigger selection algorithm for V0AND calculation
398 TString fCentralityClass; // Name of selected centrality class
399 Int_t fCentralityOpt; // Option for the returned value of the centrality, possible options 5, 10, 100
400 Int_t fCentralityBin[2]; // Minimum and maximum value of the centrality for the analysis
402 ClassDef(AliCaloTrackReader,27)
406 #endif //ALICALOTRACKREADER_H