ITS/AliITSPlotTracksV1.C

   1 #include <iostream.h>
   2 #include <fstream.h>
   3
   4 void AliITSPlotTracksV1()
   5 {
   6         TFile *file = new TFile("AliITSComparisonV1.root");
   7         TTree *tree = (TTree*)file->Get("Eval");
   8
   9         if (!tree) {
  10                 cerr << "Evaluation tree not found!" << endl;
  11                 return;
  12         }
  13
  14         // histogram definition - I (efficiency)
  15         TH1D *hFindables = new TH1D("hFindables", "Findable tracks", 10, 0, 2);
  16         TH1D *hGood = new TH1D("hGood", "Good found tracks", 10, 0, 2);
  17         TH1D *hFake = new TH1D("hFake", "Fake found tracks", 10, 0, 2);
  18         TH1D *hRatioG = new TH1D("hRatioG", "", 10, 0, 2);
  19         TH1D *hRatioF = new TH1D("hRatioF", "", 10, 0, 2);
  20         hGood->Sumw2();
  21         hFake->Sumw2();
  22         hFindables->Sumw2();
  23         hRatioG->SetLineColor(kBlue);
  24         hRatioG->SetLineWidth(2);
  25         hRatioF->SetLineColor(kRed);
  26         hRatioF->SetLineWidth(2);
  27
  28         // histograms definition - II (resolution)
  29         TH1D *hPhi    = new TH1D("hPhi", "#Phi resolution", 50, -15., 15.);
  30         TH1D *hLambda = new TH1D("hLambda", "#lambda resolution", 50, -15., 15.);
  31         TH1D *hPt     = new TH1D("hPt", "Relative Pt resolution", 40, -10., 10.);
  32         TH1D *hDtot   = new TH1D("hDtot", "Total impact parameter distribution", 100, 0., 2000.);
  33         TH1D *hDr     = new TH1D("hDr", "Transv. impact parameter distribution", 50, -1000., 1000.);
  34         TH1D *hDz     = new TH1D("hDz", "Long. impact parameter distribution", 50, -1000., 1000.);
  35         hPhi->SetFillColor(4);
  36         hLambda->SetFillColor(4);
  37         hPt->SetFillColor(2);
  38         hDtot->SetFillColor(6);
  39         hDr->SetFillColor(kGreen);
  40         hDz->SetFillColor(kBlue);
  41
  42         // Evaluation tree settings
  43         Int_t labITS, labTPC, signC;
  44         Int_t i, j, tot = (Int_t)tree->GetEntries();
  45         Double_t difpt, diflambda, difphi, Dz, Dr, Dtot, ptg;
  46         tree->SetBranchAddress("labITS"   , &labITS   );
  47         tree->SetBranchAddress("labTPC"   , &labTPC   );
  48         tree->SetBranchAddress("difpt"    , &difpt    );
  49         tree->SetBranchAddress("diflambda", &diflambda);
  50         tree->SetBranchAddress("difphi"   , &difphi   );
  51         tree->SetBranchAddress("Dz"       , &Dz       );
  52         tree->SetBranchAddress("Dr"       , &Dr       );
  53         tree->SetBranchAddress("Dtot"     , &Dtot     );
  54         tree->SetBranchAddress("ptg"      , &ptg      );
  55         tree->SetBranchAddress("signC"    , &signC    );
  56
  57         // Filling the histogram of findable tracks (w.r. to momentum)
  58         for(i = 0; i < tot; i++) {
  59                 tree->GetEntry(i);
  60                 hFindables->Fill(ptg);
  61         }
  62
  63         // Filling the evaluation histograms
  64         Int_t neglabs = 0;
  65         for(i = 0; i < tot; i++) {
  66                 tree->GetEntry(i);
  67 //              if(signC<0) continue;
  68                 if (labITS < 0) neglabs++;
  69                 if (labITS >= 0) {
  70                         hGood->Fill(ptg);
  71                         hPt->Fill(difpt);
  72                         hLambda->Fill(diflambda);
  73                         hPhi->Fill(difphi);
  74                         hDtot->Fill(Dtot);
  75                         hDr->Fill(Dr);
  76                         hDz->Fill(Dz);
  77                 }
  78                 else {
  79                         hFake->Fill(ptg);
  80                         neglabs++;
  81                 }
  82         }
  83
  84         // Drawing
  85         cerr << "Findable tracks   : " << hFindables->GetEntries() << endl;
  86         cerr << "Good found tracks : " << hGood->GetEntries() << endl;
  87         cerr << "Fake found tracks : " << hFake->GetEntries() << endl;
  88
  89         gStyle->SetOptStat(111110);
  90         gStyle->SetOptFit(1);
  91
  92         TCanvas *c1 = new TCanvas("c1","Parameter resolutions",0,0,700,700);
  93         c1->Divide(2, 2, 0.001, 0.001);
  94         c1->cd(1); hPhi->SetXTitle("(mrad)"); hPhi->Draw(); hPhi->Fit("gaus", "Q");
  95         c1->cd(2); hLambda->SetXTitle("(mrad)"); hLambda->Draw(); hLambda->Fit("gaus", "Q");
  96         c1->cd(3); hPt->SetXTitle("(%)"); hPt->Draw(); hPt->Fit("gaus", "Q");
  97         c1->cd(4); hDtot->SetXTitle("(micron)"); hDtot->Draw();
  98         c1->Update();
  99
 100         TCanvas *c2 = new TCanvas("c2","Impact parameters resolution",100,100,700,400);
 101         c2->Divide(2, 1, 0.001, 0.001);
 102         c2->cd(1); hDr->SetXTitle("(micron)"); hDr->Draw(); hDr->Fit("gaus", "Q");
 103         c2->cd(2); hDz->SetXTitle("(micron)"); hDz->Draw(); hDz->Fit("gaus", "Q");
 104         c2->Update();
 105
 106         TCanvas *c3 = new TCanvas("c3","Momentum distributions",200,200,800,500);
 107         c3->Divide(3, 1, 0.001, 0.001);
 108         c3->cd(1); hGood->Draw();
 109         c3->cd(2); hFake->Draw();
 110         c3->cd(3); hFindables->Draw();
 111         c3->Update();
 112
 113         TCanvas *c4 = new TCanvas("c4","Tracking efficiency",300,300,800,500);
 114         hRatioG->Divide(hGood, hFindables, 1., 1.);
 115         hRatioF->Divide(hFake, hFindables, 1., 1.);
 116         hRatioG->SetMaximum(1.4);
 117         hRatioG->SetYTitle("Tracking efficiency");
 118         hRatioG->SetXTitle("Pt (GeV/c)");
 119         hRatioG->Draw();  // to not plot the erro bars    hg->Draw("histo");
 120         hRatioF->SetFillColor(1);
 121         hRatioF->SetFillStyle(3013);
 122         hRatioF->SetLineColor(2);
 123         hRatioF->SetLineWidth(2);
 124         hRatioF->Draw("same");  // to not plot the error bars  hRatioF->Draw("histosame");
 125         // a line to mark the best efficiency
 126         TLine *line1 = new TLine(0,1.0,2,1.0); line1->SetLineStyle(4);
 127         line1->Draw("same");
 128         TLine *line2 = new TLine(0,0.9,2,0.9); line2->SetLineStyle(4);
 129         line2->Draw("histosame");
 130         // a text explanation
 131         TText *text = new TText(0.461176,0.248448,"Fake tracks");
 132         text->SetTextSize(0.05);
 133         text->Draw();
 134         text = new TText(0.453919,1.11408,"Good tracks");
 135         text->SetTextSize(0.05);
 136         text->Draw();
 137         c4->Update();
 138
 139         cout << "neglabs = " << neglabs << endl;  // provvisoria
 140 }