]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGCF/FEMTOSCOPY/Chaoticity/AliFourPion.h
Merge branch 'master' of https://git.cern.ch/reps/AliRoot
[u/mrichter/AliRoot.git] / PWGCF / FEMTOSCOPY / Chaoticity / AliFourPion.h
CommitLineData
be9ef9f9 1#ifndef AliFourPion_cxx
2#define AliFourPion_cxx
3//
4// Class AliFourPion
5//
6// AliFourPion
7// author:
8// Dhevan Gangadharan (dhevan.raja.gangadharan@cern.ch)
9//
10
11
12
13class TH1F;
14class TH3F;
15class TH1D;
16class TH2D;
17class TH3D;
18
19class TProfile;
20class TProfile2D;
21class TRandom3;
22
23class AliESDEvent;
24class AliAODEvent;
25class AliESDtrackCuts;
26class AliESDpid;
27
28#include "AliAnalysisTask.h"
29#include "AliAnalysisTaskSE.h"
30#include "AliESDpid.h"
31#include "AliAODPid.h"
32#include "AliFourPionEventCollection.h"
33#include "AliCentrality.h"
34
35class AliFourPion : public AliAnalysisTaskSE {
36 public:
37
38 AliFourPion();
39 AliFourPion(const Char_t *name);
40 virtual ~AliFourPion();
41 AliFourPion(const AliFourPion &obj);
42 AliFourPion &operator=(const AliFourPion &obj);
43
44
45 virtual void UserCreateOutputObjects();
46 virtual void UserExec(Option_t *option);
47 virtual void Terminate(Option_t *);
48
49 enum {
50 kPairLimit = 10000,//10000
51 kNormPairLimit = 45000,
52 kMultLimitPbPb = 1800,//1800
53 kMultLimitpp = 300,
54 kMultBinspp = 11,
55 kMCarrayLimit = 150000,// 110000
56 kQbinsWeights = 40,
57 kNDampValues = 16,
58 kRmin = 5,// EW min radii 5 fm
ab890ec3 59 kDENtypes = 44,
be9ef9f9 60 };
61
62 static const Int_t fKbinsT = 4;// Set fKstep as well !!!!
63 static const Int_t fKbinsY = 1;// Set fKstep as well !!!!
64 static const Int_t fEDbins = 2;
65 static const Int_t fCentBins = 10;// 0-50%
66 static const Int_t fRVALUES = 7;// 7 EW radii (5-11) , was 8 Gaussian radii (3-10fm)
67
68
69 Int_t GetNumKtBins() const {return AliFourPion::fKbinsT;}
70 Int_t GetNumRValues() const {return AliFourPion::fRVALUES;}
71 Int_t GetNumCentBins() const {return AliFourPion::fCentBins;}
72 Int_t GetNumEDBins() const {return AliFourPion::fEDbins;}
73 void SetWeightArrays(Bool_t legoCase=kTRUE, TH3F *histos[AliFourPion::fKbinsT][AliFourPion::fCentBins]=0x0);
74 void SetMomResCorrections(Bool_t legoCase=kTRUE, TH2D *temp2D=0x0);
75 void SetFSICorrelations(Bool_t legoCase=kTRUE, TH1D *tempss[12]=0x0, TH1D *tempos[12]=0x0);
5591748e 76 void SetMuonCorrections(Bool_t legoCase=kTRUE, TH2D *tempMuon=0x0);
be9ef9f9 77 //
78 void SetMCdecision(Bool_t mc) {fMCcase = mc;}
79 void SetTabulatePairs(Bool_t tabulate) {fTabulatePairs = tabulate;}
95567836 80 void SetInterpolationType(Bool_t linearInterp) {fLinearInterpolation = linearInterp;}
be9ef9f9 81 void SetPbPbCase(Bool_t pbpb) {fPbPbcase = pbpb;}
82 void SetGenerateSignal(Bool_t gen) {fGenerateSignal = gen;}
83 void SetGeneratorOnly(Bool_t genOnly) {fGeneratorOnly = genOnly;}
84 void SetCentBinRange(Int_t low, Int_t high) {fCentBinLowLimit = low; fCentBinHighLimit = high;}
85 void SetLEGOCase(Bool_t lego) {fLEGO = lego;}
86 void SetFilterBit(UInt_t filterbit) {fFilterBit = filterbit;}
87 void SetMaxChi2NDF(Float_t MaxChi2NDF) {fMaxChi2NDF = MaxChi2NDF;}
88 void SetMinTPCncls(Int_t MinTPCncls) {fMinTPCncls = MinTPCncls;}
89 void SetPairSeparationCutEta(Float_t pairsep) {fMinSepPairEta = pairsep;}
90 void SetPairSeparationCutPhi(Float_t pairsep) {fMinSepPairPhi = pairsep;}
91 void SetNsigmaTPC(Float_t nsig) {fSigmaCutTPC = nsig;}
92 void SetNsigmaTOF(Float_t nsig) {fSigmaCutTOF = nsig;}
93 void SetRMax(Int_t rbin) {fRMax = rbin;}
94 void SetfcSq(Float_t fcSq) {ffcSq = fcSq;}
95 //
96
97
98 private:
99
100 void ParInit();
101 Bool_t AcceptPair(AliFourPionTrackStruct, AliFourPionTrackStruct);
fcb16ac9 102 Float_t Gamov(Int_t, Int_t, Float_t);
be9ef9f9 103 void Shuffle(Int_t*, Int_t, Int_t);
104 Float_t GetQinv(Float_t[], Float_t[]);
105 void GetQosl(Float_t[], Float_t[], Float_t&, Float_t&, Float_t&);
106 void GetWeight(Float_t[], Float_t[], Float_t&, Float_t&);
107 Float_t FSICorrelation(Int_t, Int_t, Float_t);
108 Float_t MCWeight(Int_t[2], Float_t, Float_t, Float_t, Float_t);
109 Float_t MCWeightOSL(Int_t, Int_t, Int_t, Int_t, Float_t, Float_t, Float_t, Float_t);
110 Float_t MCWeight3(Int_t, Float_t, Float_t, Int_t[3], Float_t[3], Float_t[3]);
111 Float_t MCWeightFSI3(Int_t, Float_t, Float_t, Int_t[3], Float_t[3]);
112 Float_t MCWeight4(Int_t, Float_t, Float_t, Int_t[4], Float_t[6], Float_t[6]);
113 Float_t MCWeightFSI4(Int_t, Float_t, Float_t, Int_t[4], Float_t[6]);
114 //
115 void SetFillBins2(Int_t, Int_t, Int_t&, Int_t&);
116 void SetFillBins3(Int_t, Int_t, Int_t, Short_t, Int_t&, Int_t&, Int_t&, Bool_t&, Bool_t&, Bool_t&);
6bb6954b 117 void SetFillBins4(Int_t, Int_t, Int_t, Int_t, Int_t&, Int_t&, Int_t&, Int_t&, Int_t, Bool_t[13]);
be9ef9f9 118 void SetFSIindex(Float_t);
119 //
95567836 120 Float_t cubicInterpolate(Float_t[4], Float_t);
121 Float_t nCubicInterpolate(Int_t, Float_t*, Float_t[]);
be9ef9f9 122
123 const char* fname;// name of class
124 AliAODEvent *fAOD; //! // AOD object
125 TList *fOutputList; //! Compact Output list
126 AliPIDResponse *fPIDResponse; //! PID response object; equivalent to AliAODpidUtil
127
128
129 AliFourPionEventCollection ***fEC; //!
130 AliFourPionEventStruct *fEvt; //!
131 AliFourPionTrackStruct *fTempStruct; //!
132 TRandom3* fRandomNumber; //!
133
134
135
136 //////////////////////////////
137 // histogram structures
138
139
140 struct St6 {
141 TH1D *fNorm3; //!
142 TH1D *fTerms3; //!
143 TProfile *fKfactor; //!
144 TProfile *fMeanQinv; //!
145 TH2D *fIdeal; //!
146 TH2D *fSmeared; //!
147 //
5591748e 148 TH3D *fMuonSmeared; //!
149 TH3D *fMuonIdeal; //!
150 TH3D *fMuonPionK3; //!
151 TH3D *fPionPionK3; //!
152 //
be9ef9f9 153 TH2D *fTwoPartNorm; //!
6432c081 154 TH2D *fTwoPartNegNorm; //!
43046300 155 TH2D *fTwoPartNormErr; //!
be9ef9f9 156 };
157 struct St7 {
158 TH3D *fTerms2OSL; //!
159 TH3D *fTerms2OSLQW; //!
160 };
161 struct St5 {
162 TH2D *fTerms2; //!
163 TH2D *fTerms2QW; //!
164 TH3D *fTerms2ThreeD; //!
165 TProfile2D *fAvgP; //!
166 TH2D *fIdeal; //!
167 TH2D *fSmeared; //!
168 TH2D *fUnitMultBin; //!
169 //
170 TH2D *fMuonSmeared; //!
171 TH2D *fMuonIdeal; //!
172 TH2D *fMuonPionK2; //!
173 TH2D *fPionPionK2; //!
174 TH1D *fMCqinv; //!
175 TH1D *fMCqinvQW; //!
176 TH2D *fPIDpurityDen; //!
177 TH3D *fPIDpurityNum; //!
178 struct St7 OSL_ktbin[2];
179 };
180 struct StFourPT {
181 TH1D *fNorm4; //!
182 TH1D *fTerms4; //!
183 TProfile *fKfactor; //!
184 TH2D *fIdeal; //!
185 TH2D *fSmeared; //!
186 //
5591748e 187 TH3D *fMuonSmeared; //!
188 TH3D *fMuonIdeal; //!
189 TH3D *fMuonPionK4; //!
190 TH3D *fPionPionK4; //!
191 //
be9ef9f9 192 TH2D *fTwoPartNorm; //!
6432c081 193 TH2D *fTwoPartNegNorm; //!
43046300 194 TH2D *fTwoPartNormErr; //!
be9ef9f9 195 };
196 struct St_EDB {
197 struct St5 TwoPT[2];
198 struct St6 ThreePT[5];
6bb6954b 199 struct StFourPT FourPT[13];
be9ef9f9 200 };
201 struct St_M {
202 struct St_EDB EDB[fEDbins];
203 };
204 struct St4 {
205 struct St_M MB[fCentBins];
206 };
207 struct St3 {
208 struct St4 Charge4[2];
209 struct St_M MB[fCentBins];
210 };
211 struct St2 {
212 struct St3 Charge3[2];
213 struct St_M MB[fCentBins];
214 };
215 struct St1 {
216 struct St2 Charge2[2];
217 };
218 struct St1 Charge1[2];//!
219
220
221 /////////////////////
222 // 4D r3 denominator
223 struct St_Ky {
224 struct St_M MB[fCentBins];
225 };
226 struct St_Kt {
227 struct St_Ky KY[fKbinsY];
228 };
229 struct St_Kt KT[fKbinsT];//!
230
231
232 Bool_t fLEGO;
233 Bool_t fMCcase;
234 Bool_t fAODcase;
235 Bool_t fPbPbcase;
236 Bool_t fGenerateSignal;
237 Bool_t fGeneratorOnly;
be9ef9f9 238 Bool_t fTabulatePairs;
95567836 239 Bool_t fLinearInterpolation;
be9ef9f9 240 Int_t fRMax;
241 Float_t ffcSq;
242 UInt_t fFilterBit;
243 Float_t fMaxChi2NDF;
244 Int_t fMinTPCncls;
245 Double_t fBfield;
246 Int_t fMbin;
247 Int_t fFSIindex;
248 Int_t fEDbin;
249 Int_t fMbins;
250 Int_t fMultLimit;
251 Int_t fCentBinLowLimit;
252 Int_t fCentBinHighLimit;
253 Int_t fEventCounter;
254 Int_t fEventsToMix;
255 Int_t fZvertexBins;
256 Int_t fMultLimits[kMultBinspp+1];
257 Float_t fQcut;
258 Float_t fQLowerCut;
259 Float_t fNormQcutLow;
260 Float_t fNormQcutHigh;
261 Float_t fKupperBound;
262 Float_t fQupperBoundQ2;
263 Float_t fQupperBoundQ3;
264 Float_t fQupperBoundQ4;
265 Float_t fQbinsQ2;
266 Float_t fQbinsQ3;
267 Float_t fQbinsQ4;
268 Float_t fQupperBoundWeights;
269 Float_t fKstepT[fKbinsT];
270 Float_t fKstepY[fKbinsY];
271 Float_t fKmeanT[fKbinsT];
272 Float_t fKmeanY[fKbinsY];
273 Float_t fKmiddleT[fKbinsT];
274 Float_t fKmiddleY[fKbinsY];
275 Float_t fQstep;
276 Float_t fQstepWeights;
277 Float_t fQmean[kQbinsWeights];
278 Float_t fDampStart;
279 Float_t fDampStep;
280
281 Float_t fTPCTOFboundry;
282 Float_t fTOFboundry;
283 Float_t fSigmaCutTPC;
284 Float_t fSigmaCutTOF;
285
286 Float_t fMinSepPairEta;
287 Float_t fMinSepPairPhi;
288 Float_t fShareQuality;
289 Float_t fShareFraction;
290
291 Float_t fTrueMassP, fTrueMassPi, fTrueMassK, fTrueMassKs, fTrueMassLam;
292
293 Int_t fKtIndexL,fKtIndexH;
294 //
295 Int_t fQoIndexL,fQoIndexH;
296 Int_t fQsIndexL,fQsIndexH;
297 Int_t fQlIndexL,fQlIndexH;
298
299 Bool_t fDummyB;
300
301 Float_t fKT3transition;
302 Float_t fKT4transition;
303
95567836 304 Float_t farrP1[4][4][4];
305 Float_t farrP2[4][4][4];
b71263d0 306
307 /* bool LowQPairSwitch_E0E0[kMultLimitPbPb][kMultLimitPbPb];//!
be9ef9f9 308 bool LowQPairSwitch_E0E1[kMultLimitPbPb][kMultLimitPbPb];//!
309 bool LowQPairSwitch_E0E2[kMultLimitPbPb][kMultLimitPbPb];//!
310 bool LowQPairSwitch_E0E3[kMultLimitPbPb][kMultLimitPbPb];//!
311 bool LowQPairSwitch_E1E2[kMultLimitPbPb][kMultLimitPbPb];//!
312 bool LowQPairSwitch_E1E3[kMultLimitPbPb][kMultLimitPbPb];//!
313 bool LowQPairSwitch_E2E3[kMultLimitPbPb][kMultLimitPbPb];//!
314 //
315 bool NormQPairSwitch_E0E0[kMultLimitPbPb][kMultLimitPbPb];//!
316 bool NormQPairSwitch_E0E1[kMultLimitPbPb][kMultLimitPbPb];//!
317 bool NormQPairSwitch_E0E2[kMultLimitPbPb][kMultLimitPbPb];//!
318 bool NormQPairSwitch_E0E3[kMultLimitPbPb][kMultLimitPbPb];//!
319 bool NormQPairSwitch_E1E2[kMultLimitPbPb][kMultLimitPbPb];//!
320 bool NormQPairSwitch_E1E3[kMultLimitPbPb][kMultLimitPbPb];//!
321 bool NormQPairSwitch_E2E3[kMultLimitPbPb][kMultLimitPbPb];//!
b71263d0 322 */
323 //
324 Char_t fDefaultsCharSwitch[kMultLimitPbPb];//!
325 TArrayC *fLowQPairSwitch_E0E0[kMultLimitPbPb];//!
326 TArrayC *fLowQPairSwitch_E0E1[kMultLimitPbPb];//!
327 TArrayC *fLowQPairSwitch_E0E2[kMultLimitPbPb];//!
328 TArrayC *fLowQPairSwitch_E0E3[kMultLimitPbPb];//!
6bb6954b 329 TArrayC *fLowQPairSwitch_E1E1[kMultLimitPbPb];//!
b71263d0 330 TArrayC *fLowQPairSwitch_E1E2[kMultLimitPbPb];//!
331 TArrayC *fLowQPairSwitch_E1E3[kMultLimitPbPb];//!
332 TArrayC *fLowQPairSwitch_E2E3[kMultLimitPbPb];//!
333 //
334 TArrayC *fNormQPairSwitch_E0E0[kMultLimitPbPb];//!
335 TArrayC *fNormQPairSwitch_E0E1[kMultLimitPbPb];//!
336 TArrayC *fNormQPairSwitch_E0E2[kMultLimitPbPb];//!
337 TArrayC *fNormQPairSwitch_E0E3[kMultLimitPbPb];//!
6bb6954b 338 TArrayC *fNormQPairSwitch_E1E1[kMultLimitPbPb];//!
b71263d0 339 TArrayC *fNormQPairSwitch_E1E2[kMultLimitPbPb];//!
340 TArrayC *fNormQPairSwitch_E1E3[kMultLimitPbPb];//!
341 TArrayC *fNormQPairSwitch_E2E3[kMultLimitPbPb];//!
342
be9ef9f9 343
344 public:
345 TH2D *fMomResC2;
5591748e 346 TH2D *fWeightmuonCorrection;
be9ef9f9 347 TH1D *fFSIss[12];
348 TH1D *fFSIos[12];
349 TH3F *fNormWeight[fKbinsT][fCentBins];
5591748e 350
be9ef9f9 351
352 ClassDef(AliFourPion, 1);
353};
354
355#endif