Belle II Software development
CDCDedx1DCellAlgorithm Class Reference

A calibration algorithm for CDC dE/dx electron: 1D enta cleanup correction. More...

#include <CDCDedx1DCellAlgorithm.h>

Inheritance diagram for CDCDedx1DCellAlgorithm:
CalibrationAlgorithm

Public Types

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

Public Member Functions

 CDCDedx1DCellAlgorithm ()
 Constructor: Sets the description, the properties and the parameters of the algorithm.
 
virtual ~CDCDedx1DCellAlgorithm ()
 Destructor.
 
void setEntaRange (double min=-1.0, double max=1.0)
 function to set min/max range of entrance angle for calibration
 
void setEntaBins (unsigned int value=316)
 function to set number of entrance angle bins for calibration
 
void setHistBins (int value=250)
 function to set nbins of dedx dist for calibration
 
void setHistRange (double min=0.0, double max=5.0)
 function to set min/max range of dedx dist for calibration
 
void setPtLimit (double value)
 function to set pt limit
 
void setCosLimit (double value)
 function to set costheta limit
 
void setTrucationBins (double lowedge, double upedge)
 function to set bins of truncation from histogram
 
void setSplitFactor (int value)
 set bin split factor for all range
 
void setChargeType (int value)
 set charge type
 
void setAdjustmentFactor (int value)
 set adjustment factor
 
void setLayerTrunc (bool value=false)
 function to set truncation method (local vs global)
 
void setVariableBins (bool value)
 Set Var bins flag to on or off.
 
void setRotSymmetry (bool value)
 set rotation sys to copy constants from one region to other
 
void enableExtraPlots (bool value=false)
 funtion to set flag active for plotting
 
void setPrintLog (bool value=false)
 funtion to set flag to print log
 
void setMergePayload (bool value)
 set false if generating absolute (not relative) payload
 
void setSuffix (const std::string &value)
 adding suffix to control plots
 
unsigned int getRepresentativeLayer (unsigned int il) const
 Representative CDC layer for each SL group (used to access group-wise constants): SL0 => 1, SL1 => 9, SL2-8 => 17.
 
void setBaselineFactor (double charge, double factor)
 adjust baseline based on charge or global overall works for only single charge or both
 
int rotationalBin (int nbin, int ibin)
 class function to set rotation symmetry
 
void getExpRunInfo ()
 function to extract calibration run/exp
 
void CreateBinMapping ()
 class function to create vectors for bin mapping (Var->symm)
 
void defineHisto (std::array< std::vector< TH1D * >, 3 > &hdedxhit, std::array< TH1D *, 3 > &hdedxlay, std::array< TH1D *, 3 > &hentalay)
 function to define histograms
 
void getTruncatedBins (TH1D *hist, int &binlow, int &binhigh)
 function to get bins of truncation from histogram
 
double getTruncationMean (TH1D *hist, int binlow, int binhigh)
 function to get truncated mean
 
void createPayload ()
 function to generate final constants
 
void plotMergeFactor (std::map< int, std::vector< double > > bounds, const std::array< int, 2 > nDev, std::map< int, std::vector< int > > steps)
 function to plot merging factor
 
void plotdedxHist (std::array< std::vector< TH1D * >, m_kNGroups > &hdedxhit)
 function to draw the dE/dx histogram in enta bins
 
void plotLayerDist (std::array< TH1D *, m_kNGroups > &hdedxlay)
 function to draw dedx dist.
 
void plotQaPars (std::array< TH1D *, m_kNGroups > &hentalay, TH2D *hptcosth)
 function to draw pt vs costh and entrance angle distribution for Inner/Outer layer
 
void plotRelConst (std::vector< double >tempconst, std::vector< double >layerconst, int il)
 function to draw symm/Var layer constant
 
void plotConstants ()
 function to draw the old/new final constants
 
void plotEventStats ()
 function to draw the stats plots
 
const std::string & getPrefix () const
 Get the prefix used for getting calibration data.
 
const 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.
 
const 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.
 
const std::list< Database::DBImportQuery > & getPayloadValues () const
 Get constants (in TObjects) for database update from last execution.
 
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>.
 

Static Public Member Functions

static bool checkPyExpRun (PyObject *pyObj)
 Checks that a PyObject can be successfully converted to an ExpRun type.
 
static Calibration::ExpRun convertPyExpRun (PyObject *pyObj)
 Performs the conversion of PyObject to ExpRun.
 

Static Public Attributes

static constexpr int m_kNGroups = 3
 SL grouping: inner (SL0), middle (SL1), outer (SL2–8)
 

Protected Member Functions

virtual EResult calibrate () override
 1D cell algorithm
 
void setInputFileNames (const 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.
 
const 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 setDescription (const std::string &description)
 Set algorithm description (in constructor)
 
void clearCalibrationData ()
 Clear calibration data.
 
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.
 

Static Protected Member Functions

static void updateDBObjPtrs (const unsigned int event, const int run, const int experiment)
 Updates any DBObjPtrs by calling update(event) for DBStore.
 
static Calibration::ExpRun getAllGranularityExpRun ()
 Returns the Exp,Run pair that means 'Everything'. Currently unused.
 

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

double m_eaMin
 lower edge of entrance angle
 
double m_eaMax
 upper edge of entrance angle
 
double m_eaBW
 binwdith of entrance angle bin
 
int m_eaBin
 
int m_eaB
 reset # of bins for entrance angle for each experiment
 
double m_dedxMin
 lower edge of dedxhit
 
double m_dedxMax
 upper edge of dedxhit
 
int m_dedxBin
 
double m_ptMax
 a limit on transverse momentum
 
double m_cosMax
 a limit on cos theta
 
double m_truncMin
 lower threshold on truncation
 
double m_truncMax
 upper threshold on truncation
 
int m_binSplit
 multiply nbins by this factor in full range
 
double m_chargeType
 charge type for baseline adj
 
double m_adjustFac
 factor with that one what to adjust baseline
 
bool isFixTrunc
 true = fix window for all out/inner layers
 
bool isVarBins
 true: if variable bin size is requested
 
bool isRotSymm
 if rotation symmetry requested
 
bool isMakePlots
 produce plots for status
 
bool isPrintLog
 print more debug information
 
bool isMerge
 print more debug information
 
std::string m_suffix
 add suffix to all plot name
 
std::string m_runExp
 add run and exp to title of plot
 
std::string m_label [m_kNGroups] = {"SL0", "SL1", "SL2-8"}
 add inner/outer superlayer label
 
std::vector< int > m_eaBinLocal
 
std::array< std::vector< int >, m_kNGroupsm_binIndex
 symm/Var bin numbers
 
std::array< std::vector< double >, m_kNGroupsm_binValue
 enta Var bin values
 
std::vector< std::vector< double > > m_onedcors
 final vectors of calibration
 
DBObjPtr< CDCDedx1DCellm_DBOneDCell
 One cell correction 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.
 
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

A calibration algorithm for CDC dE/dx electron: 1D enta cleanup correction.

Definition at line 28 of file CDCDedx1DCellAlgorithm.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,
43 c_NotEnoughData,
44 c_Failure,
45 c_Undefined
46 };

Constructor & Destructor Documentation

◆ CDCDedx1DCellAlgorithm()

Constructor: Sets the description, the properties and the parameters of the algorithm.

Definition at line 25 of file CDCDedx1DCellAlgorithm.cc.

