ATLAS Offline Software
Loading...
Searching...
No Matches
plotting.yearwise_efficiency_vs_mu Namespace Reference

Functions

 main ()
 plot_efficiency_comb (channel, years)

Variables

 parser = argparse.ArgumentParser()
 type
 str
 help
 args = parser.parse_args()
 year = args.year
 channel = args.channel
 indir = args.indir
 outdir = args.outdir
list years = ["22", "23", "24", "25"]
str out_tag = "run3"
str time_format = "%m/%y"
 ymin
 ymax
str xtitle = 'Month / Year'
str date_tag = "Run 3, #sqrt{s} = 13.6 TeV"
float xval = 0.30
float yval = 0.33
int set_size = 1

Detailed Description

Plot trigger and reconstruction efficiencies over entire data-periods.

Function Documentation

◆ main()

plotting.yearwise_efficiency_vs_mu.main ( )

Definition at line 66 of file yearwise_efficiency_vs_mu.py.

66def main():
67 plot_efficiency_comb(channel, years)
68
int main()
Definition hello.cxx:18

◆ plot_efficiency_comb()

plotting.yearwise_efficiency_vs_mu.plot_efficiency_comb ( channel,
years )

Definition at line 69 of file yearwise_efficiency_vs_mu.py.

69def plot_efficiency_comb(channel, years):
70
71 all_graphs = []
72
73 for year in years:
74 dict_comb = {}
75 dict_comb_err = {}
76 dict_mu = {}
77
78 vec_comb = array('d')
79 vec_comb_err = array('d')
80 vec_mu = array('d')
81
82 grl = pt.get_grl(year)
83
84 for run in grl:
85 livetime, zlumi, zerr, olumi, timestamp, dfz_small = pt.get_dfz(args.indir, year, run, channel)
86
87 # Cut out short runs
88 if livetime < pt.runlivetimecut:
89 if livetime >= 0.: print(f"Skip Run {run} because of live time {livetime/60:.1f} min")
90 continue
91
92 dfz_small['CombEff'] = dfz_small[channel + 'EffComb']
93 dfz_small['CombErr'] = dfz_small[channel + 'ErrComb']
94 # safety to avoid divide by zero - maybe can be removed again later
95 dfz_small.loc[dfz_small['CombErr'] == 0., 'CombErr'] = 1.
96
97 # Scale event-level efficiency with FMC
98 campaign = "mc23a"
99 dfz_small['CombEff'] *= dq_cf.correction(dfz_small['OffMu'], channel, campaign, int(run))
100 dfz_small['CombErr'] *= dq_cf.correction(dfz_small['OffMu'], channel, campaign, int(run))
101
102 for index, event in dfz_small.iterrows():
103 pileup = int(event.OffMu)
104
105 weight_comb = 1/pow(event.CombErr, 2)
106 if pileup not in dict_mu:
107 dict_mu[pileup] = pileup
108 dict_comb[pileup] = weight_comb * event.CombEff
109 dict_comb_err[pileup] = weight_comb
110 else:
111 dict_comb[pileup] += weight_comb * event.CombEff
112 dict_comb_err[pileup] += weight_comb
113
114 if not dict_mu:
115 print("File has no filled lumi blocks!")
116 return
117
118 for pileup in dict_mu:
119 comb_weighted_average = dict_comb[pileup]/dict_comb_err[pileup]
120 comb_error = sqrt(1/dict_comb_err[pileup])
121 vec_comb.append(comb_weighted_average)
122 vec_comb_err.append(comb_error)
123
124 vec_mu.append(pileup)
125
126 all_graphs.append((year, R.TGraphErrors(len(vec_comb), vec_mu, vec_comb, R.nullptr, vec_comb_err)))
127
128 # now draw all
129 c1 = R.TCanvas()
130 leg = R.TLegend(0.645, 0.7, 0.805, 0.9)
131 if channel == "Zee":
132 ymin, ymax = 0.52, 0.74
133 elif channel == "Zmumu":
134 ymin, ymax = 0.74, 0.84
135 xmin, xmax = 0, 80
136
137 for igraph in range(len(all_graphs)):
138 comb_graph = all_graphs[igraph][1]
139 comb_graph.SetMarkerSize(1)
140 comb_graph.SetMarkerColor(R.kBlack+igraph)
141 comb_graph.SetMarkerStyle(R.kFullCircle+igraph)
142 if igraph == 0:
143 comb_graph.Draw("ap")
144 comb_graph.GetHistogram().SetYTitle("#varepsilon_{event}^{"+pt.plotlabel[channel]+"}#times F^{MC}")
145 comb_graph.GetHistogram().GetYaxis().SetRangeUser(ymin, ymax)
146 comb_graph.GetHistogram().GetXaxis().SetLimits(xmin, xmax)
147 comb_graph.GetHistogram().SetXTitle("Pileup (#mu)")
148 else:
149 comb_graph.Draw("p same")
150 leg.SetBorderSize(0)
151 leg.SetTextSize(0.07)
152 leg.AddEntry(comb_graph, "Data 20"+all_graphs[igraph][0], "ep")
153 leg.Draw()
154
155 if channel == "Zee":
156 pt.drawAtlasLabel(0.6, ymax-0.46, "Internal")
157 pt.drawText(0.2, ymax-0.46, date_tag)
158 pt.drawText(0.2, ymax-0.52, pt.plotlabel[channel] + " counting")
159 elif channel == "Zmumu":
160 pt.drawAtlasLabel(0.6, ymax-0.56, "Internal")
161 pt.drawText(0.2, ymax-0.62, pt.plotlabel[channel] + " counting")
162
163 c1.SaveAs(outdir + channel + "_eventeff_vs_mu_"+out_tag+".pdf")
164
void print(char *figname, TCanvas *c1)
constexpr int pow(int base, int exp) noexcept
STL class.

