TestBeam/bias_scan_eff_Dec2022.py at master · jreic/TestBeam · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
from ROOTTools import *
import sys

ROOT.gStyle.SetOptFit(0100) # adds Landau MPV to stat box


ps = plot_saver(plot_dir("overlay_bias_scan_Dec2022_QuadCROC_eff"), size=(600,600), log=False, pdf=True)
ps.c.SetLeftMargin(0.15)
ps.update_canvas()

plot_names = [
               "EfficiencyRef_Dut0",
               "Efficiency_Dut0"
             ]

for plot_name in plot_names :

    fpath20V = "/uscms/home/joeyr/publicweb/TFPX/Chewie_Run61907/"+plot_name+".root"
    fpath40V = "/uscms/home/joeyr/publicweb/TFPX/Chewie_Run61908/"+plot_name+".root"
    fpath60V = "/uscms/home/joeyr/publicweb/TFPX/Chewie_Run61909/"+plot_name+".root"
    fpath80V = "/uscms/home/joeyr/publicweb/TFPX/Chewie_Run61910/"+plot_name+".root"
    fpath100V = "/uscms/home/joeyr/publicweb/TFPX/Chewie_Run61911/"+plot_name+".root"
    fpath110V = "/uscms/home/joeyr/publicweb/TFPX/Chewie_Run61913/"+plot_name+".root"

    files = [ ROOT.TFile(fpath110V)
             ,ROOT.TFile(fpath100V)
             ,ROOT.TFile(fpath80V)
             ,ROOT.TFile(fpath60V)
             ,ROOT.TFile(fpath40V)
             ,ROOT.TFile(fpath20V)
            ]

    labels = [110,100,80,60,40,20]
    colors = [ROOT.kRed,ROOT.kOrange+1,ROOT.kBlue,ROOT.kGreen+2,ROOT.kViolet-3,ROOT.kOrange-6,ROOT.kCyan+1,ROOT.kMagenta,ROOT.kTeal+5,ROOT.kAzure+2,ROOT.kYellow+2,ROOT.kSpring,ROOT.kGray,ROOT.kAzure-4,ROOT.kOrange,ROOT.kRed-2,ROOT.kBlue-2,ROOT.kSpring+4,ROOT.kGray+2,ROOT.kViolet+2, ROOT.kPink+1]
    hists = []

    outhist = ROOT.TH1F(plot_name, plot_name, (max(labels)+10), min(labels), max(labels)+10)

    #leg = ROOT.TLegend(0.60, 0.75, 0.90, 0.90)
    leg = ROOT.TLegend(0.50, 0.60, 0.90, 0.90)
    leg.SetMargin(0.15)

    for index in xrange(0,len(files)) :
        f = files[index]
        can = f.Get("c0")
        h = can.GetPrimitive(plot_name)
        eff = h.GetBinContent(1)
        if eff == 0 : eff = 1e-5
        outhist.SetBinContent(outhist.FindBin(labels[index]), eff)
        #h.SetName(labels[index])
        #h.SetLineColor(colors[index])
        #h.SetMarkerColor(colors[index])
        #h.SetLineWidth(3)
        #h.Scale(1/h.Integral())

        ps.c.cd()
        #if index == 0 :
        #    h.Draw()
        #    h.GetYaxis().SetRangeUser(0.001,1)
        #    h.GetYaxis().SetTitle("fraction / bin")
        #    h.GetYaxis().SetTitleOffset(2.0)
        #else :
        #    h.Draw("same")

        #leg.AddEntry(h,h.GetName()+", mean = %.2f #pm %.2f, #sigma = %.2f #pm %.2f" % (h.GetMean(), h.GetMeanError(), h.GetStdDev(), h.GetStdDevError()),"l")
        #leg.AddEntry(h,h.GetName()+", mean = %.2f, #sigma = %.2f" % (h.GetMean(), h.GetStdDev()),"l")
        #h.SetStats(0)
        #hists.append(h)
        ps.update_canvas()

    if plot_name == "hClusterSize_Dut0" :
        histmin = 5e-6
        histmax = 1
    elif "ResidualsClusterSize2" in plot_name :
        histmin = 1e-5
        histmax = 0.06
    elif "Residuals" in plot_name :
        histmin = 5e-6
        histmax = 0.04
    elif "Efficiency" in plot_name :
        histmin = 0.98
        histmax = 1.0001

    outhist.GetYaxis().SetRangeUser(histmin, histmax)
    outhist.GetYaxis().SetTitleOffset(2.0)
    outhist.GetYaxis().SetTitle("efficiency")
    outhist.GetXaxis().SetTitle("bias voltage (V)")
    outhist.SetLineColor(ROOT.kBlack)
    outhist.SetMarkerColor(ROOT.kBlack)
    outhist.SetMarkerStyle(20)
    outhist.Draw("p")
    outhist.SetStats(0)
    #leg.Draw()
    ps.update_canvas()
    ps.save(plot_name)