25 :
26 CalibrationAlgorithm("CDCDedxElectronCollector"),
27 m_eaMin(-TMath::Pi() / 2),
28 m_eaMax(+TMath::Pi() / 2),
29 m_eaB(316),
30 m_dedxMin(0.0),
31 m_dedxMax(5.0),
32 m_dedxBin(250),
33 m_ptMax(8.0),
34 m_cosMax(1.0),
35 m_truncMin(0.05),
36 m_truncMax(0.75),
37 m_binSplit(3),
38 m_chargeType(0),
39 m_adjustFac(1.00),
40 isFixTrunc(false),
41 isVarBins(true),
42 isRotSymm(false),
43 isMakePlots(true),
44 isPrintLog(false),
45 isMerge(true),
46 m_suffix("")
47{
48 // Set module properties
49 setDescription("A calibration algorithm for the CDC dE/dx entrance angle cleanup correction");
50}
double m_eaMax
upper edge of entrance angle
double m_truncMax
upper threshold on truncation
int m_binSplit
multiply nbins by this factor in full range
double m_truncMin
lower threshold on truncation
double m_adjustFac
factor with that one what to adjust baseline
double m_chargeType
charge type for baseline adj
double m_cosMax
a limit on cos theta
bool isPrintLog
print more debug information
std::string m_suffix
add suffix to all plot name
int m_eaB
reset # of bins for entrance angle for each experiment
double m_ptMax
a limit on transverse momentum
bool isFixTrunc
true = fix window for all out/inner layers
bool isVarBins
true: if variable bin size is requested
bool isRotSymm
if rotation symmetry requested
double m_dedxMax
upper edge of dedxhit
bool isMakePlots
produce plots for status
bool isMerge
print more debug information
double m_dedxMin
lower edge of dedxhit
double m_eaMin
lower edge of entrance angle
void setDescription(const std::string &description)
Set algorithm description (in constructor)
CalibrationAlgorithm(const std::string &collectorModuleName)
Constructor - sets the prefix for collected objects (won't be accesses until execute(....

◆ ~CDCDedx1DCellAlgorithm()

virtual ~CDCDedx1DCellAlgorithm ( )
inlinevirtual

Destructor.

Definition at line 42 of file CDCDedx1DCellAlgorithm.h.

42{}

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 PXDAnalyticGainCalibrationAlgorithm, PXDValidationAlgorithm, SVD3SampleCoGTimeCalibrationAlgorithm, SVD3SampleELSTimeCalibrationAlgorithm, SVDCoGTimeCalibrationAlgorithm, TestBoundarySettingAlgorithm, and TestCalibrationAlgorithm.

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

1D cell algorithm

Implements CalibrationAlgorithm.

Definition at line 55 of file CDCDedx1DCellAlgorithm.cc.

56{
57
59
60 if (!m_DBOneDCell.isValid())
61 B2FATAL("There is no valid previous payload for CDCDedx1DCell");
62
63 //reading radiative electron collector TREE
64 auto ttree = getObjectPtr<TTree>("tree");
65 if (!ttree) return c_NotEnoughData;
66
67 std::vector<double>* dedxhit = 0, *enta = 0;
68 std::vector<int>* layer = 0;
69 double pt = 0, costh = 0;
70
71 ttree->SetBranchAddress("dedxhit", &dedxhit);
72 ttree->SetBranchAddress("entaRS", &enta);
73 ttree->SetBranchAddress("layer", &layer);
74 ttree->SetBranchAddress("pt", &pt);
75 ttree->SetBranchAddress("costh", &costh);
76
77 //repair nbins if they are not divisible accordingly
80
81 //Settings of variables bins
83
84 if (isPrintLog) {
85 B2INFO("Superlayer0 bins: " << m_eaBinLocal[0]);
86 B2INFO("Superlayer1 bins: " << m_eaBinLocal[1]);
87 B2INFO("Superlayers2-8 bins: " << m_eaBinLocal[2]);
88 }
89
90 // dedxhit vector to store dE/dx values for each enta bin
91 std::array<std::vector<TH1D*>, m_kNGroups> hdedxhit;
92 std::array<TH1D*, m_kNGroups> hdedxlay{};
93 std::array<TH1D*, m_kNGroups> hentalay{};
94
95 TH2D* hptcosth = new TH2D("hptcosth", "pt vs costh dist;pt;costh", 1000, -8.0, 8.0, 1000, -1.0, 1.0);
96
97 defineHisto(hdedxhit, hdedxlay, hentalay);
98
99 //Star filling histogram defined above
100 for (int i = 0; i < ttree->GetEntries(); ++i) {
101
102 ttree->GetEvent(i);
103
104 if (std::abs(costh) > m_cosMax) continue;
105
106 // remove wide angle bhabha tracks
107 // double mom = pt/sqrt(1-costh*costh);
108 // if(abs(pt)<2.4 && abs(mom)>3.6)continue;
109
110 if (std::abs(pt) > m_ptMax) continue;
111
112 //change to random 10%
113 int rand = gRandom->Integer(100);
114 if (rand < 10) hptcosth->Fill(pt, costh);
115
116 for (unsigned int j = 0; j < dedxhit->size(); ++j) {
117
118 if (dedxhit->at(j) == 0) continue;
119
120 double entaval = enta->at(j);
121 //Mapped bin corresponds to entaval
122 int ibin = std::floor((entaval - m_eaMin) / m_eaBW);
123 if (ibin < 0 || ibin >= m_eaBin) continue;
124
125 int lay = layer->at(j);
126 int mL = (lay < 8) ? 0 : ((lay < 14) ? 1 : 2);
127
128 hdedxlay[mL]->Fill(dedxhit->at(j));
129 if (rand < 10) hentalay[mL]->Fill(entaval);
130
131 int jbinea = ibin;
132 if (isVarBins) jbinea = m_binIndex[mL].at(ibin);
133 hdedxhit[mL][jbinea]->Fill(dedxhit->at(j));
134 }
135 }
136
137 for (int il = 0; il < m_kNGroups; il++) {
138
139 int minlay = 0, maxlay = 0;
140
141 if (isFixTrunc) {
142 getTruncatedBins(hdedxlay[il], minlay, maxlay);
143 hdedxlay[il]->SetTitle(Form("%s;%d;%d", hdedxlay[il]->GetTitle(), minlay, maxlay));
144 }
145
146 std::vector<double>tempconst;
147 tempconst.reserve(m_eaBinLocal[il]);
148
149 for (int iea = 0; iea < m_eaBinLocal[il]; iea++) {
150
151 int jea = iea;
152
153 // rotation symmtery for 1<->3 and 4<->2 but only symmetric bin
154 if (!isVarBins && isRotSymm) jea = rotationalBin(m_eaBinLocal[il], jea);
155
156 TH1D* htemp = (TH1D*)hdedxhit[il][jea]->Clone(Form("h_%s_b%d_c", m_label[il].data(), jea));
157
158 int minbin = 1, maxbin = 1;
159 if (isFixTrunc) {
160 minbin = minlay;
161 maxbin = maxlay;
162 } else {
163 //extract truncation window per bin
164 getTruncatedBins(htemp, minbin, maxbin);
165 }
166
167 double dedxmean;
168 dedxmean = getTruncationMean(htemp, minbin, maxbin);
169 tempconst.push_back(dedxmean);
170
171 hdedxhit[il][iea]->SetTitle(Form("%s, #mu_{truc} = %0.5f;%d;%d", hdedxhit[il][iea]->GetTitle(), dedxmean, minbin, maxbin));
172 }
173
174 //Expending constants
175 std::vector<double>layerconst;
176 layerconst.reserve(m_eaBin);
177
178 for (int iea = 0; iea < m_eaBin; iea++) {
179 int jea = iea;
180 if (isVarBins) jea = m_binIndex[il].at(iea);
181 layerconst.push_back(tempconst.at(jea));
182 }
183
184 // plot the rel constants var/sym bins
185 if (isMakePlots) plotRelConst(tempconst, layerconst, il);
186 m_onedcors.push_back(layerconst);
187
188 layerconst.clear();
189 tempconst.clear();
190 }
191
192 //Saving final constants
194
195 if (isMakePlots) {
196
197 //1. dE/dx dist. for entrance angle bins
198 plotdedxHist(hdedxhit);
199
200 //3. Inner and Outer layer dE/dx distributions
201 plotLayerDist(hdedxlay);
202
203 //4. entrance angle distribution sym/var bins
204 plotQaPars(hentalay, hptcosth);
205
206 //6. draw the final constants
208
209 //7. plot statistics related plots here
211 }
212
213 for (int il = 0; il < m_kNGroups; il++) {
214 delete hentalay[il];
215 delete hdedxlay[il];
216 for (int iea = 0; iea < m_eaBinLocal[il]; iea++)
217 delete hdedxhit[il][iea];
218 }
219
220 delete hptcosth;
221 m_eaBinLocal.clear();
222 for (int il = 0; il < m_kNGroups; il++) {
223 m_binValue[il].clear();
224 m_binIndex[il].clear();
225 }
226 m_suffix.clear();
227
228 return c_OK;
229}
void plotLayerDist(std::array< TH1D *, m_kNGroups > &hdedxlay)
function to draw dedx dist.
void getTruncatedBins(TH1D *hist, int &binlow, int &binhigh)
function to get bins of truncation from histogram
void CreateBinMapping()
class function to create vectors for bin mapping (Var->symm)
void getExpRunInfo()
function to extract calibration run/exp
static constexpr int m_kNGroups
SL grouping: inner (SL0), middle (SL1), outer (SL2–8)
DBObjPtr< CDCDedx1DCell > m_DBOneDCell
One cell correction DB object.
double getTruncationMean(TH1D *hist, int binlow, int binhigh)
function to get truncated mean
void plotConstants()
function to draw the old/new final constants
std::array< std::vector< int >, m_kNGroups > m_binIndex
symm/Var bin numbers
void plotQaPars(std::array< TH1D *, m_kNGroups > &hentalay, TH2D *hptcosth)
function to draw pt vs costh and entrance angle distribution for Inner/Outer layer
double m_eaBW
binwdith of entrance angle bin
void defineHisto(std::array< std::vector< TH1D * >, 3 > &hdedxhit, std::array< TH1D *, 3 > &hdedxlay, std::array< TH1D *, 3 > &hentalay)
function to define histograms
void plotEventStats()
function to draw the stats plots
int rotationalBin(int nbin, int ibin)
class function to set rotation symmetry
std::array< std::vector< double >, m_kNGroups > m_binValue
enta Var bin values
void createPayload()
function to generate final constants
void plotRelConst(std::vector< double >tempconst, std::vector< double >layerconst, int il)
function to draw symm/Var layer constant
std::vector< std::vector< double > > m_onedcors
final vectors of calibration
std::string m_label[m_kNGroups]
add inner/outer superlayer label
void plotdedxHist(std::array< std::vector< TH1D * >, m_kNGroups > &hdedxhit)
function to draw the dE/dx histogram in enta bins
@ c_OK
Finished successfully =0 in Python.
@ c_NotEnoughData
Needs more data =2 in Python.
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 t...

◆ checkPyExpRun()

bool checkPyExpRun ( PyObject * pyObj)
staticinherited

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.

324{m_data.clearCalibrationData();}

◆ 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:41
bool storeData(const std::string &name, TObject *object, const IntervalOfValidity &iov)
Store an object in the database.
Definition Database.cc:140

◆ commit() [2/2]

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

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

Definition at line 312 of file CalibrationAlgorithm.cc.

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

◆ convertPyExpRun()

ExpRun convertPyExpRun ( PyObject * pyObj)
staticinherited

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}

