A calibration algorithm for CDC dE/dx injection time (HER/LER) More...

#include <CDCDedxInjectTimeAlgorithm.h>

Inheritance diagram for CDCDedxInjectTimeAlgorithm:

Collaboration diagram for CDCDedxInjectTimeAlgorithm:

[legend]

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

Public Member Functions
	CDCDedxInjectTimeAlgorithm ()
	Constructor: Sets the description, the properties and the parameters of the algorithm.

virtual	~CDCDedxInjectTimeAlgorithm ()
	Destructor.

void	setMergePayload (bool value=true)
	function to decide merged vs relative calibration

void	setMonitoringPlots (bool value=false)
	function to enable monitoring plots

void	setFitWidth (double value)
	function to set fit range (sigma)

void	setMinTracks (int value)
	function to set min # of tracks in time bins (0-40ms)

void	setDedxPars (int value, double min, double max)
	function to set dedx hist parameters

void	setChiPars (int value, double min, double max)
	function to set chi hist parameters

void	fitGaussianWRange (TH1D *&temphist, fstatus &status)
	function to perform gaus fit for input histogram

void	getExpRunInfo ()
	function to get exp/run information (payload object, plotting)

void	defineTimeBins ()
	function to set/reset time bins

void	defineHisto (std::array< std::vector< TH1D * >, numdedx::nrings > &htemp, std::string var)
	function to define histograms for dedx and time dist. More...

void	defineTimeHisto (std::array< std::array< TH1D *, 3 >, numdedx::nrings > &htemp)
	function to define injection time bins histograms (monitoring only)

void	checkStatistics (std::array< std::vector< TH1D * >, numdedx::nrings > &hvar)
	check statistics for obtaining calibration const.

void	correctBinBias (std::map< int, std::vector< double >> &varcorr, std::map< int, std::vector< double >> &var, std::map< int, std::vector< double >> &time, TH1D *&htimes)
	function to correct dedx mean/reso and return corrected vector map

void	createPayload (std::array< double, numdedx::nrings > &scale, std::map< int, std::vector< double >> &vmeans, std::map< int, std::vector< double >> &varscal, std::string svar)
	function to store payloads after full calibration

void	getMeanReso (std::array< std::vector< TH1D * >, numdedx::nrings > &hvar, std::map< int, std::vector< double >> &vmeans, std::map< int, std::vector< double >> &vresos)
	function to get mean and reso of histogram

void	plotEventStats ()
	function to draw event/track statistics plots

void	plotBinLevelDist (std::array< std::vector< TH1D * >, numdedx::nrings > &hvar, std::string var)
	function to draw dedx, chi and time dist.

void	plotRelConstants (std::map< int, std::vector< double >> &vmeans, std::map< int, std::vector< double >> &vresos, std::map< int, std::vector< double >> &corr, std::string svar)
	function to relative constant from dedx fit mean and chi fit reso

void	plotTimeStat (std::array< std::vector< TH1D * >, numdedx::nrings > &htime)
	function to draw time stats

void	plotFinalConstants (std::map< int, std::vector< double >> &vmeans, std::map< int, std::vector< double >> &vresos, std::array< double, numdedx::nrings > &scale, std::array< double, numdedx::nrings > &scale_reso)
	function to final constant from merging or abs fits

void	plotInjectionTime (std::array< std::array< TH1D *, 3 >, numdedx::nrings > &hvar)
	function to injection time distributions (HER/LER in three bins)

void	setHistStyle (TH1D *&htemp, const int ic, const int is, const double min, const double max)
	function to set histogram cosmetics

std::string	getTimeBinLabel (const double &tedges, const int &it)
	function to return time label for histograms labeling

void	deleteHisto (std::array< std::vector< TH1D * >, numdedx::nrings > &htemp)
	function to delete histograms for dedx and time dist. More...

void	deleteTimeHisto (std::array< std::array< TH1D *, 3 >, numdedx::nrings > &htemp)
	function to define injection time bins histograms (monitoring only)

