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();
 
}