def main(): c1 = Print("transFac.pdf") c1.DoPageNum = False h = r.TH1D("","",1000,0.,1000.) xbins = [275.,325,] + [375 + i*100 for i in range(6)] transFac = [ 2.34,2.06,2.10,2.18,2.01,2.45,2.48,2.54] transFacErr = [0.1,0.14,0.04,0.16,0.16,0.25,0.33,0.40] h = h.Rebin(len(xbins)-1,"",array.array('d',xbins)) for b in range(0,h.GetNbinsX()+1): h.SetBinContent(b,transFac[b]) h.SetBinError(b,transFacErr[b]) h.GetXaxis().SetTitle("H_{T} (GeV)") h.GetYaxis().SetTitleSize(h.GetXaxis().GetTitleSize()) h.GetYaxis().SetTitle("Translation Factor") h.Draw() c1.canvas.Update() raw_input() # Do pol0 fit here by hand in the root interface. c1.Print() c1.close()
def main(): c1 = Print("transFac.pdf") c1.DoPageNum = False h = r.TH1D("", "", 1000, 0., 1000.) xbins = [ 275., 325, ] + [375 + i * 100 for i in range(6)] transFac = [2.34, 2.06, 2.10, 2.18, 2.01, 2.45, 2.48, 2.54] transFacErr = [0.1, 0.14, 0.04, 0.16, 0.16, 0.25, 0.33, 0.40] h = h.Rebin(len(xbins) - 1, "", array.array('d', xbins)) for b in range(0, h.GetNbinsX() + 1): h.SetBinContent(b, transFac[b]) h.SetBinError(b, transFacErr[b]) h.GetXaxis().SetTitle("H_{T} (GeV)") h.GetYaxis().SetTitleSize(h.GetXaxis().GetTitleSize()) h.GetYaxis().SetTitle("Translation Factor") h.Draw() c1.canvas.Update() raw_input() # Do pol0 fit here by hand in the root interface. c1.Print() c1.close()
directory = File.Get(f) a = directory.Get(hist) if h is None: h = a.Clone() else: h.Add(a) return h def nloTotalXsecMaker(individualXSecs = None): out = None for h in individualXSecs: if out is None: out = h.Clone() else: out.Add(h) return out c1 = Print("./out/JES_CMSSM_NEWSCAN.pdf") c1.DoPageNum = False c1.open() r.gPad.SetRightMargin(0.175) centalRootFile73 = r.TFile.Open("./JESandLepVetoRootFiles/results_NEWSCAN_JESCEN_had_tanB10_73.7.root") centalRootFile86 = r.TFile.Open("./JESandLepVetoRootFiles/results_NEWSCAN_JESCEN_had_tanB10_86.7.root") centalRootFile100 = r.TFile.Open("./JESandLepVetoRootFiles/results_NEWSCAN_JESCEN_had_tanB10_100.root") jesPlusRootFile73 = r.TFile.Open("./JESandLepVetoRootFiles/results_NEWSCAN_had_tanB10_73.7_+ve.root") jesPlusRootFile86 = r.TFile.Open("./JESandLepVetoRootFiles/results_NEWSCAN_had_tanB10_86.7_+ve.root") jesPlusRootFile100 = r.TFile.Open("./JESandLepVetoRootFiles/results_NEWSCAN_had_tanB10_100_+ve.root") jesNegRootFile73 = r.TFile.Open("./JESandLepVetoRootFiles/results_NEWSCAN_had_tanB10_73.7_-ve.root") jesNegRootFile86 = r.TFile.Open("./JESandLepVetoRootFiles/results_NEWSCAN_had_tanB10_86.7_-ve.root") jesNegRootFile100 = r.TFile.Open("./JESandLepVetoRootFiles/results_NEWSCAN_had_tanB10_100_-ve.root")
# print directory.ls() a = directory.Get(hist) if h is None: h = a.Clone() else: h.Add(a) return h def nloTotalXsecMaker(individualXSecs = None): out = None for h in individualXSecs: if out is None: out = h.Clone() else: out.Add(h) return out c1 = Print("EffectsOfLepVeto_GenLepVeto_T2tt.pdf") c1.open() r.gPad.SetRightMargin(0.15) # centalRootFile100 = r.TFile.Open("./NMinusOneT2tt.root") # centalRootFile73.ls() # Make cross sections/ efficiencies nocuts = threeToTwo(GetHist(File = r.TFile.Open("./results_GenLepVeto_had_T2tt_100.root"),folder = ["smsScan_before",], hist = "m0_m12_mChi_noweight", Norm = None ,rebin= 2)) nocuts = threeToTwo(nocuts) AllCuts_GenLep = threeToTwo(GetHist(File = r.TFile.Open("results_GenLepPhotonVeto_had_T2tt_100.root"),folder = ["smsScan_AlphaT55_375_475","smsScan_AlphaT55_475_575","smsScan_AlphaT55_575_675",
xTitle = "m_{stop} (GeV)" yTitle = "m_{LSP} (GeV)" if model == "T1": xTitle = "m_{gluino} (GeV)" yTitle = "m_{LSP} (GeV)" if model == "T2": xTitle = "m_{squark} (GeV)" yTitle = "m_{LSP} (GeV)" if model == "T1tttt": xTitle = "m_{gluino} (GeV)" yTitle = "m_{LSP} (GeV)" if model == "T1bbbb": xTitle = "m_{gluino} (GeV)" yTitle = "m_{LSP} (GeV)" c1 = Print("./out/JES_SMS%s.pdf" % (model)) c1.DoPageNum = False c1.open() r.gPad.SetRightMargin(0.175) r.gPad.SetLeftMargin(0.15) r.gPad.SetTopMargin(0.05) r.gPad.SetBottomMargin(0.15) # centalRootFile73 = r.TFile.Open("./JESandLepVetoRootFiles/results_FIX_NEWSCAN_had_T2bb_73.7.root") # centalRootFile86 = r.TFile.Open("./JESandLepVetoRootFiles/results_FIX_NEWSCAN_had_T2bb_86.7.root") centalRootFile100 = r.TFile.Open( "./JESandLepVetoRootFiles/results_NEWSCAN_JESCEN_had_%s_100.root" % (model)) # jesPlusRootFile73 = r.TFile.Open("./JESandLepVetoRootFiles/results_had_T2bb_73.7_+ve.root") # jesPlusRootFile86 = r.TFile.Open("./JESandLepVetoRootFiles/results_had_T2bb_86.7_+ve.root") jesPlusRootFile100 = r.TFile.Open( "./JESandLepVetoRootFiles/results_NEWSCAN_had_%s_100_+ve.root" %
if model == "T2": xTitle = "m_{squark} (GeV)" yTitle = "m_{LSP} (GeV)" if model == "T1tttt": xTitle = "m_{gluino} (GeV)" yTitle = "m_{LSP} (GeV)" if model == "T1bbbb": xTitle = "m_{gluino} (GeV)" yTitle = "m_{LSP} (GeV)" if model == "T1ttttProto": xTitle = "m_{gluino} (GeV)" yTitle = "m_{LSP} (GeV)" c1 = Print("./out/BTagScale_SMS%s.pdf"%(model)) c1.DoPageNum = False r.gPad.SetRightMargin(0.175) r.gPad.SetLeftMargin(0.15) r.gPad.SetTopMargin(0.05) r.gPad.SetBottomMargin(0.15) centalRootFile73 = r.TFile.Open("./rootFiles/sigScan_%s_had_2012_73.7_bt0.0_MChi-1.root"%(model)) centalRootFile86 = r.TFile.Open("./rootFiles/sigScan_%s_had_2012_86.7_bt0.0_MChi-1.root"%(model)) centalRootFile100 = r.TFile.Open("./rootFiles/sigScan_%s_had_2012_100.0_bt0.0_MChi-1.root"%(model)) jesPlusRootFile73 = r.TFile.Open("./rootFiles/sigScan_%s_had_2012_73.7_bt1.0_MChi-1.root"%(model)) jesPlusRootFile86 = r.TFile.Open("./rootFiles/sigScan_%s_had_2012_86.7_bt1.0_MChi-1.root"%(model)) jesPlusRootFile100 = r.TFile.Open("./rootFiles/sigScan_%s_had_2012_100.0_bt1.0_MChi-1.root"%(model))
xTitle = "m_{gluino} (GeV)" yTitle = "m_{LSP} (GeV)" maxi = .1 mini = 1E-3 m_gl_m_m_lsp = 375. m_sq = 450. if model == "T1bbbb": xTitle = "m_{gluino} (GeV)" yTitle = "m_{LSP} (GeV)" maxi = .1 mini = 1E-3 m_gl_m_m_lsp = 175. m_sq = 300. c1 = Print("./out/%s_SMS%s.pdf" % (cut, model)) c1.DoPageNum = False r.gPad.SetRightMargin(0.175) r.gPad.SetLeftMargin(0.15) r.gPad.SetTopMargin(0.05) r.gPad.SetBottomMargin(0.15) print model, cut FullCutFlowRootFile100 = r.TFile.Open( "./rootFiles/sigScan_%s_had_2012_100.0_bt0.0_MChi-1.root" % (model)) NMinus1RootFile100 = r.TFile.Open( "./rootFiles/sigScan_%s_had_2012_100.0_bt0.0_MChi-1_!%s.root" % (model, cut)) # centalRootFile73.ls() # Make cross sections/ efficiencies
def make_syst_map(central_files=[], positive_files=[], negative_files=[], bMulti="", jMulti="", syst_mode=""): xTitle = "m_{stop} (GeV)" if settings["deltaM"]: yTitle = "deltaM (GeV)" else: yTitle = "m_{LSP} (GeV)" for mode in settings["mode"]: # need to find a better way to do this! if mode != "bTag": centalRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_73.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"]) ) centalRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_86.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"]) ) centalRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_100.0_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"]) ) jesPlusRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_73.7_%s+ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"], ins) ) jesPlusRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_86.7_%s+ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"], ins) ) jesPlusRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_100.0_%s+ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"], ins) ) jesNegRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_73.7_%s-ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"], ins) ) jesNegRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_86.7_%s-ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"], ins) ) jesNegRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_100.0_%s-ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"], ins) ) else: centalRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_73.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"]) ) centalRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_86.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"]) ) centalRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_100.0_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"]) ) jesPlusRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_73.7_bt1.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"]) ) jesPlusRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_86.7_bt1.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"]) ) jesPlusRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_100.0_bt1.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"]) ) jesNegRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_73.7_bt-1.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"]) ) jesNegRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_86.7_bt-1.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"]) ) jesNegRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_100.0_bt-1.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"]) ) if mode == "ISR": ins = "isr_" else: ins = "" HTList = [] if settings["inclHT"]: HTList = ["incl"] else: HTList = settings["HTBins"] oF = open(getOutFile(model=settings["model"], format="txt", bMulti_=bMulti, jMulti_=jMulti), "w") for htbin in HTList: processCrossSections = [] cuts = [] cutsJESNeg = [] cutsJESPlus = [] nocuts = [] suf = "" if "275_" in htbin: suf = "73" elif "325_" in htbin: suf = "86" else: suf = "100" r.gROOT.SetBatch(r.kTRUE) c1 = Print(getOutFile(model=settings["model"], htbin=htbin, format="pdf")) c1.DoPageNum = False r.gPad.SetRightMargin(0.175) r.gPad.SetLeftMargin(0.15) r.gPad.SetTopMargin(0.08) r.gPad.SetBottomMargin(0.15) nocuts = GetHist( File=centalRootFile100, folder=["smsScan_before"], hist="m0_m12_mChi_weight", Norm=None, rebinX=4 ) nocuts = threeToTwo(nocuts) if settings["inclHT"]: cutsHist = GetHist( File=central_files[0], folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[0:2], hist="m0_m12_mChi_weight", Norm=None, ).Clone() cutsHist.Add( GetHist( File=central_files[1], folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[2:3], hist="m0_m12_mChi_weight", Norm=None, ) ) cutsHist.Add( GetHist( File=central_files[2], folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[3:], hist="m0_m12_mChi_weight", Norm=None, ) ) cutsJESPlusHist = GetHist( File=positive_files[0], folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[0:2], hist="m0_m12_mChi_weight", Norm=None, ).Clone() cutsJESPlusHist.Add( GetHist( File=positive_files[1], folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[2:3], hist="m0_m12_mChi_weight", Norm=None, ) ) cutsJESPlusHist.Add( GetHist( File=positive_files[2], folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[3:], hist="m0_m12_mChi_weight", Norm=None, ) ) cutsJESNegHist = GetHist( File=negative_files[0], folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[0:2], hist="m0_m12_mChi_weight", Norm=None, ).Clone() cutsJESNegHist.Add( GetHist( File=negative_files[1], folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[2:3], hist="m0_m12_mChi_weight", Norm=None, ) ) cutsJESNegHist.Add( GetHist( File=negative_files[2], folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[3:], hist="m0_m12_mChi_weight", Norm=None, ) ) else: print "f**k." exit() # convert to TH2 plots cutsHist = threeToTwo(cutsHist) cutsJESPlusHist = threeToTwo(cutsJESPlusHist) cutsJESNegHist = threeToTwo(cutsJESNegHist) if settings["deltaM"]: nocuts = deltaM(nocuts) cutsHist = deltaM(cutsHist) cutsJESPlusHist = deltaM(cutsJESPlusHist) cutsJESNegHist = deltaM(cutsJESNegHist) # zero out the above-diagonal region l = [i for i in range(301)] for a in l: xbinval = cutsHist.GetXaxis().GetBinCenter(a) for b in l: ybinval = cutsHist.GetYaxis().GetBinCenter(b) if xbinval - ybinval < 0.0 or a < 0.0: bin = cutsHist.FindBin(float(a), float(b)) cutsHist.SetBinContent(bin, 0.0) cutsJESPlusHist.SetBinContent(bin, 0.0) cutsJESNegHist.SetBinContent(bin, 0.0) if mode == "JES": mini = 0.9 maxi = 1.1 else: mini = 0.70 maxi = 1.30 c1.canvas.SetLogz() offset = 1.1 # c1.Print() c1.canvas.SetLogz(False) TotalEff = cutsHist.Clone() TotalEff.GetZaxis().SetTitle("Fraction of expected signal yield") TotalEff.GetZaxis().SetTitleOffset(offset) TotalEff.GetZaxis().SetTitleSize(0.05) TotalEff.GetXaxis().SetTitle(xTitle) TotalEff.SetTitle("Total Efficiency") TotalEff.GetYaxis().SetTitleOffset(1.3) TotalEff.GetYaxis().SetTitleSize(0.05) TotalEff.GetYaxis().SetTitle(yTitle) TotalEff.Divide(nocuts) maxVal = TotalEff.GetMaximum() # TotalEff.Scale(100.) # r.gStyle.SetPaintTextFormat("0.2f %%"); # TotalEff.SetMarkerSize(1.4) # r.gStyle.SetPalette(3) # TotalEff.Draw("COLZ TEXT40") TotalEff.Draw("COLZ") tot = 0.0 ctr = 0 for i in range(TotalEff.GetNbinsX() * TotalEff.GetNbinsY()): val = TotalEff.GetBinContent(i) if val > 0.0: tot += val ctr += 1 num0 = r.TLatex(0.17, 0.85, "Average efficiency: %.3f%%" % (float(tot / ctr) * 100)) num0.SetNDC() num0.Draw("same") c1.Print() TotalEffPlus = cutsJESPlusHist.Clone() TotalEffPlus.GetZaxis().SetTitle("Relative change in efficiency") TotalEffPlus.GetZaxis().SetTitleOffset(offset) TotalEffPlus.SetTitle("Total Up Efficiency") TotalEffPlus.Divide(nocuts) TotalEffPlus.GetXaxis().SetTitle(xTitle) TotalEffPlus.GetYaxis().SetTitle(yTitle) TotalEffPlus.GetZaxis().SetTitleSize(0.05) TotalEffPlus.GetYaxis().SetTitleOffset(1.3) TotalEffPlus.GetYaxis().SetTitleSize(0.05) TotalEffPlus.SetMaximum(maxVal) TotalEffPlus.Draw("COLZ") c1.Print() TotalEffNeg = cutsJESNegHist.Clone() TotalEffNeg.GetZaxis().SetTitle("Relative change in efficiency") TotalEffNeg.GetZaxis().SetTitleOffset(offset) TotalEffNeg.SetTitle("Total Down Efficiency") TotalEffNeg.Divide(nocuts) TotalEffNeg.GetZaxis().SetTitleSize(0.05) TotalEffNeg.GetXaxis().SetTitle(xTitle) TotalEffNeg.GetYaxis().SetTitle(yTitle) TotalEffNeg.GetYaxis().SetTitleOffset(1.3) TotalEffNeg.GetYaxis().SetTitleSize(0.05) TotalEffNeg.SetMaximum(maxVal) TotalEffNeg.Draw("COLZ") c1.Print() EffOverJESPlus = TotalEffPlus.Clone() EffOverJESPlus.SetTitle("Up deltaEff (Up / Total)") EffOverJESPlus.Divide(TotalEff) EffOverJESPlus.Draw("COLZ") EffOverJESPlus.GetZaxis().SetRangeUser(0.5, 1.5) # c1.Print() EffOverJESNeg = TotalEffNeg.Clone() EffOverJESNeg.SetTitle("Down deltaEff (Down / Total)") EffOverJESNeg.Divide(TotalEff) EffOverJESNeg.Draw("COLZ") EffOverJESNeg.GetZaxis().SetRangeUser(0.5, 1.5) # c1.Print() r.gStyle.SetOptStat(0) EffOverJESNegClone = EffOverJESNeg.Clone() EffOverJESPlusClone = EffOverJESPlus.Clone() # force the zRange accross all bins for bin in range(EffOverJESNegClone.GetNbinsX() * EffOverJESNegClone.GetNbinsY()): if EffOverJESNegClone.GetBinContent(bin) > 0: if EffOverJESNegClone.GetBinContent(bin) < mini: EffOverJESNegClone.SetBinContent(bin, mini) if EffOverJESNegClone.GetBinContent(bin) > maxi: EffOverJESNegClone.SetBinContent(bin, maxi) for bin in range(EffOverJESPlusClone.GetNbinsX() * EffOverJESPlusClone.GetNbinsY()): if EffOverJESPlusClone.GetBinContent(bin) > 0: if EffOverJESPlusClone.GetBinContent(bin) < mini: EffOverJESPlusClone.SetBinContent(bin, mini) if EffOverJESPlusClone.GetBinContent(bin) > maxi: EffOverJESPlusClone.SetBinContent(bin, maxi) # centre distribution around 0., set all others to 1000 for bin in range( EffOverJESPlusClone.GetXaxis().GetNbins() * EffOverJESPlusClone.GetYaxis().GetNbins() + 10000 ): if EffOverJESNegClone.GetBinContent(bin) > 0: EffOverJESNegClone.SetBinContent(bin, EffOverJESNegClone.GetBinContent(bin) - 1.0) else: EffOverJESNegClone.SetBinContent(bin, -1000) if EffOverJESPlusClone.GetBinContent(bin) > 0: EffOverJESPlusClone.SetBinContent(bin, EffOverJESPlusClone.GetBinContent(bin) - 1.0) else: EffOverJESPlusClone.SetBinContent(bin, -1000) # force range EffOverJESNegClone.SetMinimum(mini - 1.0) EffOverJESNegClone.SetMaximum(maxi - 1.0) EffOverJESPlusClone.SetMinimum(mini - 1.0) EffOverJESPlusClone.SetMaximum(maxi - 1.0) EffOverJESPlusClone.Draw("COLZ") c1.Print() EffOverJESNegClone.Draw("COLZ") c1.Print() if not settings["inclHT"]: oF.write( "%s\t\t%.3f +\\- %.3f\n" % (htbin, JesTotClone.GetBinLowEdge(bin68), JesTotClone.GetBinError(bin68)) ) c1.close() if not settings["inclHT"]: oF.close()
if "mhtmet" in cut: jmax = 0.4 elif "leptVeto" in cut: jmax = 0.03 elif "deadEcal" in cut: jmax = 0.3 else: jmax = 0.1 outName = "" if settings["deltaM"]: outName = "./out/LepVeto_SMS%s_%s_dM.pdf"%(model, cut) else: outName = "./out/LepVeto_SMS%s_%s.pdf"%(model, cut) c1 = Print(outName) c1.DoPageNum = False #c1.open() r.gPad.SetRightMargin(0.175) r.gPad.SetLeftMargin(0.15) r.gPad.SetTopMargin(0.05) r.gPad.SetBottomMargin(0.15) FullCutFlowRootFile100 = r.TFile.Open("./rootfiles/T2cc_v8/sigScan_%s_had_2012_100.0_bt0.0_MChi-1.0.root"%(model)) NMinus1RootFile100 = r.TFile.Open("./rootfiles/T2cc_v8/%s/sigScan_%s_had_2012_100.0_bt0.0_MChi-1.0.root"%(cut, model)) FullCutFlowRootFile87 = r.TFile.Open("./rootfiles/T2cc_v8/sigScan_%s_had_2012_86.7_bt0.0_MChi-1.0.root"%(model)) NMinus1RootFile87 = r.TFile.Open("./rootfiles/T2cc_v8/%s/sigScan_%s_had_2012_86.7_bt0.0_MChi-1.0.root"%(cut, model)) FullCutFlowRootFile74 = r.TFile.Open("./rootfiles/T2cc_v8/sigScan_%s_had_2012_73.7_bt0.0_MChi-1.0.root"%(model)) NMinus1RootFile74 = r.TFile.Open("./rootfiles/T2cc_v8/%s/sigScan_%s_had_2012_73.7_bt0.0_MChi-1.0.root"%(cut, model))
def make_syst_map_three(bMulti="", jMulti=""): set_palette() print ">>> Creating systs for", bMulti, jMulti xTitle = "m_{stop} (GeV)" if settings["deltaM"]: yTitle = "deltaM (GeV)" else: yTitle = "m_{LSP} (GeV)" total_syst_hist = r.TH2D() my_syst_hists = [] for n_syst, mode in enumerate(settings["mode"]): if "T2cc" not in settings['model'] and mode == "3jet": continue print " >>", mode if mode == "ISR": ins = "isr_" else: ins = "" cut_syst = False # set variable for a before/after type systematic if mode in ["LeptonVeto", "MHT_MET", "DeadECAL", "3jet"]: cut_syst = True # need to find a better way to do this! if mode in ["JES", "ISR"]: centalRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_73.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) centalRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_86.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) centalRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_100.0_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) jesPlusRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_73.7_%s+ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"], ins)) jesPlusRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_86.7_%s+ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"], ins)) jesPlusRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_100.0_%s+ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"], ins)) jesNegRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_73.7_%s-ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"], ins)) jesNegRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_86.7_%s-ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"], ins)) jesNegRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_100.0_%s-ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"], ins)) elif cut_syst: centalRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_73.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) centalRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_86.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) centalRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_100.0_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) jesPlusRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_73.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"])) jesPlusRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_86.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"])) jesPlusRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_100.0_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"])) else: centalRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_73.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) centalRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_86.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) centalRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_100.0_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) jesPlusRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_73.7_bt1.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"])) jesPlusRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_86.7_bt1.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"])) jesPlusRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_100.0_bt1.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"])) jesNegRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_73.7_bt-1.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"])) jesNegRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_86.7_bt-1.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"])) jesNegRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_100.0_bt-1.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"])) HTList = [] if settings["inclHT"]: HTList = ["incl"] else: print "f**k." exit() for htbin in HTList: processCrossSections = [] cuts = [] cutsJESNeg = [] cutsJESPlus = [] nocuts = [] suf = "" if "275_" in htbin: suf = "73" elif "325_" in htbin: suf = "86" else: suf = "100" r.gROOT.SetBatch(r.kTRUE) c1 = Print( getOutFile(model=settings["model"], htbin=htbin, format="pdf", bMulti_=bMulti, jMulti_=jMulti, mode_=mode)) c1.DoPageNum = False r.gPad.SetRightMargin(0.15) r.gPad.SetLeftMargin(0.15) r.gPad.SetTopMargin(0.08) r.gPad.SetBottomMargin(0.15) if settings["model"] == "T2_4body": rebin_y_val = 2 else: rebin_y_val = 1 nocuts = GetHist(File=centalRootFile100, folder=[ "smsScan_before", ], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val) nocuts = threeToTwo(nocuts) if settings["inclHT"]: cutsHist = GetHist(File=centalRootFile73, folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[0:2], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val).Clone() cutsHist.Add( GetHist(File=centalRootFile86, folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[2:3], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val)) cutsHist.Add( GetHist(File=centalRootFile100, folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[3:], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val)) cutsHist = threeToTwo(cutsHist) cutsJESPlusHist = GetHist(File=jesPlusRootFile73, folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[0:2], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val).Clone() cutsJESPlusHist.Add( GetHist(File=jesPlusRootFile86, folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[2:3], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val)) cutsJESPlusHist.Add( GetHist(File=jesPlusRootFile100, folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[3:], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val)) cutsJESPlusHist = threeToTwo(cutsJESPlusHist) if not cut_syst: cutsJESNegHist = GetHist(File=jesNegRootFile73, folder=getRootDirs( bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[0:2], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val).Clone() cutsJESNegHist.Add( GetHist(File=jesNegRootFile86, folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[2:3], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val)) cutsJESNegHist.Add( GetHist(File=jesNegRootFile100, folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[3:], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val)) cutsJESNegHist = threeToTwo(cutsJESNegHist) else: print "f**k." exit() if settings["deltaM"]: nocuts = deltaM(nocuts) cutsHist = deltaM(cutsHist) cutsJESPlusHist = deltaM(cutsJESPlusHist) if not cut_syst: cutsJESNegHist = deltaM(cutsJESNegHist) # zero out the above-diagonal region l = [i for i in range(1001)] for a in l: xbinval = cutsHist.GetXaxis().GetBinCenter(a) for b in l: ybinval = cutsHist.GetYaxis().GetBinCenter(b) if xbinval - ybinval < 0. or a < 0.: bin = cutsHist.FindBin(float(a), float(b)) cutsHist.SetBinContent(bin, 0.) cutsJESPlusHist.SetBinContent(bin, 0.) if not cut_syst: cutsJESNegHist.SetBinContent(bin, 0.) if mode in ["JES", "ISR"]: mini = 0.7 maxi = 1.3 elif mode in ["MHT_MET"]: mini = 0.65 maxi = 1.35 elif mode in ["DeadECAL"]: mini = 0.7 maxi = 1.3 elif mode in ["LeptonVeto"]: mini = 0.65 maxi = 1.35 else: mini = 0.96 maxi = 1.04 c1.canvas.SetLogz() offset = 1.1 # c1.Print() c1.canvas.SetLogz(False) TotalEff = cutsHist.Clone() TotalEff.GetZaxis().SetTitle("Fraction of expected signal yield") TotalEff.GetZaxis().SetTitleOffset(offset) TotalEff.GetZaxis().SetTitleSize(0.05) TotalEff.GetXaxis().SetTitle(xTitle) TotalEff.SetTitle("Total Efficiency - Nominal") TotalEff.GetYaxis().SetTitleOffset(1.3) TotalEff.GetYaxis().SetTitleSize(0.05) TotalEff.GetYaxis().SetTitle(yTitle) TotalEff.Divide(nocuts) maxVal = TotalEff.GetMaximum() #TotalEff.Scale(100.) if settings["text_plot"] and settings["model"] in [ "T2cc", "T2_4body" ]: r.gStyle.SetPaintTextFormat("0.4f") TotalEff.SetMarkerSize(0.8) TotalEff.Draw("COLZ TEXT") else: TotalEff.Draw("COLZ") tot = 0. ctr = 0 for i in range(TotalEff.GetNbinsX() * TotalEff.GetNbinsY()): val = TotalEff.GetBinContent(i) if val > 0.: tot += val ctr += 1 num0 = r.TLatex( 0.17, 0.85, "Average efficiency: %.3f%%" % (float(tot / ctr) * 100)) num0.SetNDC() num0.Draw("same") c1.Print() TotalEffPlus = cutsJESPlusHist.Clone() TotalEffPlus.GetZaxis().SetTitle("Relative change in efficiency") TotalEffPlus.GetZaxis().SetTitleOffset(offset) TotalEffPlus.SetTitle("Total Up Efficiency") TotalEffPlus.Divide(nocuts) TotalEffPlus.GetXaxis().SetTitle(xTitle) TotalEffPlus.GetYaxis().SetTitle(yTitle) TotalEffPlus.GetZaxis().SetTitleSize(0.05) TotalEffPlus.GetYaxis().SetTitleOffset(1.3) TotalEffPlus.GetYaxis().SetTitleSize(0.05) TotalEffPlus.SetMaximum(maxVal) if settings["text_plot"] and settings["model"] in [ "T2cc", "T2_4body" ]: r.gStyle.SetPaintTextFormat("0.4f") TotalEffPlus.SetMarkerSize(0.8) TotalEffPlus.Draw("COLZ TEXT") else: TotalEffPlus.Draw("COLZ") c1.Print() if not cut_syst: TotalEffNeg = cutsJESNegHist.Clone() TotalEffNeg.GetZaxis().SetTitle( "Relative change in efficiency") TotalEffNeg.GetZaxis().SetTitleOffset(offset) TotalEffNeg.SetTitle("Total Down Efficiency") TotalEffNeg.Divide(nocuts) TotalEffNeg.GetZaxis().SetTitleSize(0.05) TotalEffNeg.GetXaxis().SetTitle(xTitle) TotalEffNeg.GetYaxis().SetTitle(yTitle) TotalEffNeg.GetYaxis().SetTitleOffset(1.3) TotalEffNeg.GetYaxis().SetTitleSize(0.05) TotalEffNeg.SetMaximum(maxVal) if settings["text_plot"] and settings["model"] in [ "T2cc", "T2_4body" ]: r.gStyle.SetPaintTextFormat("0.4f") TotalEffNeg.SetMarkerSize(0.8) TotalEffNeg.Draw("COLZ TEXT") else: TotalEffNeg.Draw("COLZ") c1.Print() EffOverJESPlus = TotalEffPlus.Clone() if cut_syst: EffOverJESPlus.SetTitle("Efficiency Change - %s" % mode) else: EffOverJESPlus.SetTitle("Up deltaEff (Up / Total)") EffOverJESPlus.Divide(TotalEff) EffOverJESPlus.Draw("COLZ") EffOverJESPlus.GetZaxis().SetRangeUser(0.5, 1.5) # c1.Print() # invert the above division for cuts if mode in ["MHT_MET", "DeadECAL"]: for i in range(EffOverJESPlus.GetNbinsX() * EffOverJESPlus.GetNbinsY() + 1000): # print EffOverJESPlus.GetBinContent(i) if EffOverJESPlus.GetBinContent(i) > 0: # print 1./EffOverJESPlus.GetBinContent(i), EffOverJESPlus.GetBinContent(i) EffOverJESPlus.SetBinContent( i, 1. / EffOverJESPlus.GetBinContent(i)) if not cut_syst: EffOverJESNeg = TotalEffNeg.Clone() EffOverJESNeg.SetTitle("Down deltaEff (Down / Total)") EffOverJESNeg.Divide(TotalEff) EffOverJESNeg.Draw("COLZ") EffOverJESNeg.GetZaxis().SetRangeUser(0.5, 1.5) # c1.Print() r.gStyle.SetOptStat(0) if not cut_syst: EffOverJESNegClone = EffOverJESNeg.Clone() EffOverJESPlusClone = EffOverJESPlus.Clone() #force the zRange accross all bins if not cut_syst: for bin in range(EffOverJESNegClone.GetNbinsX() * EffOverJESNegClone.GetNbinsY() + 1000): if EffOverJESNegClone.GetBinContent(bin) > 0: if EffOverJESNegClone.GetBinContent(bin) < mini: EffOverJESNegClone.SetBinContent(bin, mini) if EffOverJESNegClone.GetBinContent(bin) > maxi: EffOverJESNegClone.SetBinContent(bin, maxi) for bin in range(EffOverJESPlusClone.GetNbinsX() * EffOverJESPlusClone.GetNbinsY() + 1000): if EffOverJESPlusClone.GetBinContent(bin) > 0: if EffOverJESPlusClone.GetBinContent(bin) < mini: EffOverJESPlusClone.SetBinContent(bin, mini) if EffOverJESPlusClone.GetBinContent(bin) > maxi: EffOverJESPlusClone.SetBinContent(bin, maxi) if mode not in ["MHT_MET", "DeadECAL"]: # centre distribution around 0., set all others to 1000 for bin in range(EffOverJESPlusClone.GetXaxis().GetNbins() * EffOverJESPlusClone.GetYaxis().GetNbins() + 10000): if not cut_syst: if EffOverJESNegClone.GetBinContent(bin) > 0: EffOverJESNegClone.SetBinContent( bin, EffOverJESNegClone.GetBinContent(bin) - 1.) else: EffOverJESNegClone.SetBinContent(bin, -1000) if EffOverJESPlusClone.GetBinContent(bin) > 0: EffOverJESPlusClone.SetBinContent( bin, EffOverJESPlusClone.GetBinContent(bin) - 1.) else: EffOverJESPlusClone.SetBinContent(bin, -1000) #force range if not cut_syst: EffOverJESNegClone.SetMinimum(mini - 1.) EffOverJESNegClone.SetMaximum(maxi - 1.) EffOverJESPlusClone.SetMinimum(mini - 1.) EffOverJESPlusClone.SetMaximum(maxi - 1.) else: EffOverJESPlusClone.SetTitle("Cut Efficiency - %s" % mode) if settings["text_plot"] and settings["model"] in [ "T2cc", "T2_4body" ]: r.gStyle.SetPaintTextFormat("0.4f") EffOverJESPlusClone.SetMarkerSize(0.8) EffOverJESPlusClone.Draw("COLZ TEXT") else: EffOverJESPlusClone.Draw("COLZ") if mode in ["MHT_MET", "DeadECAL"]: stat_vals = get_hist_stat_vals(EffOverJESPlusClone) stat_vals_string = "Avg=%.3f, Min=%.3f, Max=%.3f" % ( stat_vals[2], stat_vals[0], stat_vals[1]) print " >", stat_vals_string num = r.TLatex(0.16, 0.8, stat_vals_string) num.SetNDC() num.Draw("same") c1.Print() if not cut_syst: if settings["text_plot"] and settings["model"] in [ "T2cc", "T2_4body" ]: r.gStyle.SetPaintTextFormat("0.4f") EffOverJESNegClone.SetMarkerSize(0.8) EffOverJESNegClone.Draw("COLZ TEXT") else: EffOverJESNegClone.Draw("COLZ") c1.Print() if mode in ["MHT_MET", "DeadECAL", "3jet"]: c1.close() continue if mode in ["LeptonVeto"]: if settings["model"] in ["T2cc", "T2_4body"]: c1.close() continue # now calculate the overall systematics, point by point syst_hist = EffOverJESPlusClone.Clone() for i in range((EffOverJESPlusClone.GetNbinsX() * EffOverJESPlusClone.GetNbinsY()) + 1000): # skip null points if abs(EffOverJESPlusClone.GetBinContent(i)) == 1000.: continue if not cut_syst: neg_syst = abs(EffOverJESNegClone.GetBinContent(i)) pos_syst = abs(EffOverJESPlusClone.GetBinContent(i)) if cut_syst: this_syst = pos_syst else: if pos_syst >= neg_syst: this_syst = pos_syst else: this_syst = neg_syst # pick out the largest systematic - most conservative approach syst_hist.SetBinContent(i, this_syst) syst_hist.SetTitle("Systematics") syst_hist.SetMinimum(0.) if settings["text_plot"] and settings["model"] in [ "T2cc", "T2_4body" ]: r.gStyle.SetPaintTextFormat("0.4f") syst_hist.SetMarkerSize(0.8) syst_hist.Draw("COLZ TEXT") else: syst_hist.Draw("COLZ") stat_vals = get_hist_stat_vals(EffOverJESPlusClone) stat_vals_string = "Avg=%.3f, Min=%.3f, Max=%.3f" % ( stat_vals[2], stat_vals[0], stat_vals[1]) print " >", stat_vals_string num = r.TLatex(0.16, 0.8, stat_vals_string) num.SetNDC() num.Draw("same") c1.Print() c1.close() my_syst_hists.append(syst_hist.Clone()) if len(my_syst_hists) == 0: return if settings["model"] == "T2cc": # open Yossof's pdf systematics file pdf_file = r.TFile.Open("envCvRelHist.root", 'READ') pdf_hist = pdf_file.Get("acc_cvRel_m0_m12") my_syst_hists.append(pdf_hist) total_syst_hist = r.TH2D( "my_th2", "my_th2", my_syst_hists[0].GetNbinsX(), my_syst_hists[0].GetXaxis().GetBinLowEdge(1), my_syst_hists[0].GetXaxis().GetBinUpEdge(my_syst_hists[0].GetNbinsX()), my_syst_hists[0].GetNbinsY(), my_syst_hists[0].GetYaxis().GetBinLowEdge(1), my_syst_hists[0].GetYaxis().GetBinUpEdge(my_syst_hists[0].GetNbinsY())) # sum all point systs in quadrature and add to final histogram min_val = 1. max_val = 0. avg_val = 0. count = 0 # loop over specific mass points and find corresponding bins, as pdf plot is different binning scheme stop_masses = [] split_masses = [] if settings["model"] == "T2cc": stop_masses = [100. + 25. * i for i in range(11)] split_masses = [5., 10., 20., 30., 40., 60., 80.] elif settings["model"] == "T2_4body": stop_masses = [100. + 25. * i for i in range(12)] split_masses = [10. * i for i in range(1, 9)] else: for x_bin in range(1, my_syst_hists[0].GetNbinsX() + 1): for y_bin in range(1, my_syst_hists[0].GetNbinsY() + 1): if my_syst_hists[0].GetBinContent(x_bin, y_bin) > 0.: stop_mass = my_syst_hists[0].GetXaxis().GetBinLowEdge( x_bin) if stop_mass not in stop_masses: stop_masses.append(stop_mass) split_mass = my_syst_hists[0].GetXaxis().GetBinLowEdge( x_bin) - my_syst_hists[0].GetYaxis().GetBinLowEdge( y_bin) if split_mass not in split_masses: split_masses.append(split_mass) print stop_masses print split_masses for mstop in stop_masses: for mdiff in split_masses: mlsp = mstop - mdiff val = 0. this_bin = my_syst_hists[0].FindBin(mstop, mlsp) if my_syst_hists[0].GetBinContent(this_bin) <= 0.: print "Zero val bin in syst hist!" continue for s_hist in my_syst_hists: if "cvRel" in s_hist.GetName(): this_yossof_bin = s_hist.FindBin(mstop, mlsp) hist_val = abs(s_hist.GetBinContent(this_yossof_bin) - 1.) else: hist_val = s_hist.GetBinContent(this_bin) val += hist_val * hist_val val = math.sqrt(val) if val > 0: count += 1 avg_val += val if val > max_val: max_val = val if val < min_val: min_val = val total_syst_hist.SetBinContent(this_bin, val) if min_val == 1.: min_val = 0. out_name = getOutFile(model=settings["model"], htbin=htbin, format="pdf", bMulti_=bMulti, jMulti_=jMulti, mode_="total") c_total = r.TCanvas() r.gPad.SetRightMargin(0.15) total_syst_hist.SetTitle("Total Systematic") total_syst_hist.SetMaximum(0.35) if settings["text_plot"] and settings["model"] in ["T2cc", "T2_4body"]: r.gStyle.SetPaintTextFormat("0.4f") total_syst_hist.SetMarkerSize(0.8) total_syst_hist.Draw("COLZ TEXT") else: total_syst_hist.Draw("COLZ") try: avg_val /= float(count) except ZeroDivisionError: avg_val = 0. num = r.TLatex( 0.15, 0.8, "Avg=%.3f, Min=%.3f, Max=%.3f" % (avg_val, min_val, max_val)) num.SetNDC() num.Draw("same") print "\t>> Total: Avg=%.3f, Min=%.3f, Max=%.3f" % (avg_val, min_val, max_val) c_total.Print(out_name) # write the output rootfile out_root_file = r.TFile(out_name.replace(".pdf", ".root"), "RECREATE") total_syst_hist.SetName( out_name.replace(".pdf", "").replace("_v%d" % settings["version"], "").split("/")[-1]) total_syst_hist.Write() out_root_file.Close() if settings["model"] == "T2cc": pdf_file.Close() del out_root_file
def make_syst_map(central_files=[], positive_files=[], negative_files=[], bMulti="", jMulti="", syst_mode=""): xTitle = "m_{stop} (GeV)" if settings["deltaM"]: yTitle = "deltaM (GeV)" else: yTitle = "m_{LSP} (GeV)" for mode in settings["mode"]: # need to find a better way to do this! if mode != "bTag": centalRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_73.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) centalRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_86.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) centalRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_100.0_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) jesPlusRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_73.7_%s+ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"], ins)) jesPlusRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_86.7_%s+ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"], ins)) jesPlusRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_100.0_%s+ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"], ins)) jesNegRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_73.7_%s-ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"], ins)) jesNegRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_86.7_%s-ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"], ins)) jesNegRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_100.0_%s-ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"], ins)) else: centalRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_73.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) centalRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_86.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) centalRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/sigScan_%s_had_2012_100.0_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) jesPlusRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_73.7_bt1.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"])) jesPlusRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_86.7_bt1.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"])) jesPlusRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_100.0_bt1.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"])) jesNegRootFile73 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_73.7_bt-1.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"])) jesNegRootFile86 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_86.7_bt-1.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"])) jesNegRootFile100 = r.TFile.Open( "./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_100.0_bt-1.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["mode"], settings["model"])) if mode == "ISR": ins = "isr_" else: ins = "" HTList = [] if settings["inclHT"]: HTList = ["incl"] else: HTList = settings["HTBins"] oF = open( getOutFile(model=settings["model"], format="txt", bMulti_=bMulti, jMulti_=jMulti), 'w') for htbin in HTList: processCrossSections = [] cuts = [] cutsJESNeg = [] cutsJESPlus = [] nocuts = [] suf = "" if "275_" in htbin: suf = "73" elif "325_" in htbin: suf = "86" else: suf = "100" r.gROOT.SetBatch(r.kTRUE) c1 = Print( getOutFile(model=settings["model"], htbin=htbin, format="pdf")) c1.DoPageNum = False r.gPad.SetRightMargin(0.175) r.gPad.SetLeftMargin(0.15) r.gPad.SetTopMargin(0.08) r.gPad.SetBottomMargin(0.15) nocuts = GetHist(File=centalRootFile100, folder=[ "smsScan_before", ], hist="m0_m12_mChi_weight", Norm=None, rebinX=4) nocuts = threeToTwo(nocuts) if settings["inclHT"]: cutsHist = GetHist(File=central_files[0], folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[0:2], hist="m0_m12_mChi_weight", Norm=None).Clone() cutsHist.Add( GetHist(File=central_files[1], folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[2:3], hist="m0_m12_mChi_weight", Norm=None)) cutsHist.Add( GetHist(File=central_files[2], folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[3:], hist="m0_m12_mChi_weight", Norm=None)) cutsJESPlusHist = GetHist(File=positive_files[0], folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[0:2], hist="m0_m12_mChi_weight", Norm=None).Clone() cutsJESPlusHist.Add( GetHist(File=positive_files[1], folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[2:3], hist="m0_m12_mChi_weight", Norm=None)) cutsJESPlusHist.Add( GetHist(File=positive_files[2], folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[3:], hist="m0_m12_mChi_weight", Norm=None)) cutsJESNegHist = GetHist(File=negative_files[0], folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[0:2], hist="m0_m12_mChi_weight", Norm=None).Clone() cutsJESNegHist.Add( GetHist(File=negative_files[1], folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[2:3], hist="m0_m12_mChi_weight", Norm=None)) cutsJESNegHist.Add( GetHist(File=negative_files[2], folder=getRootDirs(bMulti_=bMulti, jMulti_=jMulti, sitv_=True)[3:], hist="m0_m12_mChi_weight", Norm=None)) else: print "f**k." exit() # convert to TH2 plots cutsHist = threeToTwo(cutsHist) cutsJESPlusHist = threeToTwo(cutsJESPlusHist) cutsJESNegHist = threeToTwo(cutsJESNegHist) if settings["deltaM"]: nocuts = deltaM(nocuts) cutsHist = deltaM(cutsHist) cutsJESPlusHist = deltaM(cutsJESPlusHist) cutsJESNegHist = deltaM(cutsJESNegHist) # zero out the above-diagonal region l = [i for i in range(301)] for a in l: xbinval = cutsHist.GetXaxis().GetBinCenter(a) for b in l: ybinval = cutsHist.GetYaxis().GetBinCenter(b) if xbinval - ybinval < 0. or a < 0.: bin = cutsHist.FindBin(float(a), float(b)) cutsHist.SetBinContent(bin, 0.) cutsJESPlusHist.SetBinContent(bin, 0.) cutsJESNegHist.SetBinContent(bin, 0.) if mode == "JES": mini = 0.9 maxi = 1.1 else: mini = 0.70 maxi = 1.30 c1.canvas.SetLogz() offset = 1.1 # c1.Print() c1.canvas.SetLogz(False) TotalEff = cutsHist.Clone() TotalEff.GetZaxis().SetTitle("Fraction of expected signal yield") TotalEff.GetZaxis().SetTitleOffset(offset) TotalEff.GetZaxis().SetTitleSize(0.05) TotalEff.GetXaxis().SetTitle(xTitle) TotalEff.SetTitle("Total Efficiency") TotalEff.GetYaxis().SetTitleOffset(1.3) TotalEff.GetYaxis().SetTitleSize(0.05) TotalEff.GetYaxis().SetTitle(yTitle) TotalEff.Divide(nocuts) maxVal = TotalEff.GetMaximum() #TotalEff.Scale(100.) #r.gStyle.SetPaintTextFormat("0.2f %%"); #TotalEff.SetMarkerSize(1.4) #r.gStyle.SetPalette(3) #TotalEff.Draw("COLZ TEXT40") TotalEff.Draw("COLZ") tot = 0. ctr = 0 for i in range(TotalEff.GetNbinsX() * TotalEff.GetNbinsY()): val = TotalEff.GetBinContent(i) if val > 0.: tot += val ctr += 1 num0 = r.TLatex( 0.17, 0.85, "Average efficiency: %.3f%%" % (float(tot / ctr) * 100)) num0.SetNDC() num0.Draw("same") c1.Print() TotalEffPlus = cutsJESPlusHist.Clone() TotalEffPlus.GetZaxis().SetTitle("Relative change in efficiency") TotalEffPlus.GetZaxis().SetTitleOffset(offset) TotalEffPlus.SetTitle("Total Up Efficiency") TotalEffPlus.Divide(nocuts) TotalEffPlus.GetXaxis().SetTitle(xTitle) TotalEffPlus.GetYaxis().SetTitle(yTitle) TotalEffPlus.GetZaxis().SetTitleSize(0.05) TotalEffPlus.GetYaxis().SetTitleOffset(1.3) TotalEffPlus.GetYaxis().SetTitleSize(0.05) TotalEffPlus.SetMaximum(maxVal) TotalEffPlus.Draw("COLZ") c1.Print() TotalEffNeg = cutsJESNegHist.Clone() TotalEffNeg.GetZaxis().SetTitle("Relative change in efficiency") TotalEffNeg.GetZaxis().SetTitleOffset(offset) TotalEffNeg.SetTitle("Total Down Efficiency") TotalEffNeg.Divide(nocuts) TotalEffNeg.GetZaxis().SetTitleSize(0.05) TotalEffNeg.GetXaxis().SetTitle(xTitle) TotalEffNeg.GetYaxis().SetTitle(yTitle) TotalEffNeg.GetYaxis().SetTitleOffset(1.3) TotalEffNeg.GetYaxis().SetTitleSize(0.05) TotalEffNeg.SetMaximum(maxVal) TotalEffNeg.Draw("COLZ") c1.Print() EffOverJESPlus = TotalEffPlus.Clone() EffOverJESPlus.SetTitle("Up deltaEff (Up / Total)") EffOverJESPlus.Divide(TotalEff) EffOverJESPlus.Draw("COLZ") EffOverJESPlus.GetZaxis().SetRangeUser(0.5, 1.5) # c1.Print() EffOverJESNeg = TotalEffNeg.Clone() EffOverJESNeg.SetTitle("Down deltaEff (Down / Total)") EffOverJESNeg.Divide(TotalEff) EffOverJESNeg.Draw("COLZ") EffOverJESNeg.GetZaxis().SetRangeUser(0.5, 1.5) # c1.Print() r.gStyle.SetOptStat(0) EffOverJESNegClone = EffOverJESNeg.Clone() EffOverJESPlusClone = EffOverJESPlus.Clone() #force the zRange accross all bins for bin in range(EffOverJESNegClone.GetNbinsX() * EffOverJESNegClone.GetNbinsY()): if EffOverJESNegClone.GetBinContent(bin) > 0: if EffOverJESNegClone.GetBinContent(bin) < mini: EffOverJESNegClone.SetBinContent(bin, mini) if EffOverJESNegClone.GetBinContent(bin) > maxi: EffOverJESNegClone.SetBinContent(bin, maxi) for bin in range(EffOverJESPlusClone.GetNbinsX() * EffOverJESPlusClone.GetNbinsY()): if EffOverJESPlusClone.GetBinContent(bin) > 0: if EffOverJESPlusClone.GetBinContent(bin) < mini: EffOverJESPlusClone.SetBinContent(bin, mini) if EffOverJESPlusClone.GetBinContent(bin) > maxi: EffOverJESPlusClone.SetBinContent(bin, maxi) # centre distribution around 0., set all others to 1000 for bin in range(EffOverJESPlusClone.GetXaxis().GetNbins() * EffOverJESPlusClone.GetYaxis().GetNbins() + 10000): if EffOverJESNegClone.GetBinContent(bin) > 0: EffOverJESNegClone.SetBinContent( bin, EffOverJESNegClone.GetBinContent(bin) - 1.) else: EffOverJESNegClone.SetBinContent(bin, -1000) if EffOverJESPlusClone.GetBinContent(bin) > 0: EffOverJESPlusClone.SetBinContent( bin, EffOverJESPlusClone.GetBinContent(bin) - 1.) else: EffOverJESPlusClone.SetBinContent(bin, -1000) #force range EffOverJESNegClone.SetMinimum(mini - 1.) EffOverJESNegClone.SetMaximum(maxi - 1.) EffOverJESPlusClone.SetMinimum(mini - 1.) EffOverJESPlusClone.SetMaximum(maxi - 1.) EffOverJESPlusClone.Draw("COLZ") c1.Print() EffOverJESNegClone.Draw("COLZ") c1.Print() if not settings["inclHT"]: oF.write("%s\t\t%.3f +\\- %.3f\n" % (htbin, JesTotClone.GetBinLowEdge(bin68), JesTotClone.GetBinError(bin68))) c1.close() if not settings["inclHT"]: oF.close()
if h is None: h = a.Clone() else: h.Add(a) return h def nloTotalXsecMaker(individualXSecs=None): out = None for h in individualXSecs: if out is None: out = h.Clone() else: out.Add(h) return out c1 = Print("JESSignalSMST1_250.pdf") c1.open() r.gPad.SetRightMargin(0.15) centalRootFile73 = r.TFile.Open("./rootFiles/results_had_T1_73.7_.root") centalRootFile86 = r.TFile.Open("./rootFiles/results_had_T1_86.7_.root") centalRootFile100 = r.TFile.Open("./rootFiles/results_had_T1_100_.root") jesPlusRootFile73 = r.TFile.Open("./rootFiles/results_had_T1_73.7_+ve.root") jesPlusRootFile86 = r.TFile.Open("./rootFiles/results_had_T1_86.7_+ve.root") jesPlusRootFile100 = r.TFile.Open("./rootFiles/results_had_T1_100_+ve.root") jesNegRootFile73 = r.TFile.Open("./rootFiles/results_had_T1_73.7_-ve.root") jesNegRootFile86 = r.TFile.Open("./rootFiles/results_had_T1_86.7_-ve.root") jesNegRootFile100 = r.TFile.Open("./rootFiles/results_had_T1_100_-ve.root") # jesRanRootFile73 = r.TFile.Open("./results_had_T1_73.7_ran.root") # jesRanRootFile86 = r.TFile.Open("./results_had_T1_86.7_ran.root") # jesRanRootFile100 = r.TFile.Open("./results_had_T1_100_ran.root")
def main(): """docstring for main""" c1 = Print("MHToverMET_data_to_mc.pdf") c1.DoPageNum = False Data = GetSumHist( File = ["rootFiles/MuonData.root",], Directories = ['NoMHTovMET',], Hist = "MHTovMET_all", Col = r.kBlack, Norm = None, LegendText = "Data") Data.HideOverFlow() # Data.hObj.Rebin(5) Mc = GetSumHist( File = ["rootFiles/MuonNoSmear.root",], Directories = ['NoMHTovMET',], Hist = "MHTovMET_all", Col = r.kRed, Norm =None, LegendText = "MC") Mc.HideOverFlow() print "MC Total integral = %f"%(Mc.hObj.Integral()) Data.hObj.Scale(1./Data.hObj.Integral()) Mc.hObj.Scale(1./Mc.hObj.Integral()) Data.hObj.Rebin(25) Mc.hObj.Rebin(25) Data.hObj.GetXaxis().SetRangeUser(0.,5.) # Mc.hObj.Rebin(5) DataCumu = r.TH1D(Data.hObj) McCumu = r.TH1D(Mc.hObj) DataCumu = MakeCumu(DataCumu) McCumu = MakeCumu(McCumu) # Mc.hObj.Scale(4.95) Data.Draw("h") Mc.Draw("sameh") leg = Legend(x1 = 0.6 , x2 = 0.8, y1 = 0.5 , y2 = 0.8 ) leg.AddEntry(Data.hObj,Data.legendText,"lp") leg.AddEntry(Mc.hObj,Mc.legendText,"lp") leg.Draw("") c1.Print() DataCumu.Draw("h") McCumu.Draw("sameh") leg = Legend(x1 = 0.6 , x2 = 0.8, y1 = 0.5 , y2 = 0.8 ) leg.AddEntry(Data.hObj,Data.legendText,"lp") leg.AddEntry(Mc.hObj,Mc.legendText,"lp") leg.Draw("") c1.Print() ratio = r.TH1D(Data.hObj) for bin in range(ratio.GetNbinsX()): if Mc.hObj.GetBinContent(bin) > 0 :print "BYHAND Bin lower edge = %f, value = %f"%(ratio.GetBinLowEdge(bin),Data.hObj.GetBinContent(bin)/Mc.hObj.GetBinContent(bin)) ratio.Divide(Mc.hObj) ratio.Draw("h") for bin in range(ratio.GetNbinsX()): print "Bin lower edge = %f, value = %f pm %f"%(ratio.GetBinLowEdge(bin),ratio.GetBinContent(bin),ratio.GetBinError(bin)) ratio.GetYaxis().SetRangeUser(0.,3.) ratio.GetYaxis().SetRangeUser(0.75,1.25) c1.Print() ratio = r.TH1D(DataCumu) ratio.Divide(McCumu) ratio.Draw("h") for bin in range(ratio.GetNbinsX()): print "Cumulative Bin lower edge = %f, value = %f pm %f"%(ratio.GetBinLowEdge(bin),ratio.GetBinContent(bin),ratio.GetBinError(bin)) ratio.GetYaxis().SetRangeUser(0.,3.) ratio.GetYaxis().SetRangeUser(0.75,1.25) c1.Print() c1.close() pass
# print directory.ls() a = directory.Get(hist) if h is None: h = a.Clone() else: h.Add(a) return h def nloTotalXsecMaker(individualXSecs = None): out = None for h in individualXSecs: if out is None: out = h.Clone() else: out.Add(h) return out c1 = Print("JESSignalSMST1_250.pdf") c1.open() r.gPad.SetRightMargin(0.15) centalRootFile73 = r.TFile.Open("./rootFiles/results_had_T1_73.7_.root") centalRootFile86 = r.TFile.Open("./rootFiles/results_had_T1_86.7_.root") centalRootFile100 = r.TFile.Open("./rootFiles/results_had_T1_100_.root") jesPlusRootFile73 = r.TFile.Open("./rootFiles/results_had_T1_73.7_+ve.root") jesPlusRootFile86 = r.TFile.Open("./rootFiles/results_had_T1_86.7_+ve.root") jesPlusRootFile100 = r.TFile.Open("./rootFiles/results_had_T1_100_+ve.root") jesNegRootFile73 = r.TFile.Open("./rootFiles/results_had_T1_73.7_-ve.root") jesNegRootFile86 = r.TFile.Open("./rootFiles/results_had_T1_86.7_-ve.root") jesNegRootFile100 = r.TFile.Open("./rootFiles/results_had_T1_100_-ve.root") # jesRanRootFile73 = r.TFile.Open("./results_had_T1_73.7_ran.root") # jesRanRootFile86 = r.TFile.Open("./results_had_T1_86.7_ran.root") # jesRanRootFile100 = r.TFile.Open("./results_had_T1_100_ran.root")
def make_syst_map_three(bMulti = "", jMulti = ""): set_palette() print ">>> Creating systs for", bMulti, jMulti xTitle = "m_{stop} (GeV)" if settings["deltaM"]: yTitle = "deltaM (GeV)" else: yTitle = "m_{LSP} (GeV)" total_syst_hist = r.TH2D() my_syst_hists = [] for n_syst, mode in enumerate(settings["mode"]): if "T2cc" not in settings['model'] and mode == "3jet": continue print " >>", mode if mode == "ISR": ins = "isr_" else: ins = "" cut_syst = False # set variable for a before/after type systematic if mode in ["LeptonVeto", "MHT_MET", "DeadECAL", "3jet"]: cut_syst = True # need to find a better way to do this! if mode in ["JES", "ISR"]: centalRootFile73 = r.TFile.Open("./rootFiles/%s_v%d/sigScan_%s_had_2012_73.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) centalRootFile86 = r.TFile.Open("./rootFiles/%s_v%d/sigScan_%s_had_2012_86.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) centalRootFile100 = r.TFile.Open("./rootFiles/%s_v%d/sigScan_%s_had_2012_100.0_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) jesPlusRootFile73 = r.TFile.Open("./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_73.7_%s+ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"], ins)) jesPlusRootFile86 = r.TFile.Open("./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_86.7_%s+ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"], ins)) jesPlusRootFile100 = r.TFile.Open("./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_100.0_%s+ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"], ins)) jesNegRootFile73 = r.TFile.Open("./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_73.7_%s-ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"], ins)) jesNegRootFile86 = r.TFile.Open("./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_86.7_%s-ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"], ins)) jesNegRootFile100 = r.TFile.Open("./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_100.0_%s-ve_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"], ins)) elif cut_syst: centalRootFile73 = r.TFile.Open("./rootFiles/%s_v%d/sigScan_%s_had_2012_73.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) centalRootFile86 = r.TFile.Open("./rootFiles/%s_v%d/sigScan_%s_had_2012_86.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) centalRootFile100 = r.TFile.Open("./rootFiles/%s_v%d/sigScan_%s_had_2012_100.0_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) jesPlusRootFile73 = r.TFile.Open("./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_73.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"])) jesPlusRootFile86 = r.TFile.Open("./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_86.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"])) jesPlusRootFile100 = r.TFile.Open("./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_100.0_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"])) else: centalRootFile73 = r.TFile.Open("./rootFiles/%s_v%d/sigScan_%s_had_2012_73.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) centalRootFile86 = r.TFile.Open("./rootFiles/%s_v%d/sigScan_%s_had_2012_86.7_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) centalRootFile100 = r.TFile.Open("./rootFiles/%s_v%d/sigScan_%s_had_2012_100.0_bt0.0_MChi-1.0.root" % (settings["model"], settings["version"], settings["model"])) jesPlusRootFile73 = r.TFile.Open("./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_73.7_bt1.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"])) jesPlusRootFile86 = r.TFile.Open("./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_86.7_bt1.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"])) jesPlusRootFile100 = r.TFile.Open("./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_100.0_bt1.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"])) jesNegRootFile73 = r.TFile.Open("./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_73.7_bt-1.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"])) jesNegRootFile86 = r.TFile.Open("./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_86.7_bt-1.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"])) jesNegRootFile100 = r.TFile.Open("./rootFiles/%s_v%d/%s/sigScan_%s_had_2012_100.0_bt-1.0_MChi-1.0.root" % (settings["model"], settings["version"], mode, settings["model"])) HTList = [] if settings["inclHT"]: HTList = ["incl"] else: print "f**k." exit() for htbin in HTList: processCrossSections = [] cuts = [] cutsJESNeg = [] cutsJESPlus = [] nocuts = [] suf = "" if "275_" in htbin: suf = "73" elif "325_" in htbin: suf = "86" else: suf = "100" r.gROOT.SetBatch(r.kTRUE) c1 = Print(getOutFile(model=settings["model"], htbin=htbin, format="pdf", bMulti_ = bMulti, jMulti_ = jMulti, mode_ = mode)) c1.DoPageNum = False r.gPad.SetRightMargin(0.15) r.gPad.SetLeftMargin(0.15) r.gPad.SetTopMargin(0.08) r.gPad.SetBottomMargin(0.15) if settings["model"] == "T2_4body": rebin_y_val = 2 else: rebin_y_val = 1 nocuts = GetHist(File=centalRootFile100, folder=["smsScan_before", ], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val) nocuts = threeToTwo(nocuts) if settings["inclHT"]: cutsHist = GetHist(File=centalRootFile73, folder=getRootDirs(bMulti_ = bMulti, jMulti_ = jMulti, sitv_ = True)[0:2], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val).Clone() cutsHist.Add(GetHist(File=centalRootFile86, folder=getRootDirs(bMulti_ = bMulti, jMulti_ = jMulti, sitv_ = True)[2:3], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val)) cutsHist.Add(GetHist(File=centalRootFile100, folder=getRootDirs(bMulti_ = bMulti, jMulti_ = jMulti, sitv_ = True)[3:], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val)) cutsHist = threeToTwo(cutsHist) cutsJESPlusHist = GetHist(File=jesPlusRootFile73, folder=getRootDirs(bMulti_ = bMulti, jMulti_ = jMulti, sitv_ = True)[0:2], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val).Clone() cutsJESPlusHist.Add(GetHist(File=jesPlusRootFile86, folder=getRootDirs(bMulti_ = bMulti, jMulti_ = jMulti, sitv_ = True)[2:3], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val)) cutsJESPlusHist.Add(GetHist(File=jesPlusRootFile100, folder=getRootDirs(bMulti_ = bMulti, jMulti_ = jMulti, sitv_ = True)[3:], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val)) cutsJESPlusHist = threeToTwo(cutsJESPlusHist) if not cut_syst: cutsJESNegHist = GetHist(File=jesNegRootFile73, folder=getRootDirs(bMulti_ = bMulti, jMulti_ = jMulti, sitv_ = True)[0:2], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val).Clone() cutsJESNegHist.Add(GetHist(File=jesNegRootFile86, folder=getRootDirs(bMulti_ = bMulti, jMulti_ = jMulti, sitv_ = True)[2:3], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val)) cutsJESNegHist.Add(GetHist(File=jesNegRootFile100, folder=getRootDirs(bMulti_ = bMulti, jMulti_ = jMulti, sitv_ = True)[3:], hist="m0_m12_mChi_weight", Norm=None, rebinY=rebin_y_val)) cutsJESNegHist = threeToTwo(cutsJESNegHist) else: print "f**k." exit() if settings["deltaM"]: nocuts = deltaM(nocuts) cutsHist = deltaM(cutsHist) cutsJESPlusHist = deltaM(cutsJESPlusHist) if not cut_syst: cutsJESNegHist = deltaM(cutsJESNegHist) # zero out the above-diagonal region l = [i for i in range(1001)] for a in l: xbinval = cutsHist.GetXaxis().GetBinCenter(a) for b in l: ybinval = cutsHist.GetYaxis().GetBinCenter(b) if xbinval - ybinval < 0. or a < 0.: bin = cutsHist.FindBin(float(a), float(b)) cutsHist.SetBinContent(bin, 0.) cutsJESPlusHist.SetBinContent(bin, 0.) if not cut_syst: cutsJESNegHist.SetBinContent(bin, 0.) if mode in ["JES", "ISR"]: mini = 0.7 maxi = 1.3 elif mode in ["MHT_MET"]: mini = 0.65 maxi = 1.35 elif mode in ["DeadECAL"]: mini = 0.7 maxi = 1.3 elif mode in ["LeptonVeto"]: mini = 0.65 maxi = 1.35 else: mini = 0.96 maxi = 1.04 c1.canvas.SetLogz() offset = 1.1 # c1.Print() c1.canvas.SetLogz(False) TotalEff = cutsHist.Clone() TotalEff.GetZaxis().SetTitle("Fraction of expected signal yield") TotalEff.GetZaxis().SetTitleOffset(offset) TotalEff.GetZaxis().SetTitleSize(0.05) TotalEff.GetXaxis().SetTitle(xTitle) TotalEff.SetTitle("Total Efficiency - Nominal") TotalEff.GetYaxis().SetTitleOffset(1.3) TotalEff.GetYaxis().SetTitleSize(0.05) TotalEff.GetYaxis().SetTitle(yTitle) TotalEff.Divide(nocuts) maxVal = TotalEff.GetMaximum() #TotalEff.Scale(100.) if settings["text_plot"] and settings["model"] in ["T2cc", "T2_4body"] : r.gStyle.SetPaintTextFormat("0.4f"); TotalEff.SetMarkerSize(0.8) TotalEff.Draw("COLZ TEXT") else: TotalEff.Draw("COLZ") tot = 0. ctr = 0 for i in range(TotalEff.GetNbinsX()*TotalEff.GetNbinsY()): val = TotalEff.GetBinContent(i) if val > 0.: tot += val ctr += 1 num0 = r.TLatex(0.17, 0.85, "Average efficiency: %.3f%%" % (float(tot/ctr)*100)) num0.SetNDC() num0.Draw("same") c1.Print() TotalEffPlus = cutsJESPlusHist.Clone() TotalEffPlus.GetZaxis().SetTitle("Relative change in efficiency") TotalEffPlus.GetZaxis().SetTitleOffset(offset) TotalEffPlus.SetTitle("Total Up Efficiency") TotalEffPlus.Divide(nocuts) TotalEffPlus.GetXaxis().SetTitle(xTitle) TotalEffPlus.GetYaxis().SetTitle(yTitle) TotalEffPlus.GetZaxis().SetTitleSize(0.05) TotalEffPlus.GetYaxis().SetTitleOffset(1.3) TotalEffPlus.GetYaxis().SetTitleSize(0.05) TotalEffPlus.SetMaximum(maxVal) if settings["text_plot"] and settings["model"] in ["T2cc", "T2_4body"] : r.gStyle.SetPaintTextFormat("0.4f"); TotalEffPlus.SetMarkerSize(0.8) TotalEffPlus.Draw("COLZ TEXT") else: TotalEffPlus.Draw("COLZ") c1.Print() if not cut_syst: TotalEffNeg = cutsJESNegHist.Clone() TotalEffNeg.GetZaxis().SetTitle("Relative change in efficiency") TotalEffNeg.GetZaxis().SetTitleOffset(offset) TotalEffNeg.SetTitle("Total Down Efficiency") TotalEffNeg.Divide(nocuts) TotalEffNeg.GetZaxis().SetTitleSize(0.05) TotalEffNeg.GetXaxis().SetTitle(xTitle) TotalEffNeg.GetYaxis().SetTitle(yTitle) TotalEffNeg.GetYaxis().SetTitleOffset(1.3) TotalEffNeg.GetYaxis().SetTitleSize(0.05) TotalEffNeg.SetMaximum(maxVal) if settings["text_plot"] and settings["model"] in ["T2cc", "T2_4body"] : r.gStyle.SetPaintTextFormat("0.4f"); TotalEffNeg.SetMarkerSize(0.8) TotalEffNeg.Draw("COLZ TEXT") else: TotalEffNeg.Draw("COLZ") c1.Print() EffOverJESPlus = TotalEffPlus.Clone() if cut_syst: EffOverJESPlus.SetTitle("Efficiency Change - %s" % mode) else: EffOverJESPlus.SetTitle("Up deltaEff (Up / Total)") EffOverJESPlus.Divide(TotalEff) EffOverJESPlus.Draw("COLZ") EffOverJESPlus.GetZaxis().SetRangeUser(0.5, 1.5) # c1.Print() # invert the above division for cuts if mode in ["MHT_MET", "DeadECAL"]: for i in range(EffOverJESPlus.GetNbinsX()*EffOverJESPlus.GetNbinsY()+1000): # print EffOverJESPlus.GetBinContent(i) if EffOverJESPlus.GetBinContent(i) > 0: # print 1./EffOverJESPlus.GetBinContent(i), EffOverJESPlus.GetBinContent(i) EffOverJESPlus.SetBinContent(i, 1./EffOverJESPlus.GetBinContent(i)) if not cut_syst: EffOverJESNeg = TotalEffNeg.Clone() EffOverJESNeg.SetTitle("Down deltaEff (Down / Total)") EffOverJESNeg.Divide(TotalEff) EffOverJESNeg.Draw("COLZ") EffOverJESNeg.GetZaxis().SetRangeUser(0.5, 1.5) # c1.Print() r.gStyle.SetOptStat(0) if not cut_syst: EffOverJESNegClone = EffOverJESNeg.Clone() EffOverJESPlusClone = EffOverJESPlus.Clone() #force the zRange accross all bins if not cut_syst: for bin in range(EffOverJESNegClone.GetNbinsX()*EffOverJESNegClone.GetNbinsY()+1000): if EffOverJESNegClone.GetBinContent(bin) > 0: if EffOverJESNegClone.GetBinContent(bin) < mini: EffOverJESNegClone.SetBinContent(bin, mini) if EffOverJESNegClone.GetBinContent(bin) > maxi: EffOverJESNegClone.SetBinContent(bin, maxi) for bin in range(EffOverJESPlusClone.GetNbinsX()*EffOverJESPlusClone.GetNbinsY()+1000): if EffOverJESPlusClone.GetBinContent(bin) > 0: if EffOverJESPlusClone.GetBinContent(bin) < mini: EffOverJESPlusClone.SetBinContent(bin, mini) if EffOverJESPlusClone.GetBinContent(bin) > maxi: EffOverJESPlusClone.SetBinContent(bin, maxi) if mode not in ["MHT_MET", "DeadECAL"]: # centre distribution around 0., set all others to 1000 for bin in range(EffOverJESPlusClone.GetXaxis().GetNbins()*EffOverJESPlusClone.GetYaxis().GetNbins()+10000): if not cut_syst: if EffOverJESNegClone.GetBinContent(bin) > 0: EffOverJESNegClone.SetBinContent(bin, EffOverJESNegClone.GetBinContent(bin)-1.) else: EffOverJESNegClone.SetBinContent(bin, -1000) if EffOverJESPlusClone.GetBinContent(bin) > 0: EffOverJESPlusClone.SetBinContent(bin, EffOverJESPlusClone.GetBinContent(bin)-1.) else: EffOverJESPlusClone.SetBinContent(bin, -1000) #force range if not cut_syst: EffOverJESNegClone.SetMinimum(mini-1.) EffOverJESNegClone.SetMaximum(maxi-1.) EffOverJESPlusClone.SetMinimum(mini-1.) EffOverJESPlusClone.SetMaximum(maxi-1.) else: EffOverJESPlusClone.SetTitle("Cut Efficiency - %s" % mode) if settings["text_plot"] and settings["model"] in ["T2cc", "T2_4body"] : r.gStyle.SetPaintTextFormat("0.4f"); EffOverJESPlusClone.SetMarkerSize(0.8) EffOverJESPlusClone.Draw("COLZ TEXT") else: EffOverJESPlusClone.Draw("COLZ") if mode in ["MHT_MET", "DeadECAL"]: stat_vals = get_hist_stat_vals(EffOverJESPlusClone) stat_vals_string = "Avg=%.3f, Min=%.3f, Max=%.3f"%(stat_vals[2], stat_vals[0], stat_vals[1]) print " >", stat_vals_string num = r.TLatex(0.16,0.8,stat_vals_string) num.SetNDC() num.Draw("same") c1.Print() if not cut_syst: if settings["text_plot"] and settings["model"] in ["T2cc", "T2_4body"] : r.gStyle.SetPaintTextFormat("0.4f"); EffOverJESNegClone.SetMarkerSize(0.8) EffOverJESNegClone.Draw("COLZ TEXT") else: EffOverJESNegClone.Draw("COLZ") c1.Print() if mode in ["MHT_MET", "DeadECAL", "3jet"]: c1.close() continue if mode in ["LeptonVeto"]: if settings["model"] in ["T2cc", "T2_4body"]: c1.close() continue # now calculate the overall systematics, point by point syst_hist = EffOverJESPlusClone.Clone() for i in range((EffOverJESPlusClone.GetNbinsX() * EffOverJESPlusClone.GetNbinsY()) + 1000): # skip null points if abs(EffOverJESPlusClone.GetBinContent(i)) == 1000.: continue if not cut_syst: neg_syst = abs(EffOverJESNegClone.GetBinContent(i)) pos_syst = abs(EffOverJESPlusClone.GetBinContent(i)) if cut_syst: this_syst = pos_syst else: if pos_syst >= neg_syst: this_syst = pos_syst else: this_syst = neg_syst # pick out the largest systematic - most conservative approach syst_hist.SetBinContent(i, this_syst) syst_hist.SetTitle("Systematics") syst_hist.SetMinimum(0.) if settings["text_plot"] and settings["model"] in ["T2cc", "T2_4body"]: r.gStyle.SetPaintTextFormat("0.4f"); syst_hist.SetMarkerSize(0.8) syst_hist.Draw("COLZ TEXT") else: syst_hist.Draw("COLZ") stat_vals = get_hist_stat_vals(EffOverJESPlusClone) stat_vals_string = "Avg=%.3f, Min=%.3f, Max=%.3f"%(stat_vals[2], stat_vals[0], stat_vals[1]) print " >", stat_vals_string num = r.TLatex(0.16,0.8,stat_vals_string) num.SetNDC() num.Draw("same") c1.Print() c1.close() my_syst_hists.append(syst_hist.Clone()) if len(my_syst_hists) == 0: return if settings["model"] == "T2cc": # open Yossof's pdf systematics file pdf_file = r.TFile.Open("envCvRelHist.root", 'READ') pdf_hist = pdf_file.Get("acc_cvRel_m0_m12") my_syst_hists.append(pdf_hist) total_syst_hist = r.TH2D("my_th2", "my_th2", my_syst_hists[0].GetNbinsX(), my_syst_hists[0].GetXaxis().GetBinLowEdge(1), my_syst_hists[0].GetXaxis().GetBinUpEdge(my_syst_hists[0].GetNbinsX()), my_syst_hists[0].GetNbinsY(), my_syst_hists[0].GetYaxis().GetBinLowEdge(1), my_syst_hists[0].GetYaxis().GetBinUpEdge(my_syst_hists[0].GetNbinsY())) # sum all point systs in quadrature and add to final histogram min_val = 1. max_val = 0. avg_val = 0. count = 0 # loop over specific mass points and find corresponding bins, as pdf plot is different binning scheme stop_masses = [] split_masses = [] if settings["model"] == "T2cc": stop_masses = [100.+25.*i for i in range(11)] split_masses = [5., 10., 20., 30., 40., 60., 80.] elif settings["model"] == "T2_4body": stop_masses = [100.+25.*i for i in range(12)] split_masses = [10.*i for i in range(1,9)] else: for x_bin in range(1,my_syst_hists[0].GetNbinsX()+1): for y_bin in range(1,my_syst_hists[0].GetNbinsY()+1): if my_syst_hists[0].GetBinContent(x_bin,y_bin) > 0.: stop_mass = my_syst_hists[0].GetXaxis().GetBinLowEdge(x_bin) if stop_mass not in stop_masses: stop_masses.append(stop_mass) split_mass = my_syst_hists[0].GetXaxis().GetBinLowEdge(x_bin) - my_syst_hists[0].GetYaxis().GetBinLowEdge(y_bin) if split_mass not in split_masses: split_masses.append(split_mass) print stop_masses print split_masses for mstop in stop_masses: for mdiff in split_masses: mlsp = mstop - mdiff val = 0. this_bin = my_syst_hists[0].FindBin(mstop, mlsp) if my_syst_hists[0].GetBinContent(this_bin) <= 0.: print "Zero val bin in syst hist!" continue for s_hist in my_syst_hists: if "cvRel" in s_hist.GetName(): this_yossof_bin = s_hist.FindBin(mstop, mlsp) hist_val = abs(s_hist.GetBinContent(this_yossof_bin)-1.) else: hist_val = s_hist.GetBinContent(this_bin) val += hist_val*hist_val val = math.sqrt(val) if val > 0: count += 1 avg_val += val if val > max_val: max_val = val if val < min_val: min_val = val total_syst_hist.SetBinContent(this_bin, val) if min_val == 1.: min_val = 0. out_name = getOutFile(model=settings["model"], htbin=htbin, format="pdf", bMulti_ = bMulti, jMulti_ = jMulti, mode_ = "total") c_total = r.TCanvas() r.gPad.SetRightMargin(0.15) total_syst_hist.SetTitle("Total Systematic") total_syst_hist.SetMaximum(0.35) if settings["text_plot"] and settings["model"] in ["T2cc", "T2_4body"] : r.gStyle.SetPaintTextFormat("0.4f"); total_syst_hist.SetMarkerSize(0.8) total_syst_hist.Draw("COLZ TEXT") else: total_syst_hist.Draw("COLZ") try: avg_val /= float(count) except ZeroDivisionError: avg_val = 0. num = r.TLatex(0.15,0.8,"Avg=%.3f, Min=%.3f, Max=%.3f"%(avg_val, min_val, max_val)) num.SetNDC() num.Draw("same") print "\t>> Total: Avg=%.3f, Min=%.3f, Max=%.3f"%(avg_val, min_val, max_val) c_total.Print(out_name) # write the output rootfile out_root_file = r.TFile(out_name.replace(".pdf", ".root"), "RECREATE") total_syst_hist.SetName(out_name.replace(".pdf","").replace("_v%d"%settings["version"], "").split("/")[-1]) total_syst_hist.Write() out_root_file.Close() if settings["model"] == "T2cc": pdf_file.Close() del out_root_file
processCrossSections = [] cuts = [] cutsJESNeg = [] cutsJESPlus = [] # cutsJESRan = [] nocuts = [] suf = "" if "275_" in htbin: suf = "73" elif "325_" in htbin: suf = "86" else: suf = "100" c1 = Print(getOutFile(model=model, htbin=htbin, format="pdf")) c1.DoPageNum = False r.gPad.SetRightMargin(0.175) r.gPad.SetLeftMargin(0.15) r.gPad.SetTopMargin(0.08) r.gPad.SetBottomMargin(0.15) nocuts = GetHist(File=centalRootFile100, folder=[ "smsScan_before", ], hist="m0_m12_mChi_noweight", Norm=None, rebinX=4) nocuts = threeToTwo(nocuts)
if h is None: h = a.Clone() else: h.Add(a) return h def nloTotalXsecMaker(individualXSecs=None): out = None for h in individualXSecs: if out is None: out = h.Clone() else: out.Add(h) return out c1 = Print("./out/JES_CMSSM_NEWSCAN.pdf") c1.DoPageNum = False c1.open() r.gPad.SetRightMargin(0.175) centalRootFile73 = r.TFile.Open( "./JESandLepVetoRootFiles/results_NEWSCAN_JESCEN_had_tanB10_73.7.root") centalRootFile86 = r.TFile.Open( "./JESandLepVetoRootFiles/results_NEWSCAN_JESCEN_had_tanB10_86.7.root") centalRootFile100 = r.TFile.Open( "./JESandLepVetoRootFiles/results_NEWSCAN_JESCEN_had_tanB10_100.root") jesPlusRootFile73 = r.TFile.Open( "./JESandLepVetoRootFiles/results_NEWSCAN_had_tanB10_73.7_+ve.root") jesPlusRootFile86 = r.TFile.Open(