Belle II Software development
PXDHotPixelMaskCalibrationAlgorithm Class Reference

Class implementing PXD hot pixel masking calibration algorithm. More...

#include <PXDHotPixelMaskCalibrationAlgorithm.h>

Inheritance diagram for PXDHotPixelMaskCalibrationAlgorithm:
CalibrationAlgorithm

Public Types

enum  EResult {
  c_OK ,
  c_Iterate ,
  c_NotEnoughData ,
  c_Failure ,
  c_Undefined
}
 The result of calibration. More...
 

Public Member Functions

 PXDHotPixelMaskCalibrationAlgorithm ()
 Constructor set the prefix to PXDHotPixelMaskCalibrationAlgorithm.
 
void setDebugHisto (bool debugHisto)
 setter for m_debugHisto
 
std::string getPrefix () const
 Get the prefix used for getting calibration data.
 
bool checkPyExpRun (PyObject *pyObj)
 Checks that a PyObject can be successfully converted to an ExpRun type.
 
Calibration::ExpRun convertPyExpRun (PyObject *pyObj)
 Performs the conversion of PyObject to ExpRun.
 
std::string getCollectorName () const
 Alias for prefix.
 
void setPrefix (const std::string &prefix)
 Set the prefix used to identify datastore objects.
 
void setInputFileNames (PyObject *inputFileNames)
 Set the input file names used for this algorithm from a Python list.
 
PyObject * getInputFileNames ()
 Get the input file names used for this algorithm and pass them out as a Python list of unicode strings.
 
std::vector< Calibration::ExpRun > getRunListFromAllData () const
 Get the complete list of runs from inspection of collected data.
 
RunRange getRunRangeFromAllData () const
 Get the complete RunRange from inspection of collected data.
 
IntervalOfValidity getIovFromAllData () const
 Get the complete IoV from inspection of collected data.
 
void fillRunToInputFilesMap ()
 Fill the mapping of ExpRun -> Files.
 
std::string getGranularity () const
 Get the granularity of collected data.
 
EResult execute (std::vector< Calibration::ExpRun > runs={}, int iteration=0, IntervalOfValidity iov=IntervalOfValidity())
 Runs calibration over vector of runs for a given iteration.
 
EResult execute (PyObject *runs, int iteration=0, IntervalOfValidity iov=IntervalOfValidity())
 Runs calibration over Python list of runs. Converts to C++ and then calls the other execute() function.
 
std::list< Database::DBImportQuery > & getPayloads ()
 Get constants (in TObjects) for database update from last execution.
 
std::list< Database::DBImportQuerygetPayloadValues ()
 Get constants (in TObjects) for database update from last execution but passed by VALUE.
 
bool commit ()
 Submit constants from last calibration into database.
 
bool commit (std::list< Database::DBImportQuery > payloads)
 Submit constants from a (potentially previous) set of payloads.
 
const std::string & getDescription () const
 Get the description of the algorithm (set by developers in constructor)
 
bool loadInputJson (const std::string &jsonString)
 Load the m_inputJson variable from a string (useful from Python interface). The return bool indicates success or failure.
 
const std::string dumpOutputJson () const
 Dump the JSON string of the output JSON object.
 
const std::vector< Calibration::ExpRun > findPayloadBoundaries (std::vector< Calibration::ExpRun > runs, int iteration=0)
 Used to discover the ExpRun boundaries that you want the Python CAF to execute on. This is optional and only used in some.
 
template<>
std::shared_ptr< TTree > getObjectPtr (const std::string &name, const std::vector< Calibration::ExpRun > &requestedRuns)
 Specialization of getObjectPtr<TTree>.
 

Public Attributes

bool forceContinueMasking
 Force continue masking in almost empty runs instead of returning c_NotEnoughData.
 
int minEvents
 Minimum number of collected events.
 
int minHits
 Minimum median number of hits per pixel needed for dead pixel masking.
 
float pixelMultiplier
 The occupancy threshold for masking hot single pixels is the median occupancy x pixelMultiplier.
 
bool maskDrains
 Mask hot drain lines with too high average occupancy after single pixel masking.
 
float drainMultiplier
 The occupancy threshold for masking hot drains is the median occupancy x drainMultiplier.
 
bool maskRows
 Mask hot rows with too high average occupancy after single pixel masking.
 
float rowMultiplier
 The occupancy threshold for masking hot rows is the median occupancy x rowMultiplier.
 
std::string deadPixelPayloadName
 Payload name for PXDDeadPixelPar used for more defective pixels from damaged gates.
 
float inefficientPixelMultiplier
 The occupancy threshold for inefficient (or dead) pixels is the median occupancy x inefficientPixelMultiplier.
 
int minInefficientPixels
 The minimum number of inefficient (or dead) pixels per row to define an inefficient row.
 

Protected Member Functions

virtual EResult calibrate () override
 Run algo on data.
 
void setInputFileNames (std::vector< std::string > inputFileNames)
 Set the input file names used for this algorithm.
 
virtual bool isBoundaryRequired (const Calibration::ExpRun &)
 Given the current collector data, make a decision about whether or not this run should be the start of a payload boundary.
 
virtual void boundaryFindingSetup (std::vector< Calibration::ExpRun >, int)
 If you need to make some changes to your algorithm class before 'findPayloadBoundaries' is run, make them in this function.
 
virtual void boundaryFindingTearDown ()
 Put your algorithm back into a state ready for normal execution if you need to.
 
const std::vector< Calibration::ExpRun > & getRunList () const
 Get the list of runs for which calibration is called.
 
int getIteration () const
 Get current iteration.
 
std::vector< std::string > getVecInputFileNames () const
 Get the input file names used for this algorithm as a STL vector.
 
template<class T >
std::shared_ptr< T > getObjectPtr (const std::string &name, const std::vector< Calibration::ExpRun > &requestedRuns)
 Get calibration data object by name and list of runs, the Merge function will be called to generate the overall object.
 
template<class T >
std::shared_ptr< T > getObjectPtr (std::string name)
 Get calibration data object (for all runs the calibration is requested for) This function will only work during or after execute() has been called once.
 
template<>
shared_ptr< TTree > getObjectPtr (const string &name, const vector< ExpRun > &requestedRuns)
 We cheekily cast the TChain to TTree for the returned pointer so that the user never knows Hopefully this doesn't cause issues if people do low level stuff to the tree...
 
std::string getGranularityFromData () const
 Get the granularity of collected data.
 
void saveCalibration (TClonesArray *data, const std::string &name)
 Store DBArray payload with given name with default IOV.
 
void saveCalibration (TClonesArray *data, const std::string &name, const IntervalOfValidity &iov)
 Store DBArray with given name and custom IOV.
 
void saveCalibration (TObject *data)
 Store DB payload with default name and default IOV.
 
void saveCalibration (TObject *data, const IntervalOfValidity &iov)
 Store DB payload with default name and custom IOV.
 
void saveCalibration (TObject *data, const std::string &name)
 Store DB payload with given name with default IOV.
 
void saveCalibration (TObject *data, const std::string &name, const IntervalOfValidity &iov)
 Store DB payload with given name and custom IOV.
 
void updateDBObjPtrs (const unsigned int event, const int run, const int experiment)
 Updates any DBObjPtrs by calling update(event) for DBStore.
 
void setDescription (const std::string &description)
 Set algorithm description (in constructor)
 
void clearCalibrationData ()
 Clear calibration data.
 
Calibration::ExpRun getAllGranularityExpRun () const
 Returns the Exp,Run pair that means 'Everything'. Currently unused.
 
void resetInputJson ()
 Clears the m_inputJson member variable.
 
void resetOutputJson ()
 Clears the m_outputJson member variable.
 
template<class T >
void setOutputJsonValue (const std::string &key, const T &value)
 Set a key:value pair for the outputJson object, expected to used internally during calibrate()
 
template<class T >
const T getOutputJsonValue (const std::string &key) const
 Get a value using a key from the JSON output object, not sure why you would want to do this.
 
template<class T >
const T getInputJsonValue (const std::string &key) const
 Get an input JSON value using a key. The normal exceptions are raised when the key doesn't exist.
 
const nlohmann::json & getInputJsonObject () const
 Get the entire top level JSON object. We explicitly say this must be of object type so that we might pick.
 
bool inputJsonKeyExists (const std::string &key) const
 Test for a key in the input JSON object.
 

Protected Attributes

std::vector< Calibration::ExpRun > m_boundaries
 When using the boundaries functionality from isBoundaryRequired, this is used to store the boundaries. It is cleared when.
 

Private Member Functions

void createDebugHistogram ()
 Perform debug histogram file creation.
 
std::string getExpRunString (Calibration::ExpRun &expRun) const
 Gets the "exp.run" string repr. of (exp,run)
 
std::string getFullObjectPath (const std::string &name, Calibration::ExpRun expRun) const
 constructs the full TDirectory + Key name of an object in a TFile based on its name and exprun
 

Private Attributes

const unsigned short c_nVCells = 768
 Number of vCells of Belle II PXD sensors.
 
const unsigned short c_nUCells = 250
 Number of uCells of Belle II PXD sensors.
 
const unsigned short c_nDrains = 1000
 Number of drain lines of Belle II PXD sensors.
 
int m_debugHisto = false
 Set if a debugging histogram should be created in the algorithm output directory.
 
