Page MenuHomec4science
Files F60243430

efficiency.py
No OneTemporary
Actions

File Metadata

Created
Sun, Apr 28, 14:25

efficiency.py
View Options

from cmath import sqrt
import os,sys
import ROOT
from ROOT import TH1D,TH2D,TFile,TTree,TCanvas, TChain
import sys
import array
import numpy as np
simulated_files = "mc_samples"
measured_files = "MC_Cutted_files"
########### Efficiency = measured / simulated ##############
#-----Run 1-----#
measured_1 = TChain("DecayTree")
#add all MC cutted files for Run1
measured_1.Add(os.path.join(measured_files,"Cut_Bu2KpipiJPsi-MM-MC-12145090-2011-MagDown-Sim09k-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
measured_1.Add(os.path.join(measured_files,"Cut_Bu2KpipiJPsi-MM-MC-12145090-2011-MagUp-Sim09k-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
measured_1.Add(os.path.join(measured_files,"Cut_Bu2KpipiJPsi-MM-MC-12145090-2012-MagDown-Sim09k-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
measured_1.Add(os.path.join(measured_files,"Cut_Bu2KpipiJPsi-MM-MC-12145090-2012-MagUp-Sim09k-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
simulated_1 = TChain("DecayTree")
#add all MC files for Run1
simulated_1.Add(os.path.join(simulated_files,"Bu2KpipiJPsi-MM-MC-12145090-2011-MagDown-Sim09k-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
simulated_1.Add(os.path.join(simulated_files,"Bu2KpipiJPsi-MM-MC-12145090-2011-MagUp-Sim09k-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
simulated_1.Add(os.path.join(simulated_files,"Bu2KpipiJPsi-MM-MC-12145090-2012-MagDown-Sim09k-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
simulated_1.Add(os.path.join(simulated_files,"Bu2KpipiJPsi-MM-MC-12145090-2012-MagUp-Sim09k-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
N_cut_events_1 = measured_1.Draw("B_MM>>my_histogram1(50,3700,6900)","")
N_events_1 = simulated_1.Draw("B_MM>>my_histogram11(50,3700,6900)","","same")
# efficiencies computation (estimator for probability)
efficiency_1 = N_cut_events_1/N_events_1
# Binomial uncertainty (with estimator P) : sigma = sqrt(P(1-P))/N
Proba_1 = efficiency_1
uncertainty_1 = np.sqrt(N_events_1*Proba_1*(1-Proba_1))/N_events_1
print("---------LHC RUN 1---------")
print("N_events = ", N_events_1, "N_cut_events = ",N_cut_events_1)
print(" Efficiency for Run1 = ",efficiency_1, "±",uncertainty_1)
#-----Run 2-----#
measured_2 = TChain("DecayTree")
#add all MC cutted files for Run2
measured_2.Add(os.path.join(measured_files,"Cut_Bu2KpipiJPsi-MM-MC-12145090-2015-MagDown-Sim09c-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
measured_2.Add(os.path.join(measured_files,"Cut_Bu2KpipiJPsi-MM-MC-12145090-2015-MagUp-Sim09c-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
measured_2.Add(os.path.join(measured_files,"Cut_Bu2KpipiJPsi-MM-MC-12145090-2016-MagDown-Sim09c-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
measured_2.Add(os.path.join(measured_files,"Cut_Bu2KpipiJPsi-MM-MC-12145090-2016-MagUp-Sim09c-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
measured_2.Add(os.path.join(measured_files,"Cut_Bu2KpipiJPsi-MM-MC-12145090-2017-MagDown-Sim09h-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
measured_2.Add(os.path.join(measured_files,"Cut_Bu2KpipiJPsi-MM-MC-12145090-2017-MagUp-Sim09h-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
measured_2.Add(os.path.join(measured_files,"Cut_Bu2KpipiJPsi-MM-MC-12145090-2018-MagDown-Sim09h-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
measured_2.Add(os.path.join(measured_files,"Cut_Bu2KpipiJPsi-MM-MC-12145090-2018-MagUp-Sim09h-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
simulated_2 = TChain("DecayTree")
#add all MC files for Run2
simulated_2.Add(os.path.join(simulated_files,"Bu2KpipiJPsi-MM-MC-12145090-2015-MagDown-Sim09c-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
simulated_2.Add(os.path.join(simulated_files,"Bu2KpipiJPsi-MM-MC-12145090-2015-MagUp-Sim09c-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
simulated_2.Add(os.path.join(simulated_files,"Bu2KpipiJPsi-MM-MC-12145090-2016-MagDown-Sim09c-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
simulated_2.Add(os.path.join(simulated_files,"Bu2KpipiJPsi-MM-MC-12145090-2016-MagUp-Sim09c-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
simulated_2.Add(os.path.join(simulated_files,"Bu2KpipiJPsi-MM-MC-12145090-2017-MagDown-Sim09h-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
simulated_2.Add(os.path.join(simulated_files,"Bu2KpipiJPsi-MM-MC-12145090-2017-MagUp-Sim09h-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
simulated_2.Add(os.path.join(simulated_files,"Bu2KpipiJPsi-MM-MC-12145090-2018-MagDown-Sim09h-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
simulated_2.Add(os.path.join(simulated_files,"Bu2KpipiJPsi-MM-MC-12145090-2018-MagUp-Sim09h-StrippingBu2LLK_MCTruthMM_selected_resampled.root"))
N_cut_events_2 = measured_2.Draw("B_MM>>my_histogram2(50,3700,6900)","")
N_events_2 = simulated_2.Draw("B_MM>>my_histogram22(50,3700,6900)","","same")
efficiency_2 = N_cut_events_2/N_events_2
Proba_2 = efficiency_2
uncertainty_2 = np.sqrt(N_events_2*Proba_2*(1-Proba_2))/N_events_2
print("---------LHC RUN 2---------")
print("N_events = ", N_events_2, "N_cut_events = ",N_cut_events_2)
#print("Mean mu = ", Proba_2)
#print("sigma^2 = ",Proba_2*(1-Proba_2))
print(" Efficiency for Run2 = ",efficiency_2, "±",uncertainty_2)

Event Timeline

c4science · Help