Exemplo n.º 1
0
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()
Exemplo n.º 2
0
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()
Exemplo n.º 3
0
        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")
Exemplo n.º 4
0
        # 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",
Exemplo n.º 5
0
        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" %
Exemplo n.º 6
0
    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))
    
Exemplo n.º 7
0
            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
Exemplo n.º 8
0
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()
Exemplo n.º 9
0
      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
Exemplo n.º 11
0
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()
Exemplo n.º 12
0
        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")
Exemplo n.º 13
0
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
Exemplo n.º 14
0
        # 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
Exemplo n.º 16
0
        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)
Exemplo n.º 17
0
        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(