Belle II Software development
ParticleVertexFitterModule Class Reference

Vertex fitter module. More...

#include <ParticleVertexFitterModule.h>

Inheritance diagram for ParticleVertexFitterModule:
Module PathElement

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

 ParticleVertexFitterModule ()
 Constructor.
 
virtual void initialize () override
 Initialize the Module.
 
virtual void beginRun () override
 Called when entering a new run.
 
virtual void event () override
 Event processor.
 
virtual std::vector< std::string > getFileNames (bool outputFiles)
 Return a list of output filenames for this modules.
 
virtual void endRun ()
 This method is called if the current run ends.
 
virtual void terminate ()
 This method is called at the end of the event processing.
 
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 Member Functions

bool doVertexFit (Particle *p)
 Main steering routine.
 
bool doKVertexFit (Particle *p, bool ipProfileConstraint, bool ipTubeConstraint)
 Unconstrained vertex fit using KFit.
 
bool doKMassVertexFit (Particle *p)
 Mass-constrained vertex fit using KFit.
 
bool doKMassPointingVertexFit (Particle *p)
 Mass-constrained vertex fit with additional pointing constraint using KFit.
 
bool doKMassFit (Particle *p)
 Mass fit using KFit.
 
bool doKFourCFit (Particle *p)
 FourC fit using KFit.
 
bool doKMassFourCFit (Particle *p)
 MassFourC fit using KFit.
 
bool doKRecoilMassFit (Particle *p)
 RecoilMass fit using KFit.
 
bool makeKVertexMother (analysis::VertexFitKFit &kv, Particle *p)
 Update mother particle after unconstrained vertex fit using KFit.
 
bool makeKMassVertexMother (analysis::MassVertexFitKFit &kv, Particle *p)
 Update mother particle after mass-constrained vertex fit using KFit.
 
bool makeKMassPointingVertexMother (analysis::MassPointingVertexFitKFit &kv, Particle *p)
 Update mother particle after mass-constrained vertex fit with additional pointing constraint using KFit.
 
bool makeKMassMother (analysis::MassFitKFit &kv, Particle *p)
 Update mother particle after mass fit using KFit.
 
bool makeKFourCMother (analysis::FourCFitKFit &kv, Particle *p)
 Update mother particle after FourC fit using KFit.
 
bool makeMassKFourCMother (analysis::MassFourCFitKFit &kv, Particle *p)
 Update mother particle after MassFourC fit using KFit.
 
bool makeKRecoilMassMother (analysis::RecoilMassKFit &kf, Particle *p)
 Update mother particle after RecoilMass fit using KFit.
 
void updateMapOfTrackAndDaughter (unsigned &l, std::vector< std::vector< unsigned > > &pars, std::vector< unsigned > &pard, std::vector< Particle * > &allparticles, const Particle *daughter)
 update the map of daughter and tracks, find out which tracks belong to each daughter.
 
bool addChildofParticletoKFit (analysis::FourCFitKFit &kv, const Particle *particle)
 Adds given particle's child to the FourCFitKFit.
 
bool addChildofParticletoMassKFit (analysis::MassFourCFitKFit &kf, const Particle *particle, std::vector< unsigned > &particleId)
 Adds given particle's child to the MassFourCFitKFit.
 
void addIPProfileToKFit (analysis::VertexFitKFit &kv)
 Adds IPProfile constraint to the vertex fit using KFit.
 
void addIPTubeToKFit (analysis::VertexFitKFit &kv)
 Adds IPTube constraint to the vertex fit using KFit.
 
bool fillFitParticles (const Particle *mother, std::vector< const Particle * > &fitChildren, std::vector< const Particle * > &twoPhotonChildren)
 Fills valid particle's children (with valid error matrix) in the vector of Particles that will enter the fit.
 
bool fillNotFitParticles (const Particle *mother, std::vector< const Particle * > &notFitChildren, const std::vector< const Particle * > &fitChildren)
 Fills valid particle's children (with valid error matrix) in the vector of Particles that will not enter the fit.
 
bool redoTwoPhotonDaughterMassFit (Particle *postFit, const Particle *preFit, const analysis::VertexFitKFit &kv)
 Combines preFit particle and vertex information from vertex fit kv to create new postFit particle.
 
bool doRaveFit (Particle *mother)
 Fit using Rave.
 
bool allSelectedDaughters (const Particle *mother, const std::vector< const Particle * > &tracksVertex)
 check if all the Daughters (o grand-daughters) are selected for the vertex fit
 
void findConstraintBoost (double cut)
 calculate iptube constraint (quasi cylinder along boost direction) for RAVE fit
 
void smearBeamSpot (double width)
 smear beam spot covariance
 
double getChi2TracksLBoost (const analysis::VertexFitKFit &kv)
 calculate the chi2 using only lboost information of tracks
 
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

StoreObjPtr< ParticleListm_plist
 particle list
 
std::string m_listName
 particle list name
 
double m_confidenceLevel
 required fit confidence level
 
double m_Bfield
 magnetic field from data base
 
std::string m_vertexFitter
 Vertex Fitter name.
 
std::string m_fitType
 type of the kinematic fit
 
std::string m_withConstraint
 additional constraint on vertex
 
std::string m_decayString
 daughter particles selection
 
bool m_updateDaughters
 flag for daughters update
 
DecayDescriptor m_decaydescriptor
 Decay descriptor of decays to look for.
 
bool m_hasCovMatrix = false
 flag for mother covariance matrix (PseudoFitter)
 
ROOT::Math::XYZVector m_BeamSpotCenter
 Beam spot position.
 
TMatrixDSym m_beamSpotCov
 Beam spot covariance matrix.
 
DBObjPtr< BeamSpotm_beamSpotDB
 Beam spot database object.
 
double m_smearing
 smearing width applied to IP tube
 
double m_recoilMass
 recoil mass for constraint
 
std::vector< int > m_massConstraintList
 PDG codes of the particles to be mass constraint (massfourC)
 
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

Vertex fitter module.

Definition at line 50 of file ParticleVertexFitterModule.h.

Member Typedef Documentation

◆ EAfterConditionPath

Forward the EAfterConditionPath definition from the ModuleCondition.

Definition at line 88 of file Module.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

Constructor & Destructor Documentation

◆ ParticleVertexFitterModule()

Constructor.

Definition at line 53 of file ParticleVertexFitterModule.cc.

53 : Module(),
54 m_Bfield(0)
55{
56 // set module description (e.g. insert text)
57 setDescription("Vertex fitter for modular analysis");
59
60 // Add parameters
61 addParam("listName", m_listName, "name of particle list", string(""));
62 addParam("confidenceLevel", m_confidenceLevel,
63 "Confidence level to accept the fit. Particle candidates with "
64 "p-value less than confidenceLevel are removed from the particle "
65 "list. If set to -1, all candidates are kept; if set to 0, "
66 "the candidates failing the fit are removed.",
67 0.001);
68 addParam("vertexFitter", m_vertexFitter, "KFit or Rave", string("KFit"));
69 addParam("fitType", m_fitType, "type of the kinematic fit (vertex, massvertex, mass)", string("vertex"));
70 addParam("withConstraint", m_withConstraint,
71 "additional constraint on vertex: ipprofile, iptube, mother, iptubecut, pointing, btube",
72 string(""));
73 addParam("decayString", m_decayString, "specifies which daughter particles are included in the kinematic fit", string(""));
74 addParam("updateDaughters", m_updateDaughters, "true: update the daughters after the vertex fit", false);
75 addParam("smearing", m_smearing, "smear IP tube width by given length", 0.002);
76 addParam("recoilMass", m_recoilMass, "recoil invariant mass (GeV)", 0.);
77 addParam("massConstraintList", m_massConstraintList,
78 "Type::[int]. List of daughter particles to mass constrain with int = pdg code. (only for MassFourCKFit)", {});
79}
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::string m_withConstraint
additional constraint on vertex
bool m_updateDaughters
flag for daughters update
std::string m_decayString
daughter particles selection
std::vector< int > m_massConstraintList
PDG codes of the particles to be mass constraint (massfourC)
std::string m_vertexFitter
Vertex Fitter name.
double m_recoilMass
recoil mass for constraint
double m_confidenceLevel
required fit confidence level
double m_Bfield
magnetic field from data base
double m_smearing
smearing width applied to IP tube
std::string m_fitType
type of the kinematic fit
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

◆ addChildofParticletoKFit()

bool addChildofParticletoKFit ( analysis::FourCFitKFit & kv,
const Particle * particle )
private

Adds given particle's child to the FourCFitKFit.

Parameters
kvreference to KFit FourCFit object
particlepointer to particle

Definition at line 1408 of file ParticleVertexFitterModule.cc.

1409{
1410 for (unsigned ichild = 0; ichild < particle->getNDaughters(); ichild++) {
1411 const Particle* child = particle->getDaughter(ichild);
1412 if (child->getNDaughters() > 0) addChildofParticletoKFit(kf, child);
1413 else {
1414 if (child->getPValue() < 0) return false; // error matrix not valid
1415
1416 kf.addParticle(child);
1417 }
1418 }
1419 return true;
1420}
bool addChildofParticletoKFit(analysis::FourCFitKFit &kv, const Particle *particle)
Adds given particle's child to the FourCFitKFit.

◆ addChildofParticletoMassKFit()

bool addChildofParticletoMassKFit ( analysis::MassFourCFitKFit & kf,
const Particle * particle,
std::vector< unsigned > & particleId )
private

Adds given particle's child to the MassFourCFitKFit.

Parameters
kfreference to KFit FourCFit object
particlepointer to particle
particleIdvector of daughters track id

Definition at line 1422 of file ParticleVertexFitterModule.cc.

1424{
1425 for (unsigned ichild = 0; ichild < particle->getNDaughters(); ichild++) {
1426 const Particle* child = particle->getDaughter(ichild);
1427 if (child->getNDaughters() > 0) {
1428 bool massconstraint = std::find(m_massConstraintList.begin(), m_massConstraintList.end(),
1429 std::abs(child->getPDGCode())) != m_massConstraintList.end();
1430 std::vector<unsigned> childId;
1431 addChildofParticletoMassKFit(kf, child, childId);
1432 if (massconstraint) kf.addMassConstraint(child->getPDGMass(), childId);
1433 particleId.insert(particleId.end(), childId.begin(), childId.end());
1434 } else {
1435 if (child->getPValue() < 0) return false; // error matrix not valid
1436 kf.addParticle(child);
1437 particleId.push_back(kf.getTrackCount() - 1);
1438 }
1439 }
1440 return true;
1441}
bool addChildofParticletoMassKFit(analysis::MassFourCFitKFit &kf, const Particle *particle, std::vector< unsigned > &particleId)
Adds given particle's child to the MassFourCFitKFit.
enum KFitError::ECode addParticle(const Particle *particle)
Add a particle to the fitter.
Definition KFitBase.cc:59
int getTrackCount(void) const
Get the number of added tracks.
Definition KFitBase.cc:107
enum KFitError::ECode addMassConstraint(const double m, std::vector< unsigned > &childTrackId)
Set an invariant mass of daughter particle for the mass-four-momentum-constraint fit.

◆ addIPProfileToKFit()

void addIPProfileToKFit ( analysis::VertexFitKFit & kv)
private

Adds IPProfile constraint to the vertex fit using KFit.

Definition at line 1443 of file ParticleVertexFitterModule.cc.

1444{
1445 HepPoint3D pos = ROOTToCLHEP::getPoint3D(m_BeamSpotCenter);
1446 CLHEP::HepSymMatrix covMatrix = ROOTToCLHEP::getHepSymMatrix(m_beamSpotCov);
1447
1448 kv.setIpProfile(pos, covMatrix);
1449}
TMatrixDSym m_beamSpotCov
Beam spot covariance matrix.
ROOT::Math::XYZVector m_BeamSpotCenter
Beam spot position.
enum KFitError::ECode setIpProfile(const HepPoint3D &ip, const CLHEP::HepSymMatrix &ipe)
Set an IP-ellipsoid shape for the vertex constraint fit.

◆ addIPTubeToKFit()

void addIPTubeToKFit ( analysis::VertexFitKFit & kv)
private

Adds IPTube constraint to the vertex fit using KFit.

Definition at line 1451 of file ParticleVertexFitterModule.cc.

1452{
1453 CLHEP::HepSymMatrix err(7, 0);
1454
1455 for (int i = 0; i < 3; i++) {
1456 for (int j = 0; j < 3; j++) {
1457 err[i + 4][j + 4] = m_beamSpotCov(i, j);
1458 }
1459 }
1460
1461 PCmsLabTransform T;
1462 ROOT::Math::PxPyPzEVector iptube_mom = T.getBeamFourMomentum();
1463
1465 ROOTToCLHEP::getHepLorentzVector(iptube_mom),
1466 ROOTToCLHEP::getPoint3D(m_BeamSpotCenter),
1467 err,
1468 0.);
1469}
ROOT::Math::PxPyPzEVector getBeamFourMomentum() const
Returns LAB four-momentum of e+e-, i.e.
enum KFitError::ECode setIpTubeProfile(const CLHEP::HepLorentzVector &p, const HepPoint3D &x, const CLHEP::HepSymMatrix &e, const double q)
Set a virtual IP-tube track for the vertex constraint fit.

◆ allSelectedDaughters()

bool allSelectedDaughters ( const Particle * mother,
const std::vector< const Particle * > & tracksVertex )
private

check if all the Daughters (o grand-daughters) are selected for the vertex fit

Definition at line 1384 of file ParticleVertexFitterModule.cc.

1386{
1387 bool isAll = false;
1388 if (mother->getNDaughters() == 0) return false;
1389
1390 int nNotIncluded = mother->getNDaughters();
1391
1392 for (unsigned i = 0; i < mother->getNDaughters(); i++) {
1393 bool dauOk = false;
1394 for (auto& vi : tracksVertex) {
1395 if (vi == mother->getDaughter(i)) {
1396 nNotIncluded = nNotIncluded - 1;
1397 dauOk = true;
1398 }
1399 }
1400 if (!dauOk) {
1401 if (allSelectedDaughters(mother->getDaughter(i), tracksVertex)) nNotIncluded--;
1402 }
1403 }
1404 if (nNotIncluded == 0) isAll = true;
1405 return isAll;
1406}
bool allSelectedDaughters(const Particle *mother, const std::vector< const Particle * > &tracksVertex)
check if all the Daughters (o grand-daughters) are selected for the vertex fit
unsigned getNDaughters(void) const
Returns number of daughter particles.
Definition Particle.h:747
const Particle * getDaughter(unsigned i) const
Returns a pointer to the i-th daughter particle.
Definition Particle.cc:662

◆ beginRun()

void beginRun ( void )
overridevirtual

Called when entering a new run.

Set run dependent things like run header parameters, alignment, etc.

Reimplemented from Module.

Definition at line 107 of file ParticleVertexFitterModule.cc.

108{
109 // TODO: set magnetic field for each run
110 // m_Bfield = BFieldManager::getFieldInTesla(ROOT::Math::XYZVector(0, 0, 0)).Z();
111 // TODO: set IP spot size for each run
112}

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

◆ doKFourCFit()

bool doKFourCFit ( Particle * p)
private

FourC fit using KFit.

Parameters
ppointer to particle
Returns
true for successful fit

Definition at line 726 of file ParticleVertexFitterModule.cc.

727{
728 if (mother->getNDaughters() < 2) return false;
729
730 analysis::FourCFitKFit kf;
732
733 for (unsigned ichild = 0; ichild < mother->getNDaughters(); ichild++) {
734 const Particle* child = mother->getDaughter(ichild);
735
736 if (child->getNDaughters() > 0) {
737 bool err = addChildofParticletoKFit(kf, child);
738 if (!err) return false;
739 } else {
740 if (child->getPValue() < 0) return false; // error matrix not valid
741
742 kf.addParticle(child);
743 }
744 }
745
746 // apply four momentum constraint
747 PCmsLabTransform T;
749
750 int err = kf.doFit();
751
752 if (err != 0) return false;
753
754 bool ok = makeKFourCMother(kf, mother);
755
756 return ok;
757}
bool makeKFourCMother(analysis::FourCFitKFit &kv, Particle *p)
Update mother particle after FourC fit using KFit.
enum KFitError::ECode setFourMomentum(const ROOT::Math::PxPyPzEVector &m)
Set an 4 Momentum for the FourC-constraint fit.
enum KFitError::ECode doFit(void)
Perform a four momentum-constraint fit.
enum KFitError::ECode setMagneticField(const double mf)
Change a magnetic field from the default value KFitConst::kDefaultMagneticField.
Definition KFitBase.cc:93

