コード例 #1
0
ファイル: plot_chi2.py プロジェクト: xju2/xaodtools
def plot():
    fin = ROOT.TFile.Open("output_chi2.root")
    canvas = ROOT.TCanvas("canvas", "canvas", 600, 600)
    for key in fin.GetListOfKeys():
        h1 = key.ReadObj()
        name = h1.GetName()
        if "pt" in name:
            h1.GetYaxis().SetLabelSize(0.035)
        if h1.ClassName() == "TH2F":
            h1.Draw("box")
        else:
            h1.Draw("EP")
        canvas.SaveAs(name+".pdf")

        if h1.ClassName() == "TH2F":
            py = h1.ProfileY()
            py.Draw()
            canvas.SaveAs(name+"_profileY.pdf")

    cpt.save_compare(cpt.norm_hist(fin.Get("h_chi2")), cpt.norm_hist(fin.Get("h_chi2_ts")), "chi2_cmp")
    save_compare([cpt.norm_hist(fin.Get("h_chi2")), cpt.norm_hist(fin.Get("h_chi2_ts"))],["#Upsilon(1S)#Upsilon(1S)","baseline"],"chi2_cmp")

    # compare the slices
    labels = ["#chi^{2} < 10", "#chi^{2} [10, 100]", "#chi^{2} > 100"]
    for i in range(4):
        hists = get_chi2_slice( fin.Get("h_chi2_mu"+str(i+1)+"_d0") )
        save_compare([cpt.norm_hist(x) for x in hists], labels, "cmp_mu"+str(i+1)+"_d0")

    for i in range(4):
        hists = get_chi2_slice( fin.Get("h_chi2_mu"+str(i+1)+"_z0") )
        save_compare([cpt.norm_hist(x) for x in hists], labels, "cmp_mu"+str(i+1)+"_z0")

    fin.Close()
コード例 #2
0
ファイル: compare_chi2.py プロジェクト: xju2/xaodtools
def compare_hist():
    f1 = ROOT.TFile.Open("diff.root")

    cpt.save_compare(cpt.norm_hist(f1.Get("chi2_sig_xy")), cpt.norm_hist(f1.Get("chi2_sig_ts")), "chi2_sameevents")
    cpt.save_compare(cpt.norm_hist(f1.Get("m4l_xy")), cpt.norm_hist(f1.Get("m4l_ts")), "mass_sameevents")
    cpt.save_compare(cpt.norm_hist(f1.Get("m4l_full_xy")), cpt.norm_hist(f1.Get("m4l_full_ts")), "mass_fullRange_sameevents")

    f1.Close()