'WWTo2L2Nu_13TeV_PH-skimid2859':lumisc, 'WWTo4Q_13TeV-v2_PH-skimid2871':lumisc, 'WWToLNuQQ_13TeV_PH-skimid2863':lumisc, 'WZTo3LNu_13TeV_PH-skimid3032':lumisc, 'ZZTo2L2Nu_13TeV_PH-skimid3046':lumisc, 'ZZTo4L_13TeV_PH-skimid3045':lumisc, 'TT_13TeV_PH-skimid2903':lumisc*1.138, 'TT_13TeV_ext3_PH-skimid2901':lumisc*1.138, 'TT_Mtt-1000toInf_13TeV_ext1_PH-skimid2989':lumisc*1.138/3.7, 'TT_Mtt-700to1000_13TeV_ext1_PH-skimid2868':lumisc*1.138, } bghists.addFileList(bglist) dat_hist=HistStorage(xs,lumi,path="/disk1/erdweg/television/DATA_25/merged/",isData=True) dat_hist.addFile("allData") # dat_hist.addFile("Data_Run2015B-PromptReco_251162_252126_SingleMuon-skimid81") # dat_hist.addFile("Data_Run2015C-PromptReco_253888_254914_SingleMuon-skimid48") # dat_hist.addFile("../../DATA_25/merged/Data_Run2015C-PromptReco_253888_254914_SingleMuon-skimid48") basedir="/disk1/erdweg/television/SIGNAL/merged/" sghist=HistStorage(xs,lumi,path=basedir,xstype=None) sglist=OrderedDict() sglist['RPV M=500']=[ 'RPVresonantToEMu_M-500_LLE_LQD-001_13TeV_CA-skimid3002', ] sglist['RPV M=1000']=[ 'RPVresonantToEMu_M-1000_LLE_LQD-001_13TeV_CA-skimid3022', ] sghist.additionalWeight = {'RPVresonantToEMu_M-500_LLE_LQD-001_13TeV_CA-skimid3002':lumisc}
class plotConfiguration: def __init__(self, useRoot=False ,basedir=None ,do_ddQCD=True): if basedir is None: self.basedir="/net/scratch_cms/institut_3a/materok/wprime/okt14/crab_projects/merged" #self.basedir="/net/scratch_cms/institut_3a/materok/wprime/run1/crab_projects/merged" else: self.basedir=basedir self.useRoot=useRoot self.xs= ConfigObj("xsFile13TeV_25ns_wprime.cfg") self.lumi=0 self.lumi+= 5882 #RunB self.lumi+= 2646 #RunC self.lumi+= 4353 #RunD self.lumi+= 4049 #RunE self.lumi+= 3160 #RunF #self.lumi= 11048 #RunC + RunD + RunE #self.lumi= 12881 #RunB + RunC + RunF self.bghists=HistStorage(self.xs,self.lumi,path=self.basedir,useRoot=self.useRoot) self.bghists.setDataDriven("dataDrivenQCD") self.bghists.setDataDriven("dataDrivenTree") self.bghists.setDataDriven("dataDrivenTreept") self.bghists.setDataDriven("dataDrivenTreedecay_mode") self.bghists.setDataDriven("dataDrivenTreeeta") self.bghists.setDataDriven("dataDrivenTreelepton_n") self.bghists.setDataDriven("dataDrivenTree15_pt_base") self.bghists.setDataDriven("dataDrivenTree15_pt_baseFit") self.bghists.setDataDriven("dataDrivenTree15_pt_decay_base") self.bghists.setDataDriven("dataDrivenTree15_pt_decay_baseFit") self.bghists.setDataDriven("dataDrivenTree15_pt_eta_baseFit") self.bghists.setDataDriven("dataDrivenTree15_pt_eta_base") self.bghists.setDataDriven("dataDrivenTree15_lepton_n_base") self.bghists.setDataDriven("dataDrivenTree15_decay_mode_base") #do_ddQCD=False self.bglist=OrderedDict() self.bglist["Diboson"]=[ "WW_TuneCUETP8M1_13TeV-pythia8", "WZ_TuneCUETP8M1_13TeV-pythia8", "ZZ_TuneCUETP8M1_13TeV-pythia8" ] if not do_ddQCD: self.bglist[r"$Z\rightarrow \nu\nu$"]=[ "ZJetsToNuNu_HT-100To200_13TeV-madgraph", "ZJetsToNuNu_HT-200To400_13TeV-madgraph", "ZJetsToNuNu_HT-400To600_13TeV-madgraph", "ZJetsToNuNu_HT-600To800_13TeV-madgraph", "ZJetsToNuNu_HT-800To1200_13TeV-madgraph", "ZJetsToNuNu_HT-1200To2500_13TeV-madgraph", "ZJetsToNuNu_HT-2500ToInf_13TeV-madgraph" ] self.bglist[r"$\mathsf{Z/\gamma}$"]=[ "DYJetsToLL_M-5to50_TuneCUETP8M1_13TeV-madgraphMLM-pythia8", "DYJetsToLL_M-50_TuneCUETP8M1_13TeV-madgraphMLM-pythia8" ] self.bglist[r'$\mathsf{t\bar{t}}$']=[ "TT_TuneCUETP8M1_13TeV-powheg-pythia8", "ST_t-channel_antitop_4f_leptonDecays_13TeV-powheg-pythia8_TuneCUETP8M1", "ST_t-channel_top_4f_leptonDecays_13TeV-powheg-pythia8_TuneCUETP8M1", "ST_tW_antitop_5f_inclusiveDecays_13TeV-powheg-pythia8_TuneCUETP8M1", "ST_tW_top_5f_inclusiveDecays_13TeV-powheg-pythia8_TuneCUETP8M1" ] if do_ddQCD: self.bglist["multijet"]=[ "dataDrivenQCD", ] if not do_ddQCD: self.bglist["QCD"]=[ "QCD_HT200to300_TuneCUETP8M1_13TeV-madgraphMLM-pythia8", "QCD_HT300to500_TuneCUETP8M1_13TeV-madgraphMLM-pythia8", "QCD_HT500to700_TuneCUETP8M1_13TeV-madgraphMLM-pythia8", "QCD_HT700to1000_TuneCUETP8M1_13TeV-madgraphMLM-pythia8", "QCD_HT1000to1500_TuneCUETP8M1_13TeV-madgraphMLM-pythia8", "QCD_HT1500to2000_TuneCUETP8M1_13TeV-madgraphMLM-pythia8", "QCD_HT2000toInf_TuneCUETP8M1_13TeV-madgraphMLM-pythia8" ] #self.bglist[r'$\mathsf{W\rightarrow l \nu}$ $\mathsf{M_{W}<200 GeV}$']=[ #"WJetsToLNu_HT-100To200_13TeV_MG", #"WJetsToLNu_HT-200To400_13TeV_MG", #"WJetsToLNu_HT-400To600_13TeV_MG", ##"WJetsToLNu_HT-600To800_13TeV_MG", #"WJetsToLNu_HT-600ToInf_13TeV_MG", ##"WJetsToLNu_HT-800To1200_13TeV_MG", ##"WJetsToLNu_HT-1200To2500_13TeV_MG", ##"WJetsToLNu_HT-2500ToInf_13TeV_MG", ##"WJetsToLNu_13TeV-FXFX_MC", ##"WJetsToLNu_13TeV_MG", #"WJetsToLNu_13TeV_AM", #] #self.bglist[r'$\mathsf{W\rightarrow l \nu}$ MCatNLO']=[ ###"WJetsToLNu_13TeV-FXFX_MC", #"WJetsToLNu_13TeV_AM", ###"WJetsToLNu_13TeV-FXFX_MC", ##"WJetsToLNu_13TeVMLM_MG", #] self.bglist["WJets"]=[ "WJetsToLNu_HT-100To200_TuneCUETP8M1_13TeV-madgraphMLM-pythia8", "WJetsToLNu_HT-200To400_TuneCUETP8M1_13TeV-madgraphMLM-pythia8", "WJetsToLNu_HT-400To600_TuneCUETP8M1_13TeV-madgraphMLM-pythia8", "WJetsToLNu_HT-600To800_TuneCUETP8M1_13TeV-madgraphMLM-pythia8", "WJetsToLNu_HT-800To1200_TuneCUETP8M1_13TeV-madgraphMLM-pythia8", "WJetsToLNu_HT-1200To2500_TuneCUETP8M1_13TeV-madgraphMLM-pythia8", "WJetsToLNu_HT-2500ToInf_TuneCUETP8M1_13TeV-madgraphMLM-pythia8" ] """ self.bglist[r'$\mathsf{W\rightarrow \tau \nu}$']=[ "WToTauNu_M-100_TuneCUETP8M1_13TeV-pythia8-tauola", "WToTauNu_M-200_TuneCUETP8M1_13TeV-pythia8-tauola", "WToTauNu_M-500_TuneCUETP8M1_13TeV-pythia8-tauola", "WToTauNu_M-1000_TuneCUETP8M1_13TeV-pythia8-tauola", "WToTauNu_M-2000_TuneCUETP8M1_13TeV-pythia8-tauola", "WToTauNu_M-3000_TuneCUETP8M1_13TeV-pythia8-tauola", "WToTauNu_M-4000_TuneCUETP8M1_13TeV-pythia8-tauola", "WToTauNu_M-5000_TuneCUETP8M1_13TeV-pythia8-tauola", "WToTauNu_M-6000_TuneCUETP8M1_13TeV-pythia8-tauola" ] """ self.bghists.additionalWeight["dataDrivenQCD"]=0.938091502614 self.bghists.additionalWeight["TT_13TeV_AM"]=1./3. self.bghists.additionalWeight["TT_13TeV_ext1_AM"]=1./3. self.bghists.additionalWeight["TTJets_13TeV_AM"]=1./3. self.bghists.additionalWeight["TT_13TeV_ext3_PH"]=1./2. self.bghists.additionalWeight["TT_13TeV_ext4_PH"]=1./2. self.bghists.additionalWeight["TT_Mtt-700to1000_13TeV_ext1_PH"]=1./2. self.bghists.additionalWeight["TT_Mtt-700to1000_13TeV_PH"]=1./2. self.bghists.additionalWeight["TT_Mtt-1000toInf_13TeV_ext1_PH"]=1./2. self.bghists.additionalWeight["TT_Mtt-1000toInf_13TeV_ext2_PH"]=1./2. self.bghists.additionalWeight["WWToLNuQQ_13TeV_ext1_PH"]=1./2. self.bghists.additionalWeight["WWToLNuQQ_13TeV_PH"]=1./2. #bghists.additionalWeight["WJetsToLNu_HT-100To200_13TeV_MG"]=0.5 #bghists.additionalWeight["TT_13TeV_PH"]=0.5 #bghists.additionalWeight["TT_13TeV_ext3_PH"]=0.5 #bghists.additionalWeight["ST_t-channel_4f_leptonDecays_13TeV_AM"]=0.5 #bghists.additionalWeight["DYJetsToLL_M-50_13TeV_ext1_AM"]=0.5 #bghists.additionalWeight["ST_t-channel_4f_leptonDecays_13TeV_ext1_AM"]=0.5 self.bgcolorList={} self.bgcolorList[r'$\mathsf{W\rightarrow l \nu}$']=getRGBTColor(861) self.bgcolorList[r'$\mathsf{W\rightarrow \tau \nu}$']=getRGBTColor(861) self.bgcolorList["WJets"]="lightblue" self.bgcolorList["W"]=getRGBTColor(861) self.bgcolorList["W on shell"]="lightblue" self.bgcolorList[r'$\mathsf{W\rightarrow l \nu}$ MCatNLO']="brown" self.bgcolorList[r'$\mathsf{W\rightarrow l \nu}$ $\mathsf{M_{W}<200 GeV}$']="lightblue" self.bgcolorList["W HT"]="orange" self.bgcolorList["multijet"]=getRGBTColor(213) self.bgcolorList["multijet MC"]=getRGBTColor(216) self.bgcolorList["QCD from data"]=getRGBTColor(213) self.bgcolorList["QCD jet"]=getRGBTColor(213) self.bgcolorList["QCD MC"]="brown" self.bgcolorList["QCD"]="brown" self.bgcolorList["Z (inv)"]="darkblue" self.bgcolorList["Top"]=getRGBTColor(616) self.bgcolorList["$t\bar{t}$"]=getRGBTColor(616) self.bgcolorList[r'$\mathsf{t\bar{t}}$']="orange" self.bgcolorList["Diboson"]=getRGBTColor(413) self.bgcolorList[r"$\mathsf{Z/\gamma}$"]=getRGBTColor(810) self.bgcolorList["DY"]=getRGBTColor(810) self.bgcolorList[r"$\mathsf{\gamma jets}$"]="yellow" self.bgcolorList[r"$Z\rightarrow \nu\nu$"]="yellow" # dummy signal #setup signals self.sghist=HistStorage(self.xs,self.lumi,path=self.basedir,useRoot=self.useRoot) self.sgList=OrderedDict() self.sgList["W' M=1TeV"]=["WprimeToTauNu_M-1000_TuneCUETP8M1_13TeV-pythia8-tauola"] self.sgcolorlist={} self.sgcolorlist["W' M=1TeV"]="darkred" self.dat_hist=HistStorage(self.xs,self.lumi,path=self.basedir,isData=True,useRoot=self.useRoot) #self.DataFiles=["TauRunC"] #self.DataFiles=["TauRunD"] #self.DataFiles=["TauRunE"] #self.DataFiles=["Tau_Run2016E_PromptReco_v2","Tau_Run2016C_PromptReco_v2","Tau_Run2016D_PromptReco_v2"] self.DataFiles=["Tau_Run2016B_PromptReco_v2","Tau_Run2016C_PromptReco_v2", "Tau_Run2016D_PromptReco_v2","Tau_Run2016E_PromptReco_v2","Tau_Run2016F_PromptReco_v1"] #self.DataFiles=["Tau_Run2016C_PromptReco_v2","Tau_Run2016D_PromptReco_v2"] #self.DataFiles=["TauRunC","TauRunD","TauRunE"] def compile_obj(self): self.bghists.addFileList(self.bglist) self.bghists.colorList=self.bgcolorList self.sghist.addFileList(self.sgList) self.sghist.colorList=self.sgcolorlist for data in self.DataFiles: self.dat_hist.addFile(data)
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))