]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWG4/PartCorrBase/AliCaloTrackReader.h
adding implementations to extract event type and time stamp from RawReader
[u/mrichter/AliRoot.git] / PWG4 / PartCorrBase / AliCaloTrackReader.h
CommitLineData
1c5acb87 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 */
5/* $Id: $ */
6
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.
591cc579 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)
1c5acb87 14//
15// -- Author: Gustavo Conesa (INFN-LNF)
16
17// --- ROOT system ---
18#include "TObject.h"
591cc579 19class TObjArray ;
a97cd470 20#include "TString.h"
591cc579 21class TTree ;
1c5acb87 22
23//--- ANALYSIS system ---
24class AliStack ;
25class AliHeader ;
26class AliGenEventHeader ;
477d6cee 27class AliVEvent;
8dacfd76 28class AliAODEvent;
477d6cee 29class AliMCEvent;
decca433 30#include "AliFiducialCut.h"
591cc579 31class AliAODMCHeader;
765d44e7 32#include "AliCalorimeterUtils.h"
1c5acb87 33
34class AliCaloTrackReader : public TObject {
35
03190ad7 36public:
1c5acb87 37 AliCaloTrackReader() ; // ctor
1c5acb87 38 virtual ~AliCaloTrackReader() ;//virtual dtor
03190ad7 39private:
78219bac 40 AliCaloTrackReader(const AliCaloTrackReader & g) ; // cpy ctor
03190ad7 41 AliCaloTrackReader & operator = (const AliCaloTrackReader & g) ;//cpy assignment
42
43public:
1c5acb87 44 enum inputDataType {kESD, kAOD, kMC};
45
591cc579 46 //Select generated events, depending on comparison of pT hard and jets.
47 virtual Bool_t ComparePtHardAndJetPt() ;
48 virtual Bool_t IsPtHardAndJetPtComparisonSet() const {return fComparePtHardAndJetPt ;}
49 virtual void SetPtHardAndJetPtComparison(Bool_t compare) { fComparePtHardAndJetPt = compare ;}
50 virtual Float_t GetPtHardAndJetFactor() const {return fPtHardAndJetPtFactor ;}
51 virtual void SetPtHardAndJetPtFactor(Float_t factor) { fPtHardAndJetPtFactor = factor ;}
52
1c5acb87 53 virtual void InitParameters();
54 virtual void Print(const Option_t * opt) const;
55
56 virtual Int_t GetDebug() const { return fDebug ; }
57 virtual void SetDebug(Int_t d) { fDebug = d ; }
58 virtual Int_t GetDataType() const { return fDataType ; }
59 virtual void SetDataType(Int_t data ) { fDataType = data ; }
60
a79a2424 61 virtual Int_t GetEventNumber() const {return fEventNumber ; }
62 virtual TString GetCurrentFileName() const {return fCurrentFileName ; }
63
1c5acb87 64 //Minimum pt setters and getters
65 virtual Float_t GetEMCALPtMin() const { return fEMCALPtMin ; }
66 virtual Float_t GetPHOSPtMin() const { return fPHOSPtMin ; }
67 virtual Float_t GetCTSPtMin() const { return fCTSPtMin ; }
68
69 virtual void SetEMCALPtMin(Float_t pt) { fEMCALPtMin = pt ; }
70 virtual void SetPHOSPtMin(Float_t pt) { fPHOSPtMin = pt ; }
71 virtual void SetCTSPtMin(Float_t pt) { fCTSPtMin = pt ; }
72
73 //Input setters and getters
74
75 Bool_t IsCTSSwitchedOn() const { return fFillCTS ; }
76 void SwitchOnCTS() {fFillCTS = kTRUE ; }
77 void SwitchOffCTS() {fFillCTS = kFALSE ; }
78
79 Bool_t IsEMCALSwitchedOn() const { return fFillEMCAL ; }
80 void SwitchOnEMCAL() {fFillEMCAL = kTRUE ; }
81 void SwitchOffEMCAL() {fFillEMCAL = kFALSE ; }
82
83 Bool_t IsPHOSSwitchedOn() const { return fFillPHOS ; }
84 void SwitchOnPHOS() {fFillPHOS = kTRUE ; }
85 void SwitchOffPHOS() {fFillPHOS = kFALSE ; }
86
87 Bool_t IsEMCALCellsSwitchedOn() const { return fFillEMCALCells ; }
88 void SwitchOnEMCALCells() {fFillEMCALCells = kTRUE ; }
89 void SwitchOffEMCALCells() {fFillEMCALCells = kFALSE ; }
90
91 Bool_t IsPHOSCellsSwitchedOn() const { return fFillPHOSCells ; }
92 void SwitchOnPHOSCells() {fFillPHOSCells = kTRUE ; }
93 void SwitchOffPHOSCells() {fFillPHOSCells = kFALSE ; }
94
29b2ceec 95 virtual Bool_t FillInputEvent(const Int_t iEntry, const char *currentFileName) ;
1c5acb87 96 virtual void FillInputCTS() {;}
97 virtual void FillInputEMCAL() {;}
98 virtual void FillInputPHOS() {;}
99 virtual void FillInputEMCALCells() {;}
100 virtual void FillInputPHOSCells() {;}
101
591cc579 102 virtual TObjArray* GetAODCTS() const {return fAODCTS ;}
103 virtual TObjArray* GetAODEMCAL() const {return fAODEMCAL ;}
104 virtual TObjArray* GetAODPHOS() const {return fAODPHOS ;}
477d6cee 105 virtual TNamed* GetEMCALCells() const {return fEMCALCells ;}
106 virtual TNamed* GetPHOSCells() const {return fPHOSCells ;}
1c5acb87 107
591cc579 108 //Get MC informatio
109 //Kinematics and galice.root available
477d6cee 110 virtual AliStack* GetStack() const ;
111 virtual AliHeader* GetHeader() const ;
1c5acb87 112 virtual AliGenEventHeader* GetGenEventHeader() const ;
591cc579 113 //Filtered kinematics in AOD
114 virtual TClonesArray* GetAODMCParticles(Int_t input = 0) const ;
115 virtual AliAODMCHeader* GetAODMCHeader(Int_t input = 0) const ;
116
477d6cee 117 virtual AliVEvent* GetInputEvent() const {return fInputEvent;}
118 virtual AliAODEvent* GetOutputEvent() const {return fOutputEvent;}
119 virtual AliMCEvent* GetMC() const {return fMC;}
233e0df8 120 virtual void GetVertex(Double_t *) const {;}
121 virtual void GetSecondInputAODVertex(Double_t *) const {;}
8dacfd76 122 virtual Double_t GetBField() const { return 0.;}
591cc579 123
124 virtual void Init();
125
8dacfd76 126 virtual void SetInputEvent(AliVEvent* const input) {fInputEvent = input;}
127 virtual void SetOutputEvent(AliAODEvent* const aod) {fOutputEvent = aod;}
128 virtual void SetMC(AliMCEvent* const mc) {fMC = mc;}
1c5acb87 129
130 virtual void ResetLists();
131
decca433 132 virtual AliFiducialCut * GetFiducialCut() {if(!fFiducialCut) fFiducialCut = new AliFiducialCut();
133 return fFiducialCut ;}
ff45398a 134 virtual void SetFiducialCut(AliFiducialCut * const fc) { fFiducialCut = fc ;}
decca433 135 virtual Bool_t IsFiducialCutOn() {return fCheckFidCut ; }
136 virtual void SwitchOnFiducialCut() { fCheckFidCut = kTRUE; fFiducialCut = new AliFiducialCut();}
137 virtual void SwitchOffFiducialCut() { fCheckFidCut = kFALSE;}
29b2ceec 138
477d6cee 139 virtual void SetInputOutputMCEvent(AliVEvent* /*esd*/, AliAODEvent* /*aod*/, AliMCEvent* /*mc*/) {;}
591cc579 140
141 //Methods for mixing with external input file (AOD)
142 virtual TTree* GetSecondInputAODTree() const {return fSecondInputAODTree ; }
143 //virtual void SetSecondInputAODTree(TTree * tree) {fSecondInputAODTree = tree ;
144 // fSecondInputAODEvent->ReadFromTree(tree);}//Connect tree and AOD event.
145
146 virtual AliAODEvent* GetSecondInputAODEvent() const { return fSecondInputAODEvent ; }
147
148 TString GetSecondInputFileName() const {return fSecondInputFileName ; }
149 void SetSecondInputFileName(TString name) { fSecondInputFileName = name ; }
1c5acb87 150
591cc579 151 Int_t GetSecondInputFirstEvent() const {return fSecondInputFirstEvent ; }
152 void SetSecondInputFirstEvent(Int_t iEvent0) { fSecondInputFirstEvent = iEvent0 ; }
153
154 Int_t GetAODCTSNormalInputEntries() {if(!fSecondInputAODTree) { fAODCTSNormalInputEntries = fAODCTS->GetEntriesFast() ;}
155 return fAODCTSNormalInputEntries ; }
156 Int_t GetAODEMCALNormalInputEntries() {if(!fSecondInputAODTree) { fAODEMCALNormalInputEntries = fAODEMCAL->GetEntriesFast();}
157 return fAODEMCALNormalInputEntries ; }
158 Int_t GetAODPHOSNormalInputEntries() {if(!fSecondInputAODTree) { fAODPHOSNormalInputEntries = fAODPHOS->GetEntriesFast() ;}
159 return fAODPHOSNormalInputEntries ; }
160
161 ULong_t GetTrackStatus() const {return fTrackStatus ; }
162 void SetTrackStatus(ULong_t bit) { fTrackStatus = bit ; }
163
164 void SwitchOnStack() { fReadStack = kTRUE ; }
165 void SwitchOffStack() { fReadStack = kFALSE ; }
166 void SwitchOnAODMCParticles() { fReadAODMCParticles = kTRUE ; }
167 void SwitchOffAODMCParticles() { fReadAODMCParticles = kFALSE ; }
168 Bool_t ReadStack() const { return fReadStack ; }
169 Bool_t ReadAODMCParticles() const { return fReadAODMCParticles ; }
170
08a064bc 171 void SwitchOnCleanStdAOD() {fCleanOutputStdAOD = kTRUE;}
172 void SwitchOffCleanStdAOD() {fCleanOutputStdAOD = kFALSE;}
173
42dc8e7d 174 void SetDeltaAODFileName(TString name ) {fDeltaAODFileName = name ; }
175 TString GetDeltaAODFileName() const {return fDeltaAODFileName ; }
72d2488e 176
177 void SetFiredTriggerClassName(TString name ) {fFiredTriggerClassName = name ; }
178 TString GetFiredTriggerClassName() const {return fFiredTriggerClassName ; }
179 virtual TString GetFiredTriggerClasses() {return "";}
0866d83a 180
c1ac3823 181 void AnalyzeOnlyLED() {fAnaLED = kTRUE;}
182 void AnalyzeOnlyPhysics() {fAnaLED = kFALSE;}
183
486258c9 184 TString GetTaskName() const {return fTaskName;}
185 void SetTaskName(TString name) {fTaskName = name;}
186
765d44e7 187 AliCalorimeterUtils * GetCaloUtils() const {return fCaloUtils ; }
188 void SetCaloUtils(AliCalorimeterUtils * caloutils) { fCaloUtils = caloutils ; }
189
1c5acb87 190 protected:
477d6cee 191 Int_t fEventNumber; // Event number
a79a2424 192 TString fCurrentFileName; // Current file name under analysis
1c5acb87 193 Int_t fDataType ; // Select MC:Kinematics, Data:ESD/AOD, MCData:Both
194 Int_t fDebug; // Debugging level
614701c6 195 AliFiducialCut * fFiducialCut; //! Acceptance cuts
decca433 196 Bool_t fCheckFidCut ;// Do analysis for clusters in defined region
197
591cc579 198 Bool_t fComparePtHardAndJetPt; // In MonteCarlo, jet events, reject fake events with wrong jet energy.
199 Float_t fPtHardAndJetPtFactor; // Factor between ptHard and jet pT to reject/accept event.
200
477d6cee 201 Float_t fCTSPtMin; // pT Threshold on charged particles
1c5acb87 202 Float_t fEMCALPtMin; // pT Threshold on emcal clusters
477d6cee 203 Float_t fPHOSPtMin; // pT Threshold on phos clusters
1c5acb87 204
591cc579 205 TObjArray * fAODCTS ; //! temporal referenced array with tracks
206 TObjArray * fAODEMCAL ; //! temporal referenced array with EMCAL CaloClusters
207 TObjArray * fAODPHOS ; //! temporal referenced array with PHOS CaloClusters
1c5acb87 208 TNamed * fEMCALCells ; //! temporal array with EMCAL CaloCells, ESD or AOD
209 TNamed * fPHOSCells ; //! temporal array with PHOS CaloCells, ESD or AOD
210
477d6cee 211 AliVEvent * fInputEvent; //! pointer to esd or aod input
212 AliAODEvent * fOutputEvent; //! pointer to aod output
1c5acb87 213 AliMCEvent * fMC; //! Monte Carlo Event Handler
214
215 Bool_t fFillCTS; // use data from CTS
216 Bool_t fFillEMCAL; // use data from EMCAL
217 Bool_t fFillPHOS; // use data from PHOS
218 Bool_t fFillEMCALCells; // use data from EMCAL
219 Bool_t fFillPHOSCells; // use data from PHOS
220
591cc579 221 TTree * fSecondInputAODTree; // Tree with second input AOD, for mixing analysis.
222 AliAODEvent* fSecondInputAODEvent; //! pointer to second input AOD event.
223 TString fSecondInputFileName; // File with AOD data to mix with normal stream of data.
224 Int_t fSecondInputFirstEvent; // First event to be considered in the mixing.
225
226 Int_t fAODCTSNormalInputEntries; // Number of entries in CTS in case of standard input, larger with mixing.
227 Int_t fAODEMCALNormalInputEntries; // Number of entries in EMCAL in case of standard input, larger with mixing.
228 Int_t fAODPHOSNormalInputEntries; // Number of entries in PHOS in case of standard input, larger with mixing.
229
230 ULong_t fTrackStatus ; // Track selection bit, select tracks refitted in TPC, ITS ...
231 Bool_t fReadStack ; // Access kine information from stack
232 Bool_t fReadAODMCParticles ; // Access kine information from filtered AOD MC particles
233
08a064bc 234 Bool_t fCleanOutputStdAOD; // clean the written standard tracks and caloclusters in output AOD
42dc8e7d 235 TString fDeltaAODFileName ; // Delta AOD file name
72d2488e 236 TString fFiredTriggerClassName ; // Name of trigger event type used to do the analysis
afabc52f 237
c1ac3823 238 Bool_t fAnaLED; // Analyze LED data only.
c1ac3823 239
765d44e7 240 TString fTaskName; // Name of task that executes the analysis
241
242 AliCalorimeterUtils * fCaloUtils ; // Pointer to CalorimeterUtils
243
decca433 244 ClassDef(AliCaloTrackReader,15)
1c5acb87 245} ;
246
247
248#endif //ALICALOTRACKREADER_H
249
250
251