std::map< VxdID, double > m_medianOfHitsMap
 map of VxdID to median hits of each sensor
 
std::shared_ptr< TFile > m_file
 Pointer for TFile.
 
std::vector< std::string > m_inputFileNames
 List of input files to the Algorithm, will initially be user defined but then gets the wildcards expanded during execute()
 
std::map< Calibration::ExpRun, std::vector< std::string > > m_runsToInputFiles
 Map of Runs to input files. Gets filled when you call getRunRangeFromAllData, gets cleared when setting input files again.
 
std::string m_granularityOfData
 Granularity of input data. This only changes when the input files change so it isn't specific to an execution.
 
ExecutionData m_data
 Data specific to a SINGLE execution of the algorithm. Gets reset at the beginning of execution.
 
std::string m_description {""}
 Description of the algorithm.
 
std::string m_prefix {""}
 The name of the TDirectory the collector objects are contained within.
 
nlohmann::json m_jsonExecutionInput = nlohmann::json::object()
 Optional input JSON object used to make decisions about how to execute the algorithm code.
 
nlohmann::json m_jsonExecutionOutput = nlohmann::json::object()
 Optional output JSON object that can be set during the execution by the underlying algorithm code.
 

Static Private Attributes

static const Calibration::ExpRun m_allExpRun = make_pair(-1, -1)
 allExpRun
 

Detailed Description

Class implementing PXD hot pixel masking calibration algorithm.

Definition at line 22 of file PXDHotPixelMaskCalibrationAlgorithm.h.

Member Enumeration Documentation

◆ EResult

enum EResult
inherited

The result of calibration.

Enumerator
c_OK 

Finished successfully =0 in Python.

c_Iterate 

Needs iteration =1 in Python.

c_NotEnoughData 

Needs more data =2 in Python.

c_Failure 

Failed =3 in Python.

c_Undefined 

Not yet known (before execution) =4 in Python.

Definition at line 40 of file CalibrationAlgorithm.h.

40 {
41 c_OK,
42 c_Iterate,
44 c_Failure,
46 };
@ c_OK
Finished successfully =0 in Python.
@ c_Iterate
Needs iteration =1 in Python.
@ c_NotEnoughData
Needs more data =2 in Python.
@ c_Failure
Failed =3 in Python.
@ c_Undefined
Not yet known (before execution) =4 in Python.

Constructor & Destructor Documentation

◆ PXDHotPixelMaskCalibrationAlgorithm()

Constructor set the prefix to PXDHotPixelMaskCalibrationAlgorithm.

Definition at line 25 of file PXDHotPixelMaskCalibrationAlgorithm.cc.

25 : CalibrationAlgorithm("PXDHotPixelMaskCollector"),
28 deadPixelPayloadName("PXDDeadPixelPar"),
30{
32 " -------------------------- PXDHotPixelMak Calibration Algorithm ------------------------\n"
33 " \n"
34 " Algorithm which masks hot pixels with too large occupancy and dead pixels w/o no hits. \n"
35 " ----------------------------------------------------------------------------------------\n"
36 );
37}
Base class for calibration algorithms.
void setDescription(const std::string &description)
Set algorithm description (in constructor)
float inefficientPixelMultiplier
The occupancy threshold for inefficient (or dead) pixels is the median occupancy x inefficientPixelMu...
bool maskRows
Mask hot rows with too high average occupancy after single pixel masking.
float drainMultiplier
The occupancy threshold for masking hot drains is the median occupancy x drainMultiplier.
int minInefficientPixels
The minimum number of inefficient (or dead) pixels per row to define an inefficient row.
bool maskDrains
Mask hot drain lines with too high average occupancy after single pixel masking.
float pixelMultiplier
The occupancy threshold for masking hot single pixels is the median occupancy x pixelMultiplier.
float rowMultiplier
The occupancy threshold for masking hot rows is the median occupancy x rowMultiplier.
bool forceContinueMasking
Force continue masking in almost empty runs instead of returning c_NotEnoughData.
int minHits
Minimum median number of hits per pixel needed for dead pixel masking.
std::string deadPixelPayloadName
Payload name for PXDDeadPixelPar used for more defective pixels from damaged gates.

Member Function Documentation

◆ boundaryFindingSetup()

virtual void boundaryFindingSetup ( std::vector< Calibration::ExpRun >  ,
int   
)
inlineprotectedvirtualinherited

If you need to make some changes to your algorithm class before 'findPayloadBoundaries' is run, make them in this function.

Reimplemented in TestBoundarySettingAlgorithm, TestCalibrationAlgorithm, PXDAnalyticGainCalibrationAlgorithm, PXDValidationAlgorithm, SVD3SampleCoGTimeCalibrationAlgorithm, SVD3SampleELSTimeCalibrationAlgorithm, and SVDCoGTimeCalibrationAlgorithm.

Definition at line 252 of file CalibrationAlgorithm.h.

252{};

◆ boundaryFindingTearDown()

virtual void boundaryFindingTearDown ( )
inlineprotectedvirtualinherited

Put your algorithm back into a state ready for normal execution if you need to.

Definition at line 257 of file CalibrationAlgorithm.h.

257{};

◆ calibrate()

CalibrationAlgorithm::EResult calibrate ( )
overrideprotectedvirtual

Run algo on data.

Implements CalibrationAlgorithm.

Definition at line 93 of file PXDHotPixelMaskCalibrationAlgorithm.cc.

