Belle II Software  light-2212-foldex
TrackIsoCalculatorModule Class Reference

Calculate track isolation variables on the input ParticleList. More...

#include <TrackIsoCalculatorModule.h>

Inheritance diagram for TrackIsoCalculatorModule:
Collaboration diagram for TrackIsoCalculatorModule:

Public Types

enum  EModulePropFlags {
  c_Input = 1 ,
  c_Output = 2 ,
  c_ParallelProcessingCertified = 4 ,
  c_HistogramManager = 8 ,
  c_InternalSerializer = 16 ,
  c_TerminateInAllProcesses = 32 ,
  c_DontCollectStatistics = 64
}
 Each module can be tagged with property flags, which indicate certain features of the module. More...
 
typedef ModuleCondition::EAfterConditionPath EAfterConditionPath
 Forward the EAfterConditionPath definition from the ModuleCondition.
 

Public Member Functions

 TrackIsoCalculatorModule ()
 Constructor: Sets the description, the properties and the parameters of the module.
 
 ~TrackIsoCalculatorModule () override
 Destructor, use this to clean up anything you created in the constructor.
 
void initialize () override
 Use this to initialize resources or memory your module needs. More...
 
void event () override
 Called once for each event. More...
 
virtual std::vector< std::string > getFileNames (bool outputFiles)
 Return a list of output filenames for this modules. More...
 
virtual void beginRun ()
 Called when entering a new run. More...
 
virtual void endRun ()
 This method is called if the current run ends. More...
 
virtual void terminate ()
 This method is called at the end of the event processing. More...
 
const std::string & getName () const
 Returns the name of the module. More...
 