◆ doKMassFit()

bool doKMassFit ( Particle * p)
private

Mass fit using KFit.

Parameters
ppointer to particle
Returns
true for successful fit

Definition at line 699 of file ParticleVertexFitterModule.cc.

700{
701 if (mother->getNDaughters() < 2) return false;
702
703 analysis::MassFitKFit km;
705
706 for (unsigned ichild = 0; ichild < mother->getNDaughters(); ichild++) {
707 const Particle* child = mother->getDaughter(ichild);
708
709 if (child->getPValue() < 0) return false; // error matrix not valid
710
711 km.addParticle(child);
712 }
713
714 // apply mass constraint
715 km.setInvariantMass(mother->getPDGMass());
716
717 int err = km.doFit();
718
719 if (err != 0) return false;
720
721 bool ok = makeKMassMother(km, mother);
722
723 return ok;
724}
bool makeKMassMother(analysis::MassFitKFit &kv, Particle *p)
Update mother particle after mass fit using KFit.
enum KFitError::ECode doFit(void)
Perform a mass-constraint fit.
enum KFitError::ECode setInvariantMass(const double m)
Set an invariant mass for the mass-constraint fit.

◆ doKMassFourCFit()

bool doKMassFourCFit ( Particle * p)
private

MassFourC fit using KFit.

Parameters
ppointer to particle
Returns
true for successful fit

Definition at line 759 of file ParticleVertexFitterModule.cc.

760{
761 if (mother->getNDaughters() < 2) return false;
762
763 analysis::MassFourCFitKFit kf;
765
766 for (unsigned ichild = 0; ichild < mother->getNDaughters(); ichild++) {
767 const Particle* child = mother->getDaughter(ichild);
768
769 if (child->getNDaughters() > 0) {
770 bool massconstraint = std::find(m_massConstraintList.begin(), m_massConstraintList.end(),
771 std::abs(child->getPDGCode())) != m_massConstraintList.end();
772 std::vector<unsigned> childId;
773 bool err = addChildofParticletoMassKFit(kf, child, childId);
774 if (massconstraint) kf.addMassConstraint(child->getPDGMass(), childId);
775 if (!err) return false;
776 } else {
777 if (child->getPValue() < 0) return false; // error matrix not valid
778 kf.addParticle(child);
779 }
780 }
781
782 // apply four momentum constraint
783 PCmsLabTransform T;
785
786 int err = kf.doFit();
787
788 if (err != 0) return false;
789
790 bool ok = makeMassKFourCMother(kf, mother);
791
792 return ok;
793}
bool makeMassKFourCMother(analysis::MassFourCFitKFit &kv, Particle *p)
Update mother particle after MassFourC fit using KFit.
enum KFitError::ECode setFourMomentum(const ROOT::Math::PxPyPzEVector &m)
Set an 4 Momentum for the mass-four-constraint fit.
enum KFitError::ECode doFit(void)
Perform a mass-four-momentum-constraint fit.

◆ doKMassPointingVertexFit()

bool doKMassPointingVertexFit ( Particle * p)
private

Mass-constrained vertex fit with additional pointing constraint using KFit.

Parameters
ppointer to particle
Returns
true for successful fit

Definition at line 654 of file ParticleVertexFitterModule.cc.

655{
656 if (!(mother->hasExtraInfo("prodVertX") && mother->hasExtraInfo("prodVertY") && mother->hasExtraInfo("prodVertZ"))) {
657 return false;
658 }
659
660 if (mother->getNDaughters() < 2) return false;
661
662 std::vector<const Particle*> fitChildren;
663 std::vector<const Particle*> twoPhotonChildren;
664 bool validChildren = fillFitParticles(mother, fitChildren, twoPhotonChildren);
665
666 if (!validChildren)
667 return false;
668
669 if (twoPhotonChildren.size() > 0) {
670 B2FATAL("[ParticleVertexFitterModule::doKMassPointingVertexFit] MassPointingVertex fit using KFit does not support fit with two-photon daughters (yet).");
671 }
672
673 if (fitChildren.size() < 2) {
674 B2WARNING("[ParticleVertexFitterModule::doKMassPointingVertexFit] Number of particles with valid error matrix entering the vertex fit using KFit is less than 2.");
675 return false;
676 }
677
678 bool ok = false;
679 // Initialise the Fitter
680 analysis::MassPointingVertexFitKFit kmpv;
682
683 for (auto child : fitChildren)
684 kmpv.addParticle(child);
685
686 kmpv.setInvariantMass(mother->getPDGMass());
687 HepPoint3D productionVertex(mother->getExtraInfo("prodVertX"),
688 mother->getExtraInfo("prodVertY"),
689 mother->getExtraInfo("prodVertZ"));
690 kmpv.setProductionVertex(productionVertex);
691 int err = kmpv.doFit();
692 if (err != 0) return false;
693
694 ok = makeKMassPointingVertexMother(kmpv, mother);
695
696 return ok;
697}
bool makeKMassPointingVertexMother(analysis::MassPointingVertexFitKFit &kv, Particle *p)
Update mother particle after mass-constrained vertex fit with additional pointing constraint using KF...
bool fillFitParticles(const Particle *mother, std::vector< const Particle * > &fitChildren, std::vector< const Particle * > &twoPhotonChildren)
Fills valid particle's children (with valid error matrix) in the vector of Particles that will enter ...
enum KFitError::ECode doFit(void)
Perform a mass-vertex-pointing constraint fit.
enum KFitError::ECode setInvariantMass(const double m)
Set an invariant mass for the mass-vertex-pointing constraint fit.
enum KFitError::ECode setProductionVertex(const HepPoint3D &v)
Set the production vertex of the particle.

◆ doKMassVertexFit()

bool doKMassVertexFit ( Particle * p)
private

Mass-constrained vertex fit using KFit.

Parameters
ppointer to particle
Returns
true for successful fit

Definition at line 567 of file ParticleVertexFitterModule.cc.

568{
569 if (mother->getNDaughters() < 2) return false;
570
571 std::vector<const Particle*> fitChildren;
572 std::vector<const Particle*> twoPhotonChildren;
573 bool validChildren = fillFitParticles(mother, fitChildren, twoPhotonChildren);
574
575 if (!validChildren)
576 return false;
577
578 if (twoPhotonChildren.size() > 1) {
579 B2FATAL("[ParticleVertexFitterModule::doKVertexFit] MassVertex fit using KFit does not support fit with multiple particles decaying to two photons like pi0 (yet).");
580 }
581
582 if (fitChildren.size() < 2) {
583 B2WARNING("[ParticleVertexFitterModule::doKVertexFit] Number of particles with valid error matrix entering the vertex fit using KFit is less than 2.");
584 return false;
585 }
586
587 bool ok = false;
588 if (twoPhotonChildren.size() == 0) {
589 // Initialise the Fitter
590 analysis::MassVertexFitKFit kmv;
592
593 if (mother->getV0()) {
594 HepPoint3D V0vertex_heppoint(mother->getV0()->getFittedVertexX(),
595 mother->getV0()->getFittedVertexY(),
596 mother->getV0()->getFittedVertexZ());
597 kmv.setInitialVertex(V0vertex_heppoint);
598 }
599
600 for (auto child : fitChildren)
601 kmv.addParticle(child);
602
603 kmv.setInvariantMass(mother->getPDGMass());
604 int err = kmv.doFit();
605 if (err != 0)
606 return false;
607
608 // in the case daughters do not include particles with two photon daughters like pi0 - this is it (fit done)
609 ok = makeKMassVertexMother(kmv, mother);
610 } else if (twoPhotonChildren.size() == 1) {
611 // there is a daughter reconstructed from two photons so without position information
612 // 1. determine vertex based on all other valid daughters
613 // 2. set position and error matrix of two-photon daughter to previously determined vertex
614 // 3. redo the fit using all particles (including two-photon particle this time)
615
616 analysis::VertexFitKFit kv;
618
619 for (auto child : fitChildren)
620 kv.addParticle(child);
621
622 // Perform vertex fit using only the particles with valid error matrices
623 int err = kv.doFit();
624 if (err != 0)
625 return false;
626
627 const Particle* twoPhotonDaughter = twoPhotonChildren[0];
628 Particle fixedTwoPhotonDaughter(twoPhotonDaughter->get4Vector(), twoPhotonDaughter->getPDGCode());
629 ok = redoTwoPhotonDaughterMassFit(&fixedTwoPhotonDaughter, twoPhotonDaughter, kv);
630 if (!ok)
631 return false;
632
633 // finally perform the fit using all daughter particles
634 analysis::MassVertexFitKFit kmv2;
636
637 for (auto child : fitChildren)
638 kmv2.addParticle(child);
639 kmv2.addParticle(&fixedTwoPhotonDaughter);
640
641 kmv2.setInvariantMass(mother->getPDGMass());
642 err = kmv2.doFit();
643
644 if (err != 0)
645 return false;
646
647 ok = makeKMassVertexMother(kmv2, mother);
648 }
649
650 return ok;
651
652}
bool redoTwoPhotonDaughterMassFit(Particle *postFit, const Particle *preFit, const analysis::VertexFitKFit &kv)
Combines preFit particle and vertex information from vertex fit kv to create new postFit particle.
bool makeKMassVertexMother(analysis::MassVertexFitKFit &kv, Particle *p)
Update mother particle after mass-constrained vertex fit using KFit.
int getPDGCode(void) const
Returns PDG code.
Definition Particle.h:465
ROOT::Math::PxPyPzEVector get4Vector() const
Returns Lorentz vector.
Definition Particle.h:567
enum KFitError::ECode setInitialVertex(const HepPoint3D &v)
Set an initial vertex point for the mass-vertex constraint fit.
enum KFitError::ECode doFit(void)
Perform a mass-vertex-constraint fit.
enum KFitError::ECode setInvariantMass(const double m)
Set an invariant mass for the mass-vertex constraint fit.
enum KFitError::ECode doFit(void)
Perform a vertex-constraint fit.

◆ doKRecoilMassFit()

bool doKRecoilMassFit ( Particle * p)
private

RecoilMass fit using KFit.

Parameters
ppointer to particle
Returns
true for successful fit

Definition at line 795 of file ParticleVertexFitterModule.cc.

796{
797 analysis::RecoilMassKFit kf;
799
800 for (unsigned ichild = 0; ichild < mother->getNDaughters(); ichild++) {
801 const Particle* child = mother->getDaughter(ichild);
802
803 if (child->getPValue() < 0) return false; // error matrix not valid
804
805 kf.addParticle(child);
806 }
807
808 // apply four momentum constraint
809 PCmsLabTransform T;
811
812 // apply recoil mass constraint
814
815 int err = kf.doFit();
816
817 if (err != 0) return false;
818
819 bool ok = makeKRecoilMassMother(kf, mother);
820
821 return ok;
822}
bool makeKRecoilMassMother(analysis::RecoilMassKFit &kf, Particle *p)
Update mother particle after RecoilMass fit using KFit.
enum KFitError::ECode setFourMomentum(const ROOT::Math::PxPyPzEVector &m)
Set a recoil mass .
enum KFitError::ECode setRecoilMass(const double m)
Set an invariant mass for the four momentum-constraint fit.
enum KFitError::ECode doFit(void)
Perform a recoil-mass constraint fit.

◆ doKVertexFit()

bool doKVertexFit ( Particle * p,
bool ipProfileConstraint,
bool ipTubeConstraint )
private

Unconstrained vertex fit using KFit.

Parameters
ppointer to particle
ipProfileConstraintflag for IP profile constraint
ipTubeConstraintflag for IP tube constraint
Returns
true for successful fit

Definition at line 471 of file ParticleVertexFitterModule.cc.

472{
473 if ((mother->getNDaughters() < 2 && !ipTubeConstraint) || mother->getNDaughters() < 1) return false;
474
475 std::vector<const Particle*> fitChildren;
476 std::vector<const Particle*> twoPhotonChildren;
477 bool validChildren = fillFitParticles(mother, fitChildren, twoPhotonChildren);
478
479 if (!validChildren)
480 return false;
481
482 std::vector<const Particle*> notFitChildren;
483 fillNotFitParticles(mother, notFitChildren, fitChildren);
484
485
486 if (twoPhotonChildren.size() > 1) {
487 B2FATAL("[ParticleVertexFitterModule::doKVertexFit] Vertex fit using KFit does not support fit with multiple particles decaying to two photons like pi0 (yet).");
488 }
489
490 if ((fitChildren.size() < 2 && !ipTubeConstraint) || fitChildren.size() < 1) {
491 B2WARNING("[ParticleVertexFitterModule::doKVertexFit] Number of particles with valid error matrix entering the vertex fit using KFit is too low.");
492 return false;
493 }
494
495 // Initialise the Fitter
496 analysis::VertexFitKFit kv;
498
499 if (mother->getV0()) {
500 kv.setInitialVertex(mother->getV0()->getFittedVertexPosition());
501 }
502
503 for (auto& child : fitChildren)
504 kv.addParticle(child);
505
506 if (ipProfileConstraint)
508
509 if (ipTubeConstraint)
510 addIPTubeToKFit(kv);
511
512 // Perform vertex fit using only the particles with valid error matrices
513 int err = kv.doFit();
514 if (err != 0)
515 return false;
516
517 double chi2_track = getChi2TracksLBoost(kv);
518 unsigned track_count = kv.getTrackCount();
519 mother->writeExtraInfo("chiSquared_trackL", chi2_track);
520 mother->writeExtraInfo("kFit_nTracks", track_count);
521
522 bool ok = false;
523 if (twoPhotonChildren.size() == 0)
524 // in the case daughters do not include pi0 - this is it (fit done)
525 ok = makeKVertexMother(kv, mother);
526 else if (twoPhotonChildren.size() == 1) {
527 // there is a daughter reconstructed from two photons so without position information
528 // 1. determine vertex based on all other valid daughters
529 // 2. set position and error matrix of two-photon daughter to previously determined vertex
530 // 3. redo the fit using all particles (including two-photon particle this time)
531
532 const Particle* twoPhotonDaughter = twoPhotonChildren[0];
533 Particle fixedTwoPhotonDaughter(twoPhotonDaughter->get4Vector(), twoPhotonDaughter->getPDGCode());
534 ok = redoTwoPhotonDaughterMassFit(&fixedTwoPhotonDaughter, twoPhotonDaughter, kv);
535 if (!ok)
536 return false;
537
538 // finally perform the fit using all daughter particles
539 analysis::VertexFitKFit kv2;
541
542 for (auto& child : fitChildren)
543 kv2.addParticle(child);
544
545 kv2.addParticle(&fixedTwoPhotonDaughter);
546
547 if (ipProfileConstraint)
549
550 err = kv2.doFit();
551
552 if (err != 0)
553 return false;
554
555 ok = makeKVertexMother(kv2, mother);
556 }
557
558 // update 4-vector using not-fit-particles
559 ROOT::Math::PxPyPzEVector total4Vector(mother->get4Vector());
560 for (auto& child : notFitChildren)
561 total4Vector += child->get4Vector();
562 mother->set4Vector(total4Vector);
563
564 return ok;
565}
void addIPTubeToKFit(analysis::VertexFitKFit &kv)
Adds IPTube constraint to the vertex fit using KFit.
bool makeKVertexMother(analysis::VertexFitKFit &kv, Particle *p)
Update mother particle after unconstrained vertex fit using KFit.
bool fillNotFitParticles(const Particle *mother, std::vector< const Particle * > &notFitChildren, const std::vector< const Particle * > &fitChildren)
Fills valid particle's children (with valid error matrix) in the vector of Particles that will not en...
double getChi2TracksLBoost(const analysis::VertexFitKFit &kv)
calculate the chi2 using only lboost information of tracks
void addIPProfileToKFit(analysis::VertexFitKFit &kv)
Adds IPProfile constraint to the vertex fit using KFit.
enum KFitError::ECode setInitialVertex(const HepPoint3D &v)
Set an initial vertex point for the vertex-vertex constraint fit.