94{
95
96 auto collector_pxdhits = getObjectPtr<TH1I>("PXDHits");
97 auto collector_pxdhitcounts = getObjectPtr<TH1I>("PXDHitCounts");
98
99 // Check if there is any PXD hit
100 if (!collector_pxdhits) {
102 return c_OK;
103 else
104 return c_NotEnoughData;
105 }
106
107 // We should have some minimum number of events
108 auto nevents = collector_pxdhits->GetEntries();
109 if (nevents < minEvents) {
110 if (not forceContinueMasking) {
111 B2INFO("Not enough data: Only " << nevents << " events were collected!");
112 return c_NotEnoughData;
113 } else {
114 B2WARNING("Not enough data: Only " << nevents << " events were collected! The masking continous but the mask may be empty.");
115 }
116 }
117
118 B2RESULT("Found total of " << nevents << " events in collected data.");
119
120 // Get the total number of PXD hits and sensors
121 unsigned long long int nPXDHits = 0;
122 int nPXDSensors = 0;
123 for (auto sensBin = 1; sensBin <= collector_pxdhitcounts->GetXaxis()->GetNbins(); sensBin++) {
124 // The bin label is assumed to be a string representation of VxdID
125 string sensorDescr = collector_pxdhitcounts->GetXaxis()->GetBinLabel(sensBin);
126 VxdID id(sensorDescr);
127 //Increment number of sensors
128 nPXDSensors += 1;
129 // Increment number of of collected hits
130 unsigned long long int nSensorHits = collector_pxdhitcounts->GetBinContent(sensBin);
131 nPXDHits += nSensorHits;
132
133 B2RESULT("Number of hits for sensor sensor=" << id << " is " << nSensorHits);
134 }
135
136 // We should have enough hits in the PXD before we decide a single sensor is dead
137 unsigned long long int minPXDHits = minHits * nPXDSensors * c_nUCells * c_nVCells;
138 if (nPXDHits < minPXDHits) {
139 if (not forceContinueMasking) {
140 B2INFO("Not enough data: Only " << nPXDHits << " raw hits were collected!");
141 return c_NotEnoughData;
142 } else {
143 B2WARNING("Not enough data: Only " << nPXDHits << " raw hits were collected! The masking continous but the mask may be empty.");
144 }
145 }
146
147 B2RESULT("Found total of " << nPXDHits << " raw hits in collected data.");
148
149 // Check that the median number of hits is large enough
150 for (auto sensBin = 1; sensBin <= collector_pxdhitcounts->GetXaxis()->GetNbins(); sensBin++) {
151 // The bin label is assumed to be a string representation of VxdID
152 string sensorDescr = collector_pxdhitcounts->GetXaxis()->GetBinLabel(sensBin);
153 VxdID id(sensorDescr);
154
155 m_medianOfHitsMap[id] = 0.0;
156
157 // Get hitmap from collector
158 string name = str(format("PXD_%1%_PixelHitmap") % id.getID());
159 auto collector_pxdhitmap = getObjectPtr<TH1I>(name.c_str());
160
161 // Check if there was data collected for this sensor
162 if (collector_pxdhitmap == nullptr) continue;
163
164 // Compute the median number of hits to define a robust baseline for judging a channel fires too often
165 int nBins = collector_pxdhitmap->GetXaxis()->GetNbins();
166 double prob = 0.5;
167 vector<double> hitVec(nBins);
168
169 for (auto bin = 1; bin <= nBins; bin++) {
170 hitVec[bin - 1] = (double) collector_pxdhitmap->GetBinContent(bin);
171 }
172 double medianNumberOfHits;
173 TMath::Quantiles(nBins, 1, &hitVec[0], &medianNumberOfHits, &prob, kFALSE);
174 if (medianNumberOfHits <= 0) {
175 B2WARNING("Median number of hits per senor is smaller <1. Raise median to 1 instead.");
176 medianNumberOfHits = 1;
177 } else {
178 B2RESULT("Median of hits is " << medianNumberOfHits << " for sensor " << id);
179 }
180
181 // Keep the median of later use
182 m_medianOfHitsMap[id] = medianNumberOfHits;
183
184 // Check median number of hits is large enough
185 if (medianNumberOfHits < minHits) {
186 if (not forceContinueMasking) {
187 B2INFO("Not enough data: Median number of his is only " << medianNumberOfHits << "!");
188 return c_NotEnoughData;
189 } else {
190 B2WARNING("Not enough data: Median number of hits is only " << medianNumberOfHits <<
191 "! The masking continous but the mask may be empty.");
192 }
193 }
194 }
195
196 // This is the occupancy info payload for conditions DB
197 PXDOccupancyInfoPar* occupancyInfoPar = new PXDOccupancyInfoPar();
198
199 // This is the dead pixels payload for conditions DB
200 PXDDeadPixelPar* deadPixelsPar = new PXDDeadPixelPar();
201
202 // This is the hot pixel masking payload for conditions DB
203 PXDMaskedPixelPar* maskedPixelsPar = new PXDMaskedPixelPar();
204
205 // Remember the number of events, helps to judge the reliability of calibrations.
206 occupancyInfoPar->setNumberOfEvents(nevents);
207
208 // Compute the masks for all sensors
209 for (auto sensBin = 1; sensBin <= collector_pxdhitcounts->GetXaxis()->GetNbins(); sensBin++) {
210 // The bin label is assumed to be a string representation of VxdID
211 string sensorDescr = collector_pxdhitcounts->GetXaxis()->GetBinLabel(sensBin);
212 VxdID id(sensorDescr);
213
214 // If we reach here, a sensor with no hits is deemed dead
215 int nSensorHits = collector_pxdhitcounts->GetBinContent(sensBin);
216 if (nSensorHits == 0) {
217 deadPixelsPar->maskSensor(id.getID());
218 continue;
219 }
220
221 // Get hitmap from collector
222 string name = str(format("PXD_%1%_PixelHitmap") % id.getID());
223 auto collector_pxdhitmap = getObjectPtr<TH1I>(name.c_str());
224 if (collector_pxdhitmap == nullptr) {
225 B2WARNING("Cannot find PixelHitmap although there should be hits. This is strange!");
226 continue;
227 }
228
229 double medianNumberOfHits = m_medianOfHitsMap[id];
230 int nBins = collector_pxdhitmap->GetXaxis()->GetNbins();
231
232 // Dead pixel masking
233 if (medianNumberOfHits >= minHits) {
234
235 // Bookkeeping for masking of drains and rows
236 vector<int> hitsAlongRow(c_nVCells, 0);
237 vector<int> hitsAlongDrain(c_nDrains, 0);
238 vector<int> nDeadAlongRow(c_nVCells, 0);
239
240 // Mask all single pixels exceeding medianNumberOfHits x multiplier
241 double inefficientPixelHitThr = inefficientPixelMultiplier * medianNumberOfHits;
242 B2RESULT("Pixel hit threshold for dead rows is " << inefficientPixelHitThr << " for sensor " << id);
243 // Accumulate hits along drains and rows
244 for (auto bin = 1; bin <= nBins; bin++) {
245 // Find the current pixel cell
246 int pixID = bin - 1;
247 int uCell = pixID / c_nVCells;
248 int vCell = pixID % c_nVCells;
249 int drainID = uCell * 4 + vCell % 4;
250 int nhits = collector_pxdhitmap->GetBinContent(bin);
251 hitsAlongDrain[drainID] += nhits;
252 hitsAlongRow[vCell] += nhits;
253 if (inefficientPixelMultiplier > 0 && nhits < inefficientPixelHitThr)
254 nDeadAlongRow[vCell] += 1;
255 }
256
257 // Dead row masking
258 B2INFO("Entering masking of dead rows ...");
259 for (auto vCell = 0; vCell < c_nVCells; vCell++) {
260 // Get number of hits per row
261 int nhits = hitsAlongRow[vCell];
262 int nDeadPixels = nDeadAlongRow[vCell];
263 // Mask dead row
264 if (nhits == 0 || nDeadPixels >= minInefficientPixels) {
265 deadPixelsPar->maskRow(id.getID(), vCell);
266 B2RESULT("Dead row with vCell=" << vCell << " on sensor " << id);
267 }
268
269 }
270
271 // Dead drain masking
272 B2INFO("Entering masking of dead drains ...");
273 for (auto drainID = 0; drainID < c_nDrains; drainID++) {
274 // Get number of hits per drain
275 int nhits = hitsAlongDrain[drainID];
276 // Mask dead drain
277 if (nhits == 0) {
278 deadPixelsPar->maskDrain(id.getID(), drainID);
279 B2RESULT("Dead drain line at drainID=" << drainID << " on sensor " << id);
280 }
281 }
282
283 // Dead pixel masking
284 B2INFO("Entering masking of single dead pixels ...");
285 for (auto bin = 1; bin <= nBins; bin++) {
286 // First, we mask single pixels exceeding hit threshold
287 int nhits = collector_pxdhitmap->GetBinContent(bin);
288 int pixID = bin - 1;
289 int uCell = pixID / c_nVCells;
290 int vCell = pixID % c_nVCells;
291 int drainID = uCell * 4 + vCell % 4;
292 // Mask dead pixel
293 if (nhits == 0 && !deadPixelsPar->isDeadRow(id.getID(), vCell) && !deadPixelsPar->isDeadDrain(id.getID(), drainID)) {
294 // This pixel is dead, we have to mask it
295 deadPixelsPar->maskSinglePixel(id.getID(), pixID);
296 B2RESULT("Dead single pixel with ucell=" << uCell << ", vcell=" << vCell << " on sensor " << id);
297 }
298 }
299 }
300
301 // Hot pixel masking
302
303 // Bookkeeping for masking hot drains
304 vector<float> unmaskedHitsAlongDrain(c_nDrains, 0);
305 vector<int> unmaskedCellsAlongDrain(c_nDrains, 0);
306
307 // Bookkeeping for maskign hot rows
308 vector<float> unmaskedHitsAlongRow(c_nVCells, 0);
309 vector<int> unmaskedCellsAlongRow(c_nVCells, 0);
310
311 // Mask all single pixels exceeding medianNumberOfHits x multiplier
312 double pixelHitThr = pixelMultiplier * medianNumberOfHits;
313 B2RESULT("Pixel hit threshold is " << pixelHitThr << " for sensor " << id);
314
315 // Mask all hot pixel for this sensor
316 for (auto bin = 1; bin <= nBins; bin++) {
317 // Find the current pixel cell
318 int pixID = bin - 1;
319 int uCell = pixID / c_nVCells;
320 int vCell = pixID % c_nVCells;
321 int drainID = uCell * 4 + vCell % 4;
322
323 // First, we mask single pixels exceeding hit threshold
324 float nhits = collector_pxdhitmap->GetBinContent(bin);
325 bool masked = false;
326
327 if (nhits > pixelHitThr) {
328 // This pixel is hot, we have to mask it
329 maskedPixelsPar->maskSinglePixel(id.getID(), pixID);
330 masked = true;
331 B2RESULT("Masking single pixel with ucell=" << uCell << ", vcell=" << vCell << " on sensor " << id);
332 }
333
334 // Then we accumulate hits along u and v direction for unmasked
335 // pixels
336 if (not masked) {
337 ++unmaskedCellsAlongDrain[drainID];
338 unmaskedHitsAlongDrain[drainID] += nhits;
339 ++unmaskedCellsAlongRow[vCell];
340 unmaskedHitsAlongRow[vCell] += nhits;
341 }
342 }
343
344 if (maskDrains) {
345 double drainHitThr = drainMultiplier * medianNumberOfHits;
346 B2RESULT("Drain hit threshold is " << drainHitThr << " for sensor " << id);
347
348 for (auto drainID = 0; drainID < c_nDrains; drainID++) {
349 if (unmaskedCellsAlongDrain[drainID] > 0) {
350 // Compute average number of hits per drain
351 float nhits = unmaskedHitsAlongDrain[drainID] / unmaskedCellsAlongDrain[drainID];
352 // Mask residual hot drain
353 if (nhits > drainHitThr) {
354 for (auto iGate = 0; iGate < 192; iGate++) {
355 int uCell = drainID / 4;
356 int vCell = drainID % 4 + iGate * 4;
357 maskedPixelsPar->maskSinglePixel(id.getID(), uCell * c_nVCells + vCell);
358 }
359 B2RESULT("Masking drain line with drainID=" << drainID << " on sensor " << id);
360 }
361 }
362 }
363 }
364
365 if (maskRows) {
366 double rowHitThr = rowMultiplier * medianNumberOfHits;
367 B2RESULT("Row hit threshold is " << rowHitThr << " for sensor " << id);
368
369 for (auto vCell = 0; vCell < c_nVCells; vCell++) {
370 if (unmaskedCellsAlongRow[vCell] > 0) {
371 // Compute average number of hits per row
372 float nhits = unmaskedHitsAlongRow[vCell] / unmaskedCellsAlongRow[vCell];
373 // Mask residual hot row
374 if (nhits > rowHitThr) {
375 for (auto uCell = 0; uCell < c_nUCells; uCell++)
376 maskedPixelsPar->maskSinglePixel(id.getID(), uCell * c_nVCells + vCell);
377
378 B2RESULT("Masking complete row with vCell=" << vCell << " on sensor " << id);
379 }
380 }
381 }
382 }
383
384 // After the masking is done, we compute the average sensor occupancy after
385 // hot pixel masking.
386
387 // Count all unmasked hits
388 int numberOfUnmaskedHits = 0;
389
390 for (auto bin = 1; bin <= nBins; bin++) {
391 // Find the current pixel cell
392 int pixID = bin - 1;
393
394 if (maskedPixelsPar->pixelOK(id.getID(), pixID)) {
395 numberOfUnmaskedHits += collector_pxdhitmap->GetBinContent(bin);
396 }
397 }
398
399 // Compute mean occupancy before masking
400 float meanOccupancyAfterMasking = (float)numberOfUnmaskedHits / nevents / c_nVCells / c_nUCells;
401 B2RESULT("Hotpixel filtered occupancy sensor=" << id << " is " << meanOccupancyAfterMasking);
402
403 occupancyInfoPar->setOccupancy(id.getID(), meanOccupancyAfterMasking);
404 }
405
406 for (auto elem : maskedPixelsPar->getMaskedPixelMap()) {
407 auto id = elem.first;
408 auto singles = elem.second;
409 B2RESULT("SensorID " << VxdID(id) << " has filtered occupancy of " << occupancyInfoPar->getOccupancy(id));
410 B2RESULT("SensorID " << VxdID(id) << " has fraction of masked pixels of " << (float)singles.size() / (c_nVCells * c_nUCells));
411 }
412
413 // Save the hot pixel mask to database. Note that this will set the database object name to the same as the collector but you
414 // are free to change it.
415 saveCalibration(maskedPixelsPar, "PXDMaskedPixelPar");
417 saveCalibration(occupancyInfoPar, "PXDOccupancyInfoPar");
418
419 // Create debugging histo if we asked for it
421
422 B2INFO("PXDHotPixelMask Calibration Successful");
423 return c_OK;
424}
void saveCalibration(TClonesArray *data, const std::string &name)
Store DBArray payload with given name with default IOV.
The payload telling which PXD pixel is dead (=Readout system does not receive signals)
void maskSensor(unsigned short sensorID)
Mask sensor.
void maskRow(unsigned short sensorID, unsigned int vCellID)
Mask single row.
void maskSinglePixel(unsigned short sensorID, unsigned int pixID)
Mask single pixel.
void maskDrain(unsigned short sensorID, unsigned int drainID)
Mask single drain.
bool isDeadDrain(unsigned short sensorID, unsigned int drainID) const
Check whether a drain is dead.
bool isDeadRow(unsigned short sensorID, unsigned int vCellID) const
Check whether a row is dead.
std::map< VxdID, double > m_medianOfHitsMap
map of VxdID to median hits of each sensor
const unsigned short c_nDrains
Number of drain lines of Belle II PXD sensors.
const unsigned short c_nVCells
Number of vCells of Belle II PXD sensors.
int m_debugHisto
Set if a debugging histogram should be created in the algorithm output directory.
const unsigned short c_nUCells
Number of uCells of Belle II PXD sensors.
void createDebugHistogram()
Perform debug histogram file creation.
The payload telling which PXD pixel to mask (ignore)
bool pixelOK(unsigned short sensorID, unsigned int pixID) const
Check whether a pixel on a given sensor is OK or not.
const std::unordered_map< unsigned short, MaskedSinglePixelsSet > & getMaskedPixelMap() const
Return unordered_map with all masked single pixels in PXD.
void maskSinglePixel(unsigned short sensorID, unsigned int pixID)
Mask single pixel.
The payload collecting some meta information from running the masking algorithm.
void setOccupancy(unsigned short sensorID, float occupancy)
Set occupancy.
void setNumberOfEvents(int nEvents)
Set number of events used for occupancy estimation.
float getOccupancy(unsigned short sensorID) const
Get occupancy.
Class to uniquely identify a any structure of the PXD and SVD.
Definition: VxdID.h:33
int getID(const std::vector< double > &breaks, double t)
get id of the time point t
Definition: calibTools.h:60

