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

Public Member Functions

 __init__ (self, name, tfile, rundir="maps")
 
 define_graphs (self)
 
 define_hists (self)
 
 get_db_content (self)
 
 get_graph_value (self, sensor_db_content)
 
 set_hist_content (self, h2, sensor_db_content)
 
 draw_plots (self, canvas=None, cname="PXDGain", ymin=0.5, ymax=2.5, **kwargs)
 
 run (self)
 
 objType (self)
 
 exp (self)
 
 beginRun (self)
 
 get_graph_value_from_hist (self, h2=None)
 
 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 PXDGainMapPar

Definition at line 453 of file condition_checker.py.

Constructor & Destructor Documentation

◆ __init__()

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

Definition at line 458 of file condition_checker.py.

458 def __init__(self, name, tfile, rundir="maps"):
459 """
460 """
461 super().__init__(name, Belle2.PXDGainMapPar, tfile, rundir)
462
The payload class for PXD gain corrections.

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 463 of file condition_checker.py.

463 def define_graphs(self):
464 """
465 Method to define TGraph
466 """
467 super().define_graphs(ytitle="Gain / MC default")
468

◆ define_hists()

define_hists ( self)
Method to define TH2

Reimplemented from ConditionCheckerBase.

Definition at line 469 of file condition_checker.py.

469 def define_hists(self):
470 """
471 Method to define TH2
472 """
473 nU = self.dbObj.getBinsU()
474 nV = self.dbObj.getBinsV()
475 super().define_hists(name="GainMap", title="Gain / MC default", ztitle="Relative gain", nUCells=nU, nVCells=nV)
476

◆ draw_plots()

draw_plots ( self,
canvas = None,
cname = "PXDGain",
ymin = 0.5,
ymax = 2.5,
** kwargs )
Method to draw plots on a TCanvas

Reimplemented from ConditionCheckerBase.

Definition at line 494 of file condition_checker.py.

494 def draw_plots(self, canvas=None, cname="PXDGain", ymin=0.5, ymax=2.5, **kwargs):
495 """
496 Method to draw plots on a TCanvas
497 """
498 super().draw_plots(canvas=canvas, cname=cname, ymin=ymin, ymax=ymax, **kwargs)

◆ 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 477 of file condition_checker.py.

477 def get_db_content(self):
478 gainMap = self.dbObj.getCalibrationMap()
479 content_dic = {}
480 for sensorID in sensorID_list:
481 numID = sensorID.getID()
482 content_dic[numID] = np.array(list(gainMap[numID]))
483 return content_dic
484

◆ get_graph_value()

get_graph_value ( self,
sensor_db_content )
sensor_db_content (np.array): Array of gain factors of a module

Reimplemented from ConditionCheckerBase.

Definition at line 485 of file condition_checker.py.

485 def get_graph_value(self, sensor_db_content):
486 """
487 sensor_db_content (np.array): Array of gain factors of a module
488 """
489 return sensor_db_content.mean()
490

◆ 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 )
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 from ConditionCheckerBase.

Definition at line 491 of file condition_checker.py.

491 def set_hist_content(self, h2, sensor_db_content):
492 array2hist(sensor_db_content.reshape(h2.GetNbinsX(), h2.GetNbinsY()), h2)
493

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: