errorETT = histograms['ett']['zz'].Clone()
errorETT.SetFillStyle(3013)
errorETT.Add(histograms['ett']['fakes'])
errorETT.Add(histograms['ett']['wz'])
errorETT.SetMarkerSize(0)
errorETT.SetFillColor(1)
errorETT.SetLineWidth(1)
# Apply some styles to all the histograms
for channel in [
'emt', 'mmt', 'eet', 'llem', 'llet', 'llmt', 'lltt', 'ett', 'mtt'
]:
# Use Poissonian error bars. The set_zero_bins makes it so bins w/o
# any data are blank.
histograms[channel]['poisson'] = convert(histograms[channel]['data'],
set_zero_bins=-10)
# Make sure all data points are visible
fix_maximum(histograms[channel])
# We have to draw it so things like the axes are initialized.
histograms[channel]['stack'].Draw()
histograms[channel]['stack'].GetYaxis().SetTitle(
"#bf{Events/%i GeV}" % histograms[channel]['data'].GetBinWidth(1))
if channel == 'lltt' or channel == "llet" or channel == "llmt" or channel == "llem":
histograms[channel]['stack'].GetXaxis().SetTitle(
"#bf{m_{#tau#tau} [GeV]}")
else:
histograms[channel]['stack'].GetXaxis().SetTitle(
"#bf{m_{vis} [GeV]}")
plot_suffix = "_%s_%s_%s.pdf" % ('prefit' if args.prefit else 'postfit',
args.period, 'FitByChannel' if args.MLfit
histograms['ett']['stack'].Add(histograms['ett']['wz'], "hist")
histograms['ett']['stack'].Add(histograms['ett']['signal'], "hist")
errorETT=histograms['ett']['zz'].Clone()
errorETT.SetFillStyle(3013)
errorETT.Add(histograms['ett']['fakes'])
errorETT.Add(histograms['ett']['wz'])
errorETT.SetMarkerSize(0)
errorETT.SetFillColor(1)
errorETT.SetLineWidth(1)
# Apply some styles to all the histograms
for channel in ['emt', 'mmt', 'eet', 'llem', 'llet', 'llmt', 'lltt', 'ett', 'mtt']:
# Use Poissonian error bars. The set_zero_bins makes it so bins w/o
# any data are blank.
histograms[channel]['poisson'] = convert(histograms[channel]['data'],
set_zero_bins=-10)
# Make sure all data points are visible
fix_maximum(histograms[channel])
# We have to draw it so things like the axes are initialized.
histograms[channel]['stack'].Draw()
#histograms[channel]['stack'].GetYaxis().SetTitle(
# "#bf{Events/%i GeV}" % histograms[channel]['data'].GetBinWidth(1))
histograms[channel]['stack'].GetYaxis().SetTitle(
"#bf{Events/ bin width [1/GeV]}" )
if channel=='lltt' or channel=="llet" or channel=="llmt" or channel=="llem":
histograms[channel]['stack'].GetXaxis().SetTitle("#bf{m_{#tau#tau} [GeV]}")
else:
histograms[channel]['stack'].GetXaxis().SetTitle("#bf{m_{vis} [GeV]}")
plot_suffix = "_%s_%s_%s.pdf" % (
def makePlots(dir="2012",postfix="8TeV",lumi="2.95",extra="1",var="z1Mass",varNice="Z_{M}^{1}",bins=range(60,120,5),legx=0.2,legW=0.3,legy=0.3,legH=0.3,log=False,binNorm=False):
fd=TFile(dataFile)
d=getTrees(fd,"")
eventDump(d["eeee"],extra)
eventDump(d["eemm"],extra)
eventDump(d["mmmm"],extra)
eventDump(fd.Get("llllTree"),extra)
if binNorm:
postfix = postfix + "_binNorm"
print "HOT DAMN! IT'S BIN NORMED!"
year=dir
if year=="2012":
l="8"
else:
l="7"
fzz = TFile("/scratch/iross/ZZ_wGen_proper_weights_selected.root")
zz=getTrees(fzz,"")
# fzza = TFile("/scratch/iross/aTGC_zz_f5z_0.000_f5g_0.000_processed.root")
# zza=getTrees(fzza,"")
# fzza2 = TFile("/scratch/iross/aTGC_zz_f5z_0.015_f5g_0.000_processed.root")
# zza2=getTrees(fzza2,"")
fzza = TFile("/scratch/iross/ZZ_wGen_proper_weights_selected.root")
zza=getTrees(fzza,"")
fzza2 = TFile("/scratch/iross/ZZ_wGen_proper_weights_selected.root")
zza2=getTrees(fzza2,"")
fzj=TFile(dataFile)
zj=getTrees(fzj,"")
fh126=TFile("/scratch/iross/h126.root")
h126t=getTrees(fh126,"")
fh350=TFile("/scratch/iross/h350.root")
h350t=getTrees(fh350,"")
dht=TH1F("dh","dh",len(bins)-1,array('d',bins))
dh={}
ecount={}
for tree in d:
t=d[tree]
dh[tree]=makeHist(t,var,extra,50,100,600,False,True,bins,"temp"+tree,binNorm=binNorm)
ecount[tree]=dh[tree].Integral()
dht.Add(dh[tree])
if not binNorm:
dh[tree]=poisson.convert(dh[tree],False,-1000)
dh["4l"]=dht
ecount["4l"]=dht.Integral()
if not binNorm:
dh["4l"]=poisson.convert(dh["4l"],False,-1000)
zzht=TH1F("zzh","zzh",len(bins)-1,array('d',bins))
zzh={}
for tree in zz:
t=zz[tree]
zzh[tree]=makeHist(t,var,"("+extra+")*(weight*tnp_weight_final*149/139*"+lumi+"*1000)",50,100,600,False,True,bins,name="temp_zz_"+tree,binNorm=binNorm)
zzht.Add(zzh[tree])
zzh[tree].SetFillColor(kAzure-9)
zzh[tree].SetMarkerSize(0.001)
zzh["4l"]=zzht
# add aTGC sample
zzaht=TH1F("zzh","zzh",len(bins)-1,array('d',bins))
zzah={}
for tree in zza:
t=zza[tree]
zzah[tree]=makeHist(t,var,"("+extra+")*(weight*scale_down*tnp_weight_final*"+lumi+"*1000)",50,100,600,False,True,bins,name="temp_zza_"+tree,binNorm=binNorm) #do I need this?
# # zzah[tree]=makeHist(t,var,"1*"+lumi+"*1000",50,100,600,False,True,bins)
zzaht.Add(zzah[tree])
zzah[tree].SetFillColor(kWhite-9)
zzah[tree].SetFillStyle(4000)
zzah[tree].SetMarkerSize(0.001)
zzah["4l"]=zzaht
# add aTGC sample
zzaht2=TH1F("zzh","zzh",len(bins)-1,array('d',bins))
zzah2={}
for tree in zza2:
t=zza2[tree]
zzah2[tree]=makeHist(t,var,"("+extra+")*(weight*scale_up*tnp_weight_final*"+lumi+"*1000)",50,100,600,False,True,bins,name="temp_zza2_"+tree,binNorm=binNorm) #do I need this?
zzaht2.Add(zzah2[tree])
zzah2[tree].SetFillColor(kWhite-9)
zzah2[tree].SetLineColor(kRed)
zzah2[tree].SetLineStyle(7)
zzah2[tree].SetFillStyle(4000)
zzah2[tree].SetMarkerSize(0.001)
zzah2["4l"]=zzaht2
# add Higgs sample
h126ht=TH1F("h126h","h126h",len(bins)-1,array('d',bins))
h126h={}
for tree in h126t:
t=h126t[tree]
h126h[tree]=makeHist(t,var,"("+extra+"&&(mass>100&&mass<150))*(__WEIGHT__*__CORR__*"+lumi+"*1000)",50,100,600,False,True,bins,name="temp_h126_"+tree,binNorm=binNorm) #do I need this?
h126ht.Add(h126h[tree])
h126h[tree].SetFillColor(kWhite)
h126h[tree].SetLineColor(kRed)
h126h[tree].SetLineWidth(2)
h126h[tree].SetMarkerSize(0.001)
h126h["4l"]=h126ht
h350ht=TH1F("h350h","h350h",len(bins)-1,array('d',bins))
h350h={}
for tree in h350t:
t=h350t[tree]
h350h[tree]=makeHist(t,var,"("+extra+"&&(mass>250&&mass<450))*(__WEIGHT__*__CORR__*"+lumi+"*1000)",50,100,600,False,True,bins,name="temp_h350_"+tree,binNorm=binNorm) #do I need this?
h350ht.Add(h350h[tree])
h350h[tree].SetFillColor(kWhite)
h350h[tree].SetLineColor(kGreen+3)
h350h[tree].SetLineWidth(2)
h350h[tree].SetMarkerSize(0.001)
h350h["4l"]=h350ht
zjht=TH1F("zjh","zjh",len(bins)-1,array('d',bins))
zjh={}
for tree in zj:
zjh[tree]=makeBGhist(fd,tree,var,True,bins,extra,binNorm=binNorm,fakeRates=fakeRates)
##temp use AA for shape
# t=fd.Get("eeeeAAFinal")
# zjh[tree]=makeHist(t,var,"("+extra+")",50,100,600,False,True,bins)
# if tree is "eeee":
# zjh[tree].Scale(0.000781733566027)
# elif tree is "eemm":
# zjh[tree].Scale(0.00182164048866)
# elif tree is "mmmm":
# zjh[tree].Scale(0.000998080279232)
##temp
# zjh[tree]=makeHist(t,var,"("+extra+")*(__WEIGHT__noPU*"+lumi+"*1000)",50,100,600,False,True,bins)
zjh[tree].SetFillColor(kGreen-5)
zjh[tree].SetMarkerSize(0.001)
# zjh['mmmm'].Scale(0.52/zjh['mmmm'].Integral())
# zjh['eemm'].Scale(0.58/zjh['eemm'].Integral())
# zjh['eeee'].Scale(0.25/zjh['eeee'].Integral())
for hist in zjh:
zjht.Add(zjh[hist])
zjh['4l']=zjht
#colors, etc.
zzht.SetMarkerSize(0.001)
zzht.SetFillColor(kAzure-9)
h126ht.SetMarkerSize(0.001)
h126ht.SetFillColor(kWhite)
h126ht.SetLineColor(kRed)
h126ht.SetLineWidth(2)
h350ht.SetMarkerSize(0.001)
h350ht.SetFillColor(kWhite)
h350ht.SetLineColor(kGreen+3)
h350ht.SetLineWidth(2)
zjht.SetMarkerSize(0.001)
zjht.SetFillColor(kGreen-5)
zzaht.SetMarkerSize(0.001)
zzaht2.SetMarkerSize(0.001)
f=open(dir+"/yields.txt","w")
fd.cd()
can = TCanvas("can","can",600,600)
leg=TLegend(legx,legy,legx+legW,legy+legH)
#legend
leg.SetFillColor(kWhite)
leg.AddEntry(dht,"Data","p")
leg.AddEntry(zzht,"ZZ","f")
leg.AddEntry(zjht,"Z+X","f")
# leg.AddEntry(h126ht,"H(126)","l")
# leg.AddEntry(h350ht,"H(350)","l")
# leg.AddEntry(zzaht,"Down","l")
# leg.AddEntry(zzaht2,"Up","l")
# leg.AddEntry(zzaht,"SHERPA","l")
leg.SetBorderSize(1)
for state in dh:
f.write("---"+state+"---\n")
f.write("Data: "+str(ecount[state])+"\n")
f.write("ZZ: "+str(zzh[state].Integral())+"\n")
f.write("ZJets:"+str(zjh[state].Integral())+"\n")
f.write("h126:"+str(h126h[state].Integral())+"\n")
f.write("h350:"+str(h350h[state].Integral())+"\n")
if not os.path.exists(dir+"/"+state):
os.makedirs(dir+"/"+state)
hs=THStack("hs","stack bg")
hs.Add(zjh[state])
hs.Add(zzh[state])
hse=zzh[state].Clone()
hse.Add(zjh[state])
hse.SetMarkerStyle(1)
hse.SetMarkerSize(0.0001)
hse.SetFillColor(kGray+2)
hse.SetFillStyle(3004)
hs.Add(h126h[state])
hs.Add(h350h[state])
ymax=max(dh[state].GetMaximum(),hs.GetMaximum())
if state is "4l":
ymax=max(dht.GetMaximum(),hs.GetMaximum())
print "ymax was:",ymax
if binNorm:
ymax=ceil(1.1*ymax)
else:
ymax=ceil(1.02*(ROOT.Math.gamma_quantile_c(0.3173/2,ymax+1,1) )) #for 68% coverage
print "ymax is now:",ymax
ymax=int(ymax)
dummy=TH1F("dummy","dummy",len(bins)-1,array('d',bins))
if bins[0]-bins[1] == bins[len(bins)-2]-bins[len(bins)-1]: # if spaced evenly
div=(float(bins[len(bins)-1])-float(bins[0]))/(len(bins)-1)
dummy.GetYaxis().SetTitle("Events / %.0f GeV" %div)
dummy.GetYaxis().SetRange(0,15)
else:
dummy.GetYaxis().SetTitle("Events")
dummy.GetYaxis().SetRange(0,15)
if binNorm:
varNiceNoUnits=re.sub('\([^)]*\)','',varNice)
dummy.GetYaxis().SetTitle("dN/d"+varNice)
dummy.GetXaxis().SetTitle(varNice)
if log:
dummy.GetYaxis().SetRangeUser(0.01,ymax)
else:
dummy.GetYaxis().SetRangeUser(0.,ymax)
dummy.Draw()
l1 = TLatex(bins[0]+(bins[len(bins)-1]-bins[0])/100.0,ymax*1.015,"CMS Preliminary "+year);
l1.SetTextSize(0.04);
l2 = TLatex(bins[0]+(bins[len(bins)-1]-bins[0])/2.0,ymax*1.015,"L_{int} ="+lumi+" fb^{-1}, #sqrt{s} = "+l+" TeV");
l2.SetTextSize(0.04);
if log:
l1 = TLatex(bins[0]+(bins[len(bins)-1]-bins[0])/100.0,ymax*1.08,"CMS Preliminary "+year);
l1.SetTextSize(0.04);
l2 = TLatex(bins[0]+(bins[len(bins)-1]-bins[0])/2.0,ymax*1.08,"L_{int} ="+lumi+" fb^{-1}, #sqrt{s} = "+l+" TeV");
l2.SetTextSize(0.04);
hsa=THStack("hsa","stack bg")
zzah[state].SetLineStyle(7)
zzah[state].SetLineWidth(2)
zzah[state].SetMarkerSize(0.0002)
hsa.Add(zjh[state])
hsa.Add(zzah[state])
hsa2=THStack("hsa2","stack bg")
zzah2[state].SetLineColor(kRed)
zzah2[state].SetLineStyle(7)
zzah2[state].SetLineWidth(2)
zzah2[state].SetMarkerSize(0.0002)
hsa2.Add(zjh[state])
hsa2.Add(zzah2[state])
dh[state].SetMarkerStyle(20)
dh[state].SetMarkerSize(1)
hs.Draw("HIST SAME")
# hsa.Draw("HIST E SAME")
# hsa2.Draw("HIST E SAME")
hse.Draw("e2same")
dh[state].Draw("pesame")
leg.Draw()
l1.Draw();
l2.Draw();
if log:
can.SetLogy(1)
# strip special chars out of variable name for easier saving
var=re.sub('[/,.\{\}()+><\* =]','',var)
if str(bins[0]-bins[1]) == str(bins[len(bins)-2]-bins[len(bins)-1]): # if spaced evenly
if len(var)>20 and "Phi" in var:
var="deltaPhi_l1_l2"
can.SaveAs(dir+"/"+state+"/"+state+"_"+var+""+postfix+".png")
can.SaveAs(dir+"/"+state+"/"+state+"_"+var+""+postfix+".pdf")
can.SaveAs(dir+"/"+state+"/"+state+"_"+var+""+postfix+".root")
can.SaveAs(dir+"/"+state+"/"+state+"_"+var+""+postfix+".C")
else:
if len(var)>20 and "Phi" in var:
var="deltaPhi_l1_l2"
binStr=""
print
for i in range(len(bins)):
binStr+="_"+str(bins[i])
can.SaveAs(dir+"/"+state+"/"+state+"_"+var+"_varbinning"+postfix+binStr+".png")
can.SaveAs(dir+"/"+state+"/"+state+"_"+var+"_varbinning"+postfix+binStr+".pdf")
can.SaveAs(dir+"/"+state+"/"+state+"_"+var+"_varbinning"+postfix+binStr+".root")
can.SaveAs(dir+"/"+state+"/"+state+"_"+var+"_varbinning"+postfix+binStr+".C")
f.close()
def makeBGPlots(BGtype="AA",dir="2012",postfix="8TeV",lumi="2.95",extra="1",var="z1Mass",varNice="Z_{M}^{1}",bins=range(60,120,5),legx=0.2,legW=0.3,legy=0.3,legH=0.3,log=False,binNorm=False):
year=dir
if year=="2012":
l="8"
else:
l="7"
fd=TFile(dataFile)
d=getBGTrees(fd,BGtype)
print d
fzz=TFile("qqZZ_8TeV_final_selected.root")
zz=getBGTrees(fzz,BGtype)
fzj=TFile("DYJets_8TeV_loose_9Jun-mergeFilesJob_selected.root")
zj=getBGTrees(fzj,BGtype)
print zj
ftt=TFile("TTbar_8TeV_thesisBG_loose-mergeFilesJob_selected.root")
tt=getBGTrees(ftt,BGtype)
fwz=TFile("WZ_8TeV_thesisBG_loose-mergeFilesJob_selected.root")
twz=getBGTrees(fwz,BGtype)
dht=TH1F("dh","dh",len(bins)-1,array('d',bins))
dh={}
dhmax={} #store max values, since TGraphAsymmErrors don't play well with GetMaximum used for scaling..
# fakerates=measureLeptonFakes(fd.GetName(),extra="&&z1Mass>81&&z1Mass<101") #todo: propagate 'extra' properly
for tree in d:
if "eeee" in tree or "mmee" in tree:
fr=fakeRates[0]
elif "eemm" in tree or "mmmm" in tree:
fr=fakeRates[1]
else:
print("Can't figure out which fakerate to use!")
if "AA" in BGtype:
FR=str(fr*fr/(1-fr)/(1-fr))
elif "AI" in BGtype or "IA" in BGtype:
FR=str(fr/(1-fr))
else:
print("Can't figure out which pass-fail region!")
t=d[tree]
print tree,BGtype,t
dh[tree]=makeHist(t,var,extra,50,100,600,False,True,bins,binNorm=binNorm)
dht.Add(dh[tree])
dhmax[tree]=dh[tree].GetMaximum()
if not binNorm:
temph=poisson.convert(dh[tree],False,-1000)
dh[tree]=temph
dhmax["4l"]=dht.GetMaximum()
if not binNorm:
dh["4l"]=poisson.convert(dht,False,-1000)
# zzht=TH1F("zzh","zzh",len(bins)-1,array('d',bins))
# zzh={}
# for tree in zz:
# t=zz[tree]
# print tree,BGtype,t
# zzh[tree]=makeHist(t,var,"("+extra+")*(__WEIGHT__*1.0*"+lumi+"*1000)",50,100,600,False,True,bins)
# zzht.Add(zzh[tree])
# zzh[tree].SetFillColor(kAzure-9)
# zzh[tree].SetMarkerSize(0.001)
# zzh["4l"]=zzht
ttht=TH1F("tth","tth",len(bins)-1,array('d',bins))
tth={}
for tree in tt:
t=tt[tree]
tth[tree]=makeHist(t,var,"("+extra+")*(weight*1.0*"+lumi+"*1000)",50,100,600,False,True,bins)
ttht.Add(tth[tree])
tth[tree].SetFillColor(kRed+1)
tth[tree].SetMarkerSize(0.001)
tth["4l"]=ttht
wzht=TH1F("tth","tth",len(bins)-1,array('d',bins))
wzh={}
for tree in twz:
t=twz[tree]
wzh[tree]=makeHist(t,var,"("+extra+")*(weight*1.0*"+lumi+"*1000)",50,100,600,False,True,bins)
wzht.Add(wzh[tree])
wzh[tree].SetFillColor(kYellow-9)
wzh[tree].SetMarkerSize(0.001)
wzh["4l"]=wzht
zjht=TH1F("zjh","zjh",len(bins)-1,array('d',bins))
zjh={}
for tree in zj:
print "------------",tree,"-------------"
t=zj[tree]
print t
zjh[tree]=makeHist(t,var,"("+extra+")*(weight*1.0*"+lumi+"*1000)",50,100,600,False,True,bins,binNorm=binNorm)
zjht.Add(zjh[tree])
zjh[tree].SetFillColor(kGreen-5)
zjh[tree].SetMarkerSize(0.001)
zjh['4l']=zjht
#colors, etc.
# zzht.SetMarkerSize(0.001)
# zzht.SetFillColor(kAzure-9)
zjht.SetMarkerSize(0.001)
zjht.SetFillColor(kGreen-5)
ttht.SetMarkerSize(0.001)
ttht.SetFillColor(kRed+1)
wzht.SetMarkerSize(0.001)
wzht.SetFillColor(kYellow-9)
can = TCanvas("can","can",600,600)
leg=TLegend(legx,legy,legx+legW,legy+legH)
#legend
leg.SetFillColor(kWhite)
leg.AddEntry(dht,"Data","p")
# leg.AddEntry(zzht,"ZZ","f")
leg.AddEntry(zjht,"Z+X","f")
leg.AddEntry(ttht,"ttbar","f")
leg.AddEntry(wzht,"WZ","f")
leg.SetBorderSize(1)
f=open(dir+"/yields.txt","w")
for state in dh:
if BGtype=="AA":
leg.SetHeader(state.replace("m","#mu")+", Z+2F Region")
else:
leg.SetHeader(state.replace("m","#mu")+", Z+1P1F Region")
f.write("---"+state+"---\n")
f.write("Data: "+str(dh[state].Integral())+"\n")
# f.write("ZZ: "+str(zzh[state].Integral())+"\n")
# f.write("ZJets:"+str(zjh[state].Integral())+"\n")
if not os.path.exists(dir+"/"+state):
os.makedirs(dir+"/"+state)
hs=THStack("hs","stack bg")
# hs.Add(zzh[state])
hs.Add(wzh[state])
hs.Add(tth[state])
hs.Add(zjh[state])
hse=zjh[state].Clone()
hse.Add(tth[state])
hse.Add(wzh[state])
hse.SetMarkerStyle(1)
hse.SetMarkerSize(0.0001)
hse.SetFillColor(kGray+2)
hse.SetFillStyle(3004)
ymax=max(dhmax[state],hs.GetMaximum())
if state is "4l":
ymax=max(dht.GetMaximum(),hs.GetMaximum())
ymax=ceil(1.02*(ROOT.Math.gamma_quantile_c(0.3173/2,ymax+1,1) )) #for 68% coverage
print ymax,"is y max"
ymax=int(ymax)
dummy=TH1F("dummy","dummy",len(bins)-1,array('d',bins))
dummy.GetYaxis().SetRangeUser(0.,ymax)
if bins[0]-bins[1] == bins[len(bins)-2]-bins[len(bins)-1]: # if spaced evenly
div=(float(bins[len(bins)-1])-float(bins[0]))/(len(bins)-1)
dummy.GetYaxis().SetTitle("Events / %.0f GeV" %div)
dummy.GetYaxis().SetRange(0,15)
else:
dummy.GetYaxis().SetTitle("Events")
dummy.GetYaxis().SetRange(0,15)
if binNorm:
dummy.GetYaxis().SetTitle("dN/d"+varNice)
dummy.GetXaxis().SetTitle(varNice)
dummy.Draw()
l1 = TLatex(bins[0]+(bins[len(bins)-1]-bins[0])/100.0,ymax*1.015,"CMS Preliminary "+year);
l1.SetTextSize(0.04);
l2 = TLatex(bins[0]+(bins[len(bins)-1]-bins[0])/2.0,ymax*1.015,"L_{int} ="+lumi+" fb^{-1}, #sqrt{s} = "+l+" TeV");
l2.SetTextSize(0.04);
if log:
dh[state].GetYaxis().SetRangeUser(0.1,ymax)
else:
dh[state].GetYaxis().SetRangeUser(0.,ymax)
dh[state].SetMarkerStyle(20)
dh[state].SetMarkerSize(1)
hs.Draw("HIST SAME")
# hsa.Draw("hsame")
hse.Draw("e2same")
dh[state].Draw("pesame")
leg.Draw()
l1.Draw();
l2.Draw();
if log:
can.SetLogy(1)
if str(bins[0]-bins[1]) == str(bins[len(bins)-2]-bins[len(bins)-1]): # if spaced evenly
can.SaveAs(dir+"/BG/"+state+"_"+BGtype+"_"+state+"_"+var+""+postfix+".png")
can.SaveAs(dir+"/BG/"+state+"_"+BGtype+"_"+state+"_"+var+""+postfix+".pdf")
can.SaveAs(dir+"/BG/"+state+"_"+BGtype+"_"+state+"_"+var+""+postfix+".C")
can.SaveAs(dir+"/BG/"+state+"_"+BGtype+"_"+state+"_"+var+""+postfix+".root")
else:
binStr=""
print
for i in range(len(bins)-1):
binStr+="_"+str(bins[i])
can.SaveAs(dir+"/BG/"+state+"_"+BGtype+"_"+state+"_"+var+"_varbinning"+postfix+binStr+".png")
can.SaveAs(dir+"/BG/"+state+"_"+BGtype+"_"+state+"_"+var+"_varbinning"+postfix+binStr+".pdf")
can.SaveAs(dir+"/BG/"+state+"_"+BGtype+"_"+state+"_"+var+"_varbinning"+postfix+binStr+".C")
can.SaveAs(dir+"/BG/"+state+"_"+BGtype+"_"+state+"_"+var+"_varbinning"+postfix+binStr+".root")
f.close()
b = ZZ.GetBinContent(k) + Fake.GetBinContent(k)
if channels[0] in wh_channels:
b = ZZ.GetBinContent(k) + Fake.GetBinContent(k) + WZ.GetBinContent(
k)
if (b < 0):
b = 0.000001
if (0.2 * s / (0.00001 + 0.05 * s + b)**0.5 > 0.15):
Data.SetBinContent(k, -10)
Data.SetBinError(k, 0)
# blind ZH on mass peak for high LT always
if channels[0] in zh_channels and (k == 15 or k == 16):
Data.SetBinContent(k, -10)
Data.SetBinError(k, 0)
Data.SetMinimum(0)
Poisson = convert(Data)
Poisson.GetXaxis().SetTitle("")
Poisson.GetXaxis().SetTitleSize(0)
#FIXME Poisson.GetXaxis().SetNdivisions(505)
Poisson.GetYaxis().SetLabelFont(42)
Poisson.GetYaxis().SetLabelOffset(0.01)
Poisson.GetYaxis().SetLabelSize(0.06)
Poisson.GetYaxis().SetTitleSize(0.075)
Poisson.GetYaxis().SetTitleOffset(1.04)
Poisson.SetTitle("")
Poisson.GetYaxis().SetTitle("Events/bin")
Poisson.SetMinimum(0)
Poisson.SetMarkerStyle(20)
Poisson.SetLineColor(1)
Poisson.SetMarkerSize(1)
Poisson.GetXaxis().SetLabelSize(0)
'8' : '12',
'7' : '5',
}
# Draw LLT
vh_legend = make_a_legend()
llt = llt_stack.Get(None)
llt.Draw()
llt.GetHistogram().GetXaxis().SetRangeUser(0, 200)
llt.GetHistogram().GetXaxis().SetTitle("m_{vis} [GeV]")
llt.GetHistogram().GetXaxis().SetLabelSize(0.037)
llt.GetHistogram().GetYaxis().SetTitle("Events")
llt_data = channels['llt']['obs'].Get(None)
llt.GetHistogram().GetYaxis().SetTitle("Events/%i GeV" % llt_data.GetBinWidth(1))
#llt_data.Draw('same, pe')
llt_data_pois = convert(llt_data, set_zero_bins=-10)
llt_data_pois.Draw('pe')
llt.SetMaximum(1.5*get_real_maximum(llt_data))
blurb = add_cms_blurb(period, lumi_maps[period], blurb=' ll#tau_{h}')
vh_legend.AddEntry(llt_data)
vh_legend.AddEntry(llt)
vh_legend.Draw()
canvas.Update()
canvas.SaveAs('llt' + period + pf_suffix + '.pdf')
canvas.SaveAs('llt' + period + pf_suffix + '.png')
## Draw LTT
#vh_legend = make_a_legend()
#ltt = ltt_stack.Get(None)
#ltt.Draw()
#ltt.Draw()
