]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWG/CaloTrackCorrBase/AliCaloTrackReader.h
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Fixing issues with the reconstruction efficiency, adding histograms to do a simultane...
[u/mrichter/AliRoot.git] / PWG / CaloTrackCorrBase / 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 */
1c5acb87 5
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.
591cc579 10// Mother class of : AliCaloTrackESDReader: Fills ESD data in 3 TObjArrays (PHOS, EMCAL, CTS)
743aa53a 11// : AliCaloTrackMCReader : Fills Kinematics data in 3 TObjArrays (PHOS, EMCAL, CTS)
12// : AliCaloTrackAODReader: Fills AOD data in 3 TObjArrays (PHOS, EMCAL, CTS)
1c5acb87 13// -- Author: Gustavo Conesa (INFN-LNF)
743aa53a 14//////////////////////////////////////////////////////////////////////////////
1c5acb87 15
16// --- ROOT system ---
49b53920 17#include <TObject.h>
18#include <TString.h>
cfaba834 19class TObjArray ;
591cc579 20class TTree ;
f3138ecf 21class TArrayI ;
1c5acb87 22
23//--- ANALYSIS system ---
0de1814a 24#include "AliVEvent.h"
c5693f62 25class AliVCaloCells;
26class AliStack;
27class AliHeader;
28class AliGenEventHeader;
8dacfd76 29class AliAODEvent;
477d6cee 30class AliMCEvent;
c8fe2783 31class AliMixedEvent;
591cc579 32class AliAODMCHeader;
c5693f62 33class AliESDtrackCuts;
b6dd6ad2 34class AliCentrality;
48c37e02 35class AliTriggerAnalysis;
72542aba 36class AliEventplane;
f3138ecf 37class AliVCluster;
1c5acb87 38
0de1814a 39// --- CaloTrackCorr / EMCAL ---
cfaba834 40#include "AliFiducialCut.h"
0de1814a 41class AliEMCALRecoUtils;
c5693f62 42class AliCalorimeterUtils;
cfaba834 43
1c5acb87 44class AliCaloTrackReader : public TObject {
45
03190ad7 46public:
f3138ecf 47
48 AliCaloTrackReader() ; // ctor
49 virtual ~AliCaloTrackReader() ; // virtual dtor
cfaba834 50
51 //--------------------------------
52 // General methods
53 //--------------------------------
54
55 virtual void Init();
de0b770d 56
cfaba834 57 virtual void InitParameters();
de0b770d 58
cfaba834 59 virtual void Print(const Option_t * opt) const;
de0b770d 60
cfaba834 61 virtual void ResetLists();
1c5acb87 62
3c75ddf7 63 virtual Int_t GetDebug() const { return fDebug ; }
64 virtual void SetDebug(Int_t d) { fDebug = d ; }
cfaba834 65
66 enum inputDataType {kESD, kAOD, kMC};
3c75ddf7 67 virtual Int_t GetDataType() const { return fDataType ; }
68 virtual void SetDataType(Int_t data ) { fDataType = data ; }
1c5acb87 69
3c75ddf7 70 virtual Int_t GetEventNumber() const { return fEventNumber ; }
a79a2424 71
3c75ddf7 72 TString GetTaskName() const { return fTaskName ; }
73 void SetTaskName(TString name) { fTaskName = name ; }
cfaba834 74
cfaba834 75 //---------------------------------------
76 //Input/output event setters and getters
77 //---------------------------------------
78 virtual void SetInputEvent(AliVEvent* const input) ;
3c75ddf7 79 virtual void SetOutputEvent(AliAODEvent* const aod) { fOutputEvent = aod ; }
80 virtual void SetMC(AliMCEvent* const mc) { fMC = mc ; }
be518ab0 81 virtual void SetInputOutputMCEvent(AliVEvent* /*esd*/, AliAODEvent* /*aod*/, AliMCEvent* /*mc*/) { ; }
cfaba834 82
83 // Delta AODs
3c75ddf7 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 ; }
cfaba834 90
91 //------------------------------------------------------------
92 //Clusters/Tracks arrays filtering/filling methods and switchs
93 //------------------------------------------------------------
94
95 //Minimum pt setters and getters
3c75ddf7 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 ; }
a5fb4114 102
3c75ddf7 103 void SetEMCALPtMin(Float_t pt) { fEMCALPtMin = pt ; }
104 void SetPHOSPtMin (Float_t pt) { fPHOSPtMin = pt ; }
105 void SetCTSPtMin (Float_t pt) { fCTSPtMin = pt ; }
a5fb4114 106
3c75ddf7 107 void SetEMCALPtMax(Float_t pt) { fEMCALPtMax = pt ; }
108 void SetPHOSPtMax (Float_t pt) { fPHOSPtMax = pt ; }
109 void SetCTSPtMax (Float_t pt) { fCTSPtMax = pt ; }
a5fb4114 110
3c75ddf7 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() ; }
cfaba834 115
3c75ddf7 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) ; }
cfaba834 120
cc944149 121 // Track DCA cut
122 Bool_t AcceptDCA(const Float_t pt, const Float_t dca);
689d9f15 123 Double_t GetTrackDCACut(Int_t i) const { if(i >= 0 && i < 3 ) return fTrackDCACut[i] ;
124 else return -999 ; }
125
126 void SetTrackDCACut(Int_t i, Float_t cut) { if(i >= 0 && i < 3 )
127 fTrackDCACut[i] = cut ; }
128
129 void SwitchOnUseTrackDCACut() { fUseTrackDCACut = kTRUE ; }
130 void SwitchOffUseTrackDCACut() { fUseTrackDCACut = kFALSE ; }
cc944149 131 Bool_t IsDCACutOn() const { return fUseTrackDCACut ; }
689d9f15 132
ff440946 133 //Time cut
134
975b29fa 135 Double_t GetTrackTimeCutMin() const { return fTrackTimeCutMin ; }
136 Double_t GetTrackTimeCutMax() const { return fTrackTimeCutMax ; }
137
975b29fa 138 void SetTrackTimeCut(Double_t a, Double_t b) { fTrackTimeCutMin = a ;
139 fTrackTimeCutMax = b ; } // ns
140
4b7f6e01 141 void SwitchOnUseTrackTimeCut() { fUseTrackTimeCut = kTRUE ; }
142 void SwitchOffUseTrackTimeCut() { fUseTrackTimeCut = kFALSE ; }
143
ff440946 144 Double_t GetEMCALTimeCutMin() const { return fEMCALTimeCutMin ; }
145 Double_t GetEMCALTimeCutMax() const { return fEMCALTimeCutMax ; }
d2655d46 146
3c1a2e95 147 Bool_t IsInTimeWindow(const Double_t tof, const Float_t energy) const ;
148
ff440946 149 void SetEMCALTimeCut(Double_t a, Double_t b) { fEMCALTimeCutMin = a ;
d2655d46 150 fEMCALTimeCutMax = b ; } // ns
ff440946 151
152 void SetEMCALParametrizedMinTimeCut(Int_t i, Float_t par) { fEMCALParamTimeCutMin[i] = par ; }
153 void SetEMCALParametrizedMaxTimeCut(Int_t i, Float_t par) { fEMCALParamTimeCutMax[i] = par ; }
154
4b7f6e01 155 void SwitchOnUseEMCALTimeCut() { fUseEMCALTimeCut = kTRUE ; }
156 void SwitchOffUseEMCALTimeCut() { fUseEMCALTimeCut = kFALSE ; }
157
ff440946 158 void SwitchOnUseParametrizedTimeCut() { fUseParamTimeCut = kTRUE ; }
159 void SwitchOffUseParametrizedTimeCut() { fUseParamTimeCut = kFALSE ; }
160
cfaba834 161 // Fidutial cuts
a5fb4114 162 virtual AliFiducialCut * GetFiducialCut() {
163 if(!fFiducialCut) fFiducialCut = new AliFiducialCut();
3c75ddf7 164 return fFiducialCut ; }
165 virtual void SetFiducialCut(AliFiducialCut * const fc) { fFiducialCut = fc ; }
166 virtual Bool_t IsFiducialCutOn() const { return fCheckFidCut ; }
167 virtual void SwitchOnFiducialCut() { fCheckFidCut = kTRUE ;
168 fFiducialCut = new AliFiducialCut() ; }
169 virtual void SwitchOffFiducialCut() { fCheckFidCut = kFALSE ; }
1c5acb87 170
cfaba834 171 // Cluster origin
a5fb4114 172 Bool_t IsEMCALCluster(AliVCluster *clus) const;
173 Bool_t IsPHOSCluster (AliVCluster *clus) const;
cfaba834 174 //Patch for cluster origin for Old AODs implementation
3c75ddf7 175 void SwitchOnOldAODs() { fOldAOD = kTRUE ; }
176 void SwitchOffOldAODs() { fOldAOD = kFALSE ; }
f37fa8d2 177
cfaba834 178 // Cluster/track/cells switchs
3c75ddf7 179 Bool_t IsCTSSwitchedOn() const { return fFillCTS ; }
180 void SwitchOnCTS() { fFillCTS = kTRUE ; }
181 void SwitchOffCTS() { fFillCTS = kFALSE ; }
1c5acb87 182
3c75ddf7 183 Bool_t IsEMCALSwitchedOn() const { return fFillEMCAL ; }
184 void SwitchOnEMCAL() { fFillEMCAL = kTRUE ; }
185 void SwitchOffEMCAL() { fFillEMCAL = kFALSE ; }
1c5acb87 186
3c75ddf7 187 Bool_t IsPHOSSwitchedOn() const { return fFillPHOS ; }
188 void SwitchOnPHOS() { fFillPHOS = kTRUE ; }
189 void SwitchOffPHOS() { fFillPHOS = kFALSE ; }
1c5acb87 190
3c75ddf7 191 Bool_t IsEMCALCellsSwitchedOn() const { return fFillEMCALCells ; }
192 void SwitchOnEMCALCells() { fFillEMCALCells = kTRUE ; }
193 void SwitchOffEMCALCells() { fFillEMCALCells = kFALSE ; }
1c5acb87 194
3c75ddf7 195 Bool_t IsPHOSCellsSwitchedOn() const { return fFillPHOSCells ; }
196 void SwitchOnPHOSCells() { fFillPHOSCells = kTRUE ; }
197 void SwitchOffPHOSCells() { fFillPHOSCells = kFALSE ; }
1c5acb87 198
3bfc4732 199 Bool_t AreClustersRecalculated() const { return fRecalculateClusters ; }
200 void SwitchOnClusterRecalculation() { fRecalculateClusters = kTRUE ; }
201 void SwitchOffClusterRecalculation() { fRecalculateClusters = kFALSE ; }
202
3c75ddf7 203 Bool_t IsEmbeddedClusterSelectionOn() const { return fSelectEmbeddedClusters ; }
204 void SwitchOnEmbeddedClustersSelection() { fSelectEmbeddedClusters = kTRUE ; }
205 void SwitchOffEmbeddedClustersSelection() { fSelectEmbeddedClusters = kFALSE ; }
6060ed91 206
be518ab0 207 // Filling/ filtering / detector information access methods
a5fb4114 208 virtual Bool_t FillInputEvent(const Int_t iEntry, const char *currentFileName) ;
209 virtual void FillInputCTS() ;
210 virtual void FillInputEMCAL() ;
211 virtual void FillInputEMCALAlgorithm(AliVCluster * clus, const Int_t iclus) ;
212 virtual void FillInputPHOS() ;
213 virtual void FillInputEMCALCells() ;
214 virtual void FillInputPHOSCells() ;
215 virtual void FillInputVZERO() ;
216
21812953 217 Int_t GetV0Signal(Int_t i) const { return fV0ADC[i] ; }
218 Int_t GetV0Multiplicity(Int_t i) const { return fV0Mul[i] ; }
a5fb4114 219
21812953 220 void SetEMCALClusterListName(TString &name) { fEMCALClustersListName = name ; }
221 TString GetEMCALClusterListName() const { return fEMCALClustersListName ; }
cfaba834 222
be518ab0 223 // Arrayes with clusters/track/cells access method
21812953 224 virtual TObjArray* GetCTSTracks() const { return fCTSTracks ; }
225 virtual TObjArray* GetEMCALClusters() const { return fEMCALClusters ; }
226 virtual TObjArray* GetPHOSClusters() const { return fPHOSClusters ; }
227 virtual AliVCaloCells* GetEMCALCells() const { return fEMCALCells ; }
228 virtual AliVCaloCells* GetPHOSCells() const { return fPHOSCells ; }
cfaba834 229
230 //-------------------------------------
231 // Event/track selection methods
232 //-------------------------------------
233
cd2e4ce6 234 void AcceptFastClusterEvents() { fAcceptFastCluster = kTRUE ; }
235 void RejectFastClusterEvents() { fAcceptFastCluster = kFALSE ; }
c5693f62 236 Bool_t IsFastClusterAccepted() const { return fAcceptFastCluster ; }
cd2e4ce6 237
238 void SwitchOnLEDEventsRemoval() { fRemoveLEDEvents = kTRUE ; }
afb3af8a 239 void SwitchOffLEDEventsRemoval() { fRemoveLEDEvents = kFALSE ; }
c5693f62 240 Bool_t IsLEDEventRemoved() const { return fRemoveLEDEvents ; }
a529ae05 241 Bool_t RejectLEDEvents();
242
a5fb4114 243 void SetFiredTriggerClassName(TString name ) { fFiredTriggerClassName = name ; }
244 TString GetFiredTriggerClassName() const { return fFiredTriggerClassName ; }
43074325 245 TString GetFiredTriggerClasses() ;
cfaba834 246
21812953 247 UInt_t GetEventTriggerMask() const { return fEventTriggerMask ; }
029dea5a 248 void SetEventTriggerMaks(UInt_t evtTrig = AliVEvent::kAny)
249 { fEventTriggerMask = evtTrig ; }
c2a62a94 250
afb3af8a 251 TArrayI GetL0TriggerPatches();
252 // void RejectExoticEvents(TArrayI patches);
253 //void RejectTriggeredEventsByPileUp(TArrayI patches);
254 void MatchTriggerCluster(TArrayI patches);
255
256 Bool_t IsExoticEvent() { return fIsExoticEvent ; }
257 Bool_t IsBadCellTriggerEvent() { return fIsBadCellEvent ; }
1035a8d9 258 Bool_t IsBadMaxCellTriggerEvent() { return fIsBadMaxCellEvent ; }
afb3af8a 259 Bool_t IsTriggerMatched() { return fIsTriggerMatch ; }
260 Int_t GetTriggerClusterBC() { return fTriggerClusterBC ; }
261 Int_t GetTriggerClusterIndex() { return fTriggerClusterIndex ; }
262 Int_t GetTriggerClusterId() { return fTriggerClusterId ; }
a529ae05 263
afb3af8a 264 Float_t GetEventTriggerThreshold() { return fTriggerEventThreshold ; }
265 void SetEventTriggerThreshold(Float_t tr) { fTriggerEventThreshold = tr ; }
9bf1c947 266
267 void SetTriggerPatchTimeWindow(Int_t min, Int_t max) { fTriggerPatchTimeWindow[0] = min ;
268 fTriggerPatchTimeWindow[1] = max ; }
269
afb3af8a 270 void SwitchOffBadTriggerEventsRemoval() { fRemoveBadTriggerEvents = kFALSE ; }
271 void SwitchOnBadTriggerEventsRemoval() { fRemoveBadTriggerEvents = kTRUE ; }
272
273 void SwitchOffTriggerPatchMatching() { fTriggerPatchClusterMatch = kFALSE ; }
274 void SwitchOnTriggerPatchMatching() { fTriggerPatchClusterMatch = kTRUE ; }
275
790dea42 276 UInt_t GetMixEventTriggerMask() const { return fMixEventTriggerMask ; }
277 void SetMixEventTriggerMaks(UInt_t evtTrig = AliVEvent::kAnyINT)
278 { fMixEventTriggerMask = evtTrig ; }
279
21812953 280 Bool_t IsEventTriggerAtSEOn() const { return fEventTriggerAtSE ; }
281 void SwitchOnEventTriggerAtSE() { fEventTriggerAtSE = kTRUE ; }
282 void SwitchOffEventTriggerAtSE() { fEventTriggerAtSE = kFALSE ; }
029dea5a 283
cd2e4ce6 284 void SwitchOnEventSelection() { fDoEventSelection = kTRUE ; }
285 void SwitchOffEventSelection() { fDoEventSelection = kFALSE ; }
cc944149 286 Bool_t IsEventSelectionDone() const { return fDoEventSelection ; }
cfaba834 287
a5fb4114 288 void SwitchOnV0ANDSelection() { fDoV0ANDEventSelection = kTRUE ; }
289 void SwitchOffV0ANDSelection() { fDoV0ANDEventSelection = kFALSE ; }
290 Bool_t IsV0ANDEventSelectionDone() const { return fDoV0ANDEventSelection ; }
20218aea 291
cc944149 292 void SwitchOnVertexBCEventSelection() { fDoVertexBCEventSelection = kTRUE ; }
293 void SwitchOffVertexBCEventSelection() { fDoVertexBCEventSelection = kFALSE ; }
294 Bool_t IsVertexBCEventSelectionDone() const { return fDoVertexBCEventSelection ; }
295
a5fb4114 296 void SwitchOnPrimaryVertexSelection() { fUseEventsWithPrimaryVertex = kTRUE ; }
297 void SwitchOffPrimaryVertexSelection() { fUseEventsWithPrimaryVertex = kFALSE ; }
298 Bool_t IsPrimaryVertexSelectionDone() const { return fUseEventsWithPrimaryVertex ; }
20218aea 299
2644ead9 300 void SwitchOnRejectNoTrackEvents() { fDoRejectNoTrackEvents = kTRUE ; }
301 void SwitchOffRejectNoTrackEvents() { fDoRejectNoTrackEvents = kFALSE ; }
302 Bool_t IsEventWithNoTrackRejectionDone() const { return fDoRejectNoTrackEvents ; }
303
034e885b 304 //Time Stamp
305
b370e713 306 Double_t GetRunTimeStampMin() const { return fTimeStampRunMin ; }
307 Double_t GetRunTimeStampMax() const { return fTimeStampRunMax ; }
034e885b 308
b370e713 309 void SetRunTimeStamp(Double_t a, Double_t b) { fTimeStampRunMin = a ;
034e885b 310 fTimeStampRunMax = b ; } // seconds
311
312 Float_t GetEventTimeStampFractionMin() const { return fTimeStampEventFracMin ; }
313 Float_t GetEventTimeStampFractionMax() const { return fTimeStampEventFracMax ; }
314
315 void SetEventTimeStampFraction(Float_t a, Float_t b) { fTimeStampEventFracMin = a ;
316 fTimeStampEventFracMax = b ; }
317
318 void SwitchOnSelectEventTimeStamp() { fTimeStampEventSelect = kTRUE ; }
319 void SwitchOffSelectEventTimeStamp() { fTimeStampEventSelect = kFALSE ; }
320
b8d661af 321 Bool_t IsSelectEventTimeStampOn() {return fTimeStampEventSelect ; }
034e885b 322
3c1a2e95 323 Bool_t IsPileUpFromSPD() const ;
324 Bool_t IsPileUpFromEMCal() const ;
325 Bool_t IsPileUpFromSPDAndEMCal() const ;
326 Bool_t IsPileUpFromSPDOrEMCal() const ;
327 Bool_t IsPileUpFromSPDAndNotEMCal() const ;
328 Bool_t IsPileUpFromEMCalAndNotSPD() const ;
329 Bool_t IsPileUpFromNotSPDAndNotEMCal() const ;
330
331 void SetPileUpParamForSPD (Int_t i, Double_t param)
f5500c7a 332 { fPileUpParamSPD[i] = param ; }
333 void SetPileUpParamForEMCal(Int_t param) { fNPileUpClustersCut = param ; }
4b75cb39 334
f5500c7a 335 Int_t GetNPileUpClusters() { return fNPileUpClusters ; }
336 Int_t GetNNonPileUpClusters() { return fNNonPileUpClusters ; }
975b29fa 337
338 Int_t GetEMCalEventBC(Int_t bc) const { if(bc >=0 && bc < 19) return fEMCalBCEvent [bc] ; else return 0 ; }
339 Int_t GetTrackEventBC(Int_t bc) const { if(bc >=0 && bc < 19) return fTrackBCEvent [bc] ; else return 0 ; }
340 Int_t GetEMCalEventBCcut(Int_t bc) const { if(bc >=0 && bc < 19) return fEMCalBCEventCut[bc] ; else return 0 ; }
341 Int_t GetTrackEventBCcut(Int_t bc) const { if(bc >=0 && bc < 19) return fTrackBCEventCut[bc] ; else return 0 ; }
342
343 void SetEMCalEventBC(Int_t bc) { if(bc >=0 && bc < 19) fEMCalBCEvent [bc] = 1 ; }
344 void SetTrackEventBC(Int_t bc) { if(bc >=0 && bc < 19) fTrackBCEvent [bc] = 1 ; }
345 void SetEMCalEventBCcut(Int_t bc) { if(bc >=0 && bc < 19) fEMCalBCEventCut[bc] = 1 ; }
346 void SetTrackEventBCcut(Int_t bc) { if(bc >=0 && bc < 19) fTrackBCEventCut[bc] = 1 ; }
3c1a2e95 347
c2a62a94 348 Int_t GetVertexBC(const AliVVertex * vtx);
349 Int_t GetVertexBC() const { return fVertexBC ; }
350 void SwitchOnRecalculateVertexBC() { fRecalculateVertexBC = kTRUE ; }
351 void SwitchOffRecalculateVertexBC() { fRecalculateVertexBC = kFALSE ; }
975b29fa 352
3a58eee6 353 // Track selection
f5500c7a 354 ULong_t GetTrackStatus() const { return fTrackStatus ; }
355 void SetTrackStatus(ULong_t bit) { fTrackStatus = bit ; }
a5fb4114 356
f5500c7a 357 ULong_t GetTrackFilterMask() const {return fTrackFilterMask ; }
358 void SetTrackFilterMask(ULong_t bit) { fTrackFilterMask = bit ; }
3a58eee6 359
f5500c7a 360 AliESDtrackCuts* GetTrackCuts() const { return fESDtrackCuts ; }
c5693f62 361 void SetTrackCuts(AliESDtrackCuts * cuts) ;
21812953 362
d7601185 363 AliESDtrackCuts* GetTrackComplementaryCuts() const { return fESDtrackComplementaryCuts ; }
364 void SetTrackComplementaryCuts(AliESDtrackCuts * cuts) ;
365
366
f5500c7a 367 void SwitchOnConstrainTrackToVertex() { fConstrainTrack = kTRUE ; }
d7601185 368 void SwitchOffConstrainTrackToVertex() { fConstrainTrack = kFALSE ; }
a9b8c1d0 369
21812953 370 void SwitchOnAODHybridTrackSelection() { fSelectHybridTracks = kTRUE ; }
371 void SwitchOffAODHybridTrackSelection() { fSelectHybridTracks = kFALSE ; }
c5693f62 372
f5500c7a 373 void SwitchOnTrackHitSPDSelection() { fSelectSPDHitTracks = kTRUE ; }
374 void SwitchOffTrackHitSPDSelection() { fSelectSPDHitTracks = kFALSE ; }
375
376 Int_t GetTrackMultiplicity() const { return fTrackMult ; }
377 Float_t GetTrackMultiplicityEtaCut() const { return fTrackMultEtaCut ; }
378 void SetTrackMultiplicityEtaCut(Float_t eta) { fTrackMultEtaCut = eta ; }
c8fe2783 379
a5fb4114 380 // Calorimeter specific and patches
f5500c7a 381 void AnalyzeOnlyLED() { fAnaLED = kTRUE ; }
382 void AnalyzeOnlyPhysics() { fAnaLED = kFALSE ; }
a5fb4114 383
f5500c7a 384 void SwitchOnCaloFilterPatch() { fCaloFilterPatch = kTRUE ;
385 fFillCTS = kFALSE ; }
386 void SwitchOffCaloFilterPatch() { fCaloFilterPatch = kFALSE ; }
a5fb4114 387 Bool_t IsCaloFilterPatchOn() const {
388 if(fDataType == kAOD) { return fCaloFilterPatch ; }
f5500c7a 389 else { return kFALSE ; } }
cfaba834 390
391 //-------------------------------
48c37e02 392 //Vertex methods
cfaba834 393 //-------------------------------
a5fb4114 394 virtual void GetVertex(Double_t v[3]) const ;
395 virtual Double_t* GetVertex(const Int_t evtIndex) const { return fVertex[evtIndex] ; }
396 virtual void GetVertex(Double_t vertex[3], const Int_t evtIndex) const ;
cfaba834 397 virtual void FillVertexArray();
398 virtual Bool_t CheckForPrimaryVertex();
a5fb4114 399 virtual Float_t GetZvertexCut() const { return fZvtxCut ; } //cut on vertex position
400 virtual void SetZvertexCut(Float_t zcut=10.) { fZvtxCut=zcut ; } //cut on vertex position
0b13c1f9 401
05d0d05d 402 //--------------------------
72542aba 403 // Centrality / Event Plane
05d0d05d 404 //--------------------------
ee893147 405 virtual AliCentrality* GetCentrality() const { if(fDataType!=kMC) return fInputEvent->GetCentrality() ;
406 else return 0x0 ; }
a5fb4114 407 virtual void SetCentralityClass(TString name) { fCentralityClass = name ; }
408 virtual void SetCentralityOpt(Int_t opt) { fCentralityOpt = opt ; }
409 virtual TString GetCentralityClass() const { return fCentralityClass ; }
410 virtual Int_t GetCentralityOpt() const { return fCentralityOpt ; }
411 virtual Int_t GetEventCentrality() const ;
412 virtual void SetCentralityBin(Int_t min, Int_t max) //Set the centrality bin to select the event. If used, then need to get percentile
413 { fCentralityBin[0]=min; fCentralityBin[1]=max;
414 if(min>=0 && max > 0) fCentralityOpt = 100 ; }
415 virtual Float_t GetCentralityBin(Int_t i) const { if(i < 0 || i > 1) return 0 ;
416 else return fCentralityBin[i] ; }
417
ee893147 418 virtual AliEventplane* GetEventPlane() const { if(fDataType!=kMC) return fInputEvent->GetEventplane() ;
419 else return 0x0 ; }
9929c80b 420 virtual Double_t GetEventPlaneAngle() const ;
3c75ddf7 421 virtual void SetEventPlaneMethod(TString m) { fEventPlaneMethod = m ; }
422 virtual TString GetEventPlaneMethod() const { return fEventPlaneMethod ; }
72542aba 423
39364a52 424 //--------------------
425 // Mixing
426 //--------------------
427
d783becb 428 Int_t GetLastCaloMixedEvent() const { return fLastMixedCaloEvent ; }
39364a52 429 Int_t GetLastTracksMixedEvent () const { return fLastMixedTracksEvent ; }
430
50551817 431 TList * GetListWithMixedEventsForCalo (Int_t bi) const { if(fListMixedCaloEvents) return fListMixedCaloEvents[bi] ; else return 0 ; }
432 TList * GetListWithMixedEventsForTracks(Int_t bi) const { if(fListMixedTracksEvents) return fListMixedTracksEvents [bi] ; else return 0 ; }
39364a52 433
d783becb 434 Bool_t ListWithMixedEventsForCaloExists() const { if(fListMixedCaloEvents) return kTRUE ;
435 else return kFALSE ; }
39364a52 436
d783becb 437 Bool_t ListWithMixedEventsForTracksExists() const { if(fListMixedTracksEvents) return kTRUE ;
438 else return kFALSE ; }
39364a52 439
d783becb 440 void SetLastCaloMixedEvent (Int_t e) { fLastMixedCaloEvent = e ; }
441 void SetLastTracksMixedEvent(Int_t e) { fLastMixedTracksEvent = e ; }
39364a52 442
d783becb 443 void SetListWithMixedEventsForCalo (TList ** l) {
444 if(fListMixedCaloEvents) printf("AliCaloTrackReader::SetListWithMixedEventsForCalo() - Track Mixing event list already set, nothing done\n");
445 else fListMixedCaloEvents = l ; }
39364a52 446
d783becb 447 void SetListWithMixedEventsForTracks(TList ** l) {
448 if(fListMixedTracksEvents) printf("AliCaloTrackReader::SetListWithMixedEventsForTracks() - Calorimeter Mixing event list already set, nothing done\n");
39364a52 449 else fListMixedTracksEvents = l ; }
450
cfaba834 451 //-------------------------------------
452 // Other methods
453 //-------------------------------------
3c75ddf7 454 AliCalorimeterUtils * GetCaloUtils() const { return fCaloUtils ; }
455 void SetCaloUtils(AliCalorimeterUtils * caloutils) { fCaloUtils = caloutils ; }
48c37e02 456
a5fb4114 457 virtual Double_t GetBField() const { return fInputEvent->GetMagneticField() ; }
cfaba834 458
de0b770d 459 void SetImportGeometryFromFile(Bool_t import,
55d66f31 460 TString path = "") {
de0b770d 461 fImportGeometryFromFile = import ;
462 fImportGeometryFilePath = path ; }
463
cfaba834 464 //------------------------------------------------
465 // MC analysis specific methods
466 //-------------------------------------------------
467
468 //Kinematics and galice.root available
a5fb4114 469 virtual AliStack* GetStack() const ;
470 virtual AliHeader* GetHeader() const ;
471 virtual AliGenEventHeader* GetGenEventHeader() const ;
cfaba834 472
473 //Filtered kinematics in AOD
2644ead9 474 virtual TClonesArray* GetAODMCParticles() const ;
475 virtual AliAODMCHeader* GetAODMCHeader () const ;
2644ead9 476
3c75ddf7 477 virtual AliVEvent* GetInputEvent() const { return fInputEvent ; }
478 virtual AliVEvent* GetOriginalInputEvent() const { return 0x0 ; }
479 virtual AliAODEvent* GetOutputEvent() const { return fOutputEvent ; }
480 virtual AliMCEvent* GetMC() const { return fMC ; }
481 virtual AliMixedEvent* GetMixedEvent() const { return fMixedEvent ; }
482 virtual Int_t GetNMixedEvent() const { return fNMixedEvent ; }
a5fb4114 483
3c75ddf7 484 void SwitchOnStack() { fReadStack = kTRUE ; }
485 void SwitchOffStack() { fReadStack = kFALSE ; }
486 void SwitchOnAODMCParticles() { fReadAODMCParticles = kTRUE ; }
487 void SwitchOffAODMCParticles() { fReadAODMCParticles = kFALSE ; }
488 Bool_t ReadStack() const { return fReadStack ; }
489 Bool_t ReadAODMCParticles() const { return fReadAODMCParticles ; }
cfaba834 490
491 //Select generated events, depending on comparison of pT hard and jets.
a5fb4114 492 virtual Bool_t ComparePtHardAndJetPt() ;
493 virtual Bool_t IsPtHardAndJetPtComparisonSet() const { return fComparePtHardAndJetPt ; }
494 virtual void SetPtHardAndJetPtComparison(Bool_t compare) { fComparePtHardAndJetPt = compare ; }
495 virtual Float_t GetPtHardAndJetFactor() const { return fPtHardAndJetPtFactor ; }
496 virtual void SetPtHardAndJetPtFactor(Float_t factor) { fPtHardAndJetPtFactor = factor ; }
eb3e2665 497
dbb3de7b 498 virtual Bool_t ComparePtHardAndClusterPt() ;
499 virtual Bool_t IsPtHardAndClusterPtComparisonSet() const { return fComparePtHardAndClusterPt ; }
500 virtual void SetPtHardAndClusterPtComparison(Bool_t compare) { fComparePtHardAndClusterPt = compare ; }
501 virtual Float_t GetPtHardAndClusterFactor() const { return fPtHardAndClusterPtFactor ; }
502 virtual void SetPtHardAndClusterPtFactor(Float_t factor) { fPtHardAndClusterPtFactor = factor ; }
503
504
cfaba834 505 //MC reader methods, declared there to allow compilation, they are only used in the MC reader:
0ae57829 506
a5fb4114 507 virtual void AddNeutralParticlesArray(TArrayI & /*array*/) { ; }
508 virtual void AddChargedParticlesArray(TArrayI & /*array*/) { ; }
509 virtual void AddStatusArray(TArrayI & /*array*/) { ; }
510
511 virtual void SwitchOnPi0Decay() { ; }
512 virtual void SwitchOffPi0Decay() { ; }
513 virtual void SwitchOnStatusSelection() { ; }
514 virtual void SwitchOffStatusSelection() { ; }
515 virtual void SwitchOnOverlapCheck() { ; }
516 virtual void SwitchOffOverlapCheck() { ; }
517 virtual void SwitchOnOnlyGeneratorParticles() { ; }
518 virtual void SwitchOffOnlyGeneratorParticles() { ; }
be518ab0 519
a5fb4114 520 virtual void SetEMCALOverlapAngle(Float_t /*angle*/) { ; }
521 virtual void SetPHOSOverlapAngle(Float_t /*angle*/) { ; }
af7b3903 522
0ae57829 523
1c5acb87 524 protected:
21812953 525 Int_t fEventNumber; // Event number
526 Int_t fDataType ; // Select MC:Kinematics, Data:ESD/AOD, MCData:Both
527 Int_t fDebug; // Debugging level
70752f60 528 AliFiducialCut * fFiducialCut; // Acceptance cuts
21812953 529 Bool_t fCheckFidCut ; // Do analysis for clusters in defined region
decca433 530
591cc579 531 Bool_t fComparePtHardAndJetPt; // In MonteCarlo, jet events, reject fake events with wrong jet energy.
532 Float_t fPtHardAndJetPtFactor; // Factor between ptHard and jet pT to reject/accept event.
533
dbb3de7b 534 Bool_t fComparePtHardAndClusterPt; // In MonteCarlo, jet events, reject events with too large cluster energy
535 Float_t fPtHardAndClusterPtFactor; // Factor between ptHard and cluster pT to reject/accept event.
536
21812953 537 Float_t fCTSPtMin; // pT Threshold on charged particles
538 Float_t fEMCALPtMin; // pT Threshold on emcal clusters
539 Float_t fPHOSPtMin; // pT Threshold on phos clusters
540 Float_t fCTSPtMax; // pT Threshold on charged particles
541 Float_t fEMCALPtMax; // pT Threshold on emcal clusters
542 Float_t fPHOSPtMax; // pT Threshold on phos clusters
4b7f6e01 543 Bool_t fUseEMCALTimeCut; // Do time cut selection
544 Bool_t fUseParamTimeCut; // Use simple or parametrized time cut
545 Bool_t fUseTrackTimeCut; // Do time cut selection
21812953 546 Double_t fEMCALTimeCutMin; // Remove clusters/cells with time smaller than this value, in ns
547 Double_t fEMCALTimeCutMax; // Remove clusters/cells with time larger than this value, in ns
ff440946 548 Float_t fEMCALParamTimeCutMin[4];// Remove clusters/cells with time smaller than parametrized value, in ns
549 Double_t fEMCALParamTimeCutMax[4];// Remove clusters/cells with time larger than parametrized value, in ns
975b29fa 550 Double_t fTrackTimeCutMin; // Remove tracks with time smaller than this value, in ns
551 Double_t fTrackTimeCutMax; // Remove tracks with time larger than this value, in ns
689d9f15 552 Bool_t fUseTrackDCACut; // Do DCA selection
cc944149 553 Double_t fTrackDCACut[3]; // Remove tracks with DCA larger than cut, parameters of function stored here
689d9f15 554
975b29fa 555 TList * fAODBranchList ; //-> List with AOD branches created and needed in analysis
21812953 556 TObjArray * fCTSTracks ; //-> temporal array with tracks
557 TObjArray * fEMCALClusters ; //-> temporal array with EMCAL CaloClusters
558 TObjArray * fPHOSClusters ; //-> temporal array with PHOS CaloClusters
559 AliVCaloCells * fEMCALCells ; //! temporal array with EMCAL CaloCells
560 AliVCaloCells * fPHOSCells ; //! temporal array with PHOS CaloCells
561
562 AliVEvent * fInputEvent; //! pointer to esd or aod input
563 AliAODEvent * fOutputEvent; //! pointer to aod output
564 AliMCEvent * fMC; //! Monte Carlo Event Handler
565
566 Bool_t fFillCTS; // use data from CTS
567 Bool_t fFillEMCAL; // use data from EMCAL
568 Bool_t fFillPHOS; // use data from PHOS
569 Bool_t fFillEMCALCells; // use data from EMCAL
570 Bool_t fFillPHOSCells; // use data from PHOS
3bfc4732 571 Bool_t fRecalculateClusters; // Correct clusters, recalculate them if recalibration parameters is given
572 Bool_t fSelectEmbeddedClusters; // Use only simulated clusters that come from embedding.
48c37e02 573
21812953 574 ULong_t fTrackStatus ; // Track selection bit, select tracks refitted in TPC, ITS ...
575 ULong_t fTrackFilterMask ; // Track selection bit, for AODs (any difference with track status?)
d7601185 576 AliESDtrackCuts *fESDtrackCuts ; // Track cut
577 AliESDtrackCuts *fESDtrackComplementaryCuts; // Track cut, complementary cuts for hybrids
a9b8c1d0 578 Bool_t fConstrainTrack ; // Constrain Track to vertex
21812953 579 Bool_t fSelectHybridTracks ; // Select CTS tracks of type hybrid (only for AODs)
f5500c7a 580 Bool_t fSelectSPDHitTracks ; // Ensure that track hits SPD layers
21812953 581 Int_t fTrackMult ; // Track multiplicity
582 Float_t fTrackMultEtaCut ; // Track multiplicity eta cut
583 Bool_t fReadStack ; // Access kine information from stack
584 Bool_t fReadAODMCParticles ; // Access kine information from filtered AOD MC particles
591cc579 585
21812953 586 TString fDeltaAODFileName ; // Delta AOD file name
587 TString fFiredTriggerClassName; // Name of trigger event type used to do the analysis
afabc52f 588
21812953 589 UInt_t fEventTriggerMask ; // select this triggerered event
790dea42 590 UInt_t fMixEventTriggerMask ; // select this triggerered event for mixing, tipically kMB or kAnyINT
21812953 591 Bool_t fEventTriggerAtSE; // select triggered event at SE base task or here
029dea5a 592
21812953 593 Bool_t fAnaLED; // Analyze LED data only.
c1ac3823 594
21812953 595 TString fTaskName; // Name of task that executes the analysis
765d44e7 596
21812953 597 AliCalorimeterUtils * fCaloUtils ; // Pointer to CalorimeterUtils
c8fe2783 598
21812953 599 AliMixedEvent * fMixedEvent ; //! mixed event object. This class is not the owner
600 Int_t fNMixedEvent ; // number of events in mixed event buffer
601 Double_t ** fVertex ; //! vertex array 3 dim for each mixed event buffer
c8fe2783 602
39364a52 603 TList ** fListMixedTracksEvents ; //! Container for tracks stored for different events, used in case of own mixing, set in analysis class
d783becb 604 TList ** fListMixedCaloEvents; //! Container for photon stored for different events, used in case of own mixing, set in analysis class
39364a52 605 Int_t fLastMixedTracksEvent ; // Temporary container with the last event added to the mixing list for tracks
d783becb 606 Int_t fLastMixedCaloEvent ; // Temporary container with the last event added to the mixing list for photons
39364a52 607
21812953 608 Bool_t fWriteOutputDeltaAOD; // Write the created delta AOD objects into file
609 Bool_t fOldAOD; // Old AODs, before revision 4.20
f37fa8d2 610
21812953 611 Int_t fV0ADC[2] ; // Integrated V0 signal
612 Int_t fV0Mul[2] ; // Integrated V0 Multiplicity
798a9b04 613
21812953 614 Bool_t fCaloFilterPatch; // CaloFilter patch
615 TString fEMCALClustersListName; // Alternative list of clusters produced elsewhere and not from InputEvent
cd2e4ce6 616
617 // Event selection
618 Float_t fZvtxCut ; // Cut on vertex position
619 Bool_t fAcceptFastCluster; // Accept events from fast cluster, exclude these events for LHC11a
620 Bool_t fRemoveLEDEvents; // Remove events where LED was wrongly firing - EMCAL LHC11a
9bf1c947 621
afb3af8a 622 Bool_t fRemoveBadTriggerEvents; // Remove triggered events because trigger was exotic, bad, or out of BC
623 Bool_t fTriggerPatchClusterMatch; // Search for the trigger patch and check if associated cluster was the trigger
9bf1c947 624 Int_t fTriggerPatchTimeWindow[2]; // Trigger patch selection window
afb3af8a 625 Float_t fTriggerEventThreshold; // Threshold to look for triggered events
626 Int_t fTriggerClusterBC; // Event triggered by a cluster in BC -5 0 to 5
627 Int_t fTriggerClusterIndex; // Index in clusters array of trigger cluster
628 Int_t fTriggerClusterId; // Id of trigger cluster (cluster->GetID())
629 Bool_t fIsExoticEvent; // Exotic trigger event flag
1035a8d9 630 Bool_t fIsBadCellEvent; // Bad cell triggered event flag, any cell in cluster is bad
631 Bool_t fIsBadMaxCellEvent; // Bad cell triggered event flag, only max energy cell is bad
afb3af8a 632 Int_t fIsTriggerMatch; // Could not match the event to a trigger patch
9bf1c947 633
20218aea 634 Bool_t fDoEventSelection; // Select events depending on V0, pileup, vertex well reconstructed, at least 1 track ...
635 Bool_t fDoV0ANDEventSelection; // Select events depending on V0, fDoEventSelection should be on
cc944149 636 Bool_t fDoVertexBCEventSelection; // Select events with vertex on BC=0 or -100
2644ead9 637 Bool_t fDoRejectNoTrackEvents; // Reject events with no selected tracks in event
20218aea 638 Bool_t fUseEventsWithPrimaryVertex ; // Select events with primary vertex
639 AliTriggerAnalysis* fTriggerAnalysis; // Access to trigger selection algorithm for V0AND calculation
034e885b 640
641 Bool_t fTimeStampEventSelect; // Select events within a fraction of data taking time
642 Float_t fTimeStampEventFracMin; // Minimum value of time stamp fraction event
643 Float_t fTimeStampEventFracMax; // Maximum value of time stamp fraction event
b370e713 644 Double_t fTimeStampRunMin; // Minimum value of time stamp in run
645 Double_t fTimeStampRunMax; // Maximum value of time stamp in run
034e885b 646
3c1a2e95 647 Double_t fPileUpParamSPD[5]; // Parameters to pass to method IsPileupFromSPD: Int_t minContributors,
4b75cb39 648 // Double_t minZdist,
649 // Double_t nSigmaZdist,
650 // Double_t nSigmaDiamXY,
651 // Double_t nSigmaDiamZ)
034e885b 652
3c1a2e95 653 // Pile-up in EMCal
654 Int_t fNPileUpClusters; // Number of clusters with time avobe 20 ns
655 Int_t fNNonPileUpClusters; // Number of clusters with time below 20 ns
656 Int_t fNPileUpClustersCut; // Cut to select event as pile-up
975b29fa 657 Int_t fEMCalBCEvent[19]; // Fill one entry per event if there is a cluster in a given BC
658 Int_t fEMCalBCEventCut[19]; // Fill one entry per event if there is a cluster in a given BC, depend on cluster E, acceptance cut
659 Int_t fTrackBCEvent[19]; // Fill one entry per event if there is a track in a given BC
660 Int_t fTrackBCEventCut[19]; // Fill one entry per event if there is a track in a given BC, depend on track pT, acceptance cut
cc944149 661 Int_t fVertexBC; // Vertex BC
662 Bool_t fRecalculateVertexBC; // Recalculate vertex BC from tracks pointing to vertex
663
72542aba 664 //Centrality/Event plane
21812953 665 TString fCentralityClass; // Name of selected centrality class
666 Int_t fCentralityOpt; // Option for the returned value of the centrality, possible options 5, 10, 100
667 Int_t fCentralityBin[2]; // Minimum and maximum value of the centrality for the analysis
668 TString fEventPlaneMethod; // Name of event plane method, by default "Q"
32fd29fe 669
de0b770d 670 Bool_t fImportGeometryFromFile; // Import geometry settings in geometry.root file
671 TString fImportGeometryFilePath; // path fo geometry.root file
672
3c1a2e95 673
0de1814a 674 AliCaloTrackReader( const AliCaloTrackReader & r) ; // cpy ctor
675 AliCaloTrackReader & operator = (const AliCaloTrackReader & r) ; // cpy assignment
f3138ecf 676
1035a8d9 677 ClassDef(AliCaloTrackReader,55)
f3138ecf 678
1c5acb87 679} ;
680
681
682#endif //ALICALOTRACKREADER_H
683
684
685