◆ doRaveFit()

bool doRaveFit ( Particle * mother)
private

Fit using Rave.

Parameters
motherpointer to particle
Returns
true for successful fit and update of mother

Definition at line 1229 of file ParticleVertexFitterModule.cc.

1230{
1231 if ((m_decayString.empty() ||
1232 (m_withConstraint == "" && m_fitType != "mass")) && mother->getNDaughters() < 2) return false;
1234 if (m_withConstraint == "ipprofile" || m_withConstraint == "iptube" || m_withConstraint == "mother"
1235 || m_withConstraint == "iptubecut" || m_withConstraint == "btube")
1237
1238 analysis::RaveKinematicVertexFitter rf;
1239 if (m_fitType == "mass") rf.setVertFit(false);
1240
1241 if (m_decayString.empty()) {
1242 rf.addMother(mother);
1243 } else {
1244 std::vector<const Particle*> tracksVertex = m_decaydescriptor.getSelectionParticles(mother);
1245 std::vector<std::string> tracksName = m_decaydescriptor.getSelectionNames();
1246
1247 if (allSelectedDaughters(mother, tracksVertex)) {
1248 for (auto& itrack : tracksVertex) {
1249 if (itrack != mother) rf.addTrack(itrack);
1250 }
1251 rf.setMother(mother);
1252 } else {
1253
1254 analysis::RaveKinematicVertexFitter rsf;
1255 bool mothSel = false;
1256 int nTrk = 0;
1257 for (unsigned itrack = 0; itrack < tracksVertex.size(); itrack++) {
1258 if (tracksVertex[itrack] != mother) {
1259 rsf.addTrack(tracksVertex[itrack]);
1260 B2DEBUG(1, "ParticleVertexFitterModule: Adding particle " << tracksName[itrack] << " to vertex fit ");
1261 nTrk++;
1262 }
1263 if (tracksVertex[itrack] == mother) mothSel = true;
1264 }
1265
1266
1267 // Fit one particle constrained to originate from the beam spot
1268 bool mothIPfit = false;
1269 if (tracksVertex.size() == 1 && mothSel == true && m_withConstraint != "" && nTrk == 0) {
1270 rsf.addTrack(tracksVertex[0]);
1271 if (tracksVertex[0] != mother)
1272 B2FATAL("ParticleVertexFitterModule: FATAL Error in IP constrained mother fit");
1273 nTrk++;
1274 mothIPfit = true;
1275 }
1276
1277
1278 ROOT::Math::XYZVector pos;
1279 TMatrixDSym RerrMatrix(3);
1280 int nvert = 0;
1281
1282 // one track fit is not kinematic
1283 if (nTrk == 1) {
1284 analysis::RaveVertexFitter rsg;
1285 for (auto& itrack : tracksVertex) {
1286 rsg.addTrack(itrack);
1287 nvert = rsg.fit("kalman");
1288 if (nvert > 0) {
1289 pos = rsg.getPos(0);
1290 RerrMatrix = rsg.getCov(0);
1291 double prob = rsg.getPValue(0);
1292 ROOT::Math::PxPyPzEVector mom(mother->get4Vector());
1293 TMatrixDSym errMatrix(7);
1294 for (int i = 0; i < 7; i++) {
1295 for (int j = 0; j < 7; j++) {
1296 if (i > 3 && j > 3) {errMatrix[i][j] = RerrMatrix[i - 4][j - 4];}
1297 else {errMatrix[i][j] = 0;}
1298 }
1299 }
1300 if (mothIPfit) {
1301 mother->writeExtraInfo("prodVertX", pos.X());
1302 mother->writeExtraInfo("prodVertY", pos.Y());
1303 mother->writeExtraInfo("prodVertZ", pos.Z());
1304 mother->writeExtraInfo("prodVertSxx", RerrMatrix[0][0]);
1305 mother->writeExtraInfo("prodVertSxy", RerrMatrix[0][1]);
1306 mother->writeExtraInfo("prodVertSxz", RerrMatrix[0][2]);
1307 mother->writeExtraInfo("prodVertSyx", RerrMatrix[1][0]);
1308 mother->writeExtraInfo("prodVertSyy", RerrMatrix[1][1]);
1309 mother->writeExtraInfo("prodVertSyz", RerrMatrix[1][2]);
1310 mother->writeExtraInfo("prodVertSzx", RerrMatrix[2][0]);
1311 mother->writeExtraInfo("prodVertSzy", RerrMatrix[2][1]);
1312 mother->writeExtraInfo("prodVertSzz", RerrMatrix[2][2]);
1313 } else {
1314 mother->updateMomentum(mom, pos, errMatrix, prob);
1315 }
1316 return true;
1317 } else {return false;}
1318 }
1319 } else {
1320 nvert = rsf.fit();
1321 }
1322
1323 if (nvert > 0) {
1324 pos = rsf.getPos();
1325 RerrMatrix = rsf.getVertexErrorMatrix();
1326 double prob = rsf.getPValue();
1327 ROOT::Math::PxPyPzEVector mom(mother->get4Vector());
1328 TMatrixDSym errMatrix(7);
1329 for (int i = 0; i < 7; i++) {
1330 for (int j = 0; j < 7; j++) {
1331 if (i > 3 && j > 3) {errMatrix[i][j] = RerrMatrix[i - 4][j - 4];}
1332 else {errMatrix[i][j] = 0;}
1333 }
1334 }
1335 mother->updateMomentum(mom, pos, errMatrix, prob);
1336 } else {return false;}
1337
1338
1339 if (mothSel && nTrk > 1) {
1341 rf.addMother(mother);
1342 int nKfit = rf.fit();
1343 rf.updateMother();
1345
1346 if (nKfit > 0) {return true;}
1347 else return false;
1348 } else return true;
1349 }
1350 }
1351
1352 bool okFT = false;
1353 if (m_fitType == "vertex") {
1354 okFT = true;
1355 int nVert = rf.fit();
1356 rf.updateMother();
1357 if (m_decayString.empty() && m_updateDaughters == true) rf.updateDaughters();
1358 if (nVert != 1) return false;
1359 }
1360 if (m_fitType == "mass") {
1361 // add protection
1362 okFT = true;
1363 rf.setMassConstFit(true);
1364 rf.setVertFit(false);
1365 int nVert = rf.fit();
1366 rf.updateMother();
1367 if (nVert != 1) return false;
1368 };
1369 if (m_fitType == "massvertex") {
1370 okFT = true;
1371 rf.setMassConstFit(true);
1372 int nVert = rf.fit();
1373 rf.updateMother();
1374 if (m_decayString.empty() && m_updateDaughters == true) rf.updateDaughters();
1375 if (nVert != 1) return false;
1376 };
1377 if (!okFT) {
1378 B2FATAL("fitType : " << m_fitType << " ***invalid fit type ");
1379 }
1380
1381 return true;
1382}
DecayDescriptor m_decaydescriptor
Decay descriptor of decays to look for.
void writeExtraInfo(const std::string &name, const double value)
Sets the user defined extraInfo.
Definition Particle.cc:1393
void updateMomentum(const ROOT::Math::PxPyPzEVector &p4, const ROOT::Math::XYZVector &vertex, const TMatrixFSym &errMatrix, double pValue)
Sets Lorentz vector, position, 7x7 error matrix and p-value.
Definition Particle.h:397
void addMother(const Particle *aMotherParticlePtr)
All daughters of the argument of this function will be used as input for the vertex fit.
ROOT::Math::XYZVector getPos()
get the position of the fitted vertex.
void addTrack(const Particle *aParticlePtr)
add a track (in the format of a Belle2::Particle) to set of tracks that should be fitted to a vertex
void setMother(const Particle *aMotherParticlePtr)
Set Mother particle for Vertex/momentum update.
int fit()
do the kinematic vertex fit with all tracks previously added with the addTrack or addMother function.
void setVertFit(bool isVertFit=true)
Set vertex fit: set false in case of mass fit only.
void setMassConstFit(bool isConstFit=true)
Set mass constrained fit.
double getPValue()
get the p value of the fitted vertex.
TMatrixDSym getVertexErrorMatrix()
get the covariance matrix (3x3) of the of the fitted vertex position.
void updateDaughters()
update the Daughters particles
void unsetBeamSpot()
unset beam spot constraint
Definition RaveSetup.cc:82
static RaveSetup * getInstance()
get the pointer to the instance to get/set any of options stored in RaveSetup
Definition RaveSetup.h:43
void setBeamSpot(const ROOT::Math::XYZVector &beamSpot, const TMatrixDSym &beamSpotCov)
The beam spot position and covariance is known you can set it here so that and a vertex in the beam s...
Definition RaveSetup.cc:74
TMatrixDSym getCov(VecSize vertexId=0) const
get the covariance matrix (3x3) of the of the fitted vertex position.
int fit(std::string options="default")
do the vertex fit with all tracks previously added with the addTrack or addMother function.
ROOT::Math::XYZVector getPos(VecSize vertexId=0) const
get the position of the fitted vertex.
void addTrack(const Particle *const aParticlePtr)
add a track (in the format of a Belle2::Particle) to set of tracks that should be fitted to a vertex
double getPValue(VecSize vertexId=0) const
get the p value of the fitted vertex.

◆ doVertexFit()

bool doVertexFit ( Particle * p)
private

Main steering routine.

Parameters
ppointer to particle
Returns
true for successful fit and prob(chi^2,ndf) > m_confidenceLevel

Definition at line 193 of file ParticleVertexFitterModule.cc.

194{
195 // steering starts here
196
197 if (m_Bfield == 0) {
198 B2FATAL("ParticleVertexFitter: No magnetic field");
199 }
200
201 if (m_withConstraint != "ipprofile" &&
202 m_withConstraint != "iptube" &&
203 m_withConstraint != "mother" &&
204 m_withConstraint != "iptubecut" &&
205 m_withConstraint != "pointing" &&
206 m_withConstraint != "btube" &&
207 m_withConstraint != "")
208 B2FATAL("ParticleVertexFitter: " << m_withConstraint << " ***invalid Constraint ");
209
210 bool ok = false;
211 // fits with KFit
212 if (m_vertexFitter == "KFit") {
213
214 if (m_decayString != "" and m_fitType != "vertex")
215 B2FATAL("ParticleVertexFitter: KFit does not support yet selection of daughters via decay string except for vertex fit!");
216
217 // vertex fit
218 if (m_fitType == "vertex") {
219 if (m_withConstraint == "ipprofile") {
220 ok = doKVertexFit(mother, true, false);
221 } else if (m_withConstraint == "iptube") {
222 ok = doKVertexFit(mother, false, true);
223 } else {
224 ok = doKVertexFit(mother, false, false);
225 }
226 }
227
228 // mass-constrained vertex fit
229 if (m_fitType == "massvertex") {
230 if (m_withConstraint == "ipprofile" || m_withConstraint == "iptube" || m_withConstraint == "iptubecut") {
231 B2FATAL("ParticleVertexFitter: Invalid options - mass-constrained fit using KFit does not work with iptube or ipprofile constraint.");
232 } else if (m_withConstraint == "pointing") {
233 ok = doKMassPointingVertexFit(mother);
234 } else {
235 ok = doKMassVertexFit(mother);
236 }
237 }
238
239 // mass fit
240 if (m_fitType == "mass") {
241 if (m_withConstraint == "ipprofile" || m_withConstraint == "iptube" || m_withConstraint == "iptubecut") {
242 B2FATAL("ParticleVertexFitter: Invalid options - mass fit using KFit does not work with iptube or ipprofile constraint.");
243 } else {
244 ok = doKMassFit(mother);
245 }
246 }
247
248 // four C fit
249 if (m_fitType == "fourC") {
250 if (m_withConstraint == "ipprofile" || m_withConstraint == "iptube" || m_withConstraint == "iptubecut") {
251 B2FATAL("ParticleVertexFitter: Invalid options - four C fit using KFit does not work with iptube or ipprofile constraint.");
252 } else {
253 ok = doKFourCFit(mother);
254 }
255 }
256
257 // four mass C fit
258 if (m_fitType == "massfourC") {
259 if (m_withConstraint == "ipprofile" || m_withConstraint == "iptube" || m_withConstraint == "iptubecut") {
260 B2FATAL("ParticleVertexFitter: Invalid options - four C fit using KFit does not work with iptube or ipprofile constraint.");
261 } else {
262 ok = doKMassFourCFit(mother);
263 }
264 }
265
266 // recoil mass C fit
267 if (m_fitType == "recoilmass") {
268 if (m_withConstraint == "ipprofile" || m_withConstraint == "iptube" || m_withConstraint == "iptubecut") {
269 B2FATAL("ParticleVertexFitter: Invalid options - recoil mass fit using KFit does not work with iptube or ipprofile constraint.");
270 } else {
271 ok = doKRecoilMassFit(mother);
272 }
273 }
274
275 // invalid KFit fit type
276 if (m_fitType != "vertex"
277 && m_fitType != "massvertex"
278 && m_fitType != "mass"
279 && m_fitType != "fourC"
280 && m_fitType != "massfourC"
281 && m_fitType != "recoilmass")
282 B2FATAL("ParticleVertexFitter: " << m_fitType << " ***invalid fit type for the vertex fitter ");
283 }
284
285 // fits using Rave
286 if (m_vertexFitter == "Rave") {
287 try {
288 ok = doRaveFit(mother);
289 } catch (const rave::CheckedFloatException&) {
290 B2ERROR("Invalid inputs (nan/inf)?");
291 ok = false;
292 }
293 }
294
295 // invalid fitter
296 if (m_vertexFitter != "KFit" && m_vertexFitter != "Rave")
297 B2FATAL("ParticleVertexFitter: " << m_vertexFitter << " ***invalid vertex fitter ");
298
299 if (!ok) return false;
300
301 // steering ends here
302
303 //if (mother->getPValue() < m_confidenceLevel) return false;
304 return true;
305
306}
bool doKMassPointingVertexFit(Particle *p)
Mass-constrained vertex fit with additional pointing constraint using KFit.
bool doKVertexFit(Particle *p, bool ipProfileConstraint, bool ipTubeConstraint)
Unconstrained vertex fit using KFit.
bool doKMassVertexFit(Particle *p)
Mass-constrained vertex fit using KFit.
bool doKFourCFit(Particle *p)
FourC fit using KFit.
bool doKRecoilMassFit(Particle *p)
RecoilMass fit using KFit.
bool doRaveFit(Particle *mother)
Fit using Rave.
bool doKMassFit(Particle *p)
Mass fit using KFit.
bool doKMassFourCFit(Particle *p)
MassFourC fit using KFit.

◆ endRun()

virtual void endRun ( void )
inlinevirtualinherited

This method is called if the current run ends.

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

This method can be implemented by subclasses.

