Belle II Software development
PXDOccupancyInfoChecker Class Reference
Inheritance diagram for PXDOccupancyInfoChecker:
ConditionCheckerBase

Public Member Functions

 __init__ (self, name, tfile, rundir="")
 
 define_graphs (self)
 
 get_db_content (self)
 
 get_graph_value (self, sensor_db_content)
 
 draw_plots (self, canvas=None, **kwargs)
 
 run (self)
 
 objType (self)
 
 exp (self)
 
 beginRun (self)
 
 define_hists (self, name=None, title=None, ztitle=None, **kwargs)
 
 get_graph_value_from_hist (self, h2=None)
 
 set_hist_content (self, h2, sensor_db_content)
 
 fill_plots (self)
 
 save_canvas (self, canvas, cname, logy=False)
 

Public Attributes

 name = name
 DBObj name.
 
 dbObj = Belle2.PyDBObj(self.name, self.objType.Class())
 DBObjPtr for the condition.
 
 eventMetaData = Belle2.PyStoreObj("EventMetaData")
 pointer to the event metadata
 
 tfile = tfile
 TFile for saving plots.
 
 rundir = rundir
 Directory for writing histograms of each run.
 
dict graphs = {}
 Dictionary of plots (TGraph) summarizing variables vs run.
 
dict hists = {}
 Dictionary of plots (TH1) to be saved for each run.
 
str hist_title_suffix = ""
 Hist title suffix.
 
int runstart = 999999
 The 1st run.
 
int runend = -1
 The last run.
 
int expstart = 999999
 The 1st exp.
 
 use_hist = use_hist
 Flag to get TGraph values from a histogram (TH2F)
 
 run = self.graphs[sensorID.getID()]
 

Protected Attributes

 _objType = objType
 DBObj type.
 

Detailed Description

Checker for PXDOccupancyInfoPar

Definition at line 273 of file condition_checker.py.

Constructor & Destructor Documentation

◆ __init__()

__init__ ( self,
name,
tfile,
rundir = "" )
 

Definition at line 278 of file condition_checker.py.

278 def __init__(self, name, tfile, rundir=""):
279 """
280 """
281 super().__init__(name, Belle2.PXDOccupancyInfoPar, tfile, "")
282
The payload collecting some meta information from running the masking algorithm.

Member Function Documentation

◆ beginRun()

beginRun ( self)
inherited
Call functions at beginning of a run

Definition at line 109 of file condition_checker.py.

109 def beginRun(self):
110 """
111 Call functions at beginning of a run
112 """
113 self.define_hists()
114 self.fill_plots()
115

◆ define_graphs()

define_graphs ( self)
Method to define TGraph

Reimplemented from ConditionCheckerBase.

Definition at line 283 of file condition_checker.py.

283 def define_graphs(self):
284 """
285 Method to define TGraph
286 """
287 super().define_graphs(ytitle="Occupancy (With Mask) [%]")
288

◆ define_hists()

define_hists ( self,
name = None,
title = None,
ztitle = None,
** kwargs )
inherited
Method to define TH2
Parameters:
  name (str): name for TH2, to which sensor name will be attached
  title (str): title for TH2
  ztitle (str): title for z-axis (color bar)
  kwargs: additional arguments

Reimplemented in PXDDeadPixelsChecker, PXDGainMapChecker, and PXDMaskedPixelsChecker.

Definition at line 125 of file condition_checker.py.

125 def define_hists(self, name=None, title=None, ztitle=None, **kwargs):
126 """
127 Method to define TH2
128 Parameters:
129 name (str): name for TH2, to which sensor name will be attached
130 title (str): title for TH2
131 ztitle (str): title for z-axis (color bar)
132 kwargs: additional arguments
133 """
134 if self.rundir:
135 self.tfile.cd(self.rundir)
136 self.hists.update(get_sensor_maps(name, title, ztitle, self.run, **kwargs))
137

◆ draw_plots()

draw_plots ( self,
canvas = None,
** kwargs )
Method to draw plots on a TCanvas

Reimplemented from ConditionCheckerBase.

Definition at line 302 of file condition_checker.py.