Variable Documentation

◆ args

plotting.yearwise_efficiency_vs_mu.args = parser.parse_args()

Definition at line 21 of file yearwise_efficiency_vs_mu.py.

◆ channel

plotting.yearwise_efficiency_vs_mu.channel = args.channel

Definition at line 23 of file yearwise_efficiency_vs_mu.py.

◆ date_tag

str plotting.yearwise_efficiency_vs_mu.date_tag = "Run 3, #sqrt{s} = 13.6 TeV"

Definition at line 33 of file yearwise_efficiency_vs_mu.py.

◆ help

plotting.yearwise_efficiency_vs_mu.help

Definition at line 16 of file yearwise_efficiency_vs_mu.py.

◆ indir

plotting.yearwise_efficiency_vs_mu.indir = args.indir

Definition at line 24 of file yearwise_efficiency_vs_mu.py.

◆ out_tag

str plotting.yearwise_efficiency_vs_mu.out_tag = "run3"

Definition at line 29 of file yearwise_efficiency_vs_mu.py.

◆ outdir

plotting.yearwise_efficiency_vs_mu.outdir = args.outdir

Definition at line 25 of file yearwise_efficiency_vs_mu.py.

◆ parser

plotting.yearwise_efficiency_vs_mu.parser = argparse.ArgumentParser()

Definition at line 15 of file yearwise_efficiency_vs_mu.py.

◆ set_size

int plotting.yearwise_efficiency_vs_mu.set_size = 1

Definition at line 40 of file yearwise_efficiency_vs_mu.py.

◆ str

plotting.yearwise_efficiency_vs_mu.str

Definition at line 16 of file yearwise_efficiency_vs_mu.py.

◆ time_format

str plotting.yearwise_efficiency_vs_mu.time_format = "%m/%y"

Definition at line 30 of file yearwise_efficiency_vs_mu.py.

◆ type

plotting.yearwise_efficiency_vs_mu.type

Definition at line 16 of file yearwise_efficiency_vs_mu.py.

◆ xtitle

str plotting.yearwise_efficiency_vs_mu.xtitle = 'Month / Year'

Definition at line 32 of file yearwise_efficiency_vs_mu.py.

◆ xval

float plotting.yearwise_efficiency_vs_mu.xval = 0.30

Definition at line 35 of file yearwise_efficiency_vs_mu.py.

◆ year

plotting.yearwise_efficiency_vs_mu.year = args.year

Definition at line 22 of file yearwise_efficiency_vs_mu.py.

◆ years

list plotting.yearwise_efficiency_vs_mu.years = ["22", "23", "24", "25"]

Definition at line 28 of file yearwise_efficiency_vs_mu.py.

◆ ymax

plotting.yearwise_efficiency_vs_mu.ymax

Definition at line 31 of file yearwise_efficiency_vs_mu.py.

◆ ymin

plotting.yearwise_efficiency_vs_mu.ymin

Definition at line 31 of file yearwise_efficiency_vs_mu.py.

◆ yval

float plotting.yearwise_efficiency_vs_mu.yval = 0.33

Definition at line 36 of file yearwise_efficiency_vs_mu.py.