◆ CreateBinMapping()

void CreateBinMapping ( )

class function to create vectors for bin mapping (Var->symm)

Definition at line 257 of file CDCDedx1DCellAlgorithm.cc.

258{
259
260 std::map<int, std::vector<double>> bounds;
261 std::map<int, std::vector<int>> steps;
262
263 // number of intervals for each unique configuration
264 const std::array<int, 2> nDev{8, 4};
265
266 // config 0 -> used for SL = 0,1
267 bounds[0] = {0, 108, 123, 133, 158, 183, 193, 208, 316};
268 steps[0] = {9, 3, 2, 1, 1, 2, 3, 9};
269
270 // config 1 -> used for SL = 2-8
271 bounds[1] = {0, 38, 158, 278, 316};
272 steps[1] = {2, 1, 1, 2};
273
274 // map SL -> configuration
275 const std::array<int, m_kNGroups> configIndex = {0, 0, 1};
276
277 for (int il = 0; il < m_kNGroups; il++) {
278
279 int icfg = configIndex[il];
280
281 std::vector<double> scaledBounds = bounds[icfg];
282 for (int ibin = 0; ibin <= nDev[icfg]; ibin++) {
283 scaledBounds[ibin] = scaledBounds[ibin] * m_binSplit;
284 }
285
286 int ieaprime = -1, temp = -99, ibin = 0;
287
288 double pastbin = m_eaMin;
289 m_binValue[il].push_back(pastbin);
290
291 for (int iea = 0; iea < m_eaBin; iea++) {
292
293 if (isVarBins) {
294 if (iea % int(scaledBounds[ibin + 1]) == 0 && iea > 0) ibin++;
295 int diff = iea - int(scaledBounds[ibin]);
296 if (diff % steps[icfg][ibin] == 0) ieaprime++;
297 } else {
298 ieaprime = iea;
299 }
300
301 m_binIndex[il].push_back(ieaprime);
302
303 if (ieaprime != temp) {
304 double binwidth = m_eaBW;
305 if (isVarBins) binwidth = m_eaBW * steps[icfg][ibin];
306 double binvalue = pastbin + binwidth;
307 pastbin = binvalue;
308 if (std::abs(binvalue) < 1e-5) binvalue = 0;
309 m_binValue[il].push_back(binvalue);
310 }
311 temp = ieaprime;
312 }
313
314 m_eaBinLocal.push_back(int(m_binValue[il].size()) - 1);
315 }
316
317 if (isMakePlots) plotMergeFactor(bounds, nDev, steps);
318}
void plotMergeFactor(std::map< int, std::vector< double > > bounds, const std::array< int, 2 > nDev, std::map< int, std::vector< int > > steps)
function to plot merging factor

◆ createPayload()

void createPayload ( )

function to generate final constants

Definition at line 410 of file CDCDedx1DCellAlgorithm.cc.

