1 #ifndef AliAnalysisTaskEMCalHFEpA_cxx
2 #define AliAnalysisTaskEMCalHFEpA_cxx
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5 * See cxx source for full Copyright notice */
7 ////////////////////////////////////////////////////////////////////////
9 // Task for Heavy-flavour electron analysis in pPb collisions //
10 // (+ Electron-Hadron Jetlike Azimuthal Correlation) //
12 // version: January 13, 2014. //
15 // Elienos Pereira de Oliveira Filho (epereira@cern.ch) //
16 // Cristiane Jahnke (cristiane.jahnke@cern.ch) //
18 ////////////////////////////////////////////////////////////////////////
23 class AliESDtrackCuts;
25 class AliHFEcontainer;
28 class AliHFEpidQAmanager;
35 class AliSelectNonHFE;
36 class AliEventPoolManager;
40 //______________________________________________________________________
42 #include "AliAnalysisTaskSE.h"
43 #include "AliHFEpid.h"
45 //______________________________________________________________________
47 //______________________________________________________________________
48 class AliAnalysisTaskEMCalHFEpA : public AliAnalysisTaskSE
50 //______________________________________________________________________
52 AliAnalysisTaskEMCalHFEpA();
53 AliAnalysisTaskEMCalHFEpA(const char *name);
54 virtual ~AliAnalysisTaskEMCalHFEpA();
56 virtual void UserCreateOutputObjects();
57 virtual void UserExec(Option_t *option);
58 virtual void Terminate(Option_t *);
61 void SetAssHadronPtRange(Double_t AssHadronPtMin, Double_t AssHadronPtMax) {fAssHadronPtMin = AssHadronPtMin; fAssHadronPtMax = AssHadronPtMax; };
62 void SetHFECuts(AliHFEcuts * const cuts) {fCuts = cuts;};
63 void SetRejectKinkMother(Bool_t rejectKinkMother = kFALSE) {fRejectKinkMother = rejectKinkMother;};
64 void SetCorrelationAnalysis(Bool_t CorrelationFlag=kTRUE) {fCorrelationFlag = CorrelationFlag;};
65 void SetMCanalysis() {fIsMC = kTRUE;};
66 void SetCentrality(Double_t CentralityMin, Double_t CentralityMax) { fCentralityMin = CentralityMin; fCentralityMax = CentralityMax; fHasCentralitySelection = kTRUE; };
67 void SetAODanalysis(Bool_t IsAOD) {fIsAOD = IsAOD;};
68 void SetEventMixing(Bool_t EventMixingFlag) { fEventMixingFlag = EventMixingFlag;};
69 void SetNonHFEmassCut(Double_t MassCut) { fMassCut = MassCut; fMassCutFlag = kTRUE;};
70 void SetEtaCut(Double_t EtaCutMin,Double_t EtaCutMax ) { fEtaCutMin = EtaCutMin; fEtaCutMax = EtaCutMax; };
72 void SetdPhidEtaCut(Double_t dPhiCut, Double_t dEtaCut ) { fdPhiCut = dPhiCut;fdEtaCut = dEtaCut ;};
74 void SetEoverPCut(Double_t EoverPCutMin,Double_t EoverPCutMax ) { fEoverPCutMin = EoverPCutMin; fEoverPCutMax = EoverPCutMax; };
76 void SetM02Cut(Double_t M02CutMin,Double_t M02CutMax ) { fM02CutMin = M02CutMin; fM02CutMax = M02CutMax; };
77 void SetM20Cut(Double_t M20CutMin,Double_t M20CutMax ) { fM20CutMin = M20CutMin; fM20CutMax = M20CutMax; };
80 void SetNonHFEangleCut(Double_t AngleCut) { fAngleCut = AngleCut; fAngleCutFlag = kTRUE;};
81 void SetNonHFEchi2Cut(Double_t Chi2Cut) { fChi2Cut = Chi2Cut; fChi2CutFlag = kTRUE;};
82 void SetNonHFEdcaCut(Double_t DCAcut) { fDCAcut = DCAcut; fDCAcutFlag = kTRUE;};
83 void SetUseEMCal() { fUseEMCal=kTRUE;};
84 void SetUseShowerShapeCut(Bool_t UseShowerShapeCut=kFALSE) { fUseShowerShapeCut=UseShowerShapeCut;};
85 void SetBackground(Bool_t FillBackground=kFALSE) { fFillBackground=FillBackground;};
86 void SetEMCalTriggerEG1() { fEMCEG1=kTRUE; };
87 void SetEMCalTriggerEG2() { fEMCEG2=kTRUE; };
88 void SetCentralityEstimator(Int_t Estimator) { fEstimator=Estimator; }; //0 = V0A, 1 = Other
89 void SetAdditionalCuts(Double_t PtMinAsso, Int_t TpcNclsAsso) {fPtMinAsso = PtMinAsso; fTpcNclsAsso = TpcNclsAsso;};
90 void SetSPDCutForHadrons() {fAssocWithSPD = kTRUE;};
93 AliHFEpid *GetPID() const {return fPID;};
94 //______________________________________________________________________
96 //______________________________________________________________________
99 //Function to process track cuts
100 Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track);
101 //Function to process eh analysis
102 void ElectronHadronCorrelation(AliVTrack *track, Int_t trackIndex, AliVParticle *vtrack);
103 //Function to find non-HFE and fill histos
104 void Background(AliVTrack *track, Int_t trackIndex, AliVParticle *vtrack, Bool_t IsTPConly);
105 //Selected Hadrons, for mixed event analysis
106 TObjArray* SelectedHadrons();
107 //DiHadron Correlation Background
108 void DiHadronCorrelation(AliVTrack *track, Int_t trackIndex);
109 //Find Mothers (Finde HFE and NonHFE from MC information)
110 Bool_t FindMother(Int_t mcIndex);
112 //Flags for specifics analysis
113 Bool_t fCorrelationFlag;
116 Bool_t fUseShowerShapeCut;
117 Bool_t fFillBackground;
118 Bool_t fAssocWithSPD;
124 //Used in the function FindMother
136 AliAODEvent *fAOD; /// new
137 AliVEvent *fVevent; /// new
138 AliESDtrackCuts *fPartnerCuts;
140 AliPIDResponse *fPidResponse;
141 AliSelectNonHFE *fNonHFE;
143 //For the case of AOD analysis
144 Bool_t fIsAOD; //flag for AOD analysis
146 //For Centrality Selection
147 AliCentrality *fCentrality;
148 Double_t fCentralityMin;
149 Double_t fCentralityMax;
150 Bool_t fHasCentralitySelection;
151 TH1F *fCentralityHist;
152 TH1F *fCentralityHistPass;
159 //AliESDCaloCluster *fClusESD;
179 TH1F *fPtElec_ULS_weight;
180 TH1F *fPtElec_LS_weight;
181 TH1F *fPtElec_ULS2_weight;
182 TH1F *fPtElec_LS2_weight;
198 TH1F **fTPCnsigma_pt;
200 TH2F *fTPCnsigma_pt_2D;
201 TH2F *fShowerShapeCut;
202 TH2F *fShowerShapeM02_EoverP;
203 TH2F *fShowerShapeM20_EoverP;
204 TH2F *fShowerShape_ha;
205 TH2F *fShowerShape_ele;
207 TH2F *fTPCnsigma_eta;
208 TH2F *fTPCnsigma_phi;
219 TH2F **fTPCNcls_EoverP;
227 TH2F **fNcells_EoverP;
228 TH1F **fNcells_electrons;
229 TH1F **fNcells_hadrons;
230 TH1F **fECluster_ptbins;
231 TH1F **fEoverP_ptbins;
232 TH1F **fEoverP_wSSCut;
235 TH2F **fTPCnsigma_eta_electrons;
236 TH2F **fTPCnsigma_eta_hadrons;
238 TH2F *fEoverP_pt_pions;
240 TH2F *ftpc_p_EoverPcut;
241 TH2F *fnsigma_p_EoverPcut;
243 TH2F *fEoverP_pt_pions2;
245 TH2F *fEoverP_pt_hadrons;
247 //Electron-Hadron Correlation Histograms
252 TH2F **fCEtaPhi_ULS_NoP;
253 TH2F **fCEtaPhi_LS_NoP;
255 TH2F **fCEtaPhi_ULS_Weight;
256 TH2F **fCEtaPhi_LS_Weight;
257 TH2F **fCEtaPhi_ULS_NoP_Weight;
258 TH2F **fCEtaPhi_LS_NoP_Weight;
281 Double_t fEoverPCutMin;
282 Double_t fEoverPCutMax;
292 Bool_t fAngleCutFlag;
296 //Correlation Function
297 Double_t fAssHadronPtMin;
298 Double_t fAssHadronPtMax;
300 //Non-HFE reconstruction efficiency
301 TH1F *fPtBackgroundBeforeReco;
302 TH1F *fPtBackgroundBeforeReco2;
303 TH1F *fPtBackgroundBeforeReco_weight;
304 TH1F *fPtBackgroundBeforeReco2_weight;
306 TH2F *fpT_m_electron;
307 TH2F *fpT_gm_electron;
309 TH1F *fPtBackgroundAfterReco;
314 //Tracking Efficiency
315 TH1F *fPtMCparticleAll;
316 TH1F *fPtMCparticleAll_nonPrimary;
317 TH1F *fPtMCparticleAlle_nonPrimary;
318 TH1F *fPtMCparticleAlle_Primary;
320 TH1F *fPtMCparticleReco;
321 TH1F *fPtMCparticleReco_nonPrimary;
322 TH1F *fPtMCparticleAllHfe1;
323 TH1F *fPtMCparticleRecoHfe1;
324 TH1F *fPtMCparticleAllHfe2;
325 TH1F *fPtMCparticleRecoHfe2;
326 TH1F *fPtMCelectronAfterAll;
327 TH1F *fPtMCelectronAfterAll_nonPrimary;
328 TH1F *fPtMCelectronAfterAll_Primary;
333 TH1F *fPtMC_EMCal_All;
334 TH1F *fPtMC_EMCal_Selected;
336 TH1F *fPtMC_TPC_Selected;
338 TH1F *fPtMCWithLabel;
339 TH1F *fPtMCWithoutLabel;
340 TH1F *fPtIsPhysicaPrimary;
342 //For the HFE package
343 AliHFEcuts *fCuts; // Cut Collection for HFE
344 AliCFManager *fCFM; // Correction Framework Manager
345 AliHFEpid *fPID; // PID
346 AliHFEpidQAmanager *fPIDqa; // PID QA manager
349 AliStack *fMCstack; //
350 Bool_t fRejectKinkMother; //
352 TParticle *fMCtrackMother;
353 TParticle *fMCtrackGMother;
354 TParticle *fMCtrackGGMother;
355 TParticle *fMCtrackGGGMother;
356 TClonesArray *fMCarray;
357 AliAODMCHeader *fMCheader;
358 AliAODMCParticle *fMCparticle;
359 AliAODMCParticle *fMCparticleMother;
360 AliAODMCParticle *fMCparticleGMother;
361 AliAODMCParticle *fMCparticleGGMother;
362 AliAODMCParticle *fMCparticleGGGMother;
363 AliMCEventHandler *fEventHandler;
364 AliMCEvent *fMCevent;
366 //______________________________________________________________________
367 //Mixed event analysis
368 AliEventPoolManager *fPoolMgr;
370 TObjArray *fTracksClone;
373 TH2F **fCEtaPhi_Inc_EM;
375 TH2F **fCEtaPhi_ULS_EM;
376 TH2F **fCEtaPhi_LS_EM;
378 TH2F **fCEtaPhi_ULS_Weight_EM;
379 TH2F **fCEtaPhi_LS_Weight_EM;
383 Bool_t fEventMixingFlag;
384 //______________________________________________________________________
386 //______________________________________________________________________
387 //Di-hadron correlation
388 TH2F **fCEtaPhi_Inc_DiHadron;
389 TH1F *fPtTrigger_Inc;
390 //______________________________________________________________________
392 AliAnalysisTaskEMCalHFEpA(const AliAnalysisTaskEMCalHFEpA&); // not implemented
393 AliAnalysisTaskEMCalHFEpA& operator=(const AliAnalysisTaskEMCalHFEpA&); // not implemented
395 ClassDef(AliAnalysisTaskEMCalHFEpA, 1); // example of analysis
396 //______________________________________________________________________
399 ///_________________________________________________________________________________________________
400 ///Class copied from : $ALICE_ROOT/PWGCF/Correlations/DPhi/AliAnalysisTaskLongRangeCorrelations.h
401 ///Author: Christoph Mayer
402 class AliEHCParticle : public TObject {
404 AliEHCParticle(Double_t eta=0, Double_t phi=0, Double_t pt=0)
405 : fEta(eta), fPhi(phi), fPt(pt) {}
406 virtual ~AliEHCParticle() {}
408 Double_t Eta() const { return fEta; }
409 Double_t Phi() const { return fPhi; }
410 Double_t Pt() const { return fPt; }
414 AliEHCParticle(const AliEHCParticle&);
415 AliEHCParticle& operator=(const AliEHCParticle&);
421 ClassDef(AliEHCParticle, 1);
423 ///_________________________________________________________________________________________________