◆ checkPyExpRun()

bool checkPyExpRun ( PyObject *  pyObj)
inherited

Checks that a PyObject can be successfully converted to an ExpRun type.

Checks if the PyObject can be converted to ExpRun.

Definition at line 28 of file CalibrationAlgorithm.cc.

29{
30 // Is it a sequence?
31 if (PySequence_Check(pyObj)) {
32 Py_ssize_t nObj = PySequence_Length(pyObj);
33 // Does it have 2 objects in it?
34 if (nObj != 2) {
35 B2DEBUG(29, "ExpRun was a Python sequence which didn't have exactly 2 entries!");
36 return false;
37 }
38 PyObject* item1, *item2;
39 item1 = PySequence_GetItem(pyObj, 0);
40 item2 = PySequence_GetItem(pyObj, 1);
41 // Did the GetItem work?
42 if ((item1 == NULL) || (item2 == NULL)) {
43 B2DEBUG(29, "A PyObject pointer was NULL in the sequence");
44 return false;
45 }
46 // Are they longs?
47 if (PyLong_Check(item1) && PyLong_Check(item2)) {
48 long value1, value2;
49 value1 = PyLong_AsLong(item1);
50 value2 = PyLong_AsLong(item2);
51 if (((value1 == -1) || (value2 == -1)) && PyErr_Occurred()) {
52 B2DEBUG(29, "An error occurred while converting the PyLong to long");
53 return false;
54 }
55 } else {
56 B2DEBUG(29, "One or more of the PyObjects in the ExpRun wasn't a long");
57 return false;
58 }
59 // Make sure to kill off the reference GetItem gave us responsibility for
60 Py_DECREF(item1);
61 Py_DECREF(item2);
62 } else {
63 B2DEBUG(29, "ExpRun was not a Python sequence.");
64 return false;
65 }
66 return true;
67}

◆ clearCalibrationData()

void clearCalibrationData ( )
inlineprotectedinherited

Clear calibration data.

Definition at line 324 of file CalibrationAlgorithm.h.

void clearCalibrationData()
Clear calibration data.
ExecutionData m_data
Data specific to a SINGLE execution of the algorithm. Gets reset at the beginning of execution.

◆ commit() [1/2]

bool commit ( )
inherited

Submit constants from last calibration into database.

Definition at line 302 of file CalibrationAlgorithm.cc.

303{
304 if (getPayloads().empty())
305 return false;
306 list<Database::DBImportQuery> payloads = getPayloads();
307 B2INFO("Committing " << payloads.size() << " payloads to database.");
308 return Database::Instance().storeData(payloads);
309}
std::list< Database::DBImportQuery > & getPayloads()
Get constants (in TObjects) for database update from last execution.
static Database & Instance()
Instance of a singleton Database.
Definition: Database.cc:42
bool storeData(const std::string &name, TObject *object, const IntervalOfValidity &iov)
Store an object in the database.
Definition: Database.cc:141

◆ commit() [2/2]

bool commit ( std::list< Database::DBImportQuery payloads)
inherited

Submit constants from a (potentially previous) set of payloads.

Definition at line 311 of file CalibrationAlgorithm.cc.

312{
313 if (payloads.empty())
314 return false;
315 return Database::Instance().storeData(payloads);
316}

◆ convertPyExpRun()

ExpRun convertPyExpRun ( PyObject *  pyObj)
inherited

Performs the conversion of PyObject to ExpRun.

Converts the PyObject to an ExpRun. We've preoviously checked the object so this assumes a lot about the PyObject.

Definition at line 70 of file CalibrationAlgorithm.cc.

71{
72 ExpRun expRun;
73 PyObject* itemExp, *itemRun;
74 itemExp = PySequence_GetItem(pyObj, 0);
75 itemRun = PySequence_GetItem(pyObj, 1);
76 expRun.first = PyLong_AsLong(itemExp);
77 Py_DECREF(itemExp);
78 expRun.second = PyLong_AsLong(itemRun);
79 Py_DECREF(itemRun);
80 return expRun;
81}
Struct containing exp number and run number.
Definition: Splitter.h:51

◆ createDebugHistogram()

void createDebugHistogram ( )
private

Perform debug histogram file creation.

Definition at line 39 of file PXDHotPixelMaskCalibrationAlgorithm.cc.

