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

Module to write Ntuples for KlId BKG classifier training. More...

#include <DataWriterModule.h>

Inheritance diagram for DataWriterModule:
Collaboration diagram for DataWriterModule:

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

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

Static Public Member Functions

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

Protected Member Functions

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

Private Member Functions

std::list< ModulePtrgetModules () const override
 no submodules, return empty list
 
std::string getPathString () const override
 return the module name.
 
void setParamPython (const std::string &name, const boost::python::object &pyObj)
 Implements a method for setting boost::python objects. More...
 
void setParamPythonDict (const boost::python::dict &dictionary)
 Implements a method for reading the parameter values from a boost::python dictionary. More...
 

Private Attributes

std::string m_outPath = "KlIdBKGClassifierTrainingTuples.root"
 Output path variable.
 
Float_t m_KLMnCluster
 varibales to write out. More...
 
Float_t m_KLMnLayer
 number of layers hit in KLM cluster
 
Float_t m_KLMnInnermostLayer
 number of innermost layers hit
 
Float_t m_KLMglobalZ
 global Z position in KLM

 
Float_t m_KLMtime
 timing of KLM Cluster
 
Float_t m_KLMavInterClusterDist
 average distance between all KLM clusters
 
Float_t m_KLMhitDepth
 hit depth in KLM, distance to IP
 
Float_t m_KLMenergy
 Energy deposit in KLM (0.2 GeV * nHitCells)
 
Float_t m_KLMinvM
 invariant mass calculated from root vector
 
Float_t m_KLMTruth
 target variable for KLM classification
 
Float_t m_KLMnextCluster
 distance to next KLM cluster
 
Float_t m_KLMTrackSepDist
 distance from track separation object

 
Float_t m_KLMTrackSepAngle
 angular distance from track separation object
 
Float_t m_KLMInitialTrackSepAngle
 angular distance from track to cluster at track starting point
 
Float_t m_KLMTrackRotationAngle
 angle between track at poca and trackbeginning
 
Float_t m_KLMTrackClusterSepAngle
 angle between trach momentum and cluster (measured from ip)
 
Float_t m_KLMAngleToMC
 angle between KLMcluster and Mcparticle
 
Float_t m_KLMECLDist
 distance associated ECL <-> KLM cluster
 
Float_t m_KLMECLE
 energy measured in associated ECL cluster
 
Float_t m_KLMECLdeltaL
 distance between track entry point and cluster center, might be removed
 
Float_t m_KLMECLminTrackDist
 track distance between associated ECL cluster and track extrapolated into ECL
 
Float_t m_KLMECLE9oE25
 E in surrounding 9 crystals divided by surrounding 25 crydtalls.
 
Float_t m_KLMECLTiming
 timing of associated ECL cluster
 
Float_t m_KLMECLTerror
 uncertainty on time in associated ECL cluster
 
Float_t m_KLMECLEerror
 uncertainty on E in associated ECL cluster
 
Float_t m_KLMtrackToECL
 primitive distance cluster <-> track for associated ECL cluster
 
Float_t m_KLMKLid
 KlId for that object.
 
Float_t m_KLMMCMom
 momentum of matched mc particle
 
Float_t m_KLMMCPhi
 phi of matched mc particle
 
Float_t m_KLMMCTheta
 theta of matched mc particle
 
Float_t m_KLMMom
 measured momentum
 
Float_t m_KLMPhi
 measured phi

 
Float_t m_KLMTheta
 measured theta
 
Float_t m_KLMMCStatus
 MC particles status.
 
Float_t m_KLMMCLifetime
 MC partilces life time.
 
Float_t m_KLMMCPDG
 pdg code of matched MCparticle
 
Float_t m_KLMMCPrimaryPDG
 pdg code of MCparticles mother, for example pi0 for some gammas
 
Float_t m_KLMECLHypo
 hypotheis id of closest ecl cluster 5: gamma, 6:hadron

 
Float_t m_KLMECLZMVA
 zernike mva output for closest ECL cluster (based on around 10 z-moments)
 
Float_t m_KLMECLZ40
 zernike moment 4,0 of closest ecl cluster
 
Float_t m_KLMECLZ51
 zernike moment 5,1 of closest ECL cluster
 
