]>
Commit | Line | Data |
---|---|---|
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 | 19 | class TObjArray ; |
1c5acb87 | 20 | class TLorentzVector ; |
a97cd470 | 21 | #include "TString.h" |
591cc579 | 22 | class TObjArray; |
477d6cee | 23 | class TArrayF; |
591cc579 | 24 | //#include "TTree.h" |
25 | class TTree ; | |
1c5acb87 | 26 | |
27 | //--- ANALYSIS system --- | |
28 | class AliStack ; | |
29 | class AliHeader ; | |
30 | class AliGenEventHeader ; | |
477d6cee | 31 | class AliVEvent; |
591cc579 | 32 | #include "AliAODEvent.h" |
477d6cee | 33 | class AliMCEvent; |
34 | class AliFidutialCut; | |
591cc579 | 35 | class AliAODMCHeader; |
1c5acb87 | 36 | |
37 | class AliCaloTrackReader : public TObject { | |
38 | ||
477d6cee | 39 | public: |
40 | ||
1c5acb87 | 41 | AliCaloTrackReader() ; // ctor |
42 | AliCaloTrackReader(const AliCaloTrackReader & g) ; // cpy ctor | |
43 | AliCaloTrackReader & operator = (const AliCaloTrackReader & g) ;//cpy assignment | |
44 | virtual ~AliCaloTrackReader() ;//virtual dtor | |
45 | ||
46 | enum inputDataType {kESD, kAOD, kMC}; | |
47 | ||
591cc579 | 48 | //Select generated events, depending on comparison of pT hard and jets. |
49 | virtual Bool_t ComparePtHardAndJetPt() ; | |
50 | virtual Bool_t IsPtHardAndJetPtComparisonSet() const {return fComparePtHardAndJetPt ;} | |
51 | virtual void SetPtHardAndJetPtComparison(Bool_t compare) { fComparePtHardAndJetPt = compare ;} | |
52 | virtual Float_t GetPtHardAndJetFactor() const {return fPtHardAndJetPtFactor ;} | |
53 | virtual void SetPtHardAndJetPtFactor(Float_t factor) { fPtHardAndJetPtFactor = factor ;} | |
54 | ||
1c5acb87 | 55 | virtual void InitParameters(); |
56 | virtual void Print(const Option_t * opt) const; | |
57 | ||
58 | virtual Int_t GetDebug() const { return fDebug ; } | |
59 | virtual void SetDebug(Int_t d) { fDebug = d ; } | |
60 | virtual Int_t GetDataType() const { return fDataType ; } | |
61 | virtual void SetDataType(Int_t data ) { fDataType = data ; } | |
62 | ||
a79a2424 | 63 | virtual Int_t GetEventNumber() const {return fEventNumber ; } |
64 | virtual TString GetCurrentFileName() const {return fCurrentFileName ; } | |
65 | ||
1c5acb87 | 66 | //Minimum pt setters and getters |
67 | virtual Float_t GetEMCALPtMin() const { return fEMCALPtMin ; } | |
68 | virtual Float_t GetPHOSPtMin() const { return fPHOSPtMin ; } | |
69 | virtual Float_t GetCTSPtMin() const { return fCTSPtMin ; } | |
70 | ||
71 | virtual void SetEMCALPtMin(Float_t pt) { fEMCALPtMin = pt ; } | |
72 | virtual void SetPHOSPtMin(Float_t pt) { fPHOSPtMin = pt ; } | |
73 | virtual void SetCTSPtMin(Float_t pt) { fCTSPtMin = pt ; } | |
74 | ||
75 | //Input setters and getters | |
76 | ||
77 | Bool_t IsCTSSwitchedOn() const { return fFillCTS ; } | |
78 | void SwitchOnCTS() {fFillCTS = kTRUE ; } | |
79 | void SwitchOffCTS() {fFillCTS = kFALSE ; } | |
80 | ||
81 | Bool_t IsEMCALSwitchedOn() const { return fFillEMCAL ; } | |
82 | void SwitchOnEMCAL() {fFillEMCAL = kTRUE ; } | |
83 | void SwitchOffEMCAL() {fFillEMCAL = kFALSE ; } | |
84 | ||
85 | Bool_t IsPHOSSwitchedOn() const { return fFillPHOS ; } | |
86 | void SwitchOnPHOS() {fFillPHOS = kTRUE ; } | |
87 | void SwitchOffPHOS() {fFillPHOS = kFALSE ; } | |
88 | ||
89 | Bool_t IsEMCALCellsSwitchedOn() const { return fFillEMCALCells ; } | |
90 | void SwitchOnEMCALCells() {fFillEMCALCells = kTRUE ; } | |
91 | void SwitchOffEMCALCells() {fFillEMCALCells = kFALSE ; } | |
92 | ||
93 | Bool_t IsPHOSCellsSwitchedOn() const { return fFillPHOSCells ; } | |
94 | void SwitchOnPHOSCells() {fFillPHOSCells = kTRUE ; } | |
95 | void SwitchOffPHOSCells() {fFillPHOSCells = kFALSE ; } | |
96 | ||
29b2ceec | 97 | virtual Bool_t FillInputEvent(const Int_t iEntry, const char *currentFileName) ; |
1c5acb87 | 98 | virtual void FillInputCTS() {;} |
99 | virtual void FillInputEMCAL() {;} | |
100 | virtual void FillInputPHOS() {;} | |
101 | virtual void FillInputEMCALCells() {;} | |
102 | virtual void FillInputPHOSCells() {;} | |
103 | ||
591cc579 | 104 | virtual TObjArray* GetAODCTS() const {return fAODCTS ;} |
105 | virtual TObjArray* GetAODEMCAL() const {return fAODEMCAL ;} | |
106 | virtual TObjArray* GetAODPHOS() const {return fAODPHOS ;} | |
477d6cee | 107 | virtual TNamed* GetEMCALCells() const {return fEMCALCells ;} |
108 | virtual TNamed* GetPHOSCells() const {return fPHOSCells ;} | |
1c5acb87 | 109 | |
591cc579 | 110 | //Get MC informatio |
111 | //Kinematics and galice.root available | |
477d6cee | 112 | virtual AliStack* GetStack() const ; |
113 | virtual AliHeader* GetHeader() const ; | |
1c5acb87 | 114 | virtual AliGenEventHeader* GetGenEventHeader() const ; |
591cc579 | 115 | //Filtered kinematics in AOD |
116 | virtual TClonesArray* GetAODMCParticles(Int_t input = 0) const ; | |
117 | virtual AliAODMCHeader* GetAODMCHeader(Int_t input = 0) const ; | |
118 | ||
477d6cee | 119 | virtual AliVEvent* GetInputEvent() const {return fInputEvent;} |
120 | virtual AliAODEvent* GetOutputEvent() const {return fOutputEvent;} | |
121 | virtual AliMCEvent* GetMC() const {return fMC;} | |
1c5acb87 | 122 | virtual void GetVertex(Double_t * ) const {;} |
591cc579 | 123 | |
124 | virtual void Init(); | |
125 | ||
477d6cee | 126 | virtual void SetInputEvent(AliVEvent* input) {fInputEvent = input;} |
127 | virtual void SetOutputEvent(AliAODEvent* aod) {fOutputEvent = aod;} | |
128 | virtual void SetMC(AliMCEvent* mc) {fMC = mc;} | |
1c5acb87 | 129 | |
130 | virtual void ResetLists(); | |
131 | ||
132 | virtual AliFidutialCut * GetFidutialCut() const {return fFidutialCut ;} | |
133 | virtual void SetFidutialCut(AliFidutialCut * fc) { fFidutialCut = fc ;} | |
29b2ceec | 134 | |
477d6cee | 135 | virtual void SetInputOutputMCEvent(AliVEvent* /*esd*/, AliAODEvent* /*aod*/, AliMCEvent* /*mc*/) {;} |
591cc579 | 136 | |
137 | //Methods for mixing with external input file (AOD) | |
138 | virtual TTree* GetSecondInputAODTree() const {return fSecondInputAODTree ; } | |
139 | //virtual void SetSecondInputAODTree(TTree * tree) {fSecondInputAODTree = tree ; | |
140 | // fSecondInputAODEvent->ReadFromTree(tree);}//Connect tree and AOD event. | |
141 | ||
142 | virtual AliAODEvent* GetSecondInputAODEvent() const { return fSecondInputAODEvent ; } | |
143 | ||
144 | TString GetSecondInputFileName() const {return fSecondInputFileName ; } | |
145 | void SetSecondInputFileName(TString name) { fSecondInputFileName = name ; } | |
1c5acb87 | 146 | |
591cc579 | 147 | Int_t GetSecondInputFirstEvent() const {return fSecondInputFirstEvent ; } |
148 | void SetSecondInputFirstEvent(Int_t iEvent0) { fSecondInputFirstEvent = iEvent0 ; } | |
149 | ||
150 | Int_t GetAODCTSNormalInputEntries() {if(!fSecondInputAODTree) { fAODCTSNormalInputEntries = fAODCTS->GetEntriesFast() ;} | |
151 | return fAODCTSNormalInputEntries ; } | |
152 | Int_t GetAODEMCALNormalInputEntries() {if(!fSecondInputAODTree) { fAODEMCALNormalInputEntries = fAODEMCAL->GetEntriesFast();} | |
153 | return fAODEMCALNormalInputEntries ; } | |
154 | Int_t GetAODPHOSNormalInputEntries() {if(!fSecondInputAODTree) { fAODPHOSNormalInputEntries = fAODPHOS->GetEntriesFast() ;} | |
155 | return fAODPHOSNormalInputEntries ; } | |
156 | ||
157 | ULong_t GetTrackStatus() const {return fTrackStatus ; } | |
158 | void SetTrackStatus(ULong_t bit) { fTrackStatus = bit ; } | |
159 | ||
160 | void SwitchOnStack() { fReadStack = kTRUE ; } | |
161 | void SwitchOffStack() { fReadStack = kFALSE ; } | |
162 | void SwitchOnAODMCParticles() { fReadAODMCParticles = kTRUE ; } | |
163 | void SwitchOffAODMCParticles() { fReadAODMCParticles = kFALSE ; } | |
164 | Bool_t ReadStack() const { return fReadStack ; } | |
165 | Bool_t ReadAODMCParticles() const { return fReadAODMCParticles ; } | |
166 | ||
1c5acb87 | 167 | protected: |
477d6cee | 168 | Int_t fEventNumber; // Event number |
a79a2424 | 169 | TString fCurrentFileName; // Current file name under analysis |
1c5acb87 | 170 | Int_t fDataType ; // Select MC:Kinematics, Data:ESD/AOD, MCData:Both |
171 | Int_t fDebug; // Debugging level | |
172 | AliFidutialCut * fFidutialCut; // Acceptance cuts | |
29b2ceec | 173 | |
591cc579 | 174 | Bool_t fComparePtHardAndJetPt; // In MonteCarlo, jet events, reject fake events with wrong jet energy. |
175 | Float_t fPtHardAndJetPtFactor; // Factor between ptHard and jet pT to reject/accept event. | |
176 | ||
477d6cee | 177 | Float_t fCTSPtMin; // pT Threshold on charged particles |
1c5acb87 | 178 | Float_t fEMCALPtMin; // pT Threshold on emcal clusters |
477d6cee | 179 | Float_t fPHOSPtMin; // pT Threshold on phos clusters |
1c5acb87 | 180 | |
591cc579 | 181 | TObjArray * fAODCTS ; //! temporal referenced array with tracks |
182 | TObjArray * fAODEMCAL ; //! temporal referenced array with EMCAL CaloClusters | |
183 | TObjArray * fAODPHOS ; //! temporal referenced array with PHOS CaloClusters | |
1c5acb87 | 184 | TNamed * fEMCALCells ; //! temporal array with EMCAL CaloCells, ESD or AOD |
185 | TNamed * fPHOSCells ; //! temporal array with PHOS CaloCells, ESD or AOD | |
186 | ||
477d6cee | 187 | AliVEvent * fInputEvent; //! pointer to esd or aod input |
188 | AliAODEvent * fOutputEvent; //! pointer to aod output | |
1c5acb87 | 189 | AliMCEvent * fMC; //! Monte Carlo Event Handler |
190 | ||
191 | Bool_t fFillCTS; // use data from CTS | |
192 | Bool_t fFillEMCAL; // use data from EMCAL | |
193 | Bool_t fFillPHOS; // use data from PHOS | |
194 | Bool_t fFillEMCALCells; // use data from EMCAL | |
195 | Bool_t fFillPHOSCells; // use data from PHOS | |
196 | ||
591cc579 | 197 | TTree * fSecondInputAODTree; // Tree with second input AOD, for mixing analysis. |
198 | AliAODEvent* fSecondInputAODEvent; //! pointer to second input AOD event. | |
199 | TString fSecondInputFileName; // File with AOD data to mix with normal stream of data. | |
200 | Int_t fSecondInputFirstEvent; // First event to be considered in the mixing. | |
201 | ||
202 | Int_t fAODCTSNormalInputEntries; // Number of entries in CTS in case of standard input, larger with mixing. | |
203 | Int_t fAODEMCALNormalInputEntries; // Number of entries in EMCAL in case of standard input, larger with mixing. | |
204 | Int_t fAODPHOSNormalInputEntries; // Number of entries in PHOS in case of standard input, larger with mixing. | |
205 | ||
206 | ULong_t fTrackStatus ; // Track selection bit, select tracks refitted in TPC, ITS ... | |
207 | Bool_t fReadStack ; // Access kine information from stack | |
208 | Bool_t fReadAODMCParticles ; // Access kine information from filtered AOD MC particles | |
209 | ||
210 | ClassDef(AliCaloTrackReader,6) | |
1c5acb87 | 211 | } ; |
212 | ||
213 | ||
214 | #endif //ALICALOTRACKREADER_H | |
215 | ||
216 | ||
217 |