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