Add the efficiency estimation to common FF task.
[u/mrichter/AliRoot.git] / PWG4 / JetTasks / AliAnalysisTaskFragmentationFunction.h
1 /*************************************************************************
2  * Task for Fragmentation Function Analysis in PWG4 Jet Task Force Train *
3  *************************************************************************/
4
5 #ifndef ALIANALYSISTASKFRAGMENTATIONFUNCTION_H
6 #define ALIANALYSISTASKFRAGMENTATIONFUNCTION_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
13 class AliESDEvent;
14 class AliAODEvent;
15 class TList;
16 class TH1F;
17 class TH2F;
18
19 #include "THnSparse.h"
20 #include "AliAnalysisTaskSE.h"
21
22 class AliAnalysisTaskFragmentationFunction : public AliAnalysisTaskSE {
23
24  public:
25   
26   //----------------------------------------
27   class AliFragFuncHistos : public TObject
28   {
29     
30     public:
31     
32     AliFragFuncHistos(const char* name = "FFhistos", 
33              Int_t nJetPt = 0, Float_t jetPtMin = 0, Float_t jetPtMax = 0,
34              Int_t nPt = 0, Float_t ptMin = 0, Float_t ptMax = 0,
35              Int_t nXi = 0, Float_t xiMin = 0, Float_t xiMax = 0,
36              Int_t nZ  = 0, Float_t zMin  = 0, Float_t zMax  = 0);
37     AliFragFuncHistos(const AliFragFuncHistos& copy);
38     AliFragFuncHistos& operator=(const AliFragFuncHistos &o);
39     virtual ~AliFragFuncHistos();
40     
41     virtual void DefineHistos();
42     virtual void FillFF(Float_t trackPt, Float_t jetPt,Bool_t incrementJetPt);
43     virtual void AddToOutput(TList* list) const;
44
45   private:
46
47     Int_t   fNBinsJetPt; // FF histos bins
48     Float_t fJetPtMin;   // FF histos limits
49     Float_t fJetPtMax;   // FF histos limits
50     Int_t   fNBinsPt;    // FF histos bins
51     Float_t fPtMin;      // FF histos limits
52     Float_t fPtMax;      // FF histos limits
53     Int_t   fNBinsXi;    // FF histos bins
54     Float_t fXiMin;      // FF histos limits
55     Float_t fXiMax;      // FF histos limits
56     Int_t   fNBinsZ;     // FF histos bins
57     Float_t fZMin;       // FF histos limits
58     Float_t fZMax;       // FF histos limits
59   
60     TH2F*   fh2TrackPt;   //! FF: track transverse momentum 
61     TH2F*   fh2Xi;        //! FF: xi 
62     TH2F*   fh2Z;         //! FF: z  
63     TH1F*   fh1JetPt;     //! jet pt 
64
65     TString fName;       // histo names prefix
66     
67     ClassDef(AliFragFuncHistos, 1);
68   };
69   
70   //----------------------------------------
71   class AliFragFuncQAJetHistos : public TObject
72   {
73
74   public:
75  
76     AliFragFuncQAJetHistos(const char* name = "QAJethistos",
77                 Int_t nPt  = 0, Float_t ptMin  = 0, Float_t ptMax  = 0,
78                 Int_t nEta = 0, Float_t etaMin = 0, Float_t etaMax = 0,
79                 Int_t nPhi = 0, Float_t phiMin = 0, Float_t phiMax = 0);
80       
81     AliFragFuncQAJetHistos(const AliFragFuncQAJetHistos& copy);
82     AliFragFuncQAJetHistos& operator=(const AliFragFuncQAJetHistos &o);
83     virtual ~AliFragFuncQAJetHistos();
84     virtual void DefineHistos();
85     virtual void FillJetQA(Float_t eta, Float_t phi, Float_t pt);
86     virtual void AddToOutput(TList* list) const;
87
88   private:
89     
90     Int_t   fNBinsPt;    // jet QA histos bins
91     Float_t fPtMin;      // jet QA histos limits
92     Float_t fPtMax;      // jet QA histos limits
93     Int_t   fNBinsEta;   // jet QA histos bins
94     Float_t fEtaMin;     // jet QA histos limits
95     Float_t fEtaMax;     // jet QA histos limits
96     Int_t   fNBinsPhi;   // jet QA histos bins
97     Float_t fPhiMin;     // jet QA histos limits
98     Float_t fPhiMax;     // jet QA histos limits
99     
100     TH2F*   fh2EtaPhi;   //! jet phi vs eta 
101     TH1F*   fh1Pt;       //! jet transverse momentum 
102     TString fName;       // histo names prefix
103     
104     ClassDef(AliFragFuncQAJetHistos, 1);
105   };
106   
107   //----------------------------------------
108   class AliFragFuncQATrackHistos : public TObject
109   {
110
111   public:
112
113     AliFragFuncQATrackHistos(const char* name = "QATrackhistos", 
114                   Int_t nPt  = 0, Float_t ptMin  = 0, Float_t ptMax  = 0,
115                   Int_t nEta = 0, Float_t etaMin = 0, Float_t etaMax = 0,
116                   Int_t nPhi = 0, Float_t phiMin = 0, Float_t phiMax = 0, 
117                   Float_t ptThresh = 0);
118     
119     AliFragFuncQATrackHistos(const AliFragFuncQATrackHistos& copy);
120     AliFragFuncQATrackHistos& operator=(const AliFragFuncQATrackHistos &o);
121     virtual ~AliFragFuncQATrackHistos();
122     virtual void DefineHistos();
123     virtual void FillTrackQA(Float_t eta, Float_t phi, Float_t pt);
124     virtual void AddToOutput(TList* list) const;
125
126   private:
127     
128     Int_t   fNBinsPt;    // track QA histos bins in pt
129     Float_t fPtMin;      // track QA histos limits in pt
130     Float_t fPtMax;      // track QA histos limits in pt
131     Int_t   fNBinsEta;   // track QA histos bins in eta
132     Float_t fEtaMin;     // track QA histos limits in eta
133     Float_t fEtaMax;     // track QA histos limits in eta
134     Int_t   fNBinsPhi;   // track QA histos bins in phi
135     Float_t fPhiMin;     // track QA histos limits in phi
136     Float_t fPhiMax;     // track QA histos limits in phi
137
138     Float_t fHighPtThreshold; //  high pt track phi vs eta distribution
139
140     TH2F*   fh2EtaPhi;        //! track phi vs eta 
141     TH1F*   fh1Pt;            //! track transverse momentum 
142     TH2F*   fh2HighPtEtaPhi;  //! phi vs eta for high pt (>fgHighPtThreshold) tracks
143
144     TString fName;            // histo names prefix
145     
146     ClassDef(AliFragFuncQATrackHistos, 1);
147   };
148   
149   //----------------------------------------
150   class AliFragFuncIntraJetHistos : public TObject
151   {
152
153   public:
154     
155     AliFragFuncIntraJetHistos(const char* name = "IntraJethistos", 
156              Int_t nJetPt    = 0, Float_t jetPtMin    = 0, Float_t jetPtMax    = 0,
157              Int_t nPt       = 0, Float_t ptMin       = 0, Float_t ptMax       = 0,
158              Int_t nZ        = 0, Float_t zMin        = 0, Float_t zMax        = 0,
159              Int_t nCosTheta = 0, Float_t costhetaMin = 0, Float_t costhetaMax = 0,
160              Int_t nTheta    = 0, Float_t thetaMin    = 0, Float_t thetaMax    = 0,
161              Int_t nJt       = 0, Float_t jtMin       = 0, Float_t jtMax       = 0);
162     AliFragFuncIntraJetHistos(const AliFragFuncIntraJetHistos& copy);
163     AliFragFuncIntraJetHistos& operator=(const AliFragFuncIntraJetHistos &o);
164     virtual ~AliFragFuncIntraJetHistos();
165     
166     virtual void DefineHistos();
167     virtual void FillIntraJet(TLorentzVector* trackV, TLorentzVector* jetV);
168     virtual void AddToOutput(TList* list) const;
169
170   private:
171
172     Int_t   fNBinsJetPt;    // IntraJet histos bins in jet pt
173     Float_t fJetPtMin;      // IntraJet histos limits in jet pt
174     Float_t fJetPtMax;      // IntraJet histos limits in jet pt
175     Int_t   fNBinsPt;       // IntraJet histos bins in pt
176     Float_t fPtMin;         // IntraJet histos limits in pt
177     Float_t fPtMax;         // IntraJet histos limits in pt
178     Int_t   fNBinsZ;        // IntraJet histos bins in z
179     Float_t fZMin;          // IntraJet histos limits in z
180     Float_t fZMax;          // IntraJet histos limits in z
181     Int_t   fNBinsJt;       // IntraJet histos bins in jt
182     Float_t fJtMin;         // IntraJet histos limits in jt
183     Float_t fJtMax;         // IntraJet histos limits in jt
184     Int_t   fNBinsTheta;    // IntraJet histos bins in theta
185     Float_t fThetaMin;      // IntraJet histos limits in theta
186     Float_t fThetaMax;      // IntraJet histos limits in theta
187     Int_t   fNBinsCosTheta; // IntraJet histos bins in cos(theta)
188     Float_t fCosThetaMin;   // IntraJet histos limits in cos(theta)
189     Float_t fCosThetaMax;   // IntraJet histos limits in cos(theta)
190   
191     TH2F*  fh2Theta;         //! IntraJet: theta distribution
192     TH2F*  fh2CosTheta;      //! IntraJet: cos(theta) distribution
193     TH2F*  fh2Jt;            //! IntraJet: jt distribution
194     TH2F*  fh2PtvsZ;         //! IntraJet: pt vs z distribution
195
196     THnSparseF* fhnIntraJet; //! IntraJet
197     Int_t fnDim;             // HnSparseF dimensions 
198
199     TString fName;           // histo names prefix
200     
201     ClassDef(AliFragFuncIntraJetHistos, 1);
202   };
203
204   //----------------------------------------
205   class AliFragFuncDiJetHistos : public TObject
206   {
207
208     public:
209     
210     AliFragFuncDiJetHistos(const char* name = "DiJetHistos", Int_t kindSlices = 0,
211              Int_t nJetinvMass = 0, Float_t jetInvMassMin = 0, Float_t jetInvMassMax = 0,
212              Int_t nJetPt = 0, Float_t jetPtMin = 0, Float_t jetPtMax = 0,
213              Int_t nPt = 0, Float_t ptMin = 0, Float_t ptMax = 0,
214              Int_t nXi = 0, Float_t xiMin = 0, Float_t xiMax = 0,
215              Int_t nZ  = 0, Float_t zMin  = 0, Float_t zMax  = 0);
216     AliFragFuncDiJetHistos(const AliFragFuncDiJetHistos& copy);
217     AliFragFuncDiJetHistos& operator=(const AliFragFuncDiJetHistos &o);
218     virtual ~AliFragFuncDiJetHistos();
219     
220     virtual void DefineDiJetHistos();
221     virtual void FillDiJetFF(Int_t jetType, Float_t trackPt, Float_t jetPt, Double_t jetBin, Bool_t incrementJetPt);
222     virtual void AddToOutput(TList* list) const;
223     
224     private:
225
226     Int_t   fKindSlices;
227     Int_t   fNBinsJetInvMass; // FF histos bins
228     Float_t fJetInvMassMin;   // FF histos limits
229     Float_t fJetInvMassMax;   // FF histos limits
230     Int_t   fNBinsJetPt;      // FF histos bins
231     Float_t fJetPtMin;        // FF histos limits
232     Float_t fJetPtMax;        // FF histos limits
233     Int_t   fNBinsPt;         // FF histos bins
234     Float_t fPtMin;           // FF histos limits
235     Float_t fPtMax;           // FF histos limits
236     Int_t   fNBinsXi;         // FF histos bins
237     Float_t fXiMin;           // FF histos limits
238     Float_t fXiMax;           // FF histos limits
239     Int_t   fNBinsZ;          // FF histos bins
240     Float_t fZMin;            // FF histos limits
241     Float_t fZMax;            // FF histos limits
242
243     TH2F*   fh2TrackPtJet1; //! FF dijet : track transverse momentum of jet 1 
244     TH2F*   fh2TrackPtJet2; //! FF dijet : track transverse momentum of jet 2 
245     TH2F*   fh2TrackPtJet;  //! FF dijet : track transverse momentum of jets 1 and 2   
246     TH1F*   fh1Jet1Pt;      //! jet 1 pt 
247     TH1F*   fh1Jet2Pt;      //! jet 2 pt 
248     TH1F*   fh1JetPt;       //! jet 1 and 2 pt 
249     TH2F*   fh2Xi1;         //! FF dijet : xi of jet 1 
250     TH2F*   fh2Xi2;         //! FF dijet : xi of jet 2
251     TH2F*   fh2Xi;          //! FF dijet : xi of jet 1 and 2 
252     TH2F*   fh2Z1;          //! FF dijet : z of jet 1   
253     TH2F*   fh2Z2;          //! FF dijet : z of jet 2 
254     TH2F*   fh2Z;           //! FF dijet : z of jet 1 and 2 
255     TH2F*   fh2Pt1;         //! FF dijet : z of jet 1   
256     TH2F*   fh2Pt2;         //! FF dijet : z of jet 2 
257     TH2F*   fh2Pt;          //! FF dijet : z of jet 1 and 2 
258
259     TString fName;          // histo names prefix
260     
261     ClassDef(AliFragFuncDiJetHistos, 1);
262   };
263
264   //----------------------------------------
265   class AliFragFuncQADiJetHistos : public TObject
266   {
267     public:
268     AliFragFuncQADiJetHistos(const char* name = "QADiJetHistos", Int_t kindSlices = 0,
269              Int_t nInvMass = 0, Float_t invMassMin = 0, Float_t invMassMax = 0,
270              Int_t nJetPt = 0,   Float_t jetPtMin = 0, Float_t jetPtMax = 0,
271              Int_t nDeltaPhi = 0, Float_t deltaPhiMin = 0, Float_t deltaPhiMax = 0,
272              Int_t nDeltaEta = 0, Float_t deltaEtaMin = 0, Float_t deltaEtaMax = 0,
273              Int_t nDeltaPt  = 0, Float_t deltaPtMin  = 0, Float_t deltaPtMax  = 0);
274     AliFragFuncQADiJetHistos(const AliFragFuncQADiJetHistos& copy);
275     AliFragFuncQADiJetHistos& operator=(const AliFragFuncQADiJetHistos &o);
276     virtual ~AliFragFuncQADiJetHistos();
277     
278     virtual void DefineQADiJetHistos();
279     virtual void FillDiJetQA(Double_t invMass, Double_t deltaPhi, Double_t deltaEta, Double_t deltaPt, Double_t jetBin);
280     virtual void AddToOutput(TList* list) const;
281     
282     private:
283     
284     Int_t   fKindSlices;
285     Int_t   fNBinsJetInvMass;  // FF histos bins in jet invariant mass
286     Float_t fJetInvMassMin;    // FF histos limits in jet invariant mass
287     Float_t fJetInvMassMax;    // FF histos limits in jet invariant mass
288     Int_t   fNBinsJetPt;       // FF histos bins in jet pt
289     Float_t fJetPtMin;         // FF histos limits in jet pt
290     Float_t fJetPtMax;         // FF histos limits in jet pt
291     Int_t   fNBinsDeltaPhi;    // FF histos bins in jet delta phi
292     Float_t fDeltaPhiMin;      // FF histos limits in jet delta phi
293     Float_t fDeltaPhiMax;      // FF histos limits in jet delta phi
294     Int_t   fNBinsDeltaEta;    // FF histos bins in jet delta eta
295     Float_t fDeltaEtaMin;      // FF histos limits in jet delta eta
296     Float_t fDeltaEtaMax;      // FF histos limits in jet delta eta
297     Int_t   fNBinsDeltaPt;     // FF histos bins in jet delta pt
298     Float_t fDeltaPtMin;       // FF histos limits in jet delta pt
299     Float_t fDeltaPtMax;       // FF histos limits in jet delta pt
300
301     TH2F*   fh2InvMass;        // FF dijet invariant mass histos
302     TH2F*   fh2DeltaPhi;       // FF dijet delta phi histos
303     TH2F*   fh2DeltaEta;       // FF dijet delta eta histos
304     TH2F*   fh2DeltaPt;        // FF dijet delta pt histos
305
306     TString fName;             // histo names prefix
307     
308     ClassDef(AliFragFuncQADiJetHistos, 1);
309   };
310
311   AliAnalysisTaskFragmentationFunction(); 
312   AliAnalysisTaskFragmentationFunction(const char *name);
313   AliAnalysisTaskFragmentationFunction(const  AliAnalysisTaskFragmentationFunction &copy);
314   AliAnalysisTaskFragmentationFunction& operator=(const  AliAnalysisTaskFragmentationFunction &o);
315   virtual ~AliAnalysisTaskFragmentationFunction();
316   
317   virtual void   UserCreateOutputObjects();
318   virtual void   Init();
319   virtual void   LocalInit() {Init();}
320   virtual void   UserExec(Option_t *option);
321   virtual void   Terminate(Option_t* );
322   
323   virtual void   SetTrackTypeGen(Int_t i){fTrackTypeGen = i;}
324   virtual void   SetJetTypeGen(Int_t i){fJetTypeGen = i;}
325   virtual void   SetJetTypeRecEff(Int_t i){fJetTypeRecEff = i;}
326
327   virtual void   SetBranchGenJets(const char* c){fBranchGenJets = c;}
328   virtual void   SetBranchRecJets(const char* c){fBranchRecJets = c;}
329
330   virtual void   SetTrackCuts(Float_t trackPt = 0.15, Float_t trackEtaMin = -0.9, Float_t trackEtaMax = 0.9, 
331                               Float_t trackPhiMin = 0., Float_t trackPhiMax = 2*TMath::Pi())
332   {fTrackPtCut = trackPt; fTrackEtaMin = trackEtaMin; fTrackEtaMax = trackEtaMax; 
333     fTrackPhiMin = trackPhiMin; fTrackPhiMax = trackPhiMax;}
334   virtual void   SetFilterMask(UInt_t i) {fFilterMask = i;}
335   virtual void   SetJetCuts(Float_t jetPt = 5., Float_t jetEtaMin = -0.5, Float_t jetEtaMax = 0.5, 
336                             Float_t jetPhiMin = 0., Float_t jetPhiMax = 2*TMath::Pi())
337   {fJetPtCut = jetPt; fJetEtaMin = jetEtaMin; fJetEtaMax = jetEtaMax; 
338     fJetPhiMin = jetPhiMin; fJetPhiMax = jetPhiMax;}
339   virtual void   SetDiJetCuts(Int_t cutType = 1, Float_t deltaPhiCut = 0.,  
340                               Float_t cdfCut = 0.5, Float_t ptFractionCut = 0.6)
341   {fDiJetCut = cutType; fDiJetDeltaPhiCut = deltaPhiCut;  
342     fDiJetCDFCut = cdfCut; fDiJetPtFractionCut = ptFractionCut;}
343   virtual void   SetKindSlices(Int_t slice = 1) {fDiJetKindBins = slice;}
344
345   virtual void   SetFFRadius(Float_t r = 0.4) { fFFRadius = r; }
346
347   static  void   SetProperties(TH1* h,const char* x, const char* y);
348   static  void   SetProperties(TH2* h,const char* x, const char* y,const char* z);
349   static  void   SetProperties(THnSparse* h,const Int_t dim, const char** labels);
350
351   void   SetHighPtThreshold(Float_t pt = 5.) { fQATrackHighPtThreshold = pt; }
352
353
354   void   SetFFHistoBins(Int_t nJetPt = 55, Float_t jetPtMin = 5, Float_t jetPtMax = 60, 
355                         Int_t nPt = 70, Float_t ptMin = 0., Float_t ptMax = 70., 
356                         Int_t nXi = 70, Float_t xiMin = 0., Float_t xiMax = 7.,
357                         Int_t nZ = 22,  Float_t zMin = 0.,  Float_t zMax = 1.1)
358   { fFFNBinsJetPt = nJetPt; fFFJetPtMin = jetPtMin; fFFJetPtMax = jetPtMax; 
359     fFFNBinsPt = nPt; fFFPtMin = ptMin; fFFPtMax = ptMax;
360     fFFNBinsXi = nXi; fFFXiMin = xiMin; fFFXiMax = xiMax;
361     fFFNBinsZ  = nZ;  fFFZMin  = zMin;  fFFZMax  = zMax; }
362   
363   void  SetQAJetHistoBins(Int_t nPt = 200, Float_t ptMin = 0., Float_t ptMax = 200.,
364                           Int_t nEta = 20, Float_t etaMin = -1.0, Float_t etaMax = 1.0,
365                           Int_t nPhi = 60, Float_t phiMin = 0., Float_t phiMax = 2*TMath::Pi())
366     { fQAJetNBinsPt = nPt; fQAJetPtMin = ptMin; fQAJetPtMax = ptMax;
367       fQAJetNBinsEta = nEta; fQAJetEtaMin = etaMin; fQAJetEtaMax = etaMax;
368       fQAJetNBinsPhi = nPhi; fQAJetPhiMin = phiMin; fQAJetPhiMax = phiMax; }
369   
370   void  SetQATrackHistoBins(Int_t nPt = 70, Float_t ptMin = 0., Float_t ptMax = 70.,
371                             Int_t nEta = 20, Float_t etaMin = -1.0, Float_t etaMax = 1.0,
372                             Int_t nPhi = 60, Float_t phiMin = 0., Float_t phiMax = 2*TMath::Pi())
373   { fQATrackNBinsPt = nPt; fQATrackPtMin = ptMin; fQATrackPtMax = ptMax;
374     fQATrackNBinsEta = nEta; fQATrackEtaMin = etaMin; fQATrackEtaMax = etaMax;
375     fQATrackNBinsPhi = nPhi; fQATrackPhiMin = phiMin; fQATrackPhiMax = phiMax; }
376   
377   void   SetIJHistoBins(Int_t nJetPt = 55, Float_t jetPtMin = 5, Float_t jetPtMax = 60, 
378                         Int_t nPt = 70, Float_t ptMin = 0., Float_t ptMax = 70., 
379                         Int_t nZ = 22,  Float_t zMin = 0.,  Float_t zMax = 1.1,
380                         Int_t nCosTheta = 100,  Float_t costhetaMin = 0.,  Float_t costhetaMax = 1.,
381                         Int_t nTheta = 200,  Float_t thetaMin = -0.5,  Float_t thetaMax = 1.5,
382                         Int_t nJt = 25,  Float_t jtMin = 0.,  Float_t jtMax = 5.)
383   { fIJNBinsJetPt = nJetPt; fIJJetPtMin = jetPtMin; fIJJetPtMax = jetPtMax; 
384     fIJNBinsPt = nPt; fIJPtMin = ptMin; fIJPtMax = ptMax;
385     fIJNBinsZ = nZ; fIJZMin = zMin; fIJZMax = zMax;
386     fIJNBinsCosTheta  = nCosTheta;  fIJCosThetaMin  = costhetaMin;  fIJCosThetaMax  = costhetaMax;
387     fIJNBinsTheta  = nTheta;  fIJThetaMin  = thetaMin;  fIJThetaMax  = thetaMax;
388     fIJNBinsJt  = nJt;  fIJJtMin  = jtMin;  fIJJtMax  = jtMax; }
389
390   void SetDiJetHistoBins(Int_t nJetInvMass = 55, Float_t jetInvMassMin = 5, Float_t jetInvMassMax = 60, 
391                          Int_t nJetPt = 55, Float_t jetPtMin = 5, Float_t jetPtMax = 60, 
392                          Int_t nPt = 70, Float_t ptMin = 0., Float_t ptMax = 70., 
393                          Int_t nXi = 70, Float_t xiMin = 0., Float_t xiMax = 7.,
394                          Int_t nZ = 22,  Float_t zMin = 0.,  Float_t zMax = 1.1)
395   {
396     fDiJetNBinsJetInvMass = nJetInvMass; fDiJetJetInvMassMin = jetInvMassMin; fDiJetJetInvMassMax = jetInvMassMax;
397     fDiJetNBinsJetPt = nJetPt; fDiJetJetPtMin = jetPtMin; fDiJetJetPtMax = jetPtMax;
398     fDiJetNBinsPt = nPt; fDiJetPtMin = ptMin; fDiJetPtMax = ptMax;
399     fDiJetNBinsXi = nXi; fDiJetXiMin = xiMin; fDiJetXiMax = xiMax;
400     fDiJetNBinsZ = nZ; fDiJetZMin = zMin; fDiJetZMax = zMax;
401   }
402
403   void SetQADiJetHistoBins(Int_t nInvMass = 55, Float_t invMassMin = 5., Float_t invMassMax = 60.,
404                            Int_t nJetPt = 55, Float_t jetPtMin = 5, Float_t jetPtMax = 60, 
405                            Int_t nDeltaPhi = 100, Float_t deltaPhiMin = 0., Float_t deltaPhiMax = TMath::Pi(),
406                            Int_t nDeltaEta = 22, Float_t deltaEtaMin = 0., Float_t deltaEtaMax = 1.1,
407                            Int_t nDeltaPt = 100, Float_t deltaPtMin = 0., Float_t deltaPtMax = 100.)
408   {
409     fQADiJetNBinsInvMass = nInvMass; fQADiJetInvMassMin = invMassMin; fQADiJetInvMassMax = invMassMax;
410     fQADiJetNBinsJetPt = nJetPt; fQADiJetJetPtMin = jetPtMin; fQADiJetJetPtMax = jetPtMax;
411     fQADiJetNBinsDeltaPhi = nDeltaPhi; fQADiJetDeltaPhiMin = deltaPhiMin; fQADiJetDeltaPhiMax = deltaPhiMax;
412     fQADiJetNBinsDeltaEta = nDeltaEta; fQADiJetDeltaEtaMin = deltaEtaMin; fQADiJetDeltaEtaMax = deltaEtaMax;
413     fQADiJetNBinsDeltaPt = nDeltaPt; fQADiJetDeltaPtMin = deltaPtMin; fQADiJetDeltaPtMax = deltaPtMax;
414
415   }
416
417   Float_t  GetFFRadius() const { return fFFRadius; }
418   void     GetJetTracksTrackrefs(TList* l, AliAODJet* j);
419   void     GetJetTracksPointing(TList* in, TList* out, AliAODJet* j, const Double_t r, Double_t& pt);  
420   Double_t GetDiJetBin(Double_t invMass, Double_t leadingJetPt, Double_t eMean, Int_t kindSlices); // function to find which bin fill
421   Double_t InvMass(AliAODJet* jet1, AliAODJet* jet2);
422   void     AssociateGenRec(TList* tracksAODMCCharged,TList* tracksRec, TArrayI& indexAODTr,TArrayI& indexMCTr,TArrayS& isGenPrim);
423   void     FillSingleTrackRecEffHisto(THnSparse* histo, TList* tracksGen, TList* tracksRec, TArrayI& indexAODTr, TArrayS& isGenPrim);
424   void     FillJetTrackRecEffHisto(THnSparse* histo,Double_t jetPhi,Double_t jetEta,Double_t jetPt,TList* jetTrackList, TList* tracksGen,
425                                    TArrayI& indexAODTr,TArrayS& isGenPrim);
426
427
428
429  private:
430     
431   // Consts
432   
433   enum {kTrackUndef=0, kTrackAOD, kTrackAODQualityCuts, kTrackAODCuts, kTrackKineAll, kTrackKineCharged, kTrackKineChargedAcceptance, 
434         kTrackAODMCAll, kTrackAODMCCharged, kTrackAODMCChargedAcceptance};
435   enum {kJetsUndef=0, kJetsRec, kJetsRecAcceptance, kJetsGen, kJetsGenAcceptance, kJetsKine, kJetsKineAcceptance};
436   
437   
438   Int_t   GetListOfTracks(TList* list, Int_t type);
439   Int_t   GetListOfJets(TList* list, Int_t type);
440   
441   AliESDEvent* fESD;      // ESD event
442   AliAODEvent* fAOD;      // AOD event
443   AliMCEvent*  fMCEvent;  // MC event
444   
445   TString fBranchRecJets;         // branch name for reconstructed jets
446   TString fBranchGenJets;         // branch name for generated jets
447   
448   Int_t   fTrackTypeGen;  // type of generated tracks
449   Int_t   fJetTypeGen;    // type of generated jets
450
451   Int_t   fJetTypeRecEff; // type of jets used for filling reconstruction efficiency histos
452
453   UInt_t  fFilterMask;    // filter bit for selected tracks
454         
455   // track cuts
456   Float_t fTrackPtCut;    // track transverse momentum cut
457   Float_t fTrackEtaMin;   // track eta cut
458   Float_t fTrackEtaMax;   // track eta cut
459   Float_t fTrackPhiMin;   // track phi cut
460   Float_t fTrackPhiMax;   // track phi cut
461   
462   // jet cuts
463   Float_t fJetPtCut;      // jet transverse momentum cut
464   Float_t fJetEtaMin;     // jet eta cut
465   Float_t fJetEtaMax;     // jet eta cut
466   Float_t fJetPhiMin;     // jet phi cut
467   Float_t fJetPhiMax;     // jet phi cut
468
469   // di-jet cuts
470   Int_t   fDiJetCut;            // dijet cut selection
471   Float_t fDiJetDeltaPhiCut;    // delta phi cut value
472   Float_t fDiJetPtFractionCut;  // fraction of pt cut value
473   Float_t fDiJetCDFCut;         // cdf cut value
474
475   Int_t   fDiJetKindBins;       // type of bins: invmass, etleading, emean
476   static  TArrayD* fDiJetBins;   // bining in invmass, etleading, emean
477
478   Float_t fFFRadius;        // if radius > 0 construct FF from tracks within cone around jet axis, otherwise use trackRefs  
479   
480   TList* fTracksRec;            //! reconstructed tracks
481   TList* fTracksRecCuts;        //! reconstructed tracks after cuts
482   TList* fTracksGen;            //! generated tracks 
483   TList* fTracksAODMCCharged;   //! AOD MC tracks 
484   TList* fTracksRecQualityCuts; //! reconstructed tracks after quality cuts, no acceptance/pt cut
485
486   
487   TList* fJetsRec;        //! jets from reconstructed tracks
488   TList* fJetsRecCuts;    //! jets from reonstructed tracks after jet cuts 
489   TList* fJetsGen;        //! jets from generated tracks
490   TList* fJetsRecEff;     //! jets used for reconstruction efficiency histos 
491   
492   
493   AliFragFuncQATrackHistos* fQATrackHistosRec;      //! track QA: reconstructed tracks
494   AliFragFuncQATrackHistos* fQATrackHistosRecCuts;  //! track QA: reconstructed tracks after cuts
495   AliFragFuncQATrackHistos* fQATrackHistosGen;      //! track QA: generated tracks
496   
497   AliFragFuncQAJetHistos*  fQAJetHistosRec;             //! jet QA: jets from reconstructed tracks
498   AliFragFuncQAJetHistos*  fQAJetHistosRecCuts;         //! jet QA: jets from reconstructed tracks after jet cuts 
499   AliFragFuncQAJetHistos*  fQAJetHistosRecCutsLeading;  //! jet QA: leading jet from reconstructed tracks after jet cuts 
500   AliFragFuncQAJetHistos*  fQAJetHistosGen;             //! jet QA: jets from generated tracks  
501   AliFragFuncQAJetHistos*  fQAJetHistosGenLeading;      //! jet QA: leading jet from generated tracks  
502   AliFragFuncQAJetHistos*  fQAJetHistosRecEffLeading;   //! jet QA: leading jet used for reconstruction efficiency histos  
503   
504
505   AliFragFuncHistos*  fFFHistosRecCuts;         //! FF reconstructed tracks after cuts 
506   AliFragFuncHistos*  fFFHistosRecLeading;      //! FF reconstructed tracks after cuts: all reconstructed tracks pt / leading track pt  
507   AliFragFuncHistos*  fFFHistosRecLeadingTrack; //! FF reconstructed tracks after cuts: leading track pt / jet pt
508   AliFragFuncHistos*  fFFHistosGen;             //! FF generated tracks after cuts 
509   AliFragFuncHistos*  fFFHistosGenLeading;      //! FF generated tracks after cuts: all generated tracks pt / leading track pt  
510   AliFragFuncHistos*  fFFHistosGenLeadingTrack; //! FF generated tracks after cuts: leading track pt / jet pt
511
512
513   AliFragFuncIntraJetHistos*  fIJHistosRecCuts;         //! IJ reconstructed tracks after cuts 
514   AliFragFuncIntraJetHistos*  fIJHistosRecLeading;      //! IJ reconstructed tracks after cuts: all reconstructed tracks pt / leading track pt  
515   AliFragFuncIntraJetHistos*  fIJHistosRecLeadingTrack; //! IJ reconstructed tracks after cuts: leading track pt / jet pt
516   AliFragFuncIntraJetHistos*  fIJHistosGen;             //! IJ generated tracks after cuts 
517   AliFragFuncIntraJetHistos*  fIJHistosGenLeading;      //! IJ generated tracks after cuts: all generated tracks pt / leading track pt  
518   AliFragFuncIntraJetHistos*  fIJHistosGenLeadingTrack; //! IJ generated tracks after cuts: leading track pt / jet pt
519
520   AliFragFuncDiJetHistos* fFFDiJetHistosRecCuts;         //! DiJet FF reconstructed tracks after cuts
521   AliFragFuncDiJetHistos* fFFDiJetHistosRecLeading;      //! DiJet FF reconstructed tracks after cuts: all reconstructed tracks pt / leading track pt
522   AliFragFuncDiJetHistos* fFFDiJetHistosRecLeadingTrack; //! DiJet FF reconstructed tracks after cuts: leading track pt / jet pt
523
524   AliFragFuncDiJetHistos* fFFDiJetHistosGen;             //! DiJet FF generated tracks after cuts 
525   AliFragFuncDiJetHistos* fFFDiJetHistosGenLeading;      //! DiJet FF generated tracks after cuts: all generated tracks pt / leading track pt 
526   AliFragFuncDiJetHistos* fFFDiJetHistosGenLeadingTrack; //! DiJet FF generated tracks after cuts: leading track pt / jet pt 
527
528   AliFragFuncQADiJetHistos* fQADiJetHistosRecCuts;       //! Dijet QA : reconstructed tracks after cuts
529   AliFragFuncQADiJetHistos* fQADiJetHistosGen;           //! DiJet QA: jets from generated tracks  
530
531
532   Float_t  fQATrackHighPtThreshold;       // track QA high transverse momentum threshold
533   
534   // histogram bins  
535
536   Int_t   fFFNBinsJetPt;    // FF histos bins
537   Float_t fFFJetPtMin;      // FF histos limits
538   Float_t fFFJetPtMax;      // FF histos limits
539
540   Int_t   fFFNBinsPt;       // FF histos bins
541   Float_t fFFPtMin;         // FF histos limits
542   Float_t fFFPtMax;         // FF histos limits
543
544   Int_t   fFFNBinsXi;       // FF histos bins
545   Float_t fFFXiMin;         // FF histos limits
546   Float_t fFFXiMax;         // FF histos limits
547
548   Int_t   fFFNBinsZ;        // FF histos bins
549   Float_t fFFZMin;          // FF histos limits
550   Float_t fFFZMax;          // FF histos limits
551
552   Int_t   fQAJetNBinsPt;    // jet QA histos bins
553   Float_t fQAJetPtMin;      // jet QA histos limits
554   Float_t fQAJetPtMax;      // jet QA histos limits
555   
556   Int_t   fQAJetNBinsEta;   // jet QA histos bins
557   Float_t fQAJetEtaMin;     // jet QA histos limits
558   Float_t fQAJetEtaMax;     // jet QA histos limits
559   
560   Int_t   fQAJetNBinsPhi;   // jet QA histos bins
561   Float_t fQAJetPhiMin;     // jet QA histos limits
562   Float_t fQAJetPhiMax;     // jet QA histos limits
563   
564   Int_t   fQATrackNBinsPt;  // track QA histos bins
565   Float_t fQATrackPtMin;    // track QA histos limits
566   Float_t fQATrackPtMax;    // track QA histos limits
567   
568   Int_t   fQATrackNBinsEta; // track QA histos bins
569   Float_t fQATrackEtaMin;   // track QA histos limits
570   Float_t fQATrackEtaMax;   // track QA histos limits
571   
572   Int_t   fQATrackNBinsPhi; // track QA histos bins
573   Float_t fQATrackPhiMin;   // track QA histos limits
574   Float_t fQATrackPhiMax;   // track QA histos limits
575   
576   Int_t   fIJNBinsJetPt;    // IJ histos bins
577   Float_t fIJJetPtMin;      // IJ histos limits
578   Float_t fIJJetPtMax;      // IJ histos limits
579
580   Int_t   fIJNBinsPt;       // IJ histos bins
581   Float_t fIJPtMin;         // IJ histos limits
582   Float_t fIJPtMax;         // IJ histos limits
583
584   Int_t   fIJNBinsZ;        // IJ histos bins
585   Float_t fIJZMin;          // IJ histos limits
586   Float_t fIJZMax;          // IJ histos limits
587
588   Int_t   fIJNBinsCosTheta; // IJ histos bins
589   Float_t fIJCosThetaMin;   // IJ histos limits
590   Float_t fIJCosThetaMax;   // IJ histos limits
591
592   Int_t   fIJNBinsTheta;    // IJ histos bins
593   Float_t fIJThetaMin;      // IJ histos limits
594   Float_t fIJThetaMax;      // IJ histos limits
595   
596   Int_t   fIJNBinsJt;       // IJ histos bins
597   Float_t fIJJtMin;         // IJ histos limits
598   Float_t fIJJtMax;         // IJ histos limits
599
600   Int_t   fDiJetNBinsJetInvMass; // FF dijet histos bins
601   Float_t fDiJetJetInvMassMin;   // FF dijet histos limits
602   Float_t fDiJetJetInvMassMax;   // FF dijet histos limits
603   Int_t   fDiJetNBinsJetPt;      // FF dijet histos bins
604   Float_t fDiJetJetPtMin;        // FF dijet histos limits
605   Float_t fDiJetJetPtMax;        // FF dijet histos limits
606   Int_t   fDiJetNBinsPt;         // FF dijet histos bins
607   Float_t fDiJetPtMin;           // FF dijet histos limits
608   Float_t fDiJetPtMax;           // FF dijet histos limits
609   Int_t   fDiJetNBinsXi;         // FF dijet histos bins
610   Float_t fDiJetXiMin;           // FF dijet histos limits
611   Float_t fDiJetXiMax;           // FF dijet histos limits
612   Int_t   fDiJetNBinsZ;          // FF dijet histos bins
613   Float_t fDiJetZMin;            // FF dijet histos limits
614   Float_t fDiJetZMax;            // FF dijet histos limits
615
616   Int_t   fQADiJetNBinsInvMass;  // dijet QA histos bins
617   Float_t fQADiJetInvMassMin;    // dijet QA histos limits
618   Float_t fQADiJetInvMassMax;    // dijet QA histos limits
619
620   Int_t   fQADiJetNBinsJetPt;    // dijet QA histos bins
621   Float_t fQADiJetJetPtMin;      // dijet QA histos limits
622   Float_t fQADiJetJetPtMax;      // dijet QA histos limits
623
624   Int_t   fQADiJetNBinsDeltaPhi; // dijet QA histos bins
625   Float_t fQADiJetDeltaPhiMin;   // dijet QA histos limits
626   Float_t fQADiJetDeltaPhiMax;   // dijet QA histos limits
627
628   Int_t   fQADiJetNBinsDeltaEta; // dijet QA histos bins
629   Float_t fQADiJetDeltaEtaMin;   // dijet QA histos limits
630   Float_t fQADiJetDeltaEtaMax;   // dijet QA histos limits
631
632   Int_t   fQADiJetNBinsDeltaPt;  // dijet QA histos bins
633   Float_t fQADiJetDeltaPtMin;    // dijet QA histos limits
634   Float_t fQADiJetDeltaPtMax;    // dijet QA histos limits
635
636   // Histograms
637   TList *fCommonHistList;         // List of common histos
638   
639   TH1F  *fh1EvtSelection;         //! event cuts 
640   TH1F  *fh1VertexNContributors;  //! NContributors to prim vertex
641   TH1F  *fh1VertexZ;              //! prim vertex z distribution
642   TH1F  *fh1EvtMult;              //! number of reconstructed tracks after cuts 
643   TH1F  *fh1nRecJetsCuts;         //! number of jets from reconstructed tracks per event 
644   TH1F  *fh1nGenJets;             //! number of jets from generated tracks per event
645   TH1F  *fh1nRecEffJets;          //! number of jets for reconstruction eff per event
646
647   // tracking efficiency 
648
649   THnSparseF *fhnSingleTrackRecEffHisto; //! track reconstruction efficiency 
650   THnSparseF *fhnJetTrackRecEffHisto;    //! reconstruction efficiency jet tracks 
651
652   ClassDef(AliAnalysisTaskFragmentationFunction, 4);
653 };
654
655 #endif