411{
412
413 B2INFO("dE/dx one cell calibration: Generating payloads");
414
415 for (unsigned int il = 0; il < m_kNGroups; il++) {
416
417 if (isMerge) {
418 unsigned int nbins = m_DBOneDCell->getNBins(getRepresentativeLayer(il));
419
420 if (int(nbins) != m_eaBin)
421 B2ERROR("merging failed because of unmatch bins (old " << m_eaBin << " new " << nbins << ")");
422
423 for (unsigned int iea = 0; iea < nbins; iea++) {
424 double prev = m_DBOneDCell->getMean(getRepresentativeLayer(il), iea);
425 m_onedcors[il][iea] *= prev;
426 }
427 }
428
429 double binsize = TMath::Pi() / m_onedcors[il].size();
430
431 auto computeAverages = [&](double angLow, double angHigh) {
432 unsigned int binLow = std::floor((angLow + TMath::Pi() / 2.0) / binsize);
433 unsigned int binHigh = std::floor((angHigh + TMath::Pi() / 2.0) / binsize);
434 double sum_new = 0.0, sum_prev = 0.0;
435 int count = 0;
436
437 for (unsigned int iea = binLow; iea < binHigh; ++iea) {
438 sum_new += m_onedcors[il][iea];
439 sum_prev += m_DBOneDCell->getMean(getRepresentativeLayer(il), iea);
440 ++count;
441 }
442
443 double avg_new = (count > 0) ? sum_new / count : 1.0;
444 double avg_prev = (count > 0) ? sum_prev / count : 1.0;
445 return std::make_pair(avg_new, avg_prev);
446 };
447
448 double negLow = -0.5, negHigh = -0.2;
449 double posLow = 0.2, posHigh = 0.5;
450
451 if (il == 2) {
452 negLow = -0.75; negHigh = -0.25;
453 posLow = 0.25; posHigh = 0.75;
454 }
455
456 auto [avgNewNeg, avgPrevNeg] = computeAverages(negLow, negHigh);
457 auto [avgNewPos, avgPrevPos] = computeAverages(posLow, posHigh);
458
459 double avgNew = (avgNewNeg + avgNewPos) / 2.0;
460 double avgPrev = (avgPrevNeg + avgPrevPos) / 2.0;
461 double scaleFactor = avgPrev / avgNew;
462
463 for (int iea = 0; iea < m_eaBin; iea++) {
464 m_onedcors[il][iea] *= scaleFactor;
465 }
466
467 if (m_chargeType > 0)
468 for (int ie = 0; ie < m_eaBin / 2; ie++) m_onedcors[il][ie] *= m_adjustFac;
469 if (m_chargeType < 0)
470 for (int ie = m_eaBin / 2; ie < m_eaBin; ie++) m_onedcors[il][ie] *= m_adjustFac;
471
472 }
473
474 //Saving constants
475 B2INFO("dE/dx Calibration done for CDCDedx1DCell");
476 std::vector<unsigned int> layerToGroup(56);
477
478 for (unsigned int layer = 0; layer < 56; layer++) {
479 if (layer < 8) layerToGroup[layer] = 0; // SL0
480 else if (layer < 14) layerToGroup[layer] = 1; // SL1
481 else layerToGroup[layer] = 2; // SL2-8
482 }
483
484 CDCDedx1DCell* gain = new CDCDedx1DCell(0, m_onedcors, layerToGroup);
485 saveCalibration(gain, "CDCDedx1DCell");
486}
unsigned int getRepresentativeLayer(unsigned int il) const
Representative CDC layer for each SL group (used to access group-wise constants): SL0 => 1,...
void saveCalibration(TClonesArray *data, const std::string &name)
Store DBArray payload with given name with default IOV.

◆ defineHisto()

void defineHisto ( std::array< std::vector< TH1D * >, 3 > & hdedxhit,
std::array< TH1D *, 3 > & hdedxlay,
std::array< TH1D *, 3 > & hentalay )

function to define histograms

Definition at line 322 of file CDCDedx1DCellAlgorithm.cc.

325{
326 for (int il = 0; il < m_kNGroups; il++) {
327
328 std::string title = Form("dedxhit dist (%s): %s ; dedxhit;entries", m_label[il].data(), m_runExp.data());
329 hdedxlay[il] = new TH1D(Form("hdedxlay%s", m_label[il].data()), "", m_dedxBin, m_dedxMin, m_dedxMax);
330 hdedxlay[il]->SetTitle(title.c_str());
331
332 Double_t* nvarBins = m_binValue[il].data();
333
334 if (isVarBins)
335 title = Form("entaRS dist (variable bins): %s: (%s); entaRS (#alpha);entries", m_label[il].data(), m_runExp.data());
336 else
337 title = Form("entaRS dist (sym. bins): %s: (%s); entaRS (#alpha);entries", m_label[il].data(), m_runExp.data());
338
339 hentalay[il] = new TH1D(Form("hentalay%s", m_label[il].data()), "", m_eaBinLocal[il], nvarBins);
340 hentalay[il]->SetTitle(title.c_str());
341
342 for (int iea = 0; iea < m_eaBinLocal[il]; iea++) {
343
344 double min = m_binValue[il].at(iea);
345 double max = m_binValue[il].at(iea + 1);
346 double width = max - min;
347
348 if (isPrintLog)
349 B2INFO("bin: " << iea << " ], min:" << min << " , max: " << max << " , width: " << width);
350
351 title = Form("%s: entaRS = (%0.03f to %0.03f)", m_label[il].data(), min, max);
352
353 hdedxhit[il].push_back(new TH1D(Form("hdedxhit_%s_bin%d", m_label[il].data(), iea),
355
356 hdedxhit[il][iea]->SetTitle(title.c_str());
357 }
358 }
359}
std::string m_runExp
add run and exp to title of plot

◆ 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();}

◆ enableExtraPlots()

void enableExtraPlots ( bool value = false)
inline

funtion to set flag active for plotting

Definition at line 115 of file CDCDedx1DCellAlgorithm.h.

115{isMakePlots = value;}

◆ 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++");
100 m_data.setResult(c_Failure);
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).");
108 m_data.setResult(c_Failure);
109 return c_Failure;
110 }
111 return execute(vecRuns, iteration, iov);
112}
static 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.
static Calibration::ExpRun convertPyExpRun(PyObject *pyObj)
Performs the conversion of PyObject to ExpRun.
ExecutionData m_data
Data specific to a SINGLE execution of the algorithm. Gets reset at the beginning of execution.

◆ 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()");
124 m_data.setResult(c_Failure);
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.");
144 m_data.setResult(c_Failure);
145 return c_Failure;
146 }
147 for (auto expRun : runs) {
148 B2DEBUG(29, "ExpRun requested = (" << expRun.first << ", " << expRun.second << ")");
149 }
150 }
151
152 m_data.setRequestedRuns(runs);
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 }
157 m_data.setRequestedIov(iov);
158 // After here, the getObject<...>(...) helpers start to work
159
161 m_data.setResult(result);
162 return result;
163}
std::vector< Calibration::ExpRun > getRunListFromAllData() const
Get the complete list of runs from inspection of collected data.
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.
@ c_Undefined
Not yet known (before execution) =4 in Python.
const std::string & getGranularity() const
Get the granularity of collected data.
virtual EResult calibrate()=0
Run algo on data - pure virtual: needs to be implemented.

◆ fillRunToInputFilesMap()

void fillRunToInputFilesMap ( )
inherited

Fill the mapping of ExpRun -> Files.

Definition at line 331 of file CalibrationAlgorithm.cc.

332{
333 m_runsToInputFiles.clear();
334 // Save TDirectory to change back at the end
335 TDirectory* dir = gDirectory;
336 RunRange* runRange;
337 // Construct the TDirectory name where we expect our objects to be
338 string runRangeObjName(getPrefix() + "/" + RUN_RANGE_OBJ_NAME);
339 for (const auto& fileName : m_inputFileNames) {
340 //Open TFile to get the objects
341 unique_ptr<TFile> f;
342 f.reset(TFile::Open(fileName.c_str(), "READ"));
343 runRange = dynamic_cast<RunRange*>(f->Get(runRangeObjName.c_str()));
344 if (runRange) {
345 // Insert or extend the run -> file mapping for this ExpRun
346 auto expRuns = runRange->getExpRunSet();
347 for (const auto& expRun : expRuns) {
348 auto runFiles = m_runsToInputFiles.find(expRun);
349 if (runFiles != m_runsToInputFiles.end()) {
350 (runFiles->second).push_back(fileName);
351 } else {
352 m_runsToInputFiles.insert(std::make_pair(expRun, std::vector<std::string> {fileName}));
353 }
354 }
355 } else {
356 B2WARNING("Missing a RunRange object for file: " << fileName);
357 }
358 }
359 dir->cd();
360}
const std::string & getPrefix() const
Get the prefix used for getting calibration data.
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...
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 521 of file CalibrationAlgorithm.cc.

