Homec4science
Diffusion TP4b (master) Browse

R12329/Plots_BDT.py
71d23c230be9master

Last Change
Raw File

Plots_BDT.py
View Options

import ROOT
import os,sys
from ROOT import TH1D,TH2D,TFile,TTree,TCanvas,TChain
import sys
import array
#histFileName = sys.argv [1]
plotFileName1 = sys.argv [1]
canvas1 = TCanvas("canvas1")
#canvas1.Print(plotFileName1+"[")
# Plotting #
# "B_MM"
data_files = "data_samples"
#compaired_data_files = "BDT_Cutted_files"
compaired_data_files = "Cutted_files"
muons_files = "Muons_samples"
#
#Raw data
Run1 = TChain("DecayTree")
Run1.Add(os.path.join(data_files,"Bu2KpipiMM-Data-2018-MagDown-StrippingBu2LLK_fiducialMM16_preselPIDMM16_selected.root"))
Run1.Add(os.path.join(data_files,"Bu2KpipiMM-Data-2018-MagUp-StrippingBu2LLK_fiducialMM16_preselPIDMM16_selected.root"))
#Cutted data
Run1_Cut = TChain("DecayTree")
Run1_Cut.Add(os.path.join(compaired_data_files,"Cut_Bu2KpipiMM-Data-2018-MagDown-StrippingBu2LLK_fiducialMM16_preselPIDMM16_selected.root"))
Run1_Cut.Add(os.path.join(compaired_data_files,"Cut_Bu2KpipiMM-Data-2018-MagUp-StrippingBu2LLK_fiducialMM16_preselPIDMM16_selected.root"))
# Muons data
Run1_Muons = TChain("DecayTree")
Run1_Muons.Add(os.path.join(muons_files,"BDT2mu4_added_Muon4Aliased_BDT2mu3_added_MuonAliased_BDT1added_preselectedDimuon_inclusiveJpsi_gangaAll_2018_Data_MD_211104.root"))
Run1_Muons.Add(os.path.join(muons_files,"BDT2mu4_added_Muon4Aliased_BDT2mu3_added_MuonAliased_BDT1added_preselectedDimuon_inclusiveJpsi_gangaAll_2018_Data_MU_211104.root"))
#Styling
Run1.SetLineColor(ROOT.kBlue)
Run1_Cut.SetLineColor(ROOT.kRed)
Run1_Muons.SetLineColor(ROOT.kBlack)
#Drawing
#Run1.Draw("K_PT>>raw_histo(1000,0,7000)","")
#Run1_Cut.Draw("Kaon_PT>>cut_histo(1000,0,7000)","","same")
#Run1_Muons.Draw("Kaon_PT>>muon_histo(1000,0,7000)","","same")
Run1.Draw("piplus_PT+piminus_PT>>raw_histo(1000,0,8000)","")
Run1_Cut.Draw("phi_PT>>cut_histo(1000,0,8000)","","same")
Run1_Muons.Draw("phi_PT>>muon_histo(1000,0,8000)","","same")
#B_DTF_PVandJpsi_CHI2NDOF
#Run1.Draw("log10(abs(mu1_PT)>>raw_histo","")
#Run1_Cut.Draw("log10(abs(mu1_PT)>>cut_histo","","same")
#Run1_Muons.Draw("log10(abs(mu1_PT)>>muon_histo","","same")
# Retrieve the histograms
raw_histo = ROOT.gDirectory.Get("raw_histo")
cut_histo = ROOT.gDirectory.Get("cut_histo")
muon_histo = ROOT.gDirectory.Get("muon_histo")
#Normalization
factor = 1.
raw_histo.Scale(factor/raw_histo.Integral())
cut_histo.Scale(factor/cut_histo.Integral())
muon_histo.Scale(factor/muon_histo.Integral())
#Styling
raw_histo.GetXaxis().SetTitle("phi_PT")
raw_histo.GetYaxis().SetTitle("Events (Normalised) ")
raw_histo.SetStats(0)
#cut_histo.SetStats(0)
leg1 = ROOT.TLegend(.6,.6,.9,.7)
leg1.SetBorderSize(0)
leg1.SetFillColor(0)
leg1.SetFillStyle(0)
leg1.SetTextFont(42)
leg1.SetTextSize(0.035)
leg1.AddEntry(raw_histo,"Raw data","L")
leg1.AddEntry(cut_histo,"Cutted data","L")
leg1.AddEntry(muon_histo,"Muons data","L")
leg1.Draw()
latex = ROOT.TLatex()
latex.SetNDC()
latex.SetTextSize(0.045)
latex.DrawText(0.6,0.8,"LHCb 2018")
latex.SetTextSize(0.02)
#latex.DrawText(0.7,0.77,"B \rightarrow Kpipi events")
# Update the styling part
canvas1.Update()
canvas1.SaveAs(plotFileName1)
c4science · Help