40{
41 // Run list of current IoV and all runs to be calibrated
42 auto expRuns = getRunList();
43 auto expRunsAll = getRunListFromAllData();
44
45 // Save the current directory to change back later
46 TDirectory* currentDir = gDirectory;
47
48 // Create TFile if not exist
49 if (!m_file) {
50 std::string fileName = (this->getPrefix()) + "debug.root";
51 B2INFO("Creating file " << fileName);
52 m_file = std::make_shared<TFile>(fileName.c_str(), "RECREATE");
53 }
54
55 //m_file->cd();
56 string iov_str = str(format("E%1%_R%2%_E%3%_R%4%") % expRuns.front().first
57 % expRuns.front().second % expRuns.back().first % expRuns.back().second);
58 m_file->mkdir(iov_str.c_str());
59 m_file->cd(iov_str.c_str());
60
61 // Collector info
62 auto collector_pxdhits = getObjectPtr<TH1I>("PXDHits");
63 auto collector_pxdhitcounts = getObjectPtr<TH1I>("PXDHitCounts");
64
65 auto nevents = collector_pxdhits->GetEntries();
66
67 for (auto sensBin = 1; sensBin <= collector_pxdhitcounts->GetXaxis()->GetNbins(); sensBin++) {
68 string sensorDescr = collector_pxdhitcounts->GetXaxis()->GetBinLabel(sensBin);
69 VxdID id(sensorDescr);
70
71 // Get hitmap from collector
72 string name = str(format("PXD_%1%_PixelHitmap") % id.getID());
73 auto collector_pxdhitmap = getObjectPtr<TH1I>(name.c_str());
74 if (collector_pxdhitmap == nullptr) {
75 continue;
76 }
77 TH1I* debugHisto = new TH1I(*collector_pxdhitmap);
78 // Set overflow to total number of events
79 debugHisto->SetBinContent(debugHisto->GetNbinsX() + 1, nevents);
80 debugHisto->Write();
81 delete debugHisto;
82 }
83
84 // Close TFile
85 if (expRuns.back() == expRunsAll.back()) {
86 B2INFO("Reached Final ExpRun: (" << expRuns.back().first << ", " << expRuns.back().second << ")");
87 m_file->Close();
88
89 }
90 currentDir->cd();
91}
std::vector< Calibration::ExpRun > getRunListFromAllData() const
Get the complete list of runs from inspection of collected data.
const std::vector< Calibration::ExpRun > & getRunList() const
Get the list of runs for which calibration is called.
std::string getPrefix() const
Get the prefix used for getting calibration data.
std::shared_ptr< TFile > m_file
Pointer for TFile.

◆ dumpOutputJson()

const std::string dumpOutputJson ( ) const
inlineinherited

Dump the JSON string of the output JSON object.

Definition at line 223 of file CalibrationAlgorithm.h.

223{return m_jsonExecutionOutput.dump();}
nlohmann::json m_jsonExecutionOutput
Optional output JSON object that can be set during the execution by the underlying algorithm code.

◆ execute() [1/2]

CalibrationAlgorithm::EResult execute ( PyObject *  runs,
int  iteration = 0,
IntervalOfValidity  iov = IntervalOfValidity() 
)
inherited

Runs calibration over Python list of runs. Converts to C++ and then calls the other execute() function.

Definition at line 83 of file CalibrationAlgorithm.cc.

84{
85 B2DEBUG(29, "Running execute() using Python Object as input argument");
86 // Reset the execution specific data in case the algorithm was previously called
87 m_data.reset();
88 m_data.setIteration(iteration);
89 vector<ExpRun> vecRuns;
90 // Is it a list?
91 if (PySequence_Check(runs)) {
92 boost::python::handle<> handle(boost::python::borrowed(runs));
93 boost::python::list listRuns(handle);
94
95 int nList = boost::python::len(listRuns);
96 for (int iList = 0; iList < nList; ++iList) {
97 boost::python::object pyExpRun(listRuns[iList]);
98 if (!checkPyExpRun(pyExpRun.ptr())) {
99 B2ERROR("Received Python ExpRuns couldn't be converted to C++");
101 return c_Failure;
102 } else {
103 vecRuns.push_back(convertPyExpRun(pyExpRun.ptr()));
104 }
105 }
106 } else {
107 B2ERROR("Tried to set the input runs but we didn't receive a Python sequence object (list,tuple).");
109 return c_Failure;
110 }
111 return execute(vecRuns, iteration, iov);
112}
void setResult(EResult result)
Setter for current iteration.
void setIteration(int iteration)
Setter for current iteration.
void reset()
Resets this class back to what is needed at the beginning of an execution.
bool checkPyExpRun(PyObject *pyObj)
Checks that a PyObject can be successfully converted to an ExpRun type.
EResult execute(std::vector< Calibration::ExpRun > runs={}, int iteration=0, IntervalOfValidity iov=IntervalOfValidity())
Runs calibration over vector of runs for a given iteration.
Calibration::ExpRun convertPyExpRun(PyObject *pyObj)
Performs the conversion of PyObject to ExpRun.

◆ execute() [2/2]

CalibrationAlgorithm::EResult execute ( std::vector< Calibration::ExpRun >  runs = {},
int  iteration = 0,
IntervalOfValidity  iov = IntervalOfValidity() 
)
inherited

Runs calibration over vector of runs for a given iteration.

You can also specify the IoV to save the database payload as. By default the Algorithm will create an IoV from your requested ExpRuns, or from the overall ExpRuns of the input data if you haven't specified ExpRuns in this function.

No checks are performed to make sure that a IoV you specify matches the data you ran over, it simply labels the IoV to commit to the database later.

Definition at line 114 of file CalibrationAlgorithm.cc.

115{
116 // Check if we are calling this function directly and need to reset, or through Python where it was already done.
117 if (m_data.getResult() != c_Undefined) {
118 m_data.reset();
119 m_data.setIteration(iteration);
120 }
121
122 if (m_inputFileNames.empty()) {
123 B2ERROR("There aren't any input files set. Please use CalibrationAlgorithm::setInputFiles()");
125 return c_Failure;
126 }
127
128 // Did we receive runs to execute over explicitly?
129 if (!(runs.empty())) {
130 for (auto expRun : runs) {
131 B2DEBUG(29, "ExpRun requested = (" << expRun.first << ", " << expRun.second << ")");
132 }
133 // We've asked explicitly for certain runs, but we should check if the data granularity is 'run'
134 if (strcmp(getGranularity().c_str(), "all") == 0) {
135 B2ERROR(("The data is collected with granularity=all (exp=-1,run=-1), but you seem to request calibration for specific runs."
136 " We'll continue but using ALL the input data given instead of the specific runs requested."));
137 }
138 } else {
139 // If no runs are provided, infer the runs from all collected data
140 runs = getRunListFromAllData();
141 // Let's check that we have some now
142 if (runs.empty()) {
143 B2ERROR("No collected data in input files.");
145 return c_Failure;
146 }
147 for (auto expRun : runs) {
148 B2DEBUG(29, "ExpRun requested = (" << expRun.first << ", " << expRun.second << ")");
149 }
150 }
151
153 if (iov.empty()) {
154 // If no user specified IoV we use the IoV from the executed run list
155 iov = IntervalOfValidity(runs[0].first, runs[0].second, runs[runs.size() - 1].first, runs[runs.size() - 1].second);
156 }
158 // After here, the getObject<...>(...) helpers start to work
159
161 m_data.setResult(result);
162 return result;
163}
void setRequestedIov(const IntervalOfValidity &iov=IntervalOfValidity(0, 0, -1, -1))
Sets the requested IoV for this execution, based on the.
void setRequestedRuns(const std::vector< Calibration::ExpRun > &requestedRuns)
Sets the vector of ExpRuns.
EResult getResult() const
Getter for current result.
std::vector< std::string > m_inputFileNames
List of input files to the Algorithm, will initially be user defined but then gets the wildcards expa...
EResult
The result of calibration.
virtual EResult calibrate()=0
Run algo on data - pure virtual: needs to be implemented.
std::string getGranularity() const
Get the granularity of collected data.
A class that describes the interval of experiments/runs for which an object in the database is valid.

◆ fillRunToInputFilesMap()

void fillRunToInputFilesMap ( )
inherited

Fill the mapping of ExpRun -> Files.

Definition at line 330 of file CalibrationAlgorithm.cc.

331{
332 m_runsToInputFiles.clear();
333 // Save TDirectory to change back at the end
334 TDirectory* dir = gDirectory;
335 RunRange* runRange;
336 // Construct the TDirectory name where we expect our objects to be
337 string runRangeObjName(getPrefix() + "/" + RUN_RANGE_OBJ_NAME);
338 for (const auto& fileName : m_inputFileNames) {
339 //Open TFile to get the objects
340 unique_ptr<TFile> f;
341 f.reset(TFile::Open(fileName.c_str(), "READ"));
342 runRange = dynamic_cast<RunRange*>(f->Get(runRangeObjName.c_str()));
343 if (runRange) {
344 // Insert or extend the run -> file mapping for this ExpRun
345 auto expRuns = runRange->getExpRunSet();
346 for (const auto& expRun : expRuns) {
347 auto runFiles = m_runsToInputFiles.find(expRun);
348 if (runFiles != m_runsToInputFiles.end()) {
349 (runFiles->second).push_back(fileName);
350 } else {
351 m_runsToInputFiles.insert(std::make_pair(expRun, std::vector<std::string> {fileName}));
352 }
353 }
354 } else {
355 B2WARNING("Missing a RunRange object for file: " << fileName);
356 }
357 }
358 dir->cd();
359}
std::map< Calibration::ExpRun, std::vector< std::string > > m_runsToInputFiles
Map of Runs to input files. Gets filled when you call getRunRangeFromAllData, gets cleared when setti...
Mergeable object holding (unique) set of (exp,run) pairs.
Definition: RunRange.h:25
const std::set< Calibration::ExpRun > & getExpRunSet()
Get access to the stored set.
Definition: RunRange.h:64

◆ findPayloadBoundaries()

const std::vector< ExpRun > findPayloadBoundaries ( std::vector< Calibration::ExpRun >  runs,
int  iteration = 0 
)
inherited

Used to discover the ExpRun boundaries that you want the Python CAF to execute on. This is optional and only used in some.

Definition at line 520 of file CalibrationAlgorithm.cc.

