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 */
6 //_________________________________________________________________________
7 // Base class for reading data: MonteCarlo, ESD or AOD, of PHOS EMCAL and
8 // Central Barrel Tracking detectors.
9 // Not all MC particles/tracks/clusters are kept, some kinematical restrictions are done.
10 // Mother class of : AliCaloTrackESDReader: Fills ESD data in 3 TObjArrays (PHOS, EMCAL, CTS)
11 // : AliCaloTrackMCReader : Fills Kinematics data in 3 TObjArrays (PHOS, EMCAL, CTS)
12 // : AliCaloTrackAODReader: Fills AOD data in 3 TObjArrays (PHOS, EMCAL, CTS)
13 // -- Author: Gustavo Conesa (INFN-LNF)
14 //////////////////////////////////////////////////////////////////////////////
16 // --- ROOT system ---
23 //--- ANALYSIS system ---
24 #include "AliVEvent.h"
28 class AliGenEventHeader;
33 class AliESDtrackCuts;
35 class AliTriggerAnalysis;
39 // --- CaloTrackCorr / EMCAL ---
40 #include "AliFiducialCut.h"
41 class AliEMCALRecoUtils;
42 class AliCalorimeterUtils;
44 class AliCaloTrackReader : public TObject {
48 AliCaloTrackReader() ; // ctor
49 virtual ~AliCaloTrackReader() ; // virtual dtor
51 //--------------------------------
53 //--------------------------------
57 virtual void InitParameters();
59 virtual void Print(const Option_t * opt) const;
61 virtual void ResetLists();
63 virtual Int_t GetDebug() const { return fDebug ; }
64 virtual void SetDebug(Int_t d) { fDebug = d ; }
66 enum inputDataType {kESD, kAOD, kMC};
67 virtual Int_t GetDataType() const { return fDataType ; }
68 virtual void SetDataType(Int_t data ) { fDataType = data ; }
70 virtual Int_t GetEventNumber() const { return fEventNumber ; }
72 TString GetTaskName() const { return fTaskName ; }
73 void SetTaskName(TString name) { fTaskName = name ; }
75 //---------------------------------------
76 //Input/output event setters and getters
77 //---------------------------------------
78 virtual void SetInputEvent(AliVEvent* const input) ;
79 virtual void SetOutputEvent(AliAODEvent* const aod) { fOutputEvent = aod ; }
80 virtual void SetMC(AliMCEvent* const mc) { fMC = mc ; }
81 virtual void SetInputOutputMCEvent(AliVEvent* /*esd*/, AliAODEvent* /*aod*/, AliMCEvent* /*mc*/) { ; }
84 virtual TList * GetAODBranchList() const { return fAODBranchList ; }
85 void SetDeltaAODFileName(TString name ) { fDeltaAODFileName = name ; }
86 TString GetDeltaAODFileName() const { return fDeltaAODFileName ; }
87 void SwitchOnWriteDeltaAOD() { fWriteOutputDeltaAOD = kTRUE ; }
88 void SwitchOffWriteDeltaAOD() { fWriteOutputDeltaAOD = kFALSE ; }
89 Bool_t WriteDeltaAODToFile() const { return fWriteOutputDeltaAOD ; }
91 //------------------------------------------------------------
92 //Clusters/Tracks arrays filtering/filling methods and switchs
93 //------------------------------------------------------------
95 //Minimum pt setters and getters
96 Float_t GetEMCALPtMin() const { return fEMCALPtMin ; }
97 Float_t GetPHOSPtMin() const { return fPHOSPtMin ; }
98 Float_t GetCTSPtMin() const { return fCTSPtMin ; }
99 Float_t GetEMCALPtMax() const { return fEMCALPtMax ; }
100 Float_t GetPHOSPtMax() const { return fPHOSPtMax ; }
101 Float_t GetCTSPtMax() const { return fCTSPtMax ; }
103 void SetEMCALPtMin(Float_t pt) { fEMCALPtMin = pt ; }
104 void SetPHOSPtMin (Float_t pt) { fPHOSPtMin = pt ; }
105 void SetCTSPtMin (Float_t pt) { fCTSPtMin = pt ; }
107 void SetEMCALPtMax(Float_t pt) { fEMCALPtMax = pt ; }
108 void SetPHOSPtMax (Float_t pt) { fPHOSPtMax = pt ; }
109 void SetCTSPtMax (Float_t pt) { fCTSPtMax = pt ; }
111 Float_t GetEMCALEMin() const { return GetEMCALPtMin() ; }
112 Float_t GetPHOSEMin() const { return GetPHOSPtMin() ; }
113 Float_t GetEMCALEMax() const { return GetEMCALPtMax() ; }
114 Float_t GetPHOSEMax() const { return GetPHOSPtMax() ; }
116 void SetEMCALEMin (Float_t e) { SetEMCALPtMin(e) ; }
117 void SetPHOSEMin (Float_t e) { SetPHOSPtMin (e) ; }
118 void SetEMCALEMax (Float_t e) { SetEMCALPtMax(e) ; }
119 void SetPHOSEMax (Float_t e) { SetPHOSPtMax (e) ; }
121 Double_t GetEMCALTimeCutMin() const { return fEMCALTimeCutMin ; }
122 Double_t GetEMCALTimeCutMax() const { return fEMCALTimeCutMax ; }
124 void SetEMCALTimeCut(Double_t a, Double_t b) { fEMCALTimeCutMin = a;
125 fEMCALTimeCutMax = b ; } // ns
127 virtual AliFiducialCut * GetFiducialCut() {
128 if(!fFiducialCut) fFiducialCut = new AliFiducialCut();
129 return fFiducialCut ; }
130 virtual void SetFiducialCut(AliFiducialCut * const fc) { fFiducialCut = fc ; }
131 virtual Bool_t IsFiducialCutOn() const { return fCheckFidCut ; }
132 virtual void SwitchOnFiducialCut() { fCheckFidCut = kTRUE ;
133 fFiducialCut = new AliFiducialCut() ; }
134 virtual void SwitchOffFiducialCut() { fCheckFidCut = kFALSE ; }
137 Bool_t IsEMCALCluster(AliVCluster *clus) const;
138 Bool_t IsPHOSCluster (AliVCluster *clus) const;
139 //Patch for cluster origin for Old AODs implementation
140 void SwitchOnOldAODs() { fOldAOD = kTRUE ; }
141 void SwitchOffOldAODs() { fOldAOD = kFALSE ; }
143 // Cluster/track/cells switchs
144 Bool_t IsCTSSwitchedOn() const { return fFillCTS ; }
145 void SwitchOnCTS() { fFillCTS = kTRUE ; }
146 void SwitchOffCTS() { fFillCTS = kFALSE ; }
148 Bool_t IsEMCALSwitchedOn() const { return fFillEMCAL ; }
149 void SwitchOnEMCAL() { fFillEMCAL = kTRUE ; }
150 void SwitchOffEMCAL() { fFillEMCAL = kFALSE ; }
152 Bool_t IsPHOSSwitchedOn() const { return fFillPHOS ; }
153 void SwitchOnPHOS() { fFillPHOS = kTRUE ; }
154 void SwitchOffPHOS() { fFillPHOS = kFALSE ; }
156 Bool_t IsEMCALCellsSwitchedOn() const { return fFillEMCALCells ; }
157 void SwitchOnEMCALCells() { fFillEMCALCells = kTRUE ; }
158 void SwitchOffEMCALCells() { fFillEMCALCells = kFALSE ; }
160 Bool_t IsPHOSCellsSwitchedOn() const { return fFillPHOSCells ; }
161 void SwitchOnPHOSCells() { fFillPHOSCells = kTRUE ; }
162 void SwitchOffPHOSCells() { fFillPHOSCells = kFALSE ; }
164 Bool_t AreClustersRecalculated() const { return fRecalculateClusters ; }
165 void SwitchOnClusterRecalculation() { fRecalculateClusters = kTRUE ; }
166 void SwitchOffClusterRecalculation() { fRecalculateClusters = kFALSE ; }
168 Bool_t IsEmbeddedClusterSelectionOn() const { return fSelectEmbeddedClusters ; }
169 void SwitchOnEmbeddedClustersSelection() { fSelectEmbeddedClusters = kTRUE ; }
170 void SwitchOffEmbeddedClustersSelection() { fSelectEmbeddedClusters = kFALSE ; }
172 // Filling/ filtering / detector information access methods
173 virtual Bool_t FillInputEvent(const Int_t iEntry, const char *currentFileName) ;
174 virtual void FillInputCTS() ;
175 virtual void FillInputEMCAL() ;
176 virtual void FillInputEMCALAlgorithm(AliVCluster * clus, const Int_t iclus) ;
177 virtual void FillInputPHOS() ;
178 virtual void FillInputEMCALCells() ;
179 virtual void FillInputPHOSCells() ;
180 virtual void FillInputVZERO() ;
182 Int_t GetV0Signal(Int_t i) const { return fV0ADC[i] ; }
183 Int_t GetV0Multiplicity(Int_t i) const { return fV0Mul[i] ; }
185 void SetEMCALClusterListName(TString &name) { fEMCALClustersListName = name ; }
186 TString GetEMCALClusterListName() const { return fEMCALClustersListName ; }
188 // Arrayes with clusters/track/cells access method
189 virtual TObjArray* GetCTSTracks() const { return fCTSTracks ; }
190 virtual TObjArray* GetEMCALClusters() const { return fEMCALClusters ; }
191 virtual TObjArray* GetPHOSClusters() const { return fPHOSClusters ; }
192 virtual AliVCaloCells* GetEMCALCells() const { return fEMCALCells ; }
193 virtual AliVCaloCells* GetPHOSCells() const { return fPHOSCells ; }
195 //-------------------------------------
196 // Event/track selection methods
197 //-------------------------------------
199 void AcceptFastClusterEvents() { fAcceptFastCluster = kTRUE ; }
200 void RejectFastClusterEvents() { fAcceptFastCluster = kFALSE ; }
201 Bool_t IsFastClusterAccepted() const { return fAcceptFastCluster ; }
203 void SwitchOnLEDEventsRemoval() { fRemoveLEDEvents = kTRUE ; }
204 void SwitchOffLEDEventsRemoval() { fRemoveLEDEvents = kFALSE ; }
205 Bool_t IsLEDEventRemoved() const { return fRemoveLEDEvents ; }
207 void SetFiredTriggerClassName(TString name ) { fFiredTriggerClassName = name ; }
208 TString GetFiredTriggerClassName() const { return fFiredTriggerClassName ; }
209 TString GetFiredTriggerClasses() ;
211 void SwitchOnEventSelection() { fDoEventSelection = kTRUE ; }
212 void SwitchOffEventSelection() { fDoEventSelection = kFALSE ; }
213 Bool_t IsEventSelectionDone() const { return fDoEventSelection ; }
215 void SwitchOnV0ANDSelection() { fDoV0ANDEventSelection = kTRUE ; }
216 void SwitchOffV0ANDSelection() { fDoV0ANDEventSelection = kFALSE ; }
217 Bool_t IsV0ANDEventSelectionDone() const { return fDoV0ANDEventSelection ; }
219 void SwitchOnPrimaryVertexSelection() { fUseEventsWithPrimaryVertex = kTRUE ; }
220 void SwitchOffPrimaryVertexSelection() { fUseEventsWithPrimaryVertex = kFALSE ; }
221 Bool_t IsPrimaryVertexSelectionDone() const { return fUseEventsWithPrimaryVertex ; }
224 ULong_t GetTrackStatus() const { return fTrackStatus ; }
225 void SetTrackStatus(ULong_t bit) { fTrackStatus = bit ; }
227 ULong_t GetTrackFilterMask() const {return fTrackFilterMask ; }
228 void SetTrackFilterMask(ULong_t bit) { fTrackFilterMask = bit ; }
230 AliESDtrackCuts* GetTrackCuts() const { return fESDtrackCuts ; }
231 void SetTrackCuts(AliESDtrackCuts * cuts) ;
233 Int_t GetTrackMultiplicity() const { return fTrackMult ; }
234 Float_t GetTrackMultiplicityEtaCut() const { return fTrackMultEtaCut ; }
235 void SetTrackMultiplicityEtaCut(Float_t eta) { fTrackMultEtaCut = eta ; }
237 // Calorimeter specific and patches
238 void AnalyzeOnlyLED() { fAnaLED = kTRUE ; }
239 void AnalyzeOnlyPhysics() { fAnaLED = kFALSE ; }
241 void SwitchOnCaloFilterPatch() { fCaloFilterPatch = kTRUE ;
242 fFillCTS = kFALSE ; }
243 void SwitchOffCaloFilterPatch() { fCaloFilterPatch = kFALSE ; }
244 Bool_t IsCaloFilterPatchOn() const {
245 if(fDataType == kAOD) { return fCaloFilterPatch ; }
246 else { return kFALSE ; } }
248 //-------------------------------
250 //-------------------------------
251 virtual void GetVertex(Double_t v[3]) const ;
252 virtual Double_t* GetVertex(const Int_t evtIndex) const { return fVertex[evtIndex] ; }
253 virtual void GetVertex(Double_t vertex[3], const Int_t evtIndex) const ;
254 virtual void FillVertexArray();
255 virtual Bool_t CheckForPrimaryVertex();
256 virtual Float_t GetZvertexCut() const { return fZvtxCut ; } //cut on vertex position
257 virtual void SetZvertexCut(Float_t zcut=10.) { fZvtxCut=zcut ; } //cut on vertex position
259 //--------------------------
260 // Centrality / Event Plane
261 //--------------------------
262 virtual AliCentrality* GetCentrality() const { return fInputEvent->GetCentrality() ; } //Look in AOD reader, different there
263 virtual void SetCentralityClass(TString name) { fCentralityClass = name ; }
264 virtual void SetCentralityOpt(Int_t opt) { fCentralityOpt = opt ; }
265 virtual TString GetCentralityClass() const { return fCentralityClass ; }
266 virtual Int_t GetCentralityOpt() const { return fCentralityOpt ; }
267 virtual Int_t GetEventCentrality() const ;
268 virtual void SetCentralityBin(Int_t min, Int_t max) //Set the centrality bin to select the event. If used, then need to get percentile
269 { fCentralityBin[0]=min; fCentralityBin[1]=max;
270 if(min>=0 && max > 0) fCentralityOpt = 100 ; }
271 virtual Float_t GetCentralityBin(Int_t i) const { if(i < 0 || i > 1) return 0 ;
272 else return fCentralityBin[i] ; }
274 virtual AliEventplane* GetEventPlane() const { return fInputEvent->GetEventplane() ; }
275 virtual void SetEventPlaneMethod(TString m) { fEventPlaneMethod = m ; }
276 virtual TString GetEventPlaneMethod() const { return fEventPlaneMethod ; }
278 //-------------------------------------
280 //-------------------------------------
281 AliCalorimeterUtils * GetCaloUtils() const { return fCaloUtils ; }
282 void SetCaloUtils(AliCalorimeterUtils * caloutils) { fCaloUtils = caloutils ; }
284 virtual Double_t GetBField() const { return fInputEvent->GetMagneticField() ; }
286 void SetImportGeometryFromFile(Bool_t import,
288 fImportGeometryFromFile = import ;
289 fImportGeometryFilePath = path ; }
291 //------------------------------------------------
292 // MC analysis specific methods
293 //-------------------------------------------------
295 //Kinematics and galice.root available
296 virtual AliStack* GetStack() const ;
297 virtual AliHeader* GetHeader() const ;
298 virtual AliGenEventHeader* GetGenEventHeader() const ;
300 //Filtered kinematics in AOD
301 virtual TClonesArray* GetAODMCParticles(Int_t input = 0) const ;
302 virtual AliAODMCHeader* GetAODMCHeader(Int_t input = 0) const ;
304 virtual AliVEvent* GetInputEvent() const { return fInputEvent ; }
305 virtual AliVEvent* GetOriginalInputEvent() const { return 0x0 ; }
306 virtual AliAODEvent* GetOutputEvent() const { return fOutputEvent ; }
307 virtual AliMCEvent* GetMC() const { return fMC ; }
308 virtual AliMixedEvent* GetMixedEvent() const { return fMixedEvent ; }
309 virtual Int_t GetNMixedEvent() const { return fNMixedEvent ; }
311 void SwitchOnStack() { fReadStack = kTRUE ; }
312 void SwitchOffStack() { fReadStack = kFALSE ; }
313 void SwitchOnAODMCParticles() { fReadAODMCParticles = kTRUE ; }
314 void SwitchOffAODMCParticles() { fReadAODMCParticles = kFALSE ; }
315 Bool_t ReadStack() const { return fReadStack ; }
316 Bool_t ReadAODMCParticles() const { return fReadAODMCParticles ; }
318 //Select generated events, depending on comparison of pT hard and jets.
319 virtual Bool_t ComparePtHardAndJetPt() ;
320 virtual Bool_t IsPtHardAndJetPtComparisonSet() const { return fComparePtHardAndJetPt ; }
321 virtual void SetPtHardAndJetPtComparison(Bool_t compare) { fComparePtHardAndJetPt = compare ; }
322 virtual Float_t GetPtHardAndJetFactor() const { return fPtHardAndJetPtFactor ; }
323 virtual void SetPtHardAndJetPtFactor(Float_t factor) { fPtHardAndJetPtFactor = factor ; }
325 //MC reader methods, declared there to allow compilation, they are only used in the MC reader:
327 virtual void AddNeutralParticlesArray(TArrayI & /*array*/) { ; }
328 virtual void AddChargedParticlesArray(TArrayI & /*array*/) { ; }
329 virtual void AddStatusArray(TArrayI & /*array*/) { ; }
331 virtual void SwitchOnPi0Decay() { ; }
332 virtual void SwitchOffPi0Decay() { ; }
333 virtual void SwitchOnStatusSelection() { ; }
334 virtual void SwitchOffStatusSelection() { ; }
335 virtual void SwitchOnOverlapCheck() { ; }
336 virtual void SwitchOffOverlapCheck() { ; }
337 virtual void SwitchOnOnlyGeneratorParticles() { ; }
338 virtual void SwitchOffOnlyGeneratorParticles() { ; }
340 virtual void SetEMCALOverlapAngle(Float_t /*angle*/) { ; }
341 virtual void SetPHOSOverlapAngle(Float_t /*angle*/) { ; }
345 Int_t fEventNumber; // Event number
346 Int_t fDataType ; // Select MC:Kinematics, Data:ESD/AOD, MCData:Both
347 Int_t fDebug; // Debugging level
348 AliFiducialCut * fFiducialCut; //! Acceptance cuts
349 Bool_t fCheckFidCut ; // Do analysis for clusters in defined region
351 Bool_t fComparePtHardAndJetPt; // In MonteCarlo, jet events, reject fake events with wrong jet energy.
352 Float_t fPtHardAndJetPtFactor; // Factor between ptHard and jet pT to reject/accept event.
354 Float_t fCTSPtMin; // pT Threshold on charged particles
355 Float_t fEMCALPtMin; // pT Threshold on emcal clusters
356 Float_t fPHOSPtMin; // pT Threshold on phos clusters
357 Float_t fCTSPtMax; // pT Threshold on charged particles
358 Float_t fEMCALPtMax; // pT Threshold on emcal clusters
359 Float_t fPHOSPtMax; // pT Threshold on phos clusters
360 Double_t fEMCALTimeCutMin;// Remove clusters/cells with time smaller than this value, in ns
361 Double_t fEMCALTimeCutMax;// Remove clusters/cells with time larger than this value, in ns
363 TList * fAODBranchList ; //-> List with AOD branches created and needed in analysis
364 TObjArray * fCTSTracks ; //-> temporal array with tracks
365 TObjArray * fEMCALClusters ; //-> temporal array with EMCAL CaloClusters
366 TObjArray * fPHOSClusters ; //-> temporal array with PHOS CaloClusters
367 AliVCaloCells * fEMCALCells ; //! temporal array with EMCAL CaloCells
368 AliVCaloCells * fPHOSCells ; //! temporal array with PHOS CaloCells
370 AliVEvent * fInputEvent; //! pointer to esd or aod input
371 AliAODEvent * fOutputEvent; //! pointer to aod output
372 AliMCEvent * fMC; //! Monte Carlo Event Handler
374 Bool_t fFillCTS; // use data from CTS
375 Bool_t fFillEMCAL; // use data from EMCAL
376 Bool_t fFillPHOS; // use data from PHOS
377 Bool_t fFillEMCALCells; // use data from EMCAL
378 Bool_t fFillPHOSCells; // use data from PHOS
379 Bool_t fRecalculateClusters; // Correct clusters, recalculate them if recalibration parameters is given
380 Bool_t fSelectEmbeddedClusters; // Use only simulated clusters that come from embedding.
382 ULong_t fTrackStatus ; // Track selection bit, select tracks refitted in TPC, ITS ...
383 ULong_t fTrackFilterMask ; // Track selection bit, for AODs (any difference with track status?)
384 AliESDtrackCuts *fESDtrackCuts ; // Track cut
385 Int_t fTrackMult ; // Track multiplicity
386 Float_t fTrackMultEtaCut ; // Track multiplicity eta cut
387 Bool_t fReadStack ; // Access kine information from stack
388 Bool_t fReadAODMCParticles ; // Access kine information from filtered AOD MC particles
390 TString fDeltaAODFileName ; // Delta AOD file name
391 TString fFiredTriggerClassName; // Name of trigger event type used to do the analysis
393 Bool_t fAnaLED; // Analyze LED data only.
395 TString fTaskName; // Name of task that executes the analysis
397 AliCalorimeterUtils * fCaloUtils ; // Pointer to CalorimeterUtils
399 AliMixedEvent * fMixedEvent ; //! mixed event object. This class is not the owner
400 Int_t fNMixedEvent ; // number of events in mixed event buffer
401 Double_t ** fVertex ; //! vertex array 3 dim for each mixed event buffer
403 Bool_t fWriteOutputDeltaAOD; // Write the created delta AOD objects into file
404 Bool_t fOldAOD; // Old AODs, before revision 4.20
406 Int_t fV0ADC[2] ; // Integrated V0 signal
407 Int_t fV0Mul[2] ; // Integrated V0 Multiplicity
409 Bool_t fCaloFilterPatch; // CaloFilter patch
410 TString fEMCALClustersListName; // Alternative list of clusters produced elsewhere and not from InputEvent
413 Float_t fZvtxCut ; // Cut on vertex position
414 Bool_t fAcceptFastCluster; // Accept events from fast cluster, exclude these events for LHC11a
415 Bool_t fRemoveLEDEvents; // Remove events where LED was wrongly firing - EMCAL LHC11a
416 Bool_t fDoEventSelection; // Select events depending on V0, pileup, vertex well reconstructed, at least 1 track ...
417 Bool_t fDoV0ANDEventSelection; // Select events depending on V0, fDoEventSelection should be on
418 Bool_t fUseEventsWithPrimaryVertex ; // Select events with primary vertex
419 AliTriggerAnalysis* fTriggerAnalysis; // Access to trigger selection algorithm for V0AND calculation
421 //Centrality/Event plane
422 TString fCentralityClass; // Name of selected centrality class
423 Int_t fCentralityOpt; // Option for the returned value of the centrality, possible options 5, 10, 100
424 Int_t fCentralityBin[2]; // Minimum and maximum value of the centrality for the analysis
425 TString fEventPlaneMethod; // Name of event plane method, by default "Q"
427 Bool_t fImportGeometryFromFile; // Import geometry settings in geometry.root file
428 TString fImportGeometryFilePath; // path fo geometry.root file
430 AliCaloTrackReader( const AliCaloTrackReader & r) ; // cpy ctor
431 AliCaloTrackReader & operator = (const AliCaloTrackReader & r) ; // cpy assignment
433 ClassDef(AliCaloTrackReader,36)
438 #endif //ALICALOTRACKREADER_H