]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGGA/CaloTrackCorrelations/AliAnaPi0.h
modified histo list, add jet eta cut and change to transient data members
[u/mrichter/AliRoot.git] / PWGGA / CaloTrackCorrelations / AliAnaPi0.h
CommitLineData
1c5acb87 1#ifndef ALIANAPI0_H
2#define ALIANAPI0_H
3/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
1c5acb87 5
6//_________________________________________________________________________
6175da48 7// Class to fill two-photon invariant mass histograms
1c5acb87 8// to be used to extract pi0 raw yield.
6175da48 9// Input is produced by AliAnaPhoton (or any other analysis producing output AliAODPWG4Particles),
10// it will do nothing if executed alone
1c5acb87 11//
12//-- Author: Dmitri Peressounko (RRC "KI")
745913ae 13//-- Adapted to CaloTrackCorr frame by Lamia Benhabib (SUBATECH)
1c5acb87 14//-- and Gustavo Conesa (INFN-Frascati)
15
16//Root
17class TList;
0333ede6 18class TH3F ;
19class TH2F ;
0c1383b5 20class TObjString;
1c5acb87 21
22//Analysis
745913ae 23#include "AliAnaCaloTrackCorrBaseClass.h"
1c5acb87 24class AliAODEvent ;
25class AliESDEvent ;
c8fe2783 26class AliAODPWG4Particle ;
1c5acb87 27
745913ae 28class AliAnaPi0 : public AliAnaCaloTrackCorrBaseClass {
6639984f 29
78219bac 30 public:
6639984f 31 AliAnaPi0() ; // default ctor
6639984f 32 virtual ~AliAnaPi0() ;//virtual dtor
33
6175da48 34 //-------------------------------
35 // General analysis frame methods
36 //-------------------------------
37
0c1383b5 38 TObjString * GetAnalysisCuts();
0333ede6 39
0c1383b5 40 TList * GetCreateOutputObjects();
0333ede6 41
42 void Print(const Option_t * opt) const;
43
44 void MakeAnalysisFillHistograms();
45
46 void InitParameters();
6175da48 47
48 //Calorimeter options
0333ede6 49 TString GetCalorimeter() const { return fCalorimeter ; }
50 void SetCalorimeter(TString & det) { fCalorimeter = det ; }
c8fe2783 51
6175da48 52 //-------------------------------
53 // EVENT Bin Methods
54 //-------------------------------
55
2e876d85 56 Int_t GetEventIndex(AliAODPWG4Particle * part, Double_t * vert) ;
156549ae 57
6175da48 58 //-------------------------------
59 //Opening angle pair selection
60 //-------------------------------
0333ede6 61 void SwitchOnAngleSelection() { fUseAngleCut = kTRUE ; }
62 void SwitchOffAngleSelection() { fUseAngleCut = kFALSE ; }
63
64 void SwitchOnAngleEDepSelection() { fUseAngleEDepCut = kTRUE ; }
65 void SwitchOffAngleEDepSelection() { fUseAngleEDepCut = kFALSE ; }
66
67 void SetAngleCut(Float_t a) { fAngleCut = a ; }
68 void SetAngleMaxCut(Float_t a) { fAngleMaxCut = a ; }
6175da48 69
7a972c0c 70 void SwitchOnFillAngleHisto() { fFillAngleHisto = kTRUE ; }
71 void SwitchOffFillAngleHisto() { fFillAngleHisto = kFALSE ; }
72
6175da48 73 //------------------------------------------
20218aea 74 //Do analysis only with clusters in same SM or different combinations of SM
6175da48 75 //------------------------------------------
0333ede6 76 void SwitchOnSameSM() { fSameSM = kTRUE ; }
77 void SwitchOffSameSM() { fSameSM = kFALSE ; }
6175da48 78
0333ede6 79 void SwitchOnSMCombinations() { fFillSMCombinations = kTRUE ; }
80 void SwitchOffSMCombinations() { fFillSMCombinations = kFALSE ; }
20218aea 81
6175da48 82 //-------------------------------
83 //Histogram filling options off by default
84 //-------------------------------
0333ede6 85 void SwitchOnInvPtWeight() { fMakeInvPtPlots = kTRUE ; }
86 void SwitchOffInvPtWeight() { fMakeInvPtPlots = kFALSE ; }
af7b3903 87
0333ede6 88 void SwitchOnFillBadDistHisto() { fFillBadDistHisto = kTRUE ; }
89 void SwitchOffFillBadDistHisto() { fFillBadDistHisto = kFALSE ; }
6175da48 90
91 //-------------------------------------------
92 //Cuts for multiple analysis, off by default
93 //-------------------------------------------
0333ede6 94 void SwitchOnMultipleCutAnalysis() { fMultiCutAna = kTRUE ; }
95 void SwitchOffMultipleCutAnalysis() { fMultiCutAna = kFALSE ; }
5ae09196 96
0333ede6 97 void SetNPtCuts (Int_t s) { if(s <= 10)fNPtCuts = s ; }
98 void SetNAsymCuts (Int_t s) { if(s <= 10)fNAsymCuts = s ; }
99 void SetNNCellCuts(Int_t s) { if(s <= 10)fNCellNCuts = s ; }
100 void SetNPIDBits (Int_t s) { if(s <= 10)fNPIDBits = s ; }
d7c10d78 101
0333ede6 102 void SetPtCutsAt (Int_t p,Float_t v) { if(p < 10)fPtCuts[p] = v ; }
103 void SetAsymCutsAt(Int_t p,Float_t v) { if(p < 10)fAsymCuts[p] = v ; }
104 void SetNCellCutsAt(Int_t p,Int_t v) { if(p < 10)fCellNCuts[p]= v ; }
105 void SetPIDBitsAt (Int_t p,Int_t v) { if(p < 10)fPIDBits[p] = v ; }
d7c10d78 106
7a972c0c 107 void SwitchOnFillSSCombinations() { fFillSSCombinations = kTRUE ; }
108 void SwitchOffFillSSCombinations() { fFillSSCombinations = kFALSE ; }
109
110 void SwitchOnFillAsymmetryHisto() { fFillAsymmetryHisto = kTRUE ; }
111 void SwitchOffFillAsymmetryHisto() { fFillAsymmetryHisto = kFALSE ; }
112
020e681b 113 void SwitchOnFillOriginHisto() { fFillOriginHisto = kTRUE ; }
114 void SwitchOffFillOriginHisto() { fFillOriginHisto = kFALSE ; }
be894c1d 115
29555e96 116 void SwitchOnFillArmenterosThetaStarHisto() { fFillArmenterosThetaStar = kTRUE ; }
117 void SwitchOffFillArmenterosThetaStarHisto() { fFillArmenterosThetaStar = kFALSE ; }
7a972c0c 118
6175da48 119 //MC analysis related methods
0333ede6 120
121 void SwitchOnConversionChecker() { fCheckConversion = kTRUE ; }
122 void SwitchOffConversionChecker() { fCheckConversion = kFALSE ; }
123
124 void SwitchOnMultipleCutAnalysisInSimulation() { fMultiCutAnaSim = kTRUE ; }
125 void SwitchOffMultipleCutAnalysisInSimulation() { fMultiCutAnaSim = kFALSE ; }
20218aea 126
0333ede6 127 void FillAcceptanceHistograms();
b94e038e 128 void FillMCVersusRecDataHistograms(Int_t index1, Int_t index2,
129 Float_t pt1, Float_t pt2,
130 Int_t ncells1, Int_t ncells2,
131 Double_t mass, Double_t pt, Double_t asym,
132 Double_t deta, Double_t dphi);
133
134 void FillArmenterosThetaStar(Int_t pdg, TLorentzVector meson,
135 TLorentzVector daugh1, TLorentzVector daugh2);
be894c1d 136
137
6639984f 138 private:
0333ede6 139
0333ede6 140 TList ** fEventsList ; //![GetNCentrBin()*GetNZvertBin()*GetNRPBin()] Containers for photons in stored events
141
142 TString fCalorimeter ; // Select Calorimeter for IM
143 Int_t fNModules ; // Number of EMCAL/PHOS modules, set as many histogras as modules
144
145 Bool_t fUseAngleCut ; // Select pairs depending on their opening angle
146 Bool_t fUseAngleEDepCut ; // Select pairs depending on their opening angle
147 Float_t fAngleCut ; // Select pairs with opening angle larger than a threshold
148 Float_t fAngleMaxCut ; // Select pairs with opening angle smaller than a threshold
6175da48 149
150 //Multiple cuts analysis
0333ede6 151 Bool_t fMultiCutAna; // Do analysis with several or fixed cut
152 Bool_t fMultiCutAnaSim; // Do analysis with several or fixed cut, in the simulation related part
153 Int_t fNPtCuts; // Number of pt cuts
154 Float_t fPtCuts[10]; // Array with different pt cuts
155 Int_t fNAsymCuts; // Number of assymmetry cuts
156 Float_t fAsymCuts[10]; // Array with different assymetry cuts
157 Int_t fNCellNCuts; // Number of cuts with number of cells in cluster
158 Int_t fCellNCuts[10]; // Array with different cell number cluster cuts
159 Int_t fNPIDBits ; // Number of possible PID bit combinations
160 Int_t fPIDBits[10]; // Array with different PID bits
6175da48 161
162 //Switchs of different analysis options
0333ede6 163 Bool_t fMakeInvPtPlots; // D plots with inverse pt weight
164 Bool_t fSameSM; // Select only pairs in same SM;
165 Bool_t fFillSMCombinations; // Fill histograms with different cluster pairs in SM combinations
166 Bool_t fCheckConversion; // Fill histograms with tagged photons as conversion
0333ede6 167 Bool_t fFillBadDistHisto; // Do plots for different distances to bad channels
7a972c0c 168 Bool_t fFillSSCombinations; // Do invariant mass for different combination of shower shape clusters
169 Bool_t fFillAngleHisto; // Fill histograms with pair opening angle
170 Bool_t fFillAsymmetryHisto; // Fill histograms with asymmetry vs pt
020e681b 171 Bool_t fFillOriginHisto; // Fill histograms depending on their origin
29555e96 172 Bool_t fFillArmenterosThetaStar; // Fill armenteros histograms
be894c1d 173
6639984f 174 //Histograms
6175da48 175
156549ae 176 //Event characterization
0333ede6 177 TH1F * fhAverTotECluster; //! Average number of clusters in SM
178 TH1F * fhAverTotECell; //! Average number of cells in SM
179 TH2F * fhAverTotECellvsCluster; //! Average number of cells in SM
180 TH1F * fhEDensityCluster; //! Deposited energy in event per cluster
181 TH1F * fhEDensityCell; //! Deposited energy in event per cell vs cluster
182 TH2F * fhEDensityCellvsCluster; //! Deposited energy in event per cell vs cluster
156549ae 183
7a972c0c 184 TH2F ** fhReMod ; //![fNModules] REAL two-photon invariant mass distribution for different calorimeter modules.
185 TH2F ** fhReSameSideEMCALMod ; //![fNModules-2] REAL two-photon invariant mass distribution for different clusters in different calorimeter modules.
186 TH2F ** fhReSameSectorEMCALMod ; //![fNModules/2] REAL two-photon invariant mass distribution for different clusters in different calorimeter modules.
187 TH2F ** fhReDiffPHOSMod ; //![fNModules] REAL two-photon invariant mass distribution for different clusters in different calorimeter modules.
188 TH2F ** fhMiMod ; //![fNModules] MIXED two-photon invariant mass distribution for different calorimeter modules.
189 TH2F ** fhMiSameSideEMCALMod ; //![fNModules-2] REAL two-photon invariant mass distribution for different clusters in different calorimeter modules.
190 TH2F ** fhMiSameSectorEMCALMod ; //![fNModules/2] REAL two-photon invariant mass distribution for different clusters in different calorimeter modules.
191 TH2F ** fhMiDiffPHOSMod ; //![fNModules-1] REAL two-photon invariant mass distribution for different clusters in different calorimeter modules.
6175da48 192
193 // Pairs with at least one cluster tagged as conversion
7a972c0c 194 TH2F * fhReConv ; //! REAL two-photon invariant mass distribution one of the pair was 2 clusters with small mass
195 TH2F * fhMiConv ; //! MIXED two-photon invariant mass distribution one of the pair was 2 clusters with small mass
196 TH2F * fhReConv2 ; //! REAL two-photon invariant mass distribution both pair photons recombined from 2 clusters with small mass
197 TH2F * fhMiConv2 ; //! MIXED two-photon invariant mass distribution both pair photons recombined from 2 clusters with small mass
0333ede6 198
7a972c0c 199 TH2F ** fhRe1 ; //![GetNCentrBin()*fNPIDBits*fNAsymCuts] REAL two-photon invariant mass distribution for different centralities and Asymmetry
200 TH2F ** fhMi1 ; //![GetNCentrBin()*fNPIDBits*fNAsymCuts] MIXED two-photon invariant mass distribution for different centralities and Asymmetry
201 TH2F ** fhRe2 ; //![GetNCentrBin()*fNPIDBits*fNAsymCuts] REAL two-photon invariant mass distribution for different centralities and Asymmetry
202 TH2F ** fhMi2 ; //![GetNCentrBin()*fNPIDBits*fNAsymCuts] MIXED two-photon invariant mass distribution for different centralities and Asymmetry
203 TH2F ** fhRe3 ; //![GetNCentrBin()*fNPIDBits*fNAsymCuts] REAL two-photon invariant mass distribution for different centralities and Asymmetry
204 TH2F ** fhMi3 ; //![GetNCentrBin()*fNPIDBits*fNAsymCuts] MIXED two-photon invariant mass distribution for different centralities and Asymmetry
6175da48 205
6175da48 206 //Histograms weighted by inverse pT
0333ede6 207 TH2F ** fhReInvPt1 ; //![GetNCentrBin()*fNPIDBits*fNAsymCuts] REAL two-photon invariant mass distribution for different centralities and Asymmetry, inverse pT
208 TH2F ** fhMiInvPt1 ; //![GetNCentrBin()*fNPIDBits*fNAsymCuts] MIXED two-photon invariant mass distribution for different centralities and Asymmetry, inverse pT
209 TH2F ** fhReInvPt2 ; //![GetNCentrBin()*fNPIDBits*fNAsymCuts] REAL two-photon invariant mass distribution for different centralities and Asymmetry, inverse pT
210 TH2F ** fhMiInvPt2 ; //![GetNCentrBin()*fNPIDBits*fNAsymCuts] MIXED two-photon invariant mass distribution for different centralities and Asymmetry, inverse pT
211 TH2F ** fhReInvPt3 ; //![GetNCentrBin()*fNPIDBits*fNAsymCuts] REAL two-photon invariant mass distribution for different centralities and Asymmetry, inverse pT
212 TH2F ** fhMiInvPt3 ; //![GetNCentrBin()*fNPIDBits*fNAsymCuts] MIXED two-photon invariant mass distribution for different centralities and Asymmetry, inverse pT
6175da48 213
214 //Multiple cuts: Assymmetry, pt, n cells, PID
0333ede6 215 TH2F ** fhRePtNCellAsymCuts ; //![fNPtCuts*fNAsymCuts*fNCellNCuts*] REAL two-photon invariant mass distribution for different pt cut, n cell cuts and assymetry
216 TH2F ** fhMiPtNCellAsymCuts ; //![fNPtCuts*fNAsymCuts*fNCellNCuts] Mixed two-photon invariant mass distribution for different pt cut, n cell cuts and assymetry
99b8e903 217 TH2F ** fhRePtNCellAsymCutsSM[12] ; //![fNPtCuts*fNAsymCuts*fNCellNCutsfNModules] REAL two-photon invariant mass distribution for different pt cut, n cell cuts and assymetry for each module
0333ede6 218
219 TH2F ** fhRePIDBits ; //![fNPIDBits] REAL two-photon invariant mass distribution for different PID bits
220 TH3F ** fhRePtMult ; //![fNAsymCuts] REAL two-photon invariant mass distribution for different track multiplicity and assymetry cuts
221 TH2F * fhReSS[3] ; //! Combine clusters with 3 different cuts on shower shape
6175da48 222
223 // Asymmetry vs pt, in pi0/eta regions
0333ede6 224 TH2F * fhRePtAsym ; //! REAL two-photon pt vs asymmetry
225 TH2F * fhRePtAsymPi0 ; //! REAL two-photon pt vs asymmetry, close to pi0 mass
226 TH2F * fhRePtAsymEta ; //! REAL two-photon pt vs asymmetry, close to eta mass
c4a7d28a 227
72542aba 228 //Centrality, Event plane bins
2e876d85 229 TH1I * fhEventBin; //! Number of real pairs in a particular bin (cen,vz,rp)
230 TH1I * fhEventMixBin; //! Number of mixed pairs in a particular bin (cen,vz,rp)
0333ede6 231 TH1F * fhCentrality; //! Histogram with centrality bins with at least one pare
232 TH1F * fhCentralityNoPair; //! Histogram with centrality bins with no pair
20218aea 233
0333ede6 234 TH2F * fhEventPlaneResolution; //! Histogram with Event plane resolution vs centrality
72542aba 235
6175da48 236 // Pair opening angle
0333ede6 237 TH2F * fhRealOpeningAngle ; //! Opening angle of pair versus pair energy
238 TH2F * fhRealCosOpeningAngle ; //! Cosinus of opening angle of pair version pair energy
239 TH2F * fhMixedOpeningAngle ; //! Opening angle of pair versus pair energy
240 TH2F * fhMixedCosOpeningAngle ; //! Cosinus of opening angle of pair version pair energy
6175da48 241
242 //MC analysis histograms
243 //Pi0 Acceptance
29250849 244 TH1F * fhPrimPi0E ; //! Spectrum of Primary
245 TH1F * fhPrimPi0Pt ; //! Spectrum of Primary
246 TH1F * fhPrimPi0AccE ; //! Spectrum of primary with accepted daughters
247 TH1F * fhPrimPi0AccPt ; //! Spectrum of primary with accepted daughters
0333ede6 248 TH2F * fhPrimPi0Y ; //! Rapidity distribution of primary particles vs pT
249 TH2F * fhPrimPi0AccY ; //! Rapidity distribution of primary with accepted daughters vs pT
250 TH2F * fhPrimPi0Phi ; //! Azimutal distribution of primary particles vs pT
251 TH2F * fhPrimPi0AccPhi; //! Azimutal distribution of primary with accepted daughters vs pT
252 TH2F * fhPrimPi0OpeningAngle ; //! Opening angle of pair versus pair energy, primaries
3eb6ab95 253 TH2F * fhPrimPi0OpeningAngleAsym ; //! Opening angle of pair versus pair E asymmetry, pi0 primaries
254 TH2F * fhPrimPi0CosOpeningAngle ; //! Cosinus of opening angle of pair version pair energy, pi0 primaries
c8710850 255 TH2F * fhPrimPi0PtCentrality ; //! primary pi0 reconstructed centrality vs pT
256 TH2F * fhPrimPi0PtEventPlane ; //! primary pi0 reconstructed event plane vs pT
257 TH2F * fhPrimPi0AccPtCentrality ; //! primary pi0 with accepted daughters reconstructed centrality vs pT
258 TH2F * fhPrimPi0AccPtEventPlane ; //! primary pi0 with accepted daughters reconstructed event plane vs pT
259
6175da48 260 //Eta acceptance
29250849 261 TH1F * fhPrimEtaE ; //! Spectrum of Primary
262 TH1F * fhPrimEtaPt ; //! Spectrum of Primary
263 TH1F * fhPrimEtaAccE ; //! Spectrum of primary with accepted daughters
264 TH1F * fhPrimEtaAccPt ; //! Spectrum of primary with accepted daughters
0333ede6 265 TH2F * fhPrimEtaY ; //! Rapidity distribution of primary particles vs pT
266 TH2F * fhPrimEtaAccY ; //! Rapidity distribution of primary with accepted daughters vs pT
267 TH2F * fhPrimEtaPhi ; //! Azimutal distribution of primary particles vs pT
268 TH2F * fhPrimEtaAccPhi; //! Azimutal distribution of primary with accepted daughters vs pT
c8710850 269 TH2F * fhPrimEtaOpeningAngle ; //! Opening angle of pair versus pair energy, eta primaries
270 TH2F * fhPrimEtaOpeningAngleAsym ; //! Opening angle of pair versus pair E asymmetry, eta primaries
271 TH2F * fhPrimEtaCosOpeningAngle ; //! Cosinus of opening angle of pair version pair energy, eta primaries
272 TH2F * fhPrimEtaPtCentrality ; //! primary eta reconstructed centrality vs pT
273 TH2F * fhPrimEtaPtEventPlane ; //! primary eta reconstructed event plane vs pT
274 TH2F * fhPrimEtaAccPtCentrality ; //! primary eta with accepted daughters reconstructed centrality vs pT
275 TH2F * fhPrimEtaAccPtEventPlane ; //! primary eta with accepted daughters reconstructed event plane vs pT
6175da48 276
08a56f5f 277 // Primaries origin
0333ede6 278 TH2F * fhPrimPi0PtOrigin ; //! Spectrum of generated pi0 vs mother
279 TH2F * fhPrimEtaPtOrigin ; //! Spectrum of generated eta vs mother
08a56f5f 280
6175da48 281 //Pair origin
282 //Array of histograms ordered as follows: 0-Photon, 1-electron, 2-pi0, 3-eta, 4-a-proton, 5-a-neutron, 6-stable particles,
283 // 7-other decays, 8-string, 9-final parton, 10-initial parton, intermediate, 11-colliding proton, 12-unrelated
0333ede6 284 TH2F * fhMCOrgMass[13]; //! Mass vs pt of real pairs, check common origin of pair
285 TH2F * fhMCOrgAsym[13]; //! Asymmetry vs pt of real pairs, check common origin of pair
286 TH2F * fhMCOrgDeltaEta[13]; //! Delta Eta vs pt of real pairs, check common origin of pair
287 TH2F * fhMCOrgDeltaPhi[13]; //! Delta Phi vs pt of real pairs, check common origin of pair
6175da48 288
289 //Multiple cuts in simulation, origin pi0 or eta
0333ede6 290 TH2F ** fhMCPi0MassPtRec; //![fNPtCuts*fNAsymCuts*fNCellNCuts] Real pi0 pairs, reconstructed mass vs reconstructed pt of original pair
291 TH2F ** fhMCPi0MassPtTrue; //![fNPtCuts*fNAsymCuts*fNCellNCuts] Real pi0 pairs, reconstructed mass vs generated pt of original pair
292 TH2F ** fhMCPi0PtTruePtRec; //![fNPtCuts*fNAsymCuts*fNCellNCuts] Real pi0 pairs, reconstructed pt vs generated pt of pair
293 TH2F ** fhMCEtaMassPtRec; //![fNPtCuts*fNAsymCuts*fNCellNCuts] Real eta pairs, reconstructed mass vs reconstructed pt of original pair
294 TH2F ** fhMCEtaMassPtTrue; //![fNPtCuts*fNAsymCuts*fNCellNCuts] Real eta pairs, reconstructed mass vs generated pt of original pair
295 TH2F ** fhMCEtaPtTruePtRec; //![fNPtCuts*fNAsymCuts*fNCellNCuts] Real eta pairs, reconstructed pt vs generated pt of pair
296
297 TH2F * fhMCPi0PtOrigin ; //! Mass of reoconstructed pi0 pairs in calorimeter vs mother
298 TH2F * fhMCEtaPtOrigin ; //! Mass of reoconstructed pi0 pairs in calorimeter vs mother
6175da48 299
99b8e903 300 TH2F * fhReMCFromConversion ; //! Invariant mass of 2 clusters originated in conversions
301 TH2F * fhReMCFromNotConversion ; //! Invariant mass of 2 clusters not originated in conversions
302 TH2F * fhReMCFromMixConversion ; //! Invariant mass of 2 clusters one from conversion and the other not
303
be894c1d 304 TH2F * fhArmPrimPi0[4]; //! Armenteros plots for primary pi0 in 6 energy bins
305 TH2F * fhArmPrimEta[4]; //! Armenteros plots for primary eta in 6 energy bins
29555e96 306 TH2F * fhCosThStarPrimPi0; //! cos(theta*) plots vs E for primary pi0, same as asymmetry ...
307 TH2F * fhCosThStarPrimEta; //! cos(theta*) plots vs E for primary eta, same as asymmetry ...
be894c1d 308
745913ae 309 AliAnaPi0( const AliAnaPi0 & api0) ; // cpy ctor
310 AliAnaPi0 & operator = (const AliAnaPi0 & api0) ; // cpy assignment
99b8e903 311
29555e96 312 ClassDef(AliAnaPi0,26)
1c5acb87 313} ;
314
315
316#endif //ALIANAPI0_H
317
318
319