Include dir structure in tarball for cvmfs
[u/mrichter/AliRoot.git] / STEER / AliV0Comparison.C
CommitLineData
ca28c5f5 1/****************************************************************************
2 * Very important, delicate and rather obscure macro. *
3 * *
4 * Creates list of "findable" V0s, *
5 * calculates efficiency, resolutions etc. *
6 * *
1330b1b0 7 * To get the list of the "findable" V0s, you should first get the list *
8 * of "findable" tracks, which can be done by running *
9 * TPC/AliTPCComparison.C and ITS/AliITSComparisonV2.C macros. *
10 * *
ca28c5f5 11 * Origin: I.Belikov, IReS, Strasbourg, Jouri.Belikov@cern.ch *
12 ****************************************************************************/
13
566abf75 14#if !defined(__CINT__) || defined(__MAKECINT__)
4fa1e629 15 #include <TMath.h>
16 #include <TError.h>
17 #include <Riostream.h>
18 #include <TH1F.h>
19 #include <TTree.h>
20 #include <TParticle.h>
1330b1b0 21 #include <TPDGCode.h>
4fa1e629 22 #include <TCanvas.h>
23 #include <TLine.h>
24 #include <TText.h>
25 #include <TBenchmark.h>
26 #include <TStyle.h>
8b462fd8 27 #include <TFile.h>
4fa1e629 28 #include <TROOT.h>
29
30 #include "AliStack.h"
566abf75 31 #include "AliHeader.h"
4fa1e629 32 #include "AliTrackReference.h"
566abf75 33 #include "AliRunLoader.h"
ca28c5f5 34 #include "AliRun.h"
86ad5fcb 35 #include "AliESDEvent.h"
1330b1b0 36 #include "AliESDv0.h"
ca28c5f5 37#endif
38
4fa1e629 39Int_t GoodV0s(const Char_t *dir=".");
ca28c5f5 40
566abf75 41extern AliRun *gAlice;
4fa1e629 42extern TBenchmark *gBenchmark;
43extern TROOT *gROOT;
566abf75 44
4fa1e629 45static Int_t allgood=0;
46static Int_t allfound=0;
5102bab6 47
4fa1e629 48Int_t AliV0Comparison(Int_t code=310, const Char_t *dir=".") {
49 //Lambda=3122, LambdaBar=-3122
50 gBenchmark->Start("AliV0Comparison");
51
52 ::Info("AliV0Comparison.C","Doing comparison...");
5102bab6 53
54
04b2a5f1 55 const Double_t V0window=0.05;
56 Double_t ptncut=0.13, ptpcut=0.13, kinecut=0.03;
57 Double_t V0mass=0.497672, V0width=0.020;
ca28c5f5 58 switch (code) {
04b2a5f1 59 case kK0Short:
60 break;
61 case kLambda0:
62 V0mass=1.115683; V0width=0.015; ptpcut=0.50; kinecut=0.002;
63 break;
64 case kLambda0Bar:
65 V0mass=1.115683; V0width=0.015; ptncut=0.50; kinecut=0.002;
66 break;
ca28c5f5 67 default: cerr<<"Invalid PDG code !\n"; return 1;
68 }
69
4fa1e629 70 TH1F *hp=(TH1F*)gROOT->FindObject("hp");
71 if (!hp) hp=new TH1F("hp","PHI Resolution",30,-30.,30.);
566abf75 72 hp->SetXTitle("(mrad)"); hp->SetFillColor(2);
4fa1e629 73
74 TH1F *hl=(TH1F*)gROOT->FindObject("hl");
75 if (!hl) hl=new TH1F("hl","LAMBDA Resolution",30,-30,30);
76 hl->SetXTitle("(mrad)"); hl->SetFillColor(1); hl->SetFillStyle(3013);
77
78 TH1F *hpt=(TH1F*)gROOT->FindObject("hpt");
79 if (!hpt) hpt=new TH1F("hpt","Relative Pt Resolution",30,-10.,10.);
566abf75 80 hpt->SetXTitle("(%)"); hpt->SetFillColor(2);
81
4fa1e629 82 TH1F *hx=(TH1F*)gROOT->FindObject("hx");
83 if (!hx) hx=new TH1F("hx","Position Resolution (X)",30,-3.,3.);
566abf75 84 hx->SetXTitle("(mm)"); hx->SetFillColor(6);
4fa1e629 85
86 TH1F *hy=(TH1F*)gROOT->FindObject("hy");
87 if (!hy) hy=new TH1F("hy","Position Resolution (Y)",30,-3.,3.);
566abf75 88 hy->SetXTitle("(mm)"); hy->SetFillColor(1); hy->SetFillStyle(3013);
4fa1e629 89
90 TH1F *hz=(TH1F*)gROOT->FindObject("hz");
91 if (!hz) hz=new TH1F("hz","Position Resolution (Z)",30,-3.,3.);
566abf75 92 hz->SetXTitle("(mm)"); hz->SetFillColor(6);
93
94
95 Double_t pmin=0.2, pmax=4.2; Int_t nchan=20;
4fa1e629 96 TH1F *hgood=(TH1F*)gROOT->FindObject("hgood");
97 if (!hgood) hgood=new TH1F("hgood","Good Vertices",nchan,pmin,pmax);
98
99 TH1F *hfound=(TH1F*)gROOT->FindObject("hfound");
100 if (!hfound) hfound=new TH1F("hfound","Found Vertices",nchan,pmin,pmax);
101
102 TH1F *hfake=(TH1F*)gROOT->FindObject("hfake");
103 if (!hfake) hfake=new TH1F("hfake","Fake Vertices",nchan,pmin,pmax);
104
105 TH1F *hg=(TH1F*)gROOT->FindObject("hg");
106 if (!hg) hg=new TH1F("hg","Efficiency for Good Vertices",nchan,pmin,pmax);
566abf75 107 hg->SetLineColor(4); hg->SetLineWidth(2);
4fa1e629 108
109 TH1F *hf=(TH1F*)gROOT->FindObject("hf");
110 if (!hf) hf=new TH1F("hf","Probability of Fake Vertices",nchan,pmin,pmax);
566abf75 111 hf->SetFillColor(1); hf->SetFillStyle(3013); hf->SetLineWidth(2);
112
4fa1e629 113
566abf75 114 Double_t mmin=V0mass-V0window, mmax=V0mass+V0window;
4fa1e629 115 TH1F *v0s=(TH1F*)gROOT->FindObject("v0s");
116 if (!v0s) v0s=new TH1F("v0s","V0s Effective Mass",40, mmin, mmax);
117 v0s->SetXTitle("(GeV)"); v0s->SetLineColor(4); v0s->SetLineWidth(4);
118
119 TH1F *v0sf=(TH1F*)gROOT->FindObject("v0sf");
120 if (!v0sf) v0sf=new TH1F("v0sf","Fake V0s Effective Mass",40, mmin, mmax);
566abf75 121 v0sf->SetXTitle("(GeV)"); v0sf->SetFillColor(6);
122
123
4fa1e629 124 Char_t fname[100];
125 sprintf(fname,"%s/GoodV0s.root",dir);
126
127 TFile *refFile=TFile::Open(fname,"old");
128 if (!refFile || !refFile->IsOpen()) {
129 ::Info("AliV0Comparison.C","Marking good V0s (will take a while)...");
130 if (GoodV0s(dir)) {
131 ::Error("AliV0Comparison.C","Can't generate the reference file !");
132 return 1;
133 }
134 }
135 refFile=TFile::Open(fname,"old");
136 if (!refFile || !refFile->IsOpen()) {
137 ::Error("AliV0Comparison.C","Can't open the reference file !");
138 return 1;
139 }
140
141 TTree *v0Tree=(TTree*)refFile->Get("v0Tree");
142 if (!v0Tree) {
143 ::Error("AliV0Comparison.C","Can't get the reference tree !");
144 return 2;
145 }
146 TBranch *pbranch=v0Tree->GetBranch("positive");
147 if (!pbranch) {
148 ::Error("AliV0Comparison.C","Can't get the positive daughter branch !");
149 return 3;
150 }
151 TClonesArray dummy("AliTrackReference",1000), *prefs=&dummy;
152 pbranch->SetAddress(&prefs);
ca28c5f5 153
4fa1e629 154 TBranch *nbranch=v0Tree->GetBranch("negative");
155 if (!nbranch) {
156 ::Error("AliV0Comparison.C","Can't get the negative daughter branch !");
157 return 4;
ca28c5f5 158 }
4fa1e629 159 TClonesArray dumm("AliTrackReference",1000), *nrefs=&dumm;
160 nbranch->SetAddress(&nrefs);
161
162
163 sprintf(fname,"%s/AliESDs.root",dir);
164 TFile *ef=TFile::Open(fname);
165 if ((!ef)||(!ef->IsOpen())) {
166 sprintf(fname,"%s/AliESDv0.root",dir);
167 ef=TFile::Open(fname);
168 if ((!ef)||(!ef->IsOpen())) {
169 ::Error("AliV0Comparison.C","Can't open AliESDv0.root !");
170 return 5;
171 }
ca28c5f5 172 }
4514157e 173 AliESDEvent* event = new AliESDEvent();
8b462fd8 174 TTree* esdTree = (TTree*) ef->Get("esdTree");
175 if (!esdTree) {
176 ::Error("AliV0Comparison.C", "no ESD tree found");
177 return 6;
178 }
86ad5fcb 179 event->ReadFromTree(esdTree);
4fa1e629 180
181
182 //******* Loop over events *********
183 Int_t e=0;
8b462fd8 184 while (esdTree->GetEvent(e)) {
4fa1e629 185 cout<<endl<<endl<<"********* Processing event number: "<<e<<"*******\n";
186
4fa1e629 187 Int_t nentr=event->GetNumberOfV0s();
188 allfound+=nentr;
ca28c5f5 189
4fa1e629 190 if (v0Tree->GetEvent(e++)==0) {
191 cerr<<"No reconstructable V0s !\n";
192 continue;
193 }
ca28c5f5 194
4fa1e629 195 Int_t ngood=prefs->GetEntriesFast(),ng=0;
ca28c5f5 196
4fa1e629 197 Double_t pxg=0.,pyg=0.,pzg=0.,ptg=0.;
198 Int_t nlab=-1, plab=-1;
199 Int_t i;
200 for (i=0; i<nentr; i++) {
201 AliESDv0 *vertex=event->GetV0(i);
202
d6a49f20 203 if (vertex->GetOnFlyStatus()) continue;
204
4fa1e629 205 Int_t nidx=TMath::Abs(vertex->GetNindex());
206 Int_t pidx=TMath::Abs(vertex->GetPindex());
207
208 AliESDtrack *ntrack=event->GetTrack(nidx);
209 AliESDtrack *ptrack=event->GetTrack(pidx);
210
211 nlab=TMath::Abs(ntrack->GetLabel());
212 plab=TMath::Abs(ptrack->GetLabel());
213
214 /** Kinematical cuts **/
215 Double_t pxn,pyn,pzn; vertex->GetNPxPyPz(pxn,pyn,pzn);
216 Double_t ptn=TMath::Sqrt(pxn*pxn + pyn*pyn);
217 if (ptn < ptncut) continue;
218 Double_t pxp,pyp,pzp; vertex->GetPPxPyPz(pxp,pyp,pzp);
219 Double_t ptp=TMath::Sqrt(pxp*pxp + pyp*pyp);
220 if (ptp < ptpcut) continue;
221 Double_t kine=vertex->ChangeMassHypothesis(code);
222 //if (TMath::Abs(kine)>kinecut) continue;
223
224 Double_t mass=vertex->GetEffMass();
225 v0s->Fill(mass);
226 v0sf->Fill(mass);
227
228 AliTrackReference *nref=0, *pref=0;
229 Int_t j;
230 for (j=0; j<ngood; j++) {
231 nref=(AliTrackReference*)nrefs->UncheckedAt(j);
232 pref=(AliTrackReference*)prefs->UncheckedAt(j);
233 if (nref->Label() == nlab)
234 if (pref->Label() == plab) break;
235 }
236
237 if (TMath::Abs(mass-V0mass)>V0width) continue;
238
239 Double_t px,py,pz; vertex->GetPxPyPz(px,py,pz);
240 Double_t pt=TMath::Sqrt(px*px+py*py);
241
242 if (j==ngood) {
243 hfake->Fill(pt);
244 cout<<"Fake vertex: ("<<nlab<<","<<plab<<")\n";
245 continue;
246 }
247 v0sf->Fill(mass,-1);
248
249 pxg=nref->Px()+pref->Px(); pyg=nref->Py()+pref->Py();
250 pzg=nref->Pz()+pref->Pz();
251 ptg=TMath::Sqrt(pxg*pxg+pyg*pyg);
252 Double_t phig=TMath::ATan2(pyg,pxg), phi=TMath::ATan2(py,px);
253 Double_t lamg=TMath::ATan2(pzg,ptg), lam=TMath::ATan2(pz,pt);
254 hp->Fill((phi - phig)*1000.);
255 hl->Fill((lam - lamg)*1000.);
256 hpt->Fill((1/pt - 1/ptg)/(1/ptg)*100.);
257
258 Double_t x,y,z; vertex->GetXYZ(x,y,z);
259 hx->Fill((x - nref->X())*10);
260 hy->Fill((y - nref->Y())*10);
261 hz->Fill((z - nref->Z())*10);
262
263 hfound->Fill(ptg);
264 nref->SetLabel(-1);
265
266 }
267 for (i=0; i<ngood; i++) {
268 AliTrackReference *nref=(AliTrackReference*)nrefs->UncheckedAt(i);
269 AliTrackReference *pref=(AliTrackReference*)prefs->UncheckedAt(i);
270 Int_t pdg=(Int_t)nref->GetLength();//this is the mother's PDG !
271 if (code!=pdg) continue;
272 ng++;
273 pxg=nref->Px()+pref->Px(); pyg=nref->Py()+pref->Py();
274 ptg=TMath::Sqrt(pxg*pxg+pyg*pyg);
275 hgood->Fill(ptg);
276 nlab=nref->Label(); plab=pref->Label();
277 if (nlab < 0) continue;
278 cout<<"Vertex ("<<nlab<<','<<plab<<") has not been found !\n";
279 }
280 allgood+=ng;
281
282 cout<<"Number of found V0s : "<<nentr<<endl;
283 cout<<"Number of \"good\" V0s : "<<ng<<endl;
284
285 prefs->Clear();
286 nrefs->Clear();
4fa1e629 287
288 } //**** End of the loop over events
289
8b462fd8 290 delete event;
4514157e 291 delete esdTree;
4fa1e629 292 ef->Close();
293
294 delete v0Tree;
295 refFile->Close();
ca28c5f5 296
4fa1e629 297 Stat_t ng=hgood->GetEntries(), nf=hfound->GetEntries();
298 if (ng!=0) cout<<"Integral efficiency is about "<<nf/ng*100.<<" %\n";
299 cout<<"Total number of found V0s: "<<allfound<<" ("<<nf<<" in the peak)\n";
300 cout<<"Total number of \"good\" V0s: "<<allgood<<endl;
ca28c5f5 301
302 gStyle->SetOptStat(111110);
303 gStyle->SetOptFit(1);
304
4fa1e629 305 TCanvas *c1=(TCanvas*)gROOT->FindObject("c1");
306 if (!c1) {
307 c1=new TCanvas("c1","",0,0,580,610);
308 c1->Divide(2,2);
309 }
310
311 Int_t minc=33;
ca28c5f5 312
313 c1->cd(1);
314 gPad->SetFillColor(42); gPad->SetFrameFillColor(10);
4fa1e629 315 if (hp->GetEntries()<minc) hp->Draw(); else hp->Fit("gaus");
ca28c5f5 316 hl->Draw("same"); c1->cd();
317
318 c1->cd(2);
319 gPad->SetFillColor(42); gPad->SetFrameFillColor(10);
4fa1e629 320 if (hpt->GetEntries()<minc) hpt->Draw(); else hpt->Fit("gaus");
321 c1->cd();
ca28c5f5 322
323 c1->cd(3);
324 gPad->SetFillColor(42); gPad->SetFrameFillColor(10);
4fa1e629 325 if (hx->GetEntries()<minc) hx->Draw(); else hx->Fit("gaus");
ca28c5f5 326 hy->Draw("same"); c1->cd();
327
328 c1->cd(4);
329 gPad->SetFillColor(42); gPad->SetFrameFillColor(10);
4fa1e629 330 if (hz->GetEntries()<minc) hz->Draw(); else hz->Fit("gaus");
ca28c5f5 331
4fa1e629 332 c1->Update();
ca28c5f5 333
4fa1e629 334 TCanvas *c2=(TCanvas*)gROOT->FindObject("c2");
335 if (!c2) {
336 c2=new TCanvas("c2","",600,0,580,610);
337 c2->Divide(1,2);
338 }
ca28c5f5 339
340 c2->cd(1);
341 gPad->SetFillColor(42); gPad->SetFrameFillColor(10);
342 hfound->Sumw2(); hgood->Sumw2(); hfake->Sumw2();
343 hg->Divide(hfound,hgood,1,1.,"b");
344 hf->Divide(hfake,hgood,1,1.,"b");
345 hg->SetMaximum(1.4);
346 hg->SetYTitle("V0 reconstruction efficiency");
347 hg->SetXTitle("Pt (GeV/c)");
348 hg->Draw();
349
350 TLine *line1 = new TLine(pmin,1.0,pmax,1.0); line1->SetLineStyle(4);
351 line1->Draw("same");
352 TLine *line2 = new TLine(pmin,0.9,pmax,0.9); line2->SetLineStyle(4);
353 line2->Draw("same");
354
355 hf->SetFillColor(1);
356 hf->SetFillStyle(3013);
357 hf->SetLineColor(2);
358 hf->SetLineWidth(2);
359 hf->Draw("histsame");
360 TText *text = new TText(0.461176,0.248448,"Fake vertices");
361 text->SetTextSize(0.05);
362 text->Draw();
363 text = new TText(0.453919,1.11408,"Good vertices");
364 text->SetTextSize(0.05);
365 text->Draw();
366
367
368 c2->cd(2);
369 gPad->SetFillColor(42); gPad->SetFrameFillColor(10);
4fa1e629 370 if (v0s->GetEntries()<minc) v0s->Draw();
371 else v0s->Fit("gaus","","",V0mass-V0width,V0mass+V0width);
04b2a5f1 372 v0sf->Draw("same");
ca28c5f5 373 Double_t max=v0s->GetMaximum();
374 TLine *line3 = new TLine(V0mass-V0width,0.,V0mass-V0width,max);
375 line3->Draw("same");
376 TLine *line4 = new TLine(V0mass+V0width,0.,V0mass+V0width,max);
377 line4->Draw("same");
378
4fa1e629 379 c2->Update();
380
381 TFile fc("AliV0Comparison.root","RECREATE");
382 c1->Write();
383 c2->Write();
384 fc.Close();
385
386 gBenchmark->Stop("AliV0Comparison");
387 gBenchmark->Show("AliV0Comparison");
ca28c5f5 388
389 return 0;
390}
391
ca28c5f5 392
4fa1e629 393Int_t GoodV0s(const Char_t *dir) {
394 if (gAlice) {
33c3c91a 395 delete AliRunLoader::Instance();
4fa1e629 396 delete gAlice;//if everything was OK here it is already NULL
397 gAlice = 0x0;
ca28c5f5 398 }
399
4fa1e629 400 Char_t fname[100];
401 sprintf(fname,"%s/galice.root",dir);
402
403 AliRunLoader *rl = AliRunLoader::Open(fname,"COMPARISON");
5102bab6 404 if (!rl) {
4fa1e629 405 ::Error("GoodTracksITS","Can't start session !");
406 return 1;
ca28c5f5 407 }
5102bab6 408
566abf75 409 rl->LoadgAlice();
410 rl->LoadHeader();
411 rl->LoadKinematics();
4fa1e629 412
413
414 Int_t nev=rl->GetNumberOfEvents();
415 ::Info("GoodV0s","Number of events : %d\n",nev);
416
417 sprintf(fname,"%s/GoodTracksITS.root",dir);
418 TFile *itsFile=TFile::Open(fname);
419 if ((!itsFile)||(!itsFile->IsOpen())) {
420 ::Error("GoodV0s","Can't open the GoodTracksITS.root !");
421 delete rl;
422 return 5;
423 }
424 TClonesArray dum("AliTrackReference",1000), *itsRefs=&dum;
425 TTree *itsTree=(TTree*)itsFile->Get("itsTree");
426 if (!itsTree) {
427 ::Error("GoodV0s","Can't get the ITS reference tree !");
428 delete rl;
429 return 6;
430 }
431 TBranch *itsBranch=itsTree->GetBranch("ITS");
432 if (!itsBranch) {
433 ::Error("GoodV0s","Can't get the ITS reference branch !");
434 delete rl;
435 return 7;
436 }
437 itsBranch->SetAddress(&itsRefs);
438
439
440 sprintf(fname,"%s/GoodV0s.root",dir);
441 TFile *v0File=TFile::Open(fname,"recreate");
442 TClonesArray dummy("AliTrackReference",1000), *nrefs=&dummy;
443 TClonesArray dumm("AliTrackReference",1000), *prefs=&dumm;
444 TTree v0Tree("v0Tree","Tree with info about the reconstructable V0s");
445 v0Tree.Branch("negative",&nrefs);
446 v0Tree.Branch("positive",&prefs);
447
448
449 /*** Some information about the cuts ***/
450 Double_t r2min=0.9*0.9;
451 Double_t r2max=2.9*2.9;
452
453
454
455 //******** Loop over generated events
456 for (Int_t e=0; e<nev; e++) {
457 rl->GetEvent(e); v0File->cd();
458
459 Int_t np = rl->GetHeader()->GetNtrack();
460 cout<<"Event "<<e<<" Number of particles: "<<np<<endl;
461
462 itsTree->GetEvent(e);
463 Int_t nk=itsRefs->GetEntriesFast();
464
465 AliStack *stack=rl->Stack();
466
467 AliTrackReference *nref=0, *pref=0;
468
469 Int_t nv=0;
470 while (np--) {
471 //cerr<<np<<'\r';
472 TParticle *p0=stack->Particle(np);
473
474 /*** only these V0s are "good" ***/
475 Int_t code=p0->GetPdgCode();
476 if (code!=kK0Short)
477 if (code!=kLambda0)
478 if (code!=kLambda0Bar) continue;
479
480 /*** daughter tracks should be "good" ***/
481 Int_t plab=p0->GetFirstDaughter(), nlab=p0->GetLastDaughter();
482 if (nlab==plab) continue;
483 if (nlab<0) continue;
484 if (plab<0) continue;
485 Int_t i;
60f349ea 486 TParticle * part = stack->Particle(plab);
487 if (part) {
488 TParticlePDG * partPDG = part->GetPDG();
489 if (partPDG && partPDG->Charge() < 0.) {
490 i=plab; plab=nlab; nlab=i;
491 }
492 }
4fa1e629 493
494 for (i=0; i<nk; i++) {
495 nref=(AliTrackReference*)itsRefs->UncheckedAt(i);
496 if (nref->Label()==nlab) break;
497 }
498 if (i==nk) continue;
499 for (i=0; i<nk; i++) {
500 pref=(AliTrackReference*)itsRefs->UncheckedAt(i);
501 if (pref->Label()==plab) break;
502 }
503 if (i==nk) continue;
504
505 /*** fiducial volume ***/
506 TParticle *p=stack->Particle(nlab);
507 Double_t x=p->Vx(), y=p->Vy(), z=p->Vz(), r2=x*x+y*y;
508 if (r2<r2min) continue;
509 if (r2>r2max) continue;
ca28c5f5 510
4fa1e629 511 Int_t pdg=p0->GetPdgCode();
512 nref->SetLength(pdg); //This will the V0's PDG !
513
514 new((*nrefs)[nv]) AliTrackReference(*nref);
515 new((*prefs)[nv]) AliTrackReference(*pref);
516
517 nv++;
ca28c5f5 518 }
4fa1e629 519 itsRefs->Clear();
ca28c5f5 520
4fa1e629 521 v0Tree.Fill();
522 nrefs->Clear(); prefs->Clear();
523
524 } //**** end of the loop over generated events
525
526 v0Tree.Write();
527 v0File->Close();
528
529 delete itsTree;
530 itsFile->Close();
531
532 delete rl;
533 return 0;
ca28c5f5 534}