Single_pideff.cxx

Go to the documentation of this file.

void Single_pideff( char* filename = "Single_pid.root" ) {
  TFile* f       = new TFile( filename );
  TTree* pid_kal = (TTree*)f->Get( "pid_kal" );
 
  Double_t kal_p;
  Double_t prob_pion;
  Double_t prob_kaon;
  Double_t prob_proton;
 
  pid_kal->SetBranchAddress( "kal_p", &kal_p );
  pid_kal->SetBranchAddress( "prob_pion", &prob_pion );
  pid_kal->SetBranchAddress( "prob_kaon", &prob_kaon );
  pid_kal->SetBranchAddress( "prob_proton", &prob_proton );
 
  Int_t nevent = pid_kal->GetEntries();
 
  Double_t x[10]  = { .1, .2, .3, .4, .5, .6, .7, .8, .9, 1.0 };
  Double_t ex[10] = { 0 };
 
  Int_t kal[10] = { 0 };
 
  Int_t pid1[10] = { 0 };
  Int_t pid2[10] = { 0 };
  Int_t pid3[10] = { 0 };
  Int_t pid4[10] = { 0 };
  Int_t pid5[10] = { 0 };
  Int_t pid6[10] = { 0 };
  Int_t pid7[10] = { 0 };
  Int_t pid8[10] = { 0 };
 
  Double_t eff1[10] = { 0 };
  Double_t eff2[10] = { 0 };
  Double_t eff3[10] = { 0 };
  Double_t eff4[10] = { 0 };
  Double_t eff5[10] = { 0 };
  Double_t eff6[10] = { 0 };
  Double_t eff7[10] = { 0 };
  Double_t eff8[10] = { 0 };
 
  Double_t err1[10] = { 0 };
  Double_t err2[10] = { 0 };
  Double_t err3[10] = { 0 };
  Double_t err4[10] = { 0 };
  Double_t err5[10] = { 0 };
  Double_t err6[10] = { 0 };
  Double_t err7[10] = { 0 };
  Double_t err8[10] = { 0 };
 
  for ( Int_t i = 0; i < nevent; i++ )
  {
    pid_kal->GetEntry( i );
 
    for ( Int_t j = 0; j < 10; j++ )
    {
      if ( kal_p > 0.05 + 0.1 * j && kal_p < 0.15 + 0.1 * j )
      {
        kal[j]++;
        if ( prob_pion > 0 ) pid1[j]++;
        if ( prob_pion > 0.001 ) pid2[j]++;
        if ( prob_pion > prob_kaon ) pid3[j]++;
        if ( prob_pion > prob_kaon && prob_pion > prob_proton ) pid4[j]++;
        if ( prob_pion > 0 && prob_pion > prob_kaon ) pid5[j]++;
        if ( prob_pion > 0 && prob_pion > prob_kaon && prob_pion > prob_proton ) pid6[j]++;
        if ( prob_pion > 0.001 && prob_pion > prob_kaon ) pid7[j]++;
        if ( prob_pion > 0.001 && prob_pion > prob_kaon && prob_pion > prob_proton ) pid8[j]++;
      }
    }
  }
 
  for ( Int_t i = 0; i < 10; i++ )
  {
    eff1[i] = ( pid1[i] * .1 ) / ( kal[i] * .1 );
    err1[i] = sqrt( eff1[i] * ( 1 - eff1[i] ) / kal[i] );
 
    eff2[i] = ( pid2[i] * .1 ) / ( kal[i] * .1 );
    err2[i] = sqrt( eff2[i] * ( 1 - eff2[i] ) / kal[i] );
 
    eff3[i] = ( pid3[i] * .1 ) / ( kal[i] * .1 );
    err3[i] = sqrt( eff3[i] * ( 1 - eff3[i] ) / kal[i] );
 
    eff4[i] = ( pid4[i] * .1 ) / ( kal[i] * .1 );
    err4[i] = sqrt( eff4[i] * ( 1 - eff4[i] ) / kal[i] );
 
    eff5[i] = ( pid5[i] * .1 ) / ( kal[i] * .1 );
    err5[i] = sqrt( eff5[i] * ( 1 - eff5[i] ) / kal[i] );
 
    eff6[i] = ( pid6[i] * .1 ) / ( kal[i] * .1 );
    err6[i] = sqrt( eff6[i] * ( 1 - eff6[i] ) / kal[i] );
 
    eff7[i] = ( pid7[i] * .1 ) / ( kal[i] * .1 );
    err7[i] = sqrt( eff7[i] * ( 1 - eff7[i] ) / kal[i] );
 
    eff8[i] = ( pid8[i] * .1 ) / ( kal[i] * .1 );
    err8[i] = sqrt( eff8[i] * ( 1 - eff8[i] ) / kal[i] );
  }
 
  TGraph* gr1 = new TGraphErrors( 10, x, eff1, ex, err1 );
  TGraph* gr2 = new TGraphErrors( 10, x, eff2, ex, err2 );
  TGraph* gr3 = new TGraphErrors( 10, x, eff3, ex, err3 );
  TGraph* gr4 = new TGraphErrors( 10, x, eff4, ex, err4 );
  TGraph* gr5 = new TGraphErrors( 10, x, eff5, ex, err5 );
  TGraph* gr6 = new TGraphErrors( 10, x, eff6, ex, err6 );
  TGraph* gr7 = new TGraphErrors( 10, x, eff7, ex, err7 );
  TGraph* gr8 = new TGraphErrors( 10, x, eff8, ex, err8 );
 
  TCanvas*     c1 = new TCanvas( "c1", "pion_eff_pid", 100, 10, 600, 400 );
  TMultiGraph* mg = new TMultiGraph();
 
  c1->SetGrid();
 
  mg->SetTitle( "pion_eff_pid" );
 
  // gr1->SetLineColor(2);
  gr2->SetLineColor( 1 );
  // gr3->SetLineColor(4);
  gr4->SetLineColor( 2 );
  // gr5->SetLineColor(6);
  // gr6->SetLineColor(3);
  // gr7->SetLineColor(8);
  gr8->SetLineColor( 3 );
 
  // mg->Add(gr1);
  mg->Add( gr2 );
  // mg->Add(gr3);
  mg->Add( gr4 );
  // mg->Add(gr5);
  // mg->Add(gr6);
  // mg->Add(gr7);
  mg->Add( gr8 );
 
  mg->Draw( "ALP" );
  mg->GetYaxis()->SetRangeUser( 0.9, 1.0 );
  mg->GetXaxis()->SetTitle( "Pt/GeV" );
  mg->GetYaxis()->SetTitle( "Eff" );
 
  TLegend* legend = new TLegend( 0.55, 0.15, 0.85, 0.35 );
  // legend->AddEntry(gr1,"prob(\\pi)>0","l");
  legend->AddEntry( gr2, "prob(\\pi)>0.001", "l" );
  // legend->AddEntry(gr3,"prob(\\pi)>prob(K)","l");
  legend->AddEntry( gr4, "prob(\\pi)>prob(K)&&prob(\\pi)>prob(p)", "l" );
  // legend->AddEntry(gr5,"prob(\\pi)>0&&prob(\\pi)>prob(K)","l");
  // legend->AddEntry(gr6,"prob(\\pi)>0&&prob(\\pi)>prob(K)&&prob(\\pi)>prob(p)","l");
  // legend->AddEntry(gr7,"prob(\\pi)>0.001&&prob(\\pi)>prob(K)","l");
  legend->AddEntry( gr8, "prob(\\pi)>0.001&&prob(\\pi)>prob(K)&&prob(\\pi)>prob(p)", "l" );
 
  legend->Draw();
}