Adding kaon train macros
[u/mrichter/AliRoot.git] / PWGCF / FEMTOSCOPY / AliFemtoUser / AliFemtoESDTrackCut.cxx
CommitLineData
76ce4b5b 1/*
2***************************************************************************
3*
4* $Id$
5*
6*
7***************************************************************************
8*
9*
10*
11*
12***************************************************************************
13*
14* $Log$
15* Revision 1.3 2007/05/22 09:01:42 akisiel
16* Add the possibiloity to save cut settings in the ROOT file
17*
18* Revision 1.2 2007/05/21 10:38:25 akisiel
19* More coding rule conformance
20*
21* Revision 1.1 2007/05/16 10:25:06 akisiel
22* Making the directory structure of AliFemtoUser flat. All files go into one common directory
23*
24* Revision 1.4 2007/05/03 09:46:10 akisiel
25* Fixing Effective C++ warnings
26*
27* Revision 1.3 2007/04/27 07:25:59 akisiel
28* Make revisions needed for compilation from the main AliRoot tree
29*
30* Revision 1.1.1.1 2007/04/25 15:38:41 panos
31* Importing the HBT code dir
32*
33* Revision 1.4 2007-04-03 16:00:08 mchojnacki
34* Changes to iprove memory managing
35*
36* Revision 1.3 2007/03/13 15:30:03 mchojnacki
37* adding reader for simulated data
38*
39* Revision 1.2 2007/03/08 14:58:03 mchojnacki
40* adding some alice stuff
41*
42* Revision 1.1.1.1 2007/03/07 10:14:49 mchojnacki
43* First version on CVS
44*
45**************************************************************************/
46
47#include "AliFemtoESDTrackCut.h"
48#include <cstdio>
49
50#ifdef __ROOT__
51ClassImp(AliFemtoESDTrackCut)
52#endif
53
54
55// electron
56// 0.13 - 1.8
57// 0 7.594129e-02 8.256141e-03
58// 1 -5.535827e-01 8.170825e-02
59// 2 1.728591e+00 3.104210e-01
60// 3 -2.827893e+00 5.827802e-01
61// 4 2.503553e+00 5.736207e-01
62// 5 -1.125965e+00 2.821170e-01
63// 6 2.009036e-01 5.438876e-02
64
65// pion
66// 0.13 - 2.0
67// 0 1.063457e+00 8.872043e-03
68// 1 -4.222208e-01 2.534402e-02
69// 2 1.042004e-01 1.503945e-02
70
71// kaon
72// 0.18 - 2.0
73// 0 -7.289406e-02 1.686074e-03
74// 1 4.415666e-01 1.143939e-02
75// 2 -2.996790e-01 1.840964e-02
76// 3 6.704652e-02 7.783990e-03
77
78// proton
79// 0.26 - 2.0
80// 0 -3.730200e-02 2.347311e-03
81// 1 1.163684e-01 1.319316e-02
82// 2 8.354116e-02 1.997948e-02
83// 3 -4.608098e-02 8.336400e-03
84
85
86 AliFemtoESDTrackCut::AliFemtoESDTrackCut() :
87 fCharge(0),
88 fLabel(0),
89 fStatus(0),
90 fPIDMethod(knSigma),
91 fminTPCclsF(0),
92 fminTPCncls(0),
93 fminITScls(0),
94 fMaxITSchiNdof(1000.0),
95 fMaxTPCchiNdof(1000.0),
96 fMaxSigmaToVertex(1000.0),
97 fNTracksPassed(0),
98 fNTracksFailed(0),
99 fRemoveKinks(kFALSE),
100 fRemoveITSFake(kFALSE),
101 fMostProbable(0),
102 fMaxImpactXY(1000.0),
103 fMaxImpactZ(1000.0),
104 fMaxImpactXYPtOff(1000.0),
105 fMaxImpactXYPtNrm(1000.0),
106 fMaxImpactXYPtPow(1000.0),
107 fMinPforTOFpid(0.0),
108 fMaxPforTOFpid(10000.0),
109 fMinPforTPCpid(0.0),
110 fMaxPforTPCpid(10000.0),
111 fMinPforITSpid(0.0),
112 fMaxPforITSpid(10000.0)
113{
114 // Default constructor
115 fNTracksPassed = fNTracksFailed = 0;
116 fCharge = 0; // takes both charges 0
117 fPt[0]=0.0; fPt[1] = 100.0;//100
118 fRapidity[0]=-2; fRapidity[1]=2;//-2 2
119 fEta[0]=-2; fEta[1]=2;//-2 2
120 fPidProbElectron[0]=-1;fPidProbElectron[1]=2;
121 fPidProbPion[0]=-1; fPidProbPion[1]=2;
122 fPidProbKaon[0]=-1;fPidProbKaon[1]=2;
123 fPidProbProton[0]=-1;fPidProbProton[1]=2;
124 fPidProbMuon[0]=-1;fPidProbMuon[1]=2;
125 fLabel=false;
126 fStatus=0;
127 fminTPCclsF=0;
128 fminITScls=0;
129 fPIDMethod=knSigma;
130}
131//------------------------------
132AliFemtoESDTrackCut::~AliFemtoESDTrackCut(){
133 /* noop */
134}
135//------------------------------
136bool AliFemtoESDTrackCut::Pass(const AliFemtoTrack* track)
137{
138 // test the particle and return
139 // true if it meets all the criteria
140 // false if it doesn't meet at least one of the criteria
141 float tMost[5];
142
143 //cout<<"AliFemtoESD cut"<<endl;
144 //cout<<fPidProbPion[0]<<" < pi ="<<track->PidProbPion()<<" <"<<fPidProbPion[1]<<endl;
145 if (fStatus!=0)
146 {
147 //cout<<" status "<<track->Label()<<" "<<track->Flags()<<" "<<track->TPCnclsF()<<" "<<track->ITSncls()<<endl;
148 if ((track->Flags()&fStatus)!=fStatus)
149 {
150 // cout<<track->Flags()<<" "<<fStatus<<" no go through status"<<endl;
151 return false;
152 }
153
154 }
155 if (fRemoveKinks) {
156 if ((track->KinkIndex(0)) || (track->KinkIndex(1)) || (track->KinkIndex(2)))
157 return false;
158 }
159 if (fRemoveITSFake) {
160 if (track->ITSncls() < 0)
161 return false;
162 }
163 if (fminTPCclsF>track->TPCnclsF())
164 {
165 //cout<<" No go because TPC Number of ClsF"<<fminTPCclsF<< " "<<track->TPCnclsF()<<endl;
166 return false;
167 }
168 if (fminTPCncls>track->TPCncls())
169 {
170 //cout<<" No go because TPC Number of ClsF"<<fminTPCclsF<< " "<<track->TPCnclsF()<<endl;
171 return false;
172 }
173 if (fminITScls>track->ITSncls())
174 {
175 //cout<<" No go because ITS Number of Cls"<<fminITScls<< " "<<track->ITSncls()<<endl;
176 return false;
177 }
178
179 if (fMaxImpactXY < TMath::Abs(track->ImpactD()))
180 return false;
181
182 if (fMaxImpactZ < TMath::Abs(track->ImpactZ()))
183 return false;
184
185 if (fMaxSigmaToVertex < track->SigmaToVertex()) {
186 return false;
187 }
188
189 if (track->ITSncls() > 0)
190 if ((track->ITSchi2()/track->ITSncls()) > fMaxITSchiNdof) {
191 return false;
192 }
193
194 if (track->TPCncls() > 0)
195 if ((track->TPCchi2()/track->TPCncls()) > fMaxTPCchiNdof) {
196 return false;
197 }
198
199 if (fLabel)
200 {
201 //cout<<"labels"<<endl;
202 if(track->Label()<0)
203 {
204 fNTracksFailed++;
205 // cout<<"No Go Through the cut"<<endl;
206 // cout<<fLabel<<" Label="<<track->Label()<<endl;
207 return false;
208 }
209 }
210 if (fCharge!=0)
211 {
212 //cout<<"AliFemtoESD cut ch "<<endl;
213 //cout<<fCharge<<" Charge="<<track->Charge()<<endl;
214 if (track->Charge()!= fCharge)
215 {
216 fNTracksFailed++;
217 // cout<<"No Go Through the cut"<<endl;
218 // cout<<fCharge<<" Charge="<<track->Charge()<<endl;
219 return false;
220 }
221 }
222 Bool_t tTPCPidIn = (track->Flags()&AliFemtoTrack::kTPCpid)>0;
223 Bool_t tITSPidIn = (track->Flags()&AliFemtoTrack::kITSpid)>0;
224 Bool_t tTOFPidIn = (track->Flags()&AliFemtoTrack::kTOFpid)>0;
225
226 if(fMinPforTOFpid > 0 && track->P().Mag() > fMinPforTOFpid &&
227 track->P().Mag() < fMaxPforTOFpid && !tTOFPidIn)
228 {
229 fNTracksFailed++;
230 return false;
231 }
232
233 if(fMinPforTPCpid > 0 && track->P().Mag() > fMinPforTPCpid &&
234 track->P().Mag() < fMaxPforTPCpid && !tTPCPidIn)
235 {
236 fNTracksFailed++;
237 return false;
238 }
239
240 if(fMinPforITSpid > 0 && track->P().Mag() > fMinPforITSpid &&
241 track->P().Mag() < fMaxPforITSpid && !tITSPidIn)
242 {
243 fNTracksFailed++;
244 return false;
245 }
246
247
248 float tEnergy = ::sqrt(track->P().Mag2()+fMass*fMass);
249 float tRapidity = 0.5*::log((tEnergy+track->P().z())/(tEnergy-track->P().z()));
250 float tPt = ::sqrt((track->P().x())*(track->P().x())+(track->P().y())*(track->P().y()));
251 float tEta = track->P().PseudoRapidity();
252
253 if (fMaxImpactXYPtOff < 999.0) {
254 if ((fMaxImpactXYPtOff + fMaxImpactXYPtNrm*TMath::Power(tPt, fMaxImpactXYPtPow)) < TMath::Abs(track->ImpactD())) {
255 fNTracksFailed++;
256 return false;
257 }
258 }
259
260 if ((tRapidity<fRapidity[0])||(tRapidity>fRapidity[1]))
261 {
262 fNTracksFailed++;
263 //cout<<"No Go Through the cut"<<endl;
264 //cout<<fRapidity[0]<<" < Rapidity ="<<tRapidity<<" <"<<fRapidity[1]<<endl;
265 return false;
266 }
267 if ((tEta<fEta[0])||(tEta>fEta[1]))
268 {
269 fNTracksFailed++;
270 //cout<<"No Go Through the cut"<<endl;
271 //cout<<fEta[0]<<" < Eta ="<<tEta<<" <"<<fEta[1]<<endl;
272 return false;
273 }
274 if ((tPt<fPt[0])||(tPt>fPt[1]))
275 {
276 fNTracksFailed++;
277 //cout<<"No Go Through the cut"<<endl;
278 //cout<<fPt[0]<<" < Pt ="<<Pt<<" <"<<fPt[1]<<endl;
279 return false;
280 }
281 // cout << "Track has pids: "
282 // << track->PidProbElectron() << " "
283 // << track->PidProbMuon() << " "
284 // << track->PidProbPion() << " "
285 // << track->PidProbKaon() << " "
286 // << track->PidProbProton() << " "
287 // << track->PidProbElectron()+track->PidProbMuon()+track->PidProbPion()+track->PidProbKaon()+track->PidProbProton() << endl;
288
289
290 if ((track->PidProbElectron()<fPidProbElectron[0])||(track->PidProbElectron()>fPidProbElectron[1]))
291 {
292 fNTracksFailed++;
293 //cout<<"No Go Through the cut"<<endl;
294 //cout<<fPidProbElectron[0]<<" < e ="<<track->PidProbElectron()<<" <"<<fPidProbElectron[1]<<endl;
295 return false;
296 }
297 if ((track->PidProbPion()<fPidProbPion[0])||(track->PidProbPion()>fPidProbPion[1]))
298 {
299 fNTracksFailed++;
300 //cout<<"No Go Through the cut"<<endl;
301 //cout<<fPidProbPion[0]<<" < pi ="<<track->PidProbPion()<<" <"<<fPidProbPion[1]<<endl;
302 return false;
303 }
304 if ((track->PidProbKaon()<fPidProbKaon[0])||(track->PidProbKaon()>fPidProbKaon[1]))
305 {
306 fNTracksFailed++;
307 //cout<<"No Go Through the cut"<<endl;
308 //cout<<fPidProbKaon[0]<<" < k ="<<track->PidProbKaon()<<" <"<<fPidProbKaon[1]<<endl;
309 return false;
310 }
311 if ((track->PidProbProton()<fPidProbProton[0])||(track->PidProbProton()>fPidProbProton[1]))
312 {
313 fNTracksFailed++;
314 //cout<<"No Go Through the cut"<<endl;
315 //cout<<fPidProbProton[0]<<" < p ="<<track->PidProbProton()<<" <"<<fPidProbProton[1]<<endl;
316 return false;
317 }
318 if ((track->PidProbMuon()<fPidProbMuon[0])||(track->PidProbMuon()>fPidProbMuon[1]))
319 {
320 fNTracksFailed++;
321 //cout<<"No Go Through the cut"<<endl;
322 //cout<<fPidProbMuon[0]<<" < mi="<<track->PidProbMuon()<<" <"<<fPidProbMuon[1]<<endl;
323 return false;
324 }
325
326 if (fMostProbable) {
327 int imost=0;
328 tMost[0] = track->PidProbElectron()*PidFractionElectron(track->P().Mag());
329 tMost[1] = 0.0;
330 tMost[2] = track->PidProbPion()*PidFractionPion(track->P().Mag());
331 tMost[3] = track->PidProbKaon()*PidFractionKaon(track->P().Mag());
332 tMost[4] = track->PidProbProton()*PidFractionProton(track->P().Mag());
333 float ipidmax = 0.0;
334
335
336 //****N Sigma Method****
337 if(fPIDMethod==0){
338 // Looking for pions
339 if (fMostProbable == 2) {
ce7b3d98 340 if (IsPionNSigma(track->Pt(), track->NSigmaTPCPi(), track->NSigmaTOFPi()))
76ce4b5b 341 imost = 2;
342
343 }
344 else if (fMostProbable == 3) {
ce7b3d98 345 if (IsKaonNSigma(track->Pt(), track->NSigmaTPCK(), track->NSigmaTOFK())){
76ce4b5b 346 imost = 3;
347 }
348
349 }
350 else if (fMostProbable == 4) { // proton nsigma-PID required contour adjusting
ce7b3d98 351 if (IsProtonNSigma(track->Pt(), track->NSigmaTPCP(), track->NSigmaTOFP()) && IsProtonTPCdEdx(track->P().Mag(), track->TPCsignal()))
76ce4b5b 352 imost = 4;
353 }
354
355 }
356
357 //****Contour Method****
358 if(fPIDMethod==1){
359 for (int ip=0; ip<5; ip++)
360 if (tMost[ip] > ipidmax) { ipidmax = tMost[ip]; imost = ip; };
361
362 // Looking for pions
363 if (fMostProbable == 2) {
364 if (imost == 2) {
365 // Using the TPC to reject non-pions
366 if (!(IsPionTPCdEdx(track->P().Mag(), track->TPCsignal())))
367 imost = 0;
368 if (0) {
369 // Using the TOF to reject non-pions
370 if (track->P().Mag() < 0.6) {
371 if (tTOFPidIn)
372 if (!IsPionTOFTime(track->P().Mag(), track->TOFpionTime()))
373 imost = 0;
374 }
375 else {
376 if (tTOFPidIn) {
377 if (!IsPionTOFTime(track->P().Mag(), track->TOFpionTime()))
378 imost = 0;
379 }
380 else {
381 imost = 0;
382 }
383 }
384 }
385 }
386 }
387
388 // Looking for kaons
389 else if (fMostProbable == 3) {
390 // if (imost == 3) {
391 // Using the TPC to reject non-kaons
392 if (track->P().Mag() < 0.6) {
393 if (!(IsKaonTPCdEdx(track->P().Mag(), track->TPCsignal())))
394 imost = 0;
395 else imost = 3;
396 if (1) {
397 // Using the TOF to reject non-kaons
398 if (tTOFPidIn)
399 if (!IsKaonTOFTime(track->P().Mag(), track->TOFkaonTime()))
400 imost = 0;
401 }
402 }
403 else {
404 if (1) {
405 if (tTOFPidIn) {
406 if (!IsKaonTOFTime(track->P().Mag(), track->TOFkaonTime()))
407 imost = 0;
408 else
409 imost = 3;
410 }
411 else {
412 if (!(IsKaonTPCdEdx(track->P().Mag(), track->TPCsignal())))
413 imost = 0;
414 else
415 imost = 3;
416 }
417 }
418 }
419 // }
420 }
421
422 // Looking for protons
423 else if (fMostProbable == 4) {
424 // if (imost == 3) {
425 // Using the TPC to reject non-kaons
426 if (track->P().Mag() < 0.8) {
427 if (!(IsProtonTPCdEdx(track->P().Mag(), track->TPCsignal())))
428 imost = 0;
429 else imost = 4;
430 if (0) {
431 // Using the TOF to reject non-kaons
432 if (tTOFPidIn)
433 if (!IsKaonTOFTime(track->P().Mag(), track->TOFkaonTime()))
434 imost = 0;
435 }
436 }
437 else {
438 if (0) {
439 if (tTOFPidIn) {
440 if (!IsKaonTOFTime(track->P().Mag(), track->TOFkaonTime()))
441 imost = 0;
442 else
443 imost = 3;
444 }
445 else {
446 if (!(IsKaonTPCdEdx(track->P().Mag(), track->TPCsignal())))
447 imost = 0;
448 else
449 imost = 3;
450 }
451 }
452 }
453 // }
454 }
455 }
456 if (imost != fMostProbable) return false;
457 }
458
459 //fan
460 //cout<<"****** Go Through the cut ******"<<endl;
461 // cout<<fLabel<<" Label="<<track->Label()<<endl;
462 // cout<<fCharge<<" Charge="<<track->Charge()<<endl;
463 // cout<<fPt[0]<<" < Pt ="<<Pt<<" <"<<fPt[1]<<endl;
464 //cout<<fRapidity[0]<<" < Rapidity ="<<tRapidity<<" <"<<fRapidity[1]<<endl;
465 //cout<<fPidProbElectron[0]<<" < e="<<track->PidProbElectron()<<" <"<<fPidProbElectron[1]<<endl;
466 //cout<<fPidProbPion[0]<<" < pi="<<track->PidProbPion()<<" <"<<fPidProbPion[1]<<endl;
467 //cout<<fPidProbKaon[0]<<" < k="<<track->PidProbKaon()<<" <"<<fPidProbKaon[1]<<endl;
468 //cout<<fPidProbProton[0]<<" < p="<<track->PidProbProton()<<" <"<<fPidProbProton[1]<<endl;
469 //cout<<fPidProbMuon[0]<<" < mi="<<track->PidProbMuon()<<" <"<<fPidProbMuon[1]<<endl;
470 fNTracksPassed++ ;
471 return true;
472
473
474}
475//------------------------------
476AliFemtoString AliFemtoESDTrackCut::Report()
477{
478 // Prepare report from the execution
479 string tStemp;
480 char tCtemp[100];
481 snprintf(tCtemp , 100, "Particle mass:\t%E\n",this->Mass());
482 tStemp=tCtemp;
483 snprintf(tCtemp , 100, "Particle charge:\t%d\n",fCharge);
484 tStemp+=tCtemp;
485 snprintf(tCtemp , 100, "Particle pT:\t%E - %E\n",fPt[0],fPt[1]);
486 tStemp+=tCtemp;
487 snprintf(tCtemp , 100, "Particle rapidity:\t%E - %E\n",fRapidity[0],fRapidity[1]);
488 tStemp+=tCtemp;
489 snprintf(tCtemp , 100, "Particle eta:\t%E - %E\n",fEta[0],fEta[1]);
490 tStemp+=tCtemp;
491 snprintf(tCtemp , 100, "Number of tracks which passed:\t%ld Number which failed:\t%ld\n",fNTracksPassed,fNTracksFailed);
492 tStemp += tCtemp;
493 AliFemtoString returnThis = tStemp;
494 return returnThis;
495}
496TList *AliFemtoESDTrackCut::ListSettings()
497{
498 // return a list of settings in a writable form
499 TList *tListSetttings = new TList();
500 char buf[200];
501 snprintf(buf, 200, "AliFemtoESDTrackCut.mass=%f", this->Mass());
502 tListSetttings->AddLast(new TObjString(buf));
503
504 snprintf(buf, 200, "AliFemtoESDTrackCut.charge=%i", fCharge);
505 tListSetttings->AddLast(new TObjString(buf));
506 snprintf(buf, 200, "AliFemtoESDTrackCut.pidprobpion.minimum=%f", fPidProbPion[0]);
507 tListSetttings->AddLast(new TObjString(buf));
508 snprintf(buf, 200, "AliFemtoESDTrackCut.pidprobpion.maximum=%f", fPidProbPion[1]);
509 tListSetttings->AddLast(new TObjString(buf));
510 snprintf(buf, 200, "AliFemtoESDTrackCut.pidprobkaon.minimum=%f", fPidProbKaon[0]);
511 tListSetttings->AddLast(new TObjString(buf));
512 snprintf(buf, 200, "AliFemtoESDTrackCut.pidprobkaon.maximum=%f", fPidProbKaon[1]);
513 tListSetttings->AddLast(new TObjString(buf));
514 snprintf(buf, 200, "AliFemtoESDTrackCut.pidprobproton.minimum=%f", fPidProbProton[0]);
515 tListSetttings->AddLast(new TObjString(buf));
516 snprintf(buf, 200, "AliFemtoESDTrackCut.pidprobproton.maximum=%f", fPidProbProton[1]);
517 tListSetttings->AddLast(new TObjString(buf));
518 snprintf(buf, 200, "AliFemtoESDTrackCut.pidprobelectron.minimum=%f", fPidProbElectron[0]);
519 tListSetttings->AddLast(new TObjString(buf));
520 snprintf(buf, 200, "AliFemtoESDTrackCut.pidprobelectron.maximum=%f", fPidProbElectron[1]);
521 tListSetttings->AddLast(new TObjString(buf));
522 snprintf(buf, 200, "AliFemtoESDTrackCut.pidprobMuon.minimum=%f", fPidProbMuon[0]);
523 tListSetttings->AddLast(new TObjString(buf));
524 snprintf(buf, 200, "AliFemtoESDTrackCut.pidprobMuon.maximum=%f", fPidProbMuon[1]);
525 tListSetttings->AddLast(new TObjString(buf));
526 snprintf(buf, 200, "AliFemtoESDTrackCut.minimumtpcclusters=%i", fminTPCclsF);
527 tListSetttings->AddLast(new TObjString(buf));
528 snprintf(buf, 200, "AliFemtoESDTrackCut.minimumitsclusters=%i", fminTPCclsF);
529 tListSetttings->AddLast(new TObjString(buf));
530 snprintf(buf, 200, "AliFemtoESDTrackCut.pt.minimum=%f", fPt[0]);
531 tListSetttings->AddLast(new TObjString(buf));
532 snprintf(buf, 200, "AliFemtoESDTrackCut.pt.maximum=%f", fPt[1]);
533 tListSetttings->AddLast(new TObjString(buf));
534 snprintf(buf, 200, "AliFemtoESDTrackCut.rapidity.minimum=%f", fRapidity[0]);
535 tListSetttings->AddLast(new TObjString(buf));
536 snprintf(buf, 200, "AliFemtoESDTrackCut.rapidity.maximum=%f", fRapidity[1]);
537 tListSetttings->AddLast(new TObjString(buf));
538 snprintf(buf, 200, "AliFemtoESDTrackCut.removekinks=%i", fRemoveKinks);
539 tListSetttings->AddLast(new TObjString(buf));
540 snprintf(buf, 200, "AliFemtoESDTrackCut.maxitschindof=%f", fMaxITSchiNdof);
541 tListSetttings->AddLast(new TObjString(buf));
542 snprintf(buf, 200, "AliFemtoESDTrackCut.maxtpcchindof=%f", fMaxTPCchiNdof);
543 tListSetttings->AddLast(new TObjString(buf));
544 snprintf(buf, 200, "AliFemtoESDTrackCut.maxsigmatovertex=%f", fMaxSigmaToVertex);
545 tListSetttings->AddLast(new TObjString(buf));
546 snprintf(buf, 200, "AliFemtoESDTrackCut.maximpactxy=%f", fMaxImpactXY);
547 tListSetttings->AddLast(new TObjString(buf));
548 snprintf(buf, 200, "AliFemtoESDTrackCut.maximpactz=%f", fMaxImpactZ);
549 tListSetttings->AddLast(new TObjString(buf));
550 if (fMostProbable) {
551 if (fMostProbable == 2)
552 snprintf(buf, 200, "AliFemtoESDTrackCut.mostprobable=%s", "Pion");
553 if (fMostProbable == 3)
554 snprintf(buf, 200, "AliFemtoESDTrackCut.mostprobable=%s", "Kaon");
555 if (fMostProbable == 4)
556 snprintf(buf, 200, "AliFemtoESDTrackCut.mostprobable=%s", "Proton");
557 tListSetttings->AddLast(new TObjString(buf));
558 }
559 return tListSetttings;
560}
561void AliFemtoESDTrackCut::SetRemoveKinks(const bool& flag)
562{
563 fRemoveKinks = flag;
564}
565
566void AliFemtoESDTrackCut::SetRemoveITSFake(const bool& flag)
567{
568 fRemoveITSFake = flag;
569}
570
571// electron
572// 0.13 - 1.8
573// 0 7.594129e-02 8.256141e-03
574// 1 -5.535827e-01 8.170825e-02
575// 2 1.728591e+00 3.104210e-01
576// 3 -2.827893e+00 5.827802e-01
577// 4 2.503553e+00 5.736207e-01
578// 5 -1.125965e+00 2.821170e-01
579// 6 2.009036e-01 5.438876e-02
580float AliFemtoESDTrackCut::PidFractionElectron(float mom) const
581{
582 // Provide a parameterized fraction of electrons dependent on momentum
583 if (mom<0.13)
584 return (7.594129e-02
585 -5.535827e-01*0.13
586 +1.728591e+00*0.13*0.13
587 -2.827893e+00*0.13*0.13*0.13
588 +2.503553e+00*0.13*0.13*0.13*0.13
589 -1.125965e+00*0.13*0.13*0.13*0.13*0.13
590 +2.009036e-01*0.13*0.13*0.13*0.13*0.13*0.13);
591
592 if (mom>1.8)
593 return (7.594129e-02
594 -5.535827e-01*1.8
595 +1.728591e+00*1.8*1.8
596 -2.827893e+00*1.8*1.8*1.8
597 +2.503553e+00*1.8*1.8*1.8*1.8
598 -1.125965e+00*1.8*1.8*1.8*1.8*1.8
599 +2.009036e-01*1.8*1.8*1.8*1.8*1.8*1.8);
600 return (7.594129e-02
601 -5.535827e-01*mom
602 +1.728591e+00*mom*mom
603 -2.827893e+00*mom*mom*mom
604 +2.503553e+00*mom*mom*mom*mom
605 -1.125965e+00*mom*mom*mom*mom*mom
606 +2.009036e-01*mom*mom*mom*mom*mom*mom);
607}
608
609// pion
610// 0.13 - 2.0
611// 0 1.063457e+00 8.872043e-03
612// 1 -4.222208e-01 2.534402e-02
613// 2 1.042004e-01 1.503945e-02
614float AliFemtoESDTrackCut::PidFractionPion(float mom) const
615{
616 // Provide a parameterized fraction of pions dependent on momentum
617 if (mom<0.13)
618 return ( 1.063457e+00
619 -4.222208e-01*0.13
620 +1.042004e-01*0.0169);
621 if (mom>2.0)
622 return ( 1.063457e+00
623 -4.222208e-01*2.0
624 +1.042004e-01*4.0);
625 return ( 1.063457e+00
626 -4.222208e-01*mom
627 +1.042004e-01*mom*mom);
628}
629
630// kaon
631// 0.18 - 2.0
632// 0 -7.289406e-02 1.686074e-03
633// 1 4.415666e-01 1.143939e-02
634// 2 -2.996790e-01 1.840964e-02
635// 3 6.704652e-02 7.783990e-03
636float AliFemtoESDTrackCut::PidFractionKaon(float mom) const
637{
638 // Provide a parameterized fraction of kaons dependent on momentum
639 if (mom<0.18)
640 return (-7.289406e-02
641 +4.415666e-01*0.18
642 -2.996790e-01*0.18*0.18
643 +6.704652e-02*0.18*0.18*0.18);
644 if (mom>2.0)
645 return (-7.289406e-02
646 +4.415666e-01*2.0
647 -2.996790e-01*2.0*2.0
648 +6.704652e-02*2.0*2.0*2.0);
649 return (-7.289406e-02
650 +4.415666e-01*mom
651 -2.996790e-01*mom*mom
652 +6.704652e-02*mom*mom*mom);
653}
654
655// proton
656// 0.26 - 2.0
657// 0 -3.730200e-02 2.347311e-03
658// 1 1.163684e-01 1.319316e-02
659// 2 8.354116e-02 1.997948e-02
660// 3 -4.608098e-02 8.336400e-03
661float AliFemtoESDTrackCut::PidFractionProton(float mom) const
662{
663 // Provide a parameterized fraction of protons dependent on momentum
664 if (mom<0.26) return 0.0;
665 if (mom>2.0)
666 return (-3.730200e-02
667 +1.163684e-01*2.0
668 +8.354116e-02*2.0*2.0
669 -4.608098e-02*2.0*2.0*2.0);
670 return (-3.730200e-02
671 +1.163684e-01*mom
672 +8.354116e-02*mom*mom
673 -4.608098e-02*mom*mom*mom);
674}
675
676void AliFemtoESDTrackCut::SetMomRangeTOFpidIs(const float& minp, const float& maxp)
677{
678 fMinPforTOFpid = minp;
679 fMaxPforTOFpid = maxp;
680}
681
682void AliFemtoESDTrackCut::SetMomRangeTPCpidIs(const float& minp, const float& maxp)
683{
684 fMinPforTPCpid = minp;
685 fMaxPforTPCpid = maxp;
686}
687
688void AliFemtoESDTrackCut::SetMomRangeITSpidIs(const float& minp, const float& maxp)
689{
690 fMinPforITSpid = minp;
691 fMaxPforITSpid = maxp;
692}
693
694bool AliFemtoESDTrackCut::IsPionTPCdEdx(float mom, float dEdx)
695{
696 // double a1 = -95.4545, b1 = 86.5455;
697 // double a2 = 0.0, b2 = 56.0;
698 double a1 = -343.75, b1 = 168.125;
699 double a2 = 0.0, b2 = 65.0;
700
701 if (mom < 0.32) {
702 if (dEdx < a1*mom+b1) return true;
703 }
704 if (dEdx < a2*mom+b2) return true;
705
706 return false;
707}
708
709bool AliFemtoESDTrackCut::IsKaonTPCdEdx(float mom, float dEdx)
710{
711
712// double a1 = -547.0; double b1 = 297.0;
713// double a2 = -125.0; double b2 = 145.0;
714// double a3 = -420.0; double b3 = 357.0;
715// double a4 = -110.0; double b4 = 171.0;
716// double b5 = 72.0;
717
718// if (mom<0.2) return false;
719
720// if (mom<0.36) {
721// if (dEdx < a1*mom+b1) return false;
722// if (dEdx > a3*mom+b3) return false;
723// }
724// else if (mom<0.6) {
725// if (dEdx < a2*mom+b2) return false;
726// if (dEdx > a3*mom+b3) return false;
727// }
728// else if (mom<0.9) {
729// if (dEdx > a4*mom+b4) return false;
730// if (dEdx < b5) return false;
731// }
732// else
733// return false;
734// // else {
735// // if (dEdx > b5) return false;
736// // }
737
738// return true;
739
740 double a1 = -268.896; double b1 = 198.669;
741 double a2 = -49.0012; double b2 = 88.7214;
742
743 if (mom<0.2) return false;
744
745 if (mom>0.3 && mom<0.5) {
746 if (dEdx < a1*mom+b1) return false;
747 }
748 else if (mom<1.2) {
749 if (dEdx < a2*mom+b2) return false;
750 }
751
752 return true;
753
754}
755
756bool AliFemtoESDTrackCut::IsProtonTPCdEdx(float mom, float dEdx)
757{
758 double a1 = -1800.0; double b1 = 940.0;
759 double a2 = -500.0; double b2 = 420.0;
760 double a3 = -216.7; double b3 = 250.0;
761
762 if (mom<0.2) return false;
763
764 if (mom>0.3 && mom<0.4) {
765 if (dEdx < a1*mom+b1) return false;
766 }
767 else if (mom<0.6) {
768 if (dEdx < a2*mom+b2) return false;
769 }
770 else if (mom<0.9) {
771 if (dEdx < a3*mom+b3) return false;
772 }
773
774 return true;
775
776}
777
778bool AliFemtoESDTrackCut::IsPionTOFTime(float mom, float ttof)
779{
780 double a1 = -427.0; double b1 = 916.0;
781 double a2 = 327.0; double b2 = -888.0;
782 if (mom<0.3) return kFALSE;
783 if (mom>2.0) return kFALSE;
784 if (ttof > a1*mom+b1) return kFALSE;
785 if (ttof < a2*mom+b2) return kFALSE;
786
787 return kTRUE;
788}
789
790bool AliFemtoESDTrackCut::IsKaonTOFTime(float mom, float ttof)
791{
792 double a1 = 000.0; double b1 = -500.0;
793 double a2 = 000.0; double b2 = 500.0;
794 double a3 = 850.0; double b3 = -1503.0;
795 double a4 = -1637.0; double b4 = 3621.0;
796
797 if (mom<0.3) return kFALSE;
798 if (mom>2.06) return kFALSE;
799 if (mom<1.2) {
800 if (ttof > a2*mom+b2) return kFALSE;
801 if (ttof < a1*mom+b1) return kFALSE;
802 }
803 if (mom<1.9) {
804 if (ttof > a2*mom+b2) return kFALSE;
805 if (ttof < a3*mom+b3) return kFALSE;
806 }
807 if (mom<2.06) {
808 if (ttof > a4*mom+b4) return kFALSE;
809 if (ttof < a3*mom+b3) return kFALSE;
810 }
811 return kTRUE;
812}
813
814bool AliFemtoESDTrackCut::IsProtonTOFTime(float mom, float ttof)
815{
816 double a1 = 000.0; double b1 = -915.0;
817 double a2 = 000.0; double b2 = 600.0;
818 double a3 = 572.0; double b3 = -1715.0;
819
820 if (mom<0.3) return kFALSE;
821 if (mom>3.0) return kFALSE;
822 if (mom<1.4) {
823 if (ttof > a2*mom+b2) return kFALSE;
824 if (ttof < a1*mom+b1) return kFALSE;
825 }
826 if (mom<3.0) {
827 if (ttof > a2*mom+b2) return kFALSE;
828 if (ttof < a3*mom+b3) return kFALSE;
829 }
830 return kTRUE;
831}
832
833
834
835
836bool AliFemtoESDTrackCut::IsKaonTPCdEdxNSigma(float mom, float nsigmaK)
837{
838 cout<<" AliFemtoESDTrackCut::IsKaonTPCdEdxNSigma "<<mom<<" "<<nsigmaK<<endl;
839
840
841 if(mom<0.35 && TMath::Abs(nsigmaK)<5.0)return true;
842 if(mom>=0.35 && mom<0.5 && TMath::Abs(nsigmaK)<3.0)return true;
843 if(mom>=0.5 && mom<0.7 && TMath::Abs(nsigmaK)<2.0)return true;
844
845 return false;
846}
847
848
849bool AliFemtoESDTrackCut::IsKaonTOFNSigma(float mom, float nsigmaK)
850{
851 cout<<" AliFemtoESDTrackCut::IsKaonTPCdEdxNSigma "<<mom<<" "<<nsigmaK<<endl;
852 //fan
853 // if(mom<1.5 && TMath::Abs(nsigmaK)<3.0)return true;
854 if(mom>=1.5 && TMath::Abs(nsigmaK)<2.0)return true;
855 return false;
856}
857
76ce4b5b 858bool AliFemtoESDTrackCut::IsKaonNSigma(float mom, float nsigmaTPCK, float nsigmaTOFK)
859{
ce7b3d98 860 if(mom<0.4)
861 {
862 if(nsigmaTOFK<-999.)
863 {
864 if(TMath::Abs(nsigmaTPCK)<1.0) return true;
865 }
866 else if(TMath::Abs(nsigmaTOFK)<3.0 && TMath::Abs(nsigmaTPCK)<3.0) return true;
867 }
868 else if(mom>=0.4 && mom<=0.6)
869 {
870 if(nsigmaTOFK<-999.)
871 {
872 if(TMath::Abs(nsigmaTPCK)<2.0) return true;
873 }
874 else if(TMath::Abs(nsigmaTOFK)<3.0 && TMath::Abs(nsigmaTPCK)<3.0) return true;
875 }
876 else if(nsigmaTOFK<-999.)
877 {
878 return false;
879 }
880 else if(TMath::Abs(nsigmaTOFK)<3.0 && TMath::Abs(nsigmaTPCK)<3.0) return true;
76ce4b5b 881
882 return false;
883}
884
885
76ce4b5b 886bool AliFemtoESDTrackCut::IsPionNSigma(float mom, float nsigmaTPCPi, float nsigmaTOFPi)
887{
ce7b3d98 888 if(mom<0.65)
889 {
890 if(nsigmaTOFPi<-999.)
891 {
892 if(mom<0.35 && TMath::Abs(nsigmaTPCPi)<3.0) return true;
893 else if(mom<0.5 && mom>=0.35 && TMath::Abs(nsigmaTPCPi)<3.0) return true;
894 else if(mom>=0.5 && TMath::Abs(nsigmaTPCPi)<2.0) return true;
895 else return false;
896 }
897 else if(TMath::Abs(nsigmaTOFPi)<3.0 && TMath::Abs(nsigmaTPCPi)<3.0) return true;
898 }
899 else if(nsigmaTOFPi<-999.)
900 {
901 return false;
902 }
903 else if(mom<1.5 && TMath::Abs(nsigmaTOFPi)<3.0 && TMath::Abs(nsigmaTPCPi)<5.0) return true;
904 else if(mom>=1.5 && TMath::Abs(nsigmaTOFPi)<2.0 && TMath::Abs(nsigmaTPCPi)<5.0) return true;
76ce4b5b 905
76ce4b5b 906 return false;
907}
908
909
910bool AliFemtoESDTrackCut::IsProtonNSigma(float mom, float nsigmaTPCP, float nsigmaTOFP)
911{
ce7b3d98 912 if(mom<0.8)
913 {
914 if(nsigmaTOFP<-999.)
915 {
916 if(TMath::Abs(nsigmaTPCP)<3.0) return true;
917 else return false;
918 }
919 else if(TMath::Abs(nsigmaTOFP)<3.0 && TMath::Abs(nsigmaTPCP)<3.0) return true;
920 }
921 else if(nsigmaTOFP<-999.)
922 {
923 return false;
924 }
925 else if(TMath::Abs(nsigmaTOFP)<3.0 && TMath::Abs(nsigmaTPCP)<3.0) return true;
76ce4b5b 926
927 return false;
928}
929
930
931void AliFemtoESDTrackCut::SetPIDMethod(ReadPIDMethodType newMethod)
932{
933 fPIDMethod = newMethod;
934}