521{
522 m_boundaries.clear();
523 if (m_inputFileNames.empty()) {
524 B2ERROR("There aren't any input files set. Please use CalibrationAlgorithm::setInputFiles()");
525 return m_boundaries;
526 }
527 // Reset the internal execution data just in case something is hanging around
528 m_data.reset();
529 if (runs.empty()) {
530 // Want to loop over all runs we could possibly know about
531 runs = getRunListFromAllData();
532 }
533 // Let's check that we have some now
534 if (runs.empty()) {
535 B2ERROR("No collected data in input files.");
536 return m_boundaries;
537 }
538 // In order to find run boundaries we must have collected with data granularity == 'run'
539 if (strcmp(getGranularity().c_str(), "all") == 0) {
540 B2ERROR("The data is collected with granularity='all' (exp=-1,run=-1), and we can't use that to find run boundaries.");
541 return m_boundaries;
542 }
543 m_data.setIteration(iteration);
544 // User defined setup function
545 boundaryFindingSetup(runs, iteration);
546 std::vector<ExpRun> runList;
547 // Loop over run list and call derived class "isBoundaryRequired" member function
548 for (auto currentRun : runs) {
549 runList.push_back(currentRun);
550 m_data.setRequestedRuns(runList);
551 // After here, the getObject<...>(...) helpers start to work
552 if (isBoundaryRequired(currentRun)) {
553 m_boundaries.push_back(currentRun);
554 }
555 // Only want run-by-run
556 runList.clear();
557 // Don't want memory hanging around
559 }
560 m_data.reset();
562 return m_boundaries;
563}
std::vector< Calibration::ExpRun > m_boundaries
When using the boundaries functionality from isBoundaryRequired, this is used to store the boundaries...
virtual void boundaryFindingTearDown()
Put your algorithm back into a state ready for normal execution if you need to.
virtual void boundaryFindingSetup(std::vector< Calibration::ExpRun >, int)
If you need to make some changes to your algorithm class before 'findPayloadBoundaries' is run,...
virtual bool isBoundaryRequired(const Calibration::ExpRun &)
Given the current collector data, make a decision about whether or not this run should be the start o...

◆ getAllGranularityExpRun()

Calibration::ExpRun getAllGranularityExpRun ( ) const
inlineprotectedinherited

Returns the Exp,Run pair that means 'Everything'. Currently unused.

Definition at line 327 of file CalibrationAlgorithm.h.

327{return m_allExpRun;}
static const Calibration::ExpRun m_allExpRun
allExpRun

◆ getCollectorName()

std::string getCollectorName ( ) const
inlineinherited

Alias for prefix.

For convenience and less writing, we say developers to set this to default collector module name in constructor of base class. One can however use the dublets of collector+algorithm multiple times with different settings. To bind these together correctly, the prefix has to be set the same for algo and collector. So we call the setter setPrefix rather than setModuleName or whatever. This getter will work out of the box for default cases -> return the name of module you have to add to your path to collect data for this algorithm.

Definition at line 164 of file CalibrationAlgorithm.h.

164{return getPrefix();}

◆ getDescription()

const std::string & getDescription ( ) const
inlineinherited

Get the description of the algorithm (set by developers in constructor)

Definition at line 216 of file CalibrationAlgorithm.h.

216{return m_description;}
std::string m_description
Description of the algorithm.

◆ getExpRunString()

string getExpRunString ( Calibration::ExpRun &  expRun) const
privateinherited

Gets the "exp.run" string repr. of (exp,run)

Definition at line 254 of file CalibrationAlgorithm.cc.

255{
256 string expRunString;
257 expRunString += to_string(expRun.first);
258 expRunString += ".";
259 expRunString += to_string(expRun.second);
260 return expRunString;
261}

◆ getFullObjectPath()

string getFullObjectPath ( const std::string &  name,
Calibration::ExpRun  expRun 
) const
privateinherited

constructs the full TDirectory + Key name of an object in a TFile based on its name and exprun

Definition at line 263 of file CalibrationAlgorithm.cc.

264{
265 string dirName = getPrefix() + "/" + name;
266 string objName = name + "_" + getExpRunString(expRun);
267 return dirName + "/" + objName;
268}
std::string getExpRunString(Calibration::ExpRun &expRun) const
Gets the "exp.run" string repr. of (exp,run)

◆ getGranularity()

std::string getGranularity ( ) const
inlineinherited

Get the granularity of collected data.

Definition at line 188 of file CalibrationAlgorithm.h.

188{return m_granularityOfData;};
std::string m_granularityOfData
Granularity of input data. This only changes when the input files change so it isn't specific to an e...

◆ getGranularityFromData()

string getGranularityFromData ( ) const
protectedinherited

Get the granularity of collected data.

Definition at line 383 of file CalibrationAlgorithm.cc.

384{
385 // Save TDirectory to change back at the end
386 TDirectory* dir = gDirectory;
387 RunRange* runRange;
388 string runRangeObjName(getPrefix() + "/" + RUN_RANGE_OBJ_NAME);
389 // We only check the first file
390 string fileName = m_inputFileNames[0];
391 unique_ptr<TFile> f;
392 f.reset(TFile::Open(fileName.c_str(), "READ"));
393 runRange = dynamic_cast<RunRange*>(f->Get(runRangeObjName.c_str()));
394 if (!runRange) {
395 B2FATAL("The input file " << fileName << " does not contain a RunRange object at "
396 << runRangeObjName << ". Please set your input files to exclude it.");
397 return "";
398 }
399 string granularity = runRange->getGranularity();
400 dir->cd();
401 return granularity;
402}
std::string getGranularity() const
Gets the m_granularity.
Definition: RunRange.h:110

◆ getInputFileNames()

PyObject * getInputFileNames ( )
inherited

Get the input file names used for this algorithm and pass them out as a Python list of unicode strings.

Definition at line 245 of file CalibrationAlgorithm.cc.

246{
247 PyObject* objInputFileNames = PyList_New(m_inputFileNames.size());
248 for (size_t i = 0; i < m_inputFileNames.size(); ++i) {
249 PyList_SetItem(objInputFileNames, i, Py_BuildValue("s", m_inputFileNames[i].c_str()));
250 }
251 return objInputFileNames;
252}

◆ getInputJsonObject()

const nlohmann::json & getInputJsonObject ( ) const
inlineprotectedinherited

Get the entire top level JSON object. We explicitly say this must be of object type so that we might pick.

Definition at line 357 of file CalibrationAlgorithm.h.

357{return m_jsonExecutionInput;}
nlohmann::json m_jsonExecutionInput
Optional input JSON object used to make decisions about how to execute the algorithm code.

◆ getInputJsonValue()

const T getInputJsonValue ( const std::string &  key) const
inlineprotectedinherited

Get an input JSON value using a key. The normal exceptions are raised when the key doesn't exist.

Definition at line 350 of file CalibrationAlgorithm.h.

351 {
352 return m_jsonExecutionInput.at(key);
353 }

◆ getIovFromAllData()

IntervalOfValidity getIovFromAllData ( ) const
inherited

Get the complete IoV from inspection of collected data.

Definition at line 325 of file CalibrationAlgorithm.cc.

326{
328}
RunRange getRunRangeFromAllData() const
Get the complete RunRange from inspection of collected data.
IntervalOfValidity getIntervalOfValidity()
Make IntervalOfValidity from the set, spanning all runs. Works because sets are sorted by default.
Definition: RunRange.h:70

◆ getIteration()

int getIteration ( ) const
inlineprotectedinherited

Get current iteration.

Definition at line 269 of file CalibrationAlgorithm.h.

269{ return m_data.getIteration(); }
int getIteration() const
Getter for current iteration.

◆ getObjectPtr()

std::shared_ptr< T > getObjectPtr ( std::string  name)
inlineprotectedinherited

Get calibration data object (for all runs the calibration is requested for) This function will only work during or after execute() has been called once.

Definition at line 285 of file CalibrationAlgorithm.h.

286 {
287 if (m_runsToInputFiles.size() == 0)
289 return getObjectPtr<T>(name, m_data.getRequestedRuns());
290 }
const std::vector< Calibration::ExpRun > & getRequestedRuns() const
Returns the vector of ExpRuns.
void fillRunToInputFilesMap()
Fill the mapping of ExpRun -> Files.

◆ getOutputJsonValue()

const T getOutputJsonValue ( const std::string &  key) const
inlineprotectedinherited

Get a value using a key from the JSON output object, not sure why you would want to do this.

Definition at line 342 of file CalibrationAlgorithm.h.

343 {
344 return m_jsonExecutionOutput.at(key);
345 }

◆ getPayloads()

std::list< Database::DBImportQuery > & getPayloads ( )
inlineinherited

Get constants (in TObjects) for database update from last execution.

Definition at line 204 of file CalibrationAlgorithm.h.

204{return m_data.getPayloads();}
std::list< Database::DBImportQuery > & getPayloads()
Get constants (in TObjects) for database update from last calibration.

◆ getPayloadValues()

std::list< Database::DBImportQuery > getPayloadValues ( )
inlineinherited

Get constants (in TObjects) for database update from last execution but passed by VALUE.

Definition at line 207 of file CalibrationAlgorithm.h.

207{return m_data.getPayloadValues();}
std::list< Database::DBImportQuery > getPayloadValues()
Get constants (in TObjects) for database update from last calibration but passed by VALUE.

◆ getPrefix()

std::string getPrefix ( ) const
inlineinherited

