Moved initialisation of pointer (to PIDtasks) array to UserCreateOutputObjects, since...
[u/mrichter/AliRoot.git] / PWGJE / UserTasks / AliAnalysisTaskIDFragmentationFunction.h
CommitLineData
e131b05f 1// *************************************************************************
2// * Task for Fragmentation Function Analysis in PWG4 Jet Task Force Train *
3// *************************************************************************
4
5#ifndef ALIANALYSISTASKIDFRAGMENTATIONFUNCTION_H
6#define ALIANALYSISTASKIDFRAGMENTATIONFUNCTION_H
7
8/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
9 * See cxx source for full Copyright notice */
10
11/* $Id$ */
12
13class AliESDEvent;
14class AliAODEvent;
15class AliAODJet;
16class AliAODExtension;
17class TList;
18class TH1F;
19class TH2F;
20class TH3F;
21class TProfile;
22class THnSparse;
23class TRandom3;
24class TArrayS;
25class AliAODTrack;
26class AliAODMCParticle;
27
28#include "AliAnalysisTaskSE.h"
29#include "AliPID.h"
30
31class AliAnalysisTaskIDFragmentationFunction : public AliAnalysisTaskSE {
32
33 public:
34
35 //----------------------------------------
36 class AliFragFuncHistos : public TObject
37 {
38
39 public:
40
41 AliFragFuncHistos(const char* name = "FFhistos",
42 Int_t nJetPt = 0, Float_t jetPtMin = 0, Float_t jetPtMax = 0,
43 Int_t nPt = 0, Float_t ptMin = 0, Float_t ptMax = 0,
44 Int_t nXi = 0, Float_t xiMin = 0, Float_t xiMax = 0,
45 Int_t nZ = 0, Float_t zMin = 0, Float_t zMax = 0);
46
47 AliFragFuncHistos(const AliFragFuncHistos& copy);
48 AliFragFuncHistos& operator=(const AliFragFuncHistos &o);
49 virtual ~AliFragFuncHistos();
50
51 virtual void DefineHistos();
52 virtual void FillFF(Float_t trackPt, Float_t jetPt,Bool_t incrementJetPt, Float_t norm = 0, Bool_t scaleStrangeness = kFALSE, Float_t scaleFacStrangeness = 1.);
53
54 virtual void AddToOutput(TList* list) const;
55
56 private:
57
58 Int_t fNBinsJetPt; // FF histos bins
59 Float_t fJetPtMin; // FF histos limits
60 Float_t fJetPtMax; // FF histos limits
61 Int_t fNBinsPt; // FF histos bins
62 Float_t fPtMin; // FF histos limits
63 Float_t fPtMax; // FF histos limits
64 Int_t fNBinsXi; // FF histos bins
65 Float_t fXiMin; // FF histos limits
66 Float_t fXiMax; // FF histos limits
67 Int_t fNBinsZ; // FF histos bins
68 Float_t fZMin; // FF histos limits
69 Float_t fZMax; // FF histos limits
70
71 TH2F* fh2TrackPt; //! FF: track transverse momentum
72 TH2F* fh2Xi; //! FF: xi
73 TH2F* fh2Z; //! FF: z
74 TH1F* fh1JetPt; //! jet pt
75
76 TString fNameFF; // histo names prefix
77
78 ClassDef(AliFragFuncHistos, 1);
79 };
80
81 //----------------------------------------
82 class AliFragFuncQAJetHistos : public TObject
83 {
84
85 public:
86
87 AliFragFuncQAJetHistos(const char* name = "QAJethistos",
88 Int_t nPt = 0, Float_t ptMin = 0, Float_t ptMax = 0,
89 Int_t nEta = 0, Float_t etaMin = 0, Float_t etaMax = 0,
90 Int_t nPhi = 0, Float_t phiMin = 0, Float_t phiMax = 0);
91
92 AliFragFuncQAJetHistos(const AliFragFuncQAJetHistos& copy);
93 AliFragFuncQAJetHistos& operator=(const AliFragFuncQAJetHistos &o);
94 virtual ~AliFragFuncQAJetHistos();
95 virtual void DefineHistos();
96 virtual void FillJetQA(Float_t eta, Float_t phi, Float_t pt);
97 virtual void AddToOutput(TList* list) const;
98
99 private:
100
101 Int_t fNBinsPt; // jet QA histos bins
102 Float_t fPtMin; // jet QA histos limits
103 Float_t fPtMax; // jet QA histos limits
104 Int_t fNBinsEta; // jet QA histos bins
105 Float_t fEtaMin; // jet QA histos limits
106 Float_t fEtaMax; // jet QA histos limits
107 Int_t fNBinsPhi; // jet QA histos bins
108 Float_t fPhiMin; // jet QA histos limits
109 Float_t fPhiMax; // jet QA histos limits
110
111 TH2F* fh2EtaPhi; //! jet phi vs eta
112 TH1F* fh1Pt; //! jet transverse momentum
113 TString fNameQAJ; // histo names prefix
114
115 ClassDef(AliFragFuncQAJetHistos, 1);
116 };
117
118 //----------------------------------------
119 class AliFragFuncQATrackHistos : public TObject
120 {
121
122 public:
123
124 AliFragFuncQATrackHistos(const char* name = "QATrackhistos",
125 Int_t nPt = 0, Float_t ptMin = 0, Float_t ptMax = 0,
126 Int_t nEta = 0, Float_t etaMin = 0, Float_t etaMax = 0,
127 Int_t nPhi = 0, Float_t phiMin = 0, Float_t phiMax = 0,
128 Float_t ptThresh = 0);
129
130 AliFragFuncQATrackHistos(const AliFragFuncQATrackHistos& copy);
131 AliFragFuncQATrackHistos& operator=(const AliFragFuncQATrackHistos &o);
132 virtual ~AliFragFuncQATrackHistos();
133 virtual void DefineHistos();
134 virtual void FillTrackQA(Float_t eta, Float_t phi, Float_t pt, Bool_t weightPt = kFALSE, Float_t norm = 0., Bool_t scaleStrangeness = kFALSE, Float_t scaleFacStrangeness = 1.);
135 virtual void AddToOutput(TList* list) const;
136
137 private:
138
139 Int_t fNBinsPt; // track QA histos bins in pt
140 Float_t fPtMin; // track QA histos limits in pt
141 Float_t fPtMax; // track QA histos limits in pt
142 Int_t fNBinsEta; // track QA histos bins in eta
143 Float_t fEtaMin; // track QA histos limits in eta
144 Float_t fEtaMax; // track QA histos limits in eta
145 Int_t fNBinsPhi; // track QA histos bins in phi
146 Float_t fPhiMin; // track QA histos limits in phi
147 Float_t fPhiMax; // track QA histos limits in phi
148
149 Float_t fHighPtThreshold; // high pt track phi vs eta distribution
150
151 TH2F* fh2EtaPhi; //! track phi vs eta
152 TH1F* fh1Pt; //! track transverse momentum
153 TH2F* fh2HighPtEtaPhi; //! phi vs eta for high pt (>fgHighPtThreshold) tracks
154 TH2F* fh2PhiPt; //! track phi vs pt
155
156 TString fNameQAT; // histo names prefix
157
158 ClassDef(AliFragFuncQATrackHistos, 2);
159 };
160
161
162 AliAnalysisTaskIDFragmentationFunction();
163 AliAnalysisTaskIDFragmentationFunction(const char *name);
164 AliAnalysisTaskIDFragmentationFunction(const AliAnalysisTaskIDFragmentationFunction &copy);
165 AliAnalysisTaskIDFragmentationFunction& operator=(const AliAnalysisTaskIDFragmentationFunction &o);
166 virtual ~AliAnalysisTaskIDFragmentationFunction();
167
168 virtual void UserCreateOutputObjects();
169 virtual void Init();
170 virtual void LocalInit() {Init();}
171
172 virtual void UserExec(Option_t *option);
173 virtual void Terminate(Option_t* );
174 virtual Bool_t Notify();
175
176 virtual void SetNonStdFile(char* c){fNonStdFile = c;}
177
178 virtual void SetTrackTypeGen(Int_t i){fTrackTypeGen = i;}
179 virtual void SetJetTypeGen(Int_t i){fJetTypeGen = i;}
180 virtual void SetJetTypeRecEff(Int_t i){fJetTypeRecEff = i;}
181
182 virtual void SetBranchRecBackClusters(const char* c){fBranchRecBckgClusters = c;}
183 virtual void SetBranchGenJets(const char* c){fBranchGenJets = c;}
184 virtual void SetBranchRecJets(const char* c){fBranchRecJets = c;}
185 virtual void SetBranchEmbeddedJets(const char* c){fBranchEmbeddedJets = c;}
186
187 virtual void SetTrackCuts(Float_t trackPt = 0.15, Float_t trackEtaMin = -0.9, Float_t trackEtaMax = 0.9,
188 Float_t trackPhiMin = 0., Float_t trackPhiMax = 2*TMath::Pi())
189 {fTrackPtCut = trackPt; fTrackEtaMin = trackEtaMin; fTrackEtaMax = trackEtaMax;
190 fTrackPhiMin = trackPhiMin; fTrackPhiMax = trackPhiMax;}
191
192 virtual void UseExtraTracks() { fUseExtraTracks = 1;}
193 virtual void UseExtraonlyTracks() { fUseExtraTracks = -1;}
194
195 virtual void UseExtraTracksBgr() { fUseExtraTracksBgr = 1;}
196 virtual void UseExtraonlyTracksBgr() { fUseExtraTracksBgr = -1;}
197
198 virtual void SetCutFractionPtEmbedded(Float_t cut = 0) { fCutFractionPtEmbedded = cut; }
199 virtual void SetUseEmbeddedJetAxis(Bool_t b = kTRUE) { fUseEmbeddedJetAxis = b; }
200 virtual void SetUseEmbeddedJetPt(Bool_t b = kTRUE) { fUseEmbeddedJetPt = b; }
201
202 virtual void UseAODInputJets(Bool_t b) {fUseAODInputJets = b;}
203 virtual void SetFilterMask(UInt_t i) {fFilterMask = i;}
204 virtual void UsePhysicsSelection(Bool_t b) {fUsePhysicsSelection = b;}
205 virtual void SetEventSelectionMask(UInt_t i){fEvtSelectionMask = i;}
206 virtual void SetEventClass(Int_t i){fEventClass = i;}
207 virtual void SetMaxVertexZ(Float_t z){fMaxVertexZ = z;}
208 virtual void SetJetCuts(Float_t jetPt = 5., Float_t jetEtaMin = -0.5, Float_t jetEtaMax = 0.5,
209 Float_t jetPhiMin = 0., Float_t jetPhiMax = 2*TMath::Pi())
210 {fJetPtCut = jetPt; fJetEtaMin = jetEtaMin; fJetEtaMax = jetEtaMax;
211 fJetPhiMin = jetPhiMin; fJetPhiMax = jetPhiMax;}
212
213 virtual void SetFFRadius(Float_t r = 0.4) { fFFRadius = r; }
214 virtual void SetFFMinLTrackPt(Float_t pt = -1) { fFFMinLTrackPt = pt; }
215 virtual void SetFFMaxTrackPt(Float_t pt = -1) { fFFMaxTrackPt = pt; }
216 virtual void SetFFMinNTracks(Int_t nTracks = 0) { fFFMinnTracks = nTracks; }
217 virtual void SetFFBckgRadius(Float_t r = 0.7) { fFFBckgRadius = r; }
218 virtual void SetBckgMode(Bool_t bg = 1) { fBckgMode = bg; }
219 virtual void SetBckgType(Int_t bg0 = 0, Int_t bg1 = 0,Int_t bg2 = 0, Int_t bg3 = 0, Int_t bg4 = 0)
220 { fBckgType[0] = bg0; fBckgType[1] = bg1; fBckgType[2] = bg2; fBckgType[3] = bg3; fBckgType[4] = bg4; }
221 virtual void SetQAMode(Int_t qa = 3) {fQAMode = qa;}
222 virtual void SetFFMode(Int_t ff = 1) {fFFMode = ff;}
223 virtual void SetIDFFMode(Int_t idff = 0) {fIDFFMode = idff;}
224 virtual void SetEffMode(Int_t eff = 1) {fEffMode = eff;}
225 virtual void SetJSMode(Int_t js = 1) {fJSMode = js;}
226
227 static void SetProperties(TH1* h,const char* x, const char* y);
228 static void SetProperties(TH1* h,const char* x, const char* y,const char* z);
9d7ad2e4 229 static void SetProperties(THnSparse* h,Int_t dim, const char** labels);
e131b05f 230
231 void SetHighPtThreshold(Float_t pt = 5.) { fQATrackHighPtThreshold = pt; }
232
233 void SetFFHistoBins(Int_t nJetPt = 245, Float_t jetPtMin = 5, Float_t jetPtMax = 250,
234 Int_t nPt = 200, Float_t ptMin = 0., Float_t ptMax = 200.,
235 Int_t nXi = 70, Float_t xiMin = 0., Float_t xiMax = 7.,
236 Int_t nZ = 22, Float_t zMin = 0., Float_t zMax = 1.1)
237 { fFFNBinsJetPt = nJetPt; fFFJetPtMin = jetPtMin; fFFJetPtMax = jetPtMax;
238 fFFNBinsPt = nPt; fFFPtMin = ptMin; fFFPtMax = ptMax;
239 fFFNBinsXi = nXi; fFFXiMin = xiMin; fFFXiMax = xiMax;
240 fFFNBinsZ = nZ; fFFZMin = zMin; fFFZMax = zMax; }
241
242 void SetQAJetHistoBins(Int_t nPt = 300, Float_t ptMin = 0., Float_t ptMax = 300.,
243 Int_t nEta = 20, Float_t etaMin = -1.0, Float_t etaMax = 1.0,
244 Int_t nPhi = 60, Float_t phiMin = 0., Float_t phiMax = 2*TMath::Pi())
245 { fQAJetNBinsPt = nPt; fQAJetPtMin = ptMin; fQAJetPtMax = ptMax;
246 fQAJetNBinsEta = nEta; fQAJetEtaMin = etaMin; fQAJetEtaMax = etaMax;
247 fQAJetNBinsPhi = nPhi; fQAJetPhiMin = phiMin; fQAJetPhiMax = phiMax; }
248
249 void SetQATrackHistoBins(Int_t nPt = 200, Float_t ptMin = 0., Float_t ptMax = 200.,
250 Int_t nEta = 20, Float_t etaMin = -1.0, Float_t etaMax = 1.0,
251 Int_t nPhi = 60, Float_t phiMin = 0., Float_t phiMax = 2*TMath::Pi())
252 { fQATrackNBinsPt = nPt; fQATrackPtMin = ptMin; fQATrackPtMax = ptMax;
253 fQATrackNBinsEta = nEta; fQATrackEtaMin = etaMin; fQATrackEtaMax = etaMax;
254 fQATrackNBinsPhi = nPhi; fQATrackPhiMin = phiMin; fQATrackPhiMax = phiMax; }
255
256
257
258 Float_t GetFFRadius() const { return fFFRadius; }
259 Float_t GetFFMinLTrackPt() const { return fFFMinLTrackPt; }
260 Float_t GetFFMaxTrackPt() const { return fFFMaxTrackPt; }
261 Float_t GetFFMinNTracks() const { return fFFMinnTracks; }
262 Float_t GetFFBckgRadius() const { return fFFBckgRadius; }
9d7ad2e4 263 void GetJetTracksTrackrefs(TList* l, const AliAODJet* j, Double_t minPtL, Double_t maxPt, Bool_t& isBadPt);
264 void GetJetTracksPointing(TList* in, TList* out, const AliAODJet* j, Double_t r, Double_t& sumPt, Double_t minPtL, Double_t maxPt,
265 Bool_t& isBadPt);
e131b05f 266 void GetTracksOutOfNJets(Int_t nCases, TList* in, TList* out, const TList* jets, Double_t& pt);
267 void GetTracksOutOfNJetsStat(Int_t nCases, TList* in, TList* out, const TList* jets, Double_t& pt, Double_t &normFactor);
268 void GetTracksTiltedwrpJetAxis(Float_t alpha, TList* inputlist, TList* outputlist, const AliAODJet* jet, Double_t radius, Double_t& sumPt);
269 void GetTracksTiltedwrpJetAxisWindow(Float_t alpha, TList* inputlist, TList* outputlist, const AliAODJet* jet, Double_t radius, Double_t& sumPt, Double_t &normFactor);
270
271 void AssociateGenRec(TList* tracksAODMCCharged,TList* tracksRec, TArrayI& indexAODTr,TArrayI& indexMCTr,TArrayS& isRefGen,TH2F* fh2PtRecVsGen);
272
273 void FillSingleTrackHistosRecGen(AliFragFuncQATrackHistos* trackQAGen, AliFragFuncQATrackHistos* trackQARec, TList* tracksGen,
9d7ad2e4 274 const TArrayI& indexAODTr, const TArrayS& isRefGen, Bool_t scaleStrangeness = kFALSE);
e131b05f 275
276
277 void FillJetTrackHistosRec(AliFragFuncHistos* histRec, AliAODJet* jet,
278 TList* jetTrackList, const TList* tracksGen, const TList* tracksRec, const TArrayI& indexAODTr,
9d7ad2e4 279 const TArrayS& isRefGen, TList* jetTrackListTR = 0, Bool_t scaleStrangeness = kFALSE,
e131b05f 280 Bool_t fillJS = kFALSE, TProfile* hProNtracksLeadingJet = 0, TProfile** hProDelRPtSum = 0, TProfile* hProDelR80pcPt = 0);
281
282
9d7ad2e4 283 Float_t CalcJetArea(Float_t etaJet, Float_t rc) const;
e131b05f 284 void GetClusterTracksOutOf1Jet(AliAODJet* jet, TList* outputlist, Double_t &normFactor);
285 void GetClusterTracksMedian(TList* outputlist, Double_t &normFactor);
286
287 void FillBckgHistos(Int_t type, TList* inputtracklist, TList* inputjetlist, AliAODJet* jet,
288 AliFragFuncHistos* ffbckghistocuts,AliFragFuncQATrackHistos* qabckghistos,TH1F* fh1Mult = 0);
289
9d7ad2e4 290 Double_t GetMCStrangenessFactor(Double_t pt) const;
e131b05f 291 Double_t GetMCStrangenessFactorCMS(AliAODMCParticle* daughter) const;
292
293 Bool_t IsSecondaryWithStrangeMotherMC(AliAODMCParticle* part);
294
295 void FillJetShape(AliAODJet* jet, TList* list, TProfile* hProNtracksLeadingJet, TProfile** hProDelRPtSum, TProfile* hProDelR80pcPt=0, Double_t dPhiUE=0, Double_t normUE = 0, Bool_t scaleStrangeness = kFALSE);
296
297 const TString* GetNamesOfInclusivePIDtasks() const { return fNameInclusivePIDtask; };
9d7ad2e4 298 void SetNamesOfInclusivePIDtasks(Int_t numNames, const TString* names);
e131b05f 299
300 const TString* GetNamesOfJetPIDtasks() const { return fNameJetPIDtask; };
9d7ad2e4 301 void SetNamesOfJetPIDtasks(Int_t numNames, const TString* names);
e131b05f 302
303
304 Bool_t GetIsPP() const { return fIsPP; };
305 void SetIsPP(Bool_t flag) { fIsPP = flag; };
306
c4856fb1 307 Bool_t GetOnlyLeadingJets() const { return fOnlyLeadingJets; }
308 void SetOnlyLeadingJets(Bool_t onlyLeadingJets) { fOnlyLeadingJets = onlyLeadingJets; }
309
e131b05f 310 // Consts
311 enum {kTrackUndef=0, kTrackAOD, kTrackAODQualityCuts, kTrackAODCuts,
312 kTrackAODExtra, kTrackAODExtraonly, kTrackAODExtraCuts, kTrackAODExtraonlyCuts,
313 kTrackKineAll, kTrackKineCharged, kTrackKineChargedAcceptance,
314 kTrackAODMCAll, kTrackAODMCCharged, kTrackAODMCChargedAcceptance, kTrackAODMCChargedSecS, kTrackAODMCChargedSecNS, kTrackAOCMCChargedPrimAcceptance};
315 enum {kJetsUndef=0, kJetsRec, kJetsRecAcceptance, kJetsGen, kJetsGenAcceptance, kJetsKine, kJetsKineAcceptance,kJetsEmbedded};
316 enum {kBckgNone=0, kBckgPerp, kBckgOutLJ, kBckgOut2J, kBckgClusters, kBckgClustersOutLeading, kBckgOut3J, kBckgOutAJ, kBckgOutLJStat,
317 kBckgOut2JStat, kBckgOut3JStat, kBckgOutAJStat, kBckgASide, kBckgASideWindow, kBckgPerpWindow, kBckgPerp2, kBckgPerp2Area};
318
319
320 protected:
321
322 Int_t GetListOfTracks(TList* list, Int_t type);
323 Int_t GetListOfJets(TList* list, Int_t type);
324 Int_t GetListOfBckgJets(TList *list, Int_t type);
325
9d7ad2e4 326 AliESDEvent* fESD; //! ESD event
327 AliAODEvent* fAOD; //! AOD event
328 AliAODEvent* fAODJets; //! AOD event with jet branch (case we have AOD both in input and output)
e131b05f 329 AliAODExtension *fAODExtension; //! where we take the jets from can be input or output AOD
330 TString fNonStdFile; // name of delta aod file to catch the extension
331
332
333 TString fBranchRecJets; // branch name for reconstructed jets
334 TString fBranchRecBckgClusters; // branch name for reconstructed background clusters
335 TString fBranchGenJets; // branch name for generated jets
336 TString fBranchEmbeddedJets; // branch name for embedded jets
337
338 Int_t fTrackTypeGen; // type of generated tracks
339 Int_t fJetTypeGen; // type of generated jets
340
341 Int_t fJetTypeRecEff; // type of jets used for filling reconstruction efficiency histos
342
343 Bool_t fUseAODInputJets; // take jets from in/output - only relevant if AOD event both in input AND output and we want to use output
344 UInt_t fFilterMask; // filter bit for selected tracks
345 Bool_t fUsePhysicsSelection; // switch for event selection
346 UInt_t fEvtSelectionMask; // trigger class selection
347 Int_t fEventClass; // centrality class selection
348 Float_t fMaxVertexZ; // maximum abs(z) position of primiary vertex [cm]
349
350 // track cuts
351 Float_t fTrackPtCut; // track transverse momentum cut
352 Float_t fTrackEtaMin; // track eta cut
353 Float_t fTrackEtaMax; // track eta cut
354 Float_t fTrackPhiMin; // track phi cut
355 Float_t fTrackPhiMax; // track phi cut
356
357 Int_t fUseExtraTracks; // +/- 1: embedded extra/extra only tracks, default: 0 (ignore extra tracks)
358 Int_t fUseExtraTracksBgr; // +/- 1: background: use embedded extra/extra only tracks, default: 0 (ignore extra tracks)
359 Float_t fCutFractionPtEmbedded; // cut on ratio of embedded pt found in jet
360 Bool_t fUseEmbeddedJetAxis; // use axis of embedded jet for FF
361 Bool_t fUseEmbeddedJetPt; // use axis of embedded jet for FF
362
363 // jet cuts
364 Float_t fJetPtCut; // jet transverse momentum cut
365 Float_t fJetEtaMin; // jet eta cut
366 Float_t fJetEtaMax; // jet eta cut
367 Float_t fJetPhiMin; // jet phi cut
368 Float_t fJetPhiMax; // jet phi cut
369
370 Float_t fFFRadius; // if radius > 0 construct FF from tracks within cone around jet axis, otherwise use trackRefs
371 Float_t fFFMinLTrackPt; // reject jets with leading track with pt smaller than this value
372 Float_t fFFMaxTrackPt; // reject jets containing any track with pt larger than this value
373 Int_t fFFMinnTracks; // reject jets with less tracks than this value
374 Float_t fFFBckgRadius; // compute background outside cone of this radius around jet axes
375 Bool_t fBckgMode; // Set background subtraction mode
376 Int_t fBckgType[5]; // Set background subtraction mode
377 Int_t fQAMode; // QA mode: 0x00=0 none, 0x01=1 track qa, 0x10=2 track qa, 0x11=3 both
378 Int_t fFFMode; // fragmentation function mode
379 Int_t fIDFFMode; // identified fragmentation function mode (implicitely enables also normal fragmentation function mode (but not effeciency, background etc.)!)
380 Int_t fEffMode; // efficiency mode
381 Int_t fJSMode; // jet shape mode
382
383 Float_t fAvgTrials; // average number of trials per event
384
385 TList* fTracksRecCuts; //! reconstructed tracks after cuts
386 TList* fTracksRecCutsEfficiency; //! reconstructed tracks after cuts for efficiency
387 TList* fTracksGen; //! generated tracks
388 TList* fTracksAODMCCharged; //! AOD MC tracks
389 TList* fTracksAODMCChargedSecNS; //! AOD MC tracks - secondaries (non-strangeness)
390 TList* fTracksAODMCChargedSecS; //! AOD MC tracks - secondaries (from strangeness)
391 TList* fTracksRecQualityCuts; //! reconstructed tracks after quality cuts, no acceptance/pt cut
392
393
394 TList* fJetsRec; //! jets from reconstructed tracks
395 TList* fJetsRecCuts; //! jets from reonstructed tracks after jet cuts
396 TList* fJetsGen; //! jets from generated tracks
397 TList* fJetsRecEff; //! jets used for reconstruction efficiency histos
398 TList* fJetsEmbedded; //! jets used for embedding
399
400 TList* fBckgJetsRec; //! jets from reconstructed tracks
401 TList* fBckgJetsRecCuts; //! jets from reonstructed tracks after jet cuts
402 TList* fBckgJetsGen; //! jets from generated tracks
403
404
405 AliFragFuncQATrackHistos* fQATrackHistosRecCuts; //! track QA: reconstructed tracks after cuts
406 AliFragFuncQATrackHistos* fQATrackHistosGen; //! track QA: generated tracks
407
408 AliFragFuncQAJetHistos* fQAJetHistosRec; //! jet QA: jets from reconstructed tracks
409 AliFragFuncQAJetHistos* fQAJetHistosRecCuts; //! jet QA: jets from reconstructed tracks after jet cuts
410 AliFragFuncQAJetHistos* fQAJetHistosRecCutsLeading; //! jet QA: leading jet from reconstructed tracks after jet cuts
411 AliFragFuncQAJetHistos* fQAJetHistosGen; //! jet QA: jets from generated tracks
412 AliFragFuncQAJetHistos* fQAJetHistosGenLeading; //! jet QA: leading jet from generated tracks
413 AliFragFuncQAJetHistos* fQAJetHistosRecEffLeading; //! jet QA: leading jet used for reconstruction efficiency histos
414
415
416 AliFragFuncHistos* fFFHistosRecCuts; //! FF reconstructed tracks after cuts (leading jet)
417 AliFragFuncHistos* fFFHistosRecCutsInc; //! inclusive FF (all jets)
418 AliFragFuncHistos* fFFHistosRecLeadingTrack; //! FF reconstructed tracks after cuts: leading track pt / jet pt (all jets)
419
420 AliFragFuncHistos* fFFHistosGen; //! FF generated tracks after cuts
421 AliFragFuncHistos* fFFHistosGenInc; //! inclusive FF (all jets)
422 AliFragFuncHistos* fFFHistosGenLeadingTrack; //! FF reconstructed tracks after cuts: leading track pt / jet pt (all jets)
423
424 Float_t fQATrackHighPtThreshold; // track QA high transverse momentum threshold
425
426 // histogram bins
427
428 Int_t fFFNBinsJetPt; // FF histos bins
429 Float_t fFFJetPtMin; // FF histos limits
430 Float_t fFFJetPtMax; // FF histos limits
431
432 Int_t fFFNBinsPt; // FF histos bins
433 Float_t fFFPtMin; // FF histos limits
434 Float_t fFFPtMax; // FF histos limits
435
436 Int_t fFFNBinsXi; // FF histos bins
437 Float_t fFFXiMin; // FF histos limits
438 Float_t fFFXiMax; // FF histos limits
439
440 Int_t fFFNBinsZ; // FF histos bins
441 Float_t fFFZMin; // FF histos limits
442 Float_t fFFZMax; // FF histos limits
443
444 Int_t fQAJetNBinsPt; // jet QA histos bins
445 Float_t fQAJetPtMin; // jet QA histos limits
446 Float_t fQAJetPtMax; // jet QA histos limits
447
448 Int_t fQAJetNBinsEta; // jet QA histos bins
449 Float_t fQAJetEtaMin; // jet QA histos limits
450 Float_t fQAJetEtaMax; // jet QA histos limits
451
452 Int_t fQAJetNBinsPhi; // jet QA histos bins
453 Float_t fQAJetPhiMin; // jet QA histos limits
454 Float_t fQAJetPhiMax; // jet QA histos limits
455
456 Int_t fQATrackNBinsPt; // track QA histos bins
457 Float_t fQATrackPtMin; // track QA histos limits
458 Float_t fQATrackPtMax; // track QA histos limits
459
460 Int_t fQATrackNBinsEta; // track QA histos bins
461 Float_t fQATrackEtaMin; // track QA histos limits
462 Float_t fQATrackEtaMax; // track QA histos limits
463
464 Int_t fQATrackNBinsPhi; // track QA histos bins
465 Float_t fQATrackPhiMin; // track QA histos limits
466 Float_t fQATrackPhiMax; // track QA histos limits
467
468 // Histograms
469 TList *fCommonHistList; // List of common histos
470
471 TH1F *fh1EvtSelection; //! event cuts
472 TH1F *fh1VertexNContributors; //! NContributors to prim vertex
473 TH1F *fh1VertexZ; //! prim vertex z distribution
474 TH1F *fh1EvtMult; //! number of reconstructed tracks after cuts
475 TH1F *fh1EvtCent; //! centrality percentile
476
477 TProfile* fh1Xsec; //! pythia cross section and trials
478 TH1F* fh1Trials; //! sum of trials
479 TH1F* fh1PtHard; //! pt hard of the event
480 TH1F* fh1PtHardTrials; //! pt hard of the event
481
482 TH1F *fh1nRecJetsCuts; //! number of jets from reconstructed tracks per event
483 TH1F *fh1nGenJets; //! number of jets from generated tracks per event
484 TH1F *fh1nRecEffJets; //! number of jets for reconstruction eff per event
485 TH1F *fh1nEmbeddedJets; //! number of embedded jets per event
486
487 TH1F *fh1nRecBckgJetsCuts; //! number of jets from reconstructed tracks per event
488 TH1F *fh1nGenBckgJets; //! number of jets from generated tracks per event
489 TH2F *fh2PtRecVsGenPrim; //! association rec/gen MC: rec vs gen pt, primaries
490 TH2F *fh2PtRecVsGenSec; //! association rec/gen MC: rec vs gen pt, secondaries
491
492 // tracking efficiency / secondaries
493
494 AliFragFuncQATrackHistos* fQATrackHistosRecEffGen; //! tracking efficiency: generated primaries
495 AliFragFuncQATrackHistos* fQATrackHistosRecEffRec; //! tracking efficiency: reconstructed primaries
496 AliFragFuncQATrackHistos* fQATrackHistosSecRecNS; //! reconstructed secondaries (non-strangeness)
497 AliFragFuncQATrackHistos* fQATrackHistosSecRecS; //! reconstructed secondaries (strange mothers)
498 AliFragFuncQATrackHistos* fQATrackHistosSecRecSsc; //! reconstructed secondaries (strange mothers) - scale factor
499
500 AliFragFuncHistos* fFFHistosRecEffRec; //! tracking efficiency: FF reconstructed primaries
501 AliFragFuncHistos* fFFHistosSecRecNS; //! secondary contamination: FF reconstructed secondaries (non-strangeness)
502 AliFragFuncHistos* fFFHistosSecRecS; //! secondary contamination: FF reconstructed secondaries (strange mothers)
503 AliFragFuncHistos* fFFHistosSecRecSsc; //! secondary contamination: FF reconstructed secondaries (strange mothers) - scale factor
504
505
506 // Background
507 TH1F *fh1BckgMult0; //! background multiplicity
508 TH1F *fh1BckgMult1; //! background multiplicity
509 TH1F *fh1BckgMult2; //! background multiplicity
510 TH1F *fh1BckgMult3; //! background multiplicity
511 TH1F *fh1BckgMult4; //! background multiplicity
512
513 // embedding
514 TH1F* fh1FractionPtEmbedded; //! ratio embedded pt in rec jet to embedded jet pt
515 TH1F* fh1IndexEmbedded; //! index embedded jet matching to leading rec jet
516 TH2F* fh2DeltaPtVsJetPtEmbedded; //! delta pt rec - embedded jet
517 TH2F* fh2DeltaPtVsRecJetPtEmbedded; //! delta pt rec - embedded jet
518 TH1F* fh1DeltaREmbedded; //! delta R rec - embedded jet
519
520
521 AliFragFuncQATrackHistos* fQABckgHisto0RecCuts; //! track QA: reconstructed tracks after cuts
522 AliFragFuncQATrackHistos* fQABckgHisto0Gen; //! track QA: generated tracks
523 AliFragFuncQATrackHistos* fQABckgHisto1RecCuts; //! track QA: reconstructed tracks after cuts
524 AliFragFuncQATrackHistos* fQABckgHisto1Gen; //! track QA: generated tracks
525 AliFragFuncQATrackHistos* fQABckgHisto2RecCuts; //! track QA: reconstructed tracks after cuts
526 AliFragFuncQATrackHistos* fQABckgHisto2Gen; //! track QA: generated tracks
527 AliFragFuncQATrackHistos* fQABckgHisto3RecCuts; //! track QA: reconstructed tracks after cuts
528 AliFragFuncQATrackHistos* fQABckgHisto3Gen; //! track QA: generated tracks
529 AliFragFuncQATrackHistos* fQABckgHisto4RecCuts; //! track QA: reconstructed tracks after cuts
530 AliFragFuncQATrackHistos* fQABckgHisto4Gen; //! track QA: generated tracks
531
532 AliFragFuncHistos* fFFBckgHisto0RecCuts; //! Bckg (outside leading jet or 2 jets or more) FF reconstructed tracks after cuts
533 AliFragFuncHistos* fFFBckgHisto0Gen; //! Bckg (outside leading jet or 2 jets or more) FF generated tracks after cuts
534 AliFragFuncHistos* fFFBckgHisto1RecCuts; //! Bckg (outside leading jet or 2 jets or more) FF reconstructed tracks after cuts
535 AliFragFuncHistos* fFFBckgHisto1Gen; //! Bckg (outside leading jet or 2 jets or more) FF generated tracks after cuts
536 AliFragFuncHistos* fFFBckgHisto2RecCuts; //! Bckg (outside leading jet or 2 jets or more) FF reconstructed tracks after cuts
537 AliFragFuncHistos* fFFBckgHisto2Gen; //! Bckg (outside leading jet or 2 jets or more) FF generated tracks after cuts
538 AliFragFuncHistos* fFFBckgHisto3RecCuts; //! Bckg (outside leading jet or 3 jets or more) FF reconstructed tracks after cuts
539 AliFragFuncHistos* fFFBckgHisto3Gen; //! Bckg (outside leading jet or 3 jets or more) FF generated tracks after cuts
540 AliFragFuncHistos* fFFBckgHisto4RecCuts; //! Bckg (outside leading jet or 4 jets or more) FF reconstructed tracks after cuts
541 AliFragFuncHistos* fFFBckgHisto4Gen; //! Bckg (outside leading jet or 4 jets or more) FF generated tracks after cuts
542
543 AliFragFuncHistos* fFFBckgHisto0RecEffRec; //! Bckg (outside leading jet or 2 jets or more) FF reconstructed primaries after cuts
544 AliFragFuncHistos* fFFBckgHisto0SecRecNS; //! secondary contamination: FF reconstructed secondaries (non-strangeness)
545 AliFragFuncHistos* fFFBckgHisto0SecRecS; //! secondary contamination: FF reconstructed secondaries (strange mothers)
546 AliFragFuncHistos* fFFBckgHisto0SecRecSsc; //! secondary contamination: FF reconstructed secondaries (strange mothers) - scale factor
547
548 TProfile* fProNtracksLeadingJet; //! jet shape
549 TProfile* fProDelR80pcPt; //! jet shape
550 TProfile* fProDelRPtSum[5]; //! jet shape
551
552 TProfile* fProNtracksLeadingJetGen; //! jet shape
553 TProfile* fProDelR80pcPtGen; //! jet shape
554 TProfile* fProDelRPtSumGen[5]; //! jet shape
555
556 TProfile* fProNtracksLeadingJetBgrPerp2; //! jet shape
557 TProfile* fProDelRPtSumBgrPerp2[5]; //! jet shape
558
559 TProfile* fProNtracksLeadingJetRecPrim; //! jet shape
560 TProfile* fProDelR80pcPtRecPrim; //! jet shape
561 TProfile* fProDelRPtSumRecPrim[5]; //! jet shape
562
563 TProfile* fProNtracksLeadingJetRecSecNS; //! jet shape
564 TProfile* fProDelRPtSumRecSecNS[5]; //! jet shape
565
566 TProfile* fProNtracksLeadingJetRecSecS; //! jet shape
567 TProfile* fProDelRPtSumRecSecS[5]; //! jet shape
568
569 TProfile* fProNtracksLeadingJetRecSecSsc; //! jet shape
570 TProfile* fProDelRPtSumRecSecSsc[5]; //! jet shape
571
572
9d7ad2e4 573 TRandom3* fRandom; //! TRandom3 for background estimation
c4856fb1 574
575 Bool_t fOnlyLeadingJets; // Flag indicating whether some histos are filled with leading jets only or all jets
e131b05f 576
577 // PID framework
578 Int_t fNumInclusivePIDtasks; // Number of inclusive PID tasks used
579 Int_t fNumJetPIDtasks; // Number of jet PID tasks used
580
9d7ad2e4 581 TString* fNameInclusivePIDtask; //[fNumInclusivePIDtasks] Names of the tasks for inclusive PID spectra
582 TString* fNameJetPIDtask; //[fNumJetPIDtasks] Names of the tasks for jet PID spectra
e131b05f 583
9d7ad2e4 584 AliAnalysisTaskPID** fInclusivePIDtask; //! Pointer to tasks for inclusive PID spectra
585 AliAnalysisTaskPID** fJetPIDtask; //! Pointer to tasks for jet PID spectra
e131b05f 586
587 Bool_t fUseInclusivePIDtask; // Process inclusive PID spectra?
588 Bool_t fUseJetPIDtask; // Process jet PID spectra?
589
590 Bool_t fIsPP; // Is pp collision system? -> If yes, centrality will be set to -1
591
592 AliFragFuncHistos* fIDFFHistosRecCuts[AliPID::kSPECIES]; //! Identified FF reconstructed tracks after cuts
593 AliFragFuncHistos* fIDFFHistosGen[AliPID::kSPECIES]; //! Identified FF generated tracks after cuts
594
c4856fb1 595 ClassDef(AliAnalysisTaskIDFragmentationFunction, 18);
e131b05f 596};
597
598
9d7ad2e4 599inline void AliAnalysisTaskIDFragmentationFunction::SetNamesOfInclusivePIDtasks(Int_t numNames, const TString* names)
e131b05f 600{
601 delete [] fNameInclusivePIDtask;
602 fNameInclusivePIDtask = 0x0;
e131b05f 603
604 if (!names || numNames < 0) {
605 fNumInclusivePIDtasks = 0;
606 return;
607 }
608
609 fNumInclusivePIDtasks = numNames;
610
611 if (numNames > 0) {
612 fNameInclusivePIDtask = new TString[numNames];
e131b05f 613
614 for (Int_t i = 0; i < numNames; i++) {
615 fNameInclusivePIDtask[i] = names[i];
e131b05f 616 }
617 }
618}
619
9d7ad2e4 620inline void AliAnalysisTaskIDFragmentationFunction::SetNamesOfJetPIDtasks(Int_t numNames, const TString* names)
e131b05f 621{
622 delete [] fNameJetPIDtask;
623 fNameJetPIDtask = 0x0;
e131b05f 624
625 if (!names || numNames < 0) {
626 fNumJetPIDtasks = 0;
627 return;
628 }
629
630 fNumJetPIDtasks = numNames;
631
632 if (numNames > 0) {
633 fNameJetPIDtask = new TString[numNames];
e131b05f 634
635 for (Int_t i = 0; i < numNames; i++) {
636 fNameJetPIDtask[i] = names[i];
e131b05f 637 }
638 }
639}
640
641#endif