522{
523 m_boundaries.clear();
524 if (m_inputFileNames.empty()) {
525 B2ERROR("There aren't any input files set. Please use CalibrationAlgorithm::setInputFiles()");
526 return m_boundaries;
527 }
528 // Reset the internal execution data just in case something is hanging around
529 m_data.reset();
530 if (runs.empty()) {
531 // Want to loop over all runs we could possibly know about
532 runs = getRunListFromAllData();
533 }
534 // Let's check that we have some now
535 if (runs.empty()) {
536 B2ERROR("No collected data in input files.");
537 return m_boundaries;
538 }
539 // In order to find run boundaries we must have collected with data granularity == 'run'
540 if (strcmp(getGranularity().c_str(), "all") == 0) {
541 B2ERROR("The data is collected with granularity='all' (exp=-1,run=-1), and we can't use that to find run boundaries.");
542 return m_boundaries;
543 }
544 m_data.setIteration(iteration);
545 // User defined setup function
546 boundaryFindingSetup(runs, iteration);
547 std::vector<ExpRun> runList;
548 // Loop over run list and call derived class "isBoundaryRequired" member function
549 for (auto currentRun : runs) {
550 runList.push_back(currentRun);
551 m_data.setRequestedRuns(runList);
552 // After here, the getObject<...>(...) helpers start to work
553 if (isBoundaryRequired(currentRun)) {
554 m_boundaries.push_back(currentRun);
555 }
556 // Only want run-by-run
557 runList.clear();
558 // Don't want memory hanging around
559 m_data.clearCalibrationData();
560 }
561 m_data.reset();
563 return m_boundaries;
564}
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()

static Calibration::ExpRun getAllGranularityExpRun ( )
inlinestaticprotectedinherited

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

Definition at line 327 of file CalibrationAlgorithm.h.

327{return m_allExpRun;}

◆ getCollectorName()

const 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;}

◆ getExpRunInfo()

void getExpRunInfo ( )

function to extract calibration run/exp

Definition at line 232 of file CDCDedx1DCellAlgorithm.cc.

233{
234
235 int cruns = 0;
236 for (auto expRun : getRunList()) {
237 if (cruns == 0) B2INFO("CDCDedxBadWires: start exp " << expRun.first << " and run " << expRun.second << "");
238 cruns++;
239 }
240
241 const auto erStart = getRunList()[0];
242 int estart = erStart.first;
243 int rstart = erStart.second;
244
245 const auto erEnd = getRunList()[cruns - 1];
246 int eend = erEnd.first;
247 int rend = erEnd.second;
248
249 updateDBObjPtrs(1, rstart, estart);
250
251 m_runExp = Form("Range (%d:%d,%d:%d)", estart, rstart, eend, rend);
252 if (m_suffix.length() > 0) m_suffix = Form("%s_e%d_r%dr%d", m_suffix.data(), estart, rstart, rend);
253 else m_suffix = Form("e%d_r%dr%d", estart, rstart, rend);
254}
static void updateDBObjPtrs(const unsigned int event, const int run, const int experiment)
Updates any DBObjPtrs by calling update(event) for DBStore.
const std::vector< Calibration::ExpRun > & getRunList() const
Get the list of runs for which calibration is called.

◆ 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()

const std::string & getGranularity ( ) const
inlineinherited

Get the granularity of collected data.

Definition at line 188 of file CalibrationAlgorithm.h.

188{return m_granularityOfData;};

◆ getGranularityFromData()

string getGranularityFromData ( ) const
protectedinherited

Get the granularity of collected data.

Definition at line 384 of file CalibrationAlgorithm.cc.

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

◆ getInputJsonValue()

template<class T>
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 326 of file CalibrationAlgorithm.cc.

327{
329}
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(); }

◆ getObjectPtr()

template<class T>
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)
288 fillRunToInputFilesMap();
289 return getObjectPtr<T>(name, m_data.getRequestedRuns());
290 }

◆ getOutputJsonValue()

template<class T>
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();}

◆ getPayloadValues()

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

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

Definition at line 207 of file CalibrationAlgorithm.h.

207{return m_data.getPayloadValues();}

◆ getPrefix()

const 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;}

◆ getRepresentativeLayer()

unsigned int getRepresentativeLayer ( unsigned int il) const
inline

Representative CDC layer for each SL group (used to access group-wise constants): SL0 => 1, SL1 => 9, SL2-8 => 17.

Definition at line 137 of file CDCDedx1DCellAlgorithm.h.

138 {
139 static const std::array<unsigned int, m_kNGroups> repLayer = {1, 9, 17};
140 return repLayer.at(il);
141 }

◆ 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 319 of file CalibrationAlgorithm.cc.

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

◆ getRunRangeFromAllData()

RunRange getRunRangeFromAllData ( ) const
inherited

Get the complete RunRange from inspection of collected data.

Definition at line 362 of file CalibrationAlgorithm.cc.

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

◆ getTruncatedBins()

void getTruncatedBins ( TH1D * hist,
int & binlow,
int & binhigh )

function to get bins of truncation from histogram

Definition at line 362 of file CDCDedx1DCellAlgorithm.cc.

363{
364
365 //calculating truncation average
366 double sum = hist->Integral();
367 if (sum <= 0 || hist->GetNbinsX() <= 0) {
368 binlow = 1; binhigh = 1;
369 return ;
370 }
371
372 binlow = 1.0; binhigh = 1.0;
373 double sumPer5 = 0.0, sumPer75 = 0.0;
374 for (int ibin = 1; ibin <= hist->GetNbinsX(); ibin++) {
375 double bcdedx = hist->GetBinContent(ibin);
376 if (sumPer5 <= m_truncMin * sum) {
377 sumPer5 += bcdedx;
378 binlow = ibin;
379 }
380 if (sumPer75 <= m_truncMax * sum) {
381 sumPer75 += bcdedx;
382 binhigh = ibin;
383 }
384 }
385 return;
386}

◆ getTruncationMean()

double getTruncationMean ( TH1D * hist,
int binlow,
int binhigh )

function to get truncated mean

Definition at line 389 of file CDCDedx1DCellAlgorithm.cc.

390{
391
392 //calculating truncation average
393 if (hist->Integral() < 100) return 1.0;
394
395 if (binlow <= 0 || binhigh > hist->GetNbinsX())return 1.0;
396
397 double binweights = 0., sumofbc = 0.;
398 for (int ibin = binlow; ibin <= binhigh; ibin++) {
399 double bcdedx = hist->GetBinContent(ibin);
400 if (bcdedx > 0) {
401 binweights += (bcdedx * hist->GetBinCenter(ibin));
402 sumofbc += bcdedx;
403 }
404 }
405 if (sumofbc > 0) return binweights / sumofbc;
406 else return 1.0;
407}

◆ getVecInputFileNames()

const 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 PXDAnalyticGainCalibrationAlgorithm, PXDValidationAlgorithm, SVD3SampleCoGTimeCalibrationAlgorithm, SVD3SampleELSTimeCalibrationAlgorithm, SVDCoGTimeCalibrationAlgorithm, TestBoundarySettingAlgorithm, and TestCalibrationAlgorithm.

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 503 of file CalibrationAlgorithm.cc.

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

◆ plotConstants()

void plotConstants ( )

function to draw the old/new final constants

Definition at line 672 of file CDCDedx1DCellAlgorithm.cc.

