update
[u/mrichter/AliRoot.git] / PWGHF / hfe / AliAnalysisTaskHFECal.h
CommitLineData
bfc7c23b 1/**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3 * *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
6 * *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
15
16#ifndef ALIANALYSISTASKHFECAL_H
17#define ALIANALYSISTASKHFECAL_H
18
19class THnSparse;
20class TH2F;
21class TLorentzVector;
22
23class AliEMCALTrack;
24class AliMagF;
25class AliESDEvent;
26class AliAODEvent;
27class AliEMCALGeometry;
28class AliEMCALRecoUtils;
29class AliAnalysisFilter;
30class AliESDtrackCuts;
31class AliESDtrack;
32class AliHFEcontainer;
33class AliHFEcuts;
34class AliHFEpid;
35class AliHFEpidQAmanager;
36class AliCFManager;
37
38#include "AliAnalysisTaskSE.h"
8806ce6c 39#include "AliStack.h"
bfc7c23b 40
41class AliAnalysisTaskHFECal : public AliAnalysisTaskSE {
42 public:
43 AliAnalysisTaskHFECal();
44 AliAnalysisTaskHFECal(const char *name);
45 virtual ~AliAnalysisTaskHFECal();
46
47 virtual void UserCreateOutputObjects();
48 virtual void UserExec(Option_t *option);
49 virtual void Terminate(Option_t *);
50
51 void SetHFECuts(AliHFEcuts * const cuts) { fCuts = cuts; };
52 void SetOpeningAngleCut (Double_t openingAngle) {fOpeningAngleCut = openingAngle;};
53 void SetInvariantMassCut (Double_t invmass) {fInvmassCut = invmass;};
85985bb0 54 void SetQAHist (int qahist) {fqahist = qahist;};
bfc7c23b 55 AliHFEpid *GetPID() const { return fPID; }
56 void SetRejectKinkMother(Bool_t rejectKinkMother = kFALSE) { fRejectKinkMother = rejectKinkMother; };
2b4460a6 57 void SelectPhotonicElectron(Int_t itrack, Double_t cent, AliESDtrack *track, Bool_t &fFlagPhotonicElec, Bool_t &fFlagConvinatElec, Double_t nSig, Double_t shower, Double_t ep, Double_t mce, Double_t w, Int_t ibgevent, Bool_t tagpi0, Bool_t tageta);
46305ed6 58 void FindMother(TParticle* part, int &label, int &pid);
8806ce6c 59 double GetMCweight(double mcPi0pT);
93c189c5 60 double GetMCweightEta(double mcEtapT);
fb87d707 61 void FindTriggerClusters();
bfc7c23b 62 private:
63
64 Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track);
65
66 AliESDEvent *fESD; //!ESD object
e7d87aef 67 AliMCEvent *fMC; //!MC object
46305ed6 68 AliStack *stack; //!MC object
fb87d707 69 AliEMCALGeometry *fGeom; // emcal geometry
bfc7c23b 70
71 TList *fOutputList; //! output list
85985bb0 72 Int_t fqahist;
73
bfc7c23b 74 AliESDtrackCuts *fTrackCuts; //! ESD track cuts
75 AliHFEcuts *fCuts; //! Cut Collection
76 Bool_t fIdentifiedAsOutInz; //Out Of Range in z
77 Bool_t fPassTheEventCut; //Pass The Event Cut
78 Bool_t fRejectKinkMother; //Reject Kink Mother
e7d87aef 79 Bool_t fmcData;
bfc7c23b 80 Double_t fVz; //z position of the primary vertex
81 AliCFManager *fCFM; //! Correction Framework Manager
82 AliHFEpid *fPID; //! PID
83 AliHFEpidQAmanager *fPIDqa; //! PID QA manager
84 Double_t fOpeningAngleCut; //openingAngle cut value
85 Double_t fInvmassCut; //invariant mass cut value
fb87d707 86
87 int ftriggers[48][60];//!
88 int ftriggersCut[48][60];//!
89 int ftriggersTime[48][60];//!
90
e7d87aef 91
bfc7c23b 92 TH1F *fNoEvents; //! no of events
93 THnSparseD *fEMCAccE; //! EMC acc
94 TH1F *fTrkpt; //! track pt
95 TH2F *fTrkEovPBef; //! track E/p before HFE pid
96 TH2F *fTrkEovPAft; //! track E/p after HFE pid
97 TH2F *fdEdxBef; //! track dEdx vs p before HFE pid
98 TH2F *fdEdxAft; //! track dEdx vs p after HFE pid
99 TH2F *fIncpT; //! HFE pid electron vs centrality
fb87d707 100 TH2F *fIncpTM20; //! HFE pid electron vs centrality
f09766dd 101 THnSparseD *fInvmassLS; //! Inv mass of LS (e,e)
102 THnSparseD *fInvmassULS; //! Inv mass of ULS (e,e)
8806ce6c 103 THnSparseD *fInvmassLSmc; //! Inv mass of LS (e,e)
104 THnSparseD *fInvmassULSmc; //! Inv mass of ULS (e,e)
93c189c5 105 TH2D *fInvmassLSmc0; //! Inv mass of ULS (e,e)
106 TH2D *fInvmassLSmc1; //! Inv mass of ULS (e,e)
107 TH2D *fInvmassLSmc2; //! Inv mass of ULS (e,e)
108 TH2D *fInvmassLSmc3; //! Inv mass of ULS (e,e)
109 TH2D *fInvmassULSmc0; //! Inv mass of ULS (e,e)
110 TH2D *fInvmassULSmc1; //! Inv mass of ULS (e,e)
111 TH2D *fInvmassULSmc2; //! Inv mass of ULS (e,e)
112 TH2D *fInvmassULSmc3; //! Inv mass of ULS (e,e)
bfc7c23b 113 TH1F *fOpeningAngleLS; //! opening angle for LS pairs
114 TH1F *fOpeningAngleULS; //! opening angle for ULS pairs
115 TH1F *fPhotoElecPt; //! photonic elec pt
f09766dd 116 TH2F *fPhoElecPt; //! Pho inclusive ele pt
fb87d707 117 TH2F *fPhoElecPtM20; //! Pho inclusive ele pt
f09766dd 118 TH2F *fSameElecPt; //! Same inclusive ele pt
fb87d707 119 TH2F *fSameElecPtM20; //! Same inclusive ele pt
a6df418c 120
bfc7c23b 121 TH1F *fTrackPtBefTrkCuts; //! Track pt before track cuts
122 TH1F *fTrackPtAftTrkCuts; //! Track pt after track cuts
123 TH2F *fTPCnsigma; //! TPC n sigma vs p
124
125 TH1F *fCent; //! centrality
126 THnSparseD *fEleInfo; //! EMC acc
feffe705 127 /*
fb87d707 128 //<---- trigger info
129 TH1F *fClsEBftTrigCut; //Cluster E before trigger selection
130 TH1F *fClsEAftTrigCut; //Cluster E after trigger selection
131 TH1F *fClsEAftTrigCut1; //Cluster E after trigger selection
132 TH1F *fClsEAftTrigCut2; //Cluster E after trigger selection
133 TH1F *fClsEAftTrigCut3; //Cluster E after trigger selection
134 TH1F *fClsEAftTrigCut4; //Cluster E after trigger selection
135 TH2F *fClsETime; //ClsE vs time distribution
136 TH2F *fClsETime1; //ClsE vs time distribution
137 TH1F *fTrigTimes;// trigger time
42c75692 138 TH2F *fCellCheck;// trigger time
feffe705 139 */
e7d87aef 140 //<------ MC
141 TH2F *fInputHFEMC;
feffe705 142 TH2F *fInputAlle;
e7d87aef 143 TH2F *fIncpTMChfe; //! MC HFE pid electron vs centrality
3db00c72 144 TH2F *fIncpTMChfeAll; //! MC HFE pid electron vs centrality
e7d87aef 145 TH2F *fIncpTMCM20hfe; //! MC HFE pid electron vs centrality
3db00c72 146 TH2F *fIncpTMCM20hfeAll; //! MC HFE pid electron vs centrality
60544aea 147 TH2F *fIncpTMCM20hfeCheck; //! MC HFE pid electron vs centrality
44be9e1d 148 THnSparseD *fIncpTMCpho; //! MC HFE pid electron vs centrality
149 THnSparseD *fIncpTMCM20pho; //! MC HFE pid electron vs centrality
150 THnSparseD *fPhoElecPtMC; //! Pho inclusive ele pt
151 THnSparseD *fPhoElecPtMCM20; //! Pho inclusive ele pt
152 THnSparseD *fSameElecPtMC; //! Same inclusive ele pt
153 THnSparseD *fSameElecPtMCM20; //! Same inclusive ele pt
e4b0faf2 154 THnSparseD *fIncpTMCM20pho_pi0e; //! MC HFE pid electron vs centrality
155 THnSparseD *fPhoElecPtMCM20_pi0e; //! Pho inclusive ele pt
156 THnSparseD *fSameElecPtMCM20_pi0e; //! Same inclusive ele pt
93c189c5 157 THnSparseD *fIncpTMCM20pho_eta; //! MC HFE pid electron vs centrality
158 THnSparseD *fPhoElecPtMCM20_eta; //! Pho inclusive ele pt
159 THnSparseD *fSameElecPtMCM20_eta; //! Same inclusive ele pt
d113d7cd 160 TH1D *CheckNclust;
161 TH1D *CheckNits;
162 TH2D *Hpi0pTcheck;
19325033 163 TH2D *HETApTcheck;
e4b0faf2 164 TH2D *HphopTcheck;
93c189c5 165 TH1D *fpTCheck;
50919258 166 TH2D *fMomDtoE;
d113d7cd 167
bfc7c23b 168 AliAnalysisTaskHFECal(const AliAnalysisTaskHFECal&); // not implemented
169 AliAnalysisTaskHFECal& operator=(const AliAnalysisTaskHFECal&); // not implemented
170
171 ClassDef(AliAnalysisTaskHFECal, 1); //!example of analysis
172};
173
174#endif
175
176