Reimplemented in AlignDQMModule, AnalysisPhase1StudyModule, arichBtestModule, ARICHDQMModule, AWESOMEBasicModule, B2BIIConvertMdstModule, B2BIIMCParticlesMonitorModule, B2BIIMdstInputModule, BeamabortModule, BeamabortStudyModule, BeamDigitizerModule, BeamBkgGeneratorModule, BeamBkgHitRateMonitorModule, BeamBkgMixerModule, BeamBkgTagSetterModule, BelleMCOutputModule, BgoDigitizerModule, BgoModule, BgoStudyModule, BGOverlayInputModule, BKLMAnaModule, BKLMDigitAnalyzerModule, BKLMSimHistogrammerModule, BKLMTrackingModule, CalibrationCollectorModule, CaveModule, CDCCosmicAnalysisModule, CDCCRTestModule, CDCPackerModule, CDCRecoTrackFilterModule, CDCUnpackerModule, CDCDedxDQMModule, CDCDedxValidationModule, cdcDQM7Module, CDCDQMModule, CDCTriggerNDFinderModule, CDCTriggerNeuroDQMModule, CDCTriggerNeuroDQMOnlineModule, CertifyParallelModule, ClawDigitizerModule, ClawModule, ClawStudyModule, ClawsDigitizerModule, CLAWSModule, ClawsStudyModule, Convert2RawDetModule, CosmicsAlignmentValidationModule, CsiDigitizer_v2Module, CsIDigitizerModule, CsiModule, CsiStudy_v2Module, CsIStudyModule, CurlTaggerModule, DAQPerfModule, DataWriterModule, DeSerializerPXDModule, DosiDigitizerModule, DosiModule, DosiStudyModule, DQMHistAnalysisARICHModule, DQMHistAnalysisARICHMonObjModule, DQMHistAnalysisCDCDedxModule, DQMHistAnalysisCDCEpicsModule, DQMHistAnalysisCDCMonObjModule, DQMHistAnalysisDAQMonObjModule, DQMHistAnalysisDeltaEpicsMonObjExampleModule, DQMHistAnalysisDeltaTestModule, DQMHistAnalysisECLConnectedRegionsModule, DQMHistAnalysisECLModule, DQMHistAnalysisECLOutOfTimeDigitsModule, DQMHistAnalysisECLShapersModule, DQMHistAnalysisECLSummaryModule, DQMHistAnalysisEpicsExampleModule, DQMHistAnalysisEpicsOutputModule, DQMHistAnalysisEventT0TriggerJitterModule, DQMHistAnalysisExampleFlagsModule, DQMHistAnalysisExampleModule, DQMHistAnalysisHLTMonObjModule, DQMHistAnalysisInput2Module, DQMHistAnalysisInputPVSrvModule, DQMHistAnalysisInputTestModule, DQMHistAnalysisKLM2Module, DQMHistAnalysisKLMModule, DQMHistAnalysisMiraBelleModule, DQMHistAnalysisMonObjModule, DQMHistAnalysisOutputFileModule, DQMHistAnalysisOutputMonObjModule, DQMHistAnalysisOutputRelayMsgModule, DQMHistAnalysisPhysicsModule, DQMHistAnalysisPXDChargeModule, DQMHistAnalysisPXDFitsModule, DQMHistAnalysisPXDTrackChargeModule, DQMHistAnalysisRooFitExampleModule, DQMHistAnalysisSVDClustersOnTrackModule, DQMHistAnalysisSVDDoseModule, DQMHistAnalysisSVDEfficiencyModule, DQMHistAnalysisSVDGeneralModule, DQMHistAnalysisSVDOccupancyModule, DQMHistAnalysisSVDOnMiraBelleModule, DQMHistAnalysisSVDUnpackerModule, DQMHistAnalysisTOPModule, DQMHistAnalysisTRGECLModule, DQMHistAnalysisTRGEFFModule, DQMHistAnalysisTRGGDLModule, DQMHistAnalysisTRGModule, DQMHistComparitorModule, DQMHistDeltaHistoModule, DqmHistoManagerModule, DQMHistOutputToEPICSModule, DQMHistReferenceModule, DQMHistSnapshotsModule, Ds2RawFileModule, Ds2RawModule, Ds2RbufModule, Ds2SampleModule, ECLLOMModule, ECLBackgroundModule, ECLChargedPIDDataAnalysisModule, ECLChargedPIDDataAnalysisValidationModule, ECLChargedPIDModule, ECLClusterPSDModule, ECLCovarianceMatrixModule, ECLCRFinderModule, ECLDataAnalysisModule, ECLDigitCalibratorModule, ECLDigitizerModule, ECLDigitizerPureCsIModule, EclDisplayModule, ECLDQMEXTENDEDModule, ECLDQMModule, ECLFinalizerModule, ECLHitDebugModule, ECLLocalMaximumFinderModule, ECLLocalRunCalibratorModule, ECLPackerModule, ECLShowerCorrectorModule, ECLShowerShapeModule, ECLSplitterN1Module, ECLSplitterN2Module, ECLUnpackerModule, ECLWaveformFitModule, EffPlotsModule, EKLMDataCheckerModule, ElapsedTimeModule, EventInfoPrinterModule, EventT0ValidationModule, EvReductionModule, EvtGenDecayModule, ExtModule, FANGSDigitizerModule, FANGSModule, FANGSStudyModule, FastRbuf2DsModule, FullSimModule, TRGGDLUnpackerModule, GenfitVisModule, GenRawSendModule, GetEventFromSocketModule, He3DigitizerModule, He3tubeModule, He3tubeStudyModule, HistoManagerModule, HistoModule, HitXPModule, HLTDQM2ZMQModule, HLTDs2ZMQModule, KLMClusterEfficiencyModule, KLMClustersReconstructorModule, KLMDigitizerModule, KLMDQM2Module, KLMDQMModule, KLMMuonIDDNNExpertModule, KLMPackerModule, KLMReconstructorModule, KLMScintillatorSimulatorModule, KLMTrackingModule, KLMTriggerModule, KLMUnpackerModule, KlongValidationModule, LowEnergyPi0IdentificationExpertModule, LowEnergyPi0VetoExpertModule, MCMatcherTRGECLModule, MCTrackCandClassifierModule, MCV0MatcherModule, MicrotpcModule, MicrotpcStudyModule, TpcDigitizerModule, TPCStudyModule, MonitorDataModule, MuidModule, NoKickCutsEvalModule, NtuplePhase1_v6Module, OverrideGenerationFlagsModule, PartialSeqRootReaderModule, Ph1bpipeModule, Ph1sustrModule, PhysicsObjectsDQMModule, PhysicsObjectsMiraBelleBhabhaModule, PhysicsObjectsMiraBelleDst2Module, PhysicsObjectsMiraBelleDstModule, PhysicsObjectsMiraBelleHadronModule, PhysicsObjectsMiraBelleModule, PinDigitizerModule, PindiodeModule, PindiodeStudyModule, PlumeDigitizerModule, PlumeModule, PrintDataModule, PrintEventRateModule, PXDBackgroundModule, PXDClustersFromTracksModule, PXDPerformanceModule, PXDROIFinderModule, PyModule, QcsmonitorDigitizerModule, QcsmonitorModule, QcsmonitorStudyModule, RandomBarrierModule, Raw2DsModule, RawInputModule, Rbuf2DsModule, Rbuf2RbufModule, ReceiveEventModule, ReprocessorModule, Root2BinaryModule, Root2RawModule, RT2SPTCConverterModule, RxModule, RxSocketModule, SecMapTrainerBaseModule, SecMapTrainerVXDTFModule, SectorMapBootstrapModule, SeqRootInputModule, SeqRootMergerModule, SeqRootOutputModule, SerializerModule, SPTCmomentumSeedRetrieverModule, SPTCvirtualIPRemoverModule, SrsensorModule, StatisticsSummaryModule, StorageDeserializerModule, StorageRootOutputModule, StorageSerializerModule, SubEventModule, SVD3SamplesEmulatorModule, SVDBackgroundModule, SVDClusterizerModule, SVDPackerModule, SVDRecoDigitCreatorModule, SVDUnpackerModule, SVDB4CommissioningPlotsModule, SVDClusterCalibrationsMonitorModule, SVDClusterEvaluationModule, SVDClusterEvaluationTrueInfoModule, SVDClusterFilterModule, SVDCoGTimeEstimatorModule, SVDDataFormatCheckModule, svdDumpModule, SVDHotStripFinderModule, SVDLatencyCalibrationModule, SVDLocalCalibrationsMonitorModule, SVDOccupancyAnalysisModule, SVDPerformanceModule, SVDPositionErrorScaleFactorImporterModule, SVDROIDQMModule, SVDROIFinderAnalysisModule, SVDROIFinderModule, SVDShaperDigitsFromTracksModule, SVDTimeCalibrationsMonitorModule, SVDTriggerQualityGeneratorModule, SVDUnpackerDQMModule, SwitchDataStoreModule, TOPBackgroundModule, TOPChannelT0MCModule, TOPDoublePulseGeneratorModule, TOPGainEfficiencyCalculatorModule, TOPInterimFENtupleModule, TOPLaserCalibratorModule, TOPLaserHitSelectorModule, TOPMCTrackMakerModule, TOPNtupleModule, TOPPackerModule, TOPRawDigitConverterModule, TOPTBCComparatorModule, TOPTimeBaseCalibratorModule, TOPTriggerDigitizerModule, TOPUnpackerModule, TOPWaveformFeatureExtractorModule, TOPWaveformQualityPlotterModule, TOPXTalkChargeShareSetterModule, TrackAnaModule, TrackFinderMCTruthRecoTracksModule, FindletModule< AFindlet >, FindletModule< AsicBackgroundLibraryCreator >, FindletModule< AxialSegmentPairCreator >, FindletModule< AxialSegmentPairCreator >, FindletModule< AxialStraightTrackFinder >, FindletModule< AxialStraightTrackFinder >, FindletModule< AxialTrackCreatorMCTruth >, FindletModule< AxialTrackCreatorMCTruth >, FindletModule< AxialTrackCreatorSegmentHough >, FindletModule< AxialTrackCreatorSegmentHough >, FindletModule< AxialTrackFinderHough >, FindletModule< AxialTrackFinderHough >, FindletModule< AxialTrackFinderLegendre >, FindletModule< AxialTrackFinderLegendre >, FindletModule< CDCTrackingEventLevelMdstInfoFillerFromHitsFindlet >, FindletModule< CDCTrackingEventLevelMdstInfoFillerFromSegmentsFindlet >, FindletModule< CKFToCDCFindlet >, FindletModule< CKFToCDCFromEclFindlet >, FindletModule< CKFToPXDFindlet >, FindletModule< CKFToSVDFindlet >, FindletModule< CKFToSVDSeedFindlet >, FindletModule< ClusterBackgroundDetector >, FindletModule< ClusterBackgroundDetector >, FindletModule< ClusterPreparer >, FindletModule< ClusterPreparer >, FindletModule< ClusterRefiner< BridgingWireHitRelationFilter > >, FindletModule< ClusterRefiner< BridgingWireHitRelationFilter > >, FindletModule< CosmicsTrackMergerFindlet >, FindletModule< DATCONFPGAFindlet >, FindletModule< FacetCreator >, FindletModule< FacetCreator >, FindletModule< HitBasedT0Extractor >, FindletModule< HitBasedT0Extractor >, FindletModule< HitReclaimer >, FindletModule< HitReclaimer >, FindletModule< MCVXDCDCTrackMergerFindlet >, FindletModule< MonopoleAxialTrackFinderLegendre >, FindletModule< MonopoleAxialTrackFinderLegendre >, FindletModule< MonopoleStereoHitFinder >, FindletModule< MonopoleStereoHitFinder >, FindletModule< MonopoleStereoHitFinderQuadratic >, FindletModule< MonopoleStereoHitFinderQuadratic >, FindletModule< SegmentCreatorFacetAutomaton >, FindletModule< SegmentCreatorFacetAutomaton >, FindletModule< SegmentCreatorMCTruth >, FindletModule< SegmentCreatorMCTruth >, FindletModule< SegmentFinderFacetAutomaton >, FindletModule< SegmentFinderFacetAutomaton >, FindletModule< SegmentFitter >, FindletModule< SegmentFitter >, FindletModule< SegmentLinker >, FindletModule< SegmentLinker >, FindletModule< SegmentOrienter >, FindletModule< SegmentOrienter >, FindletModule< SegmentPairCreator >, FindletModule< SegmentPairCreator >, FindletModule< SegmentRejecter >, FindletModule< SegmentRejecter >, FindletModule< SegmentTrackCombiner >, FindletModule< SegmentTrackCombiner >, FindletModule< SegmentTripleCreator >, FindletModule< SegmentTripleCreator >, FindletModule< StereoHitFinder >, FindletModule< StereoHitFinder >, FindletModule< SuperClusterCreator >, FindletModule< SuperClusterCreator >, FindletModule< TrackCombiner >, FindletModule< TrackCombiner >, FindletModule< TrackCreatorSegmentPairAutomaton >, FindletModule< TrackCreatorSegmentPairAutomaton >, FindletModule< TrackCreatorSegmentTripleAutomaton >, FindletModule< TrackCreatorSegmentTripleAutomaton >, FindletModule< TrackCreatorSingleSegments >, FindletModule< TrackCreatorSingleSegments >, FindletModule< TrackExporter >, FindletModule< TrackExporter >, FindletModule< TrackFinder >, FindletModule< TrackFinderAutomaton >, FindletModule< TrackFinderCosmics >, FindletModule< TrackFinderSegmentPairAutomaton >, FindletModule< TrackFinderSegmentPairAutomaton >, FindletModule< TrackFinderSegmentTripleAutomaton >, FindletModule< TrackFinderSegmentTripleAutomaton >, FindletModule< TrackFindingCDC::FindletStoreArrayInput< BaseEventTimeExtractorModuleFindlet< AFindlet > > >, FindletModule< TrackFindingCDC::FindletStoreArrayInput< BaseEventTimeExtractorModuleFindlet< AFindlet > > >, FindletModule< TrackFlightTimeAdjuster >, FindletModule< TrackFlightTimeAdjuster >, FindletModule< TrackLinker >, FindletModule< TrackLinker >, FindletModule< TrackOrienter >, FindletModule< TrackOrienter >, FindletModule< TrackQualityAsserter >, FindletModule< TrackQualityAsserter >, FindletModule< TrackQualityEstimator >, FindletModule< TrackQualityEstimator >, FindletModule< TrackRejecter >, FindletModule< TrackRejecter >, FindletModule< vxdHoughTracking::SVDHoughTracking >, FindletModule< WireHitBackgroundDetector >, FindletModule< WireHitBackgroundDetector >, FindletModule< WireHitCreator >, FindletModule< WireHitCreator >, FindletModule< WireHitPreparer >, FindletModule< WireHitPreparer >, TrackFinderVXDAnalizerModule, TrackingPerformanceEvaluationModule, TrackSetEvaluatorHopfieldNNDEVModule, TRGCDCETFUnpackerModule, TRGCDCModule, TRGCDCT2DDQMModule, TRGCDCT3DConverterModule, TRGCDCT3DDQMModule, TRGCDCT3DUnpackerModule, TRGCDCTSFUnpackerModule, TRGCDCTSFDQMModule, TRGCDCTSStreamModule, TRGECLBGTCHitModule, TRGECLDQMModule, TRGECLFAMModule, TRGECLModule, TRGECLQAMModule, TRGECLRawdataAnalysisModule, TRGECLTimingCalModule, TRGECLUnpackerModule, TRGEFFDQMModule, TRGGDLDQMModule, TRGGDLDSTModule, TRGGDLModule, TRGGDLSummaryModule, TRGGRLDQMModule, TRGGRLMatchModule, TRGGRLModule, TRGGRLProjectsModule, TRGGRLUnpackerModule, TRGRAWDATAModule, TRGTOPDQMModule, TRGTOPTRD2TTSConverterModule, TRGTOPUnpackerModule, TRGTOPUnpackerWaveformModule, TRGTOPWaveformPlotterModule, TxModule, TxSocketModule, V0findingPerformanceEvaluationModule, vxdDigitMaskingModule, VXDSimpleClusterizerModule, VXDTFTrainingDataCollectorModule, ZMQTxInputModule, and ZMQTxWorkerModule.

Definition at line 165 of file Module.h.

165{};

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

Event processor.

Reimplemented from Module.

Definition at line 114 of file ParticleVertexFitterModule.cc.