Get the prefix used for getting calibration data.

Definition at line 146 of file CalibrationAlgorithm.h.

146{return m_prefix;}
std::string m_prefix
The name of the TDirectory the collector objects are contained within.

◆ getRunList()

const std::vector< Calibration::ExpRun > & getRunList ( ) const
inlineprotectedinherited

Get the list of runs for which calibration is called.

Definition at line 266 of file CalibrationAlgorithm.h.

266{return m_data.getRequestedRuns();}

◆ getRunListFromAllData()

vector< ExpRun > getRunListFromAllData ( ) const
inherited

Get the complete list of runs from inspection of collected data.

Definition at line 318 of file CalibrationAlgorithm.cc.

319{
320 RunRange runRange = getRunRangeFromAllData();
321 set<ExpRun> expRunSet = runRange.getExpRunSet();
322 return vector<ExpRun>(expRunSet.begin(), expRunSet.end());
323}

◆ getRunRangeFromAllData()

RunRange getRunRangeFromAllData ( ) const
inherited

Get the complete RunRange from inspection of collected data.

Definition at line 361 of file CalibrationAlgorithm.cc.

362{
363 // Save TDirectory to change back at the end
364 TDirectory* dir = gDirectory;
365 RunRange runRange;
366 // Construct the TDirectory name where we expect our objects to be
367 string runRangeObjName(getPrefix() + "/" + RUN_RANGE_OBJ_NAME);
368 for (const auto& fileName : m_inputFileNames) {
369 //Open TFile to get the objects
370 unique_ptr<TFile> f;
371 f.reset(TFile::Open(fileName.c_str(), "READ"));
372 RunRange* runRangeOther = dynamic_cast<RunRange*>(f->Get(runRangeObjName.c_str()));
373 if (runRangeOther) {
374 runRange.merge(runRangeOther);
375 } else {
376 B2WARNING("Missing a RunRange object for file: " << fileName);
377 }
378 }
379 dir->cd();
380 return runRange;
381}
virtual void merge(const RunRange *other)
Implementation of merging - other is added to the set (union)
Definition: RunRange.h:52

◆ getVecInputFileNames()

std::vector< std::string > getVecInputFileNames ( ) const
inlineprotectedinherited

Get the input file names used for this algorithm as a STL vector.

Definition at line 275 of file CalibrationAlgorithm.h.

275{return m_inputFileNames;}

◆ inputJsonKeyExists()

bool inputJsonKeyExists ( const std::string &  key) const
inlineprotectedinherited

Test for a key in the input JSON object.

Definition at line 360 of file CalibrationAlgorithm.h.

360{return m_jsonExecutionInput.count(key);}

◆ isBoundaryRequired()

virtual bool isBoundaryRequired ( const Calibration::ExpRun &  )
inlineprotectedvirtualinherited

Given the current collector data, make a decision about whether or not this run should be the start of a payload boundary.

Reimplemented in TestBoundarySettingAlgorithm, PXDAnalyticGainCalibrationAlgorithm, PXDValidationAlgorithm, TestCalibrationAlgorithm, SVD3SampleCoGTimeCalibrationAlgorithm, SVD3SampleELSTimeCalibrationAlgorithm, and SVDCoGTimeCalibrationAlgorithm.

Definition at line 243 of file CalibrationAlgorithm.h.

244 {
245 B2ERROR("You didn't implement a isBoundaryRequired() member function in your CalibrationAlgorithm but you are calling it!");
246 return false;
247 }

◆ loadInputJson()

bool loadInputJson ( const std::string &  jsonString)
inherited

Load the m_inputJson variable from a string (useful from Python interface). The return bool indicates success or failure.

Definition at line 502 of file CalibrationAlgorithm.cc.

503{
504 try {
505 auto jsonInput = nlohmann::json::parse(jsonString);
506 // Input string has an object (dict) as the top level object?
507 if (jsonInput.is_object()) {
508 m_jsonExecutionInput = jsonInput;
509 return true;
510 } else {
511 B2ERROR("JSON input string isn't an object type i.e. not a '{}' at the top level.");
512 return false;
513 }
514 } catch (nlohmann::json::parse_error&) {
515 B2ERROR("Parsing of JSON input string failed");
516 return false;
517 }
518}

◆ resetInputJson()

void resetInputJson ( )
inlineprotectedinherited

Clears the m_inputJson member variable.

Definition at line 330 of file CalibrationAlgorithm.h.

330{m_jsonExecutionInput.clear();}

◆ resetOutputJson()

void resetOutputJson ( )
inlineprotectedinherited

Clears the m_outputJson member variable.

Definition at line 333 of file CalibrationAlgorithm.h.

333{m_jsonExecutionOutput.clear();}

◆ saveCalibration() [1/6]

void saveCalibration ( TClonesArray *  data,
const std::string &  name 
)
protectedinherited

Store DBArray payload with given name with default IOV.

Definition at line 297 of file CalibrationAlgorithm.cc.

298{
300}
const IntervalOfValidity & getRequestedIov() const
Getter for requested IOV.

◆ saveCalibration() [2/6]

void saveCalibration ( TClonesArray *  data,
const std::string &  name,
const IntervalOfValidity iov 
)
protectedinherited

Store DBArray with given name and custom IOV.

Definition at line 276 of file CalibrationAlgorithm.cc.

277{
278 B2DEBUG(29, "Saving calibration TClonesArray '" << name << "' to payloads list.");
279 getPayloads().emplace_back(name, data, iov);
280}

◆ saveCalibration() [3/6]

void saveCalibration ( TObject *  data)
protectedinherited

Store DB payload with default name and default IOV.

Definition at line 287 of file CalibrationAlgorithm.cc.

288{
289 saveCalibration(data, DataStore::objectName(data->IsA(), ""));
290}
static std::string objectName(const TClass *t, const std::string &name)
Return the storage name for an object of the given TClass and name.
Definition: DataStore.cc:151

◆ saveCalibration() [4/6]

void saveCalibration ( TObject *  data,
const IntervalOfValidity iov 
)
protectedinherited

Store DB payload with default name and custom IOV.

Definition at line 282 of file CalibrationAlgorithm.cc.

283{
284 saveCalibration(data, DataStore::objectName(data->IsA(), ""), iov);
285}

◆ saveCalibration() [5/6]

void saveCalibration ( TObject *  data,
const std::string &  name 
)
protectedinherited

Store DB payload with given name with default IOV.

Definition at line 292 of file CalibrationAlgorithm.cc.

293{
295}

◆ saveCalibration() [6/6]

void saveCalibration ( TObject *  data,
const std::string &  name,
const IntervalOfValidity iov 
)
protectedinherited

Store DB payload with given name and custom IOV.

Definition at line 270 of file CalibrationAlgorithm.cc.

271{
272 B2DEBUG(29, "Saving calibration TObject = '" << name << "' to payloads list.");
273 getPayloads().emplace_back(name, data, iov);
274}

◆ setDebugHisto()

void setDebugHisto ( bool  debugHisto)
inline

setter for m_debugHisto

Definition at line 62 of file PXDHotPixelMaskCalibrationAlgorithm.h.

62{m_debugHisto = debugHisto;}

◆ setDescription()

void setDescription ( const std::string &  description)
inlineprotectedinherited

Set algorithm description (in constructor)

Definition at line 321 of file CalibrationAlgorithm.h.

321{m_description = description;}

◆ setInputFileNames() [1/2]

void setInputFileNames ( PyObject *  inputFileNames)
inherited

Set the input file names used for this algorithm from a Python list.

Set the input file names used for this algorithm and resolve the wildcards.

Definition at line 166 of file CalibrationAlgorithm.cc.

167{
168 // The reasoning for this very 'manual' approach to extending the Python interface
169 // (instead of using boost::python) is down to my fear of putting off final users with
170 // complexity on their side.
171 //
172 // I didn't want users that inherit from this class to be forced to use boost and
173 // to have to define a new python module just to use the CAF. A derived class from
174 // from a boost exposed class would need to have its own boost python module definition
175 // to allow access from a steering file and to the base class functions (I think).
176 // I also couldn't be bothered to write a full framework to get around the issue in a similar
177 // way to Module()...maybe there's an easy way.
178 //
179 // But this way we can allow people to continue using their ROOT implemented classes and inherit
180 // easily from this one. But add in a few helper functions that work with Python objects
181 // created in their steering file i.e. instead of being forced to use STL objects as input
182 // to the algorithm.
183 if (PyList_Check(inputFileNames)) {
184 boost::python::handle<> handle(boost::python::borrowed(inputFileNames));
185 boost::python::list listInputFileNames(handle);
186 auto vecInputFileNames = PyObjConvUtils::convertPythonObject(listInputFileNames, vector<string>());
187 setInputFileNames(vecInputFileNames);
188 } else {
189 B2ERROR("Tried to set the input files but we didn't receive a Python list.");
190 }
191}
void setInputFileNames(PyObject *inputFileNames)
Set the input file names used for this algorithm from a Python list.
Scalar convertPythonObject(const boost::python::object &pyObject, Scalar)
Convert from Python to given type.

◆ setInputFileNames() [2/2]

void setInputFileNames ( std::vector< std::string >  inputFileNames)
protectedinherited

Set the input file names used for this algorithm.

Set the input file names used for this algorithm and resolve the wildcards.

Definition at line 194 of file CalibrationAlgorithm.cc.

