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