115{
116 if (m_vertexFitter == "Rave")
118
119 m_BeamSpotCenter = m_beamSpotDB->getIPPosition();
120 m_beamSpotCov.ResizeTo(3, 3);
121 TMatrixDSym beamSpotCov(3);
122 if (m_withConstraint == "ipprofile") m_beamSpotCov = m_beamSpotDB->getCovVertex();
123 if (m_withConstraint == "iptube") {
124 if (m_smearing > 0 && m_vertexFitter == "KFit") {
126 } else {
128 }
129 }
130 if (m_withConstraint == "iptubecut") { // for development purpose only
131 m_BeamSpotCenter = ROOT::Math::XYZVector(0.001, 0., .013);
133 }
134 if ((m_vertexFitter == "Rave") && (m_withConstraint == "ipprofile" || m_withConstraint == "iptube"
135 || m_withConstraint == "mother" || m_withConstraint == "iptubecut" || m_withConstraint == "btube"))
137
138 std::vector<unsigned int> toRemove;
139 unsigned int nParticles = m_plist->getListSize();
140
141 for (unsigned iPart = 0; iPart < nParticles; iPart++) {
142 Particle* particle = m_plist->getParticle(iPart);
143 m_hasCovMatrix = false;
144 if (m_updateDaughters == true) {
145 if (m_decayString.empty() || m_vertexFitter == "KFit")
147 else B2ERROR("Daughters update works only when all daughters are selected. Daughters will not be updated");
148 }
149
150 if (m_withConstraint == "mother") {
151 m_BeamSpotCenter = particle->getVertex();
152 m_beamSpotCov = particle->getVertexErrorMatrix();
153 }
154
155 TMatrixFSym mother_errMatrix(7);
156 mother_errMatrix = particle->getMomentumVertexErrorMatrix();
157 for (int k = 0; k < 7; k++) {
158 for (int j = 0; j < 7; j++) {
159 if (mother_errMatrix[k][j] > 0) {
160 m_hasCovMatrix = true;
161 }
162 }
163 }
164
165 bool hasTube = true;
166 if (m_withConstraint == "btube") {
167 Btube* Ver = particle->getRelatedTo<Btube>();
168 if (!Ver) {
169 hasTube = false;
170 toRemove.push_back(particle->getArrayIndex());
171 } else {
172 m_BeamSpotCenter.SetXYZ(Ver->getTubeCenter()(0, 0), Ver->getTubeCenter()(1, 0), Ver->getTubeCenter()(2, 0));
174 }
175 }
176 bool ok = false;
177 if (hasTube) {
178 ok = doVertexFit(particle);
179 }
180 if (!ok)
181 particle->setPValue(-1);
182 if (particle->getPValue() < m_confidenceLevel)
183 toRemove.push_back(particle->getArrayIndex());
184
185 }
186 m_plist->removeParticles(toRemove);
187
188 //free memory allocated by rave. initialize() would be enough, except that we must clean things up before program end...
189 if (m_vertexFitter == "Rave")
191}
TMatrixFSym getTubeMatrix() const
Returns Btube matrix.
Definition Btube.h:73
Eigen::Matrix< double, 3, 1 > getTubeCenter() const
Returns Btube center.
Definition Btube.h:59
void smearBeamSpot(double width)
smear beam spot covariance
bool m_hasCovMatrix
flag for mother covariance matrix (PseudoFitter)
DBObjPtr< BeamSpot > m_beamSpotDB
Beam spot database object.
bool doVertexFit(Particle *p)
Main steering routine.
void findConstraintBoost(double cut)
calculate iptube constraint (quasi cylinder along boost direction) for RAVE fit
StoreObjPtr< ParticleList > m_plist
particle list
static void initialize(int verbosity=1, double MagneticField=1.5)
Set everything up so everything needed for vertex fitting is there.
Definition RaveSetup.cc:35
void reset()
frees memory allocated by initialize().
Definition RaveSetup.cc:60
void copyDaughters(Particle *mother)
Function copies all (grand-)^n-daughter particles of the argument mother Particle.

◆ 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

◆ fillFitParticles()

bool fillFitParticles ( const Particle * mother,
std::vector< const Particle * > & fitChildren,
std::vector< const Particle * > & twoPhotonChildren )
private

Fills valid particle's children (with valid error matrix) in the vector of Particles that will enter the fit.

Particles formed from two photons (e.g. pi0) are treated separately so they are filled to another vector.

Definition at line 308 of file ParticleVertexFitterModule.cc.

310{
311 if (m_decayString.empty()) {
312 // if decayString is empty, just use all primary daughters
313 for (unsigned ichild = 0; ichild < mother->getNDaughters(); ichild++) {
314 const Particle* child = mother->getDaughter(ichild);
315 // This if allows to skip the daughters, which cannot be used in the fits, particularly K_L0 from KLM.
316 // Useful for fully-inclusive particles.
317 if (mother->getProperty() == Particle::PropertyFlags::c_IsUnspecified and child->getPValue() < 0) {
318 continue;
319 }
320 fitChildren.push_back(child);
321 }
322 } else {
323 fitChildren = m_decaydescriptor.getSelectionParticles(mother);
324 }
325
326 auto itr = fitChildren.begin();
327 while (itr != fitChildren.end()) {
328 const Particle* child = *itr;
329
330 if (child->getPValue() < 0) {
331 B2WARNING("Daughter with PDG code " << child->getPDGCode() << " does not have a valid error matrix.");
332 return false; // error matrix not valid
333 }
334 bool isTwoPhotonParticle = false;
335 if (m_hasCovMatrix == false) {
336 if (child->getPDGCode() == Const::pi0.getPDGCode() or child->getPDGCode() == 221) { // pi0 or eta
337 if (child->getNDaughters() == 2) {
338 if (child->getDaughter(0)->getPDGCode() == Const::photon.getPDGCode()
339 && child->getDaughter(1)->getPDGCode() == Const::photon.getPDGCode()) {
340 isTwoPhotonParticle = true;
341 }
342 }
343 }
344 }
345 if (isTwoPhotonParticle) {
346 // move children from fitChildren to twoPhotonChildren
347 twoPhotonChildren.push_back(child);
348 itr = fitChildren.erase(itr);
349 } else {
350 itr++;
351 }
352 }
353
354 return true;
355}
int getPDGCode() const
PDG code.
Definition Const.h:473
static const ParticleType pi0
neutral pion particle
Definition Const.h:674
static const ParticleType photon
photon particle
Definition Const.h:673
int getProperty() const
Returns particle property as a bit pattern The values are defined in the PropertyFlags enum and descr...
Definition Particle.h:518
@ c_IsUnspecified
Ordinary particles.
Definition Particle.h:120

◆ fillNotFitParticles()

bool fillNotFitParticles ( const Particle * mother,
std::vector< const Particle * > & notFitChildren,
const std::vector< const Particle * > & fitChildren )
private

Fills valid particle's children (with valid error matrix) in the vector of Particles that will not enter the fit.

Definition at line 357 of file ParticleVertexFitterModule.cc.

359{
360 if (fitChildren.empty())
361 B2WARNING("[ParticleVertexFitterModule::fillNotFitParticles] fitChildren is empty! Please call fillFitParticles firstly");
362 if (!notFitChildren.empty())
363 B2WARNING("[ParticleVertexFitterModule::fillNotFitParticles] notFitChildren is NOT empty!"
364 << " The function should be called only once");
365
366 if (m_decayString.empty())
367 // if decayString is empty, just use all primary daughters
368 return true;
369
370 std::function<bool(const Particle*)> funcCheckInFit =
371 [&funcCheckInFit, &notFitChildren, fitChildren](const Particle * part) {
372
373 // check if the given particle in fitChildren
374 // if it is included, return true
375 if (std::find(fitChildren.begin(), fitChildren.end(), part) != fitChildren.end())
376 return true;
377
378 // if not, firstly check if particle has children
379 if (part->getNDaughters() == 0)
380 // if it has no children (=final-state-particle), return false
381 return false;
382
383 // here, the given particle is not in fitChildren and has children
384 bool isAnyChildrenInFit = false;
385 vector<const Particle*> notFitChildren_tmp;
386 for (unsigned ichild = 0; ichild < part->getNDaughters(); ichild++) {
387 // call funcCheckInFit recursively for all children
388 const Particle* child = part->getDaughter(ichild);
389 bool isChildrenInFit = funcCheckInFit(child);
390 isAnyChildrenInFit = isChildrenInFit or isAnyChildrenInFit;
391
392 // if the child is not in fitChildren, fill the child in a temporary vector
393 if (!isChildrenInFit)
394 notFitChildren_tmp.push_back(child);
395 }
396
397 // if there are a sister in fitChildren, the children in the temporary vector will be filled in notFitChildren
398 if (isAnyChildrenInFit)
399 notFitChildren.insert(notFitChildren.end(), notFitChildren_tmp.begin(), notFitChildren_tmp.end());
400
401 // if no children in fitChildren, the given particle should be filled instead of all children.
402
403 return isAnyChildrenInFit;
404 };
405
406
407 // call funcCheckInFit for all primary children
408 for (unsigned ichild = 0; ichild < mother->getNDaughters(); ichild++) {
409 const Particle* child = mother->getDaughter(ichild);
410 bool isGivenParticleOrAnyChildrenInFit = funcCheckInFit(child);
411 if (!isGivenParticleOrAnyChildrenInFit)
412 notFitChildren.push_back(child);
413 }
414
415 return true;
416}

◆ findConstraintBoost()

void findConstraintBoost ( double cut)
private

calculate iptube constraint (quasi cylinder along boost direction) for RAVE fit

Definition at line 1471 of file ParticleVertexFitterModule.cc.

1472{
1473 PCmsLabTransform T;
1474
1475 ROOT::Math::XYZVector boostDir = T.getBoostVector().Unit();
1476
1477 TMatrixDSym beamSpotCov = m_beamSpotDB->getCovVertex();
1478 beamSpotCov(2, 2) = cut * cut;
1479 double thetab = boostDir.Theta();
1480 double phib = boostDir.Phi();
1481
1482 double stb = TMath::Sin(thetab);
1483 double ctb = TMath::Cos(thetab);
1484 double spb = TMath::Sin(phib);
1485 double cpb = TMath::Cos(phib);
1486
1487
1488 TMatrix rz(3, 3); rz(2, 2) = 1;
1489 rz(0, 0) = cpb; rz(0, 1) = spb;
1490 rz(1, 0) = -1 * spb; rz(1, 1) = cpb;
1491
1492 TMatrix ry(3, 3); ry(1, 1) = 1;
1493 ry(0, 0) = ctb; ry(0, 2) = -1 * stb;
1494 ry(2, 0) = stb; ry(2, 2) = ctb;
1495
1496 TMatrix r(3, 3); r.Mult(rz, ry);
1497 TMatrix rt(3, 3); rt.Transpose(r);
1498
1499 TMatrix TubePart(3, 3); TubePart.Mult(rt, beamSpotCov);
1500 TMatrix Tube(3, 3); Tube.Mult(TubePart, r);
1501
1502 m_beamSpotCov(0, 0) = Tube(0, 0); m_beamSpotCov(0, 1) = Tube(0, 1); m_beamSpotCov(0, 2) = Tube(0, 2);
1503 m_beamSpotCov(1, 0) = Tube(1, 0); m_beamSpotCov(1, 1) = Tube(1, 1); m_beamSpotCov(1, 2) = Tube(1, 2);
1504 m_beamSpotCov(2, 0) = Tube(2, 0); m_beamSpotCov(2, 1) = Tube(2, 1); m_beamSpotCov(2, 2) = Tube(2, 2);
1505}
ROOT::Math::XYZVector getBoostVector() const
Returns boost vector (beta=p/E)

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

◆ getChi2TracksLBoost()

double getChi2TracksLBoost ( const analysis::VertexFitKFit & kv)
private

calculate the chi2 using only lboost information of tracks

Definition at line 1516 of file ParticleVertexFitterModule.cc.

1517{
1518 double chi2TrackL = 0;
1519
1520 for (int iTrack = 0; iTrack < kv.getTrackCount(); iTrack++) {
1521
1522 analysis::KFitTrack trk_i = kv.getTrack(iTrack); // KFitTrack contains parameters before/after fit.
1523
1524 TMatrixFSym err = CLHEPToROOT::getTMatrixFSym(trk_i.getError(analysis::KFitConst::kBeforeFit)); // px, py, pz, E, x, y, z
1525
1526 ROOT::Math::XYZVector x_before = CLHEPToROOT::getXYZVector(trk_i.getPosition(analysis::KFitConst::kBeforeFit));
1527 ROOT::Math::XYZVector x_after = CLHEPToROOT::getXYZVector(trk_i.getPosition());
1528 ROOT::Math::XYZVector dPos = x_after - x_before;
1529
1530 PCmsLabTransform T;
1531 ROOT::Math::XYZVector boost3 = T.getBoostVector().Unit();
1532 TVectorD boostD(0, 6, 0., 0., 0., 0., boost3.X(), boost3.Y(), boost3.Z(), "END");
1533
1534 double dLBoost = dPos.Dot(boost3);
1535
1536 chi2TrackL += TMath::Power(dLBoost, 2) / err.Similarity(boostD);
1537 }
1538 return chi2TrackL;
1539}
const KFitTrack getTrack(const int id) const
Get a specified track object.
Definition KFitBase.cc:175
const CLHEP::HepSymMatrix getError(const int flag=KFitConst::kAfterFit) const
Get an error matrix of the track.
Definition KFitTrack.cc:172
const HepPoint3D getPosition(const int flag=KFitConst::kAfterFit) const
Get a position of the track.
Definition KFitTrack.cc:164
static const int kBeforeFit
Input parameter to specify before-fit when setting/getting a track attribute.
Definition KFitConst.h:35

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

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 ( void )
overridevirtual

Initialize the Module.

This method is called at the beginning of data processing.

Reimplemented from Module.

Definition at line 81 of file ParticleVertexFitterModule.cc.

82{
83 // Particle list with name m_listName has to exist
84 m_plist.isRequired(m_listName);
85
86 // magnetic field
87 m_Bfield = BFieldManager::getFieldInTesla(ROOT::Math::XYZVector(0, 0, 0)).Z();
88
89 // RAVE setup
90 if (m_vertexFitter == "Rave")
92
93 B2DEBUG(1, "ParticleVertexFitterModule : magnetic field = " << m_Bfield);
94
95
96 if (m_decayString != "")
98
99 B2INFO("ParticleVertexFitter: Performing " << m_fitType << " fit on " << m_listName << " using " << m_vertexFitter);
100 if (m_decayString != "")
101 B2INFO("ParticleVertexFitter: Using specified decay string: " << m_decayString);
102 if (m_withConstraint != "")
103 B2INFO("ParticleVertexFitter: Additional " << m_withConstraint << " will be applied");
104
105}
static ROOT::Math::XYZVector getFieldInTesla(const ROOT::Math::XYZVector &pos)
return the magnetic field at a given position in Tesla.

◆ makeKFourCMother()

bool makeKFourCMother ( analysis::FourCFitKFit & kv,
Particle * p )
private

Update mother particle after FourC fit using KFit.

Parameters
kvreference to KFit MassFit object
ppointer to particle
Returns
true for successful construction of mother

Definition at line 1042 of file ParticleVertexFitterModule.cc.

