+++ /dev/null
-/****************************************************************************
- * Legacy comparison macro, adapted to the upgraded ITS. *
- * *
- * Creates list of "trackable" tracks, *
- * calculates efficiency, resolutions etc. *
- * The ESD tracks must be in an appropriate state (kITSin or kITSrefit) *
- * *
- * Before running, load the ITSU libraries: *
- * gSystem->Load("libITSUpgradeBase");gSystem->Load("libITSUpgradeRec"); *
- * *
- * Origin: I.Belikov, IPHC, Iouri.Belikov@iphc.cnrs.fr *
- ****************************************************************************/
-
-#if !defined(__CINT__) || defined(__MAKECINT__)
- #include <TMath.h>
- #include <TError.h>
- #include <Riostream.h>
- #include <TH1F.h>
- #include <TH2F.h>
- #include <TTree.h>
- #include <TParticle.h>
- #include <TCanvas.h>
- #include <TLine.h>
- #include <TText.h>
- #include <TBenchmark.h>
- #include <TStyle.h>
- #include <TFile.h>
- #include <TROOT.h>
-
- #include "AliStack.h"
- #include "AliHeader.h"
- #include "AliTrackReference.h"
- #include "AliRunLoader.h"
- #include "AliRun.h"
- #include "AliESDEvent.h"
- #include "AliESDtrack.h"
-
- #include "UPGRADE/AliITSUClusterPix.h"
- #include "UPGRADE/AliITSULoader.h"
-#endif
-
-Int_t GoodTracksITS(const Char_t *dir=".");
-
-extern AliRun *gAlice;
-extern TBenchmark *gBenchmark;
-extern TROOT *gROOT;
-
-static Int_t allgood=0;
-static Int_t allselected=0;
-static Int_t allfound=0;
-
-Int_t AliITSUComparison
-(Float_t ptcutl=0., Float_t ptcuth=2., const Char_t *dir=".") {
- gBenchmark->Start("AliITSUComparison");
-
- ::Info("AliITSUComparison.C","Doing comparison...");
-
-
- TH1F *hp=(TH1F*)gROOT->FindObject("hp");
- if (!hp) hp=new TH1F("hp","PHI resolution",50,-20.,20.);
- hp->SetFillColor(4);
-
- TH1F *hl=(TH1F*)gROOT->FindObject("hl");
- if (!hl) hl=new TH1F("hl","LAMBDA resolution",50,-20,20);
- hl->SetFillColor(4);
-
- TH1F *hpt=(TH1F*)gROOT->FindObject("hpt");
- if (!hpt) hpt=new TH1F("hpt","Relative Pt resolution",30,-10.,10.);
- hpt->SetFillColor(2);
-
- TH1F *hmpt=(TH1F*)gROOT->FindObject("hmpt");
- if (!hmpt)
- hmpt=new TH1F("hmpt","Transverse impact parameter",30,-777,777);
- hmpt->SetFillColor(6);
-
- TH1F *hz=(TH1F*)gROOT->FindObject("hz");
- if (!hz) hz=new TH1F("hz","Longitudinal impact parameter",30,-777,777);
-
-
- Int_t nb=100;
- TH1F *hgood=(TH1F*)gROOT->FindObject("hgood");
- if (!hgood) hgood=new TH1F("hgood","Good tracks",nb,ptcutl,ptcuth);
-
- TH1F *hfound=(TH1F*)gROOT->FindObject("hfound");
- if (!hfound) hfound=new TH1F("hfound","Found tracks",nb,ptcutl,ptcuth);
-
- TH1F *hfake=(TH1F*)gROOT->FindObject("hfake");
- if (!hfake) hfake=new TH1F("hfake","Fake tracks",nb,ptcutl,ptcuth);
-
- TH1F *hg=(TH1F*)gROOT->FindObject("hg");
- if (!hg) hg=new TH1F("hg","Efficiency for good tracks",nb,ptcutl,ptcuth);
- hg->SetLineColor(4); hg->SetLineWidth(2);
-
- TH1F *hf=(TH1F*)gROOT->FindObject("hf");
- if (!hf) hf=new TH1F("hf","Efficiency for fake tracks",nb,ptcutl,ptcuth);
- hf->SetFillColor(1); hf->SetFillStyle(3013); hf->SetLineWidth(2);
-
-
- TH1F *he=(TH1F*)gROOT->FindObject("he");
- if (!he)
- he =new TH1F("he","dE/dX for pions with 0.4<p<0.5 GeV/c",50,0.,200.);
-
- TH2F *hep=(TH2F*)gROOT->FindObject("hep");
- if (!hep) hep=new TH2F("hep","dE/dX vs momentum",50,0.,2.,50,0.,400.);
- hep->SetMarkerStyle(8);
- hep->SetMarkerSize(0.4);
-
-
- Char_t fname[100];
- sprintf(fname,"%s/GoodTracksITS.root",dir);
-
- TFile *refFile=TFile::Open(fname,"old");
- if (!refFile || !refFile->IsOpen()) {
- ::Info("AliITSUComparison.C","Marking good tracks (will take a while)...");
- if (GoodTracksITS(dir)) {
- ::Error("AliITSUComparison.C","Can't generate the reference file !");
- return 1;
- }
- }
- refFile=TFile::Open(fname,"old");
- if (!refFile || !refFile->IsOpen()) {
- ::Error("AliITSUComparison.C","Can't open the reference file !");
- return 1;
- }
-
- TTree *itsTree=(TTree*)refFile->Get("itsTree");
- if (!itsTree) {
- ::Error("AliITSUComparison.C","Can't get the reference tree !");
- return 2;
- }
- TBranch *branch=itsTree->GetBranch("ITS");
- if (!branch) {
- ::Error("AliITSUComparison.C","Can't get the ITS branch !");
- return 3;
- }
- TClonesArray dummy("AliTrackReference",1000), *refs=&dummy;
- branch->SetAddress(&refs);
-
-
- sprintf(fname,"%s/AliESDs.root",dir);
- TFile *ef=TFile::Open(fname);
- if ((!ef)||(!ef->IsOpen())) {
- sprintf(fname,"%s/AliESDits.root",dir);
- ef=TFile::Open(fname);
- if ((!ef)||(!ef->IsOpen())) {
- ::Error("AliITSUComparison.C","Can't open AliESDits.root !");
- return 4;
- }
- }
- AliESDEvent* event = new AliESDEvent();
- TTree* esdTree = (TTree*) ef->Get("esdTree");
- if (!esdTree) {
- ::Error("AliITSComparison.C", "no ESD tree found");
- return 6;
- }
- event->ReadFromTree(esdTree);
-
-
- //******* Loop over events *********
- Int_t e=0;
- while (esdTree->GetEvent(e)) {
- cout<<endl<<endl<<"********* Processing event number: "<<e<<"*******\n";
-
- Int_t nentr=event->GetNumberOfTracks();
- allfound+=nentr;
-
- if (itsTree->GetEvent(e++)==0) {
- cerr<<"No reconstructable tracks !\n";
- continue;
- }
-
- Int_t ngood=refs->GetEntriesFast();
- allgood+=ngood;
-
- const Int_t MAX=15000;
- Int_t notf[MAX], nnotf=0;
- Int_t fake[MAX], nfake=0;
- Int_t mult[MAX], numb[MAX], nmult=0;
- Int_t k;
- for (k=0; k<ngood; k++) {
- AliTrackReference *ref=(AliTrackReference*)refs->UncheckedAt(k);
- Int_t lab=ref->Label(), tlab=-1;
- Float_t ptg=TMath::Sqrt(ref->Px()*ref->Px() + ref->Py()*ref->Py());
-
- if (ptg<ptcutl) continue;
- if (ptg>ptcuth) continue;
-
- allselected++;
-
- hgood->Fill(ptg);
-
- AliESDtrack *esd=0;
- Int_t cnt=0;
- for (Int_t i=0; i<nentr; i++) {
- AliESDtrack *t=event->GetTrack(i);
- UInt_t status=t->GetStatus();
-
- if ((status&AliESDtrack::kITSrefit)==0) continue;
-
- Int_t lbl=t->GetLabel();
- if (lab==TMath::Abs(lbl)) {
- if (cnt==0) {esd=t; tlab=lbl;}
- if (lbl> 0) {esd=t; tlab=lbl;}
- cnt++;
- }
- }
- if (cnt==0) {
- notf[nnotf++]=lab;
- continue;
- } else if (cnt>1){
- mult[nmult]=lab;
- numb[nmult]=cnt; nmult++;
- }
-
- if (lab==tlab) hfound->Fill(ptg);
- else {
- fake[nfake++]=lab;
- hfake->Fill(ptg);
- }
-
- Double_t alpha=esd->GetAlpha(),xv,par[5];
- esd->GetExternalParameters(xv,par);
- Float_t phi=TMath::ASin(par[2]) + alpha;
- if (phi<-TMath::Pi()) phi+=2*TMath::Pi();
- if (phi>=TMath::Pi()) phi-=2*TMath::Pi();
- Float_t lam=TMath::ATan(par[3]);
- Float_t pt_1=TMath::Abs(par[4]);
-
- Float_t phig=TMath::ATan2(ref->Py(),ref->Px());
- hp->Fill((phi - phig)*1000.);
-
- Float_t lamg=TMath::ATan2(ref->Pz(),ptg);
- hl->Fill((lam - lamg)*1000.);
-
- Float_t d,z; esd->GetImpactParameters(d,z);
- hmpt->Fill(10000*d);
- hz->Fill(10000*z);
-
- hpt->Fill((pt_1 - 1/ptg)/(1/ptg)*100.);
-
- Float_t mom=(TMath::Abs(pt_1)>1e-5) ? 1./(pt_1*TMath::Cos(lam)) : 1e+5;
- Float_t dedx=esd->GetITSsignal();
- hep->Fill(mom,dedx,1.);
-
- Int_t pdg=(Int_t)ref->GetLength(); //this is particle's PDG !
-
- if (TMath::Abs(pdg)==211) //pions
- if (mom>0.4 && mom<0.5) he->Fill(dedx,1.);
-
- }
-
- cout<<"\nList of Not found tracks :\n";
- for (k=0; k<nnotf; k++){
- cout<<notf[k]<<"\t";
- if ((k%9)==8) cout<<"\n";
- }
- cout<<"\n\nList of fake tracks :\n";
- for (k=0; k<nfake; k++){
- cout<<fake[k]<<"\t";
- if ((k%9)==8) cout<<"\n";
- }
- cout<<"\n\nList of multiple found tracks :\n";
- for (k=0; k<nmult; k++) {
- cout<<"id. "<<mult[k]
- <<" found - "<<numb[k]<<"times\n";
- }
- cout<<endl;
-
- cout<<"Number of found tracks : "<<nentr<<endl;
- cout<<"Number of \"good\" tracks : "<<ngood<<endl;
-
- refs->Clear();
- } //***** End of the loop over events
-
- delete event;
- delete esdTree;
- ef->Close();
-
- delete itsTree;
- refFile->Close();
-
- Stat_t ng=hgood->GetEntries(), nf=hfound->GetEntries();
- if (ng!=0) cout<<"\n\nIntegral efficiency is about "<<nf/ng*100.<<" %\n";
- cout<<"Total number selected of \"good\" tracks ="<<allselected<<endl<<endl;
- cout<<"Total number of found tracks ="<<allfound<<endl;
- cout<<"Total number of \"good\" tracks ="<<allgood<<endl;
- cout<<endl;
-
- gStyle->SetOptStat(111110);
- gStyle->SetOptFit(1);
-
- TCanvas *c1=new TCanvas("c1","",0,0,700,850);
-
- Int_t minc=33;
-
- TPad *p1=new TPad("p1","",0,0.3,.5,.6); p1->Draw();
- p1->cd(); p1->SetFillColor(42); p1->SetFrameFillColor(10);
- hp->SetFillColor(4); hp->SetXTitle("(mrad)");
- if (hp->GetEntries()<minc) hp->Draw(); else hp->Fit("gaus"); c1->cd();
-
- TPad *p2=new TPad("p2","",0.5,.3,1,.6); p2->Draw();
- p2->cd(); p2->SetFillColor(42); p2->SetFrameFillColor(10);
- hl->SetXTitle("(mrad)");
- if (hl->GetEntries()<minc) hl->Draw(); else hl->Fit("gaus"); c1->cd();
-
- TPad *p3=new TPad("p3","",0,0,0.5,0.3); p3->Draw();
- p3->cd(); p3->SetFillColor(42); p3->SetFrameFillColor(10);
- hpt->SetXTitle("(%)");
- if (hpt->GetEntries()<minc) hpt->Draw(); else hpt->Fit("gaus"); c1->cd();
-
- TPad *p4=new TPad("p4","",0.5,0,1,0.3); p4->Draw();
- p4->cd(); p4->SetFillColor(42); p4->SetFrameFillColor(10);
- hmpt->SetXTitle("(micron)");
- if (hmpt->GetEntries()<minc) hmpt->Draw(); else hmpt->Fit("gaus");
- hz->Draw("same"); c1->cd();
-
-
- TPad *p5=new TPad("p5","",0,0.6,1,1); p5->Draw(); p5->cd();
- p5->SetFillColor(41); p5->SetFrameFillColor(10);
- hfound->Sumw2(); hgood->Sumw2(); hfake->Sumw2();
- hg->Divide(hfound,hgood,1,1.,"b");
- hf->Divide(hfake,hgood,1,1.,"b");
- hg->SetMaximum(1.4);
- hg->SetYTitle("Tracking efficiency");
- hg->SetXTitle("Pt (GeV/c)");
- hg->Draw();
-
- TLine *line1 = new TLine(ptcutl,1.0,ptcuth,1.0); line1->SetLineStyle(4);
- line1->Draw("same");
- TLine *line2 = new TLine(ptcutl,0.9,ptcuth,0.9); line2->SetLineStyle(4);
- line2->Draw("same");
-
- hf->SetFillColor(1);
- hf->SetFillStyle(3013);
- hf->SetLineColor(2);
- hf->SetLineWidth(2);
- hf->Draw("histsame");
- TText *text = new TText(0.461176,0.248448,"Fake tracks");
- text->SetTextSize(0.05);
- text->Draw();
- text = new TText(0.453919,1.11408,"Good tracks");
- text->SetTextSize(0.05);
- text->Draw();
-
- TCanvas *c2=new TCanvas("c2","",320,32,530,590);
- TPad *p6=new TPad("p6","",0.,0.,1.,.5); p6->Draw();
- p6->cd(); p6->SetFillColor(42); p6->SetFrameFillColor(10);
- he->SetFillColor(2); he->SetFillStyle(3005);
- he->SetXTitle("Arbitrary Units");
- if (he->GetEntries()<minc) he->Draw(); else he->Fit("gaus"); c2->cd();
-
- TPad *p7=new TPad("p7","",0.,0.5,1.,1.); p7->Draw();
- p7->cd(); p7->SetFillColor(42); p7->SetFrameFillColor(10);
- hep->SetXTitle("p (Gev/c)"); hep->SetYTitle("dE/dX (Arb. Units)");
- hep->Draw(); c1->cd();
-
- TFile fc("AliITSUComparison.root","RECREATE");
- c1->Write();
- c2->Write();
- fc.Close();
-
- gBenchmark->Stop("AliITSUComparison");
- gBenchmark->Show("AliITSUComparison");
-
- return 0;
-}
-
-
-
-Int_t GoodTracksITS(const Char_t *dir) {
- if (gAlice) {
- delete AliRunLoader::Instance();
- delete gAlice;//if everything was OK here it is already NULL
- gAlice = 0x0;
- }
-
- Char_t fname[100];
- sprintf(fname,"%s/galice.root",dir);
-
- AliRunLoader *rl = AliRunLoader::Open(fname,"COMPARISON");
- if (!rl) {
- ::Error("GoodTracksITS","Can't start session !");
- return 1;
- }
-
- rl->LoadgAlice();
- rl->LoadHeader();
- rl->LoadKinematics();
-
- AliITSULoader* itsl = (AliITSULoader*)rl->GetLoader("ITSLoader");
- if (itsl == 0x0) {
- ::Error("GoodTracksITS","Can not find the ITSLoader");
- delete rl;
- return 4;
- }
- itsl->LoadRecPoints();
-
-
- Int_t nev=rl->GetNumberOfEvents();
- ::Info("GoodTracksITS","Number of events : %d\n",nev);
-
- sprintf(fname,"%s/GoodTracksTPC.root",dir);
- TFile *tpcFile=TFile::Open(fname);
- if ((!tpcFile)||(!tpcFile->IsOpen())) {
- ::Error("GoodTracksITS","Can't open the GoodTracksTPC.root !");
- delete rl;
- return 5;
- }
- TClonesArray dum("AliTrackReference",1000), *tpcRefs=&dum;
- TTree *tpcTree=(TTree*)tpcFile->Get("tpcTree");
- if (!tpcTree) {
- ::Error("GoodTracksITS","Can't get the TPC reference tree !");
- delete rl;
- return 6;
- }
- TBranch *tpcBranch=tpcTree->GetBranch("TPC");
- if (!tpcBranch) {
- ::Error("GoodTracksITS","Can't get the TPC reference branch !");
- delete rl;
- return 7;
- }
- tpcBranch->SetAddress(&tpcRefs);
-
- sprintf(fname,"%s/GoodTracksITS.root",dir);
- TFile *itsFile=TFile::Open(fname,"recreate");
- TClonesArray dummy2("AliTrackReference",1000), *itsRefs=&dummy2;
- TTree itsTree("itsTree","Tree with info about the reconstructable ITS tracks");
- itsTree.Branch("ITS",&itsRefs);
-
- //******** Loop over generated events
- for (Int_t e=0; e<nev; e++) {
- Int_t k;
-
- rl->GetEvent(e); itsFile->cd();
-
- Int_t np = rl->GetHeader()->GetNtrack();
- cout<<"Event "<<e<<" Number of particles: "<<np<<endl;
-
- //******** Fill the "good" masks
- Int_t *good=new Int_t[np]; for (k=0; k<np; k++) good[k]=0;
-
- TTree *cTree=itsl->TreeR();
- if (!cTree) {
- ::Error("GoodTracksITS","Can't get the cluster tree !");
- delete rl;
- return 8;
- }
-
- const Int_t nLayers=7;
- TBranch *branch[nLayers];
- TClonesArray clusters[nLayers];
- for (Int_t layer=0; layer<nLayers; layer++) {
- TClonesArray *ptr =
- new(clusters+layer) TClonesArray("AliITSUClusterPix",1000);
- Char_t bname[33];
- sprintf(bname,"ITSRecPoints%d\0",layer);
- branch[layer]=cTree->GetBranch(bname);
- if (!branch[layer]) {
- ::Error("GoodTracksITS","Can't get the clusters branch !");
- delete rl;
- return 9;
- }
- branch[layer]->SetAddress(&ptr);
- }
-
- Int_t entr=(Int_t)cTree->GetEntries();
- for (k=0; k<entr; k++) {
- cTree->GetEvent(k);
- for (Int_t lay=0; lay<nLayers; lay++) {
- Int_t ncl=clusters[lay].GetEntriesFast(); if (ncl==0) continue;
- while (ncl--) {
- AliITSUClusterPix *pnt=
- (AliITSUClusterPix*)clusters[lay].UncheckedAt(ncl);
- Int_t l0=pnt->GetLabel(0);
- if (l0>=np) {
-// cerr<<"Wrong label: "<<l0<<endl;
- continue;
- }
- Int_t l1=pnt->GetLabel(1);
- if (l1>=np) {
-// cerr<<"Wrong label: "<<l1<<endl;
- continue;
- }
- Int_t l2=pnt->GetLabel(2);
- if (l2>=np) {
-// cerr<<"Wrong label: "<<l2<<endl;
- continue;
- }
- Int_t mask=1<<lay;
- if (l0>=0) good[l0]|=mask;
- if (l1>=0) good[l1]|=mask;
- if (l2>=0) good[l2]|=mask;
- }
- clusters[lay].Clear();
- }
- }
-
-
-
- //****** select tracks which are "good" enough
- AliStack* stack = rl->Stack();
-
- tpcTree->GetEvent(e);
- Int_t nk=tpcRefs->GetEntriesFast();
- Int_t nt=0;
- for (k=0; k<nk; k++) {
- AliTrackReference *tpcRef=(AliTrackReference *)tpcRefs->UncheckedAt(k);
- Int_t lab=tpcRef->Label();
- if (good[lab] != 0x7F) continue;
- TParticle *p = (TParticle*)stack->Particle(lab);
- if (p == 0x0) {
- cerr<<"Can not get particle "<<lab<<endl;
- continue;
- }
-
- AliTrackReference *ref=new((*itsRefs)[nt]) AliTrackReference(*tpcRef);
- ref->SetMomentum(p->Px(),p->Py(),p->Pz());
- ref->SetPosition(p->Vx(),p->Vy(),p->Vz());
- nt++;
- }
- tpcRefs->Clear();
-
- itsTree.Fill();
- itsRefs->Clear();
-
- delete[] good;
-
- } //*** end of the loop over generated events
-
- itsTree.Write();
- itsFile->Close();
-
- delete tpcTree;
- tpcFile->Close();
-
- delete rl;
- return 0;
-}
-
-
+++ /dev/null
-/****************************************************************************
- * A standalone comparison macro for the upgraded ITS. *
- * *
- * Creates list of "trackable" tracks, *
- * calculates efficiency, resolutions etc. *
- * The ESD tracks must be in an appropriate state: kITSrefit *
- * *
- * The efficiency and resolutions are calculated *
- * wrt "primary-like" pions within the acceptance tan(lambda)~[-1,1] *
- * and for tracks having all 7 clusters correctly assigned. *
- * *
- * Before running, load the ITSU libraries: *
- * gSystem->Load("libITSUpgradeBase");gSystem->Load("libITSUpgradeRec"); *
- * *
- * Origin: I.Belikov, IPHC, Iouri.Belikov@iphc.cnrs.fr *
- ****************************************************************************/
-
-#if !defined(__CINT__) || defined(__MAKECINT__)
- #include <TMath.h>
- #include <TError.h>
- #include <Riostream.h>
- #include <TH1F.h>
- #include <TH2F.h>
- #include <TTree.h>
- #include <TParticle.h>
- #include <TCanvas.h>
- #include <TLine.h>
- #include <TText.h>
- #include <TBenchmark.h>
- #include <TStyle.h>
- #include <TFile.h>
- #include <TROOT.h>
-
- #include "AliStack.h"
- #include "AliHeader.h"
- #include "AliGenEventHeader.h"
- #include "AliTrackReference.h"
- #include "AliRunLoader.h"
- #include "AliRun.h"
- #include "AliESDEvent.h"
- #include "AliESDtrack.h"
-
- #include "UPGRADE/AliITSUClusterPix.h"
- #include "UPGRADE/AliITSULoader.h"
-#endif
-
-Int_t GoodTracksCooked(const Char_t *dir=".");
-
-extern AliRun *gAlice;
-extern TBenchmark *gBenchmark;
-extern TROOT *gROOT;
-
-static Int_t allgood=0;
-static Int_t allselected=0;
-static Int_t allfound=0;
-
-void root(TH1 *h) {
- Int_t nb=h->GetNbinsX();
- for (Int_t i=0; i<nb; i++) {
- Float_t c=h->GetBinContent(i);
- c=TMath::Sqrt(c);
- h->SetBinContent(i,c);
-
- Float_t e=h->GetBinError(i);
- if (c!=0) e = 0.5*e/c;
- h->SetBinError(i,e);
- }
-}
-void divide(TH1 *h) {
- Int_t nb=h->GetNbinsX();
- for (Int_t i=0; i<nb; i++) {
- Float_t c=h->GetBinContent(i);
- c *= h->GetBinCenter(i);
- h->SetBinContent(i,c);
-
- Float_t e=h->GetBinError(i);
- e *= h->GetBinCenter(i);
- h->SetBinError(i,e);
- }
-}
-
-Int_t AliITSUComparisonCooked
-(Float_t ptcutl=0.01, Float_t ptcuth=10., const Char_t *dir=".") {
- gBenchmark->Start("AliITSUComparisonCooked");
-
- ::Info("AliITSUComparisonCooked.C","Doing comparison...");
-
-
- TH1F *hp=(TH1F*)gROOT->FindObject("hp");
- if (!hp) hp=new TH1F("hp","PHI resolution",50,-20.,20.);
- hp->SetFillColor(4);
-
- TH1F *hl=(TH1F*)gROOT->FindObject("hl");
- if (!hl) hl=new TH1F("hl","LAMBDA resolution",50,-20,20);
- hl->SetFillColor(4);
-
- //TH1F *hz=(TH1F*)gROOT->FindObject("hz");
- //if (!hz) hz=new TH1F("hz","Longitudinal impact parameter",30,-777.,777.);
-
-
- Int_t nb=100;
- Float_t xbins[nb+1];
- Double_t a=TMath::Log(ptcuth/ptcutl)/nb;
- for (Int_t i=0; i<=nb; i++) xbins[i] = ptcutl*TMath::Exp(i*a);
-
- TH1F *hgood=(TH1F*)gROOT->FindObject("hgood");
- if (!hgood) hgood=new TH1F("hgood","Good tracks",nb,xbins);
-
- TH1F *hfound=(TH1F*)gROOT->FindObject("hfound");
- if (!hfound) hfound=new TH1F("hfound","Found tracks",nb,xbins);
-
- TH1F *hfake=(TH1F*)gROOT->FindObject("hfake");
- if (!hfake) hfake=new TH1F("hfake","Fake tracks",nb,xbins);
-
- TH1F *hg=(TH1F*)gROOT->FindObject("hg");
- if (!hg) hg=new TH1F("hg","Efficiency for good tracks",nb,xbins);
- hg->SetLineColor(4); hg->SetLineWidth(2);
-
- TH1F *hf=(TH1F*)gROOT->FindObject("hf");
- if (!hf) hf=new TH1F("hf","Efficiency for fake tracks",nb,xbins);
- hf->SetFillColor(1); hf->SetFillStyle(3013); hf->SetLineWidth(2);
-
- TH1F *hpt=(TH1F*)gROOT->FindObject("hpt");
- if (!hpt) hpt=new TH1F("hpt","Relative Pt resolution",nb,xbins);
- hpt->Sumw2();
- TH1F *hd=(TH1F*)gROOT->FindObject("hd");
- if (!hd)
- hd=new TH1F("hd","Transverse impact parameter",nb,xbins);
- hd->Sumw2();
-
-
- Char_t fname[100];
- sprintf(fname,"%s/GoodTracksCooked.root",dir);
-
- TFile *refFile=TFile::Open(fname,"old");
- if (!refFile || !refFile->IsOpen()) {
- ::Info("AliITSUComparisonCooked.C","Marking good tracks (will take a while)...");
- if (GoodTracksCooked(dir)) {
- ::Error("AliITSUComparisonCooked.C","Can't generate the reference file !");
- return 1;
- }
- }
- refFile=TFile::Open(fname,"old");
- if (!refFile || !refFile->IsOpen()) {
- ::Error("AliITSUComparisonCooked.C","Can't open the reference file !");
- return 1;
- }
-
- TTree *itsTree=(TTree*)refFile->Get("itsTree");
- if (!itsTree) {
- ::Error("AliITSUComparisonCooked.C","Can't get the reference tree !");
- return 2;
- }
- TBranch *branch=itsTree->GetBranch("ITS");
- if (!branch) {
- ::Error("AliITSUComparisonCooked.C","Can't get the ITS branch !");
- return 3;
- }
- TClonesArray dummy("AliTrackReference",1000), *refs=&dummy;
- branch->SetAddress(&refs);
-
-
- sprintf(fname,"%s/AliESDs.root",dir);
- TFile *ef=TFile::Open(fname);
- if ((!ef)||(!ef->IsOpen())) {
- sprintf(fname,"%s/AliESDits.root",dir);
- ef=TFile::Open(fname);
- if ((!ef)||(!ef->IsOpen())) {
- ::Error("AliITSUComparisonCooked.C","Can't open AliESDits.root !");
- return 4;
- }
- }
- AliESDEvent* event = new AliESDEvent();
- TTree* esdTree = (TTree*) ef->Get("esdTree");
- if (!esdTree) {
- ::Error("AliITSComparison.C", "no ESD tree found");
- return 6;
- }
- event->ReadFromTree(esdTree);
-
-
- //******* Loop over events *********
- Int_t e=0;
- while (esdTree->GetEvent(e)) {
- cout<<endl<<endl<<"********* Processing event number: "<<e<<"*******\n";
-
- Int_t nentr=event->GetNumberOfTracks();
- allfound+=nentr;
-
- if (itsTree->GetEvent(e++)==0) {
- cerr<<"No reconstructable tracks !\n";
- continue;
- }
-
- Int_t ngood=refs->GetEntriesFast();
- allgood+=ngood;
-
- const Int_t MAX=15000;
- Int_t notf[MAX], nnotf=0;
- Int_t fake[MAX], nfake=0;
- Int_t mult[MAX], numb[MAX], nmult=0;
- Int_t k;
- for (k=0; k<ngood; k++) {
- AliTrackReference *ref=(AliTrackReference*)refs->UncheckedAt(k);
- Int_t lab=ref->Label(), tlab=-1;
- Float_t ptg=TMath::Sqrt(ref->Px()*ref->Px() + ref->Py()*ref->Py());
-
- Int_t pdg=(Int_t)ref->GetLength(); //this is particle's PDG !
- if (TMath::Abs(pdg)!=211) continue; //select pions only
-
- if (ptg<ptcutl) continue;
- if (ptg>ptcuth) continue;
-
- allselected++;
-
- hgood->Fill(ptg);
-
- AliESDtrack *esd=0;
- Int_t cnt=0;
- for (Int_t i=0; i<nentr; i++) {
- AliESDtrack *t=event->GetTrack(i);
- UInt_t status=t->GetStatus();
-
- if ((status&AliESDtrack::kITSrefit)==0) continue;
- if (t->GetITSclusters(0)<4) continue;
-
- Int_t lbl=t->GetLabel();
- if (lab==TMath::Abs(lbl)) {
- if (cnt==0) {esd=t; tlab=lbl;}
- if (lbl> 0) {esd=t; tlab=lbl;}
- cnt++;
- }
- }
- if (cnt==0) {
- notf[nnotf++]=lab;
- continue;
- } else if (cnt>1){
- mult[nmult]=lab;
- numb[nmult]=cnt; nmult++;
- }
-
- if (lab==tlab) hfound->Fill(ptg);
- else {
- fake[nfake++]=lab;
- hfake->Fill(ptg);
- }
-
- if (esd->GetLabel()<0) continue; //resolutions for good tracks only
-
- Double_t alpha=esd->GetAlpha(),xv,par[5];
- esd->GetExternalParameters(xv,par);
- Float_t phi=TMath::ASin(par[2]) + alpha;
- if (phi<-TMath::Pi()) phi+=2*TMath::Pi();
- if (phi>=TMath::Pi()) phi-=2*TMath::Pi();
- Float_t lam=TMath::ATan(par[3]);
- Float_t pt_1=TMath::Abs(par[4]);
-
- Float_t phig=TMath::ATan2(ref->Py(),ref->Px());
- hp->Fill((phi - phig)*1000.);
-
- Float_t lamg=TMath::ATan2(ref->Pz(),ptg);
- hl->Fill((lam - lamg)*1000.);
-
- Float_t d,z; esd->GetImpactParameters(d,z);
- d*=10000; //microns
- Float_t w=d*d;
- hd->Fill(ptg, w);
- //z*=10000; //microns
- //hz->Fill(z);
-
- w=(pt_1 - 1/ptg)*100 * (pt_1-1/ptg)*100;
- hpt->Fill(ptg, w);
-
- }
-
- cout<<"\nList of Not found tracks :\n";
- for (k=0; k<nnotf; k++){
- cout<<notf[k]<<"\t";
- if ((k%9)==8) cout<<"\n";
- }
- cout<<"\n\nList of fake tracks :\n";
- for (k=0; k<nfake; k++){
- cout<<fake[k]<<"\t";
- if ((k%9)==8) cout<<"\n";
- }
- cout<<"\n\nList of multiple found tracks :\n";
- for (k=0; k<nmult; k++) {
- cout<<"id. "<<mult[k]
- <<" found - "<<numb[k]<<"times\n";
- }
- cout<<endl;
-
- cout<<"Number of found tracks : "<<nentr<<endl;
- cout<<"Number of \"good\" tracks : "<<ngood<<endl;
-
- refs->Clear();
- } //***** End of the loop over events
-
- delete event;
- delete esdTree;
- ef->Close();
-
- delete itsTree;
- refFile->Close();
-
- Stat_t ng=hgood->GetEntries(), nf=hfound->GetEntries();
- if (ng!=0) cout<<"\n\nIntegral efficiency is about "<<nf/ng*100.<<" %\n";
- cout<<"Total number selected of \"good\" tracks ="<<allselected<<endl<<endl;
- cout<<"Total number of found tracks ="<<allfound<<endl;
- cout<<"Total number of \"good\" tracks ="<<allgood<<endl;
- cout<<endl;
-
- gStyle->SetOptStat(111110);
- gStyle->SetOptFit(1);
-
- TCanvas *c1=new TCanvas("c1","",0,0,700,850);
-
- Int_t minc=33;
-
- TPad *p1=new TPad("p1","",0,0.3,.5,.6); p1->Draw();
- p1->cd(); p1->SetFillColor(42); p1->SetFrameFillColor(10);
- hp->SetFillColor(4); hp->SetXTitle("(mrad)");
- if (hp->GetEntries()<minc) hp->Draw(); else hp->Fit("gaus"); c1->cd();
-
- TPad *p2=new TPad("p2","",0.5,.3,1,.6); p2->Draw();
- p2->cd(); p2->SetFillColor(42); p2->SetFrameFillColor(10);
- hl->SetXTitle("(mrad)");
- if (hl->GetEntries()<minc) hl->Draw(); else hl->Fit("gaus"); c1->cd();
-
- TPad *p3=new TPad("p3","",0,0,0.5,0.3);
- p3->SetLogx(); p3->SetGridx(); p3->SetGridy();
- p3->Draw();
- p3->cd(); p3->SetFillColor(42); p3->SetFrameFillColor(10);
- hpt->SetXTitle("p_{T} (GeV/c)");
- hpt->SetYTitle("(%)");
- TH1F *hh=new TH1F(*hfound);
- //hh->Add(hfake);
- hpt->Divide(hh);
- root(hpt);
- divide(hpt);
- hpt->Draw(); c1->cd();
-
- TPad *p4=new TPad("p4","",0.5,0,1,0.3);
- p4->SetLogx(); p4->SetGridx(); p4->SetGridy();
- p4->Draw();
- p4->cd(); p4->SetFillColor(42); p4->SetFrameFillColor(10);
- hd->SetXTitle("p_{T} (GeV/c)");
- hd->SetYTitle("(micron)");
- hd->Divide(hh);
- root(hd);
- hd->Draw(); c1->cd();
-
- //hz->Draw("same"); c1->cd();
-
-
- TPad *p5=new TPad("p5","",0,0.6,1,1);
- p5->SetLogx(); p5->SetGridx(); p5->SetGridy();
- p5->Draw(); p5->cd();
- p5->SetFillColor(41); p5->SetFrameFillColor(10);
- hfound->Sumw2(); hgood->Sumw2(); hfake->Sumw2();
- hg->Divide(hfound,hgood,1,1.,"b");
- hf->Divide(hfake,hgood,1,1.,"b");
- hg->SetYTitle("Tracking efficiency (%)");
- hg->SetXTitle("p_{T} (GeV/c)");
- hg->Scale(100);
- hg->Draw();
-
- hf->SetFillColor(1);
- hf->SetFillStyle(3013);
- hf->SetLineColor(2);
- hf->SetLineWidth(2);
- hf->Scale(100);
- hf->Draw("histsame");
- TText *text = new TText(0.4, 20., "Fake tracks");
- text->SetTextSize(0.05);
- text->Draw();
- text = new TText(0.4, 80., "Good tracks");
- text->SetTextSize(0.05);
- text->Draw();
-
- TFile fc("AliITSUComparisonCooked.root","RECREATE");
- c1->Write();
- fc.Close();
-
- gBenchmark->Stop("AliITSUComparisonCooked");
- gBenchmark->Show("AliITSUComparisonCooked");
-
- return 0;
-}
-
-
-
-Int_t GoodTracksCooked(const Char_t *dir) {
- if (gAlice) {
- delete AliRunLoader::Instance();
- delete gAlice;//if everything was OK here it is already NULL
- gAlice = 0x0;
- }
-
- Char_t fname[100];
- sprintf(fname,"%s/galice.root",dir);
-
- AliRunLoader *rl = AliRunLoader::Open(fname,"COMPARISON");
- if (!rl) {
- ::Error("GoodTracksCooked","Can't start session !");
- return 1;
- }
-
- rl->LoadgAlice();
- rl->LoadHeader();
- rl->LoadKinematics();
-
- AliITSULoader* itsl = (AliITSULoader*)rl->GetLoader("ITSLoader");
- if (itsl == 0x0) {
- ::Error("GoodTracksCooked","Can not find the ITSLoader");
- delete rl;
- return 4;
- }
- itsl->LoadRecPoints();
-
-
- Int_t nev=rl->GetNumberOfEvents();
- ::Info("GoodTracksCooked","Number of events : %d\n",nev);
-
- sprintf(fname,"%s/GoodTracksCooked.root",dir);
- TFile *itsFile=TFile::Open(fname,"recreate");
- TClonesArray dummy2("AliTrackReference",1000), *itsRefs=&dummy2;
- TTree itsTree("itsTree","Tree with info about the reconstructable ITS tracks");
- itsTree.Branch("ITS",&itsRefs);
-
- //******** Loop over generated events
- for (Int_t e=0; e<nev; e++) {
- Int_t k;
-
- rl->GetEvent(e); itsFile->cd();
-
- Int_t np = rl->GetHeader()->GetNtrack();
- cout<<"Event "<<e<<" Number of particles: "<<np<<endl;
-
- AliGenEventHeader *h=rl->GetHeader()->GenEventHeader();
- TArrayF vtx(3);
- h->PrimaryVertex(vtx);
-
- Bool_t skip=kFALSE;
- if (TMath::Abs(vtx[2]) > 10.) {
- cout<<"Skipping an event with Zv="<<vtx[2]<<endl;
- skip=kTRUE;
- }
-
- //******** Fill the "good" masks
- Int_t *good=new Int_t[np]; for (k=0; k<np; k++) good[k]=0;
-
- TTree *cTree=itsl->TreeR();
- if (!cTree) {
- ::Error("GoodTracksCooked","Can't get the cluster tree !");
- delete rl;
- return 8;
- }
-
- const Int_t nLayers=7;
- TBranch *branch[nLayers];
- TClonesArray clusters[nLayers];
- for (Int_t layer=0; layer<nLayers; layer++) {
- TClonesArray *ptr =
- new(clusters+layer) TClonesArray("AliITSUClusterPix",1000);
- Char_t bname[33];
- sprintf(bname,"ITSRecPoints%d\0",layer);
- branch[layer]=cTree->GetBranch(bname);
- if (!branch[layer]) {
- ::Error("GoodTracksCooked","Can't get the clusters branch !");
- delete rl;
- return 9;
- }
- branch[layer]->SetAddress(&ptr);
- }
-
- Int_t entr=(Int_t)cTree->GetEntries();
- for (k=0; k<entr; k++) {
- cTree->GetEvent(k);
- for (Int_t lay=0; lay<nLayers; lay++) {
- Int_t ncl=clusters[lay].GetEntriesFast(); if (ncl==0) continue;
- while (ncl--) {
- AliITSUClusterPix *pnt=
- (AliITSUClusterPix*)clusters[lay].UncheckedAt(ncl);
- Int_t l0=pnt->GetLabel(0);
- if (l0>=np) {
-// cerr<<"Wrong label: "<<l0<<endl;
- continue;
- }
- Int_t l1=pnt->GetLabel(1);
- if (l1>=np) {
-// cerr<<"Wrong label: "<<l1<<endl;
- continue;
- }
- Int_t l2=pnt->GetLabel(2);
- if (l2>=np) {
-// cerr<<"Wrong label: "<<l2<<endl;
- continue;
- }
- Int_t mask=1<<lay;
- if (l0>=0) good[l0]|=mask;
- if (l1>=0) good[l1]|=mask;
- if (l2>=0) good[l2]|=mask;
- }
- clusters[lay].Clear();
- }
- }
-
-
-
- //****** select tracks which are "good" enough
- AliStack* stack = rl->Stack();
- Int_t nt=0;
- for (k=0; k<np; k++) {
- if (good[k] != 0x7F) continue;
-
- if (skip) continue; //No good primary vertex for this event
-
- TParticle *p = (TParticle*)stack->Particle(k);
- if (p == 0x0) {
- cerr<<"Can not get particle "<<k<<endl;
- continue;
- }
-
- if (p->Pt() <= 0.) continue;
- if (TMath::Abs(p->Pz()/p->Pt())>0.999) continue;
-
- Double_t dx=p->Vx()-vtx[0], dy=p->Vy()-vtx[1], dz=p->Vz()-vtx[2];
- if (TMath::Sqrt(dx*dx+dy*dy)>0.0001) continue; //Primary-like
- if (TMath::Abs(dz) > 0.0001) continue;
-
- AliTrackReference *ref=new((*itsRefs)[nt]) AliTrackReference();
- ref->SetLabel(k);
- Int_t pdg=p->GetPdgCode();
- ref->SetLength(pdg); //This will the particle's PDG !
- ref->SetMomentum(p->Px(),p->Py(),p->Pz());
- ref->SetPosition(p->Vx(),p->Vy(),p->Vz());
- nt++;
- }
-
- itsTree.Fill();
- itsRefs->Clear();
-
- delete[] good;
-
- } //*** end of the loop over generated events
-
- itsTree.Write();
- itsFile->Close();
-
- delete rl;
- return 0;
-}
-
-
+++ /dev/null
-/****************************************************************************
- * Legacy comparison macro, adapted to the upgraded ITS. *
- * *
- * Creates list of "trackable" tracks, *
- * calculates efficiency, resolutions etc. *
- * There is a possibility to run this macro over several events. *
- * *
- * Origin: I.Belikov, IPHC, Iouri.Belikov@iphc.cnrs.fr *
- * with several nice improvements by: M.Ivanov, GSI, m.ivanov@gsi.de *
- ****************************************************************************/
-
-#if !defined(__CINT__) || defined(__MAKECINT__)
- #include <TMath.h>
- #include <TError.h>
- #include <Riostream.h>
- #include <TH1F.h>
- #include <TH2F.h>
- #include <TTree.h>
- #include <TParticle.h>
- #include <TCanvas.h>
- #include <TLine.h>
- #include <TText.h>
- #include <TBenchmark.h>
- #include <TStyle.h>
- #include <TFile.h>
- #include <TROOT.h>
- #include <TGeoGlobalMagField.h>
-
- #include "AliStack.h"
- #include "AliHeader.h"
- #include "AliTrackReference.h"
- #include "AliRunLoader.h"
- #include "AliRun.h"
- #include "AliMagF.h"
- #include "AliESDEvent.h"
- #include "AliESDtrack.h"
-
- #include "Base/AliSimDigits.h"
- #include "Base/AliTPCParamSR.h"
- #include "Base/AliTPCLoader.h"
- #include "Base/AliTPCcalibDB.h"
- #include "Sim/AliTPC.h"
- #include "Rec/AliTPCClustersRow.h"
-
- #include "AliCDBManager.h"
-#endif
-
-Int_t GoodTracksTPC(const Char_t *dir=".");
-
-extern TBenchmark *gBenchmark;
-extern TROOT *gROOT;
-
-static Int_t allgood=0;
-static Int_t allselected=0;
-static Int_t allfound=0;
-
-Int_t AliTPCUComparison
-(Float_t ptcutl=0., Float_t ptcuth=2., const Char_t *dir=".") {
- gBenchmark->Start("AliTPCUComparison");
-
- ::Info("AliTPCUComparison.C","Doing comparison...");
-
-
-
- TH1F *hp=(TH1F*)gROOT->FindObject("hp");
- if (!hp) hp=new TH1F("hp","PHI resolution",50,-20.,20.);
- hp->SetFillColor(4);
-
- TH1F *hl=(TH1F*)gROOT->FindObject("hl");
- if (!hl) hl=new TH1F("hl","LAMBDA resolution",50,-20,20);
- hl->SetFillColor(4);
-
- TH1F *hpt=(TH1F*)gROOT->FindObject("hpt");
- if (!hpt) hpt=new TH1F("hpt","Relative Pt resolution",30,-10.,10.);
- hpt->SetFillColor(2);
-
- TH1F *hmpt=(TH1F*)gROOT->FindObject("hmpt");
- if (!hmpt)
- hmpt=new TH1F("hmpt","Relative Pt resolution (pt>4GeV/c)",30,-60,60);
- hmpt->SetFillColor(6);
-
-
- Int_t nb=100;
- TH1F *hgood=(TH1F*)gROOT->FindObject("hgood");
- if (!hgood) hgood=new TH1F("hgood","Good tracks",nb,ptcutl,ptcuth);
-
- TH1F *hfound=(TH1F*)gROOT->FindObject("hfound");
- if (!hfound) hfound=new TH1F("hfound","Found tracks",nb,ptcutl,ptcuth);
-
- TH1F *hfake=(TH1F*)gROOT->FindObject("hfake");
- if (!hfake) hfake=new TH1F("hfake","Fake tracks",nb,ptcutl,ptcuth);
-
- TH1F *hg=(TH1F*)gROOT->FindObject("hg");
- if (!hg) hg=new TH1F("hg","Efficiency for good tracks",nb,ptcutl,ptcuth);
- hg->SetLineColor(4); hg->SetLineWidth(2);
-
- TH1F *hf=(TH1F*)gROOT->FindObject("hf");
- if (!hf) hf=new TH1F("hf","Efficiency for fake tracks",nb,ptcutl,ptcuth);
- hf->SetFillColor(1); hf->SetFillStyle(3013); hf->SetLineWidth(2);
-
-
- TH1F *he=(TH1F*)gROOT->FindObject("he");
- if (!he)
- he =new TH1F("he","dE/dX for pions with 0.4<p<0.5 GeV/c",50,0.,100.);
-
- TH2F *hep=(TH2F*)gROOT->FindObject("hep");
- if (!hep) hep=new TH2F("hep","dE/dX vs momentum",50,0.,2.,50,0.,400.);
- hep->SetMarkerStyle(8);
- hep->SetMarkerSize(0.4);
-
-
- Char_t fname[100];
- sprintf(fname,"%s/GoodTracksTPC.root",dir);
-
- TFile *refFile=TFile::Open(fname,"old");
- if (!refFile || !refFile->IsOpen()) {
- ::Info("AliTPCUComparison.C","Marking good tracks (will take a while)...");
- if (GoodTracksTPC(dir)) {
- ::Error("AliTPCUComparison.C","Can't generate the reference file !");
- return 1;
- }
- }
- refFile=TFile::Open(fname,"old");
- if (!refFile || !refFile->IsOpen()) {
- ::Error("AliTPCUComparison.C","Can't open the reference file !");
- return 2;
- }
-
- TTree *tpcTree=(TTree*)refFile->Get("tpcTree");
- if (!tpcTree) {
- ::Error("AliTPCUComparison.C","Can't get the reference tree !");
- return 3;
- }
- TBranch *branch=tpcTree->GetBranch("TPC");
- if (!branch) {
- ::Error("AliTPCUComparison.C","Can't get the TPC branch !");
- return 4;
- }
- TClonesArray dummy("AliTrackReference",1000), *refs=&dummy;
- branch->SetAddress(&refs);
-
-
- sprintf(fname,"%s/AliESDs.root",dir);
- TFile *ef=TFile::Open(fname);
- if ((!ef)||(!ef->IsOpen())) {
- sprintf(fname,"%s/AliESDtpc.root",dir);
- ef=TFile::Open(fname);
- if ((!ef)||(!ef->IsOpen())) {
- ::Error("AliTPCUComparison.C","Can't open AliESDtpc.root !");
- return 5;
- }
- }
- AliESDEvent* event = new AliESDEvent();
- TTree* esdTree = (TTree*) ef->Get("esdTree");
- if (!esdTree) {
- ::Error("AliTPCUComparison.C", "no ESD tree found");
- return 6;
- }
- event->ReadFromTree(esdTree);
-
-
- //******* Loop over events *********
- Int_t e=0;
- while (esdTree->GetEvent(e)) {
- cout<<endl<<endl<<"********* Processing event number: "<<e<<"*******\n";
-
- Int_t nentr=event->GetNumberOfTracks();
- allfound+=nentr;
-
- if (tpcTree->GetEvent(e++)==0) {
- cerr<<"No reconstructable tracks !\n";
- continue;
- }
-
- Int_t ngood=refs->GetEntriesFast();
- allgood+=ngood;
-
- const Int_t MAX=15000;
- //MI change
- Int_t track_notfound[MAX], itrack_notfound=0;
- Int_t track_fake[MAX], itrack_fake=0;
- Int_t track_multifound[MAX],track_multifound_n[MAX],itrack_multifound=0;
-
- Int_t i;
- for (Int_t k=0; k<ngood; k++) {
- AliTrackReference *ref=(AliTrackReference*)refs->UncheckedAt(k);
- Int_t lab=ref->Label(), tlab=-1;
- Float_t ptg=TMath::Sqrt(ref->Px()*ref->Px() + ref->Py()*ref->Py());
-
- if (ptg<1e-33) continue; // for those not crossing 0 pad row
-
- if (ptg<ptcutl) continue;
- if (ptg>ptcuth) continue;
-
- allselected++;
-
- hgood->Fill(ptg);
-
- AliESDtrack *track=0;
- for (i=0; i<nentr; i++) {
- track=event->GetTrack(i);
- tlab=track->GetTPCLabel();
- if (lab==TMath::Abs(tlab)) break;
- }
- if (i==nentr) {
- track_notfound[itrack_notfound++]=lab;
- continue;
- }
-
- //MI change - addition
- Int_t micount=0;
- Int_t mi;
- AliESDtrack * mitrack;
- for (mi=0; mi<nentr; mi++) {
- mitrack=event->GetTrack(mi);
- if (lab==TMath::Abs(mitrack->GetTPCLabel())) micount++;
- }
- if (micount>1) {
- track_multifound[itrack_multifound]=lab;
- track_multifound_n[itrack_multifound]=micount;
- itrack_multifound++;
- }
- if ((track->GetStatus()&AliESDtrack::kTPCrefit)==0) continue;
- if (lab==tlab) hfound->Fill(ptg);
- else {
- track_fake[itrack_fake++]=lab;
- hfake->Fill(ptg);
- }
-
- Double_t pxpypz[3]; track->GetInnerPxPyPz(pxpypz);
- Float_t phi=TMath::ATan2(pxpypz[1],pxpypz[0]);
- if (phi<-TMath::Pi()) phi+=2*TMath::Pi();
- if (phi>=TMath::Pi()) phi-=2*TMath::Pi();
- Double_t pt=TMath::Sqrt(pxpypz[0]*pxpypz[0]+pxpypz[1]*pxpypz[1]);
- Float_t lam=TMath::ATan2(pxpypz[2],pt);
- Float_t pt_1=1/pt;
-
- Int_t pdg=(Int_t)ref->GetLength(); //this is particle's PDG !
-
- if (TMath::Abs(pdg)==11 && ptg>4.) {//high momentum electrons
- hmpt->Fill((pt_1 - 1/ptg)/(1/ptg)*100.);
- } else {
- Float_t phig=TMath::ATan2(ref->Py(),ref->Px());
- hp->Fill((phi - phig)*1000.);
-
- Float_t lamg=TMath::ATan2(ref->Pz(),ptg);
- hl->Fill((lam - lamg)*1000.);
-
- hpt->Fill((pt_1 - 1/ptg)/(1/ptg)*100.);
- }
-
- Float_t mom=pt/TMath::Cos(lam);
- Float_t dedx=track->GetTPCsignal();
- hep->Fill(mom,dedx,1.);
- if (TMath::Abs(pdg)==211) //pions
- if (mom>0.4 && mom<0.5) {
- he->Fill(dedx,1.);
- }
- }
-
- cout<<"\nList of Not found tracks :\n";
- for ( i = 0; i< itrack_notfound; i++){
- cout<<track_notfound[i]<<"\t";
- if ((i%5)==4) cout<<"\n";
- }
- cout<<"\nList of fake tracks :\n";
- for ( i = 0; i< itrack_fake; i++){
- cout<<track_fake[i]<<"\t";
- if ((i%5)==4) cout<<"\n";
- }
- cout<<"\nList of multiple found tracks :\n";
- for ( i=0; i<itrack_multifound; i++) {
- cout<<"id. "<<track_multifound[i]
- <<" found - "<<track_multifound_n[i]<<"times\n";
- }
-
- cout<<"Number of found tracks ="<<nentr<<endl;
- cout<<"Number of \"good\" tracks ="<<ngood<<endl;
-
- refs->Clear();
- }// ***** End of the loop over events
-
- delete event;
- delete esdTree;
- ef->Close();
-
- delete tpcTree;
- refFile->Close();
-
- Stat_t ng=hgood->GetEntries(), nf=hfound->GetEntries();
- if (ng!=0) cout<<"\n\nIntegral efficiency is about "<<nf/ng*100.<<" %\n";
- cout<<"Total number selected of \"good\" tracks ="<<allselected<<endl<<endl;
- cout<<"Total number of found tracks ="<<allfound<<endl;
- cout<<"Total number of \"good\" tracks ="<<allgood<<endl;
- cout<<endl;
-
- gStyle->SetOptStat(111110);
- gStyle->SetOptFit(1);
-
- TCanvas *c1=new TCanvas("c1","",0,0,700,850);
-
- Int_t minc=33;
-
- TPad *p1=new TPad("p1","",0,0.3,.5,.6); p1->Draw();
- p1->cd(); p1->SetFillColor(42); p1->SetFrameFillColor(10);
- hp->SetFillColor(4); hp->SetXTitle("(mrad)");
- if (hp->GetEntries()<minc) hp->Draw(); else hp->Fit("gaus"); c1->cd();
-
- TPad *p2=new TPad("p2","",0.5,.3,1,.6); p2->Draw();
- p2->cd(); p2->SetFillColor(42); p2->SetFrameFillColor(10);
- hl->SetXTitle("(mrad)");
- if (hl->GetEntries()<minc) hl->Draw(); else hl->Fit("gaus"); c1->cd();
-
- TPad *p3=new TPad("p3","",0,0,0.5,0.3); p3->Draw();
- p3->cd(); p3->SetFillColor(42); p3->SetFrameFillColor(10);
- hpt->SetXTitle("(%)");
- if (hpt->GetEntries()<minc) hpt->Draw(); else hpt->Fit("gaus"); c1->cd();
-
- TPad *p4=new TPad("p4","",0.5,0,1,0.3); p4->Draw();
- p4->cd(); p4->SetFillColor(42); p4->SetFrameFillColor(10);
- hmpt->SetXTitle("(%)");
- if (hmpt->GetEntries()<minc) hmpt->Draw(); else hmpt->Fit("gaus"); c1->cd();
-
- TPad *p5=new TPad("p5","",0,0.6,1,1); p5->Draw(); p5->cd();
- p5->SetFillColor(41); p5->SetFrameFillColor(10);
- hfound->Sumw2(); hgood->Sumw2(); hfake->Sumw2();
- hg->Divide(hfound,hgood,1,1.,"b");
- hf->Divide(hfake,hgood,1,1.,"b");
- hg->SetMaximum(1.4);
- hg->SetYTitle("Tracking efficiency");
- hg->SetXTitle("Pt (GeV/c)");
- hg->Draw();
-
- TLine *line1 = new TLine(ptcutl,1.0,ptcuth,1.0); line1->SetLineStyle(4);
- line1->Draw("same");
- TLine *line2 = new TLine(ptcutl,0.9,ptcuth,0.9); line2->SetLineStyle(4);
- line2->Draw("same");
-
- hf->SetFillColor(1);
- hf->SetFillStyle(3013);
- hf->SetLineColor(2);
- hf->SetLineWidth(2);
- hf->Draw("histsame");
- TText *text = new TText(0.461176,0.248448,"Fake tracks");
- text->SetTextSize(0.05);
- text->Draw();
- text = new TText(0.453919,1.11408,"Good tracks");
- text->SetTextSize(0.05);
- text->Draw();
-
-
-
- TCanvas *c2=new TCanvas("c2","",320,32,530,590);
-
- TPad *p6=new TPad("p6","",0.,0.,1.,.5); p6->Draw();
- p6->cd(); p6->SetFillColor(42); p6->SetFrameFillColor(10);
- he->SetFillColor(2); he->SetFillStyle(3005);
- he->SetXTitle("Arbitrary Units");
- if (he->GetEntries()<minc) he->Draw(); else he->Fit("gaus"); c2->cd();
-
- TPad *p7=new TPad("p7","",0.,0.5,1.,1.); p7->Draw();
- p7->cd(); p7->SetFillColor(42); p7->SetFrameFillColor(10);
- hep->SetXTitle("p (Gev/c)"); hep->SetYTitle("dE/dX (Arb. Units)");
- hep->Draw(); c1->cd();
-
- TFile fc("AliTPCUComparison.root","RECREATE");
- c1->Write();
- c2->Write();
- fc.Close();
-
- gBenchmark->Stop("AliTPCUComparison");
- gBenchmark->Show("AliTPCUComparison");
-
- return 0;
-}
-
-
-Int_t GoodTracksTPC(const Char_t *dir) {
- Char_t fname[100];
- sprintf(fname,"%s/galice.root",dir);
-
- AliRunLoader *rl = AliRunLoader::Open(fname,"COMPARISON");
- if (!rl) {
- ::Error("GoodTracksTPC","Can't start session !");
- return 1;
- }
-
- rl->LoadgAlice();
- rl->LoadHeader();
- rl->LoadKinematics();
- rl->LoadTrackRefs();
-
- AliTPCLoader *tpcl = (AliTPCLoader *)rl->GetLoader("TPCLoader");
- if (tpcl == 0x0) {
- ::Error("GoodTracksTPC","Can not find TPCLoader !");
- delete rl;
- return 2;
- }
-
- AliTPC *TPC=(AliTPC*)rl->GetAliRun()->GetDetector("TPC");
- Int_t ver = TPC->IsVersion();
- cout<<"TPC version "<<ver<<" has been found !\n";
- if (ver==1) tpcl->LoadRecPoints();
- else if (ver==2) tpcl->LoadDigits();
- else {
- ::Error("GoodTracksTPC","Wrong TPC version !");
- delete rl;
- return 3;
- }
-
- TGeoGlobalMagField::Instance()->SetField(
- new AliMagF("Maps","Maps", -1., -1., AliMagF::k5kG));
-
- AliCDBManager *man=AliCDBManager::Instance();
- man->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
- man->SetRun(0);
- AliTPCParamSR *digp=
- (AliTPCParamSR*)(AliTPCcalibDB::Instance()->GetParameters());
- if (!digp) {
- ::Error("AliTPCUComparison.C","TPC parameters have not been found !");
- delete rl;
- return 4;
- }
-
- Int_t nrow_up=digp->GetNRowUp();
- Int_t nrows=digp->GetNRowLow()+nrow_up;
- Int_t zero=digp->GetZeroSup();
- Int_t gap=Int_t(0.125*nrows), shift=Int_t(0.5*gap);
- Int_t good_number=Int_t(0.4*nrows);
-
- Int_t nev=rl->GetNumberOfEvents();
- ::Info("GoodTracksTPC","Number of events : %d\n",nev);
-
-
- sprintf(fname,"%s/GoodTracksTPC.root",dir);
- TFile *file=TFile::Open(fname,"recreate");
-
- TClonesArray dummy("AliTrackReference",1000), *refs=&dummy;
- TTree tpcTree("tpcTree","Tree with info about the reconstructable TPC tracks");
- tpcTree.Branch("TPC",&refs);
-
- //******** Loop over generated events
- for (Int_t e=0; e<nev; e++) {
- Int_t nt=0;
- refs->Clear();
-
- Int_t i;
-
- rl->GetEvent(e); file->cd();
-
- Int_t np = rl->GetHeader()->GetNtrack();
- cout<<"Event "<<e<<" Number of particles: "<<np<<endl;
-
- //******** Fill the "good" masks
- Int_t *good=new Int_t[np]; for (i=0; i<np; i++) good[i]=0;
- switch (ver) {
- case 1:
- /*
- {
- AliTPCClustersArray *pca=new AliTPCClustersArray, &ca=*pca;
- ca.Setup(digp);
- ca.SetClusterType("AliTPCcluster");
- ca.ConnectTree(tpcl->TreeR());
- Int_t nrows=Int_t(ca.GetTree()->GetEntries());
- for (Int_t n=0; n<nrows; n++) {
- AliSegmentID *s=ca.LoadEntry(n);
- Int_t sec,row;
- digp->AdjustSectorRow(s->GetID(),sec,row);
- AliTPCClustersRow &clrow = *ca.GetRow(sec,row);
- Int_t ncl=clrow.GetArray()->GetEntriesFast();
- while (ncl--) {
- AliTPCcluster *c=(AliTPCcluster*)clrow[ncl];
- Int_t lab=c->GetLabel(0);
- if (lab<0) continue; //noise cluster
- lab=TMath::Abs(lab);
-
- if (sec>=digp->GetNInnerSector())
- if (row==nrow_up-1) good[lab]|=0x4000;
- if (sec>=digp->GetNInnerSector())
- if (row==nrow_up-1-gap) good[lab]|=0x1000;
-
- if (sec>=digp->GetNInnerSector())
- if (row==nrow_up-1-shift) good[lab]|=0x2000;
- if (sec>=digp->GetNInnerSector())
- if (row==nrow_up-1-gap-shift) good[lab]|=0x800;
-
- good[lab]++;
- }
- ca.ClearRow(sec,row);
- }
- delete pca;
- }
- break;
- */
- case 2:
- {
- TTree *TD=tpcl->TreeD();
-
- AliSimDigits da, *digits=&da;
- TD->GetBranch("Segment")->SetAddress(&digits);
-
- Int_t *count = new Int_t[np];
- Int_t i;
- for (i=0; i<np; i++) count[i]=0;
-
- Int_t sectors_by_rows=(Int_t)TD->GetEntries();
- for (i=0; i<sectors_by_rows; i++) {
- if (!TD->GetEvent(i)) continue;
- Int_t sec,row;
- digp->AdjustSectorRow(digits->GetID(),sec,row);
- //cerr<<sec<<' '<<row<<'\r';
- digits->First();
- do { //Many thanks to J.Chudoba who noticed this
- Int_t it=digits->CurrentRow(), ip=digits->CurrentColumn();
- Short_t dig = digits->GetDigit(it,ip);
- Int_t idx0=digits->GetTrackID(it,ip,0);
- Int_t idx1=digits->GetTrackID(it,ip,1);
- Int_t idx2=digits->GetTrackID(it,ip,2);
- if (idx0>=0 && dig>=zero && idx0<np) count[idx0]+=1;
- if (idx1>=0 && dig>=zero && idx1<np) count[idx1]+=1;
- if (idx2>=0 && dig>=zero && idx2<np) count[idx2]+=1;
- } while (digits->Next());
- for (Int_t j=0; j<np; j++) {
- if (count[j]>1) {
- if (sec>=digp->GetNInnerSector())
- if (row==nrow_up-1 ) good[j]|=0x4000;
- if (sec>=digp->GetNInnerSector())
- if (row==nrow_up-1-gap) good[j]|=0x1000;
-
- if (sec>=digp->GetNInnerSector())
- if (row==nrow_up-1-shift) good[j]|=0x2000;
- if (sec>=digp->GetNInnerSector())
- if (row==nrow_up-1-gap-shift) good[j]|=0x800;
- good[j]++;
- }
- count[j]=0;
- }
- }
- delete[] count;
- }
- break;
- }
-
-
- //****** select tracks which are "good" enough
- AliStack* stack = rl->Stack();
- for (i=0; i<np; i++) {
- if ((good[i]&0x5000) != 0x5000)
- if ((good[i]&0x2800) != 0x2800) continue;
- if ((good[i]&0x7FF ) < good_number) continue;
- TParticle *p = (TParticle*)stack->Particle(i);
- if (p == 0x0) {
- cerr<<"Can not get particle "<<i<<endl;
- continue;
- }
- if (p->Pt() <= 0.) continue;
- if (TMath::Abs(p->Pz()/p->Pt())>0.999) continue;
-
- Double_t vx=p->Vx(),vy=p->Vy(),vz=p->Vz();
- if (TMath::Sqrt(vx*vx+vy*vy)>3.5) continue;
- if (TMath::Abs(vz) > 50.) continue;
-
- AliTrackReference *ref=new((*refs)[nt]) AliTrackReference();
-
- ref->SetLabel(i);
- Int_t pdg=p->GetPdgCode();
- ref->SetLength(pdg); //This will the particle's PDG !
- ref->SetMomentum(0.,0.,0.);
- ref->SetPosition(0.,0.,0.);
-
- nt++;
- }
-
- //**** check if there is also information at the entrance of the TPC
- TTree *TR=rl->TreeTR();
- TBranch *branch=TR->GetBranch("TrackReferences");
- if (branch==0) {
- ::Error("GoodTracksTPC","No track references !");
- delete rl;
- return 5;
- }
- TClonesArray tpcdummy("AliTrackReference",1000), *tpcRefs=&tpcdummy;
- branch->SetAddress(&tpcRefs);
- Int_t nr=(Int_t)TR->GetEntries();
- for (Int_t r=0; r<nr; r++) {
- //cerr<<r<<' '<<nr<<'\r';
- TR->GetEvent(r);
-
- Int_t nref = tpcRefs->GetEntriesFast();
- if (!nref) continue;
- AliTrackReference *tpcRef= 0x0;
- for (Int_t iref=0; iref<nref; ++iref) {
- tpcRef = (AliTrackReference*)tpcRefs->UncheckedAt(iref);
- if (tpcRef->DetectorId() == AliTrackReference::kTPC) break;
- tpcRef = 0x0;
- }
-
- if (!tpcRef) continue;
-
- Int_t j;
- AliTrackReference *ref=0;
- for (j=0; j<nt; j++) {
- ref=(AliTrackReference *)refs->UncheckedAt(j);
- if (ref->Label()==tpcRef->Label()) break;
- ref=0;
- }
- if (ref==0) continue;
-
- ref->SetMomentum(tpcRef->Px(),tpcRef->Py(),tpcRef->Pz());
- ref->SetPosition(tpcRef->LocalX(),tpcRef->LocalY(),tpcRef->Z());
-
- tpcRefs->Clear();
- }
-
- tpcTree.Fill();
-
- delete[] good;
- } //****** end of the loop over generated events
-
-
- tpcTree.Write();
- file->Close();
-
- delete rl;
- return 0;
-}
git://git.uio.no / u / mrichter / AliRoot.git / commitdiff