Added the possibility to the user to manage own PID
[u/mrichter/AliRoot.git] / PWGPP / pid / AliAnalysisTaskLambdaBayes.h
1 #ifndef ALIANALYSISTASKLAMBDABAYES_H
2 #define ALIANALYSISTASKLAMBDABayes_H
3
4 // ROOT includes
5 #include <TObject.h>
6 #include <TClonesArray.h>
7 #include <TList.h>
8 #include <TProfile.h>
9 #include "AliPIDCombined.h"
10 #include "TF1.h"
11
12 // AliRoot includes
13 #include <AliAnalysisTaskSE.h>
14 #include <AliAODEvent.h>
15 #include "AliPIDperfContainer.h"
16 #include "AliPIDperfCut.h"
17
18 class TH1F;
19 class TH2F;
20 class AliESDtrackCuts;
21
22 class AliAnalysisTaskLambdaBayes : public AliAnalysisTaskSE {
23  public:
24   AliAnalysisTaskLambdaBayes();
25   AliAnalysisTaskLambdaBayes(const char *name);
26
27   virtual ~AliAnalysisTaskLambdaBayes();
28
29   virtual void   UserCreateOutputObjects();
30   virtual void   UserExec(Option_t *option);
31   virtual void   Terminate(Option_t *); 
32   
33   Double_t GetVtxCut() { return fVtxCut; }   
34   Double_t GetEtaCut() { return fEtaCut; }     
35   Double_t GetMinPt() { return fMinPt; }   
36
37   virtual void  SetVtxCut(Double_t vtxCut){fVtxCut = vtxCut;}
38   virtual void  SetEtaCut(Double_t etaCut){fEtaCut = etaCut;}
39   virtual void  SetMinPt(Double_t value) {fMinPt = value;}   
40
41   virtual void SetMC(Bool_t flag = kTRUE){fIsMC = flag;};
42   virtual void SetQA(Bool_t flag = kTRUE){fQAsw = flag;};
43
44   void SetTPCclusterN(Int_t ncl){fNcluster=ncl;};
45
46   static const Int_t nPtBin = 13;  //! # pt ks bins
47
48   void SetFilterBit(Int_t fb){fFilterBit=fb;};
49   Int_t GetFilterBit() const {return fFilterBit;};
50
51   void SetTypeCollisions(Int_t type){fTypeCol = type;}; // 0=pp, 1=pPb, 2=PbPb
52   Int_t GetTypeCollisions() const {return fTypeCol;}; 
53
54   void SetPIDuserCut(AliPIDperfCut *usercut){fPIDuserCut = usercut;};
55
56  private:
57   AliAnalysisTaskLambdaBayes(const AliAnalysisTaskLambdaBayes &old); 
58   AliAnalysisTaskLambdaBayes& operator=(const AliAnalysisTaskLambdaBayes &source); 
59
60   Int_t PassesAODCuts(AliAODv0 *myV0, AliAODEvent *tAOD,Int_t specie);
61   Int_t FindDaugheterIndex(AliAODTrack *trk);
62
63   virtual Float_t GetVertex(AliAODEvent* aod) const;
64   virtual void Analyze(AliAODEvent* aodEvent); 
65   virtual Int_t IsChannelValid(Float_t etaAbs);
66   virtual void SelectLambda();
67
68   Double_t     fVtxCut;             // Vtx cut on z position in cm
69   Double_t     fEtaCut;             // Eta cut used to select particles
70   Double_t     fMinPt;              // Min pt - for histogram limits
71
72   Bool_t fIsMC; // if MC
73   Bool_t fQAsw;   // if QA
74
75   static const Int_t nCentrBin = 9; //! # cenrality bins
76
77   //
78   // Cuts and options
79   //
80
81   Int_t fNcluster;           // Numer of TPC cluster required
82   Int_t fFilterBit;          // filter bit to be used
83   TList *fList;              //! List for output objects
84   TList *fList2;             //! List for output objects
85   TList *fList3;             //! List for output objects
86
87   //
88   Float_t fCentrality;  //! current centrality for the tree
89   Float_t fPsi;         //! current Psi from TPC
90
91   Float_t fPtLambdaC;   //! variable to fill the tree
92   Float_t fPhiLambdaC;  //! variable to fill the tree
93   Float_t fEtaLambda;  //! variable to fill the tree
94   Float_t fMassV0; //! variable to fill the tree
95   Float_t fPtKp;  //! variable to fill the tree
96   Float_t fPhiKp; //! variable to fill the tree
97   Float_t fEtaKp; //! variable to fill the tree
98   Float_t fPtKn;  //! variable to fill the tree
99   Float_t fPhiKn; //! variable to fill the tree
100   Float_t fEtaKn; //! variable to fill the tree
101   Int_t fPidKp;   //! variable to fill the tree
102   Int_t fPidKn;   //! variable to fill the tree
103
104   TH2F *hMatching[4]; //! matching (all, pi, k ,p)
105   TH2F *hTracking[4]; //! tracking (all, pi, k ,p)
106  
107   TH2F *fTOFTPCsignal;   //! histo with tof signal
108   TH1F *fCombsignal;   //! histo with tof signal
109
110   static Float_t fPtLambdaMin[nPtBin];// ptmin bin
111   static Float_t fPtLambdaMax[nPtBin];// ptmax bin
112
113   // QA plots
114   TH2F *fPiTPC[nCentrBin];//! TPC dE/dx plot
115   TH2F *fKaTPC[nCentrBin];//! TPC dE/dx plot
116   TH2F *fPrTPC[nCentrBin];//! TPC dE/dx plot
117   TH2F *fElTPC[nCentrBin];//! TPC dE/dx plot
118
119   TH2F *fPiTOF[nCentrBin];//! TPC dE/dx plot
120   TH2F *fKaTOF[nCentrBin];//! TPC dE/dx plot
121   TH2F *fPrTOF[nCentrBin];//! TPC dE/dx plot
122   TH2F *fElTOF[nCentrBin];//! TPC dE/dx plot
123
124   AliESDtrackCuts *fCutsDaughter;
125
126   AliPIDCombined *fPIDCombined;  //! PID combined object
127   AliPIDperfContainer* fContPid;      //! results for positive
128   AliPIDperfContainer* fContPid2;     //! results for negative
129   AliPIDperfContainer* fContUser;     //! results for positive user cut
130   AliPIDperfContainer* fContUser2;    //! results for negative user cut
131
132   Int_t fNLambda; //! number of Lambda in my private selection
133   Float_t fPhiLambda[1000]; //! phi of Lambda in my private selection
134   Float_t fPtLambda[1000];//! pt of Lambda in my private selection
135   Int_t fNpPos; //! number of positive pions for Lambda selection
136   Int_t fNpNeg; //! number of negative pions for Lambda selection
137   Int_t fIPPos[1000]; //! position in the AOD stack of positive pions candidates
138   Int_t fIPNeg[1000]; //! position in the AOD stack of negative pions candidates
139   Int_t fIpP[1000]; //! position in the AOD stack of positive pions for Lambda
140   Int_t fIpN[1000]; //! position in the AOD stack of negative pions for Lambda
141   Float_t fMassLambda[1000]; //! Lambda mass
142   
143   TH1F *fHmismTOF; //! TOF mismatch distribution
144   TH1D *fHchannelTOFdistr; //! TOF channel distance w.r.t. IP
145
146   Int_t fTypeCol; // type of collision system (0=pp, 1=pPb, 2=PbPb)
147   AliPIDperfCut *fPIDuserCut;  // pid user cut to be cheked
148
149   ClassDef(AliAnalysisTaskLambdaBayes, 3);    //Analysis task for Bayesian (Lambda)
150 };
151
152 #endif