Added the possibility to the user to manage own PID
[u/mrichter/AliRoot.git] / PWGPP / pid / AliAnalysisTaskK0sBayes.h
CommitLineData
a8ad4709 1#ifndef ALIANALYSISTASKK0SBAYES_H
2#define ALIANALYSISTASKK0SBayes_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"
520899fb 16#include "AliPIDperfCut.h"
a8ad4709 17
18class TH1F;
19class TH2F;
20class AliESDtrackCuts;
21
22class AliAnalysisTaskK0sBayes : public AliAnalysisTaskSE {
23 public:
24 AliAnalysisTaskK0sBayes();
25 AliAnalysisTaskK0sBayes(const char *name);
26
27 virtual ~AliAnalysisTaskK0sBayes();
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
9c668341 51 void SetTypeCollisions(Int_t type){fTypeCol = type;}; // 0=pp, 1=pPb, 2=PbPb
52 Int_t GetTypeCollisions() const {return fTypeCol;};
53
520899fb 54 void SetPIDuserCut(AliPIDperfCut *usercut){fPIDuserCut = usercut;};
55
a8ad4709 56 private:
57 AliAnalysisTaskK0sBayes(const AliAnalysisTaskK0sBayes &old);
58 AliAnalysisTaskK0sBayes& operator=(const AliAnalysisTaskK0sBayes &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 void SelectK0s();
66 virtual Int_t IsChannelValid(Float_t etaAbs);
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 fPtKs; //! variable to fill the tree
92 Float_t fPhiKs; //! variable to fill the tree
93 Float_t fEtaKs; //! 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 fPtKsMin[nPtBin];// ptmin bin
111 static Float_t fPtKsMax[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
520899fb 129 AliPIDperfContainer* fContUser; //! results for positive user cut
130 AliPIDperfContainer* fContUser2; //! results for negative user cut
a8ad4709 131
132 Int_t fNK0s; //! number of K0s in my private selection
133 Float_t fPhiK0s[1000]; //! phi of K0s in my private selection
134 Float_t fPtK0s[1000];//! pt of K0s in my private selection
135 Int_t fNpiPos; //! number of positive pions for K0s selection
136 Int_t fNpiNeg; //! number of negative pions for K0s selection
137 Int_t fIPiPos[1000]; //! position in the AOD stack of positive pions candidates
138 Int_t fIPiNeg[1000]; //! position in the AOD stack of negative pions candidates
139 Int_t fIpiP[1000]; //! position in the AOD stack of positive pions for K0s
140 Int_t fIpiN[1000]; //! position in the AOD stack of negative pions for K0s
141 Float_t fMassKs[1000]; //! K0s mass
142
143 TH1F *fHmismTOF; //! TOF mismatch distribution
144 TH1D *fHchannelTOFdistr; //! TOF channel distance w.r.t. IP
145
9c668341 146 Int_t fTypeCol; // type of collision system (0=pp, 1=pPb, 2=PbPb)
520899fb 147 AliPIDperfCut *fPIDuserCut; // pid user cut to be cheked
9c668341 148
520899fb 149 ClassDef(AliAnalysisTaskK0sBayes, 3); //Analysis task for bayesian (K0s)
a8ad4709 150};
151
152#endif