Macro to plot the primary vertex efficiencies (Leonardo)

diff --git a/PWG1/macros/MakeVtxEffPlots.C b/PWG1/macros/MakeVtxEffPlots.C
new file mode 100644 (file)
index 0000000..f7142cb
--- /dev/null
+++ b/PWG1/macros/MakeVtxEffPlots.C
@@ -0,0 +1,213 @@
+#if !defined(__CINT__) || defined(__MAKECINT__)
+#include "TSystem.h"
+#include "TROOT.h"
+#include <TString.h>
+#include <TH1.h>
+#include <TGraphAsymmErrors.h>
+#include <TCanvas.h>
+#include <TLegend.h>
+#include <TMath.h>
+#include <TFile.h>
+#endif
+
+
+void setDrawAtt(Int_t markerstyle,Int_t markercolor,Int_t markersize,Int_t linecolor,Int_t linewidth,TGraphAsymmErrors *h1)
+{ 
+  h1->SetMarkerStyle(markerstyle);
+  h1->SetMarkerColor(markercolor);
+  h1->SetMarkerSize(markersize);
+  h1->SetLineColor(linecolor);
+  h1->SetLineWidth(linewidth);
+  }
+void setDrawAtt(Int_t markerstyle,Int_t markercolor,Int_t markersize,Int_t linecolor,Int_t linewidth,TH1 *h1)
+{ 
+  h1->SetMarkerStyle(markerstyle);
+  h1->SetMarkerColor(markercolor);
+  h1->SetMarkerSize(markersize);
+  h1->SetLineColor(linecolor);
+  h1->SetLineWidth(linewidth);
+  }
+
+void MakeVtxEffPlots(TString foldname="",Bool_t savefile=kFALSE){
+
+  //Make efficiency plots for SPD vertex and TRK Vertex
+  
+  TFile *infil=new TFile(Form("%sAnalysisResults.root",foldname.Data()),"read");
+  TDirectory *dirFile=(TDirectory*)infil->Get("Vertex_Performance");
+  TList *cOutput = (TList*)dirFile->Get("cOutputVtxESD");
+    
+  TH1F *fhTriggeredTrklets=(TH1F*)cOutput->FindObject("fhTriggeredTrklets");
+  TH1F *fhSPDZTrklets=(TH1F*)cOutput->FindObject("fhSPDZTrklets");
+  TH1F *fhSPD3DTrklets=(TH1F*)cOutput->FindObject("fhSPD3DTrklets");
+  TH1F *fhTRKTrklets=(TH1F*)cOutput->FindObject("fhTRKTrklets");
+  TH1F *fhTRKcTrklets=(TH1F*)cOutput->FindObject("fhTRKcTrklets");
+  TH1F *fhTRKncTrklets=(TH1F*)cOutput->FindObject("fhTRKncTrklets");
+  TH1F *fhSPDZZreco=(TH1F*)cOutput->FindObject("fhSPDZZreco");
+  TH1F *fhSPD3DZreco=(TH1F*)cOutput->FindObject("fhSPD3DZreco");
+  
+  TGraphAsymmErrors *fhSPDZEffTrklets=new TGraphAsymmErrors(fhSPDZTrklets,fhTriggeredTrklets,"w");
+  fhSPDZEffTrklets->SetName("fhSPDZEffTrklets");
+  fhSPDZEffTrklets->SetDrawOption("AP");
+  TGraphAsymmErrors *fhSPD3DEffTrklets=new TGraphAsymmErrors(fhSPD3DTrklets,fhTriggeredTrklets,"w");
+  fhSPD3DEffTrklets->SetName("fhSPD3DEffTrklets");
+  TH1F * fhSPDOverallTrklets=(TH1F*)fhSPDZTrklets->Clone("fhSPDOverallTrklets");
+  fhSPDOverallTrklets->Add(fhSPD3DTrklets);
+  TGraphAsymmErrors *fhSPDOverallEffTrklets=new TGraphAsymmErrors(fhSPDOverallTrklets,fhTriggeredTrklets,"w");
+  fhSPDOverallEffTrklets->SetName("fhSPDOverallEffTrklets");
+  TGraphAsymmErrors *fhTRKEffTrklets=new TGraphAsymmErrors(fhTRKTrklets,fhTriggeredTrklets,"w");
+  fhTRKEffTrklets->SetName("fhTRKEffTrklets");
+  TGraphAsymmErrors *fhTRKcEffTrklets=new TGraphAsymmErrors(fhTRKcTrklets,fhTriggeredTrklets,"w");
+  fhTRKcEffTrklets->SetName("fhTRKcEffTrklets");
+  TGraphAsymmErrors *fhTRKncEffTrklets=new TGraphAsymmErrors(fhTRKncTrklets,fhTriggeredTrklets,"w");
+  fhTRKncEffTrklets->SetName("fhTRKncEffTrklets");
+  TH1F * fhSPDOverallZreco=(TH1F*)fhSPDZZreco->Clone("fhSPDOverallZreco");
+  fhSPDOverallZreco->Add(fhSPD3DZreco);
+  TGraphAsymmErrors *fhSPDEffZreco=new TGraphAsymmErrors(fhSPD3DZreco,fhSPDOverallZreco,"w");
+  fhSPDEffZreco->SetName("fhSPDEffZreco");
+
+  TH1F *fhEff = new TH1F("hEff","hEff",6,0.5,6.5);
+  Int_t count=1;
+  if(fhSPDZTrklets->GetEntries()!=0 && fhTriggeredTrklets->GetEntries()!=0){
+    fhEff->Fill(count,fhSPDZTrklets->GetEntries()/fhTriggeredTrklets->GetEntries());
+    fhEff->SetBinError(count,fhEff->GetBinContent(count)*TMath::Sqrt(1/fhSPDZTrklets->GetEntries()+1/fhTriggeredTrklets->GetEntries()));
+  }
+  fhEff->GetXaxis()->SetBinLabel(count,"SPDZ");
+  
+  count++;
+  if(fhSPD3DTrklets->GetEntries()!=0 && fhTriggeredTrklets->GetEntries()!=0){
+    fhEff->Fill(count,fhSPD3DTrklets->GetEntries()/fhTriggeredTrklets->GetEntries());
+    fhEff->SetBinError(count,fhEff->GetBinContent(count)*TMath::Sqrt(1/fhSPD3DTrklets->GetEntries()+1/fhTriggeredTrklets->GetEntries()));
+  }
+  fhEff->GetXaxis()->SetBinLabel(count,"SPD3D");
+  
+  count++;
+  if(fhSPDOverallTrklets->GetEntries()!=0 && fhTriggeredTrklets->GetEntries()!=0){
+    fhEff->Fill(count,fhSPDOverallTrklets->GetEntries()/fhTriggeredTrklets->GetEntries());
+    fhEff->SetBinError(count,fhEff->GetBinContent(count)*TMath::Sqrt(1/fhSPDOverallTrklets->GetEntries()+1/fhTriggeredTrklets->GetEntries()));
+  }
+  fhEff->GetXaxis()->SetBinLabel(count,"SPD Overall");
+  
+  count++;
+  if(fhTRKTrklets->GetEntries()!=0 && fhTriggeredTrklets->GetEntries()!=0){
+    fhEff->Fill(count,fhTRKTrklets->GetEntries()/fhTriggeredTrklets->GetEntries());
+    fhEff->SetBinError(count,fhEff->GetBinContent(count)*TMath::Sqrt(1/fhTRKTrklets->GetEntries()+1/fhTriggeredTrklets->GetEntries()));
+  }
+  fhEff->GetXaxis()->SetBinLabel(count,"TRK");
+  
+  count++;
+  if(fhTRKcTrklets->GetEntries()!=0 && fhTriggeredTrklets->GetEntries()!=0){
+    fhEff->Fill(count,fhTRKcTrklets->GetEntries()/fhTriggeredTrklets->GetEntries());
+    fhEff->SetBinError(count,fhEff->GetBinContent(count)*TMath::Sqrt(1/fhTRKcTrklets->GetEntries()+1/fhTriggeredTrklets->GetEntries()));
+  }
+  fhEff->GetXaxis()->SetBinLabel(count,"TRKc");
+  
+  count++;
+  if(fhTRKncTrklets->GetEntries()!=0 && fhTriggeredTrklets->GetEntries()!=0){
+    fhEff->Fill(count,fhTRKncTrklets->GetEntries()/fhTriggeredTrklets->GetEntries());
+    fhEff->SetBinError(count,fhEff->GetBinContent(count)*TMath::Sqrt(1/fhTRKncTrklets->GetEntries()+1/fhTriggeredTrklets->GetEntries()));
+  }
+  fhEff->GetXaxis()->SetBinLabel(count,"TRKnc");
+  
+  count++;
+  
+  TCanvas *canvSPDTrklets=new TCanvas("canvSPDTrklets","SPD Eff vs tracklet multiplicy");
+  canvSPDTrklets->SetBottomMargin(0.14);
+  canvSPDTrklets->SetTopMargin(0.08);
+  canvSPDTrklets->SetLeftMargin(0.14);
+  canvSPDTrklets->SetRightMargin(0.08);
+  fhSPDZEffTrklets->SetMinimum(0);
+  fhSPDZEffTrklets->SetMaximum(1.2);
+  fhSPDZEffTrklets->GetXaxis()->SetLimits(0,30);
+  fhSPDZEffTrklets->GetXaxis()->SetTitle("tracklet multiplicity");
+  fhSPDZEffTrklets->GetXaxis()->SetTitleSize(0.05);
+  fhSPDZEffTrklets->GetYaxis()->SetTitle("efficiency");
+  fhSPDZEffTrklets->GetYaxis()->SetTitleSize(0.05);
+  setDrawAtt(22,4,2,4,2,fhSPDZEffTrklets);
+  setDrawAtt(23,2,2,2,2,fhSPD3DEffTrklets);
+  setDrawAtt(24,1,2,1,2,fhSPDOverallEffTrklets);
+  fhSPDZEffTrklets->SetTitle("SPDZ efficiency");
+  fhSPDZEffTrklets->Draw("ALP");
+  fhSPD3DEffTrklets->Draw("PLSAME");
+  fhSPDOverallEffTrklets->Draw("PLSAME"); 
+  TLegend *legSPD=new TLegend(0.6,0.2,0.9,0.4);
+  legSPD->AddEntry(fhSPD3DEffTrklets,"SPD3D","P");
+  legSPD->AddEntry(fhSPDZEffTrklets,"SPDZ","P");
+  legSPD->AddEntry(fhSPDOverallEffTrklets,"SPDZ + SPD3D","P");
+  legSPD->Draw();
+  
+  TCanvas *canvTRKTrklets=new TCanvas("canvTRKTrklets","TRK Eff vs tracklet multiplicy");
+  canvTRKTrklets->SetBottomMargin(0.14);
+  canvTRKTrklets->SetTopMargin(0.08);
+  canvTRKTrklets->SetLeftMargin(0.14);
+  canvTRKTrklets->SetRightMargin(0.08);
+  fhTRKEffTrklets->SetMinimum(0);
+  fhTRKEffTrklets->SetMaximum(1.2);
+  fhTRKEffTrklets->GetXaxis()->SetLimits(0,30);
+  fhTRKEffTrklets->GetXaxis()->SetTitle("tracklet multiplicity");
+  fhTRKEffTrklets->GetXaxis()->SetTitleSize(0.05);
+  fhTRKEffTrklets->GetYaxis()->SetTitle("efficiency");
+  fhTRKEffTrklets->GetYaxis()->SetTitleSize(0.05);
+  setDrawAtt(22,4,2,4,2,fhTRKEffTrklets);
+  setDrawAtt(23,2,2,2,2,fhTRKcEffTrklets);
+  setDrawAtt(24,1,2,1,2,fhTRKncEffTrklets);
+  fhTRKEffTrklets->SetTitle("TRK efficiency");
+  fhTRKEffTrklets->Draw("ALP");
+  fhTRKcEffTrklets->Draw("PLSAME");
+  fhTRKncEffTrklets->Draw("PLSAME"); 
+  TLegend *legTRK=new TLegend(0.6,0.2,0.9,0.4);
+  legTRK->AddEntry(fhTRKEffTrklets,"TRK","P");  
+  legTRK->AddEntry(fhTRKcEffTrklets,"TRKc","P");
+  legTRK->AddEntry(fhTRKncEffTrklets,"TRKnc","P");
+  legTRK->Draw();
+  
+  TCanvas *canvZ=new TCanvas("canvZ","3D/reco vs Z");
+  canvZ->SetBottomMargin(0.14);
+  canvZ->SetTopMargin(0.08);
+  canvZ->SetLeftMargin(0.14);
+  canvZ->SetRightMargin(0.08);
+  fhSPDEffZreco->SetMinimum(0);
+  fhSPDEffZreco->SetMaximum(1.2);
+  fhSPDEffZreco->GetXaxis()->SetTitle("Z - <Z> [cm]");
+  fhSPDEffZreco->GetXaxis()->SetTitleSize(0.05);
+  fhSPDEffZreco->GetYaxis()->SetTitle("3D rec / (3D+Z rec)");
+  fhSPDEffZreco->GetYaxis()->SetTitleSize(0.05);
+  setDrawAtt(22,4,2,4,2,fhSPDEffZreco);
+  fhSPDEffZreco->SetTitle("SPD3D/(SPD3D+Z) vs Zreco");
+  fhSPDEffZreco->Draw("AP");
+  TLegend *legZ=new TLegend(0.6,0.2,0.9,0.4);
+  legZ->AddEntry(fhSPDEffZreco,"SPD3D/(SPD3D+Z) vs Zreco","P");
+  legZ->Draw();
+  
+  TCanvas *canvOverall=new TCanvas("canvOverall","Eff integrated over multiplicy");
+  canvOverall->SetBottomMargin(0.14);
+  canvOverall->SetTopMargin(0.08);
+  canvOverall->SetLeftMargin(0.14);
+  canvOverall->SetRightMargin(0.08);
+  fhEff->SetMinimum(0);
+  fhEff->SetMaximum(1.2);
+  fhEff->GetXaxis()->SetLimits(0,30);
+  fhEff->GetXaxis()->SetTitleSize(0.05);
+  fhEff->GetYaxis()->SetTitle("efficiency");
+  fhEff->GetYaxis()->SetTitleSize(0.05);
+  fhEff->SetTitle("integrated over multiplicity");
+  fhEff->Draw("");
+
+  if(savefile){
+  TFile* fileEff = new TFile("VtxEff.root","recreate");
+  fhSPDZEffTrklets->Write();
+  fhSPD3DEffTrklets->Write();
+  fhSPDOverallEffTrklets->Write();
+  fhTRKEffTrklets->Write();
+  fhTRKcEffTrklets->Write();
+  fhTRKncEffTrklets->Write();
+  fhSPDEffZreco->Write();
+  fhEff->Write();
+  fileEff->Close();
+  delete fileEff;
+  }
+  
+  
+}
+
+
+