double	getCorrection (unsigned int ring, unsigned int time, std::map< int, std::vector< double >> &vmeans)
	function to get the correction factor of mean

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::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 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
virtual EResult	calibrate () override
	CDC dE/dx Injection time algorithm.

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
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::array< std::string, numdedx::nrings >	m_sring {"ler", "her"}
	injection ring name

std::vector< double >	m_vtedges
	external time vector

std::vector< double >	m_vtlocaledges
	internal time vector

double *	m_tedges
	internal time array (copy of vtlocaledges)

unsigned int	m_tbins
	internal time bins

double	m_sigmaR
	fit dedx dist in sigma range

int	m_dedxBins
	bins for dedx histogram

double	m_dedxMin
	min range of dedx

double	m_dedxMax
	max range of dedx

int	m_chiBins
	bins for chi histogram

double	m_chiMin
	min range of chi

double	m_chiMax
	max range of chi

int	m_countR
	a hack for running functions once

int	m_thersE
	min tracks to start calibration

bool	m_isminStat
	flag to merge runs for statistics thershold

bool	m_ismakePlots
	produce plots for monitoring

bool	m_isMerge
	merge payload when rel constant

std::string	m_prefix
	string prefix for plot names

std::string	m_suffix
	string suffix for object names

std::vector< std::vector< double > >	m_vinjPayload
	vector to store payload values

DBObjPtr< CDCDedxInjectionTime >	m_DBInjectTime
	Injection time DB object.

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.

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 int	c_rings = numdedx::nrings
	injection ring constants

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

Detailed Description

A calibration algorithm for CDC dE/dx injection time (HER/LER)

Definition at line 46 of file CDCDedxInjectTimeAlgorithm.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.

◆ defineHisto()

void defineHisto	(	std::array< std::vector< TH1D * >, numdedx::nrings > &	htemp,
		std::string	var
	)

function to define histograms for dedx and time dist.

(in time bins)

Definition at line 374 of file CDCDedxInjectTimeAlgorithm.cc.

 {
   for (unsigned int ir = 0; ir < c_rings; ir++) {
     htemp[ir].resize(m_tbins);
     for (unsigned int it = 0; it < m_tbins; it++) {
       std::string label = getTimeBinLabel(m_tedges[it], it);
       std::string title = Form("%s(%s), time(%s)", m_suffix.data(), m_sring[ir].data(), label.data());
       std::string hname = Form("h%s_%s_%s_t%d", var.data(), m_sring[ir].data(), m_suffix.data(), it);
       if (var == "dedx" or var == "dedx_corr") htemp[ir][it] = new TH1D(hname.data(), "", m_dedxBins, m_dedxMin, m_dedxMax);
       else if (var == "chi") htemp[ir][it] = new TH1D(hname.data(), "", m_chiBins, m_chiMin, m_chiMax);
       else htemp[ir][it] = new TH1D(hname.data(), "", 50, m_tedges[it], m_tedges[it + 1]);
       htemp[ir][it]->SetTitle(Form("%s;%s;entries", title.data(), var.data()));
     }
   }
 }

◆ deleteHisto()

void deleteHisto ( std::array< std::vector< TH1D * >, numdedx::nrings > & htemp )

inline

function to delete histograms for dedx and time dist.

(in time bins)

Definition at line 210 of file CDCDedxInjectTimeAlgorithm.h.

     {
       for (unsigned int ir = 0; ir < c_rings; ir++) {
         for (unsigned int it = 0; it < m_tbins; it++) {
           delete htemp[ir][it];
         }
       }
     }

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

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

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

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

reconstruction/calibration/include/CDCDedxInjectTimeAlgorithm.h
reconstruction/calibration/src/CDCDedxInjectTimeAlgorithm.cc

Public Types

Public Member Functions

Protected Member Functions

Protected Attributes

Private Member Functions

Private Attributes

Static Private Attributes

Detailed Description

Member Enumeration Documentation

◆ EResult

Member Function Documentation

◆ checkPyExpRun()

◆ convertPyExpRun()

◆ defineHisto()

◆ deleteHisto()

◆ execute()

◆ getCollectorName()

◆ setInputFileNames() [1/2]

◆ setInputFileNames() [2/2]