195{
196 // A lot of code below is tweaked from RootInputModule::initialize,
197 // since we're basically copying the functionality anyway.
198 if (inputFileNames.empty()) {
199 B2WARNING("You have called setInputFileNames() with an empty list. Did you mean to do that?");
200 return;
201 }
202 auto tmpInputFileNames = RootIOUtilities::expandWordExpansions(inputFileNames);
203
204 // We'll use a set to enforce sorted unique file paths as we check them
205 set<string> setInputFileNames;
206 // Check that files exist and convert to absolute paths
207 for (auto path : tmpInputFileNames) {
208 string fullPath = fs::absolute(path).string();
209 if (fs::exists(fullPath)) {
210 setInputFileNames.insert(fs::canonical(fullPath).string());
211 } else {
212 B2WARNING("Couldn't find the file " << path);
213 }
214 }
215
216 if (setInputFileNames.empty()) {
217 B2WARNING("No valid files specified!");
218 return;
219 } else {
220 // Reset the run -> files map as our files are likely different
221 m_runsToInputFiles.clear();
222 }
223
224 // Open TFile to check they can be accessed by ROOT
225 TDirectory* dir = gDirectory;
226 for (const string& fileName : setInputFileNames) {
227 unique_ptr<TFile> f;
228 try {
229 f.reset(TFile::Open(fileName.c_str(), "READ"));
230 } catch (logic_error&) {
231 //this might happen for ~invaliduser/foo.root
232 //actually undefined behaviour per standard, reported as ROOT-8490 in JIRA
233 }
234 if (!f || !f->IsOpen()) {
235 B2FATAL("Couldn't open input file " + fileName);
236 }
237 }
238 dir->cd();
239
240 // Copy the entries of the set to a vector
241 m_inputFileNames = vector<string>(setInputFileNames.begin(), setInputFileNames.end());
243}
std::string getGranularityFromData() const
Get the granularity of collected data.
std::vector< std::string > expandWordExpansions(const std::vector< std::string > &filenames)
Performs wildcard expansion using wordexp(), returns matches.

◆ setOutputJsonValue()

void setOutputJsonValue ( const std::string &  key,
const T &  value 
)
inlineprotectedinherited

Set a key:value pair for the outputJson object, expected to used internally during calibrate()

Definition at line 337 of file CalibrationAlgorithm.h.

337{m_jsonExecutionOutput[key] = value;}

◆ setPrefix()

void setPrefix ( const std::string &  prefix)
inlineinherited

Set the prefix used to identify datastore objects.

Definition at line 167 of file CalibrationAlgorithm.h.

167{m_prefix = prefix;}

◆ updateDBObjPtrs()

void updateDBObjPtrs ( const unsigned int  event = 1,
const int  run = 0,
const int  experiment = 0 
)
protectedinherited

Updates any DBObjPtrs by calling update(event) for DBStore.

Definition at line 404 of file CalibrationAlgorithm.cc.

405{
406 // Construct an EventMetaData object but NOT in the Datastore
407 EventMetaData emd(event, run, experiment);
408 // Explicitly update while avoiding registering a Datastore object
410 // Also update the intra-run objects to the event at the same time (maybe unnecessary...)
412}
Store event, run, and experiment numbers.
Definition: EventMetaData.h:33
static DBStore & Instance()
Instance of a singleton DBStore.
Definition: DBStore.cc:28
void updateEvent()
Updates all intra-run dependent objects.
Definition: DBStore.cc:142
void update()
Updates all objects that are outside their interval of validity.
Definition: DBStore.cc:79

Member Data Documentation

◆ c_nDrains

const unsigned short c_nDrains = 1000
private

Number of drain lines of Belle II PXD sensors.

Definition at line 75 of file PXDHotPixelMaskCalibrationAlgorithm.h.

◆ c_nUCells

const unsigned short c_nUCells = 250
private

Number of uCells of Belle II PXD sensors.

Definition at line 73 of file PXDHotPixelMaskCalibrationAlgorithm.h.

◆ c_nVCells

const unsigned short c_nVCells = 768
private

Number of vCells of Belle II PXD sensors.

Definition at line 71 of file PXDHotPixelMaskCalibrationAlgorithm.h.

◆ deadPixelPayloadName

std::string deadPixelPayloadName

Payload name for PXDDeadPixelPar used for more defective pixels from damaged gates.

Definition at line 53 of file PXDHotPixelMaskCalibrationAlgorithm.h.

◆ drainMultiplier

float drainMultiplier

The occupancy threshold for masking hot drains is the median occupancy x drainMultiplier.

Definition at line 44 of file PXDHotPixelMaskCalibrationAlgorithm.h.

◆ forceContinueMasking

bool forceContinueMasking

Force continue masking in almost empty runs instead of returning c_NotEnoughData.

Definition at line 29 of file PXDHotPixelMaskCalibrationAlgorithm.h.

◆ inefficientPixelMultiplier

float inefficientPixelMultiplier

The occupancy threshold for inefficient (or dead) pixels is the median occupancy x inefficientPixelMultiplier.

Definition at line 56 of file PXDHotPixelMaskCalibrationAlgorithm.h.

◆ m_allExpRun

const ExpRun m_allExpRun = make_pair(-1, -1)
staticprivateinherited

allExpRun

Definition at line 364 of file CalibrationAlgorithm.h.

◆ m_boundaries

std::vector<Calibration::ExpRun> m_boundaries
protectedinherited

When using the boundaries functionality from isBoundaryRequired, this is used to store the boundaries. It is cleared when.

Definition at line 261 of file CalibrationAlgorithm.h.

◆ m_data

ExecutionData m_data
privateinherited

Data specific to a SINGLE execution of the algorithm. Gets reset at the beginning of execution.

Definition at line 382 of file CalibrationAlgorithm.h.

◆ m_debugHisto

int m_debugHisto = false
private

Set if a debugging histogram should be created in the algorithm output directory.

Definition at line 78 of file PXDHotPixelMaskCalibrationAlgorithm.h.

◆ m_description

std::string m_description {""}
privateinherited

Description of the algorithm.

Definition at line 385 of file CalibrationAlgorithm.h.

◆ m_file

std::shared_ptr<TFile> m_file
private

Pointer for TFile.

Definition at line 84 of file PXDHotPixelMaskCalibrationAlgorithm.h.

◆ m_granularityOfData

std::string m_granularityOfData
privateinherited

Granularity of input data. This only changes when the input files change so it isn't specific to an execution.

Definition at line 379 of file CalibrationAlgorithm.h.

◆ m_inputFileNames

std::vector<std::string> m_inputFileNames
privateinherited

List of input files to the Algorithm, will initially be user defined but then gets the wildcards expanded during execute()

Definition at line 373 of file CalibrationAlgorithm.h.

◆ m_jsonExecutionInput

nlohmann::json m_jsonExecutionInput = nlohmann::json::object()
privateinherited

Optional input JSON object used to make decisions about how to execute the algorithm code.

Definition at line 397 of file CalibrationAlgorithm.h.

◆ m_jsonExecutionOutput

nlohmann::json m_jsonExecutionOutput = nlohmann::json::object()
privateinherited

Optional output JSON object that can be set during the execution by the underlying algorithm code.

Definition at line 403 of file CalibrationAlgorithm.h.

◆ m_medianOfHitsMap

std::map<VxdID, double> m_medianOfHitsMap
private

map of VxdID to median hits of each sensor

Definition at line 81 of file PXDHotPixelMaskCalibrationAlgorithm.h.

◆ m_prefix

std::string m_prefix {""}
privateinherited

The name of the TDirectory the collector objects are contained within.

Definition at line 388 of file CalibrationAlgorithm.h.

◆ m_runsToInputFiles

std::map<Calibration::ExpRun, std::vector<std::string> > m_runsToInputFiles
privateinherited

Map of Runs to input files. Gets filled when you call getRunRangeFromAllData, gets cleared when setting input files again.

Definition at line 376 of file CalibrationAlgorithm.h.

◆ maskDrains

bool maskDrains

Mask hot drain lines with too high average occupancy after single pixel masking.

Definition at line 41 of file PXDHotPixelMaskCalibrationAlgorithm.h.

◆ maskRows

bool maskRows

Mask hot rows with too high average occupancy after single pixel masking.

Definition at line 47 of file PXDHotPixelMaskCalibrationAlgorithm.h.

◆ minEvents

int minEvents

Minimum number of collected events.

Definition at line 32 of file PXDHotPixelMaskCalibrationAlgorithm.h.

◆ minHits

int minHits

Minimum median number of hits per pixel needed for dead pixel masking.

Definition at line 35 of file PXDHotPixelMaskCalibrationAlgorithm.h.

◆ minInefficientPixels

int minInefficientPixels

The minimum number of inefficient (or dead) pixels per row to define an inefficient row.

Definition at line 59 of file PXDHotPixelMaskCalibrationAlgorithm.h.

◆ pixelMultiplier

float pixelMultiplier

The occupancy threshold for masking hot single pixels is the median occupancy x pixelMultiplier.

Definition at line 38 of file PXDHotPixelMaskCalibrationAlgorithm.h.

◆ rowMultiplier

float rowMultiplier

The occupancy threshold for masking hot rows is the median occupancy x rowMultiplier.

Definition at line 50 of file PXDHotPixelMaskCalibrationAlgorithm.h.


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