1043{
1044 enum analysis::KFitError::ECode fitError;
1045 fitError = kf.updateMother(mother);
1046 if (fitError != analysis::KFitError::kNoError)
1047 return false;
1048 mother->addExtraInfo("FourCFitProb", kf.getCHIsq());
1049 mother->addExtraInfo("FourCFitChi2", kf.getNDF());
1050 if (m_decayString.empty() && m_updateDaughters == true) {
1051 // update daughter momenta as well
1052 // the order of daughters in the *fitter is the same as in the mother Particle
1053
1054 std::vector<Particle*> daughters = mother->getDaughters();
1055
1056 const unsigned nd = daughters.size();
1057 unsigned l = 0;
1058 std::vector<std::vector<unsigned>> pars;
1059 std::vector<Particle*> allparticles;
1060 for (unsigned ichild = 0; ichild < nd; ichild++) {
1061 const Particle* daughter = mother->getDaughter(ichild);
1062 std::vector<unsigned> pard;
1063 if (daughter->getNDaughters() > 0) {
1064 updateMapOfTrackAndDaughter(l, pars, pard, allparticles, daughter);
1065 pars.push_back(pard);
1066 allparticles.push_back(daughters[ichild]);
1067 } else {
1068 pard.push_back(l);
1069 pars.push_back(pard);
1070 allparticles.push_back(daughters[ichild]);
1071 l++;
1072 }
1073 }
1074
1075 unsigned track_count = kf.getTrackCount();
1076 if (l != track_count)
1077 return false;
1078
1079 for (unsigned iDaug = 0; iDaug < allparticles.size(); iDaug++) {
1080 ROOT::Math::PxPyPzEVector childMoms;
1081 ROOT::Math::XYZVector childPoss;
1082 TMatrixFSym childErrMatrixs(7);
1083 for (unsigned int iChild : pars[iDaug]) {
1084 childMoms = childMoms +
1085 CLHEPToROOT::getLorentzVector(
1086 kf.getTrackMomentum(iChild));
1087 childPoss = childPoss +
1088 CLHEPToROOT::getXYZVector(
1089 kf.getTrackPosition(iChild));
1090 TMatrixFSym childErrMatrix =
1091 CLHEPToROOT::getTMatrixFSym(kf.getTrackError(iChild));
1092 childErrMatrixs = childErrMatrixs + childErrMatrix;
1093 }
1094 allparticles[iDaug]->set4Vector(childMoms);
1095 allparticles[iDaug]->setVertex(childPoss);
1096 allparticles[iDaug]->setMomentumVertexErrorMatrix(childErrMatrixs);
1097 }
1098 }
1099
1100 return true;
1101}
void updateMapOfTrackAndDaughter(unsigned &l, std::vector< std::vector< unsigned > > &pars, std::vector< unsigned > &pard, std::vector< Particle * > &allparticles, const Particle *daughter)
update the map of daughter and tracks, find out which tracks belong to each daughter.
ECode
ECode is a error code enumerate.
Definition KFitError.h:34

◆ makeKMassMother()

bool makeKMassMother ( analysis::MassFitKFit & kv,
Particle * p )
private

Update mother particle after mass fit using KFit.

Parameters
kvreference to KFit MassFit object
ppointer to particle
Returns
true for successful construction of mother

Definition at line 1002 of file ParticleVertexFitterModule.cc.

1004{
1005 enum analysis::KFitError::ECode fitError;
1006 fitError = km.updateMother(mother);
1007 if (fitError != analysis::KFitError::kNoError)
1008 return false;
1009 if (m_decayString.empty() && m_updateDaughters == true) {
1010 // update daughter momenta as well
1011 // the order of daughters in the *fitter is the same as in the mother Particle
1012
1013 std::vector<Particle*> daughters = mother->getDaughters();
1014
1015 unsigned track_count = km.getTrackCount();
1016 if (daughters.size() != track_count)
1017 return false;
1018
1019 for (unsigned iChild = 0; iChild < track_count; iChild++) {
1020 double a = -1 * Belle2::Const::speedOfLight * 1e-4 * m_Bfield * daughters[iChild]->getCharge();
1021 double dx = km.getVertex().x() - km.getTrackPosition(iChild).x();
1022 double dy = km.getVertex().y() - km.getTrackPosition(iChild).y();
1023
1024 ROOT::Math::PxPyPzEVector i4Vector(km.getTrackMomentum(iChild).x() - a * dy,
1025 km.getTrackMomentum(iChild).y() + a * dx,
1026 km.getTrackMomentum(iChild).z(),
1027 km.getTrackMomentum(iChild).t());
1028 daughters[iChild]->set4VectorDividingByMomentumScaling(i4Vector);
1029
1030 daughters[iChild]->setVertex(
1031 CLHEPToROOT::getXYZVector(km.getTrackPosition(iChild)));
1032 daughters[iChild]->setMomentumVertexErrorMatrix(
1033 CLHEPToROOT::getTMatrixFSym(km.getTrackError(iChild)));
1034 }
1035 }
1036
1037 return true;
1038}
static const double speedOfLight
[cm/ns]
Definition Const.h:695

◆ makeKMassPointingVertexMother()

bool makeKMassPointingVertexMother ( analysis::MassPointingVertexFitKFit & kv,
Particle * p )
private

Update mother particle after mass-constrained vertex fit with additional pointing constraint using KFit.

Parameters
kvreference to KFit MassPointingVertexFit object
ppointer to particle
Returns
true for successful construction of mother

Definition at line 961 of file ParticleVertexFitterModule.cc.

963{
964 enum analysis::KFitError::ECode fitError;
965 fitError = kmpv.updateMother(mother);
966 if (fitError != analysis::KFitError::kNoError) {
967 return false;
968 }
969
970 if (m_decayString.empty() && m_updateDaughters == true) {
971 // update daughter momenta as well
972 // the order of daughters in the *fitter is the same as in the mother Particle
973
974 std::vector<Particle*> daughters = mother->getDaughters();
975
976 unsigned track_count = kmpv.getTrackCount();
977 if (daughters.size() != track_count)
978 return false;
979
980 for (unsigned iChild = 0; iChild < track_count; iChild++) {
981 double a = -1 * Belle2::Const::speedOfLight * 1e-4 * m_Bfield * daughters[iChild]->getCharge();
982 double dx = kmpv.getVertex().x() - kmpv.getTrackPosition(iChild).x();
983 double dy = kmpv.getVertex().y() - kmpv.getTrackPosition(iChild).y();
984
985 ROOT::Math::PxPyPzEVector i4Vector(kmpv.getTrackMomentum(iChild).x() - a * dy,
986 kmpv.getTrackMomentum(iChild).y() + a * dx,
987 kmpv.getTrackMomentum(iChild).z(),
988 kmpv.getTrackMomentum(iChild).t());
989 daughters[iChild]->set4VectorDividingByMomentumScaling(i4Vector);
990
991 daughters[iChild]->setVertex(
992 CLHEPToROOT::getXYZVector(kmpv.getTrackPosition(iChild)));
993 daughters[iChild]->setMomentumVertexErrorMatrix(
994 CLHEPToROOT::getTMatrixFSym(kmpv.getTrackError(iChild)));
995 }
996 }
997
998 return true;
999}

◆ makeKMassVertexMother()

bool makeKMassVertexMother ( analysis::MassVertexFitKFit & kv,
Particle * p )
private

Update mother particle after mass-constrained vertex fit using KFit.

Parameters
kvreference to KFit MassVertexFit object
ppointer to particle
Returns
true for successful construction of mother

Definition at line 923 of file ParticleVertexFitterModule.cc.

925{
926 enum analysis::KFitError::ECode fitError;
927 fitError = kmv.updateMother(mother);
928 if (fitError != analysis::KFitError::kNoError)
929 return false;
930 if (m_decayString.empty() && m_updateDaughters == true) {
931 // update daughter momenta as well
932 // the order of daughters in the *fitter is the same as in the mother Particle
933
934 std::vector<Particle*> daughters = mother->getDaughters();
935
936 unsigned track_count = kmv.getTrackCount();
937 if (daughters.size() != track_count)
938 return false;
939
940 for (unsigned iChild = 0; iChild < track_count; iChild++) {
941 double a = -1 * Belle2::Const::speedOfLight * 1e-4 * m_Bfield * daughters[iChild]->getCharge();
942 double dx = kmv.getVertex().x() - kmv.getTrackPosition(iChild).x();
943 double dy = kmv.getVertex().y() - kmv.getTrackPosition(iChild).y();
944
945 ROOT::Math::PxPyPzEVector i4Vector(kmv.getTrackMomentum(iChild).x() - a * dy,
946 kmv.getTrackMomentum(iChild).y() + a * dx,
947 kmv.getTrackMomentum(iChild).z(),
948 kmv.getTrackMomentum(iChild).t());
949 daughters[iChild]->set4VectorDividingByMomentumScaling(i4Vector);
950
951 daughters[iChild]->setVertex(
952 CLHEPToROOT::getXYZVector(kmv.getTrackPosition(iChild)));
953 daughters[iChild]->setMomentumVertexErrorMatrix(
954 CLHEPToROOT::getTMatrixFSym(kmv.getTrackError(iChild)));
955 }
956 }
957
958 return true;
959}

◆ makeKRecoilMassMother()

bool makeKRecoilMassMother ( analysis::RecoilMassKFit & kf,
Particle * p )
private

Update mother particle after RecoilMass fit using KFit.

Parameters
kfreference to KFit MassFit object
ppointer to particle
Returns
true for successful construction of mother

Definition at line 1165 of file ParticleVertexFitterModule.cc.

1166{
1167 enum analysis::KFitError::ECode fitError;
1168 fitError = kf.updateMother(mother);
1169 if (fitError != analysis::KFitError::kNoError)
1170 return false;
1171 mother->addExtraInfo("RecoilMassFitProb", TMath::Prob(kf.getCHIsq(), kf.getNDF()));
1172 mother->addExtraInfo("RecoilMassFitChi2", kf.getCHIsq());
1173 mother->addExtraInfo("RecoilMassFitNDF", kf.getNDF());
1174 if (m_decayString.empty() && m_updateDaughters == true) {
1175 // update daughter momenta as well
1176 // the order of daughters in the *fitter is the same as in the mother Particle
1177
1178 std::vector<Particle*> daughters = mother->getDaughters();
1179
1180 unsigned track_count = kf.getTrackCount();
1181 if (daughters.size() != track_count)
1182 return false;
1183
1184 for (unsigned iChild = 0; iChild < track_count; iChild++) {
1185 double a = -1 * Belle2::Const::speedOfLight * 1e-4 * m_Bfield * daughters[iChild]->getCharge();
1186 double dx = kf.getVertex().x() - kf.getTrackPosition(iChild).x();
1187 double dy = kf.getVertex().y() - kf.getTrackPosition(iChild).y();
1188
1189 ROOT::Math::PxPyPzEVector i4Vector(kf.getTrackMomentum(iChild).x() - a * dy,
1190 kf.getTrackMomentum(iChild).y() + a * dx,
1191 kf.getTrackMomentum(iChild).z(),
1192 kf.getTrackMomentum(iChild).t());
1193 daughters[iChild]->set4VectorDividingByMomentumScaling(i4Vector);
1194
1195 daughters[iChild]->setVertex(
1196 CLHEPToROOT::getXYZVector(kf.getTrackPosition(iChild)));
1197 daughters[iChild]->setMomentumVertexErrorMatrix(
1198 CLHEPToROOT::getTMatrixFSym(kf.getTrackError(iChild)));
1199 }
1200 }
1201
1202 return true;
1203}
const CLHEP::HepSymMatrix getTrackError(const int id) const
Get an error matrix of the track.
Definition KFitBase.cc:168
const CLHEP::HepLorentzVector getTrackMomentum(const int id) const
Get a Lorentz vector of the track.
Definition KFitBase.cc:154
const HepPoint3D getTrackPosition(const int id) const
Get a position of the track.
Definition KFitBase.cc:161
virtual int getNDF(void) const
Get an NDF of the fit.
Definition KFitBase.cc:114
double getCHIsq(void) const override
Get a chi-square of the fit.
enum KFitError::ECode updateMother(Particle *mother)
Update mother particle.
const HepPoint3D getVertex(const int flag=KFitConst::kAfterFit) const
Get a vertex position.

◆ makeKVertexMother()

bool makeKVertexMother ( analysis::VertexFitKFit & kv,
Particle * p )
private

Update mother particle after unconstrained vertex fit using KFit.

Parameters
kvreference to KFit VertexFit object
ppointer to particle
Returns
true for successful construction of mother

Definition at line 824 of file ParticleVertexFitterModule.cc.

826{
827 enum analysis::KFitError::ECode fitError;
828 fitError = kv.updateMother(mother);
829 if (fitError != analysis::KFitError::kNoError)
830 return false;
831 if (m_decayString.empty() && m_updateDaughters == true) {
832 // update daughter momenta as well
833 // the order of daughters in the *fitter is the same as in the mother Particle
834
835 std::vector<Particle*> daughters = mother->getDaughters();
836
837 unsigned track_count = kv.getTrackCount();
838 if (daughters.size() != track_count)
839 return false;
840
841 for (unsigned iChild = 0; iChild < track_count; iChild++) {
842 double a = -1 * Belle2::Const::speedOfLight * 1e-4 * m_Bfield * daughters[iChild]->getCharge();
843 double dx = kv.getVertex().x() - kv.getTrackPosition(iChild).x();
844 double dy = kv.getVertex().y() - kv.getTrackPosition(iChild).y();
845
846 ROOT::Math::PxPyPzEVector i4Vector(kv.getTrackMomentum(iChild).x() - a * dy,
847 kv.getTrackMomentum(iChild).y() + a * dx,
848 kv.getTrackMomentum(iChild).z(),
849 kv.getTrackMomentum(iChild).t());
850 daughters[iChild]->set4VectorDividingByMomentumScaling(i4Vector);
851
852 daughters[iChild]->setVertex(
853 CLHEPToROOT::getXYZVector(kv.getTrackPosition(iChild)));
854 daughters[iChild]->setMomentumVertexErrorMatrix(
855 CLHEPToROOT::getTMatrixFSym(kv.getTrackError(iChild)));
856 }
857
858 } else if (m_updateDaughters == true) { // if decayString is not empty
859 // first, update only the fit children
860 std::vector<const Particle*> fitChildren = m_decaydescriptor.getSelectionParticles(mother);
861
862 unsigned track_count = kv.getTrackCount();
863 if (fitChildren.size() != track_count)
864 return false;
865
866 for (unsigned iChild = 0; iChild < track_count; iChild++) {
867 auto daughter = const_cast<Particle*>(fitChildren[iChild]);
868
869 double a = -1 * Belle2::Const::speedOfLight * 1e-4 * m_Bfield * daughter->getCharge();
870 double dx = kv.getVertex().x() - kv.getTrackPosition(iChild).x();
871 double dy = kv.getVertex().y() - kv.getTrackPosition(iChild).y();
872
873 ROOT::Math::PxPyPzEVector i4Vector(kv.getTrackMomentum(iChild).x() - a * dy,
874 kv.getTrackMomentum(iChild).y() + a * dx,
875 kv.getTrackMomentum(iChild).z(),
876 kv.getTrackMomentum(iChild).t());
877 daughter->set4VectorDividingByMomentumScaling(i4Vector);
878
879 daughter->setVertex(CLHEPToROOT::getXYZVector(kv.getTrackPosition(iChild)));
880 daughter->setMomentumVertexErrorMatrix(CLHEPToROOT::getTMatrixFSym(kv.getTrackError(iChild)));
881 }
882
883 // then, update other particles that have a fit-child in decay
884 std::function<bool(Particle*)> funcUpdateMomentum =
885 [&funcUpdateMomentum, fitChildren](Particle * part) {
886
887 if (part->getNDaughters() == 0) {
888 // check if part is included in fitChildren
889 if (std::find(fitChildren.begin(), fitChildren.end(), part) != fitChildren.end())
890 return true;
891 else
892 return false;
893 }
894
895 bool includeFitChildren = false;
896
897 // Update daughters' momentum
898 for (auto daughter : part->getDaughters())
899 includeFitChildren = funcUpdateMomentum(daughter) || includeFitChildren;
900
901 if (includeFitChildren) {
902 // Using updated daughters, update part's momentum
903 ROOT::Math::PxPyPzEVector sum4Vector;
904 for (auto daughter : part->getDaughters())
905 sum4Vector += daughter->get4Vector();
906
907 part->set4VectorDividingByMomentumScaling(sum4Vector);
908 }
909
910 return includeFitChildren;
911 };
912
913 // Update all daughters
914 for (auto daughter : mother->getDaughters())
915 funcUpdateMomentum(daughter);
916
917 }
918
919
920 return true;
921}
enum KFitError::ECode updateMother(Particle *mother)
Update mother particle.
const HepPoint3D getVertex(const int flag=KFitConst::kAfterFit) const
Get a vertex position.

