]>
Commit | Line | Data |
---|---|---|
e0331fd9 | 1 | #ifndef AliAnalysisTaskPhiCorrelations_H |
2 | #define AliAnalysisTaskPhiCorrelations_H | |
3 | ||
4 | /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
5 | * See cxx source for full Copyright notice */ | |
6 | ||
7 | //////////////////////////////////////////////////////////////////////// | |
8 | // | |
9 | // Analysis class for Underlying Event studies w.r.t. leading track | |
10 | // | |
11 | // Look for correlations on the tranverse regions w.r.t | |
12 | // the leading track in the event | |
13 | // | |
14 | // This class needs input AODs. | |
15 | // The output is a list of analysis-specific containers. | |
16 | // | |
17 | // The AOD can be either connected to the InputEventHandler | |
18 | // for a chain of AOD files | |
19 | // or | |
20 | // to the OutputEventHandler | |
21 | // for a chain of ESD files, | |
22 | // in this case the class should be in the train after the jet-finder | |
23 | // | |
24 | // Authors: | |
e0331fd9 | 25 | // Jan Fiete Grosse-Oetringhaus |
e0331fd9 | 26 | // |
27 | //////////////////////////////////////////////////////////////////////// | |
28 | ||
29 | #include "AliAnalysisTask.h" | |
30 | #include "AliUEHist.h" | |
2a910c25 | 31 | #include "TString.h" |
a1c31636 | 32 | #include "AliVParticle.h" |
33 | #include "AliLog.h" | |
408d1ac9 | 34 | #include "THn.h" // in cxx file causes .../THn.h:257: error: conflicting declaration ‘typedef class THnT<float> THnF’ |
e0331fd9 | 35 | |
36 | class AliAODEvent; | |
37 | class AliAnalyseLeadingTrackUE; | |
38 | class AliInputEventHandler; | |
39 | class AliMCEvent; | |
40 | class AliMCEventHandler; | |
41 | class AliUEHistograms; | |
42 | class AliVParticle; | |
24d8278b | 43 | class TH1; |
e0331fd9 | 44 | class TObjArray; |
2a910c25 | 45 | class AliEventPoolManager; |
85bfac17 | 46 | class AliESDEvent; |
32e49607 | 47 | class AliHelperPID; |
6e84bd0a | 48 | class AliAnalysisUtils; |
97270ee2 | 49 | class TFormula; |
dc44bc1f | 50 | class TMap; |
3e09dd3e | 51 | class AliGenEventHeader; |
677dc831 | 52 | class AliVEvent; |
e0331fd9 | 53 | |
408d1ac9 | 54 | |
e0331fd9 | 55 | class AliAnalysisTaskPhiCorrelations : public AliAnalysisTask |
56 | { | |
57 | public: | |
58 | AliAnalysisTaskPhiCorrelations(const char* name="AliAnalysisTaskPhiCorrelations"); | |
59 | virtual ~AliAnalysisTaskPhiCorrelations(); | |
60 | ||
61 | ||
62 | // Implementation of interace methods | |
63 | virtual void ConnectInputData(Option_t *); | |
64 | virtual void CreateOutputObjects(); | |
65 | virtual void Exec(Option_t *option); | |
66 | ||
67 | // Setters/Getters | |
68 | // general configuration | |
69 | virtual void SetDebugLevel( Int_t level ) { fDebug = level; } | |
70 | virtual void SetMode(Int_t mode) { fMode = mode; } | |
71 | virtual void SetReduceMemoryFootprint(Bool_t flag) { fReduceMemoryFootprint = flag; } | |
eed401dc | 72 | virtual void SetEventMixing(Bool_t flag) { fFillMixed = flag; } |
ac647b0f | 73 | virtual void SetMixingTracks(Int_t tracks) { fMixingTracks = tracks; } |
1bba939a | 74 | virtual void SetTwoTrackEfficiencyStudy(Bool_t flag) { fTwoTrackEfficiencyStudy = flag; } |
7e9608f2 | 75 | virtual void SetTwoTrackEfficiencyCut(Float_t value = 0.02, Float_t min = 0.8) { fTwoTrackEfficiencyCut = value; fTwoTrackCutMinRadius = min; } |
3bbad7c1 | 76 | virtual void SetUseVtxAxis(Int_t flag) { fUseVtxAxis = flag; } |
9894bedd | 77 | virtual void SetCourseCentralityBinning(Bool_t flag) { fCourseCentralityBinning = flag; } |
04af8d15 | 78 | virtual void SetSkipTrigger(Bool_t flag) { fSkipTrigger = flag; } |
1ccd8a0a | 79 | virtual void SetInjectedSignals(Bool_t flag) { fInjectedSignals = flag; } |
f89c31a7 | 80 | void SetRandomizeReactionPlane(Bool_t flag) { fRandomizeReactionPlane = flag; } |
e0331fd9 | 81 | |
82 | // histogram settings | |
418b56c5 | 83 | void SetEfficiencyCorrectionTriggers(THnF* hist) { fEfficiencyCorrectionTriggers = hist; } |
84 | void SetEfficiencyCorrectionAssociated(THnF* hist) { fEfficiencyCorrectionAssociated = hist; } | |
24d8278b | 85 | void SetCentralityWeights(TH1* hist) { fCentralityWeights = hist; } |
e0331fd9 | 86 | |
87 | // for event QA | |
e08dff58 | 88 | void SetTracksInVertex( Int_t val ){ fnTracksVertex = val; } |
89 | void SetZVertex( Double_t val ) { fZVertex = val; } | |
90 | void SetAcceptOnlyMuEvents( Bool_t val ) { fAcceptOnlyMuEvents = val; } | |
e0331fd9 | 91 | |
92 | // track cuts | |
e08dff58 | 93 | void SetTrackEtaCut( Double_t val ) { fTrackEtaCut = val; } |
94 | void SetTrackEtaCutMin( Double_t val ) { fTrackEtaCutMin = val; } | |
418b56c5 | 95 | void SetOnlyOneEtaSide(Int_t flag) { fOnlyOneEtaSide = flag; } |
e0331fd9 | 96 | void SetPtMin(Double_t val) { fPtMin = val; } |
97 | void SetFilterBit( UInt_t val ) { fFilterBit = val; } | |
97270ee2 | 98 | void SetDCAXYCut(TFormula* value) { fDCAXYCut = value; } |
7d756130 | 99 | void SetSharedClusterCut(Float_t value) { fSharedClusterCut = value; } |
132b6963 | 100 | void SetCrossedRowsCut(Int_t value) { fCrossedRowsCut = value; } |
101 | void SetFoundFractionCut(Double_t value) { fFoundFractionCut = value; } | |
418b56c5 | 102 | void SetTrackStatus(UInt_t status) { fTrackStatus = status; } |
9e35c487 | 103 | void SetCheckMotherPDG(Bool_t checkpdg) { fCheckMotherPDG = checkpdg; } |
f334f670 | 104 | |
105 | // track cuts | |
106 | void SetTrackletDphiCut( Double_t val ) { fTrackletDphiCut = val; } | |
107 | ||
e0331fd9 | 108 | void SetEventSelectionBit( UInt_t val ) { fSelectBit = val; } |
109 | void SetUseChargeHadrons( Bool_t val ) { fUseChargeHadrons = val; } | |
5c9b9fa6 | 110 | void SetSelectParticleSpecies( Int_t trigger, Int_t associated ) { fParticleSpeciesTrigger = trigger; fParticleSpeciesAssociated = associated; } |
2a910c25 | 111 | void SetSelectCharge(Int_t selectCharge) { fSelectCharge = selectCharge; } |
7a77d480 | 112 | void SetSelectTriggerCharge(Int_t selectCharge) { fTriggerSelectCharge = selectCharge; } |
15b0fdd0 | 113 | void SetSelectAssociatedCharge(Int_t selectCharge) { fAssociatedSelectCharge = selectCharge; } |
d38fa455 | 114 | void SetTriggerRestrictEta(Float_t eta) { fTriggerRestrictEta = eta; } |
00b6f3c6 | 115 | void SetEtaOrdering(Bool_t flag) { fEtaOrdering = flag; } |
b0d56b29 | 116 | void SetPairCuts(Bool_t conversions, Bool_t resonances) { fCutConversions = conversions; fCutResonances = resonances; } |
97270ee2 | 117 | void SetRejectResonanceDaughters(Int_t value) { fRejectResonanceDaughters = value; } |
2a910c25 | 118 | void SetCentralityMethod(const char* method) { fCentralityMethod = method; } |
c05ff6be | 119 | void SetFillpT(Bool_t flag) { fFillpT = flag; } |
a26093ba | 120 | void SetStepsFillSkip(Bool_t step0, Bool_t step6) { fFillOnlyStep0 = step0; fSkipStep6 = step6; } |
5e053cad | 121 | void SetRejectCentralityOutliers(Bool_t flag = kTRUE) { fRejectCentralityOutliers = flag; } |
ff9f18e4 | 122 | void SetRejectZeroTrackEvents(Bool_t flag) { fRejectZeroTrackEvents = flag; } |
f613255f | 123 | void SetRemoveWeakDecays(Bool_t flag = kTRUE) { fRemoveWeakDecays = flag; } |
d6a8903f | 124 | void SetRemoveDuplicates(Bool_t flag = kTRUE) { fRemoveDuplicates = flag; } |
8a368fc2 | 125 | void SetSkipFastCluster(Bool_t flag = kTRUE) { fSkipFastCluster = flag; } |
126 | void SetWeightPerEvent(Bool_t flag = kTRUE) { fWeightPerEvent = flag; } | |
3f3f12d9 | 127 | void SetCustomBinning(const char* binningStr) { fCustomBinning = binningStr; } |
640b9425 | 128 | void SetPtOrder(Bool_t flag) { fPtOrder = flag; } |
2b27d6f4 | 129 | void SetTriggersFromDetector(Int_t flag) { fTriggersFromDetector = flag; } |
677dc831 | 130 | void SetAssociatedFromDetector(Int_t flag) { fAssociatedFromDetector = flag; } |
2313a5d0 | 131 | void SetMCUseUncheckedCentrality(Bool_t flag) { fMCUseUncheckedCentrality = flag; } |
e0331fd9 | 132 | |
32e49607 | 133 | AliHelperPID* GetHelperPID() { return fHelperPID; } |
134 | void SetHelperPID(AliHelperPID* pid){ fHelperPID = pid; } | |
6e84bd0a | 135 | |
136 | AliAnalysisUtils* GetAnalysisUtils() { return fAnalysisUtils; } | |
137 | void SetAnalysisUtils(AliAnalysisUtils* utils){ fAnalysisUtils = utils; } | |
dc44bc1f | 138 | |
139 | TMap* GetMap() { return fMap; } | |
140 | void SetMap(TMap* map){ fMap = map; } | |
32e49607 | 141 | |
e0331fd9 | 142 | private: |
143 | AliAnalysisTaskPhiCorrelations(const AliAnalysisTaskPhiCorrelations &det); | |
144 | AliAnalysisTaskPhiCorrelations& operator=(const AliAnalysisTaskPhiCorrelations &det); | |
145 | void AddSettingsTree(); // add list of settings to output list | |
146 | // Analysis methods | |
147 | void AnalyseCorrectionMode(); // main algorithm to get correction maps | |
148 | void AnalyseDataMode(); // main algorithm to get raw distributions | |
149 | void Initialize(); // initialize some common pointer | |
7a028750 | 150 | TObjArray* CloneAndReduceTrackList(TObjArray* tracks); |
d6a8903f | 151 | void RemoveDuplicates(TObjArray* tracks); |
3f3f12d9 | 152 | void CleanUp(TObjArray* tracks, TObject* mcObj, Int_t maxLabel); |
29cfdb5f | 153 | void SelectCharge(TObjArray* tracks); |
3e09dd3e | 154 | AliGenEventHeader* GetFirstHeader(); |
24d8278b | 155 | Bool_t AcceptEventCentralityWeight(Double_t centrality); |
f89c31a7 | 156 | void ShiftTracks(TObjArray* tracks, Double_t angle); |
677dc831 | 157 | TObjArray* GetParticlesFromDetector(AliVEvent* inputEvent, Int_t idet); |
576bfde8 | 158 | Bool_t IsMuEvent(); |
e0331fd9 | 159 | |
e0331fd9 | 160 | // General configuration |
161 | Int_t fDebug; // Debug flag | |
162 | Int_t fMode; // fMode = 0: data-like analysis | |
163 | // fMode = 1: corrections analysis | |
f89c31a7 | 164 | Bool_t fReduceMemoryFootprint; // reduce memory consumption by writing less debug histograms |
eed401dc | 165 | Bool_t fFillMixed; // enable event mixing (default: ON) |
ac647b0f | 166 | Int_t fMixingTracks; // size of track buffer for event mixing |
1bba939a | 167 | Bool_t fTwoTrackEfficiencyStudy; // two-track efficiency study on |
d4b3dbfc | 168 | Float_t fTwoTrackEfficiencyCut; // enable two-track efficiency cut |
7e9608f2 | 169 | Float_t fTwoTrackCutMinRadius; // minimum radius for two-track efficiency cut |
3bbad7c1 | 170 | Int_t fUseVtxAxis; // use z vtx as axis (needs 7-10 times more memory!) |
9894bedd | 171 | Bool_t fCourseCentralityBinning; // less centrality bins |
04af8d15 | 172 | Bool_t fSkipTrigger; // skip trigger selection |
1ccd8a0a | 173 | Bool_t fInjectedSignals; // check header to skip injected signals in MC |
f89c31a7 | 174 | Bool_t fRandomizeReactionPlane; // change the orientation of the RP by a random value by shifting all tracks |
e0331fd9 | 175 | |
e08dff58 | 176 | AliHelperPID* fHelperPID; // points to class for PID |
dc44bc1f | 177 | AliAnalysisUtils* fAnalysisUtils; // points to class with common analysis utilities |
e08dff58 | 178 | TMap* fMap; // points to TMap class containing scaling factors for VZERO A signal |
dc44bc1f | 179 | |
180 | // Pointers to external UE classes | |
e08dff58 | 181 | AliAnalyseLeadingTrackUE* fAnalyseUE; //! points to class containing common analysis algorithms |
182 | AliUEHistograms* fHistos; //! points to class to handle histograms/containers | |
183 | AliUEHistograms* fHistosMixed; //! points to class to handle mixed histograms/containers | |
e0331fd9 | 184 | |
e08dff58 | 185 | THnF* fEfficiencyCorrectionTriggers; // if non-0 this efficiency correction is applied on the fly to the filling for trigger particles. The factor is multiplicative, i.e. should contain 1/efficiency. Axes: eta, pT, centrality, z-vtx |
186 | THnF* fEfficiencyCorrectionAssociated; // if non-0 this efficiency correction is applied on the fly to the filling for associated particles. The factor is multiplicative, i.e. should contain 1/efficiency. Axes: eta, pT, centrality, z-vtx | |
187 | TH1* fCentralityWeights; // for centrality flattening | |
418b56c5 | 188 | |
e0331fd9 | 189 | // Handlers and events |
190 | AliAODEvent* fAOD; //! AOD Event | |
85bfac17 | 191 | AliESDEvent* fESD; //! ESD Event |
e0331fd9 | 192 | TClonesArray* fArrayMC; //! Array of MC particles |
193 | AliInputEventHandler* fInputHandler; //! Generic InputEventHandler | |
194 | AliMCEvent* fMcEvent; //! MC event | |
d8ee8cd0 | 195 | AliInputEventHandler* fMcHandler; //! MCEventHandler |
2a910c25 | 196 | AliEventPoolManager* fPoolMgr; //! event pool manager |
e0331fd9 | 197 | |
198 | // Histogram settings | |
e08dff58 | 199 | TList* fListOfHistos; // Output list of containers |
e0331fd9 | 200 | |
201 | // Event QA cuts | |
e08dff58 | 202 | Int_t fnTracksVertex; // QA tracks pointing to principal vertex |
203 | Double_t fZVertex; // Position of Vertex in Z direction | |
576bfde8 | 204 | Bool_t fAcceptOnlyMuEvents; // Only Events with at least one muon are accepted |
e08dff58 | 205 | TString fCentralityMethod; // Method to determine centrality |
e0331fd9 | 206 | |
207 | // Track cuts | |
97dc2b4e | 208 | Double_t fTrackEtaCut; // Maximum Eta cut on particles |
e08dff58 | 209 | Double_t fTrackEtaCutMin; // Minimum Eta cut on particles |
9da2f080 | 210 | Int_t fOnlyOneEtaSide; // decides that only trigger particle from one eta side are considered (0 = all; -1 = negative, 1 = positive) |
97270ee2 | 211 | Double_t fPtMin; // Min pT to start correlations |
212 | TFormula* fDCAXYCut; // additional pt dependent cut on DCA XY (only for AOD) | |
e08dff58 | 213 | Double_t fSharedClusterCut; // cut on shared clusters (only for AOD) |
214 | Int_t fCrossedRowsCut; // cut on crossed rows (only for AOD) | |
132b6963 | 215 | Double_t fFoundFractionCut; // cut on crossed rows/findable clusters (only for AOD) |
418b56c5 | 216 | UInt_t fFilterBit; // Select tracks from an specific track cut |
217 | UInt_t fTrackStatus; // if non-0, the bits set in this variable are required for each track | |
e0331fd9 | 218 | UInt_t fSelectBit; // Select events according to AliAnalysisTaskJetServices bit maps |
219 | Bool_t fUseChargeHadrons; // Only use charge hadrons | |
5c9b9fa6 | 220 | Int_t fParticleSpeciesTrigger; // Select which particle to use for the trigger [ -1 (all, default) 0 (pions) 1 (kaons) 2 (protons) 3 (others) particles ] |
221 | Int_t fParticleSpeciesAssociated; // Select which particle to use for the associated [ -1 (all, default) 0 (pions) 1 (kaons) 2 (protons) 3 (others) particles ] | |
e08dff58 | 222 | Bool_t fCheckMotherPDG; // Check the PDG code of mother for secondaries |
39c2cb38 | 223 | |
f334f670 | 224 | // Tracklets cuts |
225 | Double_t fTrackletDphiCut; //maximum Dphi cut on tracklets | |
226 | ||
2a910c25 | 227 | Int_t fSelectCharge; // (un)like sign selection when building correlations: 0: no selection; 1: unlike sign; 2: like sign |
15b0fdd0 | 228 | Int_t fTriggerSelectCharge; // select charge of trigger particle: 1: positive; -1 negative |
229 | Int_t fAssociatedSelectCharge; // select charge of associated particle: 1: positive; -1 negative | |
d38fa455 | 230 | Float_t fTriggerRestrictEta; // restrict eta range for trigger particle (default: -1 [off]) |
00b6f3c6 | 231 | Bool_t fEtaOrdering; // eta ordering, see AliUEHistograms.h for documentation |
b0d56b29 | 232 | Bool_t fCutConversions; // cut on conversions (inv mass) |
233 | Bool_t fCutResonances; // cut on resonances (inv mass) | |
97270ee2 | 234 | Int_t fRejectResonanceDaughters; // reject all daughters of all resonance candidates (1: test method (cut at m_inv=0.9); 2: k0; 3: lambda) |
a26093ba | 235 | Bool_t fFillOnlyStep0; // fill only step 0 |
236 | Bool_t fSkipStep6; // skip step 6 when filling | |
e08dff58 | 237 | Bool_t fRejectCentralityOutliers; // enable rejection of outliers in centrality vs no track correlation |
ff9f18e4 | 238 | Bool_t fRejectZeroTrackEvents; // reject events which have no tracks (using the eta, pT cuts defined) |
f613255f | 239 | Bool_t fRemoveWeakDecays; // remove secondaries from weak decays from tracks and particles |
d6a8903f | 240 | Bool_t fRemoveDuplicates; // remove particles with the same label (double reconstruction) |
51d0a028 | 241 | Bool_t fSkipFastCluster; // skip kFastOnly flagged events (only for data) |
8a368fc2 | 242 | Bool_t fWeightPerEvent; // weight with the number of trigger particles per event |
3f3f12d9 | 243 | TString fCustomBinning; // supersedes default binning if set, see AliUEHist::GetBinning or AliUEHistograms::AliUEHistograms for syntax and examples |
640b9425 | 244 | Bool_t fPtOrder; // apply pT,a < pt,t condition; default: kTRUE |
1db82805 | 245 | Int_t fTriggersFromDetector; // 0 = tracks (default); 1 = VZERO_A; 2 = VZERO_C; 3 = SPD tracklets; 4 = forward muons |
246 | Int_t fAssociatedFromDetector; // 0 = tracks (default); 1 = VZERO_A; 2 = VZERO_C; 3 = SPD tracklets; 4 = forward muons | |
2313a5d0 | 247 | Bool_t fMCUseUncheckedCentrality; // use unchecked centrality (only applies to MC); default: kFALSE |
d38fa455 | 248 | |
c05ff6be | 249 | Bool_t fFillpT; // fill sum pT instead of number density |
2a910c25 | 250 | |
576bfde8 | 251 | ClassDef(AliAnalysisTaskPhiCorrelations, 46); // Analysis task for delta phi correlations |
e0331fd9 | 252 | }; |
253 | ||
a1c31636 | 254 | class AliDPhiBasicParticle : public AliVParticle |
255 | { | |
256 | public: | |
257 | AliDPhiBasicParticle(Float_t eta, Float_t phi, Float_t pt, Short_t charge) | |
258 | : fEta(eta), fPhi(phi), fpT(pt), fCharge(charge) | |
259 | { | |
260 | } | |
261 | ~AliDPhiBasicParticle() {} | |
262 | ||
263 | // kinematics | |
264 | virtual Double_t Px() const { AliFatal("Not implemented"); return 0; } | |
265 | virtual Double_t Py() const { AliFatal("Not implemented"); return 0; } | |
266 | virtual Double_t Pz() const { AliFatal("Not implemented"); return 0; } | |
267 | virtual Double_t Pt() const { return fpT; } | |
268 | virtual Double_t P() const { AliFatal("Not implemented"); return 0; } | |
269 | virtual Bool_t PxPyPz(Double_t[3]) const { AliFatal("Not implemented"); return 0; } | |
270 | ||
271 | virtual Double_t Xv() const { AliFatal("Not implemented"); return 0; } | |
272 | virtual Double_t Yv() const { AliFatal("Not implemented"); return 0; } | |
273 | virtual Double_t Zv() const { AliFatal("Not implemented"); return 0; } | |
274 | virtual Bool_t XvYvZv(Double_t[3]) const { AliFatal("Not implemented"); return 0; } | |
275 | ||
276 | virtual Double_t OneOverPt() const { AliFatal("Not implemented"); return 0; } | |
277 | virtual Double_t Phi() const { return fPhi; } | |
278 | virtual Double_t Theta() const { AliFatal("Not implemented"); return 0; } | |
279 | ||
280 | ||
281 | virtual Double_t E() const { AliFatal("Not implemented"); return 0; } | |
282 | virtual Double_t M() const { AliFatal("Not implemented"); return 0; } | |
283 | ||
284 | virtual Double_t Eta() const { return fEta; } | |
285 | virtual Double_t Y() const { AliFatal("Not implemented"); return 0; } | |
286 | ||
287 | virtual Short_t Charge() const { return fCharge; } | |
288 | virtual Int_t GetLabel() const { AliFatal("Not implemented"); return 0; } | |
289 | // PID | |
290 | virtual Int_t PdgCode() const { AliFatal("Not implemented"); return 0; } | |
291 | virtual const Double_t *PID() const { AliFatal("Not implemented"); return 0; } | |
292 | ||
1d53fb34 | 293 | virtual Bool_t IsEqual(const TObject* obj) const { return (obj->GetUniqueID() == GetUniqueID()); } |
294 | ||
f89c31a7 JFGO |
295 | virtual void SetPhi(Double_t phi) { fPhi = phi; } |
296 | ||
a1c31636 | 297 | private: |
298 | Float_t fEta; // eta | |
299 | Float_t fPhi; // phi | |
300 | Float_t fpT; // pT | |
301 | Short_t fCharge; // charge | |
302 | ||
303 | ClassDef( AliDPhiBasicParticle, 1); // class which contains only quantities requires for this analysis to reduce memory consumption for event mixing | |
304 | }; | |
305 | ||
e0331fd9 | 306 | #endif |
307 | ||
308 |