673{
674
675 //Draw New/Old final constants
676 TH1D* hnewconst[m_kNGroups], *holdconst[m_kNGroups];
677 double min[m_kNGroups], max[m_kNGroups];
678
679 for (unsigned int il = 0; il < m_kNGroups; il++) {
680 unsigned int nbins = m_DBOneDCell->getNBins(getRepresentativeLayer(il));
681
682 std::string title = Form("final calibration const dist (%s): %s; entaRS (#alpha); entries", m_label[il].data(), m_runExp.data());
683 hnewconst[il] = new TH1D(Form("hnewconst_%s", m_label[il].data()), "", m_eaBin, m_eaMin, m_eaMax);
684 hnewconst[il]->SetTitle(Form("%s", title.data()));
685
686 title = Form("old calibration const dist (%s): %s; entaRS (#alpha); entries", m_label[il].data(), m_runExp.data());
687 holdconst[il] = new TH1D(Form("holdconst_%s", m_label[il].data()), "", m_eaBin, m_eaMin, m_eaMax);
688 holdconst[il]->SetTitle(Form("%s", title.data()));
689
690 for (unsigned int iea = 0; iea < nbins; iea++) {
691 double prev = m_DBOneDCell->getMean(getRepresentativeLayer(il), iea);
692 holdconst[il]->SetBinContent(iea + 1, prev);
693 hnewconst[il]->SetBinContent(iea + 1, m_onedcors[il][iea]);
694 }
695 min[il] = hnewconst[il]->GetMinimum();
696 max[il] = hnewconst[il]->GetMaximum();
697 }
698
699 //Ploting final constants
700 double globalMin = min[0];
701 double globalMax = max[0];
702 for (int il = 1; il < m_kNGroups; il++) {
703 if (min[il] < globalMin) globalMin = min[il];
704 if (max[il] > globalMax) globalMax = max[il];
705 }
706
707 gStyle->SetOptStat("ne");
708 TCanvas cfconst("cfconst", "Final calibration constants", 1600, 600);
709 cfconst.Divide(3, 1);
710
711 for (int il = 0; il < m_kNGroups; il++) {
712 cfconst.cd(il + 1);
713 hnewconst[il]->GetYaxis()->SetRangeUser(globalMin * 0.95, globalMax * 1.05);
714 hnewconst[il]->SetLineColor(kBlack);
715 hnewconst[il]->Draw("histo");
716 holdconst[il]->SetLineColor(kRed);
717 holdconst[il]->Draw("histo same");
718
719 auto legend = new TLegend(0.4, 0.75, 0.56, 0.85);
720 legend->AddEntry(holdconst[il], "Old", "lep");
721 legend->AddEntry(hnewconst[il], "New", "lep");
722 legend->Draw();
723 }
724
725 cfconst.SaveAs(Form("cdcdedx_1dcell_fconsts%s.pdf", m_suffix.data()));
726 cfconst.SaveAs(Form("cdcdedx_1dcell_fconsts%s.root", m_suffix.data()));
727
728 for (int il = 0; il < m_kNGroups; il++) {
729 delete hnewconst[il];
730 delete holdconst[il];
731 }
732}

◆ plotdedxHist()

void plotdedxHist ( std::array< std::vector< TH1D * >, m_kNGroups > & hdedxhit)

function to draw the dE/dx histogram in enta bins

Definition at line 526 of file CDCDedx1DCellAlgorithm.cc.

527{
528
529 TCanvas ctmp("tmp", "tmp", 1200, 1200);
530 ctmp.Divide(4, 4);
531 std::stringstream psname;
532
533 psname << Form("cdcdedx_1dcell_dedxhit%s.pdf[", m_suffix.data());
534 ctmp.Print(psname.str().c_str());
535 psname.str("");
536 psname << Form("cdcdedx_1dcell_dedxhit%s.pdf", m_suffix.data());
537
538 for (int il = 0; il < m_kNGroups; il++) {
539
540 for (int jea = 0; jea < m_eaBinLocal[il]; jea++) {
541
542 int minbin = std::stoi(hdedxhit[il][jea]->GetXaxis()->GetTitle());
543 int maxbin = std::stoi(hdedxhit[il][jea]->GetYaxis()->GetTitle());
544
545 ctmp.cd(jea % 16 + 1);
546 hdedxhit[il][jea]->SetFillColor(4 + il);
547
548 hdedxhit[il][jea]->SetTitle(Form("%s;dedxhit;entries", hdedxhit[il][jea]->GetTitle()));
549 hdedxhit[il][jea]->DrawClone("hist");
550 TH1D* htempC = (TH1D*)hdedxhit[il][jea]->Clone(Form("%sc2", hdedxhit[il][jea]->GetName()));
551 htempC->GetXaxis()->SetRange(minbin, maxbin);
552 htempC->SetFillColor(kGray);
553 htempC->DrawClone("same hist");
554
555 if (jea % 16 == 15 || (jea == m_eaBinLocal[il] - 1)) {
556 ctmp.Print(psname.str().c_str());
557 gPad->Clear("D");
558 ctmp.Clear("D");
559 }
560 delete htempC;
561 }
562 }
563 psname.str("");
564 psname << Form("cdcdedx_1dcell_dedxhit%s.pdf]", m_suffix.data());
565 ctmp.Print(psname.str().c_str());
566}

◆ plotEventStats()

void plotEventStats ( )

function to draw the stats plots

Definition at line 735 of file CDCDedx1DCellAlgorithm.cc.

736{
737
738 TCanvas cstats("cstats", "cstats", 1000, 500);
739 cstats.SetBatch(kTRUE);
740 cstats.Divide(2, 1);
741
742 cstats.cd(1);
743 auto hestats = getObjectPtr<TH1I>("hestats");
744 if (hestats) {
745 hestats->SetName(Form("hestats_%s", m_suffix.data()));
746 hestats->SetStats(0);
747 hestats->DrawCopy("");
748 }
749
750 cstats.cd(2);
751 auto htstats = getObjectPtr<TH1I>("htstats");
752 if (htstats) {
753 htstats->SetName(Form("htstats_%s", m_suffix.data()));
754 htstats->SetStats(0);
755 htstats->DrawCopy("");
756 }
757 cstats.Print(Form("cdcdedx_1dcell_stats_%s.pdf", m_suffix.data()));
758}

◆ plotLayerDist()

void plotLayerDist ( std::array< TH1D *, m_kNGroups > & hdedxlay)

function to draw dedx dist.

for Inner/outer layer

Definition at line 569 of file CDCDedx1DCellAlgorithm.cc.

570{
571
572 TCanvas cdedxlayer("layerdedxhit", "Inner and Outer Layer dedxhit dist", 900, 400);
573 cdedxlayer.Divide(3, 1);
574
575 for (int il = 0; il < m_kNGroups; il++) {
576 int minlay = 0, maxlay = 0;
577 if (isFixTrunc) {
578 minlay = std::stoi(hdedxlay[il]->GetXaxis()->GetTitle());
579 maxlay = std::stoi(hdedxlay[il]->GetYaxis()->GetTitle());
580 double lowedge = hdedxlay[il]->GetXaxis()->GetBinLowEdge(minlay);
581 double upedge = hdedxlay[il]->GetXaxis()->GetBinUpEdge(maxlay);
582 hdedxlay[il]->SetTitle(Form("%s, trunc #rightarrow: %0.02f - %0.02f;dedxhit;entries", hdedxlay[il]->GetTitle(), lowedge, upedge));
583 }
584
585 cdedxlayer.cd(il + 1);
586 hdedxlay[il]->SetFillColor(kYellow);
587 hdedxlay[il]->Draw("histo");
588
589 if (isFixTrunc) {
590 TH1D* hdedxlayC = (TH1D*)hdedxlay[il]->Clone(Form("hdedxlayC%d", il));
591 hdedxlayC->GetXaxis()->SetRange(minlay, maxlay);
592 hdedxlayC->SetFillColor(kAzure + 1);
593 hdedxlayC->Draw("same histo");
594 }
595 }
596
597 cdedxlayer.SaveAs(Form("cdcdedx_1dcell_dedxlayer_%s.pdf", m_suffix.data()));
598 cdedxlayer.SaveAs(Form("cdcdedx_1dcell_dedxlayer_%s.root", m_suffix.data()));
599}

◆ plotMergeFactor()

void plotMergeFactor ( std::map< int, std::vector< double > > bounds,
const std::array< int, 2 > nDev,
std::map< int, std::vector< int > > steps )

function to plot merging factor

Definition at line 489 of file CDCDedx1DCellAlgorithm.cc.

