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