def main():
#get the list of bg and data files from the unusal config
sampleinfos, basedir = getSampleInfo("201705_mergedext.config")
lumi = sampleinfos.values()[-1]["lumi"]
useRoot = False
#we will use the same colors as the .c plotter for now
colors = [100, 90, 80, 70, 60, 95, 85, 75, 65, 55]
#reverse order and change to RGB
colors = [getRGBTColor(i) for i in colors[::-1]]
#make storage containers for the bgs and data
bghists = HistStorage(xs=sampleinfos,
xstype="BSM3G",
lumi=lumi,
path=basedir,
useRoot=useRoot)
dat_hist = HistStorage(path=basedir, isData=True, useRoot=useRoot)
#make some lists that we will put into the containers
colorList = {}
bglist = OrderedDict()
#the idea is to have a readable dict:
bglist['DY+Jets'] = [
'DYJetsToLL_M-50_HT-70To100',
'DYJetsToLL_M-50_HT-600To800',
'DYJetsToLL_M-50_HT-400To600',
'DYJetsToLL_M-50_HT-800To1200',
'DYJetsToLL_M-50_HT-1200To2500',
'DYJetsToLL_M-50_HT-2500ToInf',
'DYJetsToLL_M-50_HT-200To400',
'DYJetsToLL_M-50_HT-0To70',
'DYJetsToLL_M-50_HT-100To200',
]
bglist['VV'] = [
'ZZ_TuneCUETP8M1_13TeV',
'WZ_TuneCUETP8M1_13TeV',
'WW_TuneCUETP8M1_13TeV',
]
bglist['SingleTop'] = [
'ST_tW_antitop_5f_inclusiveDecays_13TeV',
'ST_t-channel_top_4f_inclusiveDecays_13TeV',
'ST_tW_top_5f_inclusiveDecays_13TeV',
'ST_t-channel_antitop_4f_inclusiveDecays_13TeV',
]
bglist['$\mathsf{t\bar{t}}$'] = [
'TT_TuneCUETP8M2T4_13TeV',
]
bglist['W+Jets'] = [
'WJetsToLNu_HT-0To70',
]
##you can reverse the order
bglist = OrderedDict(reversed(list(bglist.items())))
#now actually add the backgrounds
bghists.addFileList(bglist)
bghists.colorList = colorList
hists = [
"NDiMuonCombinations/MHT",
"NDiMuonCombinations/Meff",
]
histContainer = HistStorageContainer(bg=bghists, data=dat_hist)
#histContainer=HistStorageContainer(bg=bghists,sg=sghist,data=dat_hist)
bghists.initStyle(style="bg")
for hist in hists:
#this is the magic line
histContainer.getHist(hist)
#now you can do what you want with the hists
allbg = sum(histContainer.getBGList())
allbg.Draw()
for bg in histContainer.getBGList():
bg.Draw("same")
data = histContainer.getData()
data.Draw("same p")
def main():
#get the list of bg and data files from the unusal config
sampleinfos, basedir = getSampleInfo("201705_mergedext.config")
lumi = sampleinfos.values()[-1]["lumi"]
useRoot = False
#we will use the same colors as the .c plotter for now
colors = [100, 90, 80, 70, 60, 95, 85, 75, 65, 55]
#reverse order and change to RGB
colors = [getRGBTColor(i) for i in colors[::-1]]
#make storage containers for the bgs and data
bghists = HistStorage(xs=sampleinfos,
xstype="BSM3G",
lumi=lumi,
path=basedir,
useRoot=useRoot)
dat_hist = HistStorage(path=basedir, isData=True, useRoot=useRoot)
#make some lists that we will put into the containers
colorList = {}
bglist = OrderedDict()
for sample in sampleinfos:
info = sampleinfos[sample]
sample = sample.replace(".root", "")
if info["type"] == "bg":
shortName = info["output"].replace(".root", "")
if "#" in shortName:
shortName = "$\mathsf{" + shortName.replace("#", "\\") + "}$"
if shortName not in bglist:
bglist[shortName] = [sample]
colorList[shortName] = colors.pop()
else:
bglist[shortName].append(sample)
#for data directly add the file
if info["type"] == "data":
dat_hist.addFile(sample)
for bg in bglist:
print("bglist['%s']=[" % (bg))
for ibg in bglist[bg]:
print("'ibg',")
print("]")
#now actually add the backgrounds
bghists.addFileList(bglist)
bghists.colorList = colorList
hists = [
"NDiMuonCombinations/MHT",
"NDiMuonCombinations/Meff",
]
histContainer = HistStorageContainer(bg=bghists, data=dat_hist)
#histContainer=HistStorageContainer(bg=bghists,sg=sghist,data=dat_hist)
binning = {
"_pt": 25,
"_mt": 25,
"_met": 25,
"MC_W_m_Gen": 10,
"_jet_ht": 10,
}
xranges = {
"_pt": [0, 1000],
"_jet_ht": [0, 1600],
"_mt": [150, 1600],
"boson_qt": [0, 600],
"_met": [0, 1300],
"_ET_MET": [0, 4],
"_DeltaPhi": [0.2, math.pi],
"MC_W_m_Gen": [0, 1300],
"relIso": [0, 0.05],
}
yranges = {
"_ET_MET": [1.01e-3, 1e8],
"_DeltaPhi": [1.01e-3, 1e8],
}
bghists.initStyle(style="bg")
#sghist.initStyle(style="sg")
for hist in hists:
##use different out dirs
dir_name = "test"
if not os.path.exists(dir_name):
os.mkdir(dir_name)
if "prof" in hist:
hist_style = sc.style_container(kind="Lines",
style='CMS',
useRoot=useRoot,
cmsPositon="upper left",
lumi=lumi,
cms=13)
else:
hist_style = sc.style_container(style='CMS',
useRoot=useRoot,
cmsPositon="upper left",
lumi=lumi,
cms=13)
#change the style if you want
#hist_style._error_bands_fcol=["grey","yellow"]
#hist_style._error_bands_ecol=["lightgrey","yellow"]
#hist_style._error_bands_alph=0.4
#hist_style.Set_error_line_width(0.0000000001)
hist_style.Set_poisson_error_off()
#hist_style.Set_minor_ticks()
#hist_style.Set_n_legend_columns(2)
#hist_style.Set_xerr()
#hist_style.Set_do_overflowbin()
#get the actual histograms
histContainer.getHist(hist)
#programmable binning
binf = getDictValue(hist, binning)
if binf is not None:
if isinstance(binf, list):
histContainer.rebin(vector=binf)
else:
histContainer.rebin(width=binf)
name = hist.replace("/", "")
#test = plotter(hist=histContainer.getBGList(),data_hist=histContainer.getData(),data=True,style=hist_style)
test = plotter(hist=histContainer.getBGList(), style=hist_style)
test.Add_data(histContainer.getData())
test.Add_plot('Ratio', pos=1, height=15)
test.Add_plot('DiffRatio_width_increase', pos=2, height=15)
mplt = test.create_plot()
test.SavePlot('%s/%s.png' % (dir_name, name))
test.SavePlot('%s/%s.pdf' % (dir_name, name))
def main():
basedir="/disk1/erdweg/out/output2015_2_13_15_47/merged//"
lumi=1000
xs= ConfigObj("/disk1/erdweg/plotting/xs_Phys14.cfg")
bghists=HistStorage(xs,lumi,path=basedir)
# bghists.setDataDriven("dataDrivenQCD")
bglist=OrderedDict()
bglist["PU20bx25"]=['ZprimeToTauTau_M-5000_PU20bx25_PHYS14',
'ZprimeToTauTau_M-1000_PU20bx25_PHYS14',
]
bglist["AVE30BX50"]=['ZprimeToTauTau_M-1000_AVE30BX50_PHYS14',
'ZprimeToTauTau_M-5000_AVE30BX50_PHYS14',
]
bglist["PU40bx25"]=['ZprimeToTauTau_M-1000_PU40bx25_PHYS14',
'ZprimeToTauTau_M-5000_PU40bx25_PHYS14',
]
colorList={}
colorList["PU20bx25"]="lightblue"
colorList["AVE30BX50"]="lightgreen"
colorList["PU40bx25"]="darkmagenta"
bghists.addFileList(bglist)
dat_hist=HistStorage(xs,lumi,path=basedir,isData=True)
# dat_hist.addFile("Data")
hists=[#'Taus/h1_1_Tau_eta',
#'Taus/h1_1_Tau_phi',
#'Taus/h1_2_Tau_pt',
#'Taus/h1_Tau_pt_resolution_0_500',
'Eles/h1_Ele_pt_resolution_0_500',
'Eles/h1_Ele_pt_resolution_500_1000',
#'Eles/h1_2_Ele_pt',
'Muons/h1_Muon_pt_resolution_0_500',
'Muons/h1_Muon_pt_resolution_500_1000',
#'Muons/h1_2_Muon_pt',
#'emu/Stage_0/h1_0_emu_Mass',
#'mutau/Stage_0/h1_0_mutau_Mass',
#'mutau/Stage_6/h1_6_mutau_Mass',
#'Ctr/h1_Ctr_Vtx_unweighted',
#'Ctr/h1_Ctr_Vtx_weighted',
#'Ctr/h1_Ctr_Vtx_emu_unweighted',
#'Ctr/h1_Ctr_Vtx_emu_weighted',
#'Ctr/h1_Ctr_HT',
#'Ctr/h1_Ctr_pT_hat',
#'emu/Stage_0/h1_0_emu_Mass_resolution',
#'mutau/Stage_0/h1_0_mutau_Mass_resolution'
]
sghist=HistStorage(xs,lumi,path=basedir)
# sgName="$\mathsf{W' \, M=2.3\,TeV \cdot 0.02}$"
# sghist.additionalWeight={"WprimeToTauNu_M-2300_TuneZ2star_8TeV-pythia6-tauola_Summer12_DR53X-PU_S10_START53_V7A-v1SIM":0.02}
# sghist.addAllFiles(tag="WprimeToTauNu_M-2300",joinName=sgName)
# sghist.colorList={sgName :"darkred"}
histContainer=HistStorageContainer(sg=bghists,data=dat_hist,bg=sghist)
binning={
"Taus/h1_2_Tau_pt":10,
# "emu/Stage_0/h1_0_emu_Mass":range(200,300,20)+range(300,400,50)+range(400,1600,100)+range(1600,2000,200),
# "mutau/Stage_0/h1_0_mutau_Mass":range(200,300,20)+range(300,400,50)+range(400,1600,100)+range(1600,2000,200),
# "mutau/Stage_6/h1_6_mutau_Mass":range(200,300,20)+range(300,400,50)+range(400,1600,100)+range(1600,2000,200),
"_met_et":30,
}
xranges={
"Taus/h1_2_Tau_pt":[0,2000],
"emu/Stage_0/h1_0_emu_Mass":[0,2000],
"mutau/Stage_0/h1_0_mutau_Mass":[0,2000],
"mutau/Stage_6/h1_6_mutau_Mass":[0,2000],
"emu/Stage_0/h1_0_emu_Mass_resolution":[-2,2],
"mutau/Stage_0/h1_0_mutau_Mass_resolution":[-2,2],
"Taus/h1_Tau_pt_resolution_0_500":[-2,2],
"Eles/h1_Ele_pt_resolution_0_500":[-2,2],
"Eles/h1_Ele_pt_resolution_500_1000":[-2,2],
"Muons/h1_Muon_pt_resolution_0_500":[-2,2],
"Muons/h1_Muon_pt_resolution_500_1000":[-2,2],
"Ctr/h1_Ctr_Vtx_unweighted":[0,40],
"Ctr/h1_Ctr_Vtx_weighted":[0,40],
"Ctr/h1_Ctr_Vtx_emu_unweighted":[0,40],
"Ctr/h1_Ctr_Vtx_emu_weighted":[0,40],
}
bghists.initStyle(style="sg",colors=colorList)
# sghist.initStyle(style="sg")
for hist in hists:
print("Now plotting: " + hist)
histContainer.getHist(hist)
binf=getDictValue(hist,binning)
if binf is not None:
# dat_hist.rebin(width=binf)
bghists.rebin(width=binf)
bghists.colorList=colorList
# bghists.setStyle(bgcolors=colorList)
# dat_hist.getHistList()[0].SetTitle("data")
hist_style = sc.style_container(style = 'CMS', useRoot = False,cms=13,lumi=19700)
hist_style.Set_additional_text('plot dummy')
test = plotter(sig=bghists.getHistList(),style=hist_style)
# test.Add_data(dat_hist.getHistList()[0])
# test.Add_plot('Signi',pos=0, height=15)
#test.Add_plot('DiffRatio',pos=1, height=15)
# test.Add_plot('Signi',pos=2, height=15)
# test.Add_plot('Diff',pos=1, height=15)
# test.Add_plot('Ratio',pos=2, height=15)
#test.Add_error_hist([sys_hist_2,sys_hist], band_center = 'ref')
test._cms_text_x = 0.12
test._cms_text_y = 0.91
#
# mxrange=getDictValue(hist,xranges)
if hist in xranges.keys():
test.Set_axis(xmin=xranges[hist][0],xmax=xranges[hist][1],ymin=1e-3,ymax=1e3)
name=hist.replace("/","")
test.create_plot()
plot_gauss_fit(test.Get_axis1(), bghists.getHistList()[0], xmin = -0.5, xmax = 0.5)
# test.show_fig()
test.SavePlot('plots/%s.pdf'%(name))
return 42