const std::string & getType () const
 Returns the type of the module (i.e. More...
 
const std::string & getPackage () const
 Returns the package this module is in.
 
const std::string & getDescription () const
 Returns the description of the module.
 
void setName (const std::string &name)
 Set the name of the module. More...
 
void setPropertyFlags (unsigned int propertyFlags)
 Sets the flags for the module properties. More...
 
LogConfiggetLogConfig ()
 Returns the log system configuration.
 
void setLogConfig (const LogConfig &logConfig)
 Set the log system configuration.
 
void setLogLevel (int logLevel)
 Configure the log level.
 
void setDebugLevel (int debugLevel)
 Configure the debug messaging level.
 
void setAbortLevel (int abortLevel)
 Configure the abort log level.
 
void setLogInfo (int logLevel, unsigned int logInfo)
 Configure the printed log information for the given level. More...
 
void if_value (const std::string &expression, const std::shared_ptr< Path > &path, EAfterConditionPath afterConditionPath=EAfterConditionPath::c_End)
 Add a condition to the module. More...
 
void if_false (const std::shared_ptr< Path > &path, EAfterConditionPath afterConditionPath=EAfterConditionPath::c_End)
 A simplified version to add a condition to the module. More...
 
void if_true (const std::shared_ptr< Path > &path, EAfterConditionPath afterConditionPath=EAfterConditionPath::c_End)
 A simplified version to set the condition of the module. More...
 
bool hasCondition () const
 Returns true if at least one condition was set for the module.
 
const ModuleConditiongetCondition () const
 Return a pointer to the first condition (or nullptr, if none was set)
 
const std::vector< ModuleCondition > & getAllConditions () const
 Return all set conditions for this module.
 
bool evalCondition () const
 If at least one condition was set, it is evaluated and true returned if at least one condition returns true. More...
 
std::shared_ptr< PathgetConditionPath () const
 Returns the path of the last true condition (if there is at least one, else reaturn a null pointer). More...
 
Module::EAfterConditionPath getAfterConditionPath () const
 What to do after the conditional path is finished. More...
 
std::vector< std::shared_ptr< Path > > getAllConditionPaths () const
 Return all condition paths currently set (no matter if the condition is true or not).
 
bool hasProperties (unsigned int propertyFlags) const
 Returns true if all specified property flags are available in this module. More...
 
bool hasUnsetForcedParams () const
 Returns true and prints error message if the module has unset parameters which the user has to set in the steering file.
 
const ModuleParamListgetParamList () const
 Return module param list.
 
template<typename T >
ModuleParam< T > & getParam (const std::string &name) const
 Returns a reference to a parameter. More...
 
bool hasReturnValue () const
 Return true if this module has a valid return value set.
 
int getReturnValue () const
 Return the return value set by this module. More...
 
std::shared_ptr< PathElementclone () const override
 Create an independent copy of this module. More...
 
std::shared_ptr< boost::python::list > getParamInfoListPython () const
 Returns a python list of all parameters. More...
 

Static Public Member Functions

static void exposePythonAPI ()
 Exposes methods of the Module class to Python.
 

Protected Member Functions

virtual void def_initialize ()
 Wrappers to make the methods without "def_" prefix callable from Python. More...
 
virtual void def_beginRun ()
 Wrapper method for the virtual function beginRun() that has the implementation to be used in a call from Python.
 
virtual void def_event ()
 Wrapper method for the virtual function event() that has the implementation to be used in a call from Python.
 
virtual void def_endRun ()
 This method can receive that the current run ends as a call from the Python side. More...
 
virtual void def_terminate ()
 Wrapper method for the virtual function terminate() that has the implementation to be used in a call from Python.
 
void setDescription (const std::string &description)
 Sets the description of the module. More...
 
void setType (const std::string &type)
 Set the module type. More...
 
template<typename T >
void addParam (const std::string &name, T &paramVariable, const std::string &description, const T &defaultValue)
 Adds a new parameter to the module. More...
 
template<typename T >
void addParam (const std::string &name, T &paramVariable, const std::string &description)
 Adds a new enforced parameter to the module. More...
 
void setReturnValue (int value)
 Sets the return value for this module as integer. More...
 
void setReturnValue (bool value)
 Sets the return value for this module as bool. More...
 
void setParamList (const ModuleParamList &params)
 Replace existing parameter list.
 

Private Member Functions

double getDistAtDetSurface (const Particle *iParticle, const Particle *jParticle, const std::string &detLayerName) const
 Calculate the distance between the points where the two input extrapolated track helices cross the given detector layer's cylindrical surface.
 
double getIsoScore (const Particle *iParticle) const
 Define a semi-continuous variable to quantify the isolation of a standard charged particle in the given detector $d$ with $N_{d}$ layers in total. More...
 
double getDistThreshold (Const::EDetector det, int layer) const
 Get the threshold value per detctor layer for the distance to closest ext. More...
 
bool onlySelectedStdChargedInDecay ()
 Check whether input particle list and reference list are of a valid charged stable particle.
 
Const::EDetector getDetEnum (const std::string &detName) const
 Get the enum type for this detector name.
 
std::list< ModulePtrgetModules () const override
 no submodules, return empty list
 
std::string getPathString () const override
 return the module name.
 
void setParamPython (const std::string &name, const boost::python::object &pyObj)
 Implements a method for setting boost::python objects. More...
 
void setParamPythonDict (const boost::python::dict &dictionary)
 Implements a method for reading the parameter values from a boost::python dictionary. More...
 

Private Attributes

StoreArray< Particlem_particles
 StoreArray of Particles.
 
std::string m_decayString
 The name of the input charged stable particle list, or composite particle w/ charged stable daughters for which distances are to be calculated.
 
unsigned short m_nSelectedDaughters
 The number of selected daughters in the decay string.
 
std::string m_pListReferenceName
 The name of the input ParticleList of reference tracks.
 
std::string m_detName
 The name of the detector at whose inner (cylindrical) surface we extrapolate each track's polar and azimuthal angle.
 
unsigned int m_nLayers
 The number of layers for the input detector.
 
std::unordered_map< std::string, std::string > m_detLayerToDistVariable
 Map that associates to each detector layer the name of the variable representing the distance to the closest particle in the reference list, based on the track helix extrapolation. More...
 
std::unordered_map< std::string, std::string > m_detLayerToRefPartIdxVariable
 Map that associates to each detector layer the name of the variable representing the mdst array index of the closest particle in the reference list. More...
 
std::string m_isoScoreVariable
 The name of the variable representing the track isolation score in this detector. More...
 
std::unordered_map< std::string, std::vector< int > > m_detToLayers
 Map that associates to each detector its list of valid layers. More...
 
std::unordered_map< std::string, DetSurfCylBoundariesm_detLayerToSurfBoundaries
 Map that associates to each detector layer its valid cylindrical surface's boundaries. More...
 
std::map< std::pair< std::string, int >, double > m_distThreshPerDetLayer
 Threshold values for the distance (in [cm]) to closest ext. More...
 
StoreObjPtr< EventMetaDatam_event_metadata
 The event information. More...
 
StoreObjPtr< ParticleListm_pListTarget
 The input ParticleList object for which distances are to be calculated.
 
DecayDescriptor m_decaydescriptor
 < Decay descriptor of decays to look for.
 
StoreObjPtr< ParticleListm_pListReference
 The input ParticleList object of reference tracks.
 
bool m_useHighestProbMassForExt
 If this option is set, the helix extrapolation for the target and reference particles will use the track fit result for the most probable mass hypothesis, namely, the one that gives the highest chi2Prob of the fit.
 
bool m_excludePIDDetWeights
 Exclude the PID detector weights for the isolation score definition.
 
std::string m_payloadName
 The name of the database payload object with the MVA weights.
 
std::unique_ptr< DBObjPtr< PIDDetectorWeights > > m_DBWeights
 Interface to get the database payload with the PID detector weights.
 
std::string m_name
 The name of the module, saved as a string (user-modifiable)
 
std::string m_type
 The type of the module, saved as a string.
 
std::string m_package
 Package this module is found in (may be empty).
 
std::string m_description
 The description of the module.
 
unsigned int m_propertyFlags
 The properties of the module as bitwise or (with |) of EModulePropFlags.
 
LogConfig m_logConfig
 The log system configuration of the module.
 
ModuleParamList m_moduleParamList
 List storing and managing all parameter of the module.
 
bool m_hasReturnValue
 True, if the return value is set.
 
int m_returnValue
 The return value.
 
std::vector< ModuleConditionm_conditions
 Module condition, only non-null if set.
 

Detailed Description

Calculate track isolation variables on the input ParticleList.

Definition at line 72 of file TrackIsoCalculatorModule.h.

Member Enumeration Documentation

◆ EModulePropFlags

enum EModulePropFlags
inherited

Each module can be tagged with property flags, which indicate certain features of the module.

Enumerator
c_Input 

This module is an input module (reads data).

c_Output 

This module is an output module (writes data).

c_ParallelProcessingCertified 

This module can be run in parallel processing mode safely (All I/O must be done through the data store, in particular, the module must not write any files.)

c_HistogramManager 

This module is used to manage histograms accumulated by other modules.

c_InternalSerializer 

This module is an internal serializer/deserializer for parallel processing.

c_TerminateInAllProcesses 

When using parallel processing, call this module's terminate() function in all processes().

This will also ensure that there is exactly one process (single-core if no parallel modules found) or at least one input, one main and one output process.

c_DontCollectStatistics 

No statistics is collected for this module.

Definition at line 77 of file Module.h.

Member Function Documentation

◆ beginRun()

◆ clone()

std::shared_ptr< PathElement > clone ( ) const
overridevirtualinherited

Create an independent copy of this module.

Note that parameters are shared, so changing them on a cloned module will also affect the original module.

Implements PathElement.

Definition at line 179 of file Module.cc.

◆ def_endRun()

virtual void def_endRun ( )
inlineprotectedvirtualinherited

This method can receive that the current run ends as a call from the Python side.

For regular C++-Modules that forwards the call to the regular endRun() method.

Reimplemented in PyModule.

Definition at line 439 of file Module.h.

◆ def_initialize()

virtual void def_initialize ( )
inlineprotectedvirtualinherited

Wrappers to make the methods without "def_" prefix callable from Python.

Overridden in PyModule. Wrapper method for the virtual function initialize() that has the implementation to be used in a call from Python.

Reimplemented in PyModule.

Definition at line 420 of file Module.h.

◆ endRun()

virtual void endRun ( )
inlinevirtualinherited

This method is called if the current run ends.

Use this method to store information, which should be aggregated over one run.

This method can be implemented by subclasses.

Reimplemented in PyModule, AWESOMEBasicModule, TxModule, RxModule, SeqRootOutputModule, SeqRootInputModule, StatisticsSummaryModule, HistoManagerModule, RandomBarrierModule, EventInfoPrinterModule, SwitchDataStoreModule, SubEventModule, HistoModule, BelleMCOutputModule, B2BIIMdstInputModule, B2BIIConvertMdstModule, B2BIIMCParticlesMonitorModule, LowEnergyPi0VetoExpertModule, LowEnergyPi0IdentificationExpertModule, and CurlTaggerModule.

Definition at line 166 of file Module.h.

◆ evalCondition()

bool evalCondition ( ) const
inherited

If at least one condition was set, it is evaluated and true returned if at least one condition returns true.

If no condition or result value was defined, the method returns false. Otherwise, the condition is evaluated and true returned, if at least one condition returns true. To speed up the evaluation, the condition strings were already parsed in the method if_value().

Returns
True if at least one condition and return value exists and at least one condition expression was evaluated to true.

Definition at line 96 of file Module.cc.

◆ event()

void event ( )
overridevirtual

Called once for each event.

This is most likely where your module will actually do anything.

Reimplemented from Module.

Definition at line 128 of file TrackIsoCalculatorModule.cc.

129 {
130  B2DEBUG(11, "Start processing EVENT: " << m_event_metadata->getEvent());
131 
132  const auto nMotherParticles = m_pListTarget->getListSize();
133 
134  // Fill transient container of all selected charged particles in the decay
135  // for which the distance to nearest neighbour is to be calculated.
136  // If the input ParticleList is that of standard charged particles, just copy
137  // the whole list over, otherwise loop over the selected charged daughters.
138  std::unordered_map<unsigned int, const Particle*> targetParticles;
139  targetParticles.reserve(nMotherParticles);
140 
141  // Loop over the layer of the input detector.
142  for (const auto& iLayer : m_detToLayers[m_detName]) {
143 
144  auto iDetLayer = m_detName + std::to_string(iLayer);
145 
146  for (unsigned int iPart(0); iPart < nMotherParticles; ++iPart) {
147 
148  auto iParticle = m_pListTarget->getParticle(iPart);
149 
150  if (m_nSelectedDaughters) {
151  for (auto* iDaughter : m_decaydescriptor.getSelectionParticles(iParticle)) {
152  // Check if the distance for this target particle has been set already,
153  // e.g. by a previous instance of this module.
154  if (iDaughter->hasExtraInfo(m_detLayerToDistVariable[iDetLayer])) {
155  continue;
156  }
157  targetParticles.insert({iDaughter->getMdstArrayIndex(), iDaughter});
158  }
159  } else {
160  if (iParticle->hasExtraInfo(m_detLayerToDistVariable[iDetLayer])) {
161  continue;
162  }
163  targetParticles.insert({iParticle->getMdstArrayIndex(), iParticle});
164  }
165 
166  }
167  }
168 
169  const auto nParticlesTarget = targetParticles.size();
170  const auto nParticlesReference = m_pListReference->getListSize();
171 
172  // Loop over the layer of the input detector.
173  for (const auto& iLayer : m_detToLayers[m_detName]) {
174 
175  auto iDetLayer = m_detName + std::to_string(iLayer);
176 
177  B2DEBUG(11, "\n"
178  << "Detector surface: " << iDetLayer << "\n"
179  << "nMotherParticles: " << nMotherParticles << "\n"
180  << "nParticlesTarget: " << nParticlesTarget << "\n"
181  << "nParticlesReference: " << nParticlesReference);
182 
183  double dummyDist(-1.0);
184 
185  // Store the pair-wise distances in a map,
186  // where the keys are pairs of mdst indexes.
187  std::map<std::pair<unsigned int, unsigned int>, double> particleMdstIdxPairsToDist;
188 
189  // Loop over input particle list
190  for (const auto& targetParticle : targetParticles) {
191 
192  auto iMdstIdx = targetParticle.first;
193  auto iParticle = targetParticle.second;
194 
195  for (unsigned int jPart(0); jPart < nParticlesReference; ++jPart) {
196 
197  auto jParticle = m_pListReference->getParticle(jPart);
198  auto jMdstIdx = jParticle->getMdstArrayIndex();
199 
200  auto partMdstIdxPair = std::make_pair(iMdstIdx, jMdstIdx);
201 
202  // Set dummy distance if same particle.
203  if (iMdstIdx == jMdstIdx) {
204  particleMdstIdxPairsToDist[partMdstIdxPair] = dummyDist;
205  continue;
206  }
207  // If:
208  //
209  // - the mass hypothesis of the best fit is used, OR
210  // - the mass hypothesis of the 'default' fit of the two particles is the same,
211  //
212  // avoid re-doing the calculation if a pair with the flipped mdst indexes in the map already exists.
213  if (m_useHighestProbMassForExt || (iParticle->getPDGCodeUsedForFit() == jParticle->getPDGCodeUsedForFit())) {
214  if (particleMdstIdxPairsToDist.count({jMdstIdx, iMdstIdx})) {
215  particleMdstIdxPairsToDist[partMdstIdxPair] = particleMdstIdxPairsToDist[ {jMdstIdx, iMdstIdx}];
216  continue;
217  }
218  }
219  // Calculate the pair-wise distance.
220  particleMdstIdxPairsToDist[partMdstIdxPair] = this->getDistAtDetSurface(iParticle, jParticle, iDetLayer);
221  }
222 
223  }
224 
225  // For each particle in the input list, find the minimum among all distances to the reference particles.
226  for (const auto& targetParticle : targetParticles) {
227 
228  auto iMdstIdx = targetParticle.first;
229  auto iParticle = targetParticle.second;
230 
231  // Save the distances and the mdst indexes of the reference particles.
232  std::vector<std::pair<double, unsigned int>> iDistancesAndRefMdstIdxs;
233  for (const auto& [mdstIdxs, dist] : particleMdstIdxPairsToDist) {
234  if (mdstIdxs.first == iMdstIdx) {
235  if (!std::isnan(dist) && dist >= 0) {
236  iDistancesAndRefMdstIdxs.push_back(std::make_pair(dist, mdstIdxs.second));
237  }
238  }
239  }
240 
241  if (!iDistancesAndRefMdstIdxs.size()) {
242  B2DEBUG(12, "The container of distances is empty. Perhaps the target and reference lists contain the same exact particles?");
243  continue;
244  }
245 
246  const auto minDist = *std::min_element(std::begin(iDistancesAndRefMdstIdxs), std::end(iDistancesAndRefMdstIdxs),
247  [](const auto & l, const auto & r) {return l.first < r.first;});
248 
249  auto jParticle = m_pListReference->getParticleWithMdstIdx(minDist.second);
250 
251  B2DEBUG(11, "\n"
252  << "Particle w/ mdstIndex[" << iMdstIdx << "] (PDG = "
253  << iParticle->getPDGCode() << "). Closest charged particle w/ mdstIndex["
254  << minDist.second
255  << "] (PDG = " << jParticle->getPDGCode()
256  << ") at " << iDetLayer
257  << " surface is found at D = " << minDist.first
258  << " [cm]\n"
259  << "Storing extraInfo variables:\n"
260  << m_detLayerToDistVariable[iDetLayer]
261  << "\n"
262  << m_detLayerToRefPartIdxVariable[iDetLayer]);
263 
264  if (!iParticle->hasExtraInfo(m_detLayerToDistVariable[iDetLayer])) {
265  m_particles[iParticle->getArrayIndex()]->addExtraInfo(m_detLayerToDistVariable[iDetLayer], minDist.first);
266  }
267  m_particles[iParticle->getArrayIndex()]->writeExtraInfo(m_detLayerToRefPartIdxVariable[iDetLayer], minDist.second);
268 
269  } // end loop over input particle list.
270 
271  } // end loop over detector layers.
272 
273  // Store the isolation score per target particle for the given detector.
274  for (const auto& targetParticle : targetParticles) {
275 
276  auto iMdstIdx = targetParticle.first;
277  auto iParticle = targetParticle.second;
278 
279  auto score = this->getIsoScore(iParticle);
280  m_particles[iParticle->getArrayIndex()]->writeExtraInfo(m_isoScoreVariable, score);
281 
282  B2DEBUG(11, "\n"
283  << "Particle w/ mdstIndex[" << iMdstIdx << "] (PDG = " << iParticle->getPDGCode() << ").\n"
284  << "Isolation score in the " << m_detName << ": s = " << score
285  << "\n"
286  << "Storing extraInfo variable:\n"
287  << m_isoScoreVariable);
288 
289  }
290 
291  B2DEBUG(11, "Finished processing EVENT: " << m_event_metadata->getEvent());
292 
293 }
std::vector< const Particle * > getSelectionParticles(const Particle *particle)
Get a vector of pointers with selected daughters in the decay tree.
StoreObjPtr< EventMetaData > m_event_metadata
The event information.
std::unordered_map< std::string, std::vector< int > > m_detToLayers
Map that associates to each detector its list of valid layers.
double getDistAtDetSurface(const Particle *iParticle, const Particle *jParticle, const std::string &detLayerName) const
Calculate the distance between the points where the two input extrapolated track helices cross the gi...
std::unordered_map< std::string, std::string > m_detLayerToDistVariable
Map that associates to each detector layer the name of the variable representing the distance to the ...
StoreObjPtr< ParticleList > m_pListTarget
The input ParticleList object for which distances are to be calculated.
std::string m_isoScoreVariable
The name of the variable representing the track isolation score in this detector.
StoreArray< Particle > m_particles
StoreArray of Particles.
std::string m_detName
The name of the detector at whose inner (cylindrical) surface we extrapolate each track's polar and a...
std::unordered_map< std::string, std::string > m_detLayerToRefPartIdxVariable
Map that associates to each detector layer the name of the variable representing the mdst array index...
double getIsoScore(const Particle *iParticle) const
Define a semi-continuous variable to quantify the isolation of a standard charged particle in the giv...
DecayDescriptor m_decaydescriptor
< Decay descriptor of decays to look for.
unsigned short m_nSelectedDaughters
The number of selected daughters in the decay string.
StoreObjPtr< ParticleList > m_pListReference
The input ParticleList object of reference tracks.
bool m_useHighestProbMassForExt
If this option is set, the helix extrapolation for the target and reference particles will use the tr...

◆ getAfterConditionPath()

Module::EAfterConditionPath getAfterConditionPath ( ) const
inherited

What to do after the conditional path is finished.

(defaults to c_End if no condition is set)

Definition at line 133 of file Module.cc.

◆ getConditionPath()

std::shared_ptr< Path > getConditionPath ( ) const
inherited

Returns the path of the last true condition (if there is at least one, else reaturn a null pointer).


Definition at line 113 of file Module.cc.

◆ getDistThreshold()

double getDistThreshold ( Const::EDetector  det,
int  layer 
) const
inlineprivate

Get the threshold value per detctor layer for the distance to closest ext.

helix that is used to define locally isolated particles at that layer.

Parameters
detthe input PID detector.
layerthe input detector layer.

Definition at line 280 of file TrackIsoCalculatorModule.h.

281  {
282  auto detAndLayer = std::make_pair(Const::parseDetectors(det), layer);
283  return m_distThreshPerDetLayer.at(detAndLayer);
284  };
static std::string parseDetectors(EDetector det)
Converts Const::EDetector object to string.
Definition: UnitConst.cc:162
std::map< std::pair< std::string, int >, double > m_distThreshPerDetLayer
Threshold values for the distance (in [cm]) to closest ext.

◆ getFileNames()

virtual std::vector<std::string> getFileNames ( bool  outputFiles)
inlinevirtualinherited

Return a list of output filenames for this modules.

This will be called when basf2 is run with "--dry-run" if the module has set either the c_Input or c_Output properties.

If the parameter outputFiles is false (for modules with c_Input) the list of input filenames should be returned (if any). If outputFiles is true (for modules with c_Output) the list of output files should be returned (if any).

If a module has sat both properties this member is called twice, once for each property.

The module should return the actual list of requested input or produced output filenames (including handling of input/output overrides) so that the grid system can handle input/output files correctly.

This function should return the same value when called multiple times. This is especially important when taking the input/output overrides from Environment as they get consumed when obtained so the finalized list of output files should be stored for subsequent calls.

Reimplemented in RootOutputModule, and RootInputModule.

Definition at line 134 of file Module.h.

◆ getIsoScore()

double getIsoScore ( const Particle iParticle) const
private

Define a semi-continuous variable to quantify the isolation of a standard charged particle in the given detector $d$ with $N_{d}$ layers in total.

The definition of the score is based on the counting of layers $n_{d}$ where a close-enough particle is found, and also (if m_excludePIDDetWeights = true) on the weight $w_{d}$ that each sub-detector has on the PID of the given particle hypothesis:

\begin{equation} s_{d} = 1 - \left(-w_{d} \cdot \frac{n_{d}}{N_{d}}\right). \end{equation}

The distance to closest track helix extrapolation defined in double getDistAtDetSurface() is used. Note that if the PID detector weighting is switched off, $w_{d} = -1$.

The per-detector score is normalised in $s_{d}\in [0, 1]$: values closer to 1 indicate well-isolated particles.

Definition at line 296 of file TrackIsoCalculatorModule.cc.

◆ getName()

const std::string& getName ( ) const
inlineinherited

Returns the name of the module.

This can be changed via e.g. set_name() in the steering file to give more useful names if there is more than one module of the same type.

For identifying the type of a module, using getType() (or type() in Python) is recommended.

Definition at line 187 of file Module.h.

◆ getParamInfoListPython()

std::shared_ptr< boost::python::list > getParamInfoListPython ( ) const
inherited

Returns a python list of all parameters.

Each item in the list consists of the name of the parameter, a string describing its type, a python list of all default values and the description of the parameter.

Returns
A python list containing the parameters of this parameter list.

Definition at line 279 of file Module.cc.

◆ getReturnValue()

int getReturnValue ( ) const
inlineinherited

Return the return value set by this module.

This value is only meaningful if hasReturnValue() is true

Definition at line 381 of file Module.h.

◆ getType()

const std::string & getType ( ) const
inherited

Returns the type of the module (i.e.

class name minus 'Module')

Definition at line 41 of file Module.cc.

◆ hasProperties()

bool hasProperties ( unsigned int  propertyFlags) const
inherited

Returns true if all specified property flags are available in this module.

Parameters
propertyFlagsOred EModulePropFlags which should be compared with the module flags.

Definition at line 160 of file Module.cc.

◆ if_false()

void if_false ( const std::shared_ptr< Path > &  path,
EAfterConditionPath  afterConditionPath = EAfterConditionPath::c_End 
)
inherited

A simplified version to add a condition to the module.

Please note that successive calls of this function will add more than one condition to the module. If more than one condition results in true, only the last of them will be used.

Please be careful: Avoid creating cyclic paths, e.g. by linking a condition to a path which is processed before the path where this module is located in.

It is equivalent to the if_value() method, using the expression "<1". This method is meant to be used together with the setReturnValue(bool value) method.

Parameters
pathShared pointer to the Path which will be executed if the return value is false.
afterConditionPathWhat to do after executing 'path'.

Definition at line 85 of file Module.cc.

◆ if_true()

void if_true ( const std::shared_ptr< Path > &  path,
EAfterConditionPath  afterConditionPath = EAfterConditionPath::c_End 
)
inherited

A simplified version to set the condition of the module.

Please note that successive calls of this function will add more than one condition to the module. If more than one condition results in true, only the last of them will be used.

Please be careful: Avoid creating cyclic paths, e.g. by linking a condition to a path which is processed before the path where this module is located in.

It is equivalent to the if_value() method, using the expression ">=1". This method is meant to be used together with the setReturnValue(bool value) method.

Parameters
pathShared pointer to the Path which will be executed if the return value is true.
afterConditionPathWhat to do after executing 'path'.

Definition at line 90 of file Module.cc.

◆ if_value()

void if_value ( const std::string &  expression,
const std::shared_ptr< Path > &  path,
EAfterConditionPath  afterConditionPath = EAfterConditionPath::c_End 
)
inherited

Add a condition to the module.

Please note that successive calls of this function will add more than one condition to the module. If more than one condition results in true, only the last of them will be used.

See https://confluence.desy.de/display/BI/Software+ModCondTut or ModuleCondition for a description of the syntax.

Please be careful: Avoid creating cyclic paths, e.g. by linking a condition to a path which is processed before the path where this module is located in.

Parameters
expressionThe expression of the condition.
pathShared pointer to the Path which will be executed if the condition is evaluated to true.
afterConditionPathWhat to do after executing 'path'.

Definition at line 79 of file Module.cc.

◆ initialize()

void initialize ( )
overridevirtual

Use this to initialize resources or memory your module needs.

Also register any outputs of your module (StoreArrays, StoreObjPtrs, relations) here, see the respective class documentation for details.

Reimplemented from Module.

Definition at line 55 of file TrackIsoCalculatorModule.cc.

◆ setDescription()

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

Sets the description of the module.

Parameters
descriptionA description of the module.

Definition at line 214 of file Module.cc.

◆ setLogInfo()

void setLogInfo ( int  logLevel,
unsigned int  logInfo 
)
inherited

Configure the printed log information for the given level.

Parameters
logLevelThe log level (one of LogConfig::ELogLevel)
logInfoWhat kind of info should be printed? ORed combination of LogConfig::ELogInfo flags.

Definition at line 73 of file Module.cc.

◆ setName()

void setName ( const std::string &  name)
inlineinherited

Set the name of the module.

Note
The module name is set when using the REG_MODULE macro, but the module can be renamed before calling process() using the set_name() function in your steering file.
Parameters
nameThe name of the module

Definition at line 214 of file Module.h.

◆ setParamPython()

void setParamPython ( const std::string &  name,
const boost::python::object &  pyObj 
)
privateinherited

Implements a method for setting boost::python objects.

The method supports the following types: list, dict, int, double, string, bool The conversion of the python object to the C++ type and the final storage of the parameter value is done in the ModuleParam class.

Parameters
nameThe unique name of the parameter.
pyObjThe object which should be converted and stored as the parameter value.

Definition at line 234 of file Module.cc.

◆ setParamPythonDict()

void setParamPythonDict ( const boost::python::dict &  dictionary)
privateinherited

Implements a method for reading the parameter values from a boost::python dictionary.

The key of the dictionary has to be the name of the parameter and the value has to be of one of the supported parameter types.

Parameters
dictionaryThe python dictionary from which the parameter values are read.

Definition at line 249 of file Module.cc.

◆ setPropertyFlags()

void setPropertyFlags ( unsigned int  propertyFlags)
inherited

Sets the flags for the module properties.

Parameters
propertyFlagsbitwise OR of EModulePropFlags

Definition at line 208 of file Module.cc.

◆ setReturnValue() [1/2]

void setReturnValue ( bool  value)
protectedinherited

Sets the return value for this module as bool.

The bool value is saved as an integer with the convention 1 meaning true and 0 meaning false. The value can be used in the steering file to divide the analysis chain into several paths.

Parameters
valueThe value of the return value.

Definition at line 227 of file Module.cc.

◆ setReturnValue() [2/2]

void setReturnValue ( int  value)
protectedinherited

Sets the return value for this module as integer.

The value can be used in the steering file to divide the analysis chain into several paths.

Parameters
valueThe value of the return value.

Definition at line 220 of file Module.cc.

◆ setType()

void setType ( const std::string &  type)
protectedinherited

Set the module type.

Only for use by internal modules (which don't use the normal REG_MODULE mechanism).

Definition at line 48 of file Module.cc.

◆ terminate()

Member Data Documentation

◆ m_detLayerToDistVariable

std::unordered_map<std::string, std::string> m_detLayerToDistVariable
private

Map that associates to each detector layer the name of the variable representing the distance to the closest particle in the reference list, based on the track helix extrapolation.

Each variable is added as particle extraInfo.

Definition at line 140 of file TrackIsoCalculatorModule.h.

◆ m_detLayerToRefPartIdxVariable

std::unordered_map<std::string, std::string> m_detLayerToRefPartIdxVariable
private

Map that associates to each detector layer the name of the variable representing the mdst array index of the closest particle in the reference list.

Each variable is added as particle extraInfo.

Definition at line 147 of file TrackIsoCalculatorModule.h.

◆ m_detLayerToSurfBoundaries

std::unordered_map<std::string, DetSurfCylBoundaries> m_detLayerToSurfBoundaries
private
Initial value:
= {
{"CDC0", DetSurfCylBoundaries(16.8, 150.0, -75.0, 0.0, 0.29, 2.61, 3.14)},
{"CDC1", DetSurfCylBoundaries(25.7, 150.0, -75.0, 0.0, 0.29, 2.61, 3.14)},
{"CDC2", DetSurfCylBoundaries(36.52, 150.0, -75.0, 0.0, 0.29, 2.61, 3.14)},
{"CDC3", DetSurfCylBoundaries(47.69, 150.0, -75.0, 0.0, 0.29, 2.61, 3.14)},
{"CDC4", DetSurfCylBoundaries(58.41, 150.0, -75.0, 0.0, 0.29, 2.61, 3.14)},
{"CDC5", DetSurfCylBoundaries(69.53, 150.0, -75.0, 0.0, 0.29, 2.61, 3.14)},
{"CDC6", DetSurfCylBoundaries(80.25, 150.0, -75.0, 0.0, 0.29, 2.61, 3.14)},
{"CDC7", DetSurfCylBoundaries(91.37, 150.0, -75.0, 0.0, 0.29, 2.61, 3.14)},
{"CDC8", DetSurfCylBoundaries(102.09, 150.0, -75.0, 0.0, 0.29, 2.61, 3.14)},
{"TOP0", DetSurfCylBoundaries(117.8, 193.0, -94.0, 0.24, 0.52, 2.23, 3.14)},
{"ARICH0", DetSurfCylBoundaries(117.8, 193.0, -94.0, 0.24, 0.52, 2.23, 3.14)},
{"ECL0", DetSurfCylBoundaries(125.0, 196.0, -102.0, 0.21, 0.56, 2.24, 2.70)},
{"ECL1", DetSurfCylBoundaries(140.0, 211.0, -117.0, 0.21, 0.56, 2.24, 2.70)},
{"KLM0", DetSurfCylBoundaries(202.0, 283.9, -189.9, 0.40, 0.82, 2.13, 2.60)},
}

Map that associates to each detector layer its valid cylindrical surface's boundaries.

Values are taken from the B2 TDR.

Definition at line 171 of file TrackIsoCalculatorModule.h.

◆ m_detToLayers

std::unordered_map<std::string, std::vector<int> > m_detToLayers
private
Initial value:
= {
{"CDC", {0, 1, 2, 3, 4, 5, 6, 7, 8}},
{"TOP", {0}},
{"ARICH", {0}},
{"ECL", {0, 1}},
{"KLM", {0}}
}

Map that associates to each detector its list of valid layers.

Definition at line 159 of file TrackIsoCalculatorModule.h.

◆ m_distThreshPerDetLayer

std::map<std::pair<std::string, int>, double> m_distThreshPerDetLayer
private
Initial value:
= {
{ {Const::parseDetectors(Const::CDC), 0}, 5.0 },
{ {Const::parseDetectors(Const::CDC), 1}, 5.0 },
{ {Const::parseDetectors(Const::CDC), 2}, 5.0 },
{ {Const::parseDetectors(Const::CDC), 3}, 5.0 },
{ {Const::parseDetectors(Const::CDC), 4}, 5.0 },
{ {Const::parseDetectors(Const::CDC), 5}, 5.0 },
{ {Const::parseDetectors(Const::CDC), 6}, 5.0 },
{ {Const::parseDetectors(Const::CDC), 7}, 5.0 },
{ {Const::parseDetectors(Const::CDC), 8}, 5.0 },
{ {Const::parseDetectors(Const::TOP), 0}, 22.0 },
{ {Const::parseDetectors(Const::ARICH), 0}, 10.0 },
{ {Const::parseDetectors(Const::ECL), 0}, 36.0 },
{ {Const::parseDetectors(Const::ECL), 1}, 36.0 },
{ {Const::parseDetectors(Const::KLM), 0}, 20.0 }
}

Threshold values for the distance (in [cm]) to closest ext.

helix to define isolated particles. One for each detector layer.

Definition at line 192 of file TrackIsoCalculatorModule.h.

◆ m_event_metadata

StoreObjPtr<EventMetaData> m_event_metadata
private

The event information.

Used for debugging purposes.

Definition at line 212 of file TrackIsoCalculatorModule.h.

◆ m_isoScoreVariable

std::string m_isoScoreVariable
private

The name of the variable representing the track isolation score in this detector.

Added as particle extraInfo.

Definition at line 154 of file TrackIsoCalculatorModule.h.


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