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 ---
26 //--- ANALYSIS system ---
29 class AliGenEventHeader ;
34 #include "AliVCaloCells.h"
35 #include "AliFiducialCut.h"
37 #include "AliCalorimeterUtils.h"
38 class AliESDtrackCuts;
40 class AliCaloTrackReader : public TObject {
43 AliCaloTrackReader() ; // ctor
44 virtual ~AliCaloTrackReader() ;//virtual dtor
46 AliCaloTrackReader(const AliCaloTrackReader & g) ; // cpy ctor
47 AliCaloTrackReader & operator = (const AliCaloTrackReader & g) ;//cpy assignment
50 enum inputDataType {kESD, kAOD, kMC};
52 //Select generated events, depending on comparison of pT hard and jets.
53 virtual Bool_t ComparePtHardAndJetPt() ;
54 virtual Bool_t IsPtHardAndJetPtComparisonSet() const {return fComparePtHardAndJetPt ;}
55 virtual void SetPtHardAndJetPtComparison(Bool_t compare) { fComparePtHardAndJetPt = compare ;}
56 virtual Float_t GetPtHardAndJetFactor() const {return fPtHardAndJetPtFactor ;}
57 virtual void SetPtHardAndJetPtFactor(Float_t factor) { fPtHardAndJetPtFactor = factor ;}
59 virtual void InitParameters();
60 virtual void Print(const Option_t * opt) const;
62 virtual Int_t GetDebug() const { return fDebug ; }
63 virtual void SetDebug(Int_t d) { fDebug = d ; }
64 virtual Int_t GetDataType() const { return fDataType ; }
65 virtual void SetDataType(Int_t data ) { fDataType = data ; }
67 virtual Int_t GetEventNumber() const {return fEventNumber ; }
68 virtual TString GetCurrentFileName() const {return fCurrentFileName ; }
70 //Minimum pt setters and getters
71 virtual Float_t GetEMCALPtMin() const { return fEMCALPtMin ; }
72 virtual Float_t GetPHOSPtMin() const { return fPHOSPtMin ; }
73 virtual Float_t GetCTSPtMin() const { return fCTSPtMin ; }
75 virtual void SetEMCALPtMin(Float_t pt) { fEMCALPtMin = pt ; }
76 virtual void SetPHOSPtMin(Float_t pt) { fPHOSPtMin = pt ; }
77 virtual void SetCTSPtMin(Float_t pt) { fCTSPtMin = pt ; }
79 //Input setters and getters
80 Bool_t IsEMCALCluster(AliVCluster *clus) const;
81 Bool_t IsPHOSCluster (AliVCluster *clus) const;
82 void SwitchOnOldAODs() {fOldAOD = kTRUE ; }
83 void SwitchOffOldAODs() {fOldAOD = kFALSE ; }
85 Bool_t IsCTSSwitchedOn() const { return fFillCTS ; }
86 void SwitchOnCTS() {fFillCTS = kTRUE ; }
87 void SwitchOffCTS() {fFillCTS = kFALSE ; }
89 Bool_t IsEMCALSwitchedOn() const { return fFillEMCAL ; }
90 void SwitchOnEMCAL() {fFillEMCAL = kTRUE ; }
91 void SwitchOffEMCAL() {fFillEMCAL = kFALSE ; }
93 Bool_t IsPHOSSwitchedOn() const { return fFillPHOS ; }
94 void SwitchOnPHOS() {fFillPHOS = kTRUE ; }
95 void SwitchOffPHOS() {fFillPHOS = kFALSE ; }
97 Bool_t IsEMCALCellsSwitchedOn() const { return fFillEMCALCells ; }
98 void SwitchOnEMCALCells() {fFillEMCALCells = kTRUE ; }
99 void SwitchOffEMCALCells() {fFillEMCALCells = kFALSE ; }
101 Bool_t IsPHOSCellsSwitchedOn() const { return fFillPHOSCells ; }
102 void SwitchOnPHOSCells() {fFillPHOSCells = kTRUE ; }
103 void SwitchOffPHOSCells() {fFillPHOSCells = kFALSE ; }
105 virtual Bool_t FillInputEvent(const Int_t iEntry, const char *currentFileName) ;
106 virtual void FillInputCTS() ;
107 virtual void FillInputEMCAL() ;
108 virtual void FillInputPHOS() ;
109 virtual void FillInputEMCALCells() ;
110 virtual void FillInputPHOSCells() ;
112 virtual TList * GetAODBranchList() const { return fAODBranchList ; }
114 virtual TObjArray* GetAODCTS() const {return fAODCTS ;}
115 virtual TObjArray* GetAODEMCAL() const {return fAODEMCAL ;}
116 virtual TObjArray* GetAODPHOS() const {return fAODPHOS ;}
117 virtual AliVCaloCells* GetEMCALCells() const { return fEMCALCells ;}
118 virtual AliVCaloCells* GetPHOSCells() const { return fPHOSCells ;}
121 //Kinematics and galice.root available
122 virtual AliStack* GetStack() const ;
123 virtual AliHeader* GetHeader() const ;
124 virtual AliGenEventHeader* GetGenEventHeader() const ;
125 //Filtered kinematics in AOD
126 virtual TClonesArray* GetAODMCParticles(Int_t input = 0) const ;
127 virtual AliAODMCHeader* GetAODMCHeader(Int_t input = 0) const ;
129 virtual AliVEvent* GetInputEvent() const {return fInputEvent;}
130 virtual AliAODEvent* GetOutputEvent() const {return fOutputEvent;}
131 virtual AliMCEvent* GetMC() const {return fMC;}
132 virtual AliMixedEvent* GetMixedEvent() const {return fMixedEvent;}
133 virtual Int_t GetNMixedEvent() const {return fNMixedEvent ; }
135 virtual Double_t GetBField() const { return 0.;}
139 // virtual void SetInputEvent(AliVEvent* const input) {fInputEvent = input;}
140 virtual void SetInputEvent(AliVEvent* const input) ;
141 virtual void SetOutputEvent(AliAODEvent* const aod) {fOutputEvent = aod;}
142 virtual void SetMC(AliMCEvent* const mc) {fMC = mc;}
144 virtual void ResetLists();
146 virtual AliFiducialCut * GetFiducialCut() {if(!fFiducialCut) fFiducialCut = new AliFiducialCut();
147 return fFiducialCut ;}
148 virtual void SetFiducialCut(AliFiducialCut * const fc) { fFiducialCut = fc ;}
149 virtual Bool_t IsFiducialCutOn() {return fCheckFidCut ; }
150 virtual void SwitchOnFiducialCut() { fCheckFidCut = kTRUE; fFiducialCut = new AliFiducialCut();}
151 virtual void SwitchOffFiducialCut() { fCheckFidCut = kFALSE;}
153 virtual void SetInputOutputMCEvent(AliVEvent* /*esd*/, AliAODEvent* /*aod*/, AliMCEvent* /*mc*/) {;}
155 //Methods for mixing with external input file (AOD)
156 //virtual TTree* GetSecondInputAODTree() const {return fSecondInputAODTree ; }
157 //virtual void SetSecondInputAODTree(TTree * tree) {fSecondInputAODTree = tree ;
158 // fSecondInputAODEvent->ReadFromTree(tree);}//Connect tree and AOD event.
160 //virtual AliAODEvent* GetSecondInputAODEvent() const { return fSecondInputAODEvent ; }
162 //TString GetSecondInputFileName() const {return fSecondInputFileName ; }
163 //void SetSecondInputFileName(TString name) { fSecondInputFileName = name ; }
165 //Int_t GetSecondInputFirstEvent() const {return fSecondInputFirstEvent ; }
166 //void SetSecondInputFirstEvent(Int_t iEvent0) { fSecondInputFirstEvent = iEvent0 ; }
168 // Int_t GetAODCTSNormalInputEntries() {if(!fSecondInputAODTree) { fAODCTSNormalInputEntries = fAODCTS->GetEntriesFast() ;}
169 // return fAODCTSNormalInputEntries ; }
170 // Int_t GetAODEMCALNormalInputEntries() {if(!fSecondInputAODTree) { fAODEMCALNormalInputEntries = fAODEMCAL->GetEntriesFast();}
171 // return fAODEMCALNormalInputEntries ; }
172 // Int_t GetAODPHOSNormalInputEntries() {if(!fSecondInputAODTree) { fAODPHOSNormalInputEntries = fAODPHOS->GetEntriesFast() ;}
173 // return fAODPHOSNormalInputEntries ; }
176 ULong_t GetTrackStatus() const {return fTrackStatus ; }
177 void SetTrackStatus(ULong_t bit) { fTrackStatus = bit ; }
179 AliESDtrackCuts* GetTrackCuts() const { return fESDtrackCuts ; }
180 void SetTrackCuts(AliESDtrackCuts * cuts) { fESDtrackCuts = cuts ; }
181 Int_t GetTrackMultiplicity() const { return fTrackMult ; }
182 Float_t GetTrackMultiplicityEtaCut() const { return fTrackMultEtaCut ; }
183 void SetTrackMultiplicityEtaCut(Float_t eta) { fTrackMultEtaCut = eta ; }
186 void SwitchOnStack() { fReadStack = kTRUE ; }
187 void SwitchOffStack() { fReadStack = kFALSE ; }
188 void SwitchOnAODMCParticles() { fReadAODMCParticles = kTRUE ; }
189 void SwitchOffAODMCParticles() { fReadAODMCParticles = kFALSE ; }
190 Bool_t ReadStack() const { return fReadStack ; }
191 Bool_t ReadAODMCParticles() const { return fReadAODMCParticles ; }
193 void SetDeltaAODFileName(TString name ) {fDeltaAODFileName = name ; }
194 TString GetDeltaAODFileName() const {return fDeltaAODFileName ; }
196 void SetFiredTriggerClassName(TString name ) {fFiredTriggerClassName = name ; }
197 TString GetFiredTriggerClassName() const {return fFiredTriggerClassName ; }
198 virtual TString GetFiredTriggerClasses() {return "";}
200 void AnalyzeOnlyLED() {fAnaLED = kTRUE;}
201 void AnalyzeOnlyPhysics() {fAnaLED = kFALSE;}
203 TString GetTaskName() const {return fTaskName;}
204 void SetTaskName(TString name) {fTaskName = name;}
206 AliCalorimeterUtils * GetCaloUtils() const {return fCaloUtils ; }
207 void SetCaloUtils(AliCalorimeterUtils * caloutils) { fCaloUtils = caloutils ; }
209 virtual void GetVertex(Double_t v[3]) const ;
210 virtual Double_t* GetVertex(const Int_t evtIndex) const {return fVertex[evtIndex];}
211 virtual void GetVertex(Double_t vertex[3], const Int_t evtIndex) const ;
212 virtual void FillVertexArray();
213 // virtual void GetSecondInputAODVertex(Double_t *) const {;}
215 void SwitchOnWriteDeltaAOD() {fWriteOutputDeltaAOD = kTRUE ; }
216 void SwitchOffWriteDeltaAOD() {fWriteOutputDeltaAOD = kFALSE ; }
217 Bool_t WriteDeltaAODToFile() const {return fWriteOutputDeltaAOD ; }
222 virtual void AddNeutralParticlesArray(TArrayI & /*array*/) { ; }
223 virtual void AddChargedParticlesArray(TArrayI & /*array*/) { ; }
224 virtual void AddStatusArray(TArrayI & /*array*/) { ; }
226 virtual void SwitchOnPi0Decay() { ; }
227 virtual void SwitchOffPi0Decay() { ; }
228 virtual void SwitchOnStatusSelection() { ; }
229 virtual void SwitchOffStatusSelection() { ; }
230 virtual void SwitchOnOverlapCheck() { ; }
231 virtual void SwitchOffOverlapCheck() { ; }
233 virtual void SetEMCALOverlapAngle(Float_t /*angle*/) { ; }
234 virtual void SetPHOSOverlapAngle(Float_t /*angle*/) { ; }
238 Int_t fEventNumber; // Event number
239 TString fCurrentFileName;// Current file name under analysis
240 Int_t fDataType ; // Select MC:Kinematics, Data:ESD/AOD, MCData:Both
241 Int_t fDebug; // Debugging level
242 AliFiducialCut * fFiducialCut; //! Acceptance cuts
243 Bool_t fCheckFidCut ; // Do analysis for clusters in defined region
245 Bool_t fComparePtHardAndJetPt; // In MonteCarlo, jet events, reject fake events with wrong jet energy.
246 Float_t fPtHardAndJetPtFactor; // Factor between ptHard and jet pT to reject/accept event.
248 Float_t fCTSPtMin; // pT Threshold on charged particles
249 Float_t fEMCALPtMin; // pT Threshold on emcal clusters
250 Float_t fPHOSPtMin; // pT Threshold on phos clusters
252 TList * fAODBranchList ; //! List with AOD branches created and needed in analysis
253 TObjArray * fAODCTS ; //! temporal referenced array with tracks
254 TObjArray * fAODEMCAL ; //! temporal referenced array with EMCAL CaloClusters
255 TObjArray * fAODPHOS ; //! temporal referenced array with PHOS CaloClusters
256 AliVCaloCells * fEMCALCells ; //! temporal array with EMCAL CaloCells, ESD or AOD
257 AliVCaloCells * fPHOSCells ; //! temporal array with PHOS CaloCells, ESD or AOD
259 AliVEvent * fInputEvent; //! pointer to esd or aod input
260 AliAODEvent * fOutputEvent; //! pointer to aod output
261 AliMCEvent * fMC; //! Monte Carlo Event Handler
263 Bool_t fFillCTS; // use data from CTS
264 Bool_t fFillEMCAL; // use data from EMCAL
265 Bool_t fFillPHOS; // use data from PHOS
266 Bool_t fFillEMCALCells; // use data from EMCAL
267 Bool_t fFillPHOSCells; // use data from PHOS
269 // TTree * fSecondInputAODTree; // Tree with second input AOD, for mixing analysis.
270 // AliAODEvent* fSecondInputAODEvent; //! pointer to second input AOD event.
271 // TString fSecondInputFileName; // File with AOD data to mix with normal stream of data.
272 // Int_t fSecondInputFirstEvent; // First event to be considered in the mixing.
274 // Int_t fAODCTSNormalInputEntries; // Number of entries in CTS in case of standard input, larger with mixing.
275 // Int_t fAODEMCALNormalInputEntries; // Number of entries in EMCAL in case of standard input, larger with mixing.
276 // Int_t fAODPHOSNormalInputEntries; // Number of entries in PHOS in case of standard input, larger with mixing.
278 ULong_t fTrackStatus ; // Track selection bit, select tracks refitted in TPC, ITS ...
279 AliESDtrackCuts *fESDtrackCuts ; // Track cut
280 Int_t fTrackMult ; // Track multiplicity
281 Float_t fTrackMultEtaCut ; // Track multiplicity eta cut
282 Bool_t fReadStack ; // Access kine information from stack
283 Bool_t fReadAODMCParticles ; // Access kine information from filtered AOD MC particles
285 TString fDeltaAODFileName ; // Delta AOD file name
286 TString fFiredTriggerClassName; // Name of trigger event type used to do the analysis
288 Bool_t fAnaLED; // Analyze LED data only.
290 TString fTaskName; // Name of task that executes the analysis
292 AliCalorimeterUtils * fCaloUtils ; // Pointer to CalorimeterUtils
294 AliMixedEvent * fMixedEvent ; //! mixed event object. This class is not the owner
295 Int_t fNMixedEvent ; // number of events in mixed event buffer
296 Double_t ** fVertex ; //! vertex array 3 dim for each mixed event buffer
298 Bool_t fWriteOutputDeltaAOD;// Write the created delta AOD objects into file
299 Bool_t fOldAOD; // Old AODs, before revision 4.20
301 ClassDef(AliCaloTrackReader,20)
305 #endif //ALICALOTRACKREADER_H