Belle II Software development
|
Class implementing the PXD gain calibration algorithm. More...
#include <PXDAnalyticGainCalibrationAlgorithm.h>
Public Types | |
enum | EResult { c_OK , c_Iterate , c_NotEnoughData , c_Failure , c_Undefined } |
The result of calibration. More... | |
Public Member Functions | |
PXDAnalyticGainCalibrationAlgorithm () | |
Constructor set the prefix to PXDAnalyticGainCalibrationAlgorithm. | |
virtual | ~PXDAnalyticGainCalibrationAlgorithm () |
Destructor. | |
void | setBoundaries (const std::vector< Calibration::ExpRun > &boundaries) |
Setter for boundaries. | |
std::vector< Calibration::ExpRun > | getBoundaries () const |
getter for current boundaries | |
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::DBImportQuery > | getPayloadValues () |
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 | |
int | minClusters |
Minimum number of collected clusters for estimating gains. | |
float | safetyFactor |
Safety factor for determining whether the collected number of clusters is enough. | |
bool | forceContinue |
Force continue in low statistics runs instead of returning c_NotEnoughData. | |
int | strategy |
strategy to used for gain calibration, 0 for medians, 1 for landau fit | |
bool | useChargeRatioHistogram |
Flag to use histogram of charge ratio (relative to expected MPV) | |
bool | correctForward |
Flag to update default gains in forward region due to low statistics. | |
Protected Member Functions | |
virtual EResult | calibrate () override |
Run algo on data. | |
virtual bool | isBoundaryRequired (const Calibration::ExpRun &) override |
Decide if a run should be a payload boundary. Only used in certain Python Algorithm Starategies. | |
virtual void | boundaryFindingSetup (std::vector< Calibration::ExpRun >, int) override |
What we do before isBoundaryRequired gets called. | |
void | setInputFileNames (std::vector< std::string > inputFileNames) |
Set the input file names used for this algorithm. | |
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 | |
double | EstimateGain (VxdID sensorID, unsigned short uBin, unsigned short vBin, TH1 *hist=nullptr) |
Estimate gain as ratio of medians from MC and data for a part of PXD. | |
double | GetCurrentGainFromDB (VxdID sensorID, unsigned short uBin, unsigned short vBin) |
Retrive current gain value from pulled in data base payload. | |
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 | |
std::vector< Calibration::ExpRun > | m_requestedBoundaries |
During findPayloadBoundaries and isBoundaryRequired this is used to define the boundaries we want. | |
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 | |
Class implementing the PXD gain calibration algorithm.
Definition at line 26 of file PXDAnalyticGainCalibrationAlgorithm.h.
|
inherited |
The result of calibration.
Definition at line 40 of file CalibrationAlgorithm.h.
Constructor set the prefix to PXDAnalyticGainCalibrationAlgorithm.
Definition at line 38 of file PXDAnalyticGainCalibrationAlgorithm.cc.
|
inlinevirtual |
|
inlineoverrideprotectedvirtual |
What we do before isBoundaryRequired gets called.
Reimplemented from CalibrationAlgorithm.
Definition at line 69 of file PXDAnalyticGainCalibrationAlgorithm.h.
|
inlineprotectedvirtualinherited |
Put your algorithm back into a state ready for normal execution if you need to.
Definition at line 257 of file CalibrationAlgorithm.h.
|
overrideprotectedvirtual |
Run algo on data.
Implements CalibrationAlgorithm.
Definition at line 51 of file PXDAnalyticGainCalibrationAlgorithm.cc.
|
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.
|
inlineprotectedinherited |
Clear calibration data.
Definition at line 324 of file CalibrationAlgorithm.h.
|
inherited |
Submit constants from last calibration into database.
Definition at line 302 of file CalibrationAlgorithm.cc.
|
inherited |
Submit constants from a (potentially previous) set of payloads.
Definition at line 311 of file CalibrationAlgorithm.cc.
|
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.
|
inlineinherited |
Dump the JSON string of the output JSON object.
Definition at line 223 of file CalibrationAlgorithm.h.
|
private |
Estimate gain as ratio of medians from MC and data for a part of PXD.
Definition at line 218 of file PXDAnalyticGainCalibrationAlgorithm.cc.
|
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.
|
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.
|
inherited |
Fill the mapping of ExpRun -> Files.
Definition at line 330 of file CalibrationAlgorithm.cc.
|
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.
|
inlineprotectedinherited |
Returns the Exp,Run pair that means 'Everything'. Currently unused.
Definition at line 327 of file CalibrationAlgorithm.h.
|
inline |
|
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.
|
private |
Retrive current gain value from pulled in data base payload.
Definition at line 276 of file PXDAnalyticGainCalibrationAlgorithm.cc.
|
inlineinherited |
Get the description of the algorithm (set by developers in constructor)
Definition at line 216 of file CalibrationAlgorithm.h.
|
privateinherited |
Gets the "exp.run" string repr. of (exp,run)
Definition at line 254 of file CalibrationAlgorithm.cc.
|
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.
|
inlineinherited |
Get the granularity of collected data.
Definition at line 188 of file CalibrationAlgorithm.h.
|
protectedinherited |
Get the granularity of collected data.
Definition at line 383 of file CalibrationAlgorithm.cc.
|
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.
|
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.
|
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.
|
inherited |
Get the complete IoV from inspection of collected data.
Definition at line 325 of file CalibrationAlgorithm.cc.
|
inlineprotectedinherited |
Get current iteration.
Definition at line 269 of file CalibrationAlgorithm.h.
|
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.
|
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.
|
inlineinherited |
Get constants (in TObjects) for database update from last execution.
Definition at line 204 of file CalibrationAlgorithm.h.
|
inlineinherited |
Get constants (in TObjects) for database update from last execution but passed by VALUE.
Definition at line 207 of file CalibrationAlgorithm.h.
|
inlineinherited |
Get the prefix used for getting calibration data.
Definition at line 146 of file CalibrationAlgorithm.h.
|
inlineprotectedinherited |
Get the list of runs for which calibration is called.
Definition at line 266 of file CalibrationAlgorithm.h.
|
inherited |
Get the complete list of runs from inspection of collected data.
Definition at line 318 of file CalibrationAlgorithm.cc.
|
inherited |
Get the complete RunRange from inspection of collected data.
Definition at line 361 of file CalibrationAlgorithm.cc.
|
inlineprotectedinherited |
Get the input file names used for this algorithm as a STL vector.
Definition at line 275 of file CalibrationAlgorithm.h.
|
inlineprotectedinherited |
|
overrideprotectedvirtual |
Decide if a run should be a payload boundary. Only used in certain Python Algorithm Starategies.
Reimplemented from CalibrationAlgorithm.
Definition at line 303 of file PXDAnalyticGainCalibrationAlgorithm.cc.
|
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.
|
inlineprotectedinherited |
Clears the m_inputJson member variable.
Definition at line 330 of file CalibrationAlgorithm.h.
|
inlineprotectedinherited |
Clears the m_outputJson member variable.
Definition at line 333 of file CalibrationAlgorithm.h.
|
protectedinherited |
Store DBArray payload with given name with default IOV.
Definition at line 297 of file CalibrationAlgorithm.cc.
|
protectedinherited |
Store DBArray with given name and custom IOV.
Definition at line 276 of file CalibrationAlgorithm.cc.
|
protectedinherited |
Store DB payload with default name and default IOV.
Definition at line 287 of file CalibrationAlgorithm.cc.
|
protectedinherited |
Store DB payload with default name and custom IOV.
Definition at line 282 of file CalibrationAlgorithm.cc.
|
protectedinherited |
Store DB payload with given name with default IOV.
Definition at line 292 of file CalibrationAlgorithm.cc.
|
protectedinherited |
Store DB payload with given name and custom IOV.
Definition at line 270 of file CalibrationAlgorithm.cc.
|
inline |
Setter for boundaries.
Definition at line 54 of file PXDAnalyticGainCalibrationAlgorithm.h.
|
inlineprotectedinherited |
Set algorithm description (in constructor)
Definition at line 321 of file CalibrationAlgorithm.h.
|
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.
|
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.
|
inlineprotectedinherited |
Set a key:value pair for the outputJson object, expected to used internally during calibrate()
Definition at line 337 of file CalibrationAlgorithm.h.
|
inlineinherited |
Set the prefix used to identify datastore objects.
Definition at line 167 of file CalibrationAlgorithm.h.
|
protectedinherited |
Updates any DBObjPtrs by calling update(event) for DBStore.
Definition at line 404 of file CalibrationAlgorithm.cc.
bool correctForward |
Flag to update default gains in forward region due to low statistics.
Definition at line 51 of file PXDAnalyticGainCalibrationAlgorithm.h.
bool forceContinue |
Force continue in low statistics runs instead of returning c_NotEnoughData.
Definition at line 42 of file PXDAnalyticGainCalibrationAlgorithm.h.
|
staticprivateinherited |
allExpRun
Definition at line 364 of file CalibrationAlgorithm.h.
|
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.
|
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.
|
privateinherited |
Description of the algorithm.
Definition at line 385 of file CalibrationAlgorithm.h.
|
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.
|
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.
|
privateinherited |
Optional input JSON object used to make decisions about how to execute the algorithm code.
Definition at line 397 of file CalibrationAlgorithm.h.
|
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.
|
privateinherited |
The name of the TDirectory the collector objects are contained within.
Definition at line 388 of file CalibrationAlgorithm.h.
|
private |
During findPayloadBoundaries and isBoundaryRequired this is used to define the boundaries we want.
Definition at line 95 of file PXDAnalyticGainCalibrationAlgorithm.h.
|
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.
int minClusters |
Minimum number of collected clusters for estimating gains.
Definition at line 36 of file PXDAnalyticGainCalibrationAlgorithm.h.
float safetyFactor |
Safety factor for determining whether the collected number of clusters is enough.
Definition at line 39 of file PXDAnalyticGainCalibrationAlgorithm.h.
int strategy |
strategy to used for gain calibration, 0 for medians, 1 for landau fit
Definition at line 45 of file PXDAnalyticGainCalibrationAlgorithm.h.
bool useChargeRatioHistogram |
Flag to use histogram of charge ratio (relative to expected MPV)
Definition at line 48 of file PXDAnalyticGainCalibrationAlgorithm.h.