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
79 vec_comb_err =
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
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
95 dfz_small.loc[dfz_small['CombErr'] == 0., 'CombErr'] = 1.
96
97
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
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