1 #ifndef AliAnalysisTaskEmcalJetHadEPpid_h
2 #define AliAnalysisTaskEmcalJetHadEPpid_h
15 class AliEventPoolManager;
16 class AliLocalRhoParameter;
22 class AliEMCALGeometry;
23 class AliEMCALRecoUtils;
26 // this whole section of includes added
27 #include <AliAnalysisTaskEmcalJet.h>
28 #include <AliEmcalJet.h>
29 #include <AliVEvent.h>
30 #include <AliVTrack.h>
31 #include <AliVCluster.h>
32 #include <TClonesArray.h>
37 #include "AliAnalysisTaskLocalRho.h"
38 #include "AliLocalRhoParameter.h"
41 #include "AliPIDResponse.h"
43 #include "AliAnalysisFilter.h"
45 class AliAnalysisTaskEmcalJetHadEPpid : public AliAnalysisTaskEmcalJet {
47 AliAnalysisTaskEmcalJetHadEPpid();
48 AliAnalysisTaskEmcalJetHadEPpid(const char *name);
49 //virtual ~AliAnalysisTaskEmcalJetHadEPpid() {}
50 virtual ~AliAnalysisTaskEmcalJetHadEPpid();
52 virtual void UserCreateOutputObjects();
53 virtual THnSparse* NewTHnSparseF(const char* name, UInt_t entries);
54 virtual void GetDimParams(Int_t iEntry,TString &label, Int_t &nbins, Double_t &xmin, Double_t &xmax);
55 virtual THnSparse* NewTHnSparseFPID(const char* name, UInt_t entries);
56 virtual void GetDimParamsPID(Int_t iEntry,TString &label, Int_t &nbins, Double_t &xmin, Double_t &xmax);
57 void SetPlotGlobalRho(Bool_t g) { doPlotGlobalRho = g; } // plot global rho switch
58 void SetVariableBinning(Bool_t v) { doVariableBinning = v; } // do variable binning switch
59 void SetdoPID(Bool_t p) { doPID = p; } // do PID switch
60 void SetvarbinTHnSparse(Bool_t vb) { dovarbinTHnSparse = vb; }
61 void SetDataType(Bool_t data) { useAOD = data; }
64 TString GetLocalRhoName() const {return fLocalRhoName; }
66 // set names of some objects
67 virtual void SetLocalRhoName(const char *ln) {fLocalRhoName = ln; }
68 virtual void SetTracksName(const char *tn) {fTracksName = tn; }
69 virtual void SetJetsName(const char *jn) {fJetsName = jn; }
72 virtual void SetAreaCut(Double_t a) { fAreacut = a; }
73 virtual void SetTrkBias(Double_t b) { fTrkBias = b; } //require a track with pt > b in jet
74 virtual void SetClusBias(Double_t b) { fClusBias = b; } //require a cluster with pt > b in jet
75 virtual void SetTrkEta(Double_t e) { fTrkEta = e; } //eta range of the associated tracks
76 virtual void SetJetPtcut(Double_t jpt) { fJetPtcut = jpt; } // jet pt cut
78 // eta and phi limits of jets
79 virtual void SetJetEta(Double_t emin, Double_t emax) { fEtamin = emin; fEtamax = emax; }
80 virtual void SetJetPhi(Double_t pmin, Double_t pmax) { fPhimin = pmin; fPhimax = pmax; }
82 // event mixing setters
83 virtual void SetEventMixing(Int_t yesno) { fDoEventMixing=yesno; }
84 virtual void SetMixingTracks(Int_t tracks) { fMixingTracks = tracks; }
90 virtual void Terminate(Option_t *);
91 virtual Int_t GetCentBin(Double_t cent) const; // centrality bin of event
92 Float_t RelativePhi(Double_t mphi,Double_t vphi) const; // relative jet track angle
93 Float_t RelativeEPJET(Double_t jetAng, Double_t EPAng) const; // relative jet event plane angle
94 virtual Int_t GetEtaBin(Double_t eta) const; // eta bins
95 virtual Int_t GetpTjetBin(Double_t pt) const; // jet pt bins
96 virtual Int_t GetpTtrackBin(Double_t pt) const; // track pt bins
97 virtual Int_t GetzVertexBin(Double_t zVtx) const; // zVertex bin
98 void SetfHistPIDcounterLabels(TH1* fHistPID) const; // PID counter
100 // parameters of detector to cut on for event
101 Double_t fPhimin; // phi min
102 Double_t fPhimax; // phi max
103 Double_t fEtamin; // eta min
104 Double_t fEtamax; // eta max
105 Double_t fAreacut; // area cut
106 Double_t fTrkBias; // track bias
107 Double_t fClusBias; // cluster bias
108 Double_t fTrkEta; // eta min/max of tracks
109 Double_t fJetPtcut; // jet pt to cut on for correlations
112 Int_t fDoEventMixing;
115 // switches for plots
116 Bool_t doPlotGlobalRho;
117 Bool_t doVariableBinning;
118 Bool_t dovarbinTHnSparse;
127 Double_t fLocalRhoVal;
136 // boolean functions for PID
137 Bool_t isPItpc, isKtpc, isPtpc;
140 Bool_t isPIits, isKits, isPits;
143 Bool_t isPItof, isKtof, isPtof;
147 TObjArray* CloneAndReduceTrackList(TObjArray* tracks);
148 AliEventPoolManager *fPoolMgr; // event pool Manager object
151 AliPIDResponse *fPIDResponse; // PID response object
152 AliTPCPIDResponse *fTPCResponse; // TPC pid response object
156 AliESDEvent *fESD; //ESD object
157 AliAODEvent *fAOD; //AOD object
160 TH2F *fHistITSsignal;
161 // TH2F *fHistTOFsignal;
163 TH2F *fHistRhovsCent; //!
164 TH2F *fHistNjetvsCent;//! number of jets versus Centrality
165 TH2F *fHistJetPtvsTrackPt[6];//!
166 TH2F *fHistRawJetPtvsTrackPt[6];//!
167 TH1F *fHistTrackPt[6];//!
168 TH1F *fHistEP0[6];//!
169 TH1F *fHistEP0A[6];//!
170 TH1F *fHistEP0C[6];//!
171 TH2F *fHistEPAvsC[6];//!
172 TH2F *fHistJetPtvsdEP[6];//!
173 TH2F *fHistJetPtvsdEPBias[6];//!
174 TH2F *fHistRhovsdEP[6]; //!
175 TH3F *fHistJetEtaPhiPt[6];//!
176 TH3F *fHistJetEtaPhiPtBias[6];//!
177 TH2F *fHistJetPtArea[6];//!
178 TH2F *fHistJetPtAreaBias[6];//!
179 TH2F *fHistJetPtNcon[6]; //!
180 TH2F *fHistJetPtNconBias[6]; //!
181 TH2F *fHistJetPtNconCh[6]; //!
182 TH2F *fHistJetPtNconBiasCh[6]; //!
183 TH2F *fHistJetPtNconEm[6]; //!
184 TH2F *fHistJetPtNconBiasEm[6]; //!
185 TH1F *fHistJetHaddPhiINcent[6];
186 TH1F *fHistJetHaddPhiOUTcent[6];
187 TH1F *fHistJetHaddPhiMIDcent[6];
188 // ***********************
189 TH1 *fHistCentrality;
194 TH1 *fHistJetHaddPhiIN;
195 TH1 *fHistJetHaddPhiOUT;
196 TH1 *fHistJetHaddPhiMID;
197 TH1 *fHistJetHaddPhiBias;
198 TH1 *fHistJetHaddPhiINBias;
199 TH1 *fHistJetHaddPhiOUTBias;
200 TH1 *fHistJetHaddPhiMIDBias;
202 TH1 *fHistMEdPHI; // phi distrubtion of mixed events
203 TH1 *fHistTrackPtallcent;
206 TH2 *fHistTrackEtaPhi[4][7];
207 TH1 *fHistJetHadbindPhi[9];
208 TH1 *fHistJetHadbindPhiIN[9];
209 TH1 *fHistJetHadbindPhiMID[9];
210 TH1 *fHistJetHadbindPhiOUT[9];
211 TH2 *fHistJetHEtaPhi;
214 TH1 *fHistJetPtBias[2];
215 TH1 *fHistJetPtTT[2];
216 TH2 *fHistAreavsRawPt[2];
217 TH2 *fHistJetH[2][5][3];
218 TH2 *fHistJetHBias[2][5][3];
219 TH2 *fHistJetHTT[2][5][3];
220 TH1F *fHistJetHdPHI[11];
221 TH2F *fHistJetHdETAdPHI[11];
222 TH2F *fHistSEphieta; // single events phi-eta distributions
223 TH2F *fHistMEphieta; // mixed events phi-eta distributions
224 TH1F *fHistJetHaddPHI;
226 // PID status histo's
230 THnSparse *fhnPID; // PID sparse
231 THnSparse *fhnMixedEvents; // mixed events matrix
232 THnSparse *fhnJH; // jet hadron events matrix
234 // ***********************************************************
236 //Declare it private to avoid compilation warning
237 AliAnalysisTaskEmcalJetHadEPpid(const AliAnalysisTaskEmcalJetHadEPpid & g) ; // cpy ctor
239 AliAnalysisTaskEmcalJetHadEPpid& operator=(const AliAnalysisTaskEmcalJetHadEPpid&); // not implemented
240 ClassDef(AliAnalysisTaskEmcalJetHadEPpid, 4); // Emcal jet hadron PID - Event plane dependence