Updated Fragmentation function task (added Di jet part and intra-jet radiation) ...
[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
326   virtual void   SetBranchGenJets(const char* c){fBranchGenJets = c;}
327   virtual void   SetBranchRecJets(const char* c){fBranchRecJets = c;}
328
329   virtual void   SetTrackCuts(Float_t trackPt = 0.15, Float_t trackEtaMin = -0.9, Float_t trackEtaMax = 0.9, 
330                               Float_t trackPhiMin = 0., Float_t trackPhiMax = 2*TMath::Pi())
331   {fTrackPtCut = trackPt; fTrackEtaMin = trackEtaMin; fTrackEtaMax = trackEtaMax; 
332     fTrackPhiMin = trackPhiMin; fTrackPhiMax = trackPhiMax;}
333   virtual void   SetFilterMask(UInt_t i) {fFilterMask = i;}
334   virtual void   SetJetCuts(Float_t jetPt = 5., Float_t jetEtaMin = -0.5, Float_t jetEtaMax = 0.5, 
335                             Float_t jetPhiMin = 0., Float_t jetPhiMax = 2*TMath::Pi())
336   {fJetPtCut = jetPt; fJetEtaMin = jetEtaMin; fJetEtaMax = jetEtaMax; 
337     fJetPhiMin = jetPhiMin; fJetPhiMax = jetPhiMax;}
338   virtual void   SetDiJetCuts(Int_t cutType = 1, Float_t deltaPhiCut = 0.,  
339                               Float_t cdfCut = 0.5, Float_t ptFractionCut = 0.6)
340   {fDiJetCut = cutType; fDiJetDeltaPhiCut = deltaPhiCut;  
341     fDiJetCDFCut = cdfCut; fDiJetPtFractionCut = ptFractionCut;}
342   virtual void   SetKindSlices(Int_t slice) {fDiJetKindBins = slice;}
343
344   virtual void   SetFFRadius(Float_t r = 0.4) { fFFRadius = r; }
345
346   static  void   SetProperties(TH1* h,const char* x, const char* y);
347   static  void   SetProperties(TH2* h,const char* x, const char* y,const char* z);
348   static  void   SetProperties(THnSparse* h, Int_t dim, const char** labels);
349
350   void   SetHighPtThreshold(Float_t pt = 5.) { fQATrackHighPtThreshold = pt; }
351
352
353   void   SetFFHistoBins(Int_t nJetPt = 55, Float_t jetPtMin = 5, Float_t jetPtMax = 60, 
354                         Int_t nPt = 70, Float_t ptMin = 0., Float_t ptMax = 70., 
355                         Int_t nXi = 70, Float_t xiMin = 0., Float_t xiMax = 7.,
356                         Int_t nZ = 22,  Float_t zMin = 0.,  Float_t zMax = 1.1)
357   { fFFNBinsJetPt = nJetPt; fFFJetPtMin = jetPtMin; fFFJetPtMax = jetPtMax; 
358     fFFNBinsPt = nPt; fFFPtMin = ptMin; fFFPtMax = ptMax;
359     fFFNBinsXi = nXi; fFFXiMin = xiMin; fFFXiMax = xiMax;
360     fFFNBinsZ  = nZ;  fFFZMin  = zMin;  fFFZMax  = zMax; }
361   
362   void  SetQAJetHistoBins(Int_t nPt = 200, Float_t ptMin = 0., Float_t ptMax = 200.,
363                           Int_t nEta = 20, Float_t etaMin = -1.0, Float_t etaMax = 1.0,
364                           Int_t nPhi = 60, Float_t phiMin = 0., Float_t phiMax = 2*TMath::Pi())
365     { fQAJetNBinsPt = nPt; fQAJetPtMin = ptMin; fQAJetPtMax = ptMax;
366       fQAJetNBinsEta = nEta; fQAJetEtaMin = etaMin; fQAJetEtaMax = etaMax;
367       fQAJetNBinsPhi = nPhi; fQAJetPhiMin = phiMin; fQAJetPhiMax = phiMax; }
368   
369   void  SetQATrackHistoBins(Int_t nPt = 70, Float_t ptMin = 0., Float_t ptMax = 70.,
370                             Int_t nEta = 20, Float_t etaMin = -1.0, Float_t etaMax = 1.0,
371                             Int_t nPhi = 60, Float_t phiMin = 0., Float_t phiMax = 2*TMath::Pi())
372   { fQATrackNBinsPt = nPt; fQATrackPtMin = ptMin; fQATrackPtMax = ptMax;
373     fQATrackNBinsEta = nEta; fQATrackEtaMin = etaMin; fQATrackEtaMax = etaMax;
374     fQATrackNBinsPhi = nPhi; fQATrackPhiMin = phiMin; fQATrackPhiMax = phiMax; }
375   
376   void   SetIJHistoBins(Int_t nJetPt = 55, Float_t jetPtMin = 5, Float_t jetPtMax = 60, 
377                         Int_t nPt = 70, Float_t ptMin = 0., Float_t ptMax = 70., 
378                         Int_t nZ = 22,  Float_t zMin = 0.,  Float_t zMax = 1.1,
379                         Int_t nCosTheta = 100,  Float_t costhetaMin = 0.,  Float_t costhetaMax = 1.,
380                         Int_t nTheta = 200,  Float_t thetaMin = -0.5,  Float_t thetaMax = 1.5,
381                         Int_t nJt = 25,  Float_t jtMin = 0.,  Float_t jtMax = 5.)
382   { fIJNBinsJetPt = nJetPt; fIJJetPtMin = jetPtMin; fIJJetPtMax = jetPtMax; 
383     fIJNBinsPt = nPt; fIJPtMin = ptMin; fIJPtMax = ptMax;
384     fIJNBinsZ = nZ; fIJZMin = zMin; fIJZMax = zMax;
385     fIJNBinsCosTheta  = nCosTheta;  fIJCosThetaMin  = costhetaMin;  fIJCosThetaMax  = costhetaMax;
386     fIJNBinsTheta  = nTheta;  fIJThetaMin  = thetaMin;  fIJThetaMax  = thetaMax;
387     fIJNBinsJt  = nJt;  fIJJtMin  = jtMin;  fIJJtMax  = jtMax; }
388
389   void SetDiJetHistoBins(Int_t nJetInvMass = 55, Float_t jetInvMassMin = 5, Float_t jetInvMassMax = 60, 
390                          Int_t nJetPt = 55, Float_t jetPtMin = 5, Float_t jetPtMax = 60, 
391                          Int_t nPt = 70, Float_t ptMin = 0., Float_t ptMax = 70., 
392                          Int_t nXi = 70, Float_t xiMin = 0., Float_t xiMax = 7.,
393                          Int_t nZ = 22,  Float_t zMin = 0.,  Float_t zMax = 1.1)
394   {
395     fDiJetNBinsJetInvMass = nJetInvMass; fDiJetJetInvMassMin = jetInvMassMin; fDiJetJetInvMassMax = jetInvMassMax;
396     fDiJetNBinsJetPt = nJetPt; fDiJetJetPtMin = jetPtMin; fDiJetJetPtMax = jetPtMax;
397     fDiJetNBinsPt = nPt; fDiJetPtMin = ptMin; fDiJetPtMax = ptMax;
398     fDiJetNBinsXi = nXi; fDiJetXiMin = xiMin; fDiJetXiMax = xiMax;
399     fDiJetNBinsZ = nZ; fDiJetZMin = zMin; fDiJetZMax = zMax;
400   }
401
402   void SetQADiJetHistoBins(Int_t nInvMass = 55, Float_t invMassMin = 5., Float_t invMassMax = 60.,
403                            Int_t nJetPt = 55, Float_t jetPtMin = 5, Float_t jetPtMax = 60, 
404                            Int_t nDeltaPhi = 100, Float_t deltaPhiMin = 0., Float_t deltaPhiMax = TMath::Pi(),
405                            Int_t nDeltaEta = 22, Float_t deltaEtaMin = 0., Float_t deltaEtaMax = 1.1,
406                            Int_t nDeltaPt = 100, Float_t deltaPtMin = 0., Float_t deltaPtMax = 100.)
407   {
408     fQADiJetNBinsInvMass = nInvMass; fQADiJetInvMassMin = invMassMin; fQADiJetInvMassMax = invMassMax;
409     fQADiJetNBinsJetPt = nJetPt; fQADiJetJetPtMin = jetPtMin; fQADiJetJetPtMax = jetPtMax;
410     fQADiJetNBinsDeltaPhi = nDeltaPhi; fQADiJetDeltaPhiMin = deltaPhiMin; fQADiJetDeltaPhiMax = deltaPhiMax;
411     fQADiJetNBinsDeltaEta = nDeltaEta; fQADiJetDeltaEtaMin = deltaEtaMin; fQADiJetDeltaEtaMax = deltaEtaMax;
412     fQADiJetNBinsDeltaPt = nDeltaPt; fQADiJetDeltaPtMin = deltaPtMin; fQADiJetDeltaPtMax = deltaPtMax;
413
414   }
415
416   Float_t  GetFFRadius() const { return fFFRadius; }
417   void     GetJetTracksTrackrefs(TList* l, AliAODJet* j);
418   void     GetJetTracksPointing(TList* in, TList* out, AliAODJet* j, const Double_t r, Double_t& pt);  
419   Double_t GetDiJetBin(Double_t invMass, Double_t leadingJetPt, Double_t eMean, Int_t kindSlices); // function to find which bin fill
420   Double_t InvMass(AliAODJet* jet1, AliAODJet* jet2);
421
422
423  private:
424     
425   // Consts
426   
427   enum {kTrackUndef=0, kTrackAOD, kTrackAODCuts, kTrackKineAll, kTrackKineCharged, kTrackKineChargedAcceptance, kTrackAODMCAll, kTrackAODMCCharged, kTrackAODMCChargedAcceptance};
428   enum {kJetsUndef=0, kJetsRec, kJetsRecAcceptance, kJetsGen, kJetsGenAcceptance, kJetsKine, kJetsKineAcceptance};
429   
430   
431   Int_t   GetListOfTracks(TList* list, Int_t type);
432   Int_t   GetListOfJets(TList* list, Int_t type);
433   
434   AliESDEvent* fESD;      // ESD event
435   AliAODEvent* fAOD;      // AOD event
436   AliMCEvent*  fMCEvent;  // MC event
437   
438   TString fBranchRecJets;         // branch name for reconstructed jets
439   TString fBranchGenJets;         // branch name for generated jets
440   
441   Int_t   fTrackTypeGen;  // type of generated tracks
442   Int_t   fJetTypeGen;    // type of generated jets
443
444   UInt_t  fFilterMask;    // filter bit for selected tracks
445         
446   // track cuts
447   Float_t fTrackPtCut;    // track transverse momentum cut
448   Float_t fTrackEtaMin;   // track eta cut
449   Float_t fTrackEtaMax;   // track eta cut
450   Float_t fTrackPhiMin;   // track phi cut
451   Float_t fTrackPhiMax;   // track phi cut
452   
453   // jet cuts
454   Float_t fJetPtCut;      // jet transverse momentum cut
455   Float_t fJetEtaMin;     // jet eta cut
456   Float_t fJetEtaMax;     // jet eta cut
457   Float_t fJetPhiMin;     // jet phi cut
458   Float_t fJetPhiMax;     // jet phi cut
459
460   // di-jet cuts
461   Int_t   fDiJetCut;            // dijet cut selection
462   Float_t fDiJetDeltaPhiCut;    // delta phi cut value
463   Float_t fDiJetPtFractionCut;  // fraction of pt cut value
464   Float_t fDiJetCDFCut;         // cdf cut value
465
466   Int_t   fDiJetKindBins;       // type of bins: invmass, etleading, emean
467   static  TArrayD* fDiJetBins;   // bining in invmass, etleading, emean
468
469   Float_t fFFRadius;        // if radius > 0 construct FF from tracks within cone around jet axis, otherwise use trackRefs  
470   
471   TList* fTracksRec;      //! reconstructed tracks
472   TList* fTracksRecCuts;  //! reconstructed tracks after cuts
473   TList* fTracksGen;      //! generated tracks 
474   
475   TList* fJetsRec;        //! jets from reconstructed tracks
476   TList* fJetsRecCuts;    //! jets from reonstructed tracks after jet cuts 
477   TList* fJetsGen;        //! jets from generated tracks
478   
479   
480   AliFragFuncQATrackHistos* fQATrackHistosRec;      //! track QA: reconstructed tracks
481   AliFragFuncQATrackHistos* fQATrackHistosRecCuts;  //! track QA: reconstructed tracks after cuts
482   AliFragFuncQATrackHistos* fQATrackHistosGen;      //! track QA: generated tracks
483   
484   AliFragFuncQAJetHistos*  fQAJetHistosRec;             //! jet QA: jets from reconstructed tracks
485   AliFragFuncQAJetHistos*  fQAJetHistosRecCuts;         //! jet QA: jets from reconstructed tracks after jet cuts 
486   AliFragFuncQAJetHistos*  fQAJetHistosRecCutsLeading;  //! jet QA: leading jet from reconstructed tracks after jet cuts 
487   AliFragFuncQAJetHistos*  fQAJetHistosGen;             //! jet QA: jets from generated tracks  
488   AliFragFuncQAJetHistos*  fQAJetHistosGenLeading;      //! jet QA: leading jet from generated tracks  
489   
490   AliFragFuncHistos*  fFFHistosRecCuts;         //! FF reconstructed tracks after cuts 
491   AliFragFuncHistos*  fFFHistosRecLeading;      //! FF reconstructed tracks after cuts: all reconstructed tracks pt / leading track pt  
492   AliFragFuncHistos*  fFFHistosRecLeadingTrack; //! FF reconstructed tracks after cuts: leading track pt / jet pt
493   AliFragFuncHistos*  fFFHistosGen;             //! FF generated tracks after cuts 
494   AliFragFuncHistos*  fFFHistosGenLeading;      //! FF generated tracks after cuts: all generated tracks pt / leading track pt  
495   AliFragFuncHistos*  fFFHistosGenLeadingTrack; //! FF generated tracks after cuts: leading track pt / jet pt
496
497
498   AliFragFuncIntraJetHistos*  fIJHistosRecCuts;         //! IJ reconstructed tracks after cuts 
499   AliFragFuncIntraJetHistos*  fIJHistosRecLeading;      //! IJ reconstructed tracks after cuts: all reconstructed tracks pt / leading track pt  
500   AliFragFuncIntraJetHistos*  fIJHistosRecLeadingTrack; //! IJ reconstructed tracks after cuts: leading track pt / jet pt
501   AliFragFuncIntraJetHistos*  fIJHistosGen;             //! IJ generated tracks after cuts 
502   AliFragFuncIntraJetHistos*  fIJHistosGenLeading;      //! IJ generated tracks after cuts: all generated tracks pt / leading track pt  
503   AliFragFuncIntraJetHistos*  fIJHistosGenLeadingTrack; //! IJ generated tracks after cuts: leading track pt / jet pt
504
505   AliFragFuncDiJetHistos* fFFDiJetHistosRecCuts;         //! DiJet FF reconstructed tracks after cuts
506   AliFragFuncDiJetHistos* fFFDiJetHistosRecLeading;      //! DiJet FF reconstructed tracks after cuts: all reconstructed tracks pt / leading track pt
507   AliFragFuncDiJetHistos* fFFDiJetHistosRecLeadingTrack; //! DiJet FF reconstructed tracks after cuts: leading track pt / jet pt
508
509   AliFragFuncDiJetHistos* fFFDiJetHistosGen;             //! DiJet FF generated tracks after cuts 
510   AliFragFuncDiJetHistos* fFFDiJetHistosGenLeading;      //! DiJet FF generated tracks after cuts: all generated tracks pt / leading track pt 
511   AliFragFuncDiJetHistos* fFFDiJetHistosGenLeadingTrack; //! DiJet FF generated tracks after cuts: leading track pt / jet pt 
512
513   AliFragFuncQADiJetHistos* fQADiJetHistosRecCuts;       //! Dijet QA : reconstructed tracks after cuts
514   AliFragFuncQADiJetHistos* fQADiJetHistosGen;           //! DiJet QA: jets from generated tracks  
515
516
517   Float_t  fQATrackHighPtThreshold;       // track QA high transverse momentum threshold
518   
519   // histogram bins  
520
521   Int_t   fFFNBinsJetPt;    // FF histos bins
522   Float_t fFFJetPtMin;      // FF histos limits
523   Float_t fFFJetPtMax;      // FF histos limits
524
525   Int_t   fFFNBinsPt;       // FF histos bins
526   Float_t fFFPtMin;         // FF histos limits
527   Float_t fFFPtMax;         // FF histos limits
528
529   Int_t   fFFNBinsXi;       // FF histos bins
530   Float_t fFFXiMin;         // FF histos limits
531   Float_t fFFXiMax;         // FF histos limits
532
533   Int_t   fFFNBinsZ;        // FF histos bins
534   Float_t fFFZMin;          // FF histos limits
535   Float_t fFFZMax;          // FF histos limits
536
537   Int_t   fQAJetNBinsPt;    // jet QA histos bins
538   Float_t fQAJetPtMin;      // jet QA histos limits
539   Float_t fQAJetPtMax;      // jet QA histos limits
540   
541   Int_t   fQAJetNBinsEta;   // jet QA histos bins
542   Float_t fQAJetEtaMin;     // jet QA histos limits
543   Float_t fQAJetEtaMax;     // jet QA histos limits
544   
545   Int_t   fQAJetNBinsPhi;   // jet QA histos bins
546   Float_t fQAJetPhiMin;     // jet QA histos limits
547   Float_t fQAJetPhiMax;     // jet QA histos limits
548   
549   Int_t   fQATrackNBinsPt;  // track QA histos bins
550   Float_t fQATrackPtMin;    // track QA histos limits
551   Float_t fQATrackPtMax;    // track QA histos limits
552   
553   Int_t   fQATrackNBinsEta; // track QA histos bins
554   Float_t fQATrackEtaMin;   // track QA histos limits
555   Float_t fQATrackEtaMax;   // track QA histos limits
556   
557   Int_t   fQATrackNBinsPhi; // track QA histos bins
558   Float_t fQATrackPhiMin;   // track QA histos limits
559   Float_t fQATrackPhiMax;   // track QA histos limits
560   
561   Int_t   fIJNBinsJetPt;    // IJ histos bins
562   Float_t fIJJetPtMin;      // IJ histos limits
563   Float_t fIJJetPtMax;      // IJ histos limits
564
565   Int_t   fIJNBinsPt;       // IJ histos bins
566   Float_t fIJPtMin;         // IJ histos limits
567   Float_t fIJPtMax;         // IJ histos limits
568
569   Int_t   fIJNBinsZ;        // IJ histos bins
570   Float_t fIJZMin;          // IJ histos limits
571   Float_t fIJZMax;          // IJ histos limits
572
573   Int_t   fIJNBinsCosTheta; // IJ histos bins
574   Float_t fIJCosThetaMin;   // IJ histos limits
575   Float_t fIJCosThetaMax;   // IJ histos limits
576
577   Int_t   fIJNBinsTheta;    // IJ histos bins
578   Float_t fIJThetaMin;      // IJ histos limits
579   Float_t fIJThetaMax;      // IJ histos limits
580   
581   Int_t   fIJNBinsJt;       // IJ histos bins
582   Float_t fIJJtMin;         // IJ histos limits
583   Float_t fIJJtMax;         // IJ histos limits
584
585   Int_t   fDiJetNBinsJetInvMass; // FF dijet histos bins
586   Float_t fDiJetJetInvMassMin;   // FF dijet histos limits
587   Float_t fDiJetJetInvMassMax;   // FF dijet histos limits
588   Int_t   fDiJetNBinsJetPt;      // FF dijet histos bins
589   Float_t fDiJetJetPtMin;        // FF dijet histos limits
590   Float_t fDiJetJetPtMax;        // FF dijet histos limits
591   Int_t   fDiJetNBinsPt;         // FF dijet histos bins
592   Float_t fDiJetPtMin;           // FF dijet histos limits
593   Float_t fDiJetPtMax;           // FF dijet histos limits
594   Int_t   fDiJetNBinsXi;         // FF dijet histos bins
595   Float_t fDiJetXiMin;           // FF dijet histos limits
596   Float_t fDiJetXiMax;           // FF dijet histos limits
597   Int_t   fDiJetNBinsZ;          // FF dijet histos bins
598   Float_t fDiJetZMin;            // FF dijet histos limits
599   Float_t fDiJetZMax;            // FF dijet histos limits
600
601   Int_t   fQADiJetNBinsInvMass;  // dijet QA histos bins
602   Float_t fQADiJetInvMassMin;    // dijet QA histos limits
603   Float_t fQADiJetInvMassMax;    // dijet QA histos limits
604
605   Int_t   fQADiJetNBinsJetPt;    // dijet QA histos bins
606   Float_t fQADiJetJetPtMin;      // dijet QA histos limits
607   Float_t fQADiJetJetPtMax;      // dijet QA histos limits
608
609   Int_t   fQADiJetNBinsDeltaPhi; // dijet QA histos bins
610   Float_t fQADiJetDeltaPhiMin;   // dijet QA histos limits
611   Float_t fQADiJetDeltaPhiMax;   // dijet QA histos limits
612
613   Int_t   fQADiJetNBinsDeltaEta; // dijet QA histos bins
614   Float_t fQADiJetDeltaEtaMin;   // dijet QA histos limits
615   Float_t fQADiJetDeltaEtaMax;   // dijet QA histos limits
616
617   Int_t   fQADiJetNBinsDeltaPt;  // dijet QA histos bins
618   Float_t fQADiJetDeltaPtMin;    // dijet QA histos limits
619   Float_t fQADiJetDeltaPtMax;    // dijet QA histos limits
620
621   // Histograms
622   TList *fCommonHistList;         // List of common histos
623   
624   TH1F  *fh1EvtSelection;         //! event cuts 
625   TH1F  *fh1VertexNContributors;  //! NContributors to prim vertex
626   TH1F  *fh1VertexZ;              //! prim vertex z distribution
627   TH1F  *fh1EvtMult;              //! number of reconstructed tracks after cuts 
628   TH1F  *fh1nRecJetsCuts;         //! number of jets from reconstructed tracks per event 
629   TH1F  *fh1nGenJets;             //! number of jets from generated tracks per event
630
631   ClassDef(AliAnalysisTaskFragmentationFunction, 3);
632 };
633
634 #endif