Belle II Software  release-08-01-10
TOPCommonT0BFAlgorithm Class Reference

Algorithm for common T0 calibration with a fit of bunch finder residuals. More...

#include <TOPCommonT0BFAlgorithm.h>

Inheritance diagram for TOPCommonT0BFAlgorithm:
Collaboration diagram for TOPCommonT0BFAlgorithm:

Public Types

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

Public Member Functions

 TOPCommonT0BFAlgorithm ()
 Constructor.
 
virtual ~TOPCommonT0BFAlgorithm ()
 Destructor.
 
void setMinEntries (int minEntries)
 Sets minimal number of histogram entries to perform a fit.
 
void setFitInitializers (double sigmaCore, double sigmaTail, double tailFract)
 Sets values for the initialization of several fit parameters. More...
 
void setCutoffEntries (int cutoffEntries)
 Sets cutoff on the number of histogram entries for steering btw. More...
 
void setMinError (double minError)
 Sets minimal fitted offset uncertainty to declare this calibration as c_OK. More...
 
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. More...
 
Calibration::ExpRun convertPyExpRun (PyObject *pyObj)
 Performs the conversion of PyObject to ExpRun. More...
 
std::string getCollectorName () const
 Alias for prefix. More...
 
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. More...
 
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. More...
 
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 algoithm (set by developers in constructor)
 
bool loadInputJson (const std::string &jsonString)
 Load the m_inputJson variable from a string (useful from Python interface). The rturn 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>.
 

Protected Member Functions

void setInputFileNames (std::vector< std::string > inputFileNames)
 Set the input file names used for this algorithm. More...
 
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 interally 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

virtual EResult calibrate () final
 algorithm implementation
 
std::shared_ptr< TH1F > getHistogram ()
 Returns histogram to be fitted.
 
int fitSingleGaus (std::shared_ptr< TH1F > h)
 Fit single gaus + constant. More...
 
int fitDoubleGaus (std::shared_ptr< TH1F > h)
 Fit double gaus w/ same mean + constant. More...
 
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

int m_minEntries = 10
 minimal number of histogram entries to perform fit
 
double m_sigmaCoreInit = 0.060
 core gaussian sigma [ns]
 
double m_sigmaTailInit = 0.120
 tail gaussian sigma [ns]
 
double m_tailFractInit = 0.20
 fraction of tail gaussian
 
int m_cutoffEntries = 100
 cutoff entries for single/double gaussian fit
 
double m_minError = 0.020
 minimal commonT0 uncertainty [ns] to declare c_OK
 
int m_expNo = 0
 experiment number
 
int m_runNo = 0
 first run number
 
int m_runLast = 0
 last run number
 
float m_fittedOffset = 0
 fitted offset
 
float m_offset = 0
 wrap-around of fitted offset (= common T0)
 
float m_offsetError = 0
 error on fitted offset
 
float m_sigmaCore = 0
 core gaussian sigma
 
float m_sigmaTail = 0
 tail gaussian sigma (set to 0 for single gauss fit)
 
float m_tailFract = 0
 tail fraction (set to 0 for single gauss fit)
 
float m_chi2 = 0
 normalized chi^2
 
float m_integral = 0
 number of fitted signal events
 
int m_numEvents = 0
 number of all events in the histogram
 
int m_fitStatus = -1
 fit status
 
double m_bunchTimeSep = 0
 bunch separation in time [ns]
 
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

Algorithm for common T0 calibration with a fit of bunch finder residuals.

method BF

Definition at line 24 of file TOPCommonT0BFAlgorithm.h.

Member Enumeration Documentation

◆ EResult

enum EResult
inherited

The result of calibration.

Enumerator
c_OK 

Finished successfuly =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.

Member Function Documentation

◆ 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.

◆ 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.

◆ execute()

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.

◆ fitDoubleGaus()

int fitDoubleGaus ( std::shared_ptr< TH1F >  h)
private