◆ makeMassKFourCMother()

bool makeMassKFourCMother ( analysis::MassFourCFitKFit & kv,
Particle * p )
private

Update mother particle after MassFourC fit using KFit.

Parameters
kvreference to KFit MassFit object
ppointer to particle
Returns
true for successful construction of mother

Definition at line 1103 of file ParticleVertexFitterModule.cc.

1104{
1105 enum analysis::KFitError::ECode fitError;
1106 fitError = kf.updateMother(mother);
1107 if (fitError != analysis::KFitError::kNoError)
1108 return false;
1109 mother->addExtraInfo("MassFourCFitProb", TMath::Prob(kf.getCHIsq(), kf.getNDF()));
1110 mother->addExtraInfo("MassFourCFitChi2", kf.getCHIsq());
1111 mother->addExtraInfo("MassFourCFitNDF", kf.getNDF());
1112 if (m_decayString.empty() && m_updateDaughters == true) {
1113 // update daughter momenta as well
1114 // the order of daughters in the *fitter is the same as in the mother Particle
1115
1116 std::vector<Particle*> daughters = mother->getDaughters();
1117
1118 const unsigned nd = daughters.size();
1119 unsigned l = 0;
1120 std::vector<std::vector<unsigned>> pars;
1121 std::vector<Particle*> allparticles;
1122 for (unsigned ichild = 0; ichild < nd; ichild++) {
1123 const Particle* daughter = mother->getDaughter(ichild);
1124 std::vector<unsigned> pard;
1125 if (daughter->getNDaughters() > 0) {
1126 updateMapOfTrackAndDaughter(l, pars, pard, allparticles, daughter);
1127 pars.push_back(pard);
1128 allparticles.push_back(daughters[ichild]);
1129 } else {
1130 pard.push_back(l);
1131 pars.push_back(pard);
1132 allparticles.push_back(daughters[ichild]);
1133 l++;
1134 }
1135 }
1136
1137 unsigned track_count = kf.getTrackCount();
1138 if (l != track_count)
1139 return false;
1140
1141 for (unsigned iDaug = 0; iDaug < allparticles.size(); iDaug++) {
1142 ROOT::Math::PxPyPzEVector childMoms;
1143 ROOT::Math::XYZVector childPoss;
1144 TMatrixFSym childErrMatrixs(7);
1145 for (unsigned int iChild : pars[iDaug]) {
1146 childMoms = childMoms +
1147 CLHEPToROOT::getLorentzVector(
1148 kf.getTrackMomentum(iChild));
1149 childPoss = childPoss +
1150 CLHEPToROOT::getXYZVector(
1151 kf.getTrackPosition(iChild));
1152 TMatrixFSym childErrMatrix =
1153 CLHEPToROOT::getTMatrixFSym(kf.getTrackError(iChild));
1154 childErrMatrixs = childErrMatrixs + childErrMatrix;
1155 }
1156 allparticles[iDaug]->set4Vector(childMoms);
1157 allparticles[iDaug]->setVertex(childPoss);
1158 allparticles[iDaug]->setMomentumVertexErrorMatrix(childErrMatrixs);
1159 }
1160 }
1161
1162 return true;
1163}

◆ redoTwoPhotonDaughterMassFit()

bool redoTwoPhotonDaughterMassFit ( Particle * postFit,
const Particle * preFit,
const analysis::VertexFitKFit & kv )
private

Combines preFit particle and vertex information from vertex fit kv to create new postFit particle.

A mass refit of this new particle is performed assuming that it originates from the point given by VertexFit.

Definition at line 418 of file ParticleVertexFitterModule.cc.

420{
421 // TODO: something like setGammaError is necessary
422 // this is just workaround for the moment
423
424 const Particle* g1Orig = preFit->getDaughter(0);
425 const Particle* g2Orig = preFit->getDaughter(1);
426 Particle g1Temp(g1Orig->get4Vector(), 22);
427 Particle g2Temp(g2Orig->get4Vector(), 22);
428
429 TMatrixFSym g1ErrMatrix = g1Orig->getMomentumVertexErrorMatrix();
430 TMatrixFSym g2ErrMatrix = g2Orig->getMomentumVertexErrorMatrix();
431
432 ROOT::Math::XYZVector pos(kv.getVertex().x(), kv.getVertex().y(), kv.getVertex().z());
433 CLHEP::HepSymMatrix posErrorMatrix = kv.getVertexError();
434
435 TMatrixFSym errMatrix(3);
436 for (int i = 0; i < 3; i++)
437 for (int j = 0; j < 3; j++)
438 errMatrix(i, j) = posErrorMatrix[i][j];
439
440 g1ErrMatrix.SetSub(4, errMatrix);
441 g2ErrMatrix.SetSub(4, errMatrix);
442
443 g1Temp.updateMomentum(g1Orig->get4Vector(), pos, g1ErrMatrix, 1.0);
444 g2Temp.updateMomentum(g2Orig->get4Vector(), pos, g2ErrMatrix, 1.0);
445
446 // perform the mass fit for the two-photon particle
447 analysis::MassFitKFit km;
449
450 km.addParticle(&g1Temp);
451 km.addParticle(&g2Temp);
452
453 km.setVertex(kv.getVertex());
455 km.setInvariantMass(preFit->getPDGMass());
456
457 int err = km.doFit();
458 if (err != 0) {
459 return false;
460 }
461
462 // The update of the daughters is disabled for this mass fit.
463 bool updateDaughters = m_updateDaughters;
464 m_updateDaughters = false;
465 bool ok = makeKMassMother(km, postFit);
466 m_updateDaughters = updateDaughters;
467
468 return ok;
469}
double getPDGMass(void) const
Returns uncertainty on the invariant mass (requires valid momentum error matrix)
Definition Particle.cc:635
TMatrixFSym getMomentumVertexErrorMatrix() const
Returns 7x7 error matrix.
Definition Particle.cc:451
enum KFitError::ECode setVertex(const HepPoint3D &v)
Set an initial vertex position for the mass-constraint fit.
enum KFitError::ECode setVertexError(const CLHEP::HepSymMatrix &e)
Set an initial vertex error matrix for the mass-constraint fit.
const CLHEP::HepSymMatrix getVertexError(void) const
Get a fitted vertex error matrix.

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

◆ smearBeamSpot()

void smearBeamSpot ( double width)
private

smear beam spot covariance

Definition at line 1507 of file ParticleVertexFitterModule.cc.

1508{
1509 TMatrixDSym beamSpotCov = m_beamSpotDB->getCovVertex();
1510 for (int i = 0; i < 3; i++)
1511 beamSpotCov(i, i) += width * width;
1512
1513 m_beamSpotCov = beamSpotCov;
1514}

◆ terminate()

virtual void terminate ( void )
inlinevirtualinherited

This method is called at the end of the event processing.

This method is called only once after the event processing finished. Use this method for cleaning up, closing files, etc.

This method can be implemented by subclasses.

