]> git.uio.no Git - u/mrichter/AliRoot.git/blame - HBTAN/AliHBTWeightNonId3DCorrFctn.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Updated to the new way AOD stores particles
[u/mrichter/AliRoot.git] / HBTAN / AliHBTWeightNonId3DCorrFctn.cxx
CommitLineData
cbcc22b0 1#include "AliHBTWeightNonId3DCorrFctn.h"
2#include <TH1.h>
3#include <Riostream.h>
4
5///////////////////////////////////////////////////////
6// //
7// AliHBTWeightNonId3DCorrFctn.h //
8// //
9// Class for calculating 3D non-id correlation //
10// functions using methods of weights //
11// //
12///////////////////////////////////////////////////////
13
14ClassImp(AliHBTWeightNonId3DCorrFctn)
15
16AliHBTWeightNonId3DCorrFctn::AliHBTWeightNonId3DCorrFctn(const char* name, const char* title):
17 AliHBTTwoPairFctn1D(name,title),
18 fWeightNumOutP(0x0),
19 fWeightDenOutP(0x0),
20 fWeightRatOutP(0x0),
21 fWeightNumOutN(0x0),
22 fWeightDenOutN(0x0),
23 fWeightRatOutN(0x0),
24 fWeightRatOut(0x0),
25 fWeightRatOutNOverP(0x0),
26 fWeightNumSideP(0x0),
27 fWeightDenSideP(0x0),
28 fWeightRatSideP(0x0),
29 fWeightNumSideN(0x0),
30 fWeightDenSideN(0x0),
31 fWeightRatSideN(0x0),
32 fWeightRatSide(0x0),
33 fWeightRatSideNOverP(0x0),
34 fWeightNumLongP(0x0),
35 fWeightDenLongP(0x0),
36 fWeightRatLongP(0x0),
37 fWeightNumLongN(0x0),
38 fWeightDenLongN(0x0),
39 fWeightRatLongN(0x0),
40 fWeightRatLong(0x0),
41 fWeightRatLongNOverP(0x0)
42{
43//ctor
44}
45/******************************************************************/
46AliHBTWeightNonId3DCorrFctn::AliHBTWeightNonId3DCorrFctn(const char* name, const char* title,
47 Int_t nbinsX, Float_t maxXval, Float_t minXval):
48 AliHBTTwoPairFctn1D(name,title,nbinsX,maxXval,minXval),
49 fWeightNumOutP(0x0),
50 fWeightDenOutP(0x0),
51 fWeightRatOutP(0x0),
52 fWeightNumOutN(0x0),
53 fWeightDenOutN(0x0),
54 fWeightRatOutN(0x0),
55 fWeightRatOut(0x0),
56 fWeightRatOutNOverP(0x0),
57 fWeightNumSideP(0x0),
58 fWeightDenSideP(0x0),
59 fWeightRatSideP(0x0),
60 fWeightNumSideN(0x0),
61 fWeightDenSideN(0x0),
62 fWeightRatSideN(0x0),
63 fWeightRatSide(0x0),
64 fWeightRatSideNOverP(0x0),
65 fWeightNumLongP(0x0),
66 fWeightDenLongP(0x0),
67 fWeightRatLongP(0x0),
68 fWeightNumLongN(0x0),
69 fWeightDenLongN(0x0),
70 fWeightRatLongN(0x0),
71 fWeightRatLong(0x0),
72 fWeightRatLongNOverP(0x0)
73{
74//ctor
75}
76
77/******************************************************************/
78AliHBTWeightNonId3DCorrFctn::AliHBTWeightNonId3DCorrFctn(const AliHBTWeightNonId3DCorrFctn& in):
79 AliHBTTwoPairFctn1D(in),
80 fWeightNumOutP(0x0),
81 fWeightDenOutP(0x0),
82 fWeightRatOutP(0x0),
83 fWeightNumOutN(0x0),
84 fWeightDenOutN(0x0),
85 fWeightRatOutN(0x0),
86 fWeightRatOut(0x0),
87 fWeightRatOutNOverP(0x0),
88 fWeightNumSideP(0x0),
89 fWeightDenSideP(0x0),
90 fWeightRatSideP(0x0),
91 fWeightNumSideN(0x0),
92 fWeightDenSideN(0x0),
93 fWeightRatSideN(0x0),
94 fWeightRatSide(0x0),
95 fWeightRatSideNOverP(0x0),
96 fWeightNumLongP(0x0),
97 fWeightDenLongP(0x0),
98 fWeightRatLongP(0x0),
99 fWeightNumLongN(0x0),
100 fWeightDenLongN(0x0),
101 fWeightRatLongN(0x0),
102 fWeightRatLong(0x0),
103 fWeightRatLongNOverP(0x0)
104{
105//ctor
106}
107
108/******************************************************************/
109
110AliHBTWeightNonId3DCorrFctn::~AliHBTWeightNonId3DCorrFctn()
111{
112 //dtor
113 delete fWeightNumOutP;
114 delete fWeightDenOutP;
115 delete fWeightRatOutP;
116 delete fWeightNumOutN;
117 delete fWeightDenOutN;
118 delete fWeightRatOutN;
119 delete fWeightRatOut;
120 delete fWeightRatOutNOverP;
121 delete fWeightNumSideP;
122 delete fWeightDenSideP;
123 delete fWeightRatSideP;
124 delete fWeightNumSideN;
125 delete fWeightDenSideN;
126 delete fWeightRatSideN;
127 delete fWeightRatSide;
128 delete fWeightRatSideNOverP;
129 delete fWeightNumLongP;
130 delete fWeightDenLongP;
131 delete fWeightRatLongP;
132 delete fWeightNumLongN;
133 delete fWeightDenLongN;
134 delete fWeightRatLongN;
135 delete fWeightRatLong;
136 delete fWeightRatLongNOverP;
137
138}
139
140//-------------------------------------
141void AliHBTWeightNonId3DCorrFctn::ProcessSameEventParticles(AliHBTPair* trackpair, AliHBTPair* partpair)
142{
143 //Fills the numerator using pair from the same event
144 partpair = CheckPair(partpair);
145 trackpair = CheckPair(trackpair);
146 if(partpair == 0x0) return;
147 if(trackpair == 0x0) return;
148
31aae134 149 double tKStar = TMath::Abs(trackpair->GetKStar());
cbcc22b0 150 double tKStarOut = trackpair->GetKStarOut();
151 double tKStarSide = trackpair->GetKStarSide();
152 double tKStarLong = trackpair->GetKStarLong();
153
154 Double_t weight = 1.0;
155
156 if ( trackpair && partpair)
157 {
158 if ( ( trackpair->Particle1()->GetPdgCode() == partpair->Particle1()->GetPdgCode()) &&
159 ( trackpair->Particle2()->GetPdgCode() == partpair->Particle2()->GetPdgCode()) )
160 {
161 weight=partpair->GetWeight();
162 }
163
164 if(tKStarOut>0.)
165 {
166 fWeightNumOutP->Fill(tKStar,weight);
167 }
168 else
169 {
170 fWeightNumOutN->Fill(tKStar,weight);
171 }
172
173 if(tKStarSide>0.)
174 {
175 fWeightNumSideP->Fill(tKStar,weight);
176 }
177 else
178 {
179 fWeightNumSideN->Fill(tKStar,weight);
180 }
181
182 if(tKStarLong>0.)
183 {
184 fWeightNumLongP->Fill(tKStar,weight);
185 }
186 else
187 {
188 fWeightNumLongN->Fill(tKStar,weight);
189 }
190 }
191
192}
193
194/****************************************************************/
195
196
197void AliHBTWeightNonId3DCorrFctn::ProcessDiffEventParticles(AliHBTPair* trackpair, AliHBTPair* partpair)
198{
199
200 trackpair = CheckPair(trackpair);
201 partpair = CheckPair(partpair);
202
31aae134 203 double tKStar = TMath::Abs(partpair->GetKStar());
cbcc22b0 204 double tKStarOut = trackpair->GetKStarOut();
205 double tKStarSide = trackpair->GetKStarSide();
206 double tKStarLong = trackpair->GetKStarLong();
207
208 if ( trackpair && partpair)
209 {
210 if(tKStarOut>0.)
211 {
212 fWeightDenOutP->Fill(tKStar);
213 }
214 else
215 {
216 fWeightDenOutN->Fill(tKStar);
217 }
218
219 if(tKStarSide>0.)
220 {
221 fWeightDenSideP->Fill(tKStar);
222 }
223 else
224 {
225 fWeightDenSideN->Fill(tKStar);
226 }
227
228 if(tKStarLong>0.)
229 {
230 fWeightDenLongP->Fill(tKStar);
231 }
232 else
233 {
234 fWeightDenLongN->Fill(tKStar);
235 }
236 }
237
238}
239
240
241/****************************************************************/
242void AliHBTWeightNonId3DCorrFctn::Init()
243{
244 AliHBTTwoPairFctn1D::Init();
245 fWeightNumOutP->Reset();
246 fWeightDenOutP->Reset();
247 fWeightRatOutP->Reset();
248 fWeightNumOutN->Reset();
249 fWeightDenOutN->Reset();
250 fWeightRatOutN->Reset();
251 fWeightRatOut->Reset();
252 fWeightRatOutNOverP->Reset();
253
254 fWeightNumSideP->Reset();
255 fWeightDenSideP->Reset();
256 fWeightRatSideP->Reset();
257 fWeightNumSideN->Reset();
258 fWeightDenSideN->Reset();
259 fWeightRatSideN->Reset();
260 fWeightRatSide->Reset();
261 fWeightRatSideNOverP->Reset();
262
263 fWeightNumLongP->Reset();
264 fWeightDenLongP->Reset();
265 fWeightRatLongP->Reset();
266 fWeightNumLongN->Reset();
267 fWeightDenLongN->Reset();
268 fWeightRatLongN->Reset();
269 fWeightRatLong->Reset();
270 fWeightRatLongNOverP->Reset();
271}
272/******************************************************************/
273
274void AliHBTWeightNonId3DCorrFctn::BuildHistos(Int_t nbins, Float_t max, Float_t min)
275{
276
277 AliHBTFunction1D::BuildHistos(nbins,max,min);
278
279 TString nameNumOutP = "WeightNumOutP";
280 TString nameDenOutP = "WeightDenOutP";
281 TString nameRatOutP = "WeightRatOutP";
282 TString nameNumOutN = "WeightNumOutN";
283 TString nameDenOutN = "WeightDenOutN";
284 TString nameRatOutN = "WeightRatOutN";
285 TString nameRatOut = "WeightRatOut";
286 TString nameRatOutNOverP = "WeightRatOutNOverP";
287 TString nameNumSideP = "WeightNumSideP";
288 TString nameDenSideP = "WeightDenSideP";
289 TString nameRatSideP = "WeightRatSideP";
290 TString nameNumSideN = "WeightNumSideN";
291 TString nameDenSideN = "WeightDenSideN";
292 TString nameRatSideN = "WeightRatSideN";
293 TString nameRatSide = "WeightRatSide";
294 TString nameRatSideNOverP = "WeightRatSideNOverP";
295 TString nameNumLongP = "WeightNumLongP";
296 TString nameDenLongP = "WeightDenLongP";
297 TString nameRatLongP = "WeightRatLongP";
298 TString nameNumLongN = "WeightNumLongN";
299 TString nameDenLongN = "WeightDenLongN";
300 TString nameRatLongN = "WeightRatLongN";
301 TString nameRatLong = "WeightRatLong";
302 TString nameRatLongNOverP = "WeightRatLongNOverP";
303
304 fWeightNumOutP = new TH1D(nameNumOutP.Data(),nameNumOutP.Data(),nbins,min,max);
305 fWeightDenOutP = new TH1D(nameDenOutP.Data(),nameDenOutP.Data(),nbins,min,max);
306 fWeightRatOutP = new TH1D(nameRatOutP.Data(),nameRatOutP.Data(),nbins,min,max);
307
308 fWeightNumOutN = new TH1D(nameNumOutN.Data(),nameNumOutN.Data(),nbins,min,max);
309 fWeightDenOutN = new TH1D(nameDenOutN.Data(),nameDenOutN.Data(),nbins,min,max);
310 fWeightRatOutN = new TH1D(nameRatOutN.Data(),nameRatOutN.Data(),nbins,min,max);
311
312 fWeightRatOut = new TH1D(nameRatOut.Data(),nameRatOut.Data(),nbins,min,max);
313 fWeightRatOutNOverP = new TH1D(nameRatOutNOverP.Data(),nameRatOutNOverP.Data(),nbins,min,max);
314
315 fWeightNumSideP = new TH1D(nameNumSideP.Data(),nameNumSideP.Data(),nbins,min,max);
316 fWeightDenSideP = new TH1D(nameDenSideP.Data(),nameDenSideP.Data(),nbins,min,max);
317 fWeightRatSideP = new TH1D(nameRatSideP.Data(),nameRatSideP.Data(),nbins,min,max);
318
319 fWeightNumSideN = new TH1D(nameNumSideN.Data(),nameNumSideN.Data(),nbins,min,max);
320 fWeightDenSideN = new TH1D(nameDenSideN.Data(),nameDenSideN.Data(),nbins,min,max);
321 fWeightRatSideN = new TH1D(nameRatSideN.Data(),nameRatSideN.Data(),nbins,min,max);
322
323 fWeightRatSide = new TH1D(nameRatSide.Data(),nameRatSide.Data(),nbins,min,max);
324 fWeightRatSideNOverP = new TH1D(nameRatSideNOverP.Data(),nameRatSideNOverP.Data(),nbins,min,max);
325
326 fWeightNumLongP = new TH1D(nameNumLongP.Data(),nameNumLongP.Data(),nbins,min,max);
327 fWeightDenLongP = new TH1D(nameDenLongP.Data(),nameDenLongP.Data(),nbins,min,max);
328 fWeightRatLongP = new TH1D(nameRatLongP.Data(),nameRatLongP.Data(),nbins,min,max);
329
330 fWeightNumLongN = new TH1D(nameNumLongN.Data(),nameNumLongN.Data(),nbins,min,max);
331 fWeightDenLongN = new TH1D(nameDenLongN.Data(),nameDenLongN.Data(),nbins,min,max);
332 fWeightRatLongN = new TH1D(nameRatLongN.Data(),nameRatLongN.Data(),nbins,min,max);
333
334 fWeightRatLong = new TH1D(nameRatLong.Data(),nameRatLong.Data(),nbins,min,max);
335 fWeightRatLongNOverP = new TH1D(nameRatLongNOverP.Data(),nameRatLongNOverP.Data(),nbins,min,max);
336
337
338 fWeightNumOutP->Sumw2();
339 fWeightDenOutP->Sumw2();
340 fWeightRatOutP->Sumw2();
341 fWeightNumOutN->Sumw2();
342 fWeightDenOutN->Sumw2();
343 fWeightRatOutN->Sumw2();
344 fWeightRatOut->Sumw2();
345 fWeightRatOutNOverP->Sumw2();
346 fWeightNumSideP->Sumw2();
347 fWeightDenSideP->Sumw2();
348 fWeightRatSideP->Sumw2();
349 fWeightNumSideN->Sumw2();
350 fWeightDenSideN->Sumw2();
351 fWeightRatSideN->Sumw2();
352 fWeightRatSide->Sumw2();
353 fWeightRatSideNOverP->Sumw2();
354 fWeightNumLongP->Sumw2();
355 fWeightDenLongP->Sumw2();
356 fWeightRatLongP->Sumw2();
357 fWeightNumLongN->Sumw2();
358 fWeightDenLongN->Sumw2();
359 fWeightRatLongN->Sumw2();
360 fWeightRatLong->Sumw2();
361 fWeightRatLongNOverP->Sumw2();
362}
363
364/******************************************************************/
365
366TH1* AliHBTWeightNonId3DCorrFctn::GetResult()
367{
368 return fWeightRatOutN;
369}
370
371/******************************************************************/
372
373void AliHBTWeightNonId3DCorrFctn::WriteFunction()
374{
375
376 Double_t outPscale = Scale(fWeightNumOutP,fWeightDenOutP);
377 cout<<"WoutPscale = "<<outPscale<<endl;
378 fWeightRatOutP->Divide(fWeightNumOutP,fWeightDenOutP,outPscale);
379
380 Double_t outNscale = Scale(fWeightNumOutN,fWeightDenOutN);
381 cout<<"WoutNscale = "<<outNscale<<endl;
382 fWeightRatOutN->Divide(fWeightNumOutN,fWeightDenOutN,outNscale);
383
384 fWeightRatOutNOverP->Divide(fWeightRatOutN,fWeightRatOutP);
385
386
387 Double_t sidePscale = Scale(fWeightNumSideP,fWeightDenSideP);
388 fWeightRatSideP->Divide(fWeightNumSideP,fWeightDenSideP,sidePscale);
389
390 Double_t sideNscale = Scale(fWeightNumSideN,fWeightDenSideN);
391 fWeightRatSideN->Divide(fWeightNumSideN,fWeightDenSideN,sideNscale);
392
393 fWeightRatSideNOverP->Divide(fWeightRatSideN,fWeightRatSideP);
394
395
396 Double_t longPscale = Scale(fWeightNumLongP,fWeightDenLongP);
397 fWeightRatLongP->Divide(fWeightNumLongP,fWeightDenLongP,longPscale);
398
399 Double_t longNscale = Scale(fWeightNumLongN,fWeightDenLongN);
400 fWeightRatLongN->Divide(fWeightNumLongN,fWeightDenLongN,longNscale);
401
402 fWeightRatLongNOverP->Divide(fWeightRatLongN,fWeightRatLongP);
403
404 fWeightNumOutP->Write();
405 fWeightDenOutP->Write();
406 fWeightRatOutP->Write();
407 fWeightNumOutN->Write();
408 fWeightDenOutN->Write();
409 fWeightRatOutN->Write();
410 fWeightRatOut->Write();
411 fWeightRatOutNOverP->Write();
412
413 fWeightNumSideP->Write();
414 fWeightDenSideP->Write();
415 fWeightRatSideP->Write();
416 fWeightNumSideN->Write();
417 fWeightDenSideN->Write();
418 fWeightRatSideN->Write();
419 fWeightRatSide->Write();
420 fWeightRatSideNOverP->Write();
421
422 fWeightNumLongP->Write();
423 fWeightDenLongP->Write();
424 fWeightRatLongP->Write();
425 fWeightNumLongN->Write();
426 fWeightDenLongN->Write();
427 fWeightRatLongN->Write();
428 fWeightRatLong->Write();
429 fWeightRatLongNOverP->Write();
430
431}
432