302 def draw_plots(self, canvas=None, **kwargs):
303 """
304 Method to draw plots on a TCanvas
305 """
306 # We don't use raw occupanncy. They will be corrected after takig out dead pixels
307 pass
308 # super().draw_plots(canvas=canvas, cname="OccupancyWithMask", ymin=0., ymax=plot_type_dict["occ_masked"]["max"])
309
310

◆ exp()

exp ( self)
inherited
Experiment number

Definition at line 103 of file condition_checker.py.

103 def exp(self):
104 """
105 Experiment number
106 """
107 return self.eventMetaData.getExperiment()
108

◆ fill_plots()

fill_plots ( self)
inherited
Method to fill plot objects

Definition at line 170 of file condition_checker.py.

170 def fill_plots(self):
171 """
172 Method to fill plot objects
173 """
174 if self.exp < self.expstart:
175 self.expstart = self.exp
176 if self.run < self.runstart:
177 self.runstart = self.run
178 if self.run > self.runend:
179 self.runend = self.run
180
181 db_content = self.get_db_content()
182 values_dict = {}
183 pre_values_dict = {}
184 for sensorID in sensorID_list:
185 sensor_db_content = db_content[sensorID.getID()]
186 if self.graphs:
187 if not self.use_hist:
188 values_dict[sensorID.getID()] = self.get_graph_value(sensor_db_content)
189 gr = self.graphs[sensorID.getID()]
190 pre_values_dict[sensorID.getID()] = gr.GetPointY(gr.GetN() - 1)
191 # Saving graph points only for different values (sensor-wise)
192 # value = self.get_graph_value(sensor_db_content)
193 # pre_value = gr.GetPointY(gr.GetN()-1)
194 # if value != pre_value:
195 # gr.SetPoint(gr.GetN(), self.run, value)
196 # Saving masked map
197 if self.rundir != "" and self.hists:
198 h2 = self.hists[sensorID.getID()]
199 self.set_hist_content(h2, sensor_db_content)
200 # Update title with total count
201 h2.SetTitle(h2.GetTitle() + self.hist_title_suffix)
202 self.tfile.cd(self.rundir)
203 h2.Write()
204 if self.use_hist:
205 values_dict[sensorID.getID()] = self.get_graph_value_from_hist(h2)
206 # Saving graph points only for different conditions
207 if values_dict != pre_values_dict:
208 for sensorID in sensorID_list:
209 gr = self.graphs[sensorID.getID()]
210 gr.SetPoint(gr.GetN(), self.run, values_dict[sensorID.getID()])
211

◆ get_db_content()

get_db_content ( self)
Abstract method to get content of a payload
Should return a Dictionary with sensorID.getID() as key and relaated calibration results as value

Reimplemented from ConditionCheckerBase.

Definition at line 289 of file condition_checker.py.

289 def get_db_content(self):
290 content_dic = {}
291 for sensorID in sensorID_list:
292 numID = sensorID.getID()
293 raw_occ = self.dbObj.getOccupancy(numID)
294 # The default value is -1. It seems 0 will not be updated into the payload.
295 content_dic[numID] = raw_occ if raw_occ >= 0 else 0
296
297 return content_dic
298

◆ get_graph_value()

get_graph_value ( self,
sensor_db_content )
Abstract method to get a value for each TGraph
Parameters:
  sensor_db_content (Any): Calibration results of a module

Reimplemented from ConditionCheckerBase.

Definition at line 299 of file condition_checker.py.

299 def get_graph_value(self, sensor_db_content):
300 return sensor_db_content * 100
301

◆ get_graph_value_from_hist()

get_graph_value_from_hist ( self,
h2 = None )
inherited
Method to get a value for each TGraph
Parameters:
  h2 (TH2): If not none, get value from h2

Reimplemented in PXDDeadPixelsChecker.

Definition at line 153 of file condition_checker.py.

153 def get_graph_value_from_hist(self, h2=None):
154 """
155 Method to get a value for each TGraph
156 Parameters:
157 h2 (TH2): If not none, get value from h2
158 """
159 return None
160

◆ objType()

objType ( self)
inherited
DBObj type (read only)

Definition at line 89 of file condition_checker.py.

89 def objType(self):
90 """
91 DBObj type (read only)
92 """
93 return self._objType
94

◆ run()

run ( self)
inherited
Run number

Definition at line 96 of file condition_checker.py.

96 def run(self):
97 """
98 Run number
99 """
100 return self.eventMetaData.getRun()
101

