9 from PixelCalibAlgs.Recovery
import ReadDbFile, ReadNewCalib
10 from PathResolver
import PathResolver
13 from matplotlib.backends.backend_agg
import FigureCanvasAgg
14 from matplotlib.figure
import Figure
23 m_array = [
charge(parameters, 3*(1+i))
for i
in range(14)]
26 m_array = [
charge(parameters, 5*(1+i))
for i
in range(8)]
28 return ar.array(
'd',m_array)
31 num = tot*parameters[2] - parameters[0]*parameters[1]
32 den = parameters[0] - tot
33 return (num/den)
if den != 0
else 0
36 return a/(a+b)*100
if (a+b) != 0
else 999.99
42 expected_charge = 20000
49 expected_charge = 16000
51 if abs(parameters[9]-expected_charge)/expected_charge*100 > 5:
52 return True, [parameters[9], abs(parameters[9]-expected_charge)/expected_charge*100, expected_charge]
55 low = expected_charge*(1-error/100)
56 high = expected_charge*(1+error/100)
57 realQ =
charge(parameters, tot)
59 if realQ < low
or realQ > high:
60 return True, [
charge(parameters, tot), abs(
charge(parameters, tot)-expected_charge)/expected_charge*100,expected_charge]
63 def EvoMon(old_calib, new_calib, mapping, old_iov, new_iov):
69 information = {
"Total_mods": 0,
71 "IBL" :{
"bad":0,
"ok":0},
72 "Blayer":{
"bad":0,
"ok":0},
73 "L1" :{
"bad":0,
"ok":0},
74 "L2" :{
"bad":0,
"ok":0},
75 "Disk" :{
"bad":0,
"ok":0}
79 for mod
in range(2048):
81 mod_str = mapping[
str(mod)]
83 print(
"%-18s - %4i" % (mod_str, mod), end=
'\r')
84 plot_range = np.array([0,75000])
86 if str(mod)
not in new_calib:
89 if mod_str.startswith(
"L0"):
91 elif mod_str.startswith(
"L1"):
93 elif mod_str.startswith(
"L2"):
95 elif mod_str.startswith(
"D"):
99 if mod_str.startswith(
"LI_S15"):
102 information[
"Total_mods"] += 1
103 fig = Figure(figsize=(13,10))
104 axs = fig.add_subplot(1,1,1)
107 for fe
in range(len(new_calib[
str(mod)])):
109 information[
"Total_FE"] += 1
115 if mod_layer !=
"IBL":
116 newCal_normal_pix = new_calib[
str(mod)][fe][12:15]
117 oldCal_normal_pix = old_calib[
str(mod)][fe][12:15]
125 newQ = new_calib[
str(mod)][fe][4:20]
126 oldQ = old_calib[
str(mod)][fe][4:20]
128 plot_range = np.array([0,35000])
131 m,b = np.polyfit(newQ ,oldQ,1)
133 boolFit = (abs((1-m)/m)*100) > MaxThreshold
135 if boolFit
or boolTOT:
136 key =
"%-18s - %i" % (mod_str, mod)
138 if key
not in log_info:
139 log_info[key] =
"\tFE%02i ---> slope: %5.2f - dev: %5.1f%%\n" % (fe,m, abs((1-m)/m)*100)
141 log_info[key] +=
"\tFE%02i ---> slope: %5.2f - dev: %5.1f%%\n" % (fe,m, abs((1-m)/m)*100)
143 if key
not in log_info:
144 log_info[key] =
"\tFE%02i ---> Charge= %5ie (dev %6.2f%%) out of error bars. Expected: %5ie\n" % (fe, realQ[0], realQ[1], realQ[2])
146 log_info[key] +=
"\tFE%02i ---> Charge= %5ie (dev %6.2f%%) out of error bars. Expected: %5ie\n" % (fe, realQ[0], realQ[1], realQ[2])
148 information[mod_layer][
"bad"] += 1
157 information[mod_layer][
"ok"] += 1
161 lstyle =
"dotted" if fe < 8
else "solid"
162 axs.plot(newQ ,oldQ, marker=
'o', linestyle=lstyle, label = (
"FE%02d" % (fe)))
163 axs.set_xlim(plot_range)
164 axs.set_ylim(plot_range)
167 fig.suptitle(
"Hash ID %d - %s" % (mod, mod_str))
168 axs.legend(loc=
'upper left', ncol=4)
169 axs.set_xlabel(new_iov+
" - Charge[e]")
170 axs.set_ylabel(old_iov+
" - Charge[e]")
171 axs.plot(plot_range,plot_range,
"k:",lw=1)
173 if mod_layer ==
'IBL':
174 axs.plot([16000,16000],plot_range,
"k:",lw=1)
175 axs.text(17000,1000,
"TOT@10 = 16 ke")
176 elif mod_layer ==
'Blayer':
177 axs.plot([20000,20000],plot_range,
"k:",lw=1)
178 axs.text(21000,1000,
"TOT@18 = 20 ke")
180 axs.plot([20000,20000],plot_range,
"k:",lw=1)
181 axs.text(21000,1000,
"TOT@30 = 20 ke")
183 storage =
"plots/" + mod_layer +
"/"
185 canvas = FigureCanvasAgg(fig)
186 canvas.print_figure(storage+
"Id"+
str(mod)+
"_" +mod_str+status+
".png", dpi=150)
189 fig = Figure(figsize=(13,10))
190 fig.suptitle(
"Fit slopes for all modules")
191 axs = fig.add_subplot(1,1,1)
192 axs.hist(np.clip(slopes, -1, 2), bins=100)
193 axs.set_xlabel(
"Fit slope")
194 axs.set_ylabel(
"Counts")
196 stats = (f
'$\\mu$ = {np.mean(np.clip(slopes, -1, 2)):.3f}\n'
197 f
'$\\sigma$ = {np.std(np.clip(slopes, -1, 2)):.3f}')
198 bbox = dict(boxstyle=
'round', fc=
'blanchedalmond', ec=
'orange', alpha=0.5)
199 axs.text(0.95, 0.07, stats, fontsize=9, bbox=bbox, transform=axs.transAxes)
201 FigureCanvasAgg(fig).print_figure(
"plots/FitSlopes.png", dpi=150)
203 return information, log_info
209 lines = fp.readlines()
211 arrayline = line.rstrip(
"\n").
split(
', ')
212 mydict[arrayline[1]] = arrayline[0]
217 with open(file)
as fp:
218 lines = fp.readlines()
220 arrayline = line.rstrip(
"\n").
split(
' ')
222 if arrayline[0]
not in mydict:
223 mydict[arrayline[0]] = []
226 while(
"" in arrayline):
229 mydict[arrayline[0]].
append([
int(arrayline[i])
if i<13
else float(arrayline[i])
for i
in range(1,len(arrayline)) ])
231 return mydict,
"[latest,0]"
234 new_iov =
"Latest IOV"
235 old_iov =
"Older IOV"
237 print(
"Files chosen for the comparison:")
239 print(
"New calibration: '%s'" % path_newCalib)
240 new_calib, new_iov =
"",
""
241 if "PIX_FINAL_calibration_candidate.txt" in path_newCalib:
243 elif "ChargeCalib_" in path_newCalib:
248 print(
"Old calibration: '%s'" % path_oldCalib)
249 old_calib, old_iov =
ReadDbFile(path_oldCalib)
254 os.makedirs(
"plots/Blayer", exist_ok=
True)
255 os.makedirs(
"plots/L1" , exist_ok=
True)
256 os.makedirs(
"plots/L2" , exist_ok=
True)
257 os.makedirs(
"plots/Disk" , exist_ok=
True)
258 os.makedirs(
"plots/IBL" , exist_ok=
True)
260 information, log_info =
EvoMon(old_calib, new_calib, mapping, old_iov, new_iov )
263 fout =
open(
"EvoMonitoring_log.txt",
"w")
265 fout.write(
"_______________ LOG _______________\n\n" )
266 fout.write(
"Files chosen for the comparison:\n" )
267 fout.write(
"New calibration: '%s'\n" % (path_newCalib) )
268 fout.write(
"Old calibration: '%s'\n" % (path_oldCalib) )
269 fout.write(
"Data base IOV: %s\n\n" % (old_iov) )
271 fout.write(
"-"*20+
" SUMMARY "+
"-"*20+
"\n" )
272 fout.write(
"%-20s: %5i\n" % (
"Total mods in det.", information[
"Total_mods"]))
273 fout.write(
"%-20s: %5i\n" % (
"Total FE in det." , information[
"Total_FE"] ))
274 fout.write(
"%-20s: %5i\n" % (
"Total FE IBL" , information[
"IBL"][
"ok"] +information[
"IBL"][
"bad"] ))
275 fout.write(
"%-20s: %5i\n" % (
"Total FE Blayer" , information[
"Blayer"][
"ok"]+information[
"Blayer"][
"bad"]))
276 fout.write(
"%-20s: %5i\n" % (
"Total FE L1" , information[
"L1"][
"ok"] +information[
"L1"][
"bad"] ))
277 fout.write(
"%-20s: %5i\n" % (
"Total FE L2" , information[
"L2"][
"ok"] +information[
"L2"][
"bad"] ))
278 fout.write(
"%-20s: %5i\n\n" % (
"Total FE Disk" , information[
"Disk"][
"ok"] +information[
"Disk"][
"bad"] ))
280 fout.write(f
'FrontEnds deviating more than {MaxThreshold}% of TOT vs charge gradient between new and previous calibration:\n')
281 fout.write(
"%-11s: %-4i (%6.2f%%)\n" % (
"IBL FEs" , information[
"IBL"][
"bad"] ,
percent(information[
"IBL"][
"bad"] ,information[
"IBL"][
"ok"]) ))
282 fout.write(
"%-11s: %-4i (%6.2f%%)\n" % (
"Blayer FEs", information[
"Blayer"][
"bad"],
percent(information[
"Blayer"][
"bad"],information[
"Blayer"][
"ok"])))
283 fout.write(
"%-11s: %-4i (%6.2f%%)\n" % (
"L1 FEs" , information[
"L1"][
"bad"] ,
percent(information[
"L1"][
"bad"] ,information[
"L1"][
"ok"]) ))
284 fout.write(
"%-11s: %-4i (%6.2f%%)\n" % (
"L2 FEs" , information[
"L2"][
"bad"] ,
percent(information[
"L2"][
"bad"] ,information[
"L2"][
"ok"]) ))
285 fout.write(
"%-11s: %-4i (%6.2f%%)\n\n" % (
"Disk FEs" , information[
"Disk"][
"bad"] ,
percent(information[
"Disk"][
"bad"] ,information[
"Disk"][
"ok"]) ))
287 fout.write(
"+"*20+
" List of bad FE "+
"+"*20+
"\n" )
288 fout.write(
"Expected TOT vs. charge for the different layers:\n")
289 fout.write(
"%-10s: TOT@%2i = %2ike\n" % (
"IBL" , 10, 16))
290 fout.write(
"%-10s: TOT@%2i = %2ike\n" % (
"Blayer" , 18, 20))
291 fout.write(
"%-10s: TOT@%2i = %2ike\n\n" % (
"L1/L2/Disk", 30, 20))
293 for key, val
in log_info.items():
300 fin =
open(
"EvoMonitoring_log.txt",
"r")
304 if __name__ ==
"__main__":
307 parser = argparse.ArgumentParser(prog=
'python -m PixelCalibAlgs.EvoMonitoring',
308 description=
"""Compares two IOV and plots the results.\n\n
309 Example: python -m PixelCalibAlgs.EvoMonitoring --new "path/to/file" --old "path/to/file" """)
311 parser.add_argument(
'--new', default=
"PIX_FINAL_calibration_candidate.txt", help=
"New calibration file (output format from the Recovery.py)")
312 parser.add_argument(
'--old', default=
"PixelChargeCalibration-DATA-RUN2-UPD4-28.log", help=
"Old DB IOV calibration")
314 args = parser.parse_args()