Fit double gaus w/ same mean + constant.

Parameters
hhistogram
Returns
fit status

Definition at line 167 of file TOPCommonT0BFAlgorithm.cc.

168  {
169  double sum = h->GetSumOfWeights();
170  if (sum < 7) return 7;
171  double maxPosition = h->GetBinCenter(h->GetMaximumBin());
172  double binWidth = h->GetBinWidth(1);
173  double xmin = h->GetXaxis()->GetXmin();
174  double xmax = h->GetXaxis()->GetXmax();
175 
176  auto* func = new TF1("func2g",
177  "[0] + [1]*((1-[4])/sqrt(2*pi)/[3]*exp(-0.5*((x-[2])/[3])**2)"
178  "+ [4]/sqrt(2*pi)/[5]*exp(-0.5*((x-[2])/[5])**2))",
179  xmin, xmax);
180  func->SetParameter(0, 0);
181  func->SetParameter(1, sum * binWidth);
182  func->SetParameter(2, maxPosition);
183  func->SetParameter(3, m_sigmaCoreInit);
184  func->SetParameter(4, m_tailFractInit);
185  func->SetParameter(5, m_sigmaTailInit);
186 
187  int status = h->Fit(func, "LRSQ");
188 
189  m_fittedOffset = func->GetParameter(2);
191  m_offsetError = func->GetParError(2);
192  m_sigmaCore = func->GetParameter(3);
193  m_sigmaTail = func->GetParameter(5);
194  m_tailFract = func->GetParameter(4);
195  m_chi2 = func->GetChisquare() / func->GetNDF();
196  m_integral = func->GetParameter(1) / binWidth;
197  m_numEvents = sum;
198  m_fitStatus = status;
199 
200  return status;
201  }
float m_offsetError
error on fitted offset
float m_offset
wrap-around of fitted offset (= common T0)
float m_tailFract
tail fraction (set to 0 for single gauss fit)
int m_numEvents
number of all events in the histogram
double m_sigmaTailInit
tail gaussian sigma [ns]
double m_tailFractInit
fraction of tail gaussian
double m_sigmaCoreInit
core gaussian sigma [ns]
float m_integral
number of fitted signal events
float m_sigmaTail
tail gaussian sigma (set to 0 for single gauss fit)
double m_bunchTimeSep
bunch separation in time [ns]

◆ fitSingleGaus()

int fitSingleGaus ( std::shared_ptr< TH1F >  h)
private

Fit single gaus + constant.

Parameters
hhistogram
Returns
fit status

Definition at line 133 of file TOPCommonT0BFAlgorithm.cc.

◆ 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 algorihtm.

Definition at line 164 of file CalibrationAlgorithm.h.

◆ setCutoffEntries()

void setCutoffEntries ( int  cutoffEntries)
inline

Sets cutoff on the number of histogram entries for steering btw.

single/double gaussian parameterization of the signal.

Parameters
cutoffEntriesdouble gaussian above that value, otherwise single gaussian

Definition at line 56 of file TOPCommonT0BFAlgorithm.h.

56 {m_cutoffEntries = cutoffEntries;}
int m_cutoffEntries
cutoff entries for single/double gaussian fit

◆ setFitInitializers()

void setFitInitializers ( double  sigmaCore,
double  sigmaTail,
double  tailFract 
)
inline

Sets values for the initialization of several fit parameters.

Parameters
sigmaCorecore gaussian sigma [ns]
sigmaTailtail gaussian sigma [ns]
tailFractfraction of tail gaussian. Single gaussian used if set to 0.

Definition at line 44 of file TOPCommonT0BFAlgorithm.h.

◆ 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.

◆ 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.

◆ setMinError()

void setMinError ( double  minError)
inline

Sets minimal fitted offset uncertainty to declare this calibration as c_OK.

Parameters
minErrorminimal uncertainty [ns]

Definition at line 62 of file TOPCommonT0BFAlgorithm.h.


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