Belle II Software light-2509-fornax
ParticleLoaderModule Class Reference

Loads MDST dataobjects as Particle objects to the StoreArray<Particle> and collects them in specified ParticleList. More...

#include <ParticleLoaderModule.h>

Inheritance diagram for ParticleLoaderModule:
Collaboration diagram for ParticleLoaderModule:

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

 ParticleLoaderModule ()
 Constructor.
 
virtual void initialize () override
 Initialize the Module.
 
virtual void event () override
 Event processor.
 
virtual void terminate () override
 Terminate the Module.
 
virtual std::vector< std::string > getFileNames (bool outputFiles)
 Return a list of output filenames for this modules.
 
virtual void beginRun ()
 Called when entering a new run.
 
virtual void endRun ()
 This method is called if the current run ends.
 
const std::string & getName () const
 Returns the name of the module.
 
const std::string & getType () const
 Returns the type of the module (i.e.
 
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.
 
void setPropertyFlags (unsigned int propertyFlags)
 Sets the flags for the module properties.
 
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.
 
void if_value (const std::string &expression, const std::shared_ptr< Path > &path, EAfterConditionPath afterConditionPath=EAfterConditionPath::c_End)
 Add a condition to the module.
 
void if_false (const std::shared_ptr< Path > &path, EAfterConditionPath afterConditionPath=EAfterConditionPath::c_End)
 A simplified version to add a condition to the module.
 
void if_true (const std::shared_ptr< Path > &path, EAfterConditionPath afterConditionPath=EAfterConditionPath::c_End)
 A simplified version to set the condition of the module.
 
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.
 
std::shared_ptr< PathgetConditionPath () const
 Returns the path of the last true condition (if there is at least one, else reaturn a null pointer).
 
Module::EAfterConditionPath getAfterConditionPath () const
 What to do after the conditional path is finished.
 
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.
 
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.
 
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.
 
std::shared_ptr< PathElementclone () const override
 Create an independent copy of this module.
 
std::shared_ptr< boost::python::list > getParamInfoListPython () const
 Returns a python list of all parameters.
 

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.
 
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.
 
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.
 
void setType (const std::string &type)
 Set the module type.
 
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.
 
template<typename T>
void addParam (const std::string &name, T &paramVariable, const std::string &description)
 Adds a new enforced parameter to the module.
 
void setReturnValue (int value)
 Sets the return value for this module as integer.
 
void setReturnValue (bool value)
 Sets the return value for this module as bool.
 
void setParamList (const ModuleParamList &params)
 Replace existing parameter list.
 

Private Types

enum  PListIndex {
  c_PListPDGCode ,
  c_PListName ,
  c_AntiPListName ,
  c_IsPListSelfConjugated
}
 Enum for describing each element in the above tuple. More...
 
typedef std::tuple< int, std::string, std::string, bool > PList
 tuple for collecting everything we know about the ParticlList to be created.
 

Private Member Functions

void mcParticlesToParticles ()
 Loads specified MCParticles as Particle to StoreArray<Particle>
 
void mdstToParticle ()
 Loads specified MSDT object as Particle to StoreArray<Particle>
 
void tracksToParticles ()
 Loads Track object as Particle to StoreArray<Particle> and adds it to the ParticleList.
 
void eclAndKLMClustersToParticles ()
 Loads ECLCluster and KLMCluster object as Particle to StoreArray<Particle> and adds it to the ParticleList.
 
bool isValidECLCluster (const ECLCluster *cluster, const int pdgCode, bool onlyNeutral) const
 Checks if the given ECLCluster is valid for the pdgCode.
 
void assignMCParticleFromECLCluster (Particle *newPart, const ECLCluster *cluster) const
 Assigns the MCParticle relation to the newPart.
 
void chargedClustersToParticles ()
 Loads ECLCluster and KLMCluster objects that are being matched with Track as Particle to StoreArray<Particle> and adds it to the ParticleList.
 
void v0sToParticles ()
 Loads V0 object as Particle of specified type to StoreArray<Particle> and adds it to the ParticleList.
 
void kinksToParticles ()
 Loads Kink object as Particle of specified type to StoreArray<Particle> and adds it to the ParticleList.
 
void roeToParticles ()
 Loads ROE object as Particle of specified type to StoreArray<Particle> and adds it to the ParticleList.
 
void addROEToParticleList (RestOfEvent *roe, int mdstIndex, int pdgCode=0, bool isSelfConjugatedParticle=true)
 Helper method to load ROE object as Particle.
 
void dummyToParticles ()
 Loads dummy object as Particle of specified type to StoreArray<Particle> and adds it to the ParticleList.
 
bool isValidPDGCode (const int pdgCode)
 returns true if the PDG code determined from the decayString is valid
 
void appendDaughtersRecursive (Particle *mother)
 recursively append bottom of a particle's decay chain (daughters of mother, granddaughters of daughter and so on).
 
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.
 
void setParamPythonDict (const boost::python::dict &dictionary)
 Implements a method for reading the parameter values from a boost::python dictionary.
 

Private Attributes

StoreArray< Particlem_particles
 StoreArray of Particles.
 
StoreArray< MCParticlem_mcparticles
 StoreArray of MCParticles.
 
StoreArray< ECLClusterm_eclclusters
 StoreArray of ECLCluster.
 
StoreArray< KLMClusterm_klmclusters
 StoreArray of KLMCluster.
 
StoreArray< PIDLikelihoodm_pidlikelihoods
 StoreArray of PIDLikelihoods.
 
StoreArray< Trackm_tracks
 StoreArray of Tracks.
 
StoreArray< TrackFitResultm_trackfitresults
 StoreArray of TrackFitResults.
 
StoreObjPtr< ParticleExtraInfoMapm_particleExtraInfoMap
 object pointer to extra info map
 
StoreArray< RestOfEventm_roes
 StoreArray of ROEs.
 
StoreArray< V0m_v0s
 StoreArray of V0s.
 
StoreArray< Kinkm_kinks
 StoreArray of Kinks.
 
bool m_useMCParticles
 Load MCParticle as Particle instead of the corresponding MDST dataobject.
 
bool m_useROEs
 Switch to load ROE as Particle.
 
bool m_useDummy
 Switch to load dummy as Particle.
 
bool m_loadChargedCluster
 Switch to load charged-cluster.
 
bool m_useOnlyMostEnergeticECLCluster
 If true, only the most energetic ECLCluster is used.
 
DecayDescriptor m_decaydescriptor
 Decay descriptor for parsing the user specified DecayString.
 
int m_properties
 Particle property to be assigned only on V0s.
 
std::vector< std::string > m_decayStrings
 Input decay strings specifying the particles being created/loaded.
 
std::vector< PListm_MCParticles2Plists
 Collection of PLists that will collect Particles created from MCParticles.
 
std::vector< PListm_Tracks2Plists
 Collection of PLists that will collect Particles created from Tracks.
 
std::vector< PListm_V02Plists
 Collection of PLists that will collect Particles created from V0.
 
std::vector< PListm_Kink2Plists
 Collection of PLists that will collect Particles created from Kink.
 
std::vector< PListm_ROE2Plists
 Collection of PLists that will collect Particles created from V0.
 
std::vector< PListm_ECLKLMClusters2Plists
 Collection of PLists that will collect Particles created from ECLClusters and KLMClusters.
 
std::vector< PListm_Dummies2Plists
 Collection of PLists that will collect Particles created from Dummies.
 
std::vector< PListm_ChargedCluster2Plists
 Collection of PLists that will collect Particles created from charged-cluster.
 
bool m_writeOut
 toggle particle list btw.
 
bool m_skipInitial
 toggle skip of initial MC particles
 
bool m_skipNonPrimary
 toggle skip of secondary MC particle
 
bool m_addDaughters
 toggle addition of the bottom part of the particle's decay chain
 
bool m_skipNonPrimaryDaughters
 toggle skip of secondary MC daughters
 
std::string m_roeMaskName
 ROE mask name to load.
 
std::string m_sourceParticleListName
 Particle list name from which we need to get related ROEs.
 
bool m_useMissing
 Use missing momentum to build a particle.
 
int m_trackHypothesis
 pdg code for track hypothesis that should be used to create the particle
 
int m_dummyMDSTIndex
 mdst index for dummy particles
 
double m_dummyCovMatrix
 diag value of cov matrix for dummy particles
 
bool m_dummyTreatAsInvisible
 should treeFitter treat the particle as invisible?
 
bool m_enforceFitHypothesis
 If true, a Particle is only created if a track fit with the particle hypothesis passed to the ParticleLoader is available.
 
std::vector< int > m_chargeZeroTrackCounts
 internally used to count number of tracks with charge zero
 
std::vector< int > m_sameChargeDaughtersV0Counts
 internally used to count the number of V0s with same charge daughters
 
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

Loads MDST dataobjects as Particle objects to the StoreArray<Particle> and collects them in specified ParticleList.

Charge conjugated particles are loaded as well and collected in the anti-ParticleList.

The type of the particle to be loaded is specified via the decayString module parameter. The type of the MDST dataobject that is used as an input is determined from the type of the particle.

The following types of the particles can be loaded:

o) charged final state particles (input MDST type = Tracks)

  • e+, mu+, pi+, K+, p, deuteron (and charge conjugated particles)

o) neutral final state particles

  • gamma (input MDST type = ECLCluster with 'n photons' ECLCluster::Hypothesis::c_nPhotons)
  • K_S0, Lambda0, converted photons (input MDST type = V0)
  • K_L0, n0 (input MDST type = KLMCluster, or ECLCluster with neutral hadron hypothesis)

o) decayed charged final state particles (input MDST type = Kinks)

  • mu+, pi+, K+ (and charge conjugated particles)

The following basf2 relations are set by the ParticleLoader:

o) in the case of charged final state particles

o) in the case of neutral final state particles

In the case the useMCParticles module parameter is set to true the module loads specific MCParticle(s) as Particle(s) instead of the MDST dataobjects. In this case any particle type can be specified via the decayString module parameter.

Definition at line 79 of file ParticleLoaderModule.h.

Member Typedef Documentation

◆ EAfterConditionPath

Forward the EAfterConditionPath definition from the ModuleCondition.

Definition at line 88 of file Module.h.

◆ PList

typedef std::tuple<int, std::string, std::string, bool> PList
private

tuple for collecting everything we know about the ParticlList to be created.

The elements are: PDGCode, name, anti-list name, and isListSelfConjugated

Definition at line 85 of file ParticleLoaderModule.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.

77 {
78 c_Input = 1,
79 c_Output = 2,
80 c_ParallelProcessingCertified = 4,
81 c_HistogramManager = 8,
82 c_InternalSerializer = 16,
83 c_TerminateInAllProcesses = 32,
84 c_DontCollectStatistics = 64,
85 };
@ c_Output
Output Process.
Definition ProcHelper.h:19
@ c_Input
Input Process.
Definition ProcHelper.h:17

◆ PListIndex

enum PListIndex
private

Enum for describing each element in the above tuple.

Definition at line 89 of file ParticleLoaderModule.h.

89 {
90 c_PListPDGCode, c_PListName, c_AntiPListName, c_IsPListSelfConjugated
91 };

Constructor & Destructor Documentation

◆ ParticleLoaderModule()

Constructor.

Definition at line 37 of file ParticleLoaderModule.cc.

