]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGGA/GammaConv/AliAnalysisTaskPi0v2.h
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Add sigma2 jet shape and fill constituent info. for subtracted jets
[u/mrichter/AliRoot.git] / PWGGA / GammaConv / AliAnalysisTaskPi0v2.h
CommitLineData
f7cf81bd 1#ifndef AliAnalysisTaskPi0v2_cxx
2#define AliAnalysisTaskPi0v2_cxx
3
4#include "AliAnalysisTaskSE.h"
5#include "TH1.h"
6#include "TH2.h"
7#include "TH3.h"
0a2b2b4b 8#include "THnSparse.h"
f7cf81bd 9#include "AliLog.h"
10#include "AliConversionSelection.h"
344100c4 11#include "AliConvEventCuts.h"
ca91a3e1 12#include "AliConversionMesonCuts.h"
f7cf81bd 13#include "AliV0ReaderV1.h"
14#include "AliEventplane.h"
15#include "TVector2.h"
16
17#include "TProfile.h"
18using namespace std;
19
20class AliAnalysisTaskPi0v2 : public AliAnalysisTaskSE{
21
344100c4 22 public:
23
24 enum EEventPlaneMethod{
25 kTPC=0,
26 kTPCEtaGap=1,
27 kV0A=2,
28 kV0C=3,
29 knEPMethod=4
30 };
31
32 enum EEventPlane{
33 kEPTPC=0,
34 kEPTPCEtaA,
35 kEPTPCEtaC,
36 kEPV0A,
37 kEPV0C,
38 knEP
39 };
40
41 enum EPDGCode{
42 kPi0=111,
43 kEta=221
44 };
45
46 static const Int_t knBinsPhi=6;
47 static const Int_t knCentMax=10;
48 static const Int_t knFlatPeriod=2;
49
50 AliAnalysisTaskPi0v2(const char *name="pi0v2",Int_t harmonic=2);
51 virtual ~AliAnalysisTaskPi0v2();
52
53 virtual void UserCreateOutputObjects();
54 virtual void UserExec(Option_t *option);
55 virtual void Terminate(Option_t *);
56
57 void SetCentralityBins(Double_t *bins,Int_t nbins);
58 void SetMeson(EPDGCode meson){fMesonPDGCode=meson;}
59
60 void SetNBinsPhi(Int_t nbins){fNBinsPhi=nbins;}
61 void SetV0Reader(AliV0ReaderV1 *v0Reader){fV0Reader=v0Reader;}
62 void SetInvMassRange(Double_t range[2]){fInvMassRange[0]=range[0];fInvMassRange[1]=range[1];};
63 void SetEtaGap(Double_t gapsize){fEtaGap=gapsize;};
64
65 void SetCuts(AliConversionSelection **conversionselection,Int_t numberOfCuts);
66 void SetFillQA(Bool_t fill){fFillQA=fill;}
67
68 void SetEPSelectionMask(Int_t mask[knEPMethod]){for(Int_t ii=0;ii<knEPMethod;ii++)fEPSelectionMask[ii]=mask[ii];};
69
70 void SetFlatteningCoeff(EEventPlane ep,Int_t period,Int_t nCent,Double_t *cc2,Double_t *cs2,Double_t *cc4,Double_t *cs4);
71
72 Int_t GetPeriodIndex(TString period);
73 Int_t GetHarmonic(){return fHarmonic;};
74
75 private:
76
77 AliAnalysisTaskPi0v2(const AliAnalysisTaskPi0v2&);// private::prevent use of copy constructor
78 AliAnalysisTaskPi0v2& operator=(const AliAnalysisTaskPi0v2&); // private::prevent use of assignment operator
79
80 Bool_t InitEvent();
81
82 void ProcessGammas(Int_t iCut,EEventPlaneMethod iEP);
83 void ProcessPi0s(Int_t iCut,EEventPlaneMethod iEP);
84 void ProcessQA();
85
86 Double_t GetPhiwrtRP(Double_t dPhi);
87 Double_t GetPhotonPhiwrtRP(AliAODConversionPhoton *gamma,EEventPlaneMethod iEP,Bool_t bDoFlattening=kTRUE);
88 Double_t GetPi0PhiwrtRP(AliAODConversionMother *pi0,EEventPlaneMethod iEP,Bool_t bDoFlattening=kTRUE);
89 Double_t GetChargedPhiwrtRP(AliVTrack *charged,EEventPlaneMethod iEP,Bool_t bDoFlattening=kTRUE);
90 void GetPhotondNdPhi(Int_t *dNdPhi,Int_t iEP,Int_t iCut=0);
91 void GetChargeddNdPhi(Int_t *dNdPhi,Int_t &ntot,Int_t iEP);
92 Int_t GetPhiBin(Double_t phiwrt);
93 Int_t GetPhotonPhiBin(AliAODConversionPhoton *gamma,Int_t iEP);
94 Double_t GetMCPhotonPhiwrtRP(TParticle *gamma,EEventPlaneMethod iEP,Bool_t bDoFlattening=kTRUE);
95 TVector2 GetEPContribution(AliAODConversionPhoton *gamma);
96 Double_t GetEventPlaneAngle(EEventPlaneMethod EPmethod,Double_t eta=0,AliAODConversionPhoton *gamma0=NULL,AliAODConversionPhoton *gamma1=NULL,Bool_t bDoFlattening=kTRUE);
97 Double_t GetTPCSubEPEta(EEventPlane ep);
98 Double_t GetCorrectedTPCEPAngle(AliAODConversionPhoton *gamma0=NULL,AliAODConversionPhoton *gamma1=NULL,Bool_t bDoFlattening=kTRUE);
99 Bool_t SetCentrality();
100 void ProcessEventPlane();
101 Int_t GetRadialBin(Double_t radius);
102 Int_t GetRunIndex(Int_t run);
103 Double_t ApplyFlattening(Double_t phi,EEventPlane ep);
104 Bool_t GetTPCEventPlane();
105
106 void GetV0EP(AliVEvent * event,Double_t &rpv0a,Double_t &rpv0c);
107 void LoadVZEROCalibration(Int_t run);
108 void LoadTPCCalibration(Int_t run);
109
110 Double_t GetWeight(TObject* track1);
111 Double_t GetPhiWeight(TObject* track1);
112 TH1F* SelectPhiDist(AliVTrack *track);
113
114 Double_t GetPsiInRange(Double_t phi);
115
116 TObjArray* GetEventPlaneTracks(Int_t &maxID);
117 TVector2 GetContributionEP(AliVTrack *track);
118 Int_t GetAODEPTrackFilterBit();
119
120 // Constants
121 enum Ebinsgamma{
122 kGammaPt=0,
123 kGammadPhi,
124 kGammaCent,
125 kGammaEPM,
126 knbinsGamma
127 };
128
129 enum Ebinspi0{
130 kPi0Pt=0,
131 kPi0Mass,
132 kPi0dPhi,
133 kPi0Cent,
134 kPi0EPM,
135 knbinsPi0
136 };
137
138 enum EEventSelection{
139 kEventIn=1,
140 kEventSelV0Reader,
141 kEventCentrality,
142 kEventRun,
143 kEventNoTPCEP,
144 kEventProcessEvent,
145 kEventSelected
146 };
147
148
149 static const Int_t knbinsGammaMult=3;
150 static const Int_t kGCnYBinsSpectra = 80;
151 static const Double_t kGCfirstYBinSpectra;
152 static const Double_t kGClastYBinSpectra;
153
154 // Class variables and pointer
155 AliV0ReaderV1 *fV0Reader; // V0Reader
156 Int_t fNCuts; // Number of Photon Cuts for v2 analysis
157 AliConversionSelection **fConversionSelection; //[fNCuts] Selection of Particles for given Cut
158 TClonesArray *fConversionGammas; //! Reconstructed Photons;
159 Int_t fNCentralityBins; // Number of Centrality Bins
160 Double_t fCentralityBins[knCentMax]; // CentralityBins for Analysis
161 Float_t fCentrality; //! Event Centrality
162 Int_t fCentralityBin; //! Event Centrality Bin
163 Int_t fNBinsPhi; // Number of Phi wrt RP bins
164 AliEventplane *fEP; //! Event Plane Pointer
165 Bool_t fUseTPCOnlyTracks; // Use TPC Only Tracks for EP
166 Double_t fEtaMax; // Eta Max for analysis;
167 Double_t fEtaGap; // Eta Gap
168 Double_t fRPTPCEtaA; //! TPCEtaA event plane
169 Double_t fRPTPCEtaC; //! TPCEtaC event plane
170 Double_t fRPV0A; //! V0A event plane
171 Double_t fRPV0C; //! V0C event plane
172 Double_t fRPTPC; //! TPC event plane
173 Double_t fRPTPCEtaABF; //! TPCEtaA event plane before flattening
174 Double_t fRPTPCEtaCBF; //! TPCEtaC event plane before flattening
175 Double_t fRPV0ABF; //! V0A event plane before flattening
176 Double_t fRPV0CBF; //! V0C event plane before flattening
177 Double_t fRPTPCBF; //! TPC event plane before flattening
178 AliConvEventCuts *fEventCuts; //! Cuts used by the V0Reader
179 AliConversionPhotonCuts *fConversionCuts; //! Cuts used by the V0Reader
180 TRandom3 *fRandomizer; //! Randomizer for Event Plane Randomisation
181 TList *fOutputList; //! List for Output (Histograms etc.)
182 EPDGCode fMesonPDGCode; // PDG Code of the processed Meson (for MC truth)
183 Double_t fInvMassRange[2]; // Inv Mass Range
184 Double_t fDeltaPsiRP; //! Difference between subEventPlane angles
185 Int_t fRunNumber; //! current run number
186 Int_t fRunIndex; //! current internal run index
187 Int_t fNEPMethods; // number of EP methods
188 Bool_t fFillQA; // Fill QA Histograms
189 Int_t fHarmonic; // Harmonic to be analyzed (v2,v3,..)
190 Double_t fPsiMax; // Range for Psi
191 TString fPeriod; //! "LHC11h","LHC10h"
192 Bool_t fIsAOD; //! Is AOD? else ESD
193 TH1F *fPhiDist[4]; //! array of Phi distributions used to calculate phi weights
194 THnSparse *fSparseDist; //! THn for eta-charge phi-weighting
195 TH1F *fHruns; //! information about runwise statistics of phi-weights
196 Bool_t fDoEPFlattening; // Do flattening
197 Int_t fEPSelectionMask[knEPMethod]; // Which EP methods shall be applied
198 Double_t fFlatc2[knFlatPeriod][knEP][knCentMax];
199 Double_t fFlats2[knFlatPeriod][knEP][knCentMax];
200 Double_t fFlatc4[knFlatPeriod][knEP][knCentMax];
201 Double_t fFlats4[knFlatPeriod][knEP][knCentMax];
202 Int_t fPeriodIndex; //!
203
204 // Histograms
205 TH1F *hNEvents; //!
206 TH1F *hEventSelection; //!
207
208 // RP
209 TH2F *hRPTPC; //!
210 TH2F *hRPV0A; //!
211 TH2F *hRPV0C; //!
212 TH2F *hRPTPCAC; //!
213 TH2F *hRPV0ATPC; //!
214 TH2F *hRPV0CTPC; //!
215 TH2F *hRPV0AC; //!
216 TH2F *hCos2TPC; //!
217 TH2F *hCos2V0ATPC; //!
218 TH2F *hCos2V0CTPC; //!
219 TH2F *hCos2V0AC; //!
220 TH2F *hRPTPCEtaA; //!
221 TH2F *hRPTPCEtaC; //!
222 TH2F *hRPTPCEtaAC; //!
223 TH2F *hCos2TPCEta; //!
224 TH2F *hCos2V0ATPCEtaA; //!
225 TH2F *hCos2V0ATPCEtaC; //!
226 TH2F *hCos2V0CTPCEtaA; //!
227 TH2F *hCos2V0CTPCEtaC; //!
228 TH2F *hCos2SumWeights; //!
229 TH2F *hEtaTPCEP; //!
230
231 // Gamma
232 TH2F *hGammaMultCent; //!
233 TH2F **hGammaPhi; //!
234 TH2F *hMultChargedvsNGamma; //!
235 TH2F *hMultChargedvsVZERO; //!
236 TH2F *hMultChargedvsSPD; //!
237
238 THnSparseF *hGammadNdPhi; //!
239 THnSparseF *hGammaMultdPhiTRUE; //!
240 THnSparseF *hGammaMultdPhiRECOTRUE; //!
241 THnSparseF *hGammaMultTRUE; //!
242 THnSparseF *hGammaMultRECOTRUE; //!
243 THnSparseF **hGammaMultdPhi; //!
244 THnSparseF **hGammaMult; //!
245
246 THnSparseF **hGamma; //!
247 THnSparseF *hGammaFull; //!
248 THnSparseF *hCharged; //!
249
250 // Pi0
251 THnSparseF **hPi0; //!
252 THnSparseF **hPi0BG; //!
253
254 //V0 Calibration
255
256 static const Int_t nCentrBinV0 = 9; // # cenrality bins
257 TProfile *fMultV0; //! object containing VZERO calibration information
258 Float_t fV0Cpol,fV0Apol; //! loaded by OADB
259 Float_t fMeanQ[nCentrBinV0][2][2]; //! and recentering
260 Float_t fWidthQ[nCentrBinV0][2][2]; //! ...
261
262 //Event Plane
263 //THnSparse *hEPVertex;
264 THnSparse *hEPQA; //!
265
266 ClassDef(AliAnalysisTaskPi0v2, 7); // example of analysis
f7cf81bd 267};
268
269#endif
270