]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGGA/CaloTrackCorrelations/AliAnaPi0EbE.h
modification of pLam HBT train macros
[u/mrichter/AliRoot.git] / PWGGA / CaloTrackCorrelations / AliAnaPi0EbE.h
CommitLineData
477d6cee 1#ifndef ALIANAPI0EBE_H
2#define ALIANAPI0EBE_H
3/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
477d6cee 5
6//_________________________________________________________________________
7//
8// Class for the analysis of high pT pi0 event by event
09273901 9// Pi0/Eta identified by one of the following:
477d6cee 10// -Invariant mass of 2 cluster in calorimeter
11// -Shower shape analysis in calorimeter
12// -Invariant mass of one cluster in calorimeter and one photon reconstructed in TPC (in near future)
13//
14//-- Author: Gustavo Conesa (INFN-LNF) & Raphaelle Ichou (SUBATECH)
15//_________________________________________________________________________
16
17
18// --- ROOT system ---
477d6cee 19class TList ;
0c1383b5 20class TObjString;
477d6cee 21
22// --- ANALYSIS system ---
745913ae 23#include "AliAnaCaloTrackCorrBaseClass.h"
477d6cee 24
745913ae 25class AliAnaPi0EbE : public AliAnaCaloTrackCorrBaseClass {
477d6cee 26
27 public:
477d6cee 28 AliAnaPi0EbE() ; // default ctor
1db06135 29 virtual ~AliAnaPi0EbE() { ; } //virtual dtor
c5693f62 30
521636d2 31 TObjString * GetAnalysisCuts();
477d6cee 32
521636d2 33 TList * GetCreateOutputObjects();
477d6cee 34
3455f821 35 Int_t GetMCIndex(const Int_t aodTag);
36
521636d2 37 void Init();
477d6cee 38
521636d2 39 void InitParameters();
40
41 void MakeAnalysisFillAOD() ;
42
43 void MakeAnalysisFillHistograms() ;
477d6cee 44
521636d2 45 void Print(const Option_t * opt) const;
477d6cee 46
521636d2 47 // Main
477d6cee 48
40d3ce60 49 void FillPileUpHistograms(const Float_t energy, const Float_t time) ;
50
51 void FillRejectedClusterHistograms(const TLorentzVector mom, const Int_t mctag);
2ad19c3d 52
5c46c992 53 void FillSelectedClusterHistograms(AliVCluster* cluster,
54 const Int_t nLocMax,
bfdcf7fb 55 const Int_t tag,
56 const Float_t asy = 0);
42d47cb7 57
58 void FillWeightHistograms(AliVCluster *clus);
0a14e9ae 59
b5dbb99b 60 void HasPairSameMCMother(AliAODPWG4Particle * photon1,
61 AliAODPWG4Particle * photon2,
62 Int_t & label, Int_t & tag);
63
521636d2 64 void MakeInvMassInCalorimeter() ;
57b97dc6 65
521636d2 66 void MakeInvMassInCalorimeterAndCTS() ;
57b97dc6 67
521636d2 68 void MakeShowerShapeIdentification() ;
dbba06ca 69
521636d2 70 //Setters Getters
71
72 //Analysis types
73 enum anaTypes {kIMCalo, kSSCalo, kIMCaloTracks};
764ab1f4 74 anaTypes GetAnalysisType() const { return fAnaType ; }
75 void SetAnalysisType(anaTypes ana) { fAnaType = ana ; }
ddc0a8a5 76
764ab1f4 77 TString GetInputAODGammaConvName() const { return fInputAODGammaConvName ; }
78 void SetInputAODGammaConvName(TString name) { fInputAODGammaConvName = name ; }
ddc0a8a5 79
521636d2 80 //Only for pi0 SS identification case
764ab1f4 81 void SetCalorimeter(TString & det) { fCalorimeter = det ; }
521636d2 82
83 void SetMinDistanceToBadChannel(Float_t m1, Float_t m2, Float_t m3) {
e671adc2 84 fMinDist = m1; fMinDist2 = m2; fMinDist3 = m3 ; }
34c16486 85
e671adc2 86 void SetNLMCut(Int_t min, Int_t max) { fNLMCutMin = min;
87 fNLMCutMax = max ; }
88 Int_t GetNLMCutMin() const { return fNLMCutMin ; }
89 Int_t GetNLMCutMax() const { return fNLMCutMax ; }
90
4d97a954 91 void SetNLMMinEnergy(Int_t i, Float_t min) { if (i < 3 && i >=0 ) fNLMECutMin[i] = min ; }
92 Float_t GetNLMMinEnergy(Int_t i) const { if( i < 3 && i >=0 ) return fNLMECutMin[i] ; else return 0 ; }
93
94 void SetTimeCut(Double_t min, Double_t max) { fTimeCutMin = min;
764ab1f4 95 fTimeCutMax = max ; }
96 Double_t GetTimeCutMin() const { return fTimeCutMin ; }
667432ef 97 Double_t GetTimeCutMax() const { return fTimeCutMax ; }
e671adc2 98
e997f026 99 Bool_t IsTrackMatchRejectionOn() const { return fRejectTrackMatch ; }
100 void SwitchOnTrackMatchRejection() { fRejectTrackMatch = kTRUE ; }
101 void SwitchOffTrackMatchRejection() { fRejectTrackMatch = kFALSE ; }
102
2ad19c3d 103 void SwitchOnFillPileUpHistograms() { fFillPileUpHistograms = kTRUE ; }
104 void SwitchOffFillPileUpHistograms() { fFillPileUpHistograms = kFALSE ; }
105
764ab1f4 106 void SwitchOnFillWeightHistograms() { fFillWeightHistograms = kTRUE ; }
107 void SwitchOffFillWeightHistograms() { fFillWeightHistograms = kFALSE ; }
108
109 void SwitchOnTMHistoFill() { fFillTMHisto = kTRUE ; }
110 void SwitchOffTMHistoFill() { fFillTMHisto = kFALSE ; }
521636d2 111
764ab1f4 112 void SwitchOnSelectedClusterHistoFill() { fFillSelectClHisto = kTRUE ; }
113 void SwitchOffSelectedClusterHistoFill() { fFillSelectClHisto = kFALSE ; }
c5693f62 114
764ab1f4 115 void SwitchOnOnlySimpleSSHistoFill() { fFillOnlySimpleSSHisto = kTRUE ; }
116 void SwitchOffOnlySimpleHistoFill() { fFillOnlySimpleSSHisto = kFALSE ; }
09273901 117
06e81356 118
4d97a954 119
521636d2 120 //For histograms
c5693f62 121 enum mcTypes { kmcPhoton = 0, kmcConversion = 1, kmcPi0 = 2,
122 kmcEta = 3, kmcElectron = 4, kmcHadron = 5 };
521636d2 123
477d6cee 124 private:
125
c5693f62 126 anaTypes fAnaType; // Select analysis type
477d6cee 127
128 //Only for pi0 SS identification case, kSSCalo
521636d2 129 TString fCalorimeter ; // Calorimeter where the gamma is searched;
130 Float_t fMinDist ; // Minimal distance to bad channel to accept cluster
131 Float_t fMinDist2; // Cuts on Minimal distance to study acceptance evaluation
132 Float_t fMinDist3; // One more cut on distance used for acceptance-efficiency study
e671adc2 133 Int_t fNLMCutMin ; // Remove clusters/cells with number of local maxima smaller than this value
134 Int_t fNLMCutMax ; // Remove clusters/cells with number of local maxima larger than this value
129e22a5 135 Float_t fNLMECutMin[3] ; // Minimum energy of the cluster, depending on nlm.
34c16486 136 Double_t fTimeCutMin ; // Remove clusters/cells with time smaller than this value, in ns
137 Double_t fTimeCutMax ; // Remove clusters/cells with time larger than this value, in ns
e997f026 138 Bool_t fRejectTrackMatch ; // Remove clusters which have an associated TPC track
139
2ad19c3d 140 Bool_t fFillPileUpHistograms; // Fill pile-up related histograms
78a28af3 141 Bool_t fFillWeightHistograms ; // Fill weigth histograms
09273901 142 Bool_t fFillTMHisto; // Fill track matching plots
06e81356 143 Bool_t fFillSelectClHisto; // Fill selected cluster histograms
764ab1f4 144 Bool_t fFillOnlySimpleSSHisto; // Fill selected cluster histograms, selected SS histograms
09273901 145
4d97a954 146
477d6cee 147 //Only for combination of calorimeter and conversion photons, kIMCaloTracks
521636d2 148 TString fInputAODGammaConvName; // Name of AOD branch with conversion photons
477d6cee 149
150 //Histograms
521636d2 151
09273901 152 TH1F * fhPt ; //! Number of identified pi0/eta vs pT
153 TH1F * fhE ; //! Number of identified pi0/eta vs E
154 TH2F * fhEEta ; //! E vs eta of identified pi0/eta
155 TH2F * fhEPhi ; //! E vs phi of identified pi0/eta
156 TH2F * fhEtaPhi ; //! eta vs phi of identified pi0/eta
7a972c0c 157
c8710850 158 TH2F * fhPtCentrality ; //! centrality vs pi0/eta pT
159 TH2F * fhPtEventPlane ; //! event plane vs pi0/eta pT
160
40d3ce60 161 TH1F * fhPtReject ; //! Number of rejected as pi0/eta vs pT
162 TH1F * fhEReject ; //! Number of rejected as pi0/eta vs E
163 TH2F * fhEEtaReject ; //! E vs eta of rejected as pi0/eta
164 TH2F * fhEPhiReject ; //! E vs phi of rejected as pi0/eta
165 TH2F * fhEtaPhiReject ; //! eta vs phi of rejected as pi0/eta
166
f02db2c0 167 TH2F * fhMass ; //! pair mass vs E, for all pairs
4650f5cf 168 TH2F * fhAsymmetry ; //! cluster E vs asymmetry of 2 splitted clusters
f02db2c0 169 TH2F * fhSelectedMass ; //! pair mass vs E, for selected pairs
4650f5cf 170 TH2F * fhSelectedAsymmetry ; //! cluster E vs asymmetry of 2 splitted clusters, for selected pairs
cfdf2b91 171 TH1F * fhSplitE ; //! split sub-cluster pair energy sum
172 TH1F * fhSplitPt ; //! split sub-cluster pair pT sum
173
09273901 174 TH1F * fhPtDecay ; //! Number of identified pi0/eta decay photons vs pT
175 TH1F * fhEDecay ; //! Number of identified pi0/eta decay photons vs E
176
42d47cb7 177 TH2F * fhEDispersion ; //! E vs disp of selected cluster
178 TH2F * fhELambda0 ; //! E vs lambda0 of selected cluster
179 TH2F * fhELambda1 ; //! E vs lambda1 of selected cluster
180 TH2F * fhELambda0NoTRD ; //! E vs lambda0 of selected cluster, not behind TRD
181 TH2F * fhELambda0FracMaxCellCut ;//! E vs lambda0 of selected cluster, fraction of cluster energy in max cell cut
182 TH2F * fhEFracMaxCell ; //! E vs frac max cell of selected cluster
183 TH2F * fhEFracMaxCellNoTRD ; //! E vs frac max cell of selected cluster, not behind TRD
184 TH2F * fhENCells; //! E vs N cells in selected cluster
185 TH2F * fhETime; //! E vs Time of selected cluster
186 TH2F * fhEPairDiffTime; //! E vs Pair of clusters time difference vs E
5c46c992 187
34c16486 188 TH2F * fhDispEtaE ; //! shower dispersion in eta direction
189 TH2F * fhDispPhiE ; //! shower dispersion in phi direction
d2655d46 190 TH2F * fhLambda0DispEta[7] ; //! shower shape correlation l0 vs disp eta
191 TH2F * fhLambda0DispPhi[7] ; //! shower shape correlation l0 vs disp phi
34c16486 192 TH2F * fhSumEtaE ; //! shower dispersion in eta direction
193 TH2F * fhSumPhiE ; //! shower dispersion in phi direction
194 TH2F * fhSumEtaPhiE ; //! shower dispersion in eta and phi direction
195 TH2F * fhDispEtaPhiDiffE ; //! shower dispersion eta - phi
196 TH2F * fhSphericityE ; //! shower sphericity in eta vs phi
d2655d46 197 TH2F * fhDispEtaDispPhi[7] ; //! shower dispersion in eta direction vs phi direction for 5 E bins [0-2],[2-4],[4-6],[6-10],[> 10]
d2655d46 198 TH2F * fhAsymmetryLambda0[7] ; //! E asymmetry of 2 splitted clusters vs lam0 for 5 E bins
199 TH2F * fhAsymmetryDispEta[7] ; //! E asymmetry of 2 splitted clusters vs lam0 for 5 E bins
200 TH2F * fhAsymmetryDispPhi[7] ; //! E asymmetry of 2 splitted clusters vs lam0 for 5 E bins
bfdcf7fb 201
c4a7d28a 202 //MC histograms
203
3455f821 204 TH2F * fhEMCLambda0[6] ; //! E vs lambda0 of pi0 pairs but really from MC particle
205 TH2F * fhEMCLambda1[6] ; //! E vs lambda1 of pi0 pairs but really from MC particle
206 TH2F * fhEMCDispersion[6] ; //! E vs dispersion of pi0 pairs but really from MC particle
3bfcb597 207 TH2F * fhEMCLambda0NoTRD[6] ; //! E vs lambda0 of pi0 pairs but really from MC particle, not behind TRD
208 TH2F * fhEMCLambda0FracMaxCellCut[6] ;//! E vs lambda0 of pi0 pairs but really from MC particle, fraction of cluster energy in max cell cut
3455f821 209 TH2F * fhEMCFracMaxCell[6] ; //! E vs fraction of max cell
210
211 TH2F * fhMCEDispEta[6] ; //! shower dispersion in eta direction
212 TH2F * fhMCEDispPhi[6] ; //! shower dispersion in phi direction
213 TH2F * fhMCLambda0DispEta[7][6] ; //! shower shape correlation l0 vs disp eta
214 TH2F * fhMCLambda0DispPhi[7][6] ; //! shower shape correlation l0 vs disp phi
215 TH2F * fhMCESumEtaPhi[6] ; //! shower dispersion in eta vs phi direction
216 TH2F * fhMCEDispEtaPhiDiff[6] ; //! shower dispersion in eta -phi direction
217 TH2F * fhMCESphericity[6] ; //! shower sphericity, eta vs phi
218 TH2F * fhMCDispEtaDispPhi[7][6] ; //! shower dispersion in eta direction vs phi direction for 5 E bins [0-2],[2-4],[4-6],[6-10],[> 10]
bfdcf7fb 219 TH2F * fhMCEAsymmetry[6] ; //! E asymmetry of 2 splitted clusters vs cluster E
d2655d46 220 TH2F * fhMCAsymmetryLambda0[7][6] ; //! E asymmetry of 2 splitted clusters vs lam0 for 5 E bins
221 TH2F * fhMCAsymmetryDispEta[7][6] ; //! E asymmetry of 2 splitted clusters vs lam0 for 5 E bins
222 TH2F * fhMCAsymmetryDispPhi[7][6] ; //! E asymmetry of 2 splitted clusters vs lam0 for 5 E bins
34c16486 223
40d3ce60 224 TH1F * fhMCE[6]; //! Number of identified as pi0 vs E coming from X
225 TH1F * fhMCPt[6]; //! Number of identified as pi0 vs Pt coming from X
3455f821 226 TH2F * fhMCPhi[6]; //! Phi of identified as pi0, coming from X
227 TH2F * fhMCEta[6]; //! eta of identified as pi0, coming from X
40d3ce60 228 TH1F * fhMCEReject[6]; //! Number of rejected as pi0 vs E coming from X
229 TH1F * fhMCPtReject[6]; //! Number of rejected as pi0 vs Pt coming from X
3455f821 230
cfdf2b91 231 TH1F * fhMCSplitE[6]; //! Number of identified as pi0 vs sum E split coming from X
232 TH1F * fhMCSplitPt[6]; //! Number of identified as pi0 vs sum Pt split coming from X
233
234
17f5b4b6 235 TH2F * fhMCPtCentrality[6] ; //! centrality vs pi0/eta pT coming from X
236
237
883411b2 238 TH2F * fhMCPi0PtGenRecoFraction; //! SS id, clusters id as pi0 (eta), coming from 2 photon, pi0 primary, pt vs E prim pi0 / E reco
239 TH2F * fhMCEtaPtGenRecoFraction; //! SS id, clusters id as pi0 (eta), coming from 2 photon, eta primary, pt vs E prim eta / E reco
240 TH1F * fhMCPi0DecayPt; //! SS id, clusters id as pi0 (eta), coming from 1 photon, pi0 decay primary, pt
241 TH2F * fhMCPi0DecayPtFraction; //! SS id, clusters id as pi0 (eta), coming from 1 photon, pi0 decay primary, pt vs pt decay / pt mother
242 TH1F * fhMCEtaDecayPt; //! SS id, clusters id as pi0 (eta), coming from 1 photon, eta decay primary, pt
243 TH2F * fhMCEtaDecayPtFraction; //! SS id, clusters id as pi0 (eta), coming from 1 photon, eta decay primary, pt vs pt decay / pt mother
244 TH1F * fhMCOtherDecayPt; //! SS id, clusters id as pi0 (eta), coming from 1 photon, other decay primary, pt
b5dbb99b 245
3455f821 246 TH2F * fhMassPairMCPi0; //! pair mass, origin is same pi0
247 TH2F * fhMassPairMCEta; //! pair mass, origin is same eta
248 TH2F * fhAnglePairMCPi0; //! pair opening angle, origin is same pi0
249 TH2F * fhAnglePairMCEta; //! pair opening angle, origin is same eta
521636d2 250
78a28af3 251 // Weight studies
252
253 TH2F * fhECellClusterRatio; //! e cell / e cluster vs e cluster for selected photons
254 TH2F * fhECellClusterLogRatio; //! log (e cell / e cluster) vs e cluster for selected photons
255 TH2F * fhEMaxCellClusterRatio; //! e max cell / e cluster vs e cluster for selected photons
256 TH2F * fhEMaxCellClusterLogRatio;//! log (e max cell / e cluster) vs e cluster for selected photons
c5693f62 257 TH2F * fhLambda0ForW0[14]; //! L0 for 7 defined w0= 3, 3.5 ... 6 for selected photons
1a72f6c5 258 //TH2F * fhLambda1ForW0[7]; //! L1 for 7 defined w0= 3, 3.5 ... 6 for selected photons
78a28af3 259
09273901 260 // Track Matching
261 TH2F * fhTrackMatchedDEta ; //! Eta distance between track and cluster vs cluster E
262 TH2F * fhTrackMatchedDPhi ; //! Phi distance between track and cluster vs cluster E
263 TH2F * fhTrackMatchedDEtaDPhi ; //! Eta vs Phi distance between track and cluster, E cluster > 0.5 GeV
5c46c992 264 TH2F * fhTrackMatchedMCParticle; //! Trace origin of matched particle
31ae6d59 265 TH2F * fhdEdx ; //! matched track dEdx vs cluster E
b5dbb99b 266 TH2F * fhEOverP; //! matched track E cluster over P track vs cluster E
b5dbb99b 267 TH2F * fhEOverPNoTRD; //! matched track E cluster over P track vs cluster E, not behind TRD
268
5c46c992 269 // Local maxima
270 TH2F * fhNLocMax; //! number of maxima in selected clusters
34c16486 271 TH2F * fhELambda0LocMax[3] ; //! E vs lambda0 of selected cluster, 1,2,>2 local maxima in cluster
272 TH2F * fhELambda1LocMax[3] ; //! E vs lambda1 of selected cluster, 1,2,>2 local maxima in cluster
273 TH2F * fhEDispersionLocMax[3] ; //! E vs lambda1 of selected cluster, 1,2,>2 local maxima in cluster
274 TH2F * fhEDispEtaLocMax[3] ; //! E vs eta dispersion of selected cluster, 1,2,>2 local maxima in cluster
275 TH2F * fhEDispPhiLocMax[3] ; //! E vs phi dispersion of selected cluster, 1,2,>2 local maxima in cluster
276 TH2F * fhESumEtaPhiLocMax[3] ; //! E vs dispersion in eta and phi direction
277 TH2F * fhEDispEtaPhiDiffLocMax[3] ; //! E vs dispersion eta - phi
278 TH2F * fhESphericityLocMax[3] ; //! E vs sphericity in eta vs phi
bfdcf7fb 279 TH2F * fhEAsymmetryLocMax[3] ; //! E asymmetry of 2 splitted clusters vs cluster E for different NLM
280
3c1d9afb 281 TH2F * fhMassPairLocMax[8]; //! pair mass, origin is same pi0, combine clusters depending on number of maxima
5c46c992 282
2ad19c3d 283 // Pile-up
5e5e056f 284 TH1F * fhPtPi0PileUp[7]; //! pT distribution of selected pi0/eta
285 TH2F * fhTimeENoCut; //! time of cluster vs E, no cut
2ad19c3d 286 TH2F * fhTimeESPD; //! time of cluster vs E, IsSPDPileUp
287 TH2F * fhTimeESPDMulti; //! time of cluster vs E, IsSPDPileUpMulti
288 TH2F * fhTimeNPileUpVertSPD; //! time of cluster vs n pile-up vertices from SPD
289 TH2F * fhTimeNPileUpVertTrack; //! time of cluster vs n pile-up vertices from Tracks
290 TH2F * fhTimeNPileUpVertContributors; //! time of cluster vs n pile-up vertex from SPD contributors
291 TH2F * fhTimePileUpMainVertexZDistance; //! time of cluster vs difference of z main vertex and pile-up vertex
292 TH2F * fhTimePileUpMainVertexZDiamond; //! time of cluster vs difference of z diamond and pile-up vertex
293
bfdcf7fb 294 AliAnaPi0EbE( const AliAnaPi0EbE & pi0ebe) ; // cpy ctor
295 AliAnaPi0EbE & operator = (const AliAnaPi0EbE & pi0ebe) ; // cpy assignment
c5693f62 296
e997f026 297 ClassDef(AliAnaPi0EbE,25)
c4a7d28a 298} ;
477d6cee 299
300
301#endif //ALIANAPI0EBE_H
302
303
304