Float_t m_KLMECLUncertaintyPhi
 phi uncertainty oof closeest ecl cluster
 
Float_t m_KLMECLUncertaintyTheta
 theta uncertainty of closest ECL cluster
 
Float_t m_KLMMCWeight
 mc weight
 
Float_t m_KLMtrackFlag
 track flag for belle comparision
 
Float_t m_KLMeclFlag
 ecl flag for belle comparision
 
Float_t m_ECLE
 measured energy
 
Float_t m_ECLE9oE25
 energy of 9/25 chrystall rings (E dispersion shape)
 
Float_t m_ECLTiming
 timing of ECL
 
Float_t m_ECLR
 distance of cluster to IP
 
Float_t m_ECLEerror
 uncertainty on E measurement in ECL
 
Float_t m_ECLminTrkDistance
 more sophisticated distaqnce to track in ECL
 
Float_t m_ECLdeltaL
 distance between track entrace into cluster and cluster center
 
Float_t m_ECLZ51
 Zernike moment 5,1 see Belle2 note on that.
 
Float_t m_ECLZ40
 Zernike moment 4,0 see Belle2 note on that.
 
Float_t m_ECLE1oE9
 central crystal devided by 3x3 area with it in its center
 
Float_t m_ECL2ndMom
 second moment, shower shape
 
Float_t m_ECLnumChrystals
 number of crystals in the cluster
 
Float_t m_ECLLAT
 lateral shower shape
 
Float_t m_ECLZMVA
 output of a BDT that was fitted on some Zernike Moments on a connected region
 
Float_t m_ECLKLid
 classifier output
 
Float_t m_ECLMCStatus
 mc status, seen in detector etc. More...
 
Float_t m_ECLMCLifetime
 MC particles lifetime.
 
Float_t m_ECLMCPDG
 pdg code of the MCparticle directly related to the cluster
 
Float_t m_ECLMCPrimaryPDG
 pdg code of higher order MC particle, a cluster related to a photon that originates from a pi0 decay get the pi0 code
 
Float_t m_ECLDeltaTime
 KlId for that object.
 
Float_t m_ECLUncertaintyEnergy
 measured energy uncertainty
 
Float_t m_ECLUncertaintyTheta
 measured uncertainty on theta
 
Float_t m_ECLUncertaintyPhi
 measured uncertainty of phi
 
Float_t m_ECLMCMom
 MC particle momentum; -999 if not MCparticle.
 
Float_t m_ECLMCPhi
 MC particle phi; -999 if not MCparticle

 
Float_t m_ECLMCTheta
 MC particle momentum; -999 if not MCparticle.
 
Float_t m_ECLMom
 measured momentum
 
Float_t m_ECLPhi
 measured phi
 
Float_t m_ECLTheta
 measured theta
 
Float_t m_ECLZ
 measured Z-coordinate
 
Float_t m_ECLTruth
 ECL trarget variable.
 
Float_t m_isBeamBKG
 is beam bkg
 
Float_t m_ECLMCWeight
 mc weight
 
Float_t m_isSignal
 isSignal for the classifier
 
StoreArray< MCParticlem_mcParticles
 Store array

 
StoreArray< KLMClusterm_klmClusters
 Store array

 
StoreArray< ECLClusterm_eclClusters
 Store array

 
TFile * m_f = nullptr
 root file
 
TTree * m_treeKLM = nullptr
 tree for klm data
 
TTree * m_treeECLhadron = nullptr
 tree containing ntuples for ECL cluster with N2 (hadron hypothesis)
 
TTree * m_treeECLgamma = nullptr
 tree containing ntuples for ECL cluster with N1 (photon hypothesis)
 
bool m_useKLM
 write out KLM data
 
bool m_useECL
 write out KLM data
 
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

Module to write Ntuples for KlId BKG classifier training.

Writes tuples for both ECL and KLM. You have to specify an outputpath outPath.

Definition at line 29 of file DataWriterModule.h.

Member Enumeration Documentation

◆ EModulePropFlags

enum EModulePropFlags
inherited

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

Enumerator
c_Input 

This module is an input module (reads data).

c_Output 

This module is an output module (writes data).

c_ParallelProcessingCertified 

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