Reimplemented in AafhInputModule, AnalysisPhase1StudyModule, ARICHBackgroundModule, arichBtestModule, ARICHNtupleModule, arichToNtupleModule, AsyncWrapper, AWESOMEBasicModule, B2BIIConvertMdstModule, B2BIIFixMdstModule, B2BIIMCParticlesMonitorModule, B2BIIMdstInputModule, BabayagaNLOInputModule, BBBremInputModule, BeamabortModule, BeamabortStudyModule, BeamDigitizerModule, BeamBkgGeneratorModule, BeamBkgHitRateMonitorModule, BeamBkgMixerModule, BeamBkgTagSetterModule, BeamSpotMonitorModule, BelleMCOutputModule, BgoDigitizerModule, BgoModule, BgoStudyModule, BGOverlayInputModule, BHWideInputModule, BKLMAnaModule, BKLMDigitAnalyzerModule, BKLMSimHistogrammerModule, BKLMTrackingModule, CalibrationCollectorModule, CaveModule, CDCCosmicAnalysisModule, CDCCRTestModule, CDCPackerModule, CDCRecoTrackFilterModule, CDCUnpackerModule, CDCCrossTalkAdderModule, CDCDedxCorrectionModule, CDCDedxDQMModule, CDCDedxPIDModule, CDCDedxScanModule, CDCDedxSkimCDSTModule, CDCDedxSkimModule, CDCDedxValidationModule, CDCDigitizerModule, cdcDQM7Module, CDCDQMModule, CDCInitialT0DeterminationModule, CDCTrigger2DFinderModule, CDCTriggerHoughETFModule, CDCTriggerNDFinderModule, CDCTriggerNeuroDataModule, CDCTriggerNeuroDQMModule, CDCTriggerNeuroDQMOnlineModule, CDCTriggerNeuroIDHistModule, CDCTriggerTSFFirmwareModule, CDCTriggerTSFModule, CDCTriggerUnpackerModule, CertifyParallelModule, CheckErrorEventModule, Chi2MCTrackMatcherModule, ClawDigitizerModule, ClawModule, ClawStudyModule, ClawsDigitizerModule, CLAWSModule, ClawsStudyModule, CollectorTestModule, Convert2RawDetModule, CosmicsAlignmentValidationModule, CRYInputModule, CsiDigitizer_v2Module, CsIDigitizerModule, CsiModule, CsiStudy_v2Module, CsIStudyModule, CurlingTrackCandSplitterModule, CurlTaggerModule, DAQPerfModule, DataWriterModule, DeSerializerModule, DeSerializerPXDModule, DisplayModule, DosiDigitizerModule, DosiModule, DosiStudyModule, DQMHistAnalysisARICHModule, DQMHistAnalysisARICHMonObjModule, DQMHistAnalysisCDCDedxModule, DQMHistAnalysisCDCEpicsModule, DQMHistAnalysisCDCMonObjModule, DQMHistAnalysisDAQMonObjModule, DQMHistAnalysisDeltaEpicsMonObjExampleModule, DQMHistAnalysisDeltaTestModule, DQMHistAnalysisECLConnectedRegionsModule, DQMHistAnalysisECLModule, DQMHistAnalysisECLOutOfTimeDigitsModule, DQMHistAnalysisECLShapersModule, DQMHistAnalysisECLSummaryModule, DQMHistAnalysisEpicsExampleModule, DQMHistAnalysisEventT0EfficiencyModule, DQMHistAnalysisEventT0TriggerJitterModule, DQMHistAnalysisExampleFlagsModule, DQMHistAnalysisExampleModule, DQMHistAnalysisHLTModule, DQMHistAnalysisHLTMonObjModule, DQMHistAnalysisInput2Module, DQMHistAnalysisInputPVSrvModule, DQMHistAnalysisInputTestModule, DQMHistAnalysisKLMModule, DQMHistAnalysisMiraBelleModule, DQMHistAnalysisMonObjModule, DQMHistAnalysisOutputFileModule, DQMHistAnalysisOutputMonObjModule, DQMHistAnalysisOutputRelayMsgModule, DQMHistAnalysisPeakModule, DQMHistAnalysisPhysicsModule, DQMHistAnalysisPXDChargeModule, DQMHistAnalysisPXDCMModule, DQMHistAnalysisPXDDAQModule, DQMHistAnalysisPXDEffModule, DQMHistAnalysisPXDERModule, DQMHistAnalysisPXDFitsModule, DQMHistAnalysisPXDInjectionModule, DQMHistAnalysisPXDReductionModule, DQMHistAnalysisPXDTrackChargeModule, DQMHistAnalysisRooFitExampleModule, DQMHistAnalysisRunNrModule, DQMHistAnalysisSVDClustersOnTrackModule, DQMHistAnalysisSVDEfficiencyModule, DQMHistAnalysisSVDGeneralModule, DQMHistAnalysisSVDOccupancyModule, DQMHistAnalysisSVDOnMiraBelleModule, DQMHistAnalysisSVDUnpackerModule, DQMHistAnalysisTOPModule, DQMHistAnalysisTrackingAbortModule, DQMHistAnalysisTRGECLModule, DQMHistAnalysisTRGEFFModule, DQMHistAnalysisTRGGDLModule, DQMHistAnalysisTRGModule, DQMHistAutoCanvasModule, DQMHistComparitorModule, DQMHistDeltaHistoModule, DqmHistoManagerModule, DQMHistOutputToEPICSModule, DQMHistReferenceModule, DQMHistSnapshotsModule, Ds2RawFileModule, Ds2RawModule, Ds2RbufModule, Ds2SampleModule, ECLLOMModule, ECLBackgroundModule, ECLChargedPIDDataAnalysisModule, ECLChargedPIDDataAnalysisValidationModule, ECLChargedPIDModule, ECLClusterPSDModule, ECLCompressBGOverlayModule, ECLCovarianceMatrixModule, EclCovMatrixNtupleModule, ECLCRFinderModule, ECLDataAnalysisModule, ECLDigiStudyModule, ECLDigitCalibratorModule, ECLDigitizerModule, ECLDigitizerPureCsIModule, EclDisplayModule, ECLDQMEXTENDEDModule, ECLDQMModule, ECLExpertModule, ECLFillCellIdMappingModule, ECLFinalizerModule, ECLHitDebugModule, ECLLocalMaximumFinderModule, ECLMatchingPerformanceExpertModule, ECLPackerModule, ECLShowerCorrectorModule, ECLShowerShapeModule, ECLSplitterN1Module, ECLSplitterN2Module, ECLTrackClusterMatchingModule, ECLTrackClusterMatchingParametrizationExpertModule, ECLTrackClusterMatchingPerformanceModule, ECLTRGInformationModule, ECLTrimShowersAndDigitsModule, ECLUnpackerModule, eclWaveformCalibCollectorModule, ECLWaveformFitModule, EffPlotsModule, EKLMDataCheckerModule, ElapsedTimeModule, EventKinematicsModule, EvReductionModule, EvtGenDecayModule, ExtModule, FANGSDigitizerModule, FANGSModule, FANGSStudyModule, FastBDTClassifierAnalyzerModule, FastBDTClassifierTrainingModule, FastRbuf2DsModule, FillTrackFitNtupleModule, FlavorTaggerInfoFillerModule, FragmentationModule, FullSimModule, FullSimTimingModule, TRGGDLUnpackerModule, GeneratedVertexDisplacerModule, GeneratorPreselectionModule, GenfitVisModule, GenRawSendModule, GeometryModule, GetEventFromSocketModule, GFTC2SPTCConverterModule, GRLNeuroModule, GRLNeuroTrainerModule, He3DigitizerModule, He3tubeModule, He3tubeStudyModule, HepevtOutputModule, HepMCInputModule, HepMCOutputModule, HistoManagerModule, HistoModule, HitLevelInfoWriterModule, HitXPModule, HLTDQM2ZMQModule, HLTDs2ZMQModule, KinkFinderModule, KKGenInputModule, KLMClusterEfficiencyModule, KLMClustersReconstructorModule, KLMDigitizerModule, KLMDQM2Module, KLMDQMModule, KLMExpertModule, KLMMuonIDDNNExpertModule, KLMPackerModule, KLMReconstructorModule, KLMScintillatorSimulatorModule, KLMTrackingModule, KLMTriggerModule, KLMUnpackerModule, KlongValidationModule, KoralWInputModule, LowEnergyPi0IdentificationExpertModule, LowEnergyPi0VetoExpertModule, MasterClassModule, MCMatcherTRGECLModule, MCTrackCandClassifierModule, MCV0MatcherModule, MergeDataStoreModule, MicrotpcModule, MicrotpcStudyModule, TpcDigitizerModule, MLSegmentNetworkProducerModule, MonitorDataModule, MuidModule, MVAExpertModule, MVAMultipleExpertsModule, NoKickCutsEvalModule, NtuplePhase1_v6Module, ParticleKinematicFitterModule, OverrideGenerationFlagsModule, PartialSeqRootReaderModule, ParticleLoaderModule, ParticleMassHypothesesUpdaterModule, ParticleMassUpdaterModule, ParticleMCDecayStringModule, ParticleMomentumUpdaterModule, ParticleStatsModule, ParticleWeightingLookUpCreatorModule, Ph1bpipeModule, Ph1sustrModule, PhaseSpaceAnalysisModule, PhokharaInputModule, PhysicsObjectsDQMModule, PhysicsObjectsMiraBelleBhabhaModule, PhysicsObjectsMiraBelleDst2Module, PhysicsObjectsMiraBelleDstModule, PhysicsObjectsMiraBelleHadronModule, PhysicsObjectsMiraBelleModule, PIDNtupleModule, PinDigitizerModule, PindiodeModule, PindiodeStudyModule, PlumeDigitizerModule, PlumeModule, ProfileModule, ProgressBarModule, PXDBackgroundModule, PXDBgTupleProducerModule, PXDMCBgTupleProducerModule, PXDPackerErrModule, PXDPackerModule, PXDReadRawBonnDAQMatchedModule, PXDReadRawBonnDAQModule, PXDReadRawONSENModule, PXDUnpackerModule, PXDUnpackerOldModule, PXDUnpackerOTModule, PXDClustersFromTracksModule, PXDDQMEfficiencyNtupleModule, PXDDQMEfficiencyNtupleSelftrackModule, PXDDQMTrackRawNtupleModule, PXDPerformanceModule, PXDROIFinderAnalysisModule, PXDSpacePointCreatorModule, PyModule, QcsmonitorDigitizerModule, QcsmonitorModule, QcsmonitorStudyModule, RandomBarrierModule, Raw2DsModule, RawInputModule, Rbuf2DsModule, Rbuf2RbufModule, ReceiveEventModule, RemoveMCParticlesModule, RemoveParticlesNotInListsModule, ReprocessorModule, ROIDQMModule, ROIReadTestModule, ROISenderModule, Root2BinaryModule, Root2RawModule, RootInputModule, RootOutputModule, RT2SPTCConverterModule, RxModule, RxSocketModule, ScanCDCGeoModule, SecMapTrainerBaseModule, SecMapTrainerVXDTFModule, SegmentNetworkAnalyzerModule, SeqRootInputModule, SeqRootMergerModule, SeqRootOutputModule, SerializerModule, SkimFilterModule, SoftwareTriggerModule, SoftwareTriggerResultPrinterModule, SpacePoint2TrueHitConnectorModule, SpacePointCreatorTestModule, SPTC2GFTCConverterModule, SPTC2RTConverterModule, SPTCRefereeModule, SrsensorModule, StandardTrackingPerformanceModule, StorageDeserializerModule, StorageRootOutputModule, StorageSerializerModule, StudyMaterialEffectsModule, SubEventModule, SVD3SamplesEmulatorModule, SVDBackgroundModule, SVDDigitizerModule, SVDPackerModule, SVDChannelMappingModule, SVDChargeSharingAnalysisModule, SVDClusterEvaluationTrueInfoModule, SVDClusterFilterModule, SVDClusterQualityEstimatorCalibrationModule, SVDClusterQualityEstimatorModule, SVDCoGTimeEstimatorModule, SVDCrossTalkFinderModule, SVDDataFormatCheckModule, SVDDQMClustersOnTrackModule, SVDDQMExpressRecoModule, svdDumpModule, SVDEventT0PerformanceTTreeModule, SVDHotStripFinderModule, SVDMaxStripTTreeModule, SVDPerformanceTTreeModule, SVDROIFinderAnalysisDataModule, SVDROIFinderAnalysisModule, SVDROIFinderModule, SVDShaperDigitsFromTracksModule, SVDSpacePointCreatorModule, SVDSpacePointQICalibrationModule, SVDTriggerQualityGeneratorModule, SwitchDataStoreModule, TCConvertersTestModule, TeeggInputModule, TOPAlignerModule, TOPBackgroundModule, TOPBunchFinderModule, TOPChannelT0CalibratorModule, TOPChannelT0MCModule, TOPCommonT0CalibratorModule, TOPCosmicT0FinderModule, TOPDoublePulseGeneratorModule, TOPGainEfficiencyCalculatorModule, TOPInterimFENtupleModule, TOPLaserCalibratorModule, TOPLaserHitSelectorModule, TOPLLScannerModule, TOPMCTrackMakerModule, TOPModuleT0CalibratorModule, TOPNtupleModule, TOPPackerModule, TOPPDFCheckerModule, TOPRawDigitConverterModule, TOPRingPlotterModule, TOPTBCComparatorModule, TOPTimeBaseCalibratorModule, TOPTriggerDigitizerModule, TOPUnpackerModule, TOPWaveformFeatureExtractorModule, TOPXTalkChargeShareSetterModule, TrackAnaModule, TrackFilterModule, FindletModule< AFindlet >, FindletModule< AsicBackgroundLibraryCreator >, FindletModule< AxialSegmentPairCreator >, FindletModule< AxialSegmentPairCreator >, FindletModule< AxialStraightTrackFinder >, FindletModule< AxialStraightTrackFinder >, FindletModule< AxialTrackCreatorMCTruth >, FindletModule< AxialTrackCreatorMCTruth >, FindletModule< AxialTrackCreatorSegmentHough >, FindletModule< AxialTrackCreatorSegmentHough >, FindletModule< AxialTrackFinderHough >, FindletModule< AxialTrackFinderHough >, FindletModule< AxialTrackFinderLegendre >, FindletModule< AxialTrackFinderLegendre >, FindletModule< CDCTrackingEventLevelMdstInfoFillerFromHitsFindlet >, FindletModule< CDCTrackingEventLevelMdstInfoFillerFromSegmentsFindlet >, FindletModule< CKFToCDCFindlet >, FindletModule< CKFToCDCFromEclFindlet >, FindletModule< CKFToPXDFindlet >, FindletModule< CKFToSVDFindlet >, FindletModule< CKFToSVDSeedFindlet >, FindletModule< ClusterBackgroundDetector >, FindletModule< ClusterBackgroundDetector >, FindletModule< ClusterPreparer >, FindletModule< ClusterPreparer >, FindletModule< ClusterRefiner< BridgingWireHitRelationFilter > >, FindletModule< ClusterRefiner< BridgingWireHitRelationFilter > >, FindletModule< CosmicsTrackMergerFindlet >, FindletModule< DATCONFPGAFindlet >, FindletModule< FacetCreator >, FindletModule< FacetCreator >, FindletModule< HitBasedT0Extractor >, FindletModule< HitBasedT0Extractor >, FindletModule< HitReclaimer >, FindletModule< HitReclaimer >, FindletModule< MCVXDCDCTrackMergerFindlet >, FindletModule< MonopoleAxialTrackFinderLegendre >, FindletModule< MonopoleAxialTrackFinderLegendre >, FindletModule< MonopoleStereoHitFinder >, FindletModule< MonopoleStereoHitFinder >, FindletModule< MonopoleStereoHitFinderQuadratic >, FindletModule< MonopoleStereoHitFinderQuadratic >, FindletModule< SegmentCreatorFacetAutomaton >, FindletModule< SegmentCreatorFacetAutomaton >, FindletModule< SegmentCreatorMCTruth >, FindletModule< SegmentCreatorMCTruth >, FindletModule< SegmentFinderFacetAutomaton >, FindletModule< SegmentFinderFacetAutomaton >, FindletModule< SegmentFitter >, FindletModule< SegmentFitter >, FindletModule< SegmentLinker >, FindletModule< SegmentLinker >, FindletModule< SegmentOrienter >, FindletModule< SegmentOrienter >, FindletModule< SegmentPairCreator >, FindletModule< SegmentPairCreator >, FindletModule< SegmentRejecter >, FindletModule< SegmentRejecter >, FindletModule< SegmentTrackCombiner >, FindletModule< SegmentTrackCombiner >, FindletModule< SegmentTripleCreator >, FindletModule< SegmentTripleCreator >, FindletModule< StereoHitFinder >, FindletModule< StereoHitFinder >, FindletModule< SuperClusterCreator >, FindletModule< SuperClusterCreator >, FindletModule< TrackCombiner >, FindletModule< TrackCombiner >, FindletModule< TrackCreatorSegmentPairAutomaton >, FindletModule< TrackCreatorSegmentPairAutomaton >, FindletModule< TrackCreatorSegmentTripleAutomaton >, FindletModule< TrackCreatorSegmentTripleAutomaton >, FindletModule< TrackCreatorSingleSegments >, FindletModule< TrackCreatorSingleSegments >, FindletModule< TrackExporter >, FindletModule< TrackExporter >, FindletModule< TrackFinder >, FindletModule< TrackFinderAutomaton >, FindletModule< TrackFinderCosmics >, FindletModule< TrackFinderSegmentPairAutomaton >, FindletModule< TrackFinderSegmentPairAutomaton >, FindletModule< TrackFinderSegmentTripleAutomaton >, FindletModule< TrackFinderSegmentTripleAutomaton >, FindletModule< TrackFindingCDC::FindletStoreArrayInput< BaseEventTimeExtractorModuleFindlet< AFindlet > > >, FindletModule< TrackFindingCDC::FindletStoreArrayInput< BaseEventTimeExtractorModuleFindlet< AFindlet > > >, FindletModule< TrackFlightTimeAdjuster >, FindletModule< TrackFlightTimeAdjuster >, FindletModule< TrackLinker >, FindletModule< TrackLinker >, FindletModule< TrackOrienter >, FindletModule< TrackOrienter >, FindletModule< TrackQualityAsserter >, FindletModule< TrackQualityAsserter >, FindletModule< TrackQualityEstimator >, FindletModule< TrackQualityEstimator >, FindletModule< TrackRejecter >, FindletModule< TrackRejecter >, FindletModule< vxdHoughTracking::SVDHoughTracking >, FindletModule< WireHitBackgroundDetector >, FindletModule< WireHitBackgroundDetector >, FindletModule< WireHitCreator >, FindletModule< WireHitCreator >, FindletModule< WireHitPreparer >, FindletModule< WireHitPreparer >, TrackFinderVXDAnalizerModule, TrackingPerformanceEvaluationModule, TrackQETrainingDataCollectorModule, TreeFitterModule, TrepsInputModule, TRGCDCETFUnpackerModule, TRGCDCModule, TRGCDCT2DDQMModule, TRGCDCT3DConverterModule, TRGCDCT3DDQMModule, TRGCDCT3DUnpackerModule, TRGCDCTSFUnpackerModule, TRGCDCTSFDQMModule, TRGCDCTSStreamModule, TRGECLBGTCHitModule, TRGECLDQMModule, TRGECLFAMModule, TRGECLModule, TRGECLQAMModule, TRGECLRawdataAnalysisModule, TRGECLTimingCalModule, TRGECLUnpackerModule, TRGEFFDQMModule, TRGGDLDQMModule, TRGGDLDSTModule, TRGGDLModule, TRGGDLSummaryModule, TRGGRLDQMModule, TRGGRLMatchModule, TRGGRLModule, TRGGRLProjectsModule, TRGGRLUnpackerModule, TRGRAWDATAModule, TRGTOPDQMModule, TRGTOPTRD2TTSConverterModule, TRGTOPUnpackerModule, TRGTOPUnpackerWaveformModule, TRGTOPWaveformPlotterModule, TxModule, TxSocketModule, V0FinderModule, V0findingPerformanceEvaluationModule, VariablesToEventBasedTreeModule, VariablesToHistogramModule, VariablesToNtupleModule, VXDMisalignmentModule, VXDDedxPIDModule, VXDQETrainingDataCollectorModule, VXDTFTrainingDataCollectorModule, ZMQRxOutputModule, ZMQRxWorkerModule, ZMQTxInputModule, and ZMQTxWorkerModule.

Definition at line 175 of file Module.h.

175{};

◆ updateMapOfTrackAndDaughter()

void updateMapOfTrackAndDaughter ( unsigned & l,
std::vector< std::vector< unsigned > > & pars,
std::vector< unsigned > & pard,
std::vector< Particle * > & allparticles,
const Particle * daughter )
private

update the map of daughter and tracks, find out which tracks belong to each daughter.

Parameters
lrepresent the tracks ID
parsmap of all parameters
pardvector of parameters
allparticlesvector of all particles
daughterpointer to particle

Definition at line 1206 of file ParticleVertexFitterModule.cc.

1208{
1209 std::vector <Belle2::Particle*> daughters = daughter->getDaughters();
1210 for (unsigned ichild = 0; ichild < daughter->getNDaughters(); ichild++) {
1211 const Particle* child = daughter->getDaughter(ichild);
1212 std::vector<unsigned> pard;
1213 if (child->getNDaughters() > 0) {
1214 updateMapOfTrackAndDaughter(l, pars, pard, allparticles, child);
1215 parm.insert(parm.end(), pard.begin(), pard.end());
1216 pars.push_back(pard);
1217 allparticles.push_back(daughters[ichild]);
1218 } else {
1219 pard.push_back(l);
1220 parm.push_back(l);
1221 pars.push_back(pard);
1222 allparticles.push_back(daughters[ichild]);
1223 l++;
1224 }
1225 }
1226}

Member Data Documentation

◆ m_BeamSpotCenter

ROOT::Math::XYZVector m_BeamSpotCenter
private

Beam spot position.

Definition at line 89 of file ParticleVertexFitterModule.h.

◆ m_beamSpotCov

TMatrixDSym m_beamSpotCov
private

Beam spot covariance matrix.

Definition at line 90 of file ParticleVertexFitterModule.h.

◆ m_beamSpotDB

DBObjPtr<BeamSpot> m_beamSpotDB
private

Beam spot database object.

Definition at line 91 of file ParticleVertexFitterModule.h.

◆ m_Bfield

double m_Bfield
private

magnetic field from data base

Definition at line 81 of file ParticleVertexFitterModule.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_confidenceLevel

double m_confidenceLevel
private

required fit confidence level

Definition at line 80 of file ParticleVertexFitterModule.h.

◆ m_decaydescriptor

DecayDescriptor m_decaydescriptor
private

Decay descriptor of decays to look for.

Definition at line 87 of file ParticleVertexFitterModule.h.

◆ m_decayString

std::string m_decayString
private

daughter particles selection

Definition at line 85 of file ParticleVertexFitterModule.h.

◆ m_description

std::string m_description
privateinherited

The description of the module.

Definition at line 510 of file Module.h.

◆ m_fitType

std::string m_fitType
private

type of the kinematic fit

Definition at line 83 of file ParticleVertexFitterModule.h.

◆ m_hasCovMatrix

bool m_hasCovMatrix = false
private

flag for mother covariance matrix (PseudoFitter)

Definition at line 88 of file ParticleVertexFitterModule.h.

◆ m_hasReturnValue

bool m_hasReturnValue
privateinherited

True, if the return value is set.

Definition at line 517 of file Module.h.

◆ m_listName

std::string m_listName
private

particle list name

Definition at line 79 of file ParticleVertexFitterModule.h.

◆ m_logConfig

LogConfig m_logConfig
privateinherited

The log system configuration of the module.

Definition at line 513 of file Module.h.

◆ m_massConstraintList

std::vector<int> m_massConstraintList
private

PDG codes of the particles to be mass constraint (massfourC)

Definition at line 94 of file ParticleVertexFitterModule.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_plist

StoreObjPtr<ParticleList> m_plist
private

particle list

Definition at line 78 of file ParticleVertexFitterModule.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_recoilMass

double m_recoilMass
private

recoil mass for constraint

Definition at line 93 of file ParticleVertexFitterModule.h.

◆ m_returnValue

int m_returnValue
privateinherited

The return value.

Definition at line 518 of file Module.h.

◆ m_smearing

double m_smearing
private

smearing width applied to IP tube

Definition at line 92 of file ParticleVertexFitterModule.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_updateDaughters

bool m_updateDaughters
private

flag for daughters update

Definition at line 86 of file ParticleVertexFitterModule.h.

◆ m_vertexFitter

std::string m_vertexFitter
private

Vertex Fitter name.

Definition at line 82 of file ParticleVertexFitterModule.h.

◆ m_withConstraint

std::string m_withConstraint
private

additional constraint on vertex

Definition at line 84 of file ParticleVertexFitterModule.h.


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