◆ save_canvas()

save_canvas ( self,
canvas,
cname,
logy = False )
inherited
Save TCanvas to png/pdf format and do not write it to the ROOT file by default
Parameters:
  canvas (TCanvas): ROOT TCanvas for plotting
  canme (str): Name of the canvas
  logy (bool): Flag to use log scale for y-axis

Definition at line 246 of file condition_checker.py.

246 def save_canvas(self, canvas, cname, logy=False):
247 """
248 Save TCanvas to png/pdf format and do not write it to the ROOT file by default
249 Parameters:
250 canvas (TCanvas): ROOT TCanvas for plotting
251 canme (str): Name of the canvas
252 logy (bool): Flag to use log scale for y-axis
253 """
254 if cname:
255 canvas.SetName(cname)
256 exp_run = f"e{self.expstart:05}_r{self.runstart:04}-{self.runend:04}"
257 # Draw Belle II label
258 latex_r.DrawLatex(0.95, 0.92, "Belle II Experiment: " + exp_run.replace("_", " "))
259 # Print and write TCanvas
260 if logy:
261 canvas.SetLogy(1)
262 canvas.Update()
263 canvas.Modified()
264 canvas.Print(f"{exp_run}_{cname}_vs_run_logy.png")
265 canvas.Print(f"{exp_run}_{cname}_vs_run_logy.pdf")
266 else:
267 canvas.SetLogy(0)
268 canvas.Print(f"{exp_run}_{cname}_vs_run.png")
269 canvas.Print(f"{exp_run}_{cname}_vs_run.pdf")
270 self.tfile.cd()
271
272

◆ set_hist_content()

set_hist_content ( self,
h2,
sensor_db_content )
inherited
Method to set TH2 bins
Parameters:
  h2 (TH2): TH2F object for handling values of a pixel matrix
  sensor_db_content (Any): Calibration results of a module

Reimplemented in PXDDeadPixelsChecker, PXDGainMapChecker, and PXDMaskedPixelsChecker.

Definition at line 161 of file condition_checker.py.

161 def set_hist_content(self, h2, sensor_db_content):
162 """
163 Method to set TH2 bins
164 Parameters:
165 h2 (TH2): TH2F object for handling values of a pixel matrix
166 sensor_db_content (Any): Calibration results of a module
167 """
168 pass
169

Member Data Documentation

◆ _objType

_objType = objType
protectedinherited

DBObj type.

Definition at line 59 of file condition_checker.py.

◆ dbObj

dbObj = Belle2.PyDBObj(self.name, self.objType.Class())
inherited

DBObjPtr for the condition.

Definition at line 61 of file condition_checker.py.

◆ eventMetaData

eventMetaData = Belle2.PyStoreObj("EventMetaData")
inherited

pointer to the event metadata

Definition at line 63 of file condition_checker.py.

◆ expstart

int expstart = 999999
inherited

The 1st exp.

Definition at line 82 of file condition_checker.py.

◆ graphs

dict graphs = {}
inherited

Dictionary of plots (TGraph) summarizing variables vs run.

Definition at line 72 of file condition_checker.py.

◆ hist_title_suffix

hist_title_suffix = ""
inherited

Hist title suffix.

Definition at line 76 of file condition_checker.py.

◆ hists

dict hists = {}
inherited

Dictionary of plots (TH1) to be saved for each run.

Definition at line 74 of file condition_checker.py.

◆ name

name = name
inherited

DBObj name.

Definition at line 57 of file condition_checker.py.

◆ run

run = self.graphs[sensorID.getID()]
inherited

Definition at line 136 of file condition_checker.py.

◆ rundir

rundir = rundir
inherited

Directory for writing histograms of each run.

Definition at line 67 of file condition_checker.py.

◆ runend

int runend = -1
inherited

The last run.

Definition at line 80 of file condition_checker.py.

◆ runstart

int runstart = 999999
inherited

The 1st run.

Definition at line 78 of file condition_checker.py.

◆ tfile

tfile = tfile
inherited

TFile for saving plots.

Definition at line 65 of file condition_checker.py.

◆ use_hist

use_hist = use_hist
inherited

Flag to get TGraph values from a histogram (TH2F)

Definition at line 84 of file condition_checker.py.


The documentation for this class was generated from the following file: