]> 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
from Redmer Bertens:
[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 ; }
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 TArrayI GetL0TriggerPatches();
251 void RejectExoticEvents(TArrayI patches);
a529ae05 252 Bool_t IsExoticEvent() { return fIsExoticEvent ; }
c2a62a94 253 void SwitchOffExoticEventsRemoval() { fRemoveExoticEvents = kFALSE ; }
08f2fa0f 254 void SwitchOnExoticEventsRemoval(Bool_t all = kFALSE)
255 { fRemoveExoticEvents = kTRUE ;
256 fForceExoticRejection = all ; }
c2a62a94 257
258 Int_t IsPileUpClusterTriggeredEvent() { return fIsTriggerEventOutBC ; }
259 void SwitchOffTriggerOutBCEventsRemoval() { fRemoveTriggerOutBCEvents = kFALSE ; }
260 void SwitchOnTriggerOutBCEventsRemoval() { fRemoveTriggerOutBCEvents = kTRUE ; }
261
262 void RejectTriggeredEventsByPileUp(TArrayI patches);
263
264 void SetEventTriggerThreshold(Float_t tr) { fTriggerEventThreshold = tr ; }
a529ae05 265
9bf1c947 266 void SwitchOffExoticEventsFromTriggerPatch() { fTriggerPatchExoticRejection = kFALSE ; }
267 void SwitchOnExoticEventsFromTriggerPatch() { fTriggerPatchExoticRejection = kTRUE ; }
268
269 void SetTriggerPatchTimeWindow(Int_t min, Int_t max) { fTriggerPatchTimeWindow[0] = min ;
270 fTriggerPatchTimeWindow[1] = max ; }
271
790dea42 272 UInt_t GetMixEventTriggerMask() const { return fMixEventTriggerMask ; }
273 void SetMixEventTriggerMaks(UInt_t evtTrig = AliVEvent::kAnyINT)
274 { fMixEventTriggerMask = evtTrig ; }
275
21812953 276 Bool_t IsEventTriggerAtSEOn() const { return fEventTriggerAtSE ; }
277 void SwitchOnEventTriggerAtSE() { fEventTriggerAtSE = kTRUE ; }
278 void SwitchOffEventTriggerAtSE() { fEventTriggerAtSE = kFALSE ; }
029dea5a 279
cd2e4ce6 280 void SwitchOnEventSelection() { fDoEventSelection = kTRUE ; }
281 void SwitchOffEventSelection() { fDoEventSelection = kFALSE ; }
cc944149 282 Bool_t IsEventSelectionDone() const { return fDoEventSelection ; }
cfaba834 283
a5fb4114 284 void SwitchOnV0ANDSelection() { fDoV0ANDEventSelection = kTRUE ; }
285 void SwitchOffV0ANDSelection() { fDoV0ANDEventSelection = kFALSE ; }
286 Bool_t IsV0ANDEventSelectionDone() const { return fDoV0ANDEventSelection ; }
20218aea 287
cc944149 288 void SwitchOnVertexBCEventSelection() { fDoVertexBCEventSelection = kTRUE ; }
289 void SwitchOffVertexBCEventSelection() { fDoVertexBCEventSelection = kFALSE ; }
290 Bool_t IsVertexBCEventSelectionDone() const { return fDoVertexBCEventSelection ; }
291
a5fb4114 292 void SwitchOnPrimaryVertexSelection() { fUseEventsWithPrimaryVertex = kTRUE ; }
293 void SwitchOffPrimaryVertexSelection() { fUseEventsWithPrimaryVertex = kFALSE ; }
294 Bool_t IsPrimaryVertexSelectionDone() const { return fUseEventsWithPrimaryVertex ; }
20218aea 295
2644ead9 296 void SwitchOnRejectNoTrackEvents() { fDoRejectNoTrackEvents = kTRUE ; }
297 void SwitchOffRejectNoTrackEvents() { fDoRejectNoTrackEvents = kFALSE ; }
298 Bool_t IsEventWithNoTrackRejectionDone() const { return fDoRejectNoTrackEvents ; }
299
034e885b 300 //Time Stamp
301
b370e713 302 Double_t GetRunTimeStampMin() const { return fTimeStampRunMin ; }
303 Double_t GetRunTimeStampMax() const { return fTimeStampRunMax ; }
034e885b 304
b370e713 305 void SetRunTimeStamp(Double_t a, Double_t b) { fTimeStampRunMin = a ;
034e885b 306 fTimeStampRunMax = b ; } // seconds
307
308 Float_t GetEventTimeStampFractionMin() const { return fTimeStampEventFracMin ; }
309 Float_t GetEventTimeStampFractionMax() const { return fTimeStampEventFracMax ; }
310
311 void SetEventTimeStampFraction(Float_t a, Float_t b) { fTimeStampEventFracMin = a ;
312 fTimeStampEventFracMax = b ; }
313
314 void SwitchOnSelectEventTimeStamp() { fTimeStampEventSelect = kTRUE ; }
315 void SwitchOffSelectEventTimeStamp() { fTimeStampEventSelect = kFALSE ; }
316
b8d661af 317 Bool_t IsSelectEventTimeStampOn() {return fTimeStampEventSelect ; }
034e885b 318
3c1a2e95 319 Bool_t IsPileUpFromSPD() const ;
320 Bool_t IsPileUpFromEMCal() const ;
321 Bool_t IsPileUpFromSPDAndEMCal() const ;
322 Bool_t IsPileUpFromSPDOrEMCal() const ;
323 Bool_t IsPileUpFromSPDAndNotEMCal() const ;
324 Bool_t IsPileUpFromEMCalAndNotSPD() const ;
325 Bool_t IsPileUpFromNotSPDAndNotEMCal() const ;
326
327 void SetPileUpParamForSPD (Int_t i, Double_t param)
f5500c7a 328 { fPileUpParamSPD[i] = param ; }
329 void SetPileUpParamForEMCal(Int_t param) { fNPileUpClustersCut = param ; }
4b75cb39 330
f5500c7a 331 Int_t GetNPileUpClusters() { return fNPileUpClusters ; }
332 Int_t GetNNonPileUpClusters() { return fNNonPileUpClusters ; }
975b29fa 333
334 Int_t GetEMCalEventBC(Int_t bc) const { if(bc >=0 && bc < 19) return fEMCalBCEvent [bc] ; else return 0 ; }
335 Int_t GetTrackEventBC(Int_t bc) const { if(bc >=0 && bc < 19) return fTrackBCEvent [bc] ; else return 0 ; }
336 Int_t GetEMCalEventBCcut(Int_t bc) const { if(bc >=0 && bc < 19) return fEMCalBCEventCut[bc] ; else return 0 ; }
337 Int_t GetTrackEventBCcut(Int_t bc) const { if(bc >=0 && bc < 19) return fTrackBCEventCut[bc] ; else return 0 ; }
338
339 void SetEMCalEventBC(Int_t bc) { if(bc >=0 && bc < 19) fEMCalBCEvent [bc] = 1 ; }
340 void SetTrackEventBC(Int_t bc) { if(bc >=0 && bc < 19) fTrackBCEvent [bc] = 1 ; }
341 void SetEMCalEventBCcut(Int_t bc) { if(bc >=0 && bc < 19) fEMCalBCEventCut[bc] = 1 ; }
342 void SetTrackEventBCcut(Int_t bc) { if(bc >=0 && bc < 19) fTrackBCEventCut[bc] = 1 ; }
3c1a2e95 343
c2a62a94 344 Int_t GetVertexBC(const AliVVertex * vtx);
345 Int_t GetVertexBC() const { return fVertexBC ; }
346 void SwitchOnRecalculateVertexBC() { fRecalculateVertexBC = kTRUE ; }
347 void SwitchOffRecalculateVertexBC() { fRecalculateVertexBC = kFALSE ; }
975b29fa 348
3a58eee6 349 // Track selection
f5500c7a 350 ULong_t GetTrackStatus() const { return fTrackStatus ; }
351 void SetTrackStatus(ULong_t bit) { fTrackStatus = bit ; }
a5fb4114 352
f5500c7a 353 ULong_t GetTrackFilterMask() const {return fTrackFilterMask ; }
354 void SetTrackFilterMask(ULong_t bit) { fTrackFilterMask = bit ; }
3a58eee6 355
f5500c7a 356 AliESDtrackCuts* GetTrackCuts() const { return fESDtrackCuts ; }
c5693f62 357 void SetTrackCuts(AliESDtrackCuts * cuts) ;
21812953 358
d7601185 359 AliESDtrackCuts* GetTrackComplementaryCuts() const { return fESDtrackComplementaryCuts ; }
360 void SetTrackComplementaryCuts(AliESDtrackCuts * cuts) ;
361
362
f5500c7a 363 void SwitchOnConstrainTrackToVertex() { fConstrainTrack = kTRUE ; }
d7601185 364 void SwitchOffConstrainTrackToVertex() { fConstrainTrack = kFALSE ; }
a9b8c1d0 365
21812953 366 void SwitchOnAODHybridTrackSelection() { fSelectHybridTracks = kTRUE ; }
367 void SwitchOffAODHybridTrackSelection() { fSelectHybridTracks = kFALSE ; }
c5693f62 368
f5500c7a 369 void SwitchOnTrackHitSPDSelection() { fSelectSPDHitTracks = kTRUE ; }
370 void SwitchOffTrackHitSPDSelection() { fSelectSPDHitTracks = kFALSE ; }
371
372 Int_t GetTrackMultiplicity() const { return fTrackMult ; }
373 Float_t GetTrackMultiplicityEtaCut() const { return fTrackMultEtaCut ; }
374 void SetTrackMultiplicityEtaCut(Float_t eta) { fTrackMultEtaCut = eta ; }
c8fe2783 375
a5fb4114 376 // Calorimeter specific and patches
f5500c7a 377 void AnalyzeOnlyLED() { fAnaLED = kTRUE ; }
378 void AnalyzeOnlyPhysics() { fAnaLED = kFALSE ; }
a5fb4114 379
f5500c7a 380 void SwitchOnCaloFilterPatch() { fCaloFilterPatch = kTRUE ;
381 fFillCTS = kFALSE ; }
382 void SwitchOffCaloFilterPatch() { fCaloFilterPatch = kFALSE ; }
a5fb4114 383 Bool_t IsCaloFilterPatchOn() const {
384 if(fDataType == kAOD) { return fCaloFilterPatch ; }
f5500c7a 385 else { return kFALSE ; } }
cfaba834 386
387 //-------------------------------
48c37e02 388 //Vertex methods
cfaba834 389 //-------------------------------
a5fb4114 390 virtual void GetVertex(Double_t v[3]) const ;
391 virtual Double_t* GetVertex(const Int_t evtIndex) const { return fVertex[evtIndex] ; }
392 virtual void GetVertex(Double_t vertex[3], const Int_t evtIndex) const ;
cfaba834 393 virtual void FillVertexArray();
394 virtual Bool_t CheckForPrimaryVertex();
a5fb4114 395 virtual Float_t GetZvertexCut() const { return fZvtxCut ; } //cut on vertex position
396 virtual void SetZvertexCut(Float_t zcut=10.) { fZvtxCut=zcut ; } //cut on vertex position
0b13c1f9 397
05d0d05d 398 //--------------------------
72542aba 399 // Centrality / Event Plane
05d0d05d 400 //--------------------------
ee893147 401 virtual AliCentrality* GetCentrality() const { if(fDataType!=kMC) return fInputEvent->GetCentrality() ;
402 else return 0x0 ; }
a5fb4114 403 virtual void SetCentralityClass(TString name) { fCentralityClass = name ; }
404 virtual void SetCentralityOpt(Int_t opt) { fCentralityOpt = opt ; }
405 virtual TString GetCentralityClass() const { return fCentralityClass ; }
406 virtual Int_t GetCentralityOpt() const { return fCentralityOpt ; }
407 virtual Int_t GetEventCentrality() const ;
408 virtual void SetCentralityBin(Int_t min, Int_t max) //Set the centrality bin to select the event. If used, then need to get percentile
409 { fCentralityBin[0]=min; fCentralityBin[1]=max;
410 if(min>=0 && max > 0) fCentralityOpt = 100 ; }
411 virtual Float_t GetCentralityBin(Int_t i) const { if(i < 0 || i > 1) return 0 ;
412 else return fCentralityBin[i] ; }
413
ee893147 414 virtual AliEventplane* GetEventPlane() const { if(fDataType!=kMC) return fInputEvent->GetEventplane() ;
415 else return 0x0 ; }
9929c80b 416 virtual Double_t GetEventPlaneAngle() const ;
3c75ddf7 417 virtual void SetEventPlaneMethod(TString m) { fEventPlaneMethod = m ; }
418 virtual TString GetEventPlaneMethod() const { return fEventPlaneMethod ; }
72542aba 419
39364a52 420 //--------------------
421 // Mixing
422 //--------------------
423
d783becb 424 Int_t GetLastCaloMixedEvent() const { return fLastMixedCaloEvent ; }
39364a52 425 Int_t GetLastTracksMixedEvent () const { return fLastMixedTracksEvent ; }
426
50551817 427 TList * GetListWithMixedEventsForCalo (Int_t bi) const { if(fListMixedCaloEvents) return fListMixedCaloEvents[bi] ; else return 0 ; }
428 TList * GetListWithMixedEventsForTracks(Int_t bi) const { if(fListMixedTracksEvents) return fListMixedTracksEvents [bi] ; else return 0 ; }
39364a52 429
d783becb 430 Bool_t ListWithMixedEventsForCaloExists() const { if(fListMixedCaloEvents) return kTRUE ;
431 else return kFALSE ; }
39364a52 432
d783becb 433 Bool_t ListWithMixedEventsForTracksExists() const { if(fListMixedTracksEvents) return kTRUE ;
434 else return kFALSE ; }
39364a52 435
d783becb 436 void SetLastCaloMixedEvent (Int_t e) { fLastMixedCaloEvent = e ; }
437 void SetLastTracksMixedEvent(Int_t e) { fLastMixedTracksEvent = e ; }
39364a52 438
d783becb 439 void SetListWithMixedEventsForCalo (TList ** l) {
440 if(fListMixedCaloEvents) printf("AliCaloTrackReader::SetListWithMixedEventsForCalo() - Track Mixing event list already set, nothing done\n");
441 else fListMixedCaloEvents = l ; }
39364a52 442
d783becb 443 void SetListWithMixedEventsForTracks(TList ** l) {
444 if(fListMixedTracksEvents) printf("AliCaloTrackReader::SetListWithMixedEventsForTracks() - Calorimeter Mixing event list already set, nothing done\n");
39364a52 445 else fListMixedTracksEvents = l ; }
446
cfaba834 447 //-------------------------------------
448 // Other methods
449 //-------------------------------------
3c75ddf7 450 AliCalorimeterUtils * GetCaloUtils() const { return fCaloUtils ; }
451 void SetCaloUtils(AliCalorimeterUtils * caloutils) { fCaloUtils = caloutils ; }
48c37e02 452
a5fb4114 453 virtual Double_t GetBField() const { return fInputEvent->GetMagneticField() ; }
cfaba834 454
de0b770d 455 void SetImportGeometryFromFile(Bool_t import,
55d66f31 456 TString path = "") {
de0b770d 457 fImportGeometryFromFile = import ;
458 fImportGeometryFilePath = path ; }
459
cfaba834 460 //------------------------------------------------
461 // MC analysis specific methods
462 //-------------------------------------------------
463
464 //Kinematics and galice.root available
a5fb4114 465 virtual AliStack* GetStack() const ;
466 virtual AliHeader* GetHeader() const ;
467 virtual AliGenEventHeader* GetGenEventHeader() const ;
cfaba834 468
469 //Filtered kinematics in AOD
2644ead9 470 virtual TClonesArray* GetAODMCParticles() const ;
471 virtual AliAODMCHeader* GetAODMCHeader () const ;
2644ead9 472
3c75ddf7 473 virtual AliVEvent* GetInputEvent() const { return fInputEvent ; }
474 virtual AliVEvent* GetOriginalInputEvent() const { return 0x0 ; }
475 virtual AliAODEvent* GetOutputEvent() const { return fOutputEvent ; }
476 virtual AliMCEvent* GetMC() const { return fMC ; }
477 virtual AliMixedEvent* GetMixedEvent() const { return fMixedEvent ; }
478 virtual Int_t GetNMixedEvent() const { return fNMixedEvent ; }
a5fb4114 479
3c75ddf7 480 void SwitchOnStack() { fReadStack = kTRUE ; }
481 void SwitchOffStack() { fReadStack = kFALSE ; }
482 void SwitchOnAODMCParticles() { fReadAODMCParticles = kTRUE ; }
483 void SwitchOffAODMCParticles() { fReadAODMCParticles = kFALSE ; }
484 Bool_t ReadStack() const { return fReadStack ; }
485 Bool_t ReadAODMCParticles() const { return fReadAODMCParticles ; }
cfaba834 486
487 //Select generated events, depending on comparison of pT hard and jets.
a5fb4114 488 virtual Bool_t ComparePtHardAndJetPt() ;
489 virtual Bool_t IsPtHardAndJetPtComparisonSet() const { return fComparePtHardAndJetPt ; }
490 virtual void SetPtHardAndJetPtComparison(Bool_t compare) { fComparePtHardAndJetPt = compare ; }
491 virtual Float_t GetPtHardAndJetFactor() const { return fPtHardAndJetPtFactor ; }
492 virtual void SetPtHardAndJetPtFactor(Float_t factor) { fPtHardAndJetPtFactor = factor ; }
eb3e2665 493
dbb3de7b 494 virtual Bool_t ComparePtHardAndClusterPt() ;
495 virtual Bool_t IsPtHardAndClusterPtComparisonSet() const { return fComparePtHardAndClusterPt ; }
496 virtual void SetPtHardAndClusterPtComparison(Bool_t compare) { fComparePtHardAndClusterPt = compare ; }
497 virtual Float_t GetPtHardAndClusterFactor() const { return fPtHardAndClusterPtFactor ; }
498 virtual void SetPtHardAndClusterPtFactor(Float_t factor) { fPtHardAndClusterPtFactor = factor ; }
499
500
cfaba834 501 //MC reader methods, declared there to allow compilation, they are only used in the MC reader:
0ae57829 502
a5fb4114 503 virtual void AddNeutralParticlesArray(TArrayI & /*array*/) { ; }
504 virtual void AddChargedParticlesArray(TArrayI & /*array*/) { ; }
505 virtual void AddStatusArray(TArrayI & /*array*/) { ; }
506
507 virtual void SwitchOnPi0Decay() { ; }
508 virtual void SwitchOffPi0Decay() { ; }
509 virtual void SwitchOnStatusSelection() { ; }
510 virtual void SwitchOffStatusSelection() { ; }
511 virtual void SwitchOnOverlapCheck() { ; }
512 virtual void SwitchOffOverlapCheck() { ; }
513 virtual void SwitchOnOnlyGeneratorParticles() { ; }
514 virtual void SwitchOffOnlyGeneratorParticles() { ; }
be518ab0 515
a5fb4114 516 virtual void SetEMCALOverlapAngle(Float_t /*angle*/) { ; }
517 virtual void SetPHOSOverlapAngle(Float_t /*angle*/) { ; }
af7b3903 518
0ae57829 519
1c5acb87 520 protected:
21812953 521 Int_t fEventNumber; // Event number
522 Int_t fDataType ; // Select MC:Kinematics, Data:ESD/AOD, MCData:Both
523 Int_t fDebug; // Debugging level
70752f60 524 AliFiducialCut * fFiducialCut; // Acceptance cuts
21812953 525 Bool_t fCheckFidCut ; // Do analysis for clusters in defined region
decca433 526
591cc579 527 Bool_t fComparePtHardAndJetPt; // In MonteCarlo, jet events, reject fake events with wrong jet energy.
528 Float_t fPtHardAndJetPtFactor; // Factor between ptHard and jet pT to reject/accept event.
529
dbb3de7b 530 Bool_t fComparePtHardAndClusterPt; // In MonteCarlo, jet events, reject events with too large cluster energy
531 Float_t fPtHardAndClusterPtFactor; // Factor between ptHard and cluster pT to reject/accept event.
532
21812953 533 Float_t fCTSPtMin; // pT Threshold on charged particles
534 Float_t fEMCALPtMin; // pT Threshold on emcal clusters
535 Float_t fPHOSPtMin; // pT Threshold on phos clusters
536 Float_t fCTSPtMax; // pT Threshold on charged particles
537 Float_t fEMCALPtMax; // pT Threshold on emcal clusters
538 Float_t fPHOSPtMax; // pT Threshold on phos clusters
4b7f6e01 539 Bool_t fUseEMCALTimeCut; // Do time cut selection
540 Bool_t fUseParamTimeCut; // Use simple or parametrized time cut
541 Bool_t fUseTrackTimeCut; // Do time cut selection
21812953 542 Double_t fEMCALTimeCutMin; // Remove clusters/cells with time smaller than this value, in ns
543 Double_t fEMCALTimeCutMax; // Remove clusters/cells with time larger than this value, in ns
ff440946 544 Float_t fEMCALParamTimeCutMin[4];// Remove clusters/cells with time smaller than parametrized value, in ns
545 Double_t fEMCALParamTimeCutMax[4];// Remove clusters/cells with time larger than parametrized value, in ns
975b29fa 546 Double_t fTrackTimeCutMin; // Remove tracks with time smaller than this value, in ns
547 Double_t fTrackTimeCutMax; // Remove tracks with time larger than this value, in ns
689d9f15 548 Bool_t fUseTrackDCACut; // Do DCA selection
cc944149 549 Double_t fTrackDCACut[3]; // Remove tracks with DCA larger than cut, parameters of function stored here
689d9f15 550
975b29fa 551 TList * fAODBranchList ; //-> List with AOD branches created and needed in analysis
21812953 552 TObjArray * fCTSTracks ; //-> temporal array with tracks
553 TObjArray * fEMCALClusters ; //-> temporal array with EMCAL CaloClusters
554 TObjArray * fPHOSClusters ; //-> temporal array with PHOS CaloClusters
555 AliVCaloCells * fEMCALCells ; //! temporal array with EMCAL CaloCells
556 AliVCaloCells * fPHOSCells ; //! temporal array with PHOS CaloCells
557
558 AliVEvent * fInputEvent; //! pointer to esd or aod input
559 AliAODEvent * fOutputEvent; //! pointer to aod output
560 AliMCEvent * fMC; //! Monte Carlo Event Handler
561
562 Bool_t fFillCTS; // use data from CTS
563 Bool_t fFillEMCAL; // use data from EMCAL
564 Bool_t fFillPHOS; // use data from PHOS
565 Bool_t fFillEMCALCells; // use data from EMCAL
566 Bool_t fFillPHOSCells; // use data from PHOS
3bfc4732 567 Bool_t fRecalculateClusters; // Correct clusters, recalculate them if recalibration parameters is given
568 Bool_t fSelectEmbeddedClusters; // Use only simulated clusters that come from embedding.
48c37e02 569
21812953 570 ULong_t fTrackStatus ; // Track selection bit, select tracks refitted in TPC, ITS ...
571 ULong_t fTrackFilterMask ; // Track selection bit, for AODs (any difference with track status?)
d7601185 572 AliESDtrackCuts *fESDtrackCuts ; // Track cut
573 AliESDtrackCuts *fESDtrackComplementaryCuts; // Track cut, complementary cuts for hybrids
a9b8c1d0 574 Bool_t fConstrainTrack ; // Constrain Track to vertex
21812953 575 Bool_t fSelectHybridTracks ; // Select CTS tracks of type hybrid (only for AODs)
f5500c7a 576 Bool_t fSelectSPDHitTracks ; // Ensure that track hits SPD layers
21812953 577 Int_t fTrackMult ; // Track multiplicity
578 Float_t fTrackMultEtaCut ; // Track multiplicity eta cut
579 Bool_t fReadStack ; // Access kine information from stack
580 Bool_t fReadAODMCParticles ; // Access kine information from filtered AOD MC particles
591cc579 581
21812953 582 TString fDeltaAODFileName ; // Delta AOD file name
583 TString fFiredTriggerClassName; // Name of trigger event type used to do the analysis
afabc52f 584
21812953 585 UInt_t fEventTriggerMask ; // select this triggerered event
790dea42 586 UInt_t fMixEventTriggerMask ; // select this triggerered event for mixing, tipically kMB or kAnyINT
21812953 587 Bool_t fEventTriggerAtSE; // select triggered event at SE base task or here
029dea5a 588
21812953 589 Bool_t fAnaLED; // Analyze LED data only.
c1ac3823 590
21812953 591 TString fTaskName; // Name of task that executes the analysis
765d44e7 592
21812953 593 AliCalorimeterUtils * fCaloUtils ; // Pointer to CalorimeterUtils
c8fe2783 594
21812953 595 AliMixedEvent * fMixedEvent ; //! mixed event object. This class is not the owner
596 Int_t fNMixedEvent ; // number of events in mixed event buffer
597 Double_t ** fVertex ; //! vertex array 3 dim for each mixed event buffer
c8fe2783 598
39364a52 599 TList ** fListMixedTracksEvents ; //! Container for tracks stored for different events, used in case of own mixing, set in analysis class
d783becb 600 TList ** fListMixedCaloEvents; //! Container for photon stored for different events, used in case of own mixing, set in analysis class
39364a52 601 Int_t fLastMixedTracksEvent ; // Temporary container with the last event added to the mixing list for tracks
d783becb 602 Int_t fLastMixedCaloEvent ; // Temporary container with the last event added to the mixing list for photons
39364a52 603
21812953 604 Bool_t fWriteOutputDeltaAOD; // Write the created delta AOD objects into file
605 Bool_t fOldAOD; // Old AODs, before revision 4.20
f37fa8d2 606
21812953 607 Int_t fV0ADC[2] ; // Integrated V0 signal
608 Int_t fV0Mul[2] ; // Integrated V0 Multiplicity
798a9b04 609
21812953 610 Bool_t fCaloFilterPatch; // CaloFilter patch
611 TString fEMCALClustersListName; // Alternative list of clusters produced elsewhere and not from InputEvent
cd2e4ce6 612
613 // Event selection
614 Float_t fZvtxCut ; // Cut on vertex position
615 Bool_t fAcceptFastCluster; // Accept events from fast cluster, exclude these events for LHC11a
616 Bool_t fRemoveLEDEvents; // Remove events where LED was wrongly firing - EMCAL LHC11a
9bf1c947 617
a529ae05 618 Bool_t fRemoveExoticEvents; // Remove events triggered by exotic cluster
c2a62a94 619 Bool_t fRemoveTriggerOutBCEvents; // Remove events triggered by pile-up cluster
9bf1c947 620 Bool_t fTriggerPatchExoticRejection; // Search for the trigger patch and check if associated cluster was the trigger
621 Int_t fTriggerPatchTimeWindow[2]; // Trigger patch selection window
c2a62a94 622 Float_t fTriggerEventThreshold; // Threshold to look for triggered events
a529ae05 623 Bool_t fIsExoticEvent; // Exotic event flag
c2a62a94 624 Int_t fIsTriggerEventOutBC; // Event triggered by pile-up in BC
08f2fa0f 625 Bool_t fForceExoticRejection; // Reject events triggered by exotic only on EMC triggered events, except in MC to study false rejections
9bf1c947 626
20218aea 627 Bool_t fDoEventSelection; // Select events depending on V0, pileup, vertex well reconstructed, at least 1 track ...
628 Bool_t fDoV0ANDEventSelection; // Select events depending on V0, fDoEventSelection should be on
cc944149 629 Bool_t fDoVertexBCEventSelection; // Select events with vertex on BC=0 or -100
2644ead9 630 Bool_t fDoRejectNoTrackEvents; // Reject events with no selected tracks in event
20218aea 631 Bool_t fUseEventsWithPrimaryVertex ; // Select events with primary vertex
632 AliTriggerAnalysis* fTriggerAnalysis; // Access to trigger selection algorithm for V0AND calculation
034e885b 633
634 Bool_t fTimeStampEventSelect; // Select events within a fraction of data taking time
635 Float_t fTimeStampEventFracMin; // Minimum value of time stamp fraction event
636 Float_t fTimeStampEventFracMax; // Maximum value of time stamp fraction event
b370e713 637 Double_t fTimeStampRunMin; // Minimum value of time stamp in run
638 Double_t fTimeStampRunMax; // Maximum value of time stamp in run
034e885b 639
3c1a2e95 640 Double_t fPileUpParamSPD[5]; // Parameters to pass to method IsPileupFromSPD: Int_t minContributors,
4b75cb39 641 // Double_t minZdist,
642 // Double_t nSigmaZdist,
643 // Double_t nSigmaDiamXY,
644 // Double_t nSigmaDiamZ)
034e885b 645
3c1a2e95 646 // Pile-up in EMCal
647 Int_t fNPileUpClusters; // Number of clusters with time avobe 20 ns
648 Int_t fNNonPileUpClusters; // Number of clusters with time below 20 ns
649 Int_t fNPileUpClustersCut; // Cut to select event as pile-up
975b29fa 650 Int_t fEMCalBCEvent[19]; // Fill one entry per event if there is a cluster in a given BC
651 Int_t fEMCalBCEventCut[19]; // Fill one entry per event if there is a cluster in a given BC, depend on cluster E, acceptance cut
652 Int_t fTrackBCEvent[19]; // Fill one entry per event if there is a track in a given BC
653 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 654 Int_t fVertexBC; // Vertex BC
655 Bool_t fRecalculateVertexBC; // Recalculate vertex BC from tracks pointing to vertex
656
72542aba 657 //Centrality/Event plane
21812953 658 TString fCentralityClass; // Name of selected centrality class
659 Int_t fCentralityOpt; // Option for the returned value of the centrality, possible options 5, 10, 100
660 Int_t fCentralityBin[2]; // Minimum and maximum value of the centrality for the analysis
661 TString fEventPlaneMethod; // Name of event plane method, by default "Q"
32fd29fe 662
de0b770d 663 Bool_t fImportGeometryFromFile; // Import geometry settings in geometry.root file
664 TString fImportGeometryFilePath; // path fo geometry.root file
665
3c1a2e95 666
0de1814a 667 AliCaloTrackReader( const AliCaloTrackReader & r) ; // cpy ctor
668 AliCaloTrackReader & operator = (const AliCaloTrackReader & r) ; // cpy assignment
f3138ecf 669
9bf1c947 670 ClassDef(AliCaloTrackReader,53)
f3138ecf 671
1c5acb87 672} ;
673
674
675#endif //ALICALOTRACKREADER_H
676
677
678