]>
Commit | Line | Data |
---|---|---|
3a9a3487 | 1 | /************************************************************************** |
2 | * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
3 | * * | |
4 | * Author: The ALICE Off-line Project. * | |
5 | * Contributors are mentioned in the code where appropriate. * | |
6 | * * | |
7 | * Permission to use, copy, modify and distribute this software and its * | |
8 | * documentation strictly for non-commercial purposes is hereby granted * | |
9 | * without fee, provided that the above copyright notice appears in all * | |
10 | * copies and that both the copyright notice and this permission notice * | |
11 | * appear in the supporting documentation. The authors make no claims * | |
12 | * about the suitability of this software for any purpose. It is * | |
13 | * provided "as is" without express or implied warranty. * | |
14 | **************************************************************************/ | |
15 | ||
16 | //---------------------------------------------------------------------------- | |
17 | // Implementation of the D0toKpi class | |
741f4d30 | 18 | // for pp and PbPb interactions |
3a9a3487 | 19 | // Note: the two decay tracks are labelled: 0 (positive track) |
20 | // 1 (negative track) | |
3a9a3487 | 21 | // Origin: A. Dainese andrea.dainese@pd.infn.it |
22 | //---------------------------------------------------------------------------- | |
741f4d30 | 23 | |
3a9a3487 | 24 | #include <TH1.h> |
25 | #include <TH2.h> | |
26 | #include <TCanvas.h> | |
27 | #include <TPaveLabel.h> | |
dab957fb | 28 | #include <TVector3.h> |
3a9a3487 | 29 | |
30 | #include "AliD0toKpi.h" | |
31 | ||
32 | ClassImp(AliD0toKpi) | |
33 | ||
34 | //---------------------------------------------------------------------------- | |
35 | AliD0toKpi::AliD0toKpi() { | |
36 | // Default constructor | |
37 | ||
38 | fSignal = kFALSE; | |
39 | ||
40 | fEvent = 0; | |
41 | ||
42 | fTrkNum[0] = 0; | |
43 | fTrkNum[1] = 0; | |
44 | ||
45 | fV1x = 0.; | |
46 | fV1y = 0.; | |
47 | fV1z = 0.; | |
48 | fV2x = 0.; | |
49 | fV2y = 0.; | |
50 | fV2z = 0.; | |
51 | fDCA = 0.; | |
52 | ||
53 | fPx[0] = 0.; | |
54 | fPy[0] = 0.; | |
55 | fPz[0] = 0.; | |
56 | fPx[1] = 0.; | |
57 | fPy[1] = 0.; | |
58 | fPz[1] = 0.; | |
59 | ||
60 | fd0[0] = 0.; | |
61 | fd0[1] = 0.; | |
62 | ||
63 | fPdg[0] = 0; | |
64 | fPdg[1] = 0; | |
65 | fMum[0] = 0; | |
66 | fMum[1] = 0; | |
67 | ||
68 | fTagPi[0] = 0.; | |
69 | fTagPi[1] = 0.; | |
70 | fTagKa[0] = 0.; | |
71 | fTagKa[1] = 0.; | |
72 | fTagNid[0] = 0.; | |
73 | fTagNid[1] = 0.; | |
74 | ||
75 | fWgtAD0=fWgtAD0bar=fWgtBD0=fWgtBD0bar=fWgtCD0=fWgtCD0bar=fWgtDD0=fWgtDD0bar=0; | |
76 | ||
77 | } | |
78 | //---------------------------------------------------------------------------- | |
79 | AliD0toKpi::AliD0toKpi(Int_t ev,Int_t trkNum[2], | |
80 | Double_t v1[3],Double_t v2[3], | |
81 | Double_t dca, | |
82 | Double_t mom[6],Double_t d0[2]) { | |
83 | // Constructor | |
84 | ||
85 | fSignal = kFALSE; | |
86 | ||
87 | fEvent = ev; | |
88 | fTrkNum[0] = trkNum[0]; | |
89 | fTrkNum[1] = trkNum[1]; | |
90 | ||
91 | fV1x = v1[0]; | |
92 | fV1y = v1[1]; | |
93 | fV1z = v1[2]; | |
94 | fV2x = v2[0]; | |
95 | fV2y = v2[1]; | |
96 | fV2z = v2[2]; | |
97 | fDCA = dca; | |
98 | ||
99 | fPx[0] = mom[0]; | |
100 | fPy[0] = mom[1]; | |
101 | fPz[0] = mom[2]; | |
102 | fPx[1] = mom[3]; | |
103 | fPy[1] = mom[4]; | |
104 | fPz[1] = mom[5]; | |
105 | ||
106 | fd0[0] = d0[0]; | |
107 | fd0[1] = d0[1]; | |
108 | ||
109 | fPdg[0] = 0; | |
110 | fPdg[1] = 0; | |
111 | fMum[0] = 0; | |
112 | fMum[1] = 0; | |
113 | ||
114 | fTagPi[0] = 0.; | |
115 | fTagPi[1] = 0.; | |
116 | fTagKa[0] = 0.; | |
117 | fTagKa[1] = 0.; | |
118 | fTagNid[0] = 0.; | |
119 | fTagNid[1] = 0.; | |
120 | ||
121 | fWgtAD0=fWgtAD0bar=fWgtBD0=fWgtBD0bar=fWgtCD0=fWgtCD0bar=fWgtDD0=fWgtDD0bar=0; | |
122 | } | |
123 | //---------------------------------------------------------------------------- | |
124 | AliD0toKpi::~AliD0toKpi() {} | |
125 | //____________________________________________________________________________ | |
126 | AliD0toKpi::AliD0toKpi( const AliD0toKpi& d0toKpi):TObject(d0toKpi) { | |
127 | // dummy copy constructor | |
128 | } | |
129 | //---------------------------------------------------------------------------- | |
7b84ea57 | 130 | void AliD0toKpi::ApplyPID(const Char_t * pidScheme) { |
741f4d30 | 131 | // Applies particle identification |
7b84ea57 | 132 | const char *tofparampbpb = strstr(pidScheme,"TOFparamPbPb"); |
133 | const char *tofparampp = strstr(pidScheme,"TOFparamPP"); | |
3a9a3487 | 134 | |
135 | if((tofparampbpb || tofparampp) && fPdg[0]==0) { | |
136 | printf("AliD0toKpi::ApplyPID :\n Warning: TOF parameterized PID can be used only for simulation!\n"); | |
137 | return; | |
138 | } | |
139 | ||
140 | if(tofparampbpb) { | |
141 | // tagging of the positive track | |
142 | if(TMath::Abs(fPdg[0])==211 || TMath::Abs(fPdg[0])==13 | |
143 | || TMath::Abs(fPdg[0])==11) { // pion,muon,electron | |
741f4d30 | 144 | fTagPi[0] = LinearInterpolation(PChild(0),kPiBinsPbPb,kPiBinWidthPbPb,kPiTagPiPbPb); |
3a9a3487 | 145 | fTagNid[0] = 1.-fTagPi[0]; |
146 | fTagKa[0] = 0.; | |
147 | fTagPr[0] = 0.; | |
148 | } | |
149 | if(TMath::Abs(fPdg[0])==321) { // kaon | |
741f4d30 | 150 | fTagKa[0] = LinearInterpolation(PChild(0),kKBinsPbPb,kKBinWidthPbPb,kKTagKPbPb); |
151 | fTagNid[0] = LinearInterpolation(PChild(0),kKBinsPbPb,kKBinWidthPbPb,kKTagNidPbPb); | |
3a9a3487 | 152 | if((fTagNid[0]+fTagKa[0])>1.) fTagNid[0] = 1.-fTagKa[0]; |
153 | fTagPi[0] = 1.-fTagNid[0]-fTagKa[0]; | |
154 | fTagPr[0] = 0.; | |
155 | } | |
156 | if(TMath::Abs(fPdg[0])==2212) { // proton | |
741f4d30 | 157 | fTagPr[0] = LinearInterpolation(PChild(0),kPBinsPbPb,kPBinWidthPbPb,kPTagPPbPb); |
158 | fTagNid[0] = LinearInterpolation(PChild(0),kPBinsPbPb,kPBinWidthPbPb,kPTagNidPbPb); | |
3a9a3487 | 159 | if((fTagNid[0]+fTagPr[0])>1.) fTagNid[0] = 1.-fTagPr[0]; |
160 | fTagPi[0] = 1.-fTagNid[0]-fTagPr[0]; | |
161 | fTagKa[0] = 0.; | |
162 | } | |
163 | // tagging of the negative track | |
164 | if(TMath::Abs(fPdg[1])==211 || TMath::Abs(fPdg[1])==13 | |
165 | || TMath::Abs(fPdg[1])==11) { // pion,muon,electron | |
741f4d30 | 166 | fTagPi[1] = LinearInterpolation(PChild(1),kPiBinsPbPb,kPiBinWidthPbPb,kPiTagPiPbPb); |
3a9a3487 | 167 | fTagNid[1] = 1.-fTagPi[1]; |
168 | fTagKa[1] = 0.; | |
169 | fTagPr[1] = 0.; | |
170 | } | |
171 | if(TMath::Abs(fPdg[1])==321) { // kaon | |
741f4d30 | 172 | fTagKa[1] = LinearInterpolation(PChild(1),kKBinsPbPb,kKBinWidthPbPb,kKTagKPbPb); |
173 | fTagNid[1] = LinearInterpolation(PChild(1),kKBinsPbPb,kKBinWidthPbPb,kKTagNidPbPb); | |
3a9a3487 | 174 | if((fTagNid[1]+fTagKa[1])>1.) fTagNid[1] = 1.-fTagKa[1]; |
175 | fTagPi[1] = 1.-fTagNid[1]-fTagKa[1]; | |
176 | fTagPr[1] = 0.; | |
177 | } | |
178 | if(TMath::Abs(fPdg[1])==2212) { // proton | |
741f4d30 | 179 | fTagPr[1] = LinearInterpolation(PChild(1),kPBinsPbPb,kPBinWidthPbPb,kPTagPPbPb); |
180 | fTagNid[1] = LinearInterpolation(PChild(1),kPBinsPbPb,kPBinWidthPbPb,kPTagNidPbPb); | |
3a9a3487 | 181 | if((fTagNid[1]+fTagPr[1])>1.) fTagNid[1] = 1.-fTagPr[1]; |
182 | fTagPi[1] = 1.-fTagNid[1]-fTagPr[1]; | |
183 | fTagKa[1] = 0.; | |
184 | } | |
185 | } | |
186 | ||
187 | ||
188 | if(tofparampp) { | |
189 | // tagging of the positive track | |
190 | if(TMath::Abs(fPdg[0])==211 || TMath::Abs(fPdg[0])==13 | |
191 | || TMath::Abs(fPdg[0])==11) { // pion,muon,electron | |
741f4d30 | 192 | fTagPi[0] = LinearInterpolation(PChild(0),kPiBinsPP,kPiBinWidthPP,kPiTagPiPP); |
3a9a3487 | 193 | fTagNid[0] = 1.-fTagPi[0]; |
194 | fTagKa[0] = 0.; | |
195 | fTagPr[0] = 0.; | |
196 | } | |
197 | if(TMath::Abs(fPdg[0])==321) { // kaon | |
741f4d30 | 198 | fTagKa[0] = LinearInterpolation(PChild(0),kKBinsPP,kKBinWidthPP,kKTagKPP); |
199 | fTagNid[0] = LinearInterpolation(PChild(0),kKBinsPP,kKBinWidthPP,kKTagNidPP); | |
3a9a3487 | 200 | if((fTagNid[0]+fTagKa[0])>1.) fTagNid[0] = 1.-fTagKa[0]; |
201 | fTagPi[0] = 1.-fTagNid[0]-fTagKa[0]; | |
202 | fTagPr[0] = 0.; | |
203 | } | |
204 | if(TMath::Abs(fPdg[0])==2212) { // proton | |
741f4d30 | 205 | fTagPr[0] = LinearInterpolation(PChild(0),kPBinsPP,kPBinWidthPP,kPTagPPP); |
206 | fTagNid[0] = LinearInterpolation(PChild(0),kPBinsPP,kPBinWidthPP,kPTagNidPP); | |
3a9a3487 | 207 | if((fTagNid[0]+fTagPr[0])>1.) fTagNid[0] = 1.-fTagPr[0]; |
208 | fTagPi[0] = 1.-fTagNid[0]-fTagPr[0]; | |
209 | fTagKa[0] = 0.; | |
210 | } | |
211 | // tagging of the negative track | |
212 | if(TMath::Abs(fPdg[1])==211 || TMath::Abs(fPdg[1])==13 | |
213 | || TMath::Abs(fPdg[1])==11) { // pion,muon,electron | |
741f4d30 | 214 | fTagPi[1] = LinearInterpolation(PChild(1),kPiBinsPP,kPiBinWidthPP,kPiTagPiPP); |
3a9a3487 | 215 | fTagNid[1] = 1.-fTagPi[1]; |
216 | fTagKa[1] = 0.; | |
217 | fTagPr[1] = 0.; | |
218 | } | |
219 | if(TMath::Abs(fPdg[1])==321) { // kaon | |
741f4d30 | 220 | fTagKa[1] = LinearInterpolation(PChild(1),kKBinsPP,kKBinWidthPP,kKTagKPP); |
221 | fTagNid[1] = LinearInterpolation(PChild(1),kKBinsPP,kKBinWidthPP,kKTagNidPP); | |
3a9a3487 | 222 | if((fTagNid[1]+fTagKa[1])>1.) fTagNid[1] = 1.-fTagKa[1]; |
223 | fTagPi[1] = 1.-fTagNid[1]-fTagKa[1]; | |
224 | fTagPr[1] = 0.; | |
225 | } | |
226 | if(TMath::Abs(fPdg[1])==2212) { // proton | |
741f4d30 | 227 | fTagPr[1] = LinearInterpolation(PChild(1),kPBinsPP,kPBinWidthPP,kPTagPPP); |
228 | fTagNid[1] = LinearInterpolation(PChild(1),kPBinsPP,kPBinWidthPP,kPTagNidPP); | |
3a9a3487 | 229 | if((fTagNid[1]+fTagPr[1])>1.) fTagNid[1] = 1.-fTagPr[1]; |
230 | fTagPi[1] = 1.-fTagNid[1]-fTagPr[1]; | |
231 | fTagKa[1] = 0.; | |
232 | } | |
233 | } | |
234 | ||
235 | return; | |
236 | } | |
237 | //---------------------------------------------------------------------------- | |
238 | Double_t AliD0toKpi::ChildrenRelAngle() const { | |
239 | // relative angle between K and pi | |
240 | ||
241 | TVector3 mom0(fPx[0],fPy[0],fPz[0]); | |
242 | TVector3 mom1(fPx[1],fPy[1],fPz[1]); | |
243 | ||
244 | Double_t angle = mom0.Angle(mom1); | |
245 | ||
246 | return angle; | |
247 | } | |
248 | //---------------------------------------------------------------------------- | |
249 | void AliD0toKpi::ComputeWgts() { | |
250 | // calculate the weights for PID | |
251 | ||
252 | ||
253 | // assignement of the weights from PID | |
254 | fWgtAD0 = fTagKa[1]*(fTagPi[0]+fTagNid[0]); | |
255 | fWgtAD0bar = fTagKa[0]*(fTagPi[1]+fTagNid[1]); | |
256 | fWgtBD0 = fTagPi[0]*fTagNid[1]; | |
257 | fWgtBD0bar = fTagPi[1]*fTagNid[0]; | |
258 | fWgtCD0 = fTagNid[0]*fTagNid[1]; | |
259 | fWgtCD0bar = fTagNid[0]*fTagNid[1]; | |
260 | fWgtDD0 = 1.-fWgtAD0-fWgtBD0-fWgtCD0; | |
261 | fWgtDD0bar = 1.-fWgtAD0bar-fWgtBD0bar-fWgtCD0bar; | |
262 | ||
263 | /* | |
264 | cerr<<fWgtAD0<<" "<<fWgtAD0bar<<endl; | |
265 | cerr<<fWgtBD0<<" "<<fWgtBD0bar<<endl; | |
266 | cerr<<fWgtCD0<<" "<<fWgtCD0bar<<endl; | |
267 | ||
268 | if(fWgtAD0<0.) cerr<<"AliD0toKpi::ComputeWgts() Negative weight!!!\n"; | |
269 | if(fWgtAD0bar<0.) cerr<<"AliD0toKpi::ComputeWgts() Negative weight!!!\n"; | |
270 | if(fWgtBD0<0.) cerr<<"AliD0toKpi::ComputeWgts() Negative weight!!!\n"; | |
271 | if(fWgtBD0bar<0.) cerr<<"AliD0toKpi::ComputeWgts() Negative weight!!!\n"; | |
272 | if(fWgtCD0<0.) cerr<<"AliD0toKpi::ComputeWgts() Negative weight!!!\n"; | |
273 | if(fWgtCD0bar<0.) cerr<<"AliD0toKpi::ComputeWgts() Negative weight!!!\n"; | |
274 | */ | |
275 | ||
276 | return; | |
277 | } | |
278 | //---------------------------------------------------------------------------- | |
279 | void AliD0toKpi::CorrectWgt4BR(Double_t factor) { | |
280 | // correct weights of background from charm | |
281 | ||
282 | fWgtAD0 *= factor; | |
283 | fWgtAD0bar *= factor; | |
284 | fWgtBD0 *= factor; | |
285 | fWgtBD0bar *= factor; | |
286 | fWgtCD0 *= factor; | |
287 | fWgtCD0bar *= factor; | |
288 | fWgtDD0 *= factor; | |
289 | fWgtDD0bar *= factor; | |
290 | ||
291 | return; | |
292 | } | |
293 | //---------------------------------------------------------------------------- | |
294 | Double_t AliD0toKpi::CosPointing() const { | |
295 | // cosine of pointing angle in space | |
296 | ||
297 | TVector3 mom(Px(),Py(),Pz()); | |
298 | TVector3 flight(fV2x-fV1x,fV2y-fV1y,fV2z-fV1z); | |
299 | ||
300 | Double_t pta = mom.Angle(flight); | |
301 | ||
302 | return TMath::Cos(pta); | |
303 | } | |
304 | //---------------------------------------------------------------------------- | |
305 | Double_t AliD0toKpi::CosPointingXY() const { | |
306 | // cosine of pointing angle in transverse plane | |
307 | ||
308 | TVector3 momXY(Px(),Py(),0.); | |
309 | TVector3 flightXY(fV2x-fV1x,fV2y-fV1y,0.); | |
310 | ||
311 | Double_t ptaXY = momXY.Angle(flightXY); | |
312 | ||
313 | return TMath::Cos(ptaXY); | |
314 | } | |
315 | //---------------------------------------------------------------------------- | |
316 | void AliD0toKpi::CosThetaStar(Double_t &ctsD0,Double_t &ctsD0bar) const { | |
317 | // cosine of decay angle in the D0 rest frame | |
318 | ||
319 | Double_t pStar = TMath::Sqrt(TMath::Power(kMD0*kMD0-kMK*kMK-kMPi*kMPi,2.)-4.*kMK*kMK*kMPi*kMPi)/(2.*kMD0); | |
320 | ||
321 | Double_t beta = P()/Energy(); | |
322 | Double_t gamma = Energy()/kMD0; | |
323 | ||
324 | ctsD0 = (Ql(1)/gamma-beta*TMath::Sqrt(pStar*pStar+kMK*kMK))/pStar; | |
325 | // if(ctsD0 > 1.) { cerr<<"AliD0toKpi::CosThetaStar: > 1 "<<ctsD0<<"!\n"; } | |
326 | // if(ctsD0 < -1.) { cerr<<"AliD0toKpi::CosThetaStar: < -1 "<<ctsD0<<"!\n"; } | |
327 | ||
328 | ctsD0bar = (Ql(0)/gamma-beta*TMath::Sqrt(pStar*pStar+kMK*kMK))/pStar; | |
329 | // if(ctsD0bar > 1.) { cerr<<"AliD0toKpi::CosThetaStar: > 1 "<<ctsD0bar<<"!\n"; } | |
330 | // if(ctsD0bar < -1.) { cerr<<"AliD0toKpi::CosThetaStar: < -1 "<<ctsD0bar<<"!\n";} | |
331 | ||
332 | return; | |
333 | } | |
334 | //---------------------------------------------------------------------------- | |
335 | Double_t AliD0toKpi::Eta() const { | |
336 | // pseudorapidity of the D0 | |
337 | ||
338 | Double_t theta = TMath::Pi()/2.-TMath::ATan2(Pz(),Pt()); | |
339 | Double_t eta = -TMath::Log(TMath::Tan(theta/2.)); | |
340 | return eta; | |
341 | } | |
342 | //---------------------------------------------------------------------------- | |
343 | Double_t AliD0toKpi::EtaChild(Int_t child) const { | |
344 | // pseudorapidity of the decay tracks | |
345 | ||
346 | Double_t theta = TMath::Pi()/2.-TMath::ATan2(fPz[child],PtChild(child)); | |
347 | Double_t eta = -TMath::Log(TMath::Tan(theta/2.)); | |
348 | return eta; | |
349 | } | |
350 | //---------------------------------------------------------------------------- | |
351 | void AliD0toKpi::GetWgts(Double_t &WgtD0,Double_t &WgtD0bar,TString sample) | |
352 | const { | |
353 | // returns the weights for pid | |
354 | ||
355 | const char *sampleA = strstr(sample.Data(),"A"); | |
356 | const char *sampleB = strstr(sample.Data(),"B"); | |
357 | const char *sampleC = strstr(sample.Data(),"C"); | |
358 | const char *sampleD = strstr(sample.Data(),"D"); | |
359 | const char *sampleABCD = strstr(sample.Data(),"ABCD"); | |
360 | const char *sampleABC = strstr(sample.Data(),"ABC"); | |
361 | const char *sampleBC = strstr(sample.Data(),"BC"); | |
362 | ||
363 | if(sampleA) { WgtD0 = fWgtAD0; WgtD0bar = fWgtAD0bar; } | |
364 | if(sampleB) { WgtD0 = fWgtBD0; WgtD0bar = fWgtBD0bar; } | |
365 | if(sampleC) { WgtD0 = fWgtCD0; WgtD0bar = fWgtCD0bar; } | |
366 | if(sampleD) { WgtD0 = fWgtDD0; WgtD0bar = fWgtDD0bar; } | |
367 | if(sampleABCD) { | |
368 | WgtD0 = fWgtAD0+fWgtBD0+fWgtCD0+fWgtDD0; | |
369 | WgtD0bar = fWgtAD0bar+fWgtBD0bar+fWgtCD0bar+fWgtDD0bar; | |
370 | } | |
371 | if(sampleABC) { | |
372 | WgtD0 = fWgtAD0+fWgtBD0+fWgtCD0; | |
373 | WgtD0bar = fWgtAD0bar+fWgtBD0bar+fWgtCD0bar; | |
374 | } | |
375 | if(sampleBC) { | |
376 | WgtD0 = fWgtBD0+fWgtCD0; | |
377 | WgtD0bar = fWgtBD0bar+fWgtCD0bar; | |
378 | } | |
379 | ||
380 | ||
381 | if(fSignal) { | |
382 | if(fMum[0]==421) WgtD0bar = 0.; | |
383 | if(fMum[0]==-421) WgtD0 = 0.; | |
384 | } | |
385 | ||
386 | return; | |
387 | } | |
388 | //---------------------------------------------------------------------------- | |
389 | void AliD0toKpi::InvMass(Double_t &mD0,Double_t &mD0bar) const { | |
390 | // invariant mass as D0 and as D0bar | |
391 | ||
392 | Double_t energy[2]; | |
393 | ||
394 | // D0 -> K- Pi+ | |
395 | energy[1] = TMath::Sqrt(kMK*kMK+PChild(1)*PChild(1)); | |
396 | energy[0] = TMath::Sqrt(kMPi*kMPi+PChild(0)*PChild(0)); | |
397 | ||
398 | mD0 = TMath::Sqrt((energy[0]+energy[1])*(energy[0]+energy[1])-P()*P()); | |
399 | ||
400 | ||
401 | // D0bar -> K+ Pi- | |
402 | energy[0] = TMath::Sqrt(kMK*kMK+PChild(0)*PChild(0)); | |
403 | energy[1] = TMath::Sqrt(kMPi*kMPi+PChild(1)*PChild(1)); | |
404 | ||
405 | mD0bar = TMath::Sqrt((energy[0]+energy[1])*(energy[0]+energy[1])-P()*P()); | |
406 | ||
407 | return; | |
408 | ||
409 | } | |
410 | //---------------------------------------------------------------------------- | |
411 | Double_t AliD0toKpi::Ql(Int_t child) const { | |
412 | // longitudinal momentum of decay tracks w.r.t. to D0 momentum | |
413 | ||
414 | Double_t qL; | |
415 | TVector3 mom(fPx[child],fPy[child],fPz[child]); | |
416 | TVector3 momD(Px(),Py(),Pz()); | |
417 | ||
418 | qL = mom.Dot(momD)/momD.Mag(); | |
419 | ||
420 | return qL ; | |
421 | } | |
422 | //---------------------------------------------------------------------------- | |
423 | Double_t AliD0toKpi::Qt() const { | |
424 | // transverse momentum of decay tracks w.r.t. to D0 momentum | |
425 | ||
426 | TVector3 mom0(fPx[0],fPy[0],fPz[0]); | |
427 | TVector3 momD(Px(),Py(),Pz()); | |
428 | ||
429 | return mom0.Perp(momD); | |
430 | } | |
431 | //---------------------------------------------------------------------------- | |
432 | Bool_t AliD0toKpi::Select(const Double_t* cuts,Int_t& okD0,Int_t& okD0bar) | |
433 | const { | |
434 | // | |
435 | // This function compares the D0 with a set of cuts: | |
436 | // | |
437 | // cuts[0] = inv. mass half width [GeV] | |
438 | // cuts[1] = dca [micron] | |
439 | // cuts[2] = cosThetaStar | |
440 | // cuts[3] = pTK [GeV/c] | |
441 | // cuts[4] = pTPi [GeV/c] | |
442 | // cuts[5] = d0K [micron] upper limit! | |
443 | // cuts[6] = d0Pi [micron] upper limit! | |
444 | // cuts[7] = d0d0 [micron^2] | |
445 | // cuts[8] = cosThetaPoint | |
446 | // | |
447 | // If the D0/D0bar doesn't pass the cuts it sets the weights to 0 | |
448 | // If neither D0 nor D0bar pass the cuts return kFALSE | |
449 | // | |
450 | Double_t mD0,mD0bar,ctsD0,ctsD0bar; | |
451 | okD0=1; okD0bar=1; | |
452 | ||
453 | if(PtChild(1) < cuts[3] || PtChild(0) < cuts[4]) okD0 = 0; | |
454 | if(PtChild(0) < cuts[3] || PtChild(1) < cuts[4]) okD0bar = 0; | |
455 | if(!okD0 && !okD0bar) return kFALSE; | |
456 | ||
457 | if(TMath::Abs(Getd0Child(1)) > cuts[5] || | |
458 | TMath::Abs(Getd0Child(0)) > cuts[6]) okD0 = 0; | |
459 | if(TMath::Abs(Getd0Child(0)) > cuts[6] || | |
460 | TMath::Abs(Getd0Child(1)) > cuts[5]) okD0bar = 0; | |
461 | if(!okD0 && !okD0bar) return kFALSE; | |
462 | ||
463 | if(GetDCA() > cuts[1]) { okD0 = okD0bar = 0; return kFALSE; } | |
464 | ||
465 | InvMass(mD0,mD0bar); | |
466 | if(TMath::Abs(mD0-kMD0) > cuts[0]) okD0 = 0; | |
467 | if(TMath::Abs(mD0bar-kMD0) > cuts[0]) okD0bar = 0; | |
468 | if(!okD0 && !okD0bar) return kFALSE; | |
469 | ||
470 | CosThetaStar(ctsD0,ctsD0bar); | |
471 | if(TMath::Abs(ctsD0) > cuts[2]) okD0 = 0; | |
472 | if(TMath::Abs(ctsD0bar) > cuts[2]) okD0bar = 0; | |
473 | if(!okD0 && !okD0bar) return kFALSE; | |
474 | ||
475 | if(ProdImpParams() > cuts[7]) { okD0 = okD0bar = 0; return kFALSE; } | |
476 | ||
477 | if(CosPointing() < cuts[8]) { okD0 = okD0bar = 0; return kFALSE; } | |
478 | ||
479 | return kTRUE; | |
480 | } | |
481 | //----------------------------------------------------------------------------- | |
482 | void AliD0toKpi::SetPIDresponse(Double_t resp0[5],Double_t resp1[5]) { | |
483 | // Set combined PID detector response probabilities | |
484 | ||
485 | fPIDrespEl[0] = resp0[0]; | |
486 | fPIDrespEl[1] = resp1[0]; | |
487 | fPIDrespMu[0] = resp0[1]; | |
488 | fPIDrespMu[1] = resp1[1]; | |
489 | fPIDrespPi[0] = resp0[2]; | |
490 | fPIDrespPi[1] = resp1[2]; | |
491 | fPIDrespKa[0] = resp0[3]; | |
492 | fPIDrespKa[1] = resp1[3]; | |
493 | fPIDrespPr[0] = resp0[4]; | |
494 | fPIDrespPr[1] = resp1[4]; | |
495 | ||
496 | return; | |
497 | } | |
498 | //----------------------------------------------------------------------------- | |
7b84ea57 | 499 | void AliD0toKpi::DrawPIDinTOF(const Char_t * pidScheme) const { |
3a9a3487 | 500 | // Draw parameterized PID probabilities in TOF |
501 | ||
7b84ea57 | 502 | const char *tofparampbpb = strstr(pidScheme,"TOFparamPbPb"); |
503 | const char *tofparampp = strstr(pidScheme,"TOFparamPP"); | |
3a9a3487 | 504 | |
505 | TH2F* framePi = new TH2F("framePi","Tag probabilities for PIONS",2,0,2.5,2,0,1); | |
506 | framePi->SetXTitle("p [GeV/c]"); | |
507 | framePi->SetStats(0); | |
508 | TH2F* frameK = new TH2F("frameK","Tag probabilities for KAONS",2,0,2.5,2,0,1); | |
509 | frameK->SetXTitle("p [GeV/c]"); | |
510 | frameK->SetStats(0); | |
511 | TH2F* frameP = new TH2F("frameP","Tag probabilities for PROTONS",2,0,4.5,2,0,1); | |
512 | frameP->SetXTitle("p [GeV/c]"); | |
513 | frameP->SetStats(0); | |
514 | ||
741f4d30 | 515 | TH1F* hPiPi = new TH1F("hPiPi","Tag probabilities for PIONS",kPiBinsPbPb,0,2.5); |
516 | TH1F* hPiNid = new TH1F("hPiNid","Tag probabilities for PIONS",kPiBinsPbPb,0,2.5); | |
3a9a3487 | 517 | |
741f4d30 | 518 | TH1F* hKK = new TH1F("hKK","Tag probabilities for KAONS",kKBinsPbPb,0,2.5); |
519 | TH1F* hKNid = new TH1F("hKNid","Tag probabilities for KAONS",kKBinsPbPb,0,2.5); | |
520 | TH1F* hKPi = new TH1F("hKPi","Tag probabilities for KAONS",kKBinsPbPb,0,2.5); | |
3a9a3487 | 521 | |
741f4d30 | 522 | TH1F* hPP = new TH1F("hPP","Tag probabilities for PROTONS",kPBinsPbPb,0,4.5); |
523 | TH1F* hPNid = new TH1F("hPNid","Tag probabilities for PROTONS",kPBinsPbPb,0,4.5); | |
524 | TH1F* hPPi = new TH1F("hPPi","Tag probabilities for PROTONS",kPBinsPbPb,0,4.5); | |
3a9a3487 | 525 | |
526 | ||
527 | if(tofparampbpb) { | |
528 | ||
741f4d30 | 529 | for(Int_t i=1; i<=kPiBinsPbPb; i++) { |
530 | hPiPi->SetBinContent(i,kPiTagPiPbPb[i-1]); | |
531 | hPiNid->SetBinContent(i,kPiTagPiPbPb[i-1]+kPiTagNidPbPb[i-1]); | |
3a9a3487 | 532 | |
741f4d30 | 533 | hKK->SetBinContent(i,kKTagKPbPb[i-1]); |
534 | hKPi->SetBinContent(i,kKTagKPbPb[i-1]+kKTagPiPbPb[i-1]); | |
535 | hKNid->SetBinContent(i,kKTagKPbPb[i-1]+kKTagPiPbPb[i-1]+kKTagNidPbPb[i-1]); | |
3a9a3487 | 536 | } |
741f4d30 | 537 | for(Int_t i=1; i<=kPBinsPbPb; i++) { |
538 | hPP->SetBinContent(i,kPTagPPbPb[i-1]); | |
539 | hPPi->SetBinContent(i,kPTagPPbPb[i-1]+kPTagPiPbPb[i-1]); | |
540 | hPNid->SetBinContent(i,kPTagPPbPb[i-1]+kPTagPiPbPb[i-1]+kPTagNidPbPb[i-1]); | |
3a9a3487 | 541 | } |
542 | ||
543 | } else if(tofparampp) { | |
544 | ||
741f4d30 | 545 | for(Int_t i=1; i<=kPiBinsPP; i++) { |
546 | hPiPi->SetBinContent(i,kPiTagPiPP[i-1]); | |
547 | hPiNid->SetBinContent(i,kPiTagPiPP[i-1]+kPiTagNidPP[i-1]); | |
3a9a3487 | 548 | |
741f4d30 | 549 | hKK->SetBinContent(i,kKTagKPP[i-1]); |
550 | hKPi->SetBinContent(i,kKTagKPP[i-1]+kKTagPiPP[i-1]); | |
551 | hKNid->SetBinContent(i,kKTagKPP[i-1]+kKTagPiPP[i-1]+kKTagNidPP[i-1]); | |
3a9a3487 | 552 | } |
741f4d30 | 553 | for(Int_t i=1; i<=kPBinsPP; i++) { |
554 | hPP->SetBinContent(i,kPTagPPP[i-1]); | |
555 | hPPi->SetBinContent(i,kPTagPPP[i-1]+kPTagPiPP[i-1]); | |
556 | hPNid->SetBinContent(i,kPTagPPP[i-1]+kPTagPiPP[i-1]+kPTagNidPP[i-1]); | |
3a9a3487 | 557 | } |
558 | ||
559 | } | |
560 | ||
561 | ||
562 | TCanvas* c = new TCanvas("c","Parameterized PID in TOF",0,0,1000,400); | |
563 | c->Divide(3,1); | |
564 | c->cd(1); | |
565 | framePi->Draw(); | |
566 | hPiNid->SetFillColor(18); hPiNid->Draw("same"); | |
567 | hPiPi->SetFillColor(4); hPiPi->Draw("same"); | |
568 | TPaveLabel* pav1 = new TPaveLabel(1,.2,1.4,.3,"#pi"); | |
569 | pav1->SetBorderSize(0); | |
570 | pav1->Draw("same"); | |
571 | TPaveLabel* pav2 = new TPaveLabel(1,.8,1.8,.9,"non-id"); | |
572 | pav2->SetBorderSize(0); | |
573 | pav2->Draw("same"); | |
574 | ||
575 | c->cd(2); | |
576 | frameK->Draw(); | |
577 | hKNid->SetFillColor(18); hKNid->Draw("same"); | |
578 | hKPi->SetFillColor(4); hKPi->Draw("same"); | |
579 | hKK->SetFillColor(7); hKK->Draw("same"); | |
580 | TPaveLabel* pav3 = new TPaveLabel(1,.2,1.5,.3,"K"); | |
581 | pav3->SetBorderSize(0); | |
582 | pav3->Draw("same"); | |
583 | TPaveLabel* pav4 = new TPaveLabel(1,.8,1.8,.9,"non-id"); | |
584 | pav4->SetBorderSize(0); | |
585 | pav4->Draw("same"); | |
586 | TPaveLabel* pav5 = new TPaveLabel(.4,.5,.8,.6,"#pi"); | |
587 | pav5->SetBorderSize(0); | |
588 | pav5->Draw("same"); | |
589 | ||
590 | c->cd(3); | |
591 | frameP->Draw(); | |
592 | hPNid->SetFillColor(18); hPNid->Draw("same"); | |
593 | hPPi->SetFillColor(4); hPPi->Draw("same"); | |
594 | hPP->SetFillColor(3); hPP->Draw("same"); | |
595 | TPaveLabel* pav6 = new TPaveLabel(1,.2,1.5,.3,"p"); | |
596 | pav6->SetBorderSize(0); | |
597 | pav6->Draw("same"); | |
598 | TPaveLabel* pav7 = new TPaveLabel(1,.8,2.6,.9,"non-id"); | |
599 | pav7->SetBorderSize(0); | |
600 | pav7->Draw("same"); | |
601 | TPaveLabel* pav8 = new TPaveLabel(.2,.5,1,.6,"#pi"); | |
602 | pav8->SetBorderSize(0); | |
603 | pav8->Draw("same"); | |
604 | ||
605 | ||
606 | return; | |
607 | } | |
608 | //---------------------------------------------------------------------------- | |
609 | Double_t AliD0toKpi::LinearInterpolation(Double_t p,Int_t nBins,Double_t Bin, | |
610 | const Double_t *values) const { | |
611 | // a linear interpolation method | |
612 | ||
613 | Double_t value=0; | |
614 | Double_t slope; | |
615 | ||
616 | if(p<0.5*Bin) { | |
617 | value = values[0]; | |
618 | } else if(p>=(nBins-0.5)*Bin) { | |
619 | slope = (2*values[nBins-1]-values[nBins-2]-values[nBins-3])/Bin/2; | |
620 | value = values[nBins-2]+slope*(p-Bin*(nBins-1.5)); | |
621 | } else { | |
622 | for(Int_t i=0; i<nBins; i++) { | |
623 | if(p<(i+0.5)*Bin) { | |
624 | slope = (values[i]-values[i-1])/Bin; | |
625 | value = values[i-1]+slope*(p-Bin*(i-0.5)); | |
626 | break; | |
627 | } | |
628 | } | |
629 | } | |
630 | ||
631 | if(value<0.) value=0.; | |
632 | if(value>1.) value=1.; | |
633 | ||
634 | return value; | |
635 | } | |
636 | //---------------------------------------------------------------------------- | |
637 | ||
638 | ||
639 | ||
640 | ||
641 | ||
642 | ||
643 | /* | |
644 | //____________________________________________________________________________ | |
645 | void AliD0toKpi::SetPtWgts4pp() { | |
646 | // Correct pt distribution in order to reproduce MNR pt slope | |
647 | // (for pp generated with PYTHIA min. bias) | |
648 | ||
649 | if(TMath::Abs(fMum[0]) != 421 && TMath::Abs(fMum[1]) != 421 && | |
650 | TMath::Abs(fMum[0]) != 411 && TMath::Abs(fMum[1]) != 411) return; | |
651 | ||
652 | Double_t ptWgt = 1.; | |
653 | ptWgt = 2.05-0.47*Pt()+0.02*Pt()*Pt(); | |
654 | if(Pt() >= 5.) ptWgt = 0.56*TMath::Exp(-0.12*Pt()); | |
655 | ||
656 | fWgtAD0 *= ptWgt; | |
657 | fWgtAD0bar *= ptWgt; | |
658 | fWgtBD0 *= ptWgt; | |
659 | fWgtBD0bar *= ptWgt; | |
660 | fWgtCD0 *= ptWgt; | |
661 | fWgtCD0bar *= ptWgt; | |
662 | fWgtDD0 *= ptWgt; | |
663 | fWgtDD0bar *= ptWgt; | |
664 | ||
665 | return; | |
666 | } | |
667 | //____________________________________________________________________________ | |
668 | */ | |
669 | ||
670 | ||
671 | ||
672 | ||
673 | ||
674 | ||
675 | ||
676 | ||
677 |