492{
493 TCanvas cmfactor("cmfactor", "Merging factors", 800, 400);
494 cmfactor.Divide(2, 1);
495
496 std::array<TH1I*, 2> hists{};
497
498 for (int icfg = 0; icfg < 2; icfg++) {
499 Double_t* nvarBins = bounds[icfg].data();
500
501 hists[icfg] = new TH1I(Form("hist_cfg%d", icfg), "", nDev[icfg], nvarBins);
502
503 if (icfg == 0)
504 hists[icfg]->SetTitle("Merging factor for SL0/SL1 bins;binindex;merge-factors");
505 else
506 hists[icfg]->SetTitle("Merging factor for SL2-8 bins;binindex;merge-factors");
507
508 for (int ibin = 0; ibin < nDev[icfg]; ibin++) {
509 hists[icfg]->SetBinContent(ibin + 1, steps[icfg][ibin]);
510 }
511
512 cmfactor.cd(icfg + 1);
513 hists[icfg]->SetFillColor(kYellow);
514 hists[icfg]->Draw("hist");
515 }
516
517 cmfactor.SaveAs(Form("cdcdedx_1dcell_mergefactor%s.pdf", m_suffix.data()));
518 cmfactor.SaveAs(Form("cdcdedx_1dcell_mergefactor%s.root", m_suffix.data()));
519
520 for (int icfg = 0; icfg < 2; icfg++) {
521 delete hists[icfg];
522 }
523}

◆ plotQaPars()

void plotQaPars ( std::array< TH1D *, m_kNGroups > & hentalay,
TH2D * hptcosth )

function to draw pt vs costh and entrance angle distribution for Inner/Outer layer

Definition at line 602 of file CDCDedx1DCellAlgorithm.cc.

603{
604
605 TCanvas ceadist("ceadist", "Enta distributions", 1600, 800);
606 ceadist.Divide(3, 1);
607
608 for (int il = 0; il < m_kNGroups; il++) {
609
610 ceadist.cd(il + 1);
611 gPad->SetLogy();
612 hentalay[il]->SetFillColor(kYellow);
613 hentalay[il]->Draw("hist");
614 }
615
616 TCanvas cptcos("cptcos", "pt vs costh dist.", 400, 400);
617 cptcos.cd();
618 hptcosth->Draw("colz");
619
620 cptcos.SaveAs(Form("cdcdedx_ptcosth_%s.pdf", m_suffix.data()));
621 ceadist.SaveAs(Form("cdcdedx_1dcell_enta%s.pdf", m_suffix.data()));
622 ceadist.SaveAs(Form("cdcdedx_1dcell_enta%s.root", m_suffix.data()));
623}

◆ plotRelConst()

void plotRelConst ( std::vector< double > tempconst,
std::vector< double > layerconst,
int il )

function to draw symm/Var layer constant

Definition at line 626 of file CDCDedx1DCellAlgorithm.cc.

627{
628
629 TH1D* hconst, *hconstvar;
630
631 Double_t* nvarBins;
632 nvarBins = &m_binValue[il][0];
633
634 hconst = new TH1D(Form("hconst%s", m_label[il].data()), "", m_eaBin, m_eaMin, m_eaMax);
635 std::string title = Form("calibration const dist: %s: (%s); entaRS (#alpha); entries", m_label[il].data(), m_runExp.data());
636 hconst->SetTitle(Form("%s", title.data()));
637
638 hconstvar = new TH1D(Form("hconstvar%s", m_label[il].data()), "", m_eaBinLocal[il], nvarBins);
639 title = Form("calibration const dist (var bins): %s: (%s); entaRS (#alpha);entries", m_label[il].data(), m_runExp.data());
640 hconstvar->SetTitle(Form("%s", title.data()));
641
642 if (isVarBins) {
643 for (int iea = 0; iea < m_eaBinLocal[il]; iea++)
644 hconstvar->SetBinContent(iea + 1, tempconst.at(iea));
645 }
646
647 for (int jea = 0; jea < m_eaBin; jea++) hconst->SetBinContent(jea + 1, layerconst.at(jea));
648
649 gStyle->SetOptStat("ne");
650 TCanvas cconst("cconst", "calibration Constants", 800, 400);
651 if (isVarBins) {
652 cconst.Divide(2, 1);
653 cconst.SetWindowSize(1000, 800);
654 }
655
656 cconst.cd(1);
657 hconst->SetFillColor(kYellow);
658 hconst->Draw("histo");
659 if (isVarBins) {
660 cconst.cd(2);
661 hconstvar->SetFillColor(kBlue);
662 hconstvar->Draw("hist");
663 }
664 cconst.SaveAs(Form("cdcdedx_1dcell_relconst%s_%s.pdf", m_label[il].data(), m_suffix.data()));
665 cconst.SaveAs(Form("cdcdedx_1dcell_relconst%s_%s.root", m_label[il].data(), m_suffix.data()));
666
667 delete hconst;
668 delete hconstvar;
669}

◆ 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();}

◆ rotationalBin()

int rotationalBin ( int nbin,
int ibin )
inline

class function to set rotation symmetry

Definition at line 161 of file CDCDedx1DCellAlgorithm.h.

162 {
163 if (nbin % 4 != 0)return ibin;
164 int jbin = ibin;
165 if (ibin < nbin / 4) jbin = ibin + nbin / 2 ;
166 else if (ibin >= 3 * nbin / 4) jbin = ibin - nbin / 2 ;
167 return jbin;
168 }

◆ 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{
299 saveCalibration(data, name, m_data.getRequestedIov());
300}

◆ 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:150

◆ 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{
294 saveCalibration(data, name, m_data.getRequestedIov());
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}

◆ setAdjustmentFactor()

void setAdjustmentFactor ( int value)
inline

set adjustment factor

Definition at line 95 of file CDCDedx1DCellAlgorithm.h.

95{m_adjustFac = value;}

◆ setBaselineFactor()

void setBaselineFactor ( double charge,
double factor )
inline

adjust baseline based on charge or global overall works for only single charge or both

Definition at line 147 of file CDCDedx1DCellAlgorithm.h.

148 {
149
150 m_adjustFac = factor;
151 if (charge < 0)m_chargeType = -1.0;
152 else if (charge > 0)m_chargeType = 1.0;
153 else if (charge == 0)m_chargeType = 0.0;
154 else
155 B2FATAL("Choose charge value either +/-1 or 0");
156 }

◆ setChargeType()

void setChargeType ( int value)
inline

set charge type

Definition at line 90 of file CDCDedx1DCellAlgorithm.h.

90{m_chargeType = value;}

◆ setCosLimit()

void setCosLimit ( double value)
inline

function to set costheta limit

Definition at line 72 of file CDCDedx1DCellAlgorithm.h.

72{m_cosMax = value;}

◆ 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;}

◆ setEntaBins()

void setEntaBins ( unsigned int value = 316)
inline

function to set number of entrance angle bins for calibration

Definition at line 52 of file CDCDedx1DCellAlgorithm.h.

52{m_eaB = value;}

◆ setEntaRange()

void setEntaRange ( double min = -1.0,
double max = 1.0 )
inline

function to set min/max range of entrance angle for calibration

Definition at line 47 of file CDCDedx1DCellAlgorithm.h.

47{m_eaMin = min; m_eaMax = max;}

◆ setHistBins()

void setHistBins ( int value = 250)
inline

function to set nbins of dedx dist for calibration

Definition at line 57 of file CDCDedx1DCellAlgorithm.h.

57{m_dedxBin = value;}

◆ setHistRange()

void setHistRange ( double min = 0.0,
double max = 5.0 )
inline

function to set min/max range of dedx dist for calibration

Definition at line 62 of file CDCDedx1DCellAlgorithm.h.

62{m_dedxMin = min; m_dedxMax = max;}

◆ setInputFileNames() [1/2]

void setInputFileNames ( const 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 m_granularityOfData
Granularity of input data. This only changes when the input files change so it isn't specific to an e...
void setInputFileNames(PyObject *inputFileNames)
Set the input file names used for this algorithm from a Python list.
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.

◆ setInputFileNames() [2/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}
Scalar convertPythonObject(const boost::python::object &pyObject, Scalar)
Convert from Python to given type.

◆ setLayerTrunc()

void setLayerTrunc ( bool value = false)
inline

function to set truncation method (local vs global)

Definition at line 100 of file CDCDedx1DCellAlgorithm.h.

100{isFixTrunc = value;}

◆ setMergePayload()

void setMergePayload ( bool value)
inline

set false if generating absolute (not relative) payload

Definition at line 126 of file CDCDedx1DCellAlgorithm.h.

126{ isMerge = value;}

◆ setOutputJsonValue()

template<class T>
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;}