37 : Module()
38
39{
40 setDescription("Loads MDST dataobjects as Particle objects to the StoreArray<Particle> and collects them in specified ParticleList.");
42
43 // Add parameters
44 addParam("decayStrings", m_decayStrings,
45 "List of decay strings (see :ref:`DecayString` for syntax) that specify all output ParticleLists to be created by the module.",
46 {});
47
48 addParam("useMCParticles", m_useMCParticles,
49 "Use MCParticles instead of reconstructed MDST dataobjects (tracks, ECL, KLM, clusters, V0s, ...)", false);
50
51 addParam("useROEs", m_useROEs,
52 "Use ROE instead of reconstructed MDST dataobjects (tracks, ECL, KLM, clusters, V0s, ...)", false);
53
54 addParam("roeMaskName", m_roeMaskName,
55 "ROE mask name to load", std::string(RestOfEvent::c_defaultMaskName));
56
57 addParam("sourceParticleListName", m_sourceParticleListName,
58 "Particle list name from which we need to get ROEs", std::string(""));
59
60 addParam("useMissing", m_useMissing,
61 "If true, the Particle List will be filled with missing momentum from the ROE and signal particle.", false);
62
63 addParam("writeOut", m_writeOut,
64 "If true, the output ParticleList will be saved by RootOutput. If false, it will be ignored when writing the file.", false);
65
66 addParam("skipInitial", m_skipInitial,
67 "If true, initial MCParticles will be skipped (default). If false, initial MCParticles will be included.", true);
68
69 addParam("skipNonPrimary", m_skipNonPrimary,
70 "If true, the secondary MC particle will be skipped, default is false",
71 false);
72
73 addParam("addDaughters", m_addDaughters,
74 "If true, the particles from the bottom part of the selected particle's decay chain will also be created in the datastore and mother-daughter relations are recursively set",
75 false);
76
77 addParam("skipNonPrimaryDaughters", m_skipNonPrimaryDaughters,
78 "If true, the secondary MC daughters will be skipped, default is false",
79 false);
80
81 addParam("trackHypothesis", m_trackHypothesis,
82 "Track hypothesis to use when loading the particle. By default, use the particle's own hypothesis.",
83 0);
84
85 addParam("enforceFitHypothesis", m_enforceFitHypothesis,
86 "If true, a Particle is only created if a track fit with the particle hypothesis passed to the ParticleLoader is available.",
88
89 addParam("useDummy", m_useDummy,
90 "Use Dummy instead of reconstructed MDST dataobjects (tracks, ECL, KLM, clusters, V0s, ...)", false);
91
92 addParam("dummyMDSTIndex", m_dummyMDSTIndex,
93 "mdst index to use for dummy particle", 0);
94
95 addParam("dummyCovMatrix", m_dummyCovMatrix,
96 "Diagonal value of covariance matrix to use for dummy particle", 10000.);
97
98 addParam("dummyTreatAsInvisible", m_dummyTreatAsInvisible,
99 "Should treeFitter treat the particle as invisible?", true);
100
101 addParam("loadChargedCluster", m_loadChargedCluster,
102 "Load neutral Particles from the clusters being matched with the track of the sourceParticleList",
103 false);
104 addParam("useOnlyMostEnergeticECLCluster", m_useOnlyMostEnergeticECLCluster,
105 "If true, the most energetic ECLCluster among ones matching with the Track is used. "
106 "If false, all matched ECLCluster are used. "
107 "This option is checked only when loadChargedCluster=True.",
108 true);
109}
void setDescription(const std::string &description)
Sets the description of the module.
Definition Module.cc:214
void setPropertyFlags(unsigned int propertyFlags)
Sets the flags for the module properties.
Definition Module.cc:208
Module()
Constructor.
Definition Module.cc:30
@ c_ParallelProcessingCertified
This module can be run in parallel processing mode safely (All I/O must be done through the data stor...
Definition Module.h:80
std::vector< std::string > m_decayStrings
Input decay strings specifying the particles being created/loaded.
bool m_dummyTreatAsInvisible
should treeFitter treat the particle as invisible?
int m_trackHypothesis
pdg code for track hypothesis that should be used to create the particle
std::string m_sourceParticleListName
Particle list name from which we need to get related ROEs.
bool m_skipInitial
toggle skip of initial MC particles
int m_dummyMDSTIndex
mdst index for dummy particles
std::string m_roeMaskName
ROE mask name to load.
double m_dummyCovMatrix
diag value of cov matrix for dummy particles
bool m_addDaughters
toggle addition of the bottom part of the particle's decay chain
bool m_skipNonPrimaryDaughters
toggle skip of secondary MC daughters
bool m_enforceFitHypothesis
If true, a Particle is only created if a track fit with the particle hypothesis passed to the Particl...
bool m_useDummy
Switch to load dummy as Particle.
bool m_writeOut
toggle particle list btw.
bool m_useOnlyMostEnergeticECLCluster
If true, only the most energetic ECLCluster is used.
bool m_skipNonPrimary
toggle skip of secondary MC particle
bool m_useROEs
Switch to load ROE as Particle.
bool m_loadChargedCluster
Switch to load charged-cluster.
bool m_useMissing
Use missing momentum to build a particle.
bool m_useMCParticles
Load MCParticle as Particle instead of the corresponding MDST dataobject.
static constexpr const char * c_defaultMaskName
Default mask name.
Definition RestOfEvent.h:58
void addParam(const std::string &name, T &paramVariable, const std::string &description, const T &defaultValue)
Adds a new parameter to the module.
Definition Module.h:559

Member Function Documentation

◆ addROEToParticleList()

void addROEToParticleList ( RestOfEvent * roe,
int mdstIndex,
int pdgCode = 0,
bool isSelfConjugatedParticle = true )
private

Helper method to load ROE object as Particle.

Definition at line 462 of file ParticleLoaderModule.cc.

463{
464
465 Particle* newPart = nullptr;
466 if (!m_useMissing) {
467 // Convert ROE to particle
468 newPart = roe->convertToParticle(m_roeMaskName, pdgCode, isSelfConjugatedParticle);
469 } else {
470 // Create a particle from missing momentum
471 auto* signalSideParticle = roe->getRelatedFrom<Particle>();
472 PCmsLabTransform T;
473 ROOT::Math::PxPyPzEVector boost4Vector = T.getBeamFourMomentum();
474
475 ROOT::Math::PxPyPzEVector signal4Vector = signalSideParticle->get4Vector();
476 ROOT::Math::PxPyPzEVector roe4Vector = roe->get4Vector(m_roeMaskName);
477 ROOT::Math::PxPyPzEVector missing4Vector = boost4Vector - signal4Vector - roe4Vector;
478 auto isFlavored = (isSelfConjugatedParticle) ? Particle::EFlavorType::c_Unflavored : Particle::EFlavorType::c_Flavored;
479 newPart = m_particles.appendNew(missing4Vector, pdgCode, isFlavored, Particle::EParticleSourceObject::c_Undefined, mdstIndex);
480 }
481
482 roe->addRelationTo(newPart);
483
484 for (auto roe2Plist : m_ROE2Plists) {
485 string listName = get<c_PListName>(roe2Plist);
486 StoreObjPtr<ParticleList> plist(listName);
487 plist->addParticle(newPart);
488 }
489}
StoreArray< Particle > m_particles
StoreArray of Particles.
std::vector< PList > m_ROE2Plists
Collection of PLists that will collect Particles created from V0.
@ c_Unflavored
Is its own antiparticle or we don't know whether it is a particle/antiparticle.
Definition Particle.h:97
void addRelationTo(const RelationsInterface< BASE > *object, float weight=1.0, const std::string &namedRelation="") const
Add a relation from this object to another object (with caching).
FROM * getRelatedFrom(const std::string &name="", const std::string &namedRelation="") const
Get the object from which this object has a relation.
ROOT::Math::PxPyPzEVector get4Vector(const std::string &maskName=c_defaultMaskName) const
Get 4-momentum vector all (no mask) or a subset (use mask) of all Tracks and ECLClusters in ROE.
Particle * convertToParticle(const std::string &maskName=c_defaultMaskName, int pdgCode=0, bool isSelfConjugated=true)
Converts ROE to Particle and adds it to StoreArray.

◆ appendDaughtersRecursive()

void appendDaughtersRecursive ( Particle * mother)
private

recursively append bottom of a particle's decay chain (daughters of mother, granddaughters of daughter and so on).

Definition at line 1108 of file ParticleLoaderModule.cc.

1109{
1110 auto* mcmother = mother->getRelated<MCParticle>();
1111
1112 if (!mcmother)
1113 return;
1114
1115 vector<MCParticle*> mcdaughters = mcmother->getDaughters();
1116
1117 for (auto& mcdaughter : mcdaughters) {
1118 if (!mcdaughter->hasStatus(MCParticle::c_PrimaryParticle) and m_skipNonPrimaryDaughters) continue;
1119 Particle particle(mcdaughter);
1120 Particle* daughter = m_particles.appendNew(particle);
1121 daughter->addRelationTo(mcdaughter);
1122 mother->appendDaughter(daughter, false);
1123
1124 if (mcdaughter->getNDaughters() > 0)
1125 appendDaughtersRecursive(daughter);
1126 }
1127}
@ c_PrimaryParticle
bit 0: Particle is primary particle.
Definition MCParticle.h:47
void appendDaughtersRecursive(Particle *mother)
recursively append bottom of a particle's decay chain (daughters of mother, granddaughters of daughte...
void appendDaughter(const Particle *daughter, const bool updateType=true, const int daughterProperty=c_Ordinary)
Appends index of daughter to daughters index array.
Definition Particle.cc:707
std::vector< Particle * > getDaughters() const
Returns a vector of pointers to daughter particles.
Definition Particle.cc:668
T * getRelated(const std::string &name="", const std::string &namedRelation="") const
Get the object to or from which this object has a relation.

◆ assignMCParticleFromECLCluster()

void assignMCParticleFromECLCluster ( Particle * newPart,
const ECLCluster * cluster ) const
private

Assigns the MCParticle relation to the newPart.

The given cluster is used to find the best MCParticle matching.

Definition at line 894 of file ParticleLoaderModule.cc.

895{
896 // ECLCluster can be matched to multiple MCParticles
897 // order the relations by weights and set Particle -> multiple MCParticle relation
898 // preserve the weight
899 const RelationVector<MCParticle> mcRelations = cluster->getRelationsTo<MCParticle>();
900
901 // order relations by weights
902 std::vector<std::pair<double, int>> weightsAndIndices;
903 for (unsigned int iMCParticle = 0; iMCParticle < mcRelations.size(); iMCParticle++) {
904 const MCParticle* relMCParticle = mcRelations[iMCParticle];
905 if (relMCParticle) {
906 double weight = mcRelations.weight(iMCParticle);
907 weightsAndIndices.emplace_back(weight, relMCParticle->getArrayIndex());
908 }
909 }
910
911 // sort descending by weight
912 std::sort(weightsAndIndices.begin(), weightsAndIndices.end(),
913 ValueIndexPairSorting::higherPair<decltype(weightsAndIndices)::value_type>);
914
915 // set relations to mcparticles
916 for (auto& weightsAndIndex : weightsAndIndices) {
917 const MCParticle* relMCParticle = m_mcparticles[weightsAndIndex.second];
918 double weight = weightsAndIndex.first;
919
920 // TODO: study this further and avoid hard-coded values
921 // set the relation only if the MCParticle(reconstructed Particle)'s
922 // energy contribution to this cluster amounts to at least 30(20)%
923 if (relMCParticle)
924 if (weight / newPart->getECLClusterEnergy() > 0.20
925 && weight / relMCParticle->getEnergy() > 0.30)
926 newPart->addRelationTo(relMCParticle, weight);
927 }
928
929}
float getEnergy() const
Return particle energy in GeV.
Definition MCParticle.h:136
int getArrayIndex() const
Get 0-based index of the particle in the corresponding MCParticle list.
Definition MCParticle.h:233
StoreArray< MCParticle > m_mcparticles
StoreArray of MCParticles.
double getECLClusterEnergy() const
Returns the energy of the ECLCluster for the particle.
Definition Particle.cc:1004
size_t size() const
Get number of relations.
float weight(int index) const
Get weight with index.

◆ beginRun()

◆ chargedClustersToParticles()

void chargedClustersToParticles ( )
private

Loads ECLCluster and KLMCluster objects that are being matched with Track as Particle to StoreArray<Particle> and adds it to the ParticleList.

Definition at line 931 of file ParticleLoaderModule.cc.

932{
933 if (m_ChargedCluster2Plists.empty()) // nothing to do
934 return;
935
936 // loop over all ParticleLists
937 for (auto chargedCluster2Plist : m_ChargedCluster2Plists) {
938
939 string listName = get<c_PListName>(chargedCluster2Plist);
940 string antiListName = get<c_AntiPListName>(chargedCluster2Plist);
941 int pdgCode = get<c_PListPDGCode>(chargedCluster2Plist);
942 bool isSelfConjugatedParticle = get<c_IsPListSelfConjugated>(chargedCluster2Plist);
943 Const::ParticleType thisType(pdgCode);
944
945 StoreObjPtr<ParticleList> plist(listName);
946 // since a particle list in the ParticleLoader always contains all possible objects
947 // we check whether it already exists in this path and can skip any further steps if it does
948 if (plist.isValid())
949 continue;
950 plist.create();
951 plist->initialize(pdgCode, listName);
952
953 // create anti-particle list if necessary
954 if (!isSelfConjugatedParticle) {
955 StoreObjPtr<ParticleList> antiPlist(antiListName);
956 antiPlist.create();
957 antiPlist->initialize(-1 * pdgCode, antiListName);
958
959 antiPlist->bindAntiParticleList(*(plist));
960 }
961
962 const StoreObjPtr<ParticleList> sourceList(m_sourceParticleListName);
963 if (!sourceList.isValid())
964 B2FATAL("ParticleList " << m_sourceParticleListName << " could not be found or is not valid!");
965
966 for (unsigned int iPart = 0; iPart < sourceList->getListSize(); iPart++) {
967
968 const Particle* sourcePart = sourceList->getParticle(iPart);
969 const Track* sourceTrack = sourceList->getParticle(iPart)->getTrack();
970 if (!sourceTrack)
971 continue;
972
973 // ECL clusters
975
976 const ECLCluster* cluster = sourcePart->getECLCluster();
977 if (isValidECLCluster(cluster, pdgCode, false)) {
978 Particle particle(cluster, thisType);
979 if (particle.getParticleSource() != Particle::c_ECLCluster) {
980 B2FATAL("Particle created from ECLCluster does not have ECLCluster type.");
981 }
982 Particle* newPart = m_particles.appendNew(particle);
983 assignMCParticleFromECLCluster(newPart, cluster);
984 plist->addParticle(newPart);
985 }
986 } else {
987
988 // loop over all clusters matched to this track
989 const RelationVector<ECLCluster> clusters = sourceTrack->getRelationsTo<ECLCluster>();
990 for (unsigned int iCluster = 0; iCluster < clusters.size(); iCluster++) {
991 const ECLCluster* cluster = clusters[iCluster];
992 if (!isValidECLCluster(cluster, pdgCode, false))
993 continue; // go to next iCluster
994
995 Particle particle(cluster, thisType);
996 if (particle.getParticleSource() != Particle::c_ECLCluster) {
997 B2FATAL("Particle created from ECLCluster does not have ECLCluster type.");
998 }
999 Particle* newPart = m_particles.appendNew(particle);
1000 assignMCParticleFromECLCluster(newPart, cluster);
1001 plist->addParticle(newPart);
1002 }
1003 }
1004
1005 // KLM clusters
1006 const KLMCluster* cluster = sourcePart->getKLMCluster();
1007 if (!cluster) continue; // go to next iPart
1008
1009 if (std::isnan(cluster->getMomentumMag())) {
1010 B2DEBUG(19, "Skipping KLMCluster because its momentum is NaN. "
1011 "This can happen if the timing calibration is missing or wrong, so that the velocity is calculated to be negative.");
1012 continue;
1013 }
1014
1015 // create particle and check its type before adding it to list
1016 Particle particle(cluster, pdgCode);
1017 if (particle.getParticleSource() != Particle::c_KLMCluster) {
1018 B2FATAL("Particle created from KLMCluster does not have KLMCluster type.");
1019 }
1020 Particle* newPart = m_particles.appendNew(particle);
1021
1022 const MCParticle* mcParticle = cluster->getRelated<MCParticle>();
1023 if (mcParticle) newPart->addRelationTo(mcParticle);
1024
1025 plist->addParticle(newPart);
1026 }
1027
1028 } // loop over particle lists
1029}
bool isValidECLCluster(const ECLCluster *cluster, const int pdgCode, bool onlyNeutral) const
Checks if the given ECLCluster is valid for the pdgCode.
std::vector< PList > m_ChargedCluster2Plists
Collection of PLists that will collect Particles created from charged-cluster.
void assignMCParticleFromECLCluster(Particle *newPart, const ECLCluster *cluster) const
Assigns the MCParticle relation to the newPart.
const KLMCluster * getKLMCluster() const
Returns the pointer to the KLMCluster object that was used to create this Particle (ParticleType == c...
Definition Particle.cc:1011
const ECLCluster * getECLCluster() const
Returns the pointer to the ECLCluster object that was used to create this Particle (if ParticleType =...
Definition Particle.cc:976
RelationVector< TO > getRelationsTo(const std::string &name="", const std::string &namedRelation="") const
Get the relations that point from this object to another store array.

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

180{
183 newModule->setName(getName());
184 newModule->m_package = m_package;
185 newModule->m_propertyFlags = m_propertyFlags;
186 newModule->m_logConfig = m_logConfig;
187 newModule->m_conditions = m_conditions;
188
189 return newModule;
190}
std::shared_ptr< Module > registerModule(const std::string &moduleName, std::string sharedLibPath="") noexcept(false)
Creates an instance of a module and registers it to the ModuleManager.
static ModuleManager & Instance()
Exception is thrown if the requested module could not be created by the ModuleManager.
void setParameters(const ModuleParamList &params)
Set values for parameters from other parameter list.
const ModuleParamList & getParamList() const
Return module param list.
Definition Module.h:362
const std::string & getName() const
Returns the name of the module.
Definition Module.h:186
const std::string & getType() const
Returns the type of the module (i.e.
Definition Module.cc:41
unsigned int m_propertyFlags
The properties of the module as bitwise or (with |) of EModulePropFlags.
Definition Module.h:511
ModuleParamList m_moduleParamList
List storing and managing all parameter of the module.
Definition Module.h:515
void setName(const std::string &name)
Set the name of the module.
Definition Module.h:213
LogConfig m_logConfig
The log system configuration of the module.
Definition Module.h:513
std::vector< ModuleCondition > m_conditions
Module condition, only non-null if set.
Definition Module.h:520
std::string m_package
Package this module is found in (may be empty).
Definition Module.h:509
std::shared_ptr< Module > ModulePtr
Defines a pointer to a module object as a boost shared pointer.
Definition Module.h:43

◆ def_beginRun()

virtual void def_beginRun ( )
inlineprotectedvirtualinherited

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

Reimplemented in PyModule.

Definition at line 425 of file Module.h.

425{ beginRun(); }

◆ 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 438 of file Module.h.

438{ endRun(); }

◆ def_event()

virtual void def_event ( )
inlineprotectedvirtualinherited

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

Reimplemented in PyModule.

Definition at line 431 of file Module.h.

431{ event(); }

◆ 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 419 of file Module.h.

419{ initialize(); }

◆ def_terminate()

virtual void def_terminate ( )
inlineprotectedvirtualinherited

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

Reimplemented in PyModule.

Definition at line 444 of file Module.h.

444{ terminate(); }

◆ dummyToParticles()

void dummyToParticles ( )
private

Loads dummy object as Particle of specified type to StoreArray<Particle> and adds it to the ParticleList.

Definition at line 356 of file ParticleLoaderModule.cc.

357{
358 if (m_Dummies2Plists.empty()) // nothing to do
359 return;
360 if (m_Dummies2Plists.size() != 1)
361 B2ERROR("ParticleLoaderModule::dummyToParticles Multiple particle lists are not supported!");
362 auto dummy2Plist = m_Dummies2Plists[0];
363 string dummyListName = get<c_PListName>(dummy2Plist);
364 string antiDummyListName = get<c_AntiPListName>(dummy2Plist);
365 int pdgCode = get<c_PListPDGCode>(dummy2Plist);
366 bool isSelfConjugatedParticle = get<c_IsPListSelfConjugated>(dummy2Plist);
367
368 StoreObjPtr<ParticleList> plist(dummyListName);
369 plist.create();
370 plist->initialize(pdgCode, dummyListName);
371
372 if (!isSelfConjugatedParticle) {
373 StoreObjPtr<ParticleList> antiPlist(antiDummyListName);
374 antiPlist.create();
375 antiPlist->initialize(-1 * pdgCode, antiDummyListName);
376 antiPlist->bindAntiParticleList(*(plist));
377 }
378
379 TMatrixFSym covariance(7);
380 for (int row = 0; row < 7; ++row) { //diag
381 covariance(row, row) = m_dummyCovMatrix;
382 }
383
384 Particle* newPart = nullptr;
385 Particle* newAntiPart = nullptr;
386
387 auto isFlavored = (isSelfConjugatedParticle) ? Particle::EFlavorType::c_Unflavored : Particle::EFlavorType::c_Flavored;
388
389 ROOT::Math::PxPyPzEVector zero4Vector = {0., 0., 0., 0.};
390
391 newPart = m_particles.appendNew(zero4Vector, pdgCode, isFlavored, Particle::EParticleSourceObject::c_NoMDSTSource,
393 if (m_dummyCovMatrix > 0.) newPart->setMomentumVertexErrorMatrix(covariance);
394 if (m_dummyTreatAsInvisible) newPart->writeExtraInfo("treeFitterTreatMeAsInvisible", 1);
395 plist->addParticle(newPart);
396
397 if (!isSelfConjugatedParticle) {
398 newAntiPart = m_particles.appendNew(zero4Vector, -pdgCode, isFlavored, Particle::EParticleSourceObject::c_NoMDSTSource,
400 if (m_dummyCovMatrix > 0.) newAntiPart->setMomentumVertexErrorMatrix(covariance);
401 if (m_dummyTreatAsInvisible) newAntiPart->writeExtraInfo("treeFitterTreatMeAsInvisible", 1);
402 plist->addParticle(newAntiPart);
403 }
404
405}
std::vector< PList > m_Dummies2Plists
Collection of PLists that will collect Particles created from Dummies.
void writeExtraInfo(const std::string &name, const double value)
Sets the user defined extraInfo.
Definition Particle.cc:1393
void setMomentumVertexErrorMatrix(const TMatrixFSym &errMatrix)
Sets 7x7 error matrix.
Definition Particle.cc:424

◆ eclAndKLMClustersToParticles()

void eclAndKLMClustersToParticles ( )
private

Loads ECLCluster and KLMCluster object as Particle to StoreArray<Particle> and adds it to the ParticleList.

Definition at line 792 of file ParticleLoaderModule.cc.

793{
794 if (m_ECLKLMClusters2Plists.empty()) // nothing to do
795 return;
796
797 // loop over all ParticleLists
798 for (auto eclKLMCluster2Plist : m_ECLKLMClusters2Plists) {
799 string listName = get<c_PListName>(eclKLMCluster2Plist);
800 int pdgCode = get<c_PListPDGCode>(eclKLMCluster2Plist);
801 Const::ParticleType thisType(pdgCode);
802
803 StoreObjPtr<ParticleList> plist(listName);
804 // since a particle list in the ParticleLoader always contains all possible objects
805 // we check whether it already exists in this path and can skip any further steps if it does
806 if (plist.isValid())
807 continue;
808 plist.create();
809 plist->initialize(pdgCode, listName);
810
811 plist->setEditable(true); // :all list is originally reserved. we have to set it as editable.
812
813 // load reconstructed neutral ECL clusters as photons or Klongs or neutrons
814 for (int i = 0; i < m_eclclusters.getEntries(); i++) {
815 const ECLCluster* cluster = m_eclclusters[i];
816
817 if (!isValidECLCluster(cluster, pdgCode, true))
818 continue;
819
820 // create particle and check it before adding to list
821 Particle particle(cluster, thisType);
822 if (particle.getParticleSource() != Particle::c_ECLCluster) {
823 B2FATAL("Particle created from ECLCluster does not have ECLCluster type.");
824 continue;
825 }
826 Particle* newPart = m_particles.appendNew(particle);
827
828 assignMCParticleFromECLCluster(newPart, cluster);
829
830 // add particle to list
831 plist->addParticle(newPart);
832 }
833
834 // load reconstructed KLM clusters as Klongs or neutrons or photons
835 for (int i = 0; i < m_klmclusters.getEntries(); i++) {
836 const KLMCluster* cluster = m_klmclusters[i];
837
838 if (std::isnan(cluster->getMomentumMag())) {
839 B2DEBUG(19, "Skipping KLMCluster because its momentum is NaN. "
840 "This can happen if the timing calibration is missing or wrong, so that the velocity is calculated to be negative.");
841 continue;
842 }
843
844 const MCParticle* mcParticle = cluster->getRelated<MCParticle>();
845
846 // create particle and check its type before adding it to list
847 Particle particle(cluster, pdgCode);
848 if (particle.getParticleSource() != Particle::c_KLMCluster) {
849 B2FATAL("Particle created from KLMCluster does not have KLMCluster type.");
850 }
851 Particle* newPart = m_particles.appendNew(particle);
852
853 if (mcParticle)
854 newPart->addRelationTo(mcParticle);
855
856 // add particle to list
857 plist->addParticle(newPart);
858 }
859
860 plist->setEditable(false); // set the :all list as not editable.
861 } // loop over particle lists
862}
std::vector< PList > m_ECLKLMClusters2Plists
Collection of PLists that will collect Particles created from ECLClusters and KLMClusters.
StoreArray< KLMCluster > m_klmclusters
StoreArray of KLMCluster.
StoreArray< ECLCluster > m_eclclusters
StoreArray of ECLCluster.

◆ endRun()

virtual void endRun ( )
inlinevirtualinherited

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

97{
98 if (m_conditions.empty()) return false;
99
100 //okay, a condition was set for this Module...
101 if (!m_hasReturnValue) {
102 B2FATAL("A condition was set for '" << getName() << "', but the module did not set a return value!");
103 }
104
105 for (const auto& condition : m_conditions) {
106 if (condition.evaluate(m_returnValue)) {
107 return true;
108 }
109 }
110 return false;
111}
int m_returnValue
The return value.
Definition Module.h:518
bool m_hasReturnValue
True, if the return value is set.
Definition Module.h:517

◆ event()

void event ( )
overridevirtual

Event processor.

Reimplemented from Module.

Definition at line 312 of file ParticleLoaderModule.cc.

313{
314 if (not m_particleExtraInfoMap) {
315 m_particleExtraInfoMap.create();
316 }
317
318 if (m_useROEs)
320 else if (m_useDummy)
322 else if (m_useMCParticles)
324 else if (m_loadChargedCluster)
326 else {
331
332 }
333}
void mcParticlesToParticles()
Loads specified MCParticles as Particle to StoreArray<Particle>
void v0sToParticles()
Loads V0 object as Particle of specified type to StoreArray<Particle> and adds it to the ParticleList...
void kinksToParticles()
Loads Kink object as Particle of specified type to StoreArray<Particle> and adds it to the ParticleLi...
void roeToParticles()
Loads ROE object as Particle of specified type to StoreArray<Particle> and adds it to the ParticleLis...
void tracksToParticles()
Loads Track object as Particle to StoreArray<Particle> and adds it to the ParticleList.
StoreObjPtr< ParticleExtraInfoMap > m_particleExtraInfoMap
object pointer to extra info map
void dummyToParticles()
Loads dummy object as Particle of specified type to StoreArray<Particle> and adds it to the ParticleL...
void chargedClustersToParticles()
Loads ECLCluster and KLMCluster objects that are being matched with Track as Particle to StoreArray<P...
void eclAndKLMClustersToParticles()
Loads ECLCluster and KLMCluster object as Particle to StoreArray<Particle> and adds it to the Particl...

◆ exposePythonAPI()

void exposePythonAPI ( )
staticinherited

Exposes methods of the Module class to Python.

Definition at line 325 of file Module.cc.

326{
327 // to avoid confusion between std::arg and boost::python::arg we want a shorthand namespace as well
328 namespace bp = boost::python;
329
330 docstring_options options(true, true, false); //userdef, py sigs, c++ sigs
331
332 void (Module::*setReturnValueInt)(int) = &Module::setReturnValue;
333
334 enum_<Module::EAfterConditionPath>("AfterConditionPath",
335 R"(Determines execution behaviour after a conditional path has been executed:
336
337.. attribute:: END
338
339 End processing of this path after the conditional path. (this is the default for if_value() etc.)
340
341.. attribute:: CONTINUE
342
343 After the conditional path, resume execution after this module.)")
344 .value("END", Module::EAfterConditionPath::c_End)
345 .value("CONTINUE", Module::EAfterConditionPath::c_Continue)
346 ;
347
348 /* Do not change the names of >, <, ... we use them to serialize conditional paths */
349 enum_<Belle2::ModuleCondition::EConditionOperators>("ConditionOperator")
356 ;
357
358 enum_<Module::EModulePropFlags>("ModulePropFlags",
359 R"(Flags to indicate certain low-level features of modules, see :func:`Module.set_property_flags()`, :func:`Module.has_properties()`. Most useful flags are:
360
361.. attribute:: PARALLELPROCESSINGCERTIFIED
362
363 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.)
364
365.. attribute:: HISTOGRAMMANAGER
366
367 This module is used to manage histograms accumulated by other modules
368
369.. attribute:: TERMINATEINALLPROCESSES
370
371 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.
372)")
373 .value("INPUT", Module::EModulePropFlags::c_Input)
374 .value("OUTPUT", Module::EModulePropFlags::c_Output)
375 .value("PARALLELPROCESSINGCERTIFIED", Module::EModulePropFlags::c_ParallelProcessingCertified)
376 .value("HISTOGRAMMANAGER", Module::EModulePropFlags::c_HistogramManager)
377 .value("INTERNALSERIALIZER", Module::EModulePropFlags::c_InternalSerializer)
378 .value("TERMINATEINALLPROCESSES", Module::EModulePropFlags::c_TerminateInAllProcesses)
379 ;
380
381 //Python class definition
382 class_<Module, PyModule> module("Module", R"(
383Base class for Modules.
384
385A module is the smallest building block of the framework.
386A typical event processing chain consists of a Path containing
387modules. By inheriting from this base class, various types of
388modules can be created. To use a module, please refer to
389:func:`Path.add_module()`. A list of modules is available by running
390``basf2 -m`` or ``basf2 -m package``, detailed information on parameters is
391given by e.g. ``basf2 -m RootInput``.
392
393The 'Module Development' section in the manual provides detailed information
394on how to create modules, setting parameters, or using return values/conditions:
395https://xwiki.desy.de/xwiki/rest/p/f4fa4/#HModuleDevelopment
396
397)");
398 module
399 .def("__str__", &Module::getPathString)
400 .def("name", &Module::getName, return_value_policy<copy_const_reference>(),
401 "Returns the name of the module. Can be changed via :func:`set_name() <Module.set_name()>`, use :func:`type() <Module.type()>` for identifying a particular module class.")
402 .def("type", &Module::getType, return_value_policy<copy_const_reference>(),
403 "Returns the type of the module (i.e. class name minus 'Module')")
404 .def("set_name", &Module::setName, args("name"), R"(
405Set custom name, e.g. to distinguish multiple modules of the same type.
406
407>>> path.add_module('EventInfoSetter')
408>>> ro = path.add_module('RootOutput', branchNames=['EventMetaData'])
409>>> ro.set_name('RootOutput_metadata_only')
410>>> print(path)
411[EventInfoSetter -> RootOutput_metadata_only]
412
413)")
414 .def("description", &Module::getDescription, return_value_policy<copy_const_reference>(),
415 "Returns the description of this module.")
416 .def("package", &Module::getPackage, return_value_policy<copy_const_reference>(),
417 "Returns the package this module belongs to.")
418 .def("available_params", &_getParamInfoListPython,
419 "Return list of all module parameters as `ModuleParamInfo` instances")
420 .def("has_properties", &Module::hasProperties, (bp::arg("properties")),
421 R"DOCSTRING(Allows to check if the module has the given properties out of `ModulePropFlags` set.
422
423>>> if module.has_properties(ModulePropFlags.PARALLELPROCESSINGCERTIFIED):
424>>> ...
425
426Parameters:
427 properties (int): bitmask of `ModulePropFlags` to check for.
428)DOCSTRING")
429 .def("set_property_flags", &Module::setPropertyFlags, args("property_mask"),
430 "Set module properties in the form of an OR combination of `ModulePropFlags`.");
431 {
432 // python signature is too crowded, make ourselves
433 docstring_options subOptions(true, false, false); //userdef, py sigs, c++ sigs
434 module
435 .def("if_value", &Module::if_value,
436 (bp::arg("expression"), bp::arg("condition_path"), bp::arg("after_condition_path")= Module::EAfterConditionPath::c_End),
437 R"DOCSTRING(if_value(expression, condition_path, after_condition_path=AfterConditionPath.END)
438
439Sets a conditional sub path which will be executed after this
440module if the return value set in the module passes the given ``expression``.
441
442Modules can define a return value (int or bool) using ``setReturnValue()``,
443which can be used in the steering file to split the Path based on this value, for example
444
445>>> module_with_condition.if_value("<1", another_path)
446
447In case the return value of the ``module_with_condition`` for a given event is
448less than 1, the execution will be diverted into ``another_path`` for this event.
449
450You could for example set a special return value if an error occurs, and divert
451the execution into a path containing :b2:mod:`RootOutput` if it is found;
452saving only the data producing/produced by the error.
453
454After a conditional path has executed, basf2 will by default stop processing
455the path for this event. This behaviour can be changed by setting the
456``after_condition_path`` argument.
457
458Parameters:
459 expression (str): Expression to determine if the conditional path should be executed.
460 This should be one of the comparison operators ``<``, ``>``, ``<=``,
461 ``>=``, ``==``, or ``!=`` followed by a numerical value for the return value
462 condition_path (Path): path to execute in case the expression is fulfilled
463 after_condition_path (AfterConditionPath): What to do once the ``condition_path`` has been executed.
464)DOCSTRING")
465 .def("if_false", &Module::if_false,
466 (bp::arg("condition_path"), bp::arg("after_condition_path")= Module::EAfterConditionPath::c_End),
467 R"DOC(if_false(condition_path, after_condition_path=AfterConditionPath.END)
468
469Sets a conditional sub path which will be executed after this module if
470the return value of the module evaluates to False. This is equivalent to
471calling `if_value` with ``expression=\"<1\"``)DOC")
472 .def("if_true", &Module::if_true,
473 (bp::arg("condition_path"), bp::arg("after_condition_path")= Module::EAfterConditionPath::c_End),
474 R"DOC(if_true(condition_path, after_condition_path=AfterConditionPath.END)
475
476Sets a conditional sub path which will be executed after this module if
477the return value of the module evaluates to True. It is equivalent to
478calling `if_value` with ``expression=\">=1\"``)DOC");
479 }
480 module
481 .def("has_condition", &Module::hasCondition,
482 "Return true if a conditional path has been set for this module "
483 "using `if_value`, `if_true` or `if_false`")
484 .def("get_all_condition_paths", &_getAllConditionPathsPython,
485 "Return a list of all conditional paths set for this module using "
486 "`if_value`, `if_true` or `if_false`")
487 .def("get_all_conditions", &_getAllConditionsPython,
488 "Return a list of all conditional path expressions set for this module using "
489 "`if_value`, `if_true` or `if_false`")
490 .add_property("logging", make_function(&Module::getLogConfig, return_value_policy<reference_existing_object>()),
491 &Module::setLogConfig)
@ c_GE
Greater or equal than: ">=".
@ c_SE
Smaller or equal than: "<=".
@ c_GT
Greater than: ">".
@ c_NE
Not equal: "!=".
@ c_EQ
Equal: "=" or "==".
@ c_ST
Smaller than: "<".
void if_value(const std::string &expression, const std::shared_ptr< Path > &path, EAfterConditionPath afterConditionPath=EAfterConditionPath::c_End)
Add a condition to the module.
Definition Module.cc:79
void if_true(const std::shared_ptr< Path > &path, EAfterConditionPath afterConditionPath=EAfterConditionPath::c_End)
A simplified version to set the condition of the module.
Definition Module.cc:90
void setReturnValue(int value)
Sets the return value for this module as integer.
Definition Module.cc:220
@ c_HistogramManager
This module is used to manage histograms accumulated by other modules.
Definition Module.h:81
@ c_Input
This module is an input module (reads data).
Definition Module.h:78
@ c_InternalSerializer
This module is an internal serializer/deserializer for parallel processing.
Definition Module.h:82
@ c_Output
This module is an output module (writes data).
Definition Module.h:79
@ c_TerminateInAllProcesses
When using parallel processing, call this module's terminate() function in all processes().
Definition Module.h:83
void if_false(const std::shared_ptr< Path > &path, EAfterConditionPath afterConditionPath=EAfterConditionPath::c_End)
A simplified version to add a condition to the module.
Definition Module.cc:85
ModuleCondition::EAfterConditionPath EAfterConditionPath
Forward the EAfterConditionPath definition from the ModuleCondition.
Definition Module.h:88

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

134{
135 if (m_conditions.empty()) return EAfterConditionPath::c_End;
136
137 //okay, a condition was set for this Module...
138 if (!m_hasReturnValue) {
139 B2FATAL("A condition was set for '" << getName() << "', but the module did not set a return value!");
140 }
141
142 for (const auto& condition : m_conditions) {
143 if (condition.evaluate(m_returnValue)) {
144 return condition.getAfterConditionPath();
145 }
146 }
147
148 return EAfterConditionPath::c_End;
149}

◆ getAllConditionPaths()

std::vector< std::shared_ptr< Path > > getAllConditionPaths ( ) const
inherited

Return all condition paths currently set (no matter if the condition is true or not).

Definition at line 150 of file Module.cc.

151{
152 std::vector<std::shared_ptr<Path>> allConditionPaths;
153 for (const auto& condition : m_conditions) {
154 allConditionPaths.push_back(condition.getPath());
155 }
156
157 return allConditionPaths;
158}

◆ getAllConditions()

const std::vector< ModuleCondition > & getAllConditions ( ) const
inlineinherited

Return all set conditions for this module.

Definition at line 323 of file Module.h.

324 {
325 return m_conditions;
326 }

◆ getCondition()

const ModuleCondition * getCondition ( ) const
inlineinherited

Return a pointer to the first condition (or nullptr, if none was set)

Definition at line 313 of file Module.h.

314 {
315 if (m_conditions.empty()) {
316 return nullptr;
317 } else {
318 return &m_conditions.front();
319 }
320 }

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

114{
115 PathPtr p;
116 if (m_conditions.empty()) return p;
117
118 //okay, a condition was set for this Module...
119 if (!m_hasReturnValue) {
120 B2FATAL("A condition was set for '" << getName() << "', but the module did not set a return value!");
121 }
122
123 for (const auto& condition : m_conditions) {
124 if (condition.evaluate(m_returnValue)) {
125 return condition.getPath();
126 }
127 }
128
129 // if none of the conditions were true, return a null pointer.
130 return p;
131}
std::shared_ptr< Path > PathPtr
Defines a pointer to a path object as a boost shared pointer.
Definition Path.h:35

◆ getDescription()

const std::string & getDescription ( ) const
inlineinherited

Returns the description of the module.

Definition at line 201 of file Module.h.

201{return m_description;}

◆ 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 RootInputModule, and RootOutputModule.

Definition at line 133 of file Module.h.

134 {
135 return std::vector<std::string>();
136 }

◆ getLogConfig()

LogConfig & getLogConfig ( )
inlineinherited

Returns the log system configuration.

Definition at line 224 of file Module.h.

224{return m_logConfig;}

◆ getModules()

std::list< ModulePtr > getModules ( ) const
inlineoverrideprivatevirtualinherited

no submodules, return empty list

Implements PathElement.

Definition at line 505 of file Module.h.

505{ return std::list<ModulePtr>(); }

◆ 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 186 of file Module.h.

186{return m_name;}

◆ getPackage()

const std::string & getPackage ( ) const
inlineinherited

Returns the package this module is in.

Definition at line 196 of file Module.h.

196{return m_package;}

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

280{
282}
std::shared_ptr< boost::python::list > getParamInfoListPython() const
Returns a python list of all parameters.

◆ getParamList()

const ModuleParamList & getParamList ( ) const
inlineinherited

Return module param list.

Definition at line 362 of file Module.h.

362{ return m_moduleParamList; }

◆ getPathString()

std::string getPathString ( ) const
overrideprivatevirtualinherited

return the module name.

Implements PathElement.

Definition at line 192 of file Module.cc.

193{
194
195 std::string output = getName();
196
197 for (const auto& condition : m_conditions) {
198 output += condition.getString();
199 }
200
201 return output;
202}

◆ 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 380 of file Module.h.

380{ return m_returnValue; }

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

42{
43 if (m_type.empty())
44 B2FATAL("Module type not set for " << getName());
45 return m_type;
46}
std::string m_type
The type of the module, saved as a string.
Definition Module.h:508

◆ hasCondition()

bool hasCondition ( ) const
inlineinherited

Returns true if at least one condition was set for the module.

Definition at line 310 of file Module.h.

310{ return not m_conditions.empty(); };

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

161{
162 return (propertyFlags & m_propertyFlags) == propertyFlags;
163}

◆ hasReturnValue()

bool hasReturnValue ( ) const
inlineinherited

Return true if this module has a valid return value set.

Definition at line 377 of file Module.h.

377{ return m_hasReturnValue; }

◆ hasUnsetForcedParams()

bool hasUnsetForcedParams ( ) const
inherited

Returns true and prints error message if the module has unset parameters which the user has to set in the steering file.

Definition at line 166 of file Module.cc.

167{
169 std::string allMissing = "";
170 for (const auto& s : missing)
171 allMissing += s + " ";
172 if (!missing.empty())
173 B2ERROR("The following required parameters of Module '" << getName() << "' were not specified: " << allMissing <<
174 "\nPlease add them to your steering file.");
175 return !missing.empty();
176}
std::vector< std::string > getUnsetForcedParams() const
Returns list of unset parameters (if they are required to have a value.

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

86{
87 if_value("<1", path, afterConditionPath);
88}

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

91{
92 if_value(">=1", path, afterConditionPath);
93}

◆ 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://xwiki.desy.de/xwiki/rest/p/a94f2 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.

80{
81 m_conditions.emplace_back(expression, path, afterConditionPath);
82}

◆ initialize()

void initialize ( )
overridevirtual

Initialize the Module.

This method is called at the beginning of data processing.

Reimplemented from Module.

Definition at line 112 of file ParticleLoaderModule.cc.

113{
114 B2INFO("ParticleLoader's Summary of Actions:");
115
116 if ((int)m_useMCParticles + (int)m_useROEs + (int)m_useDummy + (int)m_loadChargedCluster > 1)
117 B2FATAL("The options on how to load the Particle are not valid. The incompatible combination of options is selected. "
118 << "useMCParticles: " << m_useMCParticles << ", useROEs: " << m_useROEs << ", useDummy: " << m_useDummy
119 << ", loadChargedCluster: " << m_loadChargedCluster);
120
121 m_particles.registerInDataStore();
122 // regsiter temporary StoreArray of Particles needed for matchedMC variable
123 StoreArray<Particle>("tempParticles").registerInDataStore(DataStore::c_DontWriteOut);
124 m_particleExtraInfoMap.registerInDataStore();
125 //register relations if these things exists
126 if (m_mcparticles.isOptional()) {
127 m_particles.registerRelationTo(m_mcparticles);
128 StoreArray<Particle>("tempParticles").registerRelationTo(m_mcparticles);
129 }
130 if (m_pidlikelihoods.isOptional()) {
131 m_particles.registerRelationTo(m_pidlikelihoods);
132 }
133 if (m_trackfitresults.isOptional()) {
134 m_particles.registerRelationTo(m_trackfitresults);
135 }
136
137 if (m_useMCParticles) {
138 m_mcparticles.isRequired();
139 }
140
141 if (m_useROEs) {
142 m_roes.isRequired();
143 m_roes.registerRelationTo(m_particles);
144
145 StoreArray<RestOfEvent> nestedRoes("NestedRestOfEvents");
146 if (nestedRoes.isOptional())
147 nestedRoes.registerRelationTo(m_particles);
148 }
149
150 if (m_decayStrings.empty()) {
151 B2WARNING("Obsolete usage of the ParticleLoader module (load all MDST objects as all possible Particle object types). Specify the particle type via decayStrings module parameter instead.");
152 } else {
153 for (auto decayString : m_decayStrings) {
154
155 // obtain the output particle lists from the decay string
156 const bool valid = m_decaydescriptor.init(decayString);
157 if (!valid)
158 B2ERROR("ParticleLoaderModule::initialize Invalid input DecayString: " << decayString);
159
160 // Mother particle
161 const DecayDescriptorParticle* mother = m_decaydescriptor.getMother();
162 int nProducts = m_decaydescriptor.getNDaughters();
163
164 int pdgCode = mother->getPDGCode();
165 // The default list name is "all"
166 string listName = mother->getName() + ":all";
167
168 // Full name for ROE, dummy, MCParticle, chargedCluster particles, and (anti-)neutrons
169 if (m_useROEs or m_useDummy or m_useMCParticles or m_loadChargedCluster or abs(pdgCode) == abs(Const::neutron.getPDGCode()))
170 listName = mother->getFullName();
171 // Kinks get the label "kink"
172 else if (nProducts == 1)
173 listName = mother->getName() + ":kink";
174 // V0s get the label "V0"
175 else if (nProducts > 1)
176 listName = mother->getName() + ":V0";
177
178 string antiListName = ParticleListName::antiParticleListName(listName);
179 bool isSelfConjugatedParticle = (listName == antiListName);
180
181 StoreObjPtr<ParticleList> particleList(listName);
182 // if the particle list doesn't exist, we have to register it
183 if (!particleList.isOptional()) {
185 particleList.registerInDataStore(flags);
186 // Neutrons are not self-conjugated, but we cannot distinguish particle from anti-particle
187 if (!isSelfConjugatedParticle and (abs(pdgCode) != abs(Const::neutron.getPDGCode()) or m_useMCParticles)) {
188 StoreObjPtr<ParticleList> antiParticleList(antiListName);
189 antiParticleList.registerInDataStore(flags);
190 }
191 } else if (m_useMCParticles) {
192 B2WARNING("ParticleList " << listName << " already exists and will not be created again. " <<
193 "Please note that the given options (addDaughters, skipNonPrimaryDaughters, skipNonPrimary, skipInitial) do not apply to "
194 << listName);
195 } else if (m_loadChargedCluster) {
196 B2WARNING("ParticleList " << listName << " already exists and will not be created again. " <<
197 "Please note that the given option, useOnlyMostEnergeticECLCluster, does not apply to "
198 << listName);
199 }
200
201 if (not isValidPDGCode(pdgCode) and (m_useMCParticles == false and m_useROEs == false and m_useDummy == false))
202 B2ERROR("Invalid particle type requested to be loaded. Set a valid decayString module parameter.");
203
204 // if we're not loading MCParticles and we are loading K0S, Lambdas, or photons --> ee then this decaystring is a V0
205 bool mdstSourceIsV0 = false;
206 if (!m_useMCParticles &&
207 (abs(pdgCode) == abs(Const::Kshort.getPDGCode()) || abs(pdgCode) == abs(Const::Lambda.getPDGCode())
208 || (abs(pdgCode) == abs(Const::photon.getPDGCode()) && m_addDaughters == true)))
209 mdstSourceIsV0 = true;
210
211 // if we're not loading MCParticles, and we are loading kaon/pion/muon/charged sigmas with one daughter,
212 // then this decaystring is a kink
213 bool mdstSourceIsKink = false;
214 if (!m_useMCParticles && nProducts == 1 &&
215 (abs(pdgCode) == abs(Const::kaon.getPDGCode()) || abs(pdgCode) == abs(Const::pion.getPDGCode())
216 || abs(pdgCode) == abs(Const::muon.getPDGCode()) || abs(pdgCode) == abs(3222) || abs(pdgCode) == abs(3112)))
217 mdstSourceIsKink = true;
218
219 if (mdstSourceIsV0) {
220 if (nProducts == 2) {
221 m_properties = m_decaydescriptor.getProperty() | mother->getProperty(); // only used for V0s
222 if (m_decaydescriptor.getDaughter(0)->getMother()->getPDGCode() * m_decaydescriptor.getDaughter(1)->getMother()->getPDGCode() > 0)
223 B2ERROR("MDST source of the particle list is V0, the two daughters should have opposite charge");
224 } else {
225 B2ERROR("ParticleLoaderModule::initialize Invalid input DecayString " << decayString
226 << ". MDST source of the particle list is V0, DecayString should contain exactly two daughters, as well as the mother particle.");
227 }
228 } else {
229 if (!mdstSourceIsKink && nProducts > 0) {
230 if (m_useROEs or m_useDummy) {
231 B2INFO("ParticleLoaderModule: Replacing the source particle list name by " <<
232 m_decaydescriptor.getDaughter(0)->getMother()->getFullName()
233 << " all other daughters will be ignored.");
234 m_sourceParticleListName = m_decaydescriptor.getDaughter(0)->getMother()->getFullName();
235 } else {
236 B2ERROR("ParticleLoaderModule::initialize Invalid input DecayString " << decayString
237 << ". DecayString should not contain any daughters, only the mother particle.");
238 }
239 }
240 }
241
243 B2ERROR("The sourceParticleListName is not given. The charged ParticleList is required for the chargedCluster loading.");
244
245 // add PList to corresponding collection of Lists
246 if (isSelfConjugatedParticle or (abs(pdgCode) == Const::neutron.getPDGCode() and not m_useMCParticles))
247 B2INFO(" o) creating ParticleList with name: " << listName);
248 else
249 B2INFO(" o) creating (anti-)ParticleList with name: " << listName << " (" << antiListName << ")");
250 if (m_useROEs) {
251 B2INFO(" -> MDST source: RestOfEvents");
252 m_ROE2Plists.emplace_back(pdgCode, listName, antiListName, isSelfConjugatedParticle);
253 } else if (m_useDummy) {
254 B2INFO(" -> MDST source: No MDST source");
255 m_Dummies2Plists.emplace_back(pdgCode, listName, antiListName, isSelfConjugatedParticle);
256 } else if (m_useMCParticles) {
257 B2INFO(" -> MDST source: MCParticles");
258 m_MCParticles2Plists.emplace_back(pdgCode, listName, antiListName, isSelfConjugatedParticle);
259 } else if (m_loadChargedCluster) {
260 if (abs(pdgCode) == abs(Const::photon.getPDGCode()) || abs(pdgCode) == abs(Const::Klong.getPDGCode())
261 || abs(pdgCode) == abs(Const::neutron.getPDGCode())) {
262 m_ChargedCluster2Plists.emplace_back(pdgCode, listName, antiListName, isSelfConjugatedParticle);
263 B2INFO(" -> MDST source: ECLClusters and KLMClusters being matched with Tracks");
264 } else {
265 B2ERROR("The Particle type must be gamma, K_L0, or (anti-)n0 for the loadChargedCluster option.");
266 }
267 } else {
268 bool chargedFSP = Const::chargedStableSet.contains(Const::ParticleType(abs(pdgCode)));
269 if (chargedFSP) {
270 if (!mdstSourceIsKink) {
271 B2INFO(" -> MDST source: Tracks");
272 m_Tracks2Plists.emplace_back(pdgCode, listName, antiListName, isSelfConjugatedParticle);
273 } else {
274 B2INFO(" -> MDST source: Kinks");
275 m_Kink2Plists.emplace_back(pdgCode, listName, antiListName, isSelfConjugatedParticle);
276 }
277 }
278
279 if (abs(pdgCode) == abs(Const::photon.getPDGCode())) {
280 if (m_addDaughters == false) {
281 m_ECLKLMClusters2Plists.emplace_back(pdgCode, listName, antiListName, isSelfConjugatedParticle);
282 B2INFO(" -> MDST source: ECLClusters and KLMClusters");
283 } else {
284 B2INFO(" -> MDST source: V0");
285 m_V02Plists.emplace_back(pdgCode, listName, antiListName, isSelfConjugatedParticle);
286 }
287 }
288
289 if (abs(pdgCode) == abs(Const::Kshort.getPDGCode())) {
290 B2INFO(" -> MDST source: V0");
291 m_V02Plists.emplace_back(pdgCode, listName, antiListName, isSelfConjugatedParticle);
292 }
293
294 if (abs(pdgCode) == abs(Const::Klong.getPDGCode()) || abs(pdgCode) == abs(Const::neutron.getPDGCode())) {
295 B2INFO(" -> MDST source: exclusively KLMClusters or exclusively ECLClusters (matching between those not used)");
296 m_ECLKLMClusters2Plists.emplace_back(pdgCode, listName, antiListName, isSelfConjugatedParticle);
297 }
298
299 if (abs(pdgCode) == abs(Const::Lambda.getPDGCode())) {
300 B2INFO(" -> MDST source: V0");
301 m_V02Plists.emplace_back(pdgCode, listName, antiListName, isSelfConjugatedParticle);
302 }
303 }
304 }
305 }
306
307
308 m_chargeZeroTrackCounts = std::vector<int>(m_Tracks2Plists.size(), 0);
309 m_sameChargeDaughtersV0Counts = std::vector<int>(m_V02Plists.size(), 0);
310}
bool contains(const ParticleType &p) const
Returns true if and only if the set contains 'p'.
Definition UnitConst.cc:421
static const ParticleType neutron
neutron particle
Definition Const.h:675
static const ParticleType Lambda
Lambda particle.
Definition Const.h:679
static const ChargedStable muon
muon particle
Definition Const.h:660
static const ParticleSet chargedStableSet
set of charged stable particles
Definition Const.h:618
static const ChargedStable pion
charged pion particle
Definition Const.h:661
static const ParticleType Klong
K^0_L particle.
Definition Const.h:678
static const ParticleType Kshort
K^0_S particle.
Definition Const.h:677
static const ChargedStable kaon
charged kaon particle
Definition Const.h:662
static const ParticleType photon
photon particle
Definition Const.h:673
EStoreFlags
Flags describing behaviours of objects etc.
Definition DataStore.h:69
@ c_WriteOut
Object/array should be saved by output modules.
Definition DataStore.h:70
@ c_DontWriteOut
Object/array should be NOT saved by output modules.
Definition DataStore.h:71
int getPDGCode() const
Return PDG code.
std::string getFullName() const
returns the full name of the particle full_name = name:label
std::string getName() const
evt.pdl name of the particle.
int getProperty() const
return property of the particle.
StoreArray< TrackFitResult > m_trackfitresults
StoreArray of TrackFitResults.
std::vector< int > m_sameChargeDaughtersV0Counts
internally used to count the number of V0s with same charge daughters
bool isValidPDGCode(const int pdgCode)
returns true if the PDG code determined from the decayString is valid
std::vector< PList > m_Kink2Plists
Collection of PLists that will collect Particles created from Kink.
StoreArray< PIDLikelihood > m_pidlikelihoods
StoreArray of PIDLikelihoods.
StoreArray< RestOfEvent > m_roes
StoreArray of ROEs.
std::vector< PList > m_MCParticles2Plists
Collection of PLists that will collect Particles created from MCParticles.
std::vector< PList > m_V02Plists
Collection of PLists that will collect Particles created from V0.
DecayDescriptor m_decaydescriptor
Decay descriptor for parsing the user specified DecayString.
std::vector< int > m_chargeZeroTrackCounts
internally used to count number of tracks with charge zero
std::vector< PList > m_Tracks2Plists
Collection of PLists that will collect Particles created from Tracks.
int m_properties
Particle property to be assigned only on V0s.
std::string antiParticleListName(const std::string &listName)
Returns name of anti-particle-list corresponding to listName.

◆ isValidECLCluster()

bool isValidECLCluster ( const ECLCluster * cluster,
const int pdgCode,
bool onlyNeutral ) const
private

Checks if the given ECLCluster is valid for the pdgCode.

Definition at line 864 of file ParticleLoaderModule.cc.

865{
866 if (!cluster)
867 return false;
868
869 // ECLClusters can be reconstructed under different hypotheses, for
870 // example photons or neutral hadrons, we only load particles from these
871 // for now
872 if (!cluster->isNeutral() and onlyNeutral)
873 return false;
874
875 if (not cluster->hasHypothesis(ECLCluster::EHypothesisBit::c_nPhotons)
876 and not cluster->hasHypothesis(ECLCluster::EHypothesisBit::c_neutralHadron))
877 return false;
878
879 // don't fill photon list with clusters that don't have
880 // the nPhotons hypothesis (ECL people call this N1)
881 if (pdgCode == Const::photon.getPDGCode()
882 and not cluster->hasHypothesis(ECLCluster::EHypothesisBit::c_nPhotons))
883 return false;
884
885 // don't fill a KLong nor a (anti-)neutron list with clusters that don't have the neutral
886 // hadron hypothesis set (ECL people call this N2)
887 if ((pdgCode == Const::Klong.getPDGCode() or abs(pdgCode) == Const::neutron.getPDGCode())
888 and not cluster->hasHypothesis(ECLCluster::EHypothesisBit::c_neutralHadron))
889 return false;
890
891 return true;
892}
@ c_nPhotons
CR is split into n photons (N1)
Definition ECLCluster.h:41
@ c_neutralHadron
CR is reconstructed as a neutral hadron (N2)
Definition ECLCluster.h:43

◆ isValidPDGCode()

bool isValidPDGCode ( const int pdgCode)
private

returns true if the PDG code determined from the decayString is valid

Definition at line 1082 of file ParticleLoaderModule.cc.

1083{
1084 bool result = false;
1085
1086 // is particle type = charged final state particle?
1087 if (Const::chargedStableSet.find(abs(pdgCode)) != Const::invalidParticle)
1088 return true;
1089
1090 if (abs(pdgCode) == abs(Const::photon.getPDGCode()))
1091 return true;
1092
1093 if (abs(pdgCode) == abs(Const::Kshort.getPDGCode()))
1094 return true;
1095
1096 if (abs(pdgCode) == abs(Const::Klong.getPDGCode()))
1097 return true;
1098
1099 if (abs(pdgCode) == abs(Const::Lambda.getPDGCode()))
1100 return true;
1101
1102 if (abs(pdgCode) == abs(Const::neutron.getPDGCode()))
1103 return true;
1104
1105 return result;
1106}
static const ParticleType invalidParticle
Invalid particle, used internally.
Definition Const.h:681

◆ kinksToParticles()

void kinksToParticles ( )
private

Loads Kink object as Particle of specified type to StoreArray<Particle> and adds it to the ParticleList.

Definition at line 636 of file ParticleLoaderModule.cc.

637{
638
639 if (m_Kink2Plists.empty()) // nothing to do
640 return;
641
642 // loop over all ParticleLists
643 for (size_t ilist = 0; ilist < m_Kink2Plists.size(); ilist++) {
644 auto kink2Plist = m_Kink2Plists[ilist];
645 string listName = get<c_PListName>(kink2Plist);
646 string antiListName = get<c_AntiPListName>(kink2Plist);
647 int pdgCode = get<c_PListPDGCode>(kink2Plist);
648 bool isSelfConjugatedParticle = get<c_IsPListSelfConjugated>(kink2Plist);
649
650 StoreObjPtr<ParticleList> plist(listName);
651 // since a particle list in the ParticleLoader always contains all possible objects
652 // we check whether it already exists in this path and can skip any further steps if it does
653 if (plist.isValid())
654 continue;
655 plist.create();
656 plist->initialize(pdgCode, listName);
657
658 if (!isSelfConjugatedParticle) {
659 StoreObjPtr<ParticleList> antiPlist(antiListName);
660 antiPlist.create();
661 antiPlist->initialize(-1 * pdgCode, antiListName);
662
663 antiPlist->bindAntiParticleList(*(plist));
664 }
665
666 plist->setEditable(true); // :kink list is originally reserved. we have to set it as editable.
667
668 // load reconstructed Kinks according to requested mother hypothesis
669 for (int i = 0; i < m_kinks.getEntries(); i++) {
670 const Kink* kink = m_kinks[i];
671
672 Const::ChargedStable motherType(abs(pdgCode));
673 const Track* motherTrack = kink->getMotherTrack();
674 const TrackFitResult* motherTrackFit = kink->getMotherTrackFitResultStart();
675
676 int motherCharge = motherTrackFit->getChargeSign();
677 if (motherCharge == 0) {
678 B2DEBUG(19, "Kink track with charge = 0 skipped!");
679 continue;
680 }
681
682 const auto& motherMCParticleWithWeight = motherTrack->getRelatedToWithWeight<MCParticle>();
683 const PIDLikelihood* motherPID = motherTrack->getRelated<PIDLikelihood>();
684
685 // a particle object creation from kink mother with the correct option
686 Particle kinkP(kink, motherType, kink->getTrackFitResultIndexMotherStart());
687
688 // append the particle to the Particle StoreArray
689 Particle* newPart = m_particles.appendNew(kinkP);
690
691 if (motherPID)
692 newPart->addRelationTo(motherPID);
693 if (motherMCParticleWithWeight.first)
694 newPart->addRelationTo(motherMCParticleWithWeight.first, motherMCParticleWithWeight.second);
695 newPart->writeExtraInfo("kinkDaughterPDGCode", m_decaydescriptor.getDaughter(0)->getMother()->getPDGCode());
696
697 // add the new particle to the ParticleList
698 plist->addParticle(newPart);
699 }
700
701 plist->setEditable(false); // set the :kink list as not editable.
702 }
703}
short getTrackFitResultIndexMotherStart() const
Get index of the TrackFitResult of mother at the starting point.
Definition Kink.h:88
Track * getMotherTrack() const
Get mother Track.
Definition Kink.cc:27
TrackFitResult * getMotherTrackFitResultStart() const
Get the TrackFitResult of mother at the starting point.
Definition Kink.cc:43
StoreArray< Kink > m_kinks
StoreArray of Kinks.
std::pair< TO *, float > getRelatedToWithWeight(const std::string &name="", const std::string &namedRelation="") const
Get first related object & weight of relation pointing to an array.
short getChargeSign() const
Return track charge (1 or -1).

◆ mcParticlesToParticles()

void mcParticlesToParticles ( )
private

Loads specified MCParticles as Particle to StoreArray<Particle>

Definition at line 1031 of file ParticleLoaderModule.cc.

1032{
1033 if (m_MCParticles2Plists.empty()) // nothing to do
1034 return;
1035
1036 // create all lists
1037 for (auto mcParticle2Plist : m_MCParticles2Plists) {
1038 string listName = get<c_PListName>(mcParticle2Plist);
1039 string antiListName = get<c_AntiPListName>(mcParticle2Plist);
1040 int pdgCode = get<c_PListPDGCode>(mcParticle2Plist);
1041 bool isSelfConjugatedParticle = get<c_IsPListSelfConjugated>(mcParticle2Plist);
1042
1043 StoreObjPtr<ParticleList> plist(listName);
1044 if (plist.isValid())
1045 continue;
1046 plist.create();
1047 plist->initialize(pdgCode, listName);
1048
1049 if (!isSelfConjugatedParticle) {
1050 StoreObjPtr<ParticleList> antiPlist(antiListName);
1051 antiPlist.create();
1052 antiPlist->initialize(-1 * pdgCode, antiListName);
1053
1054 antiPlist->bindAntiParticleList(*(plist));
1055 }
1056
1057 for (int i = 0; i < m_mcparticles.getEntries(); i++) {
1058 const MCParticle* mcParticle = m_mcparticles[i];
1059
1060 if (abs(pdgCode) != abs(mcParticle->getPDG()))
1061 continue;
1062
1064 continue;
1065
1066 if (m_skipInitial and mcParticle->isInitial())
1067 continue;
1068
1069 Particle particle(mcParticle);
1070 Particle* newPart = m_particles.appendNew(particle);
1071 newPart->addRelationTo(mcParticle);
1072
1073 //append the whole bottom part of the decay tree to this particle
1075
1076 plist->addParticle(newPart);
1077 }
1078
1079 }
1080}
bool hasStatus(unsigned short int bitmask) const
Return if specific status bit is set.
Definition MCParticle.h:118
int getPDG() const
Return PDG code of particle.
Definition MCParticle.h:101
bool isInitial() const
Check if particle is an initial particle such as ISR.
Definition MCParticle.h:580

◆ roeToParticles()

void roeToParticles ( )
private

Loads ROE object as Particle of specified type to StoreArray<Particle> and adds it to the ParticleList.

Definition at line 408 of file ParticleLoaderModule.cc.

409{
410 if (m_ROE2Plists.empty()) // nothing to do
411 return;
412 // Multiple particle lists are not supported
413 auto roe2Plist = m_ROE2Plists[0];
414 string listName = get<c_PListName>(roe2Plist);
415 string antiListName = get<c_AntiPListName>(roe2Plist);
416 int pdgCode = get<c_PListPDGCode>(roe2Plist);
417 bool isSelfConjugatedParticle = get<c_IsPListSelfConjugated>(roe2Plist);
418
419 StoreObjPtr<ParticleList> plist(listName);
420 // since a particle list in the ParticleLoader always contains all possible objects
421 // we check whether it already exists in this path and can skip any further steps if it does
422 if (plist.isValid())
423 return;
424 plist.create();
425 plist->initialize(pdgCode, listName);
426
427 if (!isSelfConjugatedParticle) {
428 StoreObjPtr<ParticleList> antiPlist(antiListName);
429 antiPlist.create();
430 antiPlist->initialize(-1 * pdgCode, antiListName);
431 antiPlist->bindAntiParticleList(*(plist));
432 }
433 if (m_sourceParticleListName != "") {
434 // Take related ROEs from a particle list
435 StoreObjPtr<ParticleList> pList(m_sourceParticleListName);
436 if (!pList.isValid())
437 B2FATAL("ParticleList " << m_sourceParticleListName << " could not be found or is not valid!");
438
439 for (unsigned int i = 0; i < pList->getListSize(); i++) {
440 RestOfEvent* roe = pList->getParticle(i)->getRelatedTo<RestOfEvent>("ALL");
441 if (!roe) {
442 B2ERROR("ParticleList " << m_sourceParticleListName << " has no associated ROEs!");
443 } else {
444
445 if (isSelfConjugatedParticle)
446 addROEToParticleList(roe, i, pdgCode, isSelfConjugatedParticle);
447 else if (i < pList->getListSize(false))
448 addROEToParticleList(roe, i, pdgCode, isSelfConjugatedParticle);
449 else
450 addROEToParticleList(roe, i, -1 * pdgCode, isSelfConjugatedParticle);
451 }
452 }
453
454 } else {
455 // Take all ROE if no particle list provided
456 for (int i = 0; i < m_roes.getEntries(); i++) {
458 }
459 }
460}
void addROEToParticleList(RestOfEvent *roe, int mdstIndex, int pdgCode=0, bool isSelfConjugatedParticle=true)
Helper method to load ROE object as Particle.

◆ setAbortLevel()

void setAbortLevel ( int abortLevel)
inherited

Configure the abort log level.

Definition at line 67 of file Module.cc.

68{
69 m_logConfig.setAbortLevel(static_cast<LogConfig::ELogLevel>(abortLevel));
70}
ELogLevel
Definition of the supported log levels.
Definition LogConfig.h:26
void setAbortLevel(ELogLevel abortLevel)
Configure the abort level.
Definition LogConfig.h:112

◆ setDebugLevel()

void setDebugLevel ( int debugLevel)
inherited

Configure the debug messaging level.

Definition at line 61 of file Module.cc.

62{
63 m_logConfig.setDebugLevel(debugLevel);
64}
void setDebugLevel(int debugLevel)
Configure the debug messaging level.
Definition LogConfig.h:98

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

215{
216 m_description = description;
217}
std::string m_description
The description of the module.
Definition Module.h:510

◆ setLogConfig()

void setLogConfig ( const LogConfig & logConfig)
inlineinherited

Set the log system configuration.

Definition at line 229 of file Module.h.

229{m_logConfig = logConfig;}

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

74{
75 m_logConfig.setLogInfo(static_cast<LogConfig::ELogLevel>(logLevel), logInfo);
76}
void setLogInfo(ELogLevel logLevel, unsigned int logInfo)
Configure the printed log information for the given level.
Definition LogConfig.h:127

◆ setLogLevel()

void setLogLevel ( int logLevel)
inherited

Configure the log level.

Definition at line 55 of file Module.cc.

56{
57 m_logConfig.setLogLevel(static_cast<LogConfig::ELogLevel>(logLevel));
58}
void setLogLevel(ELogLevel logLevel)
Configure the log level.
Definition LogConfig.cc:25

◆ 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 213 of file Module.h.

213{ m_name = name; };

◆ setParamList()

void setParamList ( const ModuleParamList & params)
inlineprotectedinherited

Replace existing parameter list.

Definition at line 500 of file Module.h.

500{ m_moduleParamList = params; }

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

235{
236 LogSystem& logSystem = LogSystem::Instance();
237 logSystem.updateModule(&(getLogConfig()), getName());
238 try {
240 } catch (std::runtime_error& e) {
241 throw std::runtime_error("Cannot set parameter '" + name + "' for module '"
242 + m_name + "': " + e.what());
243 }
244
245 logSystem.updateModule(nullptr);
246}
void updateModule(const LogConfig *moduleLogConfig=nullptr, const std::string &moduleName="")
Sets the log configuration to the given module log configuration and sets the module name This method...
Definition LogSystem.h:200
static LogSystem & Instance()
Static method to get a reference to the LogSystem instance.
Definition LogSystem.cc:28
LogConfig & getLogConfig()
Returns the log system configuration.
Definition Module.h:224
std::string m_name
The name of the module, saved as a string (user-modifiable)
Definition Module.h:507
void setParamPython(const std::string &name, const PythonObject &pyObj)
Implements a method for setting boost::python objects.

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

250{
251
252 LogSystem& logSystem = LogSystem::Instance();
253 logSystem.updateModule(&(getLogConfig()), getName());
254
255 boost::python::list dictKeys = dictionary.keys();
256 int nKey = boost::python::len(dictKeys);
257
258 //Loop over all keys in the dictionary
259 for (int iKey = 0; iKey < nKey; ++iKey) {
260 boost::python::object currKey = dictKeys[iKey];
261 boost::python::extract<std::string> keyProxy(currKey);
262
263 if (keyProxy.check()) {
264 const boost::python::object& currValue = dictionary[currKey];
265 setParamPython(keyProxy, currValue);
266 } else {
267 B2ERROR("Setting the module parameters from a python dictionary: invalid key in dictionary!");
268 }
269 }
270
271 logSystem.updateModule(nullptr);
272}
void setParamPython(const std::string &name, const boost::python::object &pyObj)
Implements a method for setting boost::python objects.
Definition Module.cc:234

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

209{
210 m_propertyFlags = propertyFlags;
211}

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

228{
229 m_hasReturnValue = true;
230 m_returnValue = value;
231}

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

221{
222 m_hasReturnValue = true;
223 m_returnValue = value;
224}

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

49{
50 if (!m_type.empty())
51 B2FATAL("Trying to change module type from " << m_type << " is not allowed, the value is assumed to be fixed.");
52 m_type = type;
53}

◆ terminate()

void terminate ( )
overridevirtual

Terminate the Module.

This method is called at the end of data processing.

Reimplemented from Module.

Definition at line 335 of file ParticleLoaderModule.cc.

336{
337 // report track errors integrated
338 for (size_t i = 0; i < m_Tracks2Plists.size(); i++)
339 if (m_chargeZeroTrackCounts[i] > 0) {
340 auto track2Plist = m_Tracks2Plists[i];
341 B2WARNING("There were " << m_chargeZeroTrackCounts[i]
342 << " tracks skipped because of zero charge for "
343 << get<c_PListName>(track2Plist));
344 }
345 // report V0 errors integrated
346 for (size_t i = 0; i < m_V02Plists.size(); i++)
348 auto v02Plist = m_V02Plists[i];
349 B2WARNING("There were " << m_sameChargeDaughtersV0Counts[i]
350 << " v0s skipped because of same charge daughters for "
351 << get<c_PListName>(v02Plist));
352 }
353}

◆ tracksToParticles()

void tracksToParticles ( )
private

Loads Track object as Particle to StoreArray<Particle> and adds it to the ParticleList.

Definition at line 705 of file ParticleLoaderModule.cc.

706{
707 if (m_Tracks2Plists.empty()) // nothing to do
708 return;
709
710 // loop over all requested particle lists
711 for (size_t ilist = 0; ilist < m_Tracks2Plists.size(); ilist++) {
712 auto track2Plist = m_Tracks2Plists[ilist];
713 string listName = get<c_PListName>(track2Plist);
714 string antiListName = get<c_AntiPListName>(track2Plist);
715 int pdgCode = get<c_PListPDGCode>(track2Plist);
716 bool isSelfConjugatedParticle = get<c_IsPListSelfConjugated>(track2Plist);
717
718 StoreObjPtr<ParticleList> plist(listName);
719 // since a particle list in the ParticleLoader always contains all possible objects
720 // we check whether it already exists in this path and can skip any further steps if it does
721 if (plist.isValid())
722 continue;
723 plist.create();
724 plist->initialize(pdgCode, listName);
725
726 // if cc exists then also create and bind that list
727 if (!isSelfConjugatedParticle) {
728 StoreObjPtr<ParticleList> antiPlist(antiListName);
729 antiPlist.create();
730 antiPlist->initialize(-1 * pdgCode, antiListName);
731
732 antiPlist->bindAntiParticleList(*(plist));
733 }
734
735 plist->setEditable(true); // :all list is originally reserved. we have to set it as editable.
736
737 // the inner loop over all tracks from which Particles
738 // are created, and get sorted in the particle lists
739 for (int i = 0; i < m_tracks.getEntries(); i++) {
740 const Track* track = m_tracks[i];
741 const PIDLikelihood* pid = track->getRelated<PIDLikelihood>();
742 const auto& mcParticleWithWeight = track->getRelatedToWithWeight<MCParticle>();
743
744 // if a special track hypothesis is requested, use it
745 if (m_trackHypothesis != 0) pdgCode = m_trackHypothesis;
746 Const::ChargedStable type(abs(pdgCode));
747
748 // load the TrackFitResult for the requested particle or if not available use
749 // the one with the closest mass
750 const TrackFitResult* trackFit = track->getTrackFitResultWithClosestMass(type);
751
752 if (!trackFit) { // should never happen with the "closest mass" getter - leave as a sanity check
753 B2WARNING("Track returned null TrackFitResult pointer for ChargedStable::getPDGCode() = " << type.getPDGCode());
754 continue;
755 }
756
757 if (m_enforceFitHypothesis && (trackFit->getParticleType().getPDGCode() != type.getPDGCode())) {
758 // the required hypothesis does not exist for this track, skip it
759 continue;
760 }
761
762 // charge zero tracks can appear, filter them and
763 // count number of tracks filtered out
764 int charge = trackFit->getChargeSign();
765 if (charge == 0) {
766 B2DEBUG(19, "Track with charge = 0 skipped!");
768 continue;
769 }
770
771 // create particle and add it to the Particle list.
772 Particle particle(track->getArrayIndex(), trackFit, type);
773
774 if (particle.getParticleSource() == Particle::c_Track) { // should always hold but...
775
776 Particle* newPart = m_particles.appendNew(particle);
777 if (pid)
778 newPart->addRelationTo(pid);
779 if (mcParticleWithWeight.first)
780 newPart->addRelationTo(mcParticleWithWeight.first, mcParticleWithWeight.second);
781 newPart->addRelationTo(trackFit);
782
783 plist->addParticle(newPart);
784
785 } // sanity check correct particle type
786 } // loop over tracks
787
788 plist->setEditable(false); // set the :all list as not editable.
789 } // particle lists
790}
int getPDGCode() const
PDG code.
Definition Const.h:473
StoreArray< Track > m_tracks
StoreArray of Tracks.
Const::ParticleType getParticleType() const
Getter for ParticleType of the mass hypothesis of the track fit.
double charge(int pdgCode)
Returns electric charge of a particle with given pdg code.
Definition EvtPDLUtil.cc:44

◆ v0sToParticles()

void v0sToParticles ( )
private

Loads V0 object as Particle of specified type to StoreArray<Particle> and adds it to the ParticleList.

Definition at line 492 of file ParticleLoaderModule.cc.

493{
494 if (m_V02Plists.empty()) // nothing to do
495 return;
496
497 // check if the order of the daughters in the decay string (decided by the user) is the same as the v0 daughters' order (fixed)
498 bool matchingDaughtersOrder = true;
499 if (m_decaydescriptor.getDaughter(0)->getMother()->getPDGCode() < 0
500 && m_decaydescriptor.getDaughter(1)->getMother()->getPDGCode() > 0)
501 matchingDaughtersOrder = false;
502
503 // loop over all ParticleLists
504 for (size_t ilist = 0; ilist < m_V02Plists.size(); ilist++) {
505 auto v02Plist = m_V02Plists[ilist];
506 string listName = get<c_PListName>(v02Plist);
507 string antiListName = get<c_AntiPListName>(v02Plist);
508 int pdgCode = get<c_PListPDGCode>(v02Plist);
509 bool isSelfConjugatedParticle = get<c_IsPListSelfConjugated>(v02Plist);
510
511 StoreObjPtr<ParticleList> plist(listName);
512 // since a particle list in the ParticleLoader always contains all possible objects
513 // we check whether it already exists in this path and can skip any further steps if it does
514 if (plist.isValid())
515 continue;
516 plist.create();
517 plist->initialize(pdgCode, listName);
518
519 if (!isSelfConjugatedParticle) {
520 StoreObjPtr<ParticleList> antiPlist(antiListName);
521 antiPlist.create();
522 antiPlist->initialize(-1 * pdgCode, antiListName);
523
524 antiPlist->bindAntiParticleList(*(plist));
525 }
526
527 plist->setEditable(true); // :V0 list is originally reserved. we have to set it as editable.
528
529 // load reconstructed V0s as Kshorts (pi-pi+ combination), Lambdas (p+pi- combinations), and converted photons (e-e+ combinations)
530 for (int i = 0; i < m_v0s.getEntries(); i++) {
531 const V0* v0 = m_v0s[i];
532 Const::ParticleType v0Type = v0->getV0Hypothesis();
533
534 if (abs(pdgCode) != abs(v0Type.getPDGCode()))
535 continue;
536
537 // check if the charge of the 2 V0's daughters is opposite
538 if (v0->getTrackFitResults().first->getChargeSign() == v0->getTrackFitResults().second->getChargeSign()) {
539 B2DEBUG(19, "V0 with same charge daughters skipped!");
541 continue;
542 }
543
544 Const::ChargedStable pTypeP(Const::pion);
545 Const::ChargedStable pTypeM(Const::pion);
547
548 if (v0Type.getPDGCode() == Const::Kshort.getPDGCode()) { // K0s -> pi+ pi-
549 pTypeP = Const::pion;
550 pTypeM = Const::pion;
551 } else if (v0Type.getPDGCode() == Const::Lambda.getPDGCode()) { // Lambda -> p+ pi-
552 pTypeP = Const::proton;
553 pTypeM = Const::pion;
554 v0FlavorType = Particle::c_Flavored; // K0s are not flavoured, lambdas are
555 } else if (v0Type.getPDGCode() == Const::antiLambda.getPDGCode()) { // anti-Lambda -> pi+ anti-p-
556 pTypeP = Const::pion;
557 pTypeM = Const::proton;
558 v0FlavorType = Particle::c_Flavored;
559 } else if (v0Type.getPDGCode() == Const::photon.getPDGCode()) { // gamma -> e+ e-
560 pTypeP = Const::electron;
561 pTypeM = Const::electron;
562 } else {
563 B2WARNING("Unknown V0 hypothesis!");
564 }
565
566 // check if, given the initial user's decay descriptor, the current v0 is a particle or an anti-particle.
567 // in the V0 the order of the daughters is fixed, first the positive and then the negative; to be coherent with the decay descriptor, when creating
568 // one particle list and one anti-particle, the v0 daughters' order has to be switched only in one case
569 bool correctOrder = matchingDaughtersOrder;
570 if (abs(v0Type.getPDGCode()) == abs(m_decaydescriptor.getMother()->getPDGCode())
571 && v0Type.getPDGCode() != m_decaydescriptor.getMother()->getPDGCode())
572 correctOrder = !correctOrder;
573
574 std::pair<Track*, Track*> v0Tracks = v0->getTracks();
575 std::pair<TrackFitResult*, TrackFitResult*> v0TrackFitResults = v0->getTrackFitResults();
576
577 Particle daugP((v0Tracks.first)->getArrayIndex(), v0TrackFitResults.first, pTypeP);
578 Particle daugM((v0Tracks.second)->getArrayIndex(), v0TrackFitResults.second, pTypeM);
579
580 const PIDLikelihood* pidP = (v0Tracks.first)->getRelated<PIDLikelihood>();
581 const PIDLikelihood* pidM = (v0Tracks.second)->getRelated<PIDLikelihood>();
582
583 const auto& mcParticlePWithWeight = (v0Tracks.first)->getRelatedToWithWeight<MCParticle>();
584 const auto& mcParticleMWithWeight = (v0Tracks.second)->getRelatedToWithWeight<MCParticle>();
585
586 // add V0 daughters to the Particle StoreArray
587 Particle* newDaugP;
588 Particle* newDaugM;
589
590 if (correctOrder) {
591 newDaugP = m_particles.appendNew(daugP);
592 newDaugM = m_particles.appendNew(daugM);
593 } else {
594 newDaugM = m_particles.appendNew(daugM);
595 newDaugP = m_particles.appendNew(daugP);
596 }
597
598 // if there are PIDLikelihoods and MCParticles then also add relations to the particles
599 if (pidP)
600 newDaugP->addRelationTo(pidP);
601 if (mcParticlePWithWeight.first)
602 newDaugP->addRelationTo(mcParticlePWithWeight.first, mcParticlePWithWeight.second);
603 newDaugP->addRelationTo(v0TrackFitResults.first);
604
605 if (pidM)
606 newDaugM->addRelationTo(pidM);
607 if (mcParticleMWithWeight.first)
608 newDaugM->addRelationTo(mcParticleMWithWeight.first, mcParticleMWithWeight.second);
609 newDaugM->addRelationTo(v0TrackFitResults.second);
610
611 // sum the 4-momenta of the daughters and construct a particle object
612 ROOT::Math::PxPyPzEVector v0Momentum = newDaugP->get4Vector() + newDaugM->get4Vector();
613 Particle v0P(v0Momentum, v0Type.getPDGCode(), v0FlavorType,
614 Particle::EParticleSourceObject::c_V0, v0->getArrayIndex());
615 v0P.setProperty(m_properties);
616
617 // add the daughters of the V0 (in the correct order) and don't update
618 // the type to c_Composite (i.e. maintain c_V0)
619 if (correctOrder) {
620 v0P.appendDaughter(newDaugP, false);
621 v0P.appendDaughter(newDaugM, false);
622 } else {
623 v0P.appendDaughter(newDaugM, false);
624 v0P.appendDaughter(newDaugP, false);
625 }
626
627 // append the particle to the Particle StoreArray and add the new particle to the ParticleList
628 Particle* newPart = m_particles.appendNew(v0P);
629 plist->addParticle(newPart);
630 }
631
632 plist->setEditable(false); // set the :V0 list as not editable.
633 }
634}
static const ParticleType antiLambda
Anti-Lambda particle.
Definition Const.h:680
static const ChargedStable proton
proton particle
Definition Const.h:663
static const ChargedStable electron
electron particle
Definition Const.h:659
StoreArray< V0 > m_v0s
StoreArray of V0s.
ROOT::Math::PxPyPzEVector get4Vector() const
Returns Lorentz vector.
Definition Particle.h:567
EFlavorType
describes flavor type, see getFlavorType().
Definition Particle.h:96
@ c_Flavored
Is either particle or antiparticle.
Definition Particle.h:98
int getArrayIndex() const
Returns this object's array index (in StoreArray), or -1 if not found.
std::pair< Track *, Track * > getTracks() const
Get pair of the Tracks, that are part of the V0 particle.
Definition V0.h:45
std::pair< TrackFitResult *, TrackFitResult * > getTrackFitResults() const
Get pair of the TrackFitResults, that are part of the V0 particle.
Definition V0.h:58
Const::ParticleType getV0Hypothesis() const
Get the hypothesis under which the V0 particle was created.
Definition V0.cc:25

Member Data Documentation

◆ m_addDaughters

bool m_addDaughters
private

toggle addition of the bottom part of the particle's decay chain

Definition at line 230 of file ParticleLoaderModule.h.

◆ m_ChargedCluster2Plists

std::vector<PList> m_ChargedCluster2Plists
private

Collection of PLists that will collect Particles created from charged-cluster.

Definition at line 224 of file ParticleLoaderModule.h.

◆ m_chargeZeroTrackCounts

std::vector<int> m_chargeZeroTrackCounts
private

internally used to count number of tracks with charge zero

Definition at line 244 of file ParticleLoaderModule.h.

◆ m_conditions

std::vector<ModuleCondition> m_conditions
privateinherited

Module condition, only non-null if set.

Definition at line 520 of file Module.h.

◆ m_decaydescriptor

DecayDescriptor m_decaydescriptor
private

Decay descriptor for parsing the user specified DecayString.

Definition at line 211 of file ParticleLoaderModule.h.

◆ m_decayStrings

std::vector<std::string> m_decayStrings
private

Input decay strings specifying the particles being created/loaded.

Definition at line 214 of file ParticleLoaderModule.h.

◆ m_description

std::string m_description
privateinherited

The description of the module.

Definition at line 510 of file Module.h.

◆ m_Dummies2Plists

std::vector<PList> m_Dummies2Plists
private

Collection of PLists that will collect Particles created from Dummies.

Definition at line 223 of file ParticleLoaderModule.h.

◆ m_dummyCovMatrix

double m_dummyCovMatrix
private

diag value of cov matrix for dummy particles

Definition at line 238 of file ParticleLoaderModule.h.

◆ m_dummyMDSTIndex

int m_dummyMDSTIndex
private

mdst index for dummy particles

Definition at line 237 of file ParticleLoaderModule.h.

◆ m_dummyTreatAsInvisible

bool m_dummyTreatAsInvisible
private

should treeFitter treat the particle as invisible?

Definition at line 239 of file ParticleLoaderModule.h.

◆ m_eclclusters

StoreArray<ECLCluster> m_eclclusters
private

StoreArray of ECLCluster.

Definition at line 192 of file ParticleLoaderModule.h.

◆ m_ECLKLMClusters2Plists

std::vector<PList> m_ECLKLMClusters2Plists
private

Collection of PLists that will collect Particles created from ECLClusters and KLMClusters.

Definition at line 222 of file ParticleLoaderModule.h.

◆ m_enforceFitHypothesis

bool m_enforceFitHypothesis
private
Initial value:
=
false

If true, a Particle is only created if a track fit with the particle hypothesis passed to the ParticleLoader is available.

Definition at line 241 of file ParticleLoaderModule.h.

◆ m_hasReturnValue

bool m_hasReturnValue
privateinherited

True, if the return value is set.

Definition at line 517 of file Module.h.

◆ m_Kink2Plists

std::vector<PList> m_Kink2Plists
private

Collection of PLists that will collect Particles created from Kink.

Definition at line 219 of file ParticleLoaderModule.h.

◆ m_kinks

StoreArray<Kink> m_kinks
private

StoreArray of Kinks.

Definition at line 200 of file ParticleLoaderModule.h.

◆ m_klmclusters

StoreArray<KLMCluster> m_klmclusters
private

StoreArray of KLMCluster.

Definition at line 193 of file ParticleLoaderModule.h.

◆ m_loadChargedCluster

bool m_loadChargedCluster
private

Switch to load charged-cluster.

Definition at line 208 of file ParticleLoaderModule.h.

◆ m_logConfig

LogConfig m_logConfig
privateinherited

The log system configuration of the module.

Definition at line 513 of file Module.h.

◆ m_mcparticles

StoreArray<MCParticle> m_mcparticles
private

StoreArray of MCParticles.

Definition at line 191 of file ParticleLoaderModule.h.

◆ m_MCParticles2Plists

std::vector<PList> m_MCParticles2Plists
private

Collection of PLists that will collect Particles created from MCParticles.

Definition at line 216 of file ParticleLoaderModule.h.

◆ m_moduleParamList

ModuleParamList m_moduleParamList
privateinherited

List storing and managing all parameter of the module.

Definition at line 515 of file Module.h.

◆ m_name

std::string m_name
privateinherited

The name of the module, saved as a string (user-modifiable)

Definition at line 507 of file Module.h.

◆ m_package

std::string m_package
privateinherited

Package this module is found in (may be empty).

Definition at line 509 of file Module.h.

◆ m_particleExtraInfoMap

StoreObjPtr<ParticleExtraInfoMap> m_particleExtraInfoMap
private

object pointer to extra info map

Definition at line 197 of file ParticleLoaderModule.h.

◆ m_particles

StoreArray<Particle> m_particles
private

StoreArray of Particles.

Definition at line 190 of file ParticleLoaderModule.h.

◆ m_pidlikelihoods

StoreArray<PIDLikelihood> m_pidlikelihoods
private

StoreArray of PIDLikelihoods.

Definition at line 194 of file ParticleLoaderModule.h.

◆ m_properties

int m_properties
private

Particle property to be assigned only on V0s.

Flags are defined in Particle::PropertyFlags

Definition at line 212 of file ParticleLoaderModule.h.

◆ m_propertyFlags

unsigned int m_propertyFlags
privateinherited

The properties of the module as bitwise or (with |) of EModulePropFlags.

Definition at line 511 of file Module.h.

◆ m_returnValue

int m_returnValue
privateinherited

The return value.

Definition at line 518 of file Module.h.

◆ m_ROE2Plists

std::vector<PList> m_ROE2Plists
private

Collection of PLists that will collect Particles created from V0.

Definition at line 220 of file ParticleLoaderModule.h.

◆ m_roeMaskName

std::string m_roeMaskName
private

ROE mask name to load.

Definition at line 232 of file ParticleLoaderModule.h.

◆ m_roes

StoreArray<RestOfEvent> m_roes
private

StoreArray of ROEs.

Definition at line 198 of file ParticleLoaderModule.h.

◆ m_sameChargeDaughtersV0Counts

std::vector<int> m_sameChargeDaughtersV0Counts
private

internally used to count the number of V0s with same charge daughters

Definition at line 245 of file ParticleLoaderModule.h.

◆ m_skipInitial

bool m_skipInitial
private

toggle skip of initial MC particles

Definition at line 228 of file ParticleLoaderModule.h.

◆ m_skipNonPrimary

bool m_skipNonPrimary
private

toggle skip of secondary MC particle

Definition at line 229 of file ParticleLoaderModule.h.

◆ m_skipNonPrimaryDaughters

bool m_skipNonPrimaryDaughters
private

toggle skip of secondary MC daughters

Definition at line 231 of file ParticleLoaderModule.h.

◆ m_sourceParticleListName

std::string m_sourceParticleListName
private

Particle list name from which we need to get related ROEs.

Definition at line 233 of file ParticleLoaderModule.h.

◆ m_trackfitresults

StoreArray<TrackFitResult> m_trackfitresults
private

StoreArray of TrackFitResults.

Definition at line 196 of file ParticleLoaderModule.h.

◆ m_trackHypothesis

int m_trackHypothesis
private

pdg code for track hypothesis that should be used to create the particle

Definition at line 235 of file ParticleLoaderModule.h.

◆ m_tracks

StoreArray<Track> m_tracks
private

StoreArray of Tracks.

Definition at line 195 of file ParticleLoaderModule.h.

◆ m_Tracks2Plists

std::vector<PList> m_Tracks2Plists
private

Collection of PLists that will collect Particles created from Tracks.

Definition at line 217 of file ParticleLoaderModule.h.

◆ m_type

std::string m_type
privateinherited

The type of the module, saved as a string.

Definition at line 508 of file Module.h.

◆ m_useDummy

bool m_useDummy
private

Switch to load dummy as Particle.

Definition at line 206 of file ParticleLoaderModule.h.

◆ m_useMCParticles

bool m_useMCParticles
private

Load MCParticle as Particle instead of the corresponding MDST dataobject.

Definition at line 202 of file ParticleLoaderModule.h.

◆ m_useMissing

bool m_useMissing
private

Use missing momentum to build a particle.

Definition at line 234 of file ParticleLoaderModule.h.

◆ m_useOnlyMostEnergeticECLCluster

bool m_useOnlyMostEnergeticECLCluster
private

If true, only the most energetic ECLCluster is used.

Definition at line 209 of file ParticleLoaderModule.h.

◆ m_useROEs

bool m_useROEs
private

Switch to load ROE as Particle.

Definition at line 204 of file ParticleLoaderModule.h.

◆ m_V02Plists

std::vector<PList> m_V02Plists
private

Collection of PLists that will collect Particles created from V0.

Definition at line 218 of file ParticleLoaderModule.h.

◆ m_v0s

StoreArray<V0> m_v0s
private

StoreArray of V0s.

Definition at line 199 of file ParticleLoaderModule.h.

◆ m_writeOut

bool m_writeOut
private

toggle particle list btw.

transient/persistent

Definition at line 227 of file ParticleLoaderModule.h.


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