]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWGHF/hfe/AliAnalysisTaskEMCalHFEpA.h
Update EMCal pPb
[u/mrichter/AliRoot.git] / PWGHF / hfe / AliAnalysisTaskEMCalHFEpA.h
1 #ifndef AliAnalysisTaskEMCalHFEpA_cxx
2 #define AliAnalysisTaskEMCalHFEpA_cxx
3
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6
7         ////////////////////////////////////////////////////////////////////////
8         //                                                                    //
9         //      Task for Heavy-flavour electron analysis in pPb collisions    //
10         //      (+ Electron-Hadron Jetlike Azimuthal Correlation)             //
11         //                                                                                                                                        //
12         //              version: July 28, 2014.                                                               //
13         //                                                                    //
14         //          Authors                                                                               //
15         //              Elienos Pereira de Oliveira Filho (epereira@cern.ch)          //
16         //              Cristiane Jahnke                (cristiane.jahnke@cern.ch)                    //
17         //                                                                    //
18         ////////////////////////////////////////////////////////////////////////
19
20 class TH1F;
21 class TH2F;
22 class AliESDEvent;
23 class AliESDtrackCuts;
24 class AliESDtrack;
25 class AliHFEcontainer;
26 class AliHFEcuts;
27 class AliHFEpid;
28 class AliHFEpidQAmanager;
29 class AliCFManager;
30 class AliPIDResponse;
31 class AliCentrality;
32 class AliAODEvent;
33 class AliVEvent;
34 class AliAODMCHeader;
35 class AliSelectNonHFE;
36 class AliEventPoolManager;
37 class AliEventPool;
38 class TObjArray;
39         //Lucile
40 class AliCaloTrackAODReader;
41 class AliCaloTrackReader;
42         //exotic
43 class AliEMCALRecoUtils;
44 class AliAODReader;
45 class AliCalorimeterUtils;
46
47         // --- ROOT system ---
48 #include <TObject.h> 
49 #include <TString.h>
50 #include <TObjArray.h>
51 class TArrayF;  
52 #include <TH2I.h>
53 #include <TGeoMatrix.h>
54
55         //--- ANALYSIS system ---
56 class AliVEvent;
57 class AliVTrack;
58 class AliAODPWG4Particle;
59 class AliAODCaloCluster;
60 class AliVCaloCells;
61 class AliPHOSGeoUtils;
62 class AliEMCALGeometry;
63 #include "AliEMCALRecoUtils.h"
64
65
66         //______________________________________________________________________
67         //Library
68 #include "AliAnalysisTaskSE.h"
69 #include "AliHFEpid.h"
70 #include "AliLog.h"
71         //______________________________________________________________________
72
73         //______________________________________________________________________
74 class AliAnalysisTaskEMCalHFEpA : public AliAnalysisTaskSE 
75 {
76                 //______________________________________________________________________
77 public:
78         AliAnalysisTaskEMCalHFEpA();
79         AliAnalysisTaskEMCalHFEpA(const char *name);
80         virtual ~AliAnalysisTaskEMCalHFEpA();
81         
82         virtual void   UserCreateOutputObjects();
83         virtual void   UserExec(Option_t *option);
84         virtual void   Terminate(Option_t *);
85         
86                 //Setters
87         void SetAssHadronPtRange(Double_t AssHadronPtMin, Double_t AssHadronPtMax) {fAssHadronPtMin = AssHadronPtMin; fAssHadronPtMax = AssHadronPtMax; };
88         void SetHFECuts(AliHFEcuts * const cuts) {fCuts = cuts;};
89         void SetRejectKinkMother(Bool_t rejectKinkMother = kFALSE) {fRejectKinkMother = rejectKinkMother;};
90         void SetCorrelationAnalysis(Bool_t CorrelationFlag=kTRUE) {fCorrelationFlag = CorrelationFlag;};
91         void SetMCanalysis() {fIsMC = kTRUE;};
92         void SetCentrality(Double_t CentralityMin, Double_t CentralityMax) { fCentralityMin = CentralityMin; fCentralityMax = CentralityMax; fHasCentralitySelection = kTRUE; };
93         void SetAODanalysis(Bool_t IsAOD) {fIsAOD = IsAOD;};
94         void SetEventMixing(Bool_t EventMixingFlag) { fEventMixingFlag = EventMixingFlag;};
95         void SetNonHFEmassCut(Double_t MassCut) { fMassCut = MassCut; fMassCutFlag = kTRUE;};
96         void SetEtaCut(Double_t EtaCutMin,Double_t EtaCutMax ) { fEtaCutMin = EtaCutMin; fEtaCutMax = EtaCutMax; };
97         
98         void SetdPhidEtaCut(Double_t dPhiCut, Double_t dEtaCut ) { fdPhiCut = dPhiCut;fdEtaCut = dEtaCut ;};
99         
100         void SetEoverPCut(Double_t EoverPCutMin,Double_t EoverPCutMax ) { fEoverPCutMin = EoverPCutMin; fEoverPCutMax = EoverPCutMax; };
101         
102         void SetM02Cut(Double_t M02CutMin,Double_t M02CutMax ) { fM02CutMin = M02CutMin; fM02CutMax = M02CutMax; };
103         void SetM20Cut(Double_t M20CutMin,Double_t M20CutMax ) { fM20CutMin = M20CutMin; fM20CutMax = M20CutMax; };
104         
105         
106         void SetNonHFEangleCut(Double_t AngleCut) { fAngleCut = AngleCut; fAngleCutFlag = kTRUE;};
107         void SetNonHFEchi2Cut(Double_t Chi2Cut) { fChi2Cut = Chi2Cut; fChi2CutFlag = kTRUE;};
108         void SetNonHFEdcaCut(Double_t DCAcut) { fDCAcut = DCAcut; fDCAcutFlag = kTRUE;};
109         
110         //DCA cut main particle
111         void SetdcaCut(Double_t DCAcutr, Double_t DCAcutz) { fDCAcutr = DCAcutr; fDCAcutz = DCAcutz;};
112         
113         void SetUseEMCal() { fUseEMCal=kTRUE;};
114         void SetUseTrigger() { fUseTrigger=kTRUE;};
115         void SetUseShowerShapeCut(Bool_t UseShowerShapeCut=kFALSE) { fUseShowerShapeCut=UseShowerShapeCut;};
116         void SetBackground(Bool_t FillBackground=kFALSE) { fFillBackground=FillBackground;};
117         void SetEMCalTriggerEG1() { fEMCEG1=kTRUE; };
118         void SetEMCalTriggerEG2() { fEMCEG2=kTRUE; };
119         void SetCentralityEstimator(Int_t Estimator) { fEstimator=Estimator; }; //0 = V0A, 1 = Other
120         void SetAdditionalCuts(Double_t PtMinAsso, Int_t TpcNclsAsso) {fPtMinAsso = PtMinAsso; fTpcNclsAsso = TpcNclsAsso;};
121         void SetSPDCutForHadrons() {fAssocWithSPD = kTRUE;};
122         
123                 //Getters
124         AliHFEpid *GetPID() const {return fPID;};
125                 //bad channel
126                 //AliEMCALGeometry * GetEMCALGeometry()              const { return fEMCALGeo; }
127                 //AliCalorimeterUtils * GetCaloUtils()               const { return fCaloUtils; }
128         /*AliCalorimeterUtils * GetCaloUtils()                                { if(!fCaloUtils) fCaloUtils = new AliCalorimeterUtils(); 
129                 return fCaloUtils      ; }*/
130
131                 //______________________________________________________________________
132         
133                 //______________________________________________________________________
134 private:
135         
136                 //Function to process track cuts
137         Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track);
138                 //Function to process eh analysis
139         void ElectronHadronCorrelation(AliVTrack *track, Int_t trackIndex, AliVParticle *vtrack);
140                 //Function to find non-HFE and fill histos
141         void Background(AliVTrack *track, Int_t trackIndex, AliVParticle *vtrack, Bool_t IsTPConly);
142                 //Selected Hadrons, for mixed event analysis
143         TObjArray* SelectedHadrons();
144                 //DiHadron Correlation Background
145         void DiHadronCorrelation(AliVTrack *track, Int_t trackIndex);
146                 //Find Mothers (Finde HFE and NonHFE from MC information)
147         Bool_t FindMother(Int_t mcIndex);
148         Bool_t ContainsBadChannel(TString calorimeter,UShort_t* cellList, Int_t nCells);
149         TArrayI GetTriggerPatches(Bool_t IsEventEMCALL0, Bool_t IsEventEMCALL1);
150         Double_t CalculateWeight(Int_t pdg_particle, Double_t x);
151         
152                 //Flags for specifics analysis
153         Bool_t                          fCorrelationFlag;
154         Bool_t                          fIsMC;
155         Bool_t                          fUseEMCal;
156         Bool_t                          fUseTrigger;
157         Bool_t                          fUseShowerShapeCut;
158         Bool_t                          fFillBackground;
159         Bool_t                          fAssocWithSPD;
160         
161         
162         Bool_t                          fEMCEG1;
163         Bool_t                          fEMCEG2;
164         
165                 //Used in the function FindMother
166         Bool_t                          fIsHFE1;
167         Bool_t                          fIsHFE2;
168         Bool_t                          fIsNonHFE;
169         Bool_t                          fIsFromD;
170         Bool_t                          fIsFromB;
171         Bool_t                          fIsFromPi0;
172         Bool_t                          fIsFromEta;
173         Bool_t                          fIsFromGamma;
174         
175                 //General variables
176         AliESDEvent                     *fESD;
177         AliAODEvent                     *fAOD;                          /// new
178         AliVEvent                       *fVevent;                       /// new
179         AliESDtrackCuts         *fPartnerCuts;
180         TList                           *fOutputList;
181         AliPIDResponse                  *fPidResponse;
182         AliSelectNonHFE                 *fNonHFE;
183         
184                 //For the case of AOD analysis
185         Bool_t                                  fIsAOD;                                 //flag for AOD analysis
186         
187                 //For Centrality Selection
188         AliCentrality                   *fCentrality;
189         Double_t                                fCentralityMin;
190         Double_t                                fCentralityMax;
191         Bool_t                                  fHasCentralitySelection;
192         TH1F                                    *fCentralityHist;
193         TH1F                                    *fCentralityHistPass;
194         Float_t                                 fZvtx;  
195         Int_t                                   fEstimator;
196         
197                 //EMCal
198         
199         AliVCluster                             *fClus;
200                 //AliESDCaloCluster             *fClusESD;
201         
202                 //Histograms
203         TH1F                            *fNevent;
204         TH1F                            *fNevent2;
205         TH1F                            *fPtElec_Inc;
206         
207         TH1F                            *fPtPrim;
208         TH1F                            *fPtSec;
209         TH1F                            *fPtPrim2;
210         TH1F                            *fPtSec2;
211         
212         
213         TH1F                            *fCharge_n;
214         TH1F                            *fCharge_p;
215         
216         TH2D                            *fTime;
217         TH2D                            *fTime2;
218         TH2D                            *ftimingEle;
219         TH2D                            *ftimingEle2;   
220         
221         TH1F                            *fPtElec_ULS;
222         TH1F                            *fPtElec_LS;
223         
224         TH1F                            *fPtElec_ULS_NoPid;
225         TH1F                            *fPtElec_LS_NoPid;
226         
227         TH1F                            *fPtElec_ULS_MC;
228         TH1F                            *fPtElec_ULS_MC_weight;
229         
230         TH1F                            *fPtElec_ULS2;
231         TH1F                            *fPtElec_LS2;
232         
233         TH1F                            *fPtElec_ULS_weight;
234         TH1F                            *fPtElec_LS_weight;
235         TH1F                            *fPtElec_ULS2_weight;
236         TH1F                            *fPtElec_LS2_weight;
237         
238                 //PID Histograms
239         
240         TH2F                            *fTOF01;
241         TH2F                            *fTOF02;
242         TH2F                            *fTOF03;
243         TH1F                            *fpid;          
244         TH2F                            **fEoverP_pt;
245         TH2F                            **fEoverP_tpc;
246         TH2F                            **fEoverP_tpc_p_trigger;
247         TH2F                            **fEoverP_tpc_pt_trigger;
248         TH1F                            **fTPC_pt;
249         TH2F                            **fTPC_p;
250         TH1F                            **fTPCnsigma_pt;
251         TH2F                            **fTPCnsigma_p;
252         
253         TH2F                            *fTPCnsigma_p_TPC;
254         TH2F                            *fTPCnsigma_p_TPC_on_EMCal_acc;
255         TH2F                            *fTPCnsigma_p_TPC_EoverP_cut;
256         
257         TH2F                            *fTPCnsigma_pt_2D;
258         TH2F                            *fShowerShapeCut;
259         TH2F                            *fShowerShapeM02_EoverP;
260         TH2F                            *fShowerShapeM20_EoverP;
261         TH2F                            *fShowerShape_ha;
262         TH2F                            *fShowerShape_ele;
263         TH2F                            *fTPCnsigma_eta;
264         TH2F                            *fTPCnsigma_phi;
265         TH1F                            **fECluster;
266         TH1F                            *fECluster_pure;
267         TH1F                            *fECluster_not_exotic;
268         TH1F                            *fECluster_not_exotic1;
269         TH1F                            *fECluster_not_exotic2;
270         TH1F                            *fECluster_exotic;
271         TH1F                            *fNCluster_pure;
272         TH1F                            *fNCluster_pure_aod;
273         TH2F                            *fNCluster_ECluster;
274         TH2F                            *fNcells_energy;
275         TH2F                            *fNcells_energy_elec_selected;
276         TH2F                            *fNcells_energy_not_exotic;
277         TH2F                            **fEtaPhi;
278         TH2F                            *fEtaPhi_num;
279         TH2F                            *fEtaPhi_den;
280         TH2F                            *fEtaPhi_data;
281         TH2F                            *fpt_reco_pt_MC_num;
282         TH2F                            *fpt_reco_pt_MC_den;
283         TH1F                            **fVtxZ;
284         TH1F                            *fVtxZ_new1;
285         TH1F                            *fVtxZ_new2;
286         TH1F                            *fVtxZ_new3;
287         TH1F                            *fVtxZ_new4;
288         
289         TH1F                    *fzRes1;
290         TH1F                    *fzRes2;
291         TH1F                    *fSPD_track_vtx1;
292         TH1F                        *fSPD_track_vtx2;
293         
294         TH1F                            **fEtad;
295         TH1F                            **fNTracks;
296         TH1F                            *fTrack_Multi;
297         TH2F                            **fNTracks_pt;
298         TH2F                            **fNTracks_eta;
299         TH2F                            **fNTracks_phi;
300         TH1F                            **fNClusters;
301         TH2F                            **fTPCNcls_EoverP;
302         TH2F                            **fTPCNcls_pid;
303         TH1F                            **fEta;
304         TH1F                            **fPhi;
305         TH1F                            **fR;
306         TH2F                            **fR_EoverP;
307         TH1F                            **fNcells;
308         TH2F                            **fNcells_EoverP;
309         TH1F                            **fNcells_electrons;
310         TH1F                            **fNcells_hadrons;
311         TH1F                            **fECluster_ptbins;
312         TH1F                            **fEoverP_ptbins;
313         TH1F                            **fEoverP_wSSCut;
314         TH2F                            **fM02_EoverP;
315         TH2F                            **fM20_EoverP;
316         TH2F                            **fTPCnsigma_eta_electrons;
317         TH2F                            **fTPCnsigma_eta_hadrons;
318         TH2F                            *fEoverP_pt_pions;
319         TH2F                            *ftpc_p_EoverPcut;
320         TH2F                            *fnsigma_p_EoverPcut;
321         TH2F                            *fEoverP_pt_pions2;
322         TH2F                            *fNcells_pt;
323         TH2F                            *fEoverP_pt_hadrons;
324                 //Electron-Hadron Correlation Histograms
325         TH2F                            **fCEtaPhi_Inc;
326         TH2F                            **fCEtaPhi_ULS;
327         TH2F                            **fCEtaPhi_LS;
328         TH2F                            **fCEtaPhi_ULS_NoP;
329         TH2F                            **fCEtaPhi_LS_NoP;
330         TH2F                            **fCEtaPhi_ULS_Weight;
331         TH2F                            **fCEtaPhi_LS_Weight;
332         TH2F                            **fCEtaPhi_ULS_NoP_Weight;
333         TH2F                            **fCEtaPhi_LS_NoP_Weight;
334         TH1F                            *fInvMass;
335         TH1F                            *fInvMassBack;
336         TH1F                            *fDCA;
337         TH1F                            *fDCABack;
338         TH1F                            *fOpAngle;
339         TH1F                            *fOpAngleBack;
340         TH1F                            *fInvMass2;
341         TH1F                            *fInvMassBack2;
342         TH1F                            *fDCA2;
343         TH1F                            *fDCABack2;
344         TH1F                            *fOpAngle2;
345         TH1F                            *fOpAngleBack2;
346         Double_t                        fMassCut;
347         Double_t                        fEtaCutMin;
348         Double_t                        fEtaCutMax;
349         Double_t                        fdPhiCut;
350         Double_t                        fdEtaCut;
351         Double_t                        fEoverPCutMin;
352         Double_t                        fEoverPCutMax;
353         Double_t                        fM20CutMin;
354         Double_t                        fM20CutMax;
355         Double_t                        fM02CutMin;
356         Double_t                        fM02CutMax;
357         Double_t                        fAngleCut;
358         Double_t                        fChi2Cut;
359         Double_t                        fDCAcut;
360         Double_t                        fDCAcutr;
361         Double_t                        fDCAcutz;
362         
363         Bool_t                          fMassCutFlag;
364         Bool_t                          fAngleCutFlag;
365         Bool_t                          fChi2CutFlag;
366         Bool_t                          fDCAcutFlag;
367         //Correlation Function
368         Double_t                        fAssHadronPtMin;
369         Double_t                        fAssHadronPtMax;
370         //Non-HFE reconstruction efficiency
371         TH1F                            *fPtBackgroundBeforeReco;
372         TH1F                            *fPtBackgroundBeforeReco2;
373         TH1F                            *fPtBackgroundBeforeReco_weight;
374         TH1F                            *fPtBackgroundBeforeReco2_weight;
375         TH2F                            *fpT_m_electron;
376         TH2F                            *fpT_gm_electron;
377         TH1F                            *fPtBackgroundAfterReco;
378         Double_t                        fPtMinAsso;
379         Int_t                   fTpcNclsAsso;
380                 //Tracking Efficiency
381         TH1F                            *fPtMCparticleAll;
382         TH1F                            *fPtMCparticleAll_nonPrimary;
383         TH1F                            *fPtMCparticleAlle_nonPrimary;
384         TH1F                            *fPtMCparticleAlle_Primary;
385         TH1F                            *fPtMCparticleReco;
386         TH1F                            *fPtMCparticleReco_nonPrimary;
387         TH1F                            *fPtMCparticleAllHfe1;
388         TH1F                            *fPtMCparticleRecoHfe1;
389         TH1F                            *fPtMCparticleAllHfe2;
390         TH1F                            *fPtMCparticleRecoHfe2;
391         TH1F                            *fPtMCelectronAfterAll;
392         TH1F                            *fPtMCelectronAfterAll_unfolding;
393         TH1F                            *fPtMCelectronAfterAll_nonPrimary;
394         TH1F                            *fPtMCelectronAfterAll_Primary;
395         
396         TH1F                            *fPtMCpi0;
397         TH1F                            *fPtMCeta;
398         TH1F                            *fPtMCpi02;
399         TH1F                            *fPtMCeta2;
400         TH1F                            *fPtMCpi03;
401         TH1F                            *fPtMCeta3;
402         
403         TH1F                            *fPtMC_EMCal_All;
404         TH1F                            *fPtMC_EMCal_Selected;
405         TH1F                            *fPtMC_TPC_All;
406         TH1F                            *fPtMC_TPC_Selected;
407         TH1F                            *fPt_track_match_den;
408         TH1F                            *fPt_track_match_num;
409         
410                 
411         TH1F                            *fPtMCWithLabel;
412         TH1F                            *fPtMCWithoutLabel;
413         TH1F                            *fPtIsPhysicaPrimary;
414         
415                 //For the HFE package
416         AliHFEcuts                      *fCuts;                                 // Cut Collection for HFE
417                                                                                                                 //Lucile
418                                                                                                                 //AliCaloTrackAODReader                         *reader; 
419         AliCFManager            *fCFM;                                  // Correction Framework Manager
420         AliHFEpid                       *fPID;                                  // PID
421         AliHFEpidQAmanager      *fPIDqa;                                                // PID QA manager
422         
423                 //Others
424         AliStack                        *fMCstack;                                              //
425         Bool_t              fRejectKinkMother;                          //
426         TParticle                       *fMCtrack;
427         TParticle                       *fMCtrackMother;
428         TParticle                       *fMCtrackGMother;
429         TParticle                       *fMCtrackGGMother;
430         TParticle                       *fMCtrackGGGMother;
431         TClonesArray            *fMCarray;
432         AliAODMCHeader          *fMCheader;
433         AliAODMCParticle        *fMCparticle;
434         AliAODMCParticle        *fMCparticle2;
435         AliAODMCParticle        *fMCparticleMother;
436         AliAODMCParticle        *fMCparticleGMother;
437         AliAODMCParticle        *fMCparticleGGMother;
438         AliAODMCParticle        *fMCparticleGGGMother;
439         AliMCEventHandler       *fEventHandler;
440         AliMCEvent                      *fMCevent;
441         
442                 //______________________________________________________________________
443                 //Mixed event analysis
444         AliEventPoolManager *fPoolMgr;
445         AliEventPool            *fPool;
446         TObjArray                       *fTracksClone;
447         TObjArray                       *fTracks;
448         
449         TH2F                            **fCEtaPhi_Inc_EM;
450         
451         TH2F                            **fCEtaPhi_ULS_EM;
452         TH2F                            **fCEtaPhi_LS_EM;
453         
454         TH2F                            **fCEtaPhi_ULS_Weight_EM;
455         TH2F                            **fCEtaPhi_LS_Weight_EM;
456         
457         TH1F                            *fPoolNevents;
458         
459         Bool_t                          fEventMixingFlag;
460                 //______________________________________________________________________
461         
462                 //______________________________________________________________________
463                 //Di-hadron correlation
464         TH2F                            **fCEtaPhi_Inc_DiHadron;
465         TH1F                            *fPtTrigger_Inc;
466         
467                 //AliEMCALRecoUtils             *fEMCALRecoUtils;   // EMCAL Reco Utils //exotic
468                                                                                                 //AliEMCALGeometry *fEMCALGeo ;             //! EMCAL geometry pointer
469                                                                                                 //AliCalorimeterUtils *fCaloUtils;
470         
471         Int_t            fBitEGA;                    // Trigger bit on VCaloTrigger for EGA
472
473          
474                 //______________________________________________________________________
475         
476         AliAnalysisTaskEMCalHFEpA(const AliAnalysisTaskEMCalHFEpA&);                    // not implemented
477         AliAnalysisTaskEMCalHFEpA& operator=(const AliAnalysisTaskEMCalHFEpA&);                 // not implemented
478         
479         ClassDef(AliAnalysisTaskEMCalHFEpA, 1);                                                                 // example of analysis
480                                                                                                                                                         //______________________________________________________________________
481 };
482
483         ///_________________________________________________________________________________________________
484         ///Class copied from : $ALICE_ROOT/PWGCF/Correlations/DPhi/AliAnalysisTaskLongRangeCorrelations.h
485         ///Author: Christoph Mayer
486 class AliEHCParticle : public TObject {
487 public:
488         AliEHCParticle(Double_t eta=0, Double_t phi=0, Double_t pt=0)
489     : fEta(eta), fPhi(phi), fPt(pt) {}
490         virtual ~AliEHCParticle() {}
491         
492         Double_t Eta() const { return fEta; }
493         Double_t Phi() const { return fPhi; }
494         Double_t Pt() const { return fPt; }
495         
496 protected:
497 private:
498         AliEHCParticle(const AliEHCParticle&);
499         AliEHCParticle& operator=(const AliEHCParticle&);
500         
501         Double_t fEta;
502         Double_t fPhi;
503         Double_t fPt;
504         
505         ClassDef(AliEHCParticle, 1);
506 } ;
507         ///_________________________________________________________________________________________________
508
509 #endif