◆ setPrintLog()

void setPrintLog ( bool value = false)
inline

funtion to set flag to print log

Definition at line 120 of file CDCDedx1DCellAlgorithm.h.

120{isPrintLog = value;}

◆ setPtLimit()

void setPtLimit ( double value)
inline

function to set pt limit

Definition at line 67 of file CDCDedx1DCellAlgorithm.h.

67{m_ptMax = value;}

◆ setRotSymmetry()

void setRotSymmetry ( bool value)
inline

set rotation sys to copy constants from one region to other

Definition at line 110 of file CDCDedx1DCellAlgorithm.h.

110{isRotSymm = value;}

◆ setSplitFactor()

void setSplitFactor ( int value)
inline

set bin split factor for all range

Definition at line 85 of file CDCDedx1DCellAlgorithm.h.

85{m_binSplit = value;}

◆ setSuffix()

void setSuffix ( const std::string & value)
inline

adding suffix to control plots

Definition at line 131 of file CDCDedx1DCellAlgorithm.h.

131{m_suffix = value;}

◆ setTrucationBins()

void setTrucationBins ( double lowedge,
double upedge )
inline

function to set bins of truncation from histogram

Definition at line 77 of file CDCDedx1DCellAlgorithm.h.

78 {
79 m_truncMin = lowedge; m_truncMax = upedge ;
80 }

◆ setVariableBins()

void setVariableBins ( bool value)
inline

Set Var bins flag to on or off.

Definition at line 105 of file CDCDedx1DCellAlgorithm.h.

105{isVarBins = value;}

◆ updateDBObjPtrs()

void updateDBObjPtrs ( const unsigned int event,
const int run,
const int experiment )
staticprotectedinherited

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

Definition at line 405 of file CalibrationAlgorithm.cc.

406{
407 // Construct an EventMetaData object but NOT in the Datastore
408 EventMetaData emd(event, run, experiment);
409 // Explicitly update while avoiding registering a Datastore object
411 // Also update the intra-run objects to the event at the same time (maybe unnecessary...)
413}
static DBStore & Instance()
Instance of a singleton DBStore.
Definition DBStore.cc:26
void updateEvent()
Updates all intra-run dependent objects.
Definition DBStore.cc:140
void update()
Updates all objects that are outside their interval of validity.
Definition DBStore.cc:77

Member Data Documentation

◆ isFixTrunc

bool isFixTrunc
private

true = fix window for all out/inner layers

Definition at line 267 of file CDCDedx1DCellAlgorithm.h.

◆ isMakePlots

bool isMakePlots
private

produce plots for status

Definition at line 270 of file CDCDedx1DCellAlgorithm.h.

◆ isMerge

bool isMerge
private

print more debug information

Definition at line 272 of file CDCDedx1DCellAlgorithm.h.

◆ isPrintLog

bool isPrintLog
private

print more debug information

Definition at line 271 of file CDCDedx1DCellAlgorithm.h.

◆ isRotSymm

bool isRotSymm
private

if rotation symmetry requested

Definition at line 269 of file CDCDedx1DCellAlgorithm.h.

◆ isVarBins

bool isVarBins
private

true: if variable bin size is requested

Definition at line 268 of file CDCDedx1DCellAlgorithm.h.

◆ m_adjustFac

double m_adjustFac
private

factor with that one what to adjust baseline

Definition at line 265 of file CDCDedx1DCellAlgorithm.h.

◆ m_allExpRun

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

allExpRun

Definition at line 364 of file CalibrationAlgorithm.h.

◆ m_binIndex

std::array<std::vector<int>, m_kNGroups> m_binIndex
private

symm/Var bin numbers

Definition at line 279 of file CDCDedx1DCellAlgorithm.h.

◆ m_binSplit

int m_binSplit
private

multiply nbins by this factor in full range

Definition at line 262 of file CDCDedx1DCellAlgorithm.h.

◆ m_binValue

std::array<std::vector<double>, m_kNGroups> m_binValue
private

enta Var bin values

Definition at line 280 of file CDCDedx1DCellAlgorithm.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_chargeType

double m_chargeType
private

charge type for baseline adj

Definition at line 264 of file CDCDedx1DCellAlgorithm.h.

◆ m_cosMax

double m_cosMax
private

a limit on cos theta

Definition at line 257 of file CDCDedx1DCellAlgorithm.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_DBOneDCell

DBObjPtr<CDCDedx1DCell> m_DBOneDCell
private

One cell correction DB object.

Definition at line 284 of file CDCDedx1DCellAlgorithm.h.

◆ m_dedxBin

int m_dedxBin
private

of bins for dedxhit range

Definition at line 254 of file CDCDedx1DCellAlgorithm.h.

◆ m_dedxMax

double m_dedxMax
private

upper edge of dedxhit

Definition at line 253 of file CDCDedx1DCellAlgorithm.h.

◆ m_dedxMin

double m_dedxMin
private

lower edge of dedxhit

Definition at line 252 of file CDCDedx1DCellAlgorithm.h.

◆ m_description

std::string m_description {""}
privateinherited

Description of the algorithm.

Definition at line 385 of file CalibrationAlgorithm.h.

385{""};

◆ m_eaB

int m_eaB
private

reset # of bins for entrance angle for each experiment

Definition at line 250 of file CDCDedx1DCellAlgorithm.h.

◆ m_eaBin

int m_eaBin
private

of bins for entrance angle

Definition at line 249 of file CDCDedx1DCellAlgorithm.h.

◆ m_eaBinLocal

std::vector<int> m_eaBinLocal
private

of var bins for enta angle

Definition at line 278 of file CDCDedx1DCellAlgorithm.h.

◆ m_eaBW

double m_eaBW
private

binwdith of entrance angle bin

Definition at line 248 of file CDCDedx1DCellAlgorithm.h.

◆ m_eaMax

double m_eaMax
private

upper edge of entrance angle

Definition at line 247 of file CDCDedx1DCellAlgorithm.h.

◆ m_eaMin

double m_eaMin
private

lower edge of entrance angle

Definition at line 246 of file CDCDedx1DCellAlgorithm.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_kNGroups

int m_kNGroups = 3
staticconstexpr

SL grouping: inner (SL0), middle (SL1), outer (SL2–8)

Definition at line 32 of file CDCDedx1DCellAlgorithm.h.

◆ m_label

std::string m_label[m_kNGroups] = {"SL0", "SL1", "SL2-8"}
private

add inner/outer superlayer label

Definition at line 276 of file CDCDedx1DCellAlgorithm.h.

276{"SL0", "SL1", "SL2-8"};

◆ m_onedcors

std::vector<std::vector<double> > m_onedcors
private

final vectors of calibration

Definition at line 282 of file CDCDedx1DCellAlgorithm.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.

388{""};

◆ m_ptMax

double m_ptMax
private

a limit on transverse momentum

Definition at line 256 of file CDCDedx1DCellAlgorithm.h.

◆ m_runExp

std::string m_runExp
private

add run and exp to title of plot

Definition at line 275 of file CDCDedx1DCellAlgorithm.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.

◆ m_suffix

std::string m_suffix
private

add suffix to all plot name

Definition at line 274 of file CDCDedx1DCellAlgorithm.h.

◆ m_truncMax

double m_truncMax
private

upper threshold on truncation

Definition at line 260 of file CDCDedx1DCellAlgorithm.h.

◆ m_truncMin

double m_truncMin
private

lower threshold on truncation

Definition at line 259 of file CDCDedx1DCellAlgorithm.h.


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