c_HistogramManager 

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

c_InternalSerializer 

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

c_TerminateInAllProcesses 

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

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

c_DontCollectStatistics 

No statistics is collected for this module.

Definition at line 77 of file Module.h.

Member Function Documentation

◆ clone()

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

Create an independent copy of this module.

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

Implements PathElement.

Definition at line 179 of file Module.cc.

◆ def_endRun()

virtual void def_endRun ( )
inlineprotectedvirtualinherited

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

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

Reimplemented in PyModule.

Definition at line 439 of file Module.h.

◆ def_initialize()

virtual void def_initialize ( )
inlineprotectedvirtualinherited

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

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

Reimplemented in PyModule.

Definition at line 420 of file Module.h.

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

◆ getAfterConditionPath()

Module::EAfterConditionPath getAfterConditionPath ( ) const
inherited

What to do after the conditional path is finished.

(defaults to c_End if no condition is set)

Definition at line 133 of file Module.cc.

◆ getConditionPath()

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

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


Definition at line 113 of file Module.cc.

◆ getFileNames()

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

Return a list of output filenames for this modules.

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

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

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

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

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

Reimplemented in RootOutputModule, StorageRootOutputModule, and RootInputModule.

Definition at line 134 of file Module.h.

◆ getName()

const std::string& getName ( ) const
inlineinherited

Returns the name of the module.

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

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

Definition at line 187 of file Module.h.

◆ getParamInfoListPython()

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

Returns a python list of all parameters.

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

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

Definition at line 279 of file Module.cc.

◆ getReturnValue()

int getReturnValue ( ) const
inlineinherited

Return the return value set by this module.

This value is only meaningful if hasReturnValue() is true

Definition at line 381 of file Module.h.

◆ getType()

const std::string & getType ( ) const
inherited

Returns the type of the module (i.e.

class name minus 'Module')

Definition at line 41 of file Module.cc.

◆ hasProperties()

bool hasProperties ( unsigned int  propertyFlags) const
inherited

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

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

Definition at line 160 of file Module.cc.

◆ if_false()

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

A simplified version to add a condition to the module.

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

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

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

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

Definition at line 85 of file Module.cc.

◆ if_true()

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

A simplified version to set the condition of the module.

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

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

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

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

Definition at line 90 of file Module.cc.

◆ if_value()

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

Add a condition to the module.

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

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

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

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

Definition at line 79 of file Module.cc.

◆ setDescription()

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

Sets the description of the module.

Parameters
descriptionA description of the module.

Definition at line 214 of file Module.cc.

◆ setLogInfo()

void setLogInfo ( int  logLevel,
unsigned int  logInfo 
)
inherited

Configure the printed log information for the given level.

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

Definition at line 73 of file Module.cc.

◆ setName()

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

Set the name of the module.

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

Definition at line 214 of file Module.h.

◆ setParamPython()

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

Implements a method for setting boost::python objects.

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

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

Definition at line 234 of file Module.cc.

◆ setParamPythonDict()

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

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

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

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

Definition at line 249 of file Module.cc.

◆ setPropertyFlags()

void setPropertyFlags ( unsigned int  propertyFlags)
inherited

Sets the flags for the module properties.

Parameters
propertyFlagsbitwise OR of EModulePropFlags

Definition at line 208 of file Module.cc.

◆ setReturnValue() [1/2]

void setReturnValue ( bool  value)
protectedinherited

Sets the return value for this module as bool.

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

Parameters
valueThe value of the return value.

Definition at line 227 of file Module.cc.

◆ setReturnValue() [2/2]

void setReturnValue ( int  value)
protectedinherited

Sets the return value for this module as integer.

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

Parameters
valueThe value of the return value.

Definition at line 220 of file Module.cc.

◆ setType()

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

Set the module type.

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

Definition at line 48 of file Module.cc.

Member Data Documentation

◆ m_ECLMCStatus

Float_t m_ECLMCStatus
private

mc status, seen in detector etc.

...

Definition at line 196 of file DataWriterModule.h.

◆ m_KLMnCluster

Float_t m_KLMnCluster
private

varibales to write out.

used for classification of clusters
number of clusters

Definition at line 64 of file DataWriterModule.h.


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