another unnecessary layer of abstraction More...

#include <InternalParticle.h>

Inheritance diagram for InternalParticle:

Collaboration diagram for InternalParticle:

[legend]

Public Types
enum	TFParticleType { kInteractionPoint , kOrigin , kComposite , kRecoResonance , kInternalParticle , kRecoTrack , kResonance , kRecoPhoton , kRecoKlong , kMissingParticle }
	particle types

typedef std::vector< Constraint >	constraintlist
	alias

typedef std::vector< std::pair< const ParticleBase *, int > >	indexmap
	alias

Public Member Functions
	InternalParticle (Belle2::Particle particle, const ParticleBase mother, const ConstraintConfiguration &config, bool forceFitAll)
	constructor

virtual	~InternalParticle ()
	destructor

virtual ErrCode	initCovariance (FitParams &) const override
	init covariance

ErrCode	projectKineConstraint (const FitParams &, Projection &) const
	project kinematical constraint

ErrCode	projectBeamConstraint (const FitParams &, Projection &) const
	project beam four momentum constraint

virtual void	forceP4Sum (FitParams &) const override
	enforce conservation of momentum sum

virtual ErrCode	initParticleWithMother (FitParams &fitparams) override
	init particle in case it has a mother

virtual ErrCode	initMotherlessParticle (FitParams &fitparams) override
	init particle in case it has no mother More...

ErrCode	projectConstraint (const Constraint::Type type, const FitParams &fitparams, Projection &p) const override
	find out which constraint it is and project

virtual int	dim () const override
	space reserved in fit params, if has mother then it has tau

virtual int	type () const override
	type

virtual int	posIndex () const override
	position index in fit params

virtual int	tauIndex () const override
	tau index in fit params only if it has a mother More...

virtual int	momIndex () const override
	momentum index in fit params depending on whether it has a mother More...

virtual bool	hasEnergy () const override
	has energy in fitparams

virtual bool	hasPosition () const override
	has position index

virtual std::string	parname (int index) const override
	name

virtual void	addToConstraintList (constraintlist &list, int depth) const override
	add to constraint list

void	setMassConstraint (bool b)
	set mass constraint flag

virtual void	updateIndex (int &offset)
	this sets the index for momentum, position, etc. More...

const ParticleBase *	locate (Belle2::Particle *particle) const
	get particle base from basf2 particle

Belle2::Particle *	particle () const
	get basf2 particle

int	index () const
	get index

const ParticleBase *	mother () const
	getMother() / hasMother()

virtual ErrCode	projectGeoConstraint (const FitParams &, Projection &) const
	project geometrical constraint More...

virtual ErrCode	projectMassConstraintParticle (const FitParams &, Projection &) const
	project mass constraint using the particles parameters More...

virtual ErrCode	projectMassConstraintDaughters (const FitParams &, Projection &) const
	project mass constraint using the parameters of the daughters More...

virtual ErrCode	projectMassConstraint (const FitParams &, Projection &) const
	project mass constraint abstract

int	eneIndex () const
	get energy index

virtual double	chiSquare (const FitParams &) const
	get chi2

int	charge () const
	get charge

virtual ParticleBase *	addDaughter (Belle2::Particle *, const ConstraintConfiguration &config, bool forceFitAll=false)
	add daughter

virtual void	removeDaughter (const ParticleBase *pb)
	remove daughter

virtual void	retrieveIndexMap (indexmap &anindexmap) const
	get index map

void	setMother (const ParticleBase *m)
	set mother

void	collectVertexDaughters (std::vector< ParticleBase * > &particles, int posindex)
	get vertex daughters

virtual int	nFinalChargedCandidates () const
	number of charged candidates

Static Public Member Functions
static ParticleBase *	createParticle (Belle2::Particle particle, const ParticleBase mother, const ConstraintConfiguration &config, bool forceFitAll=false)
	create the according treeFitter particle obj for a basf2 particle type

static ParticleBase *	createOrigin (Belle2::Particle *daughter, const ConstraintConfiguration &config, bool forceFitAll)
	create a custom origin particle or a beamspot

Protected Types
typedef std::vector< ParticleBase * >	ParticleContainer
	just an alias

Protected Member Functions
ErrCode	initMomentum (FitParams &fitparams) const
	init momentum of *this and daughters

ErrCode	initTau (FitParams &par) const
	initialises tau as a length

void	setIndex (int i)
	set Index (in statevector)

Static Protected Member Functions
static bool	isAResonance (Belle2::Particle *particle)
	controls if a particle is treated as a resonance(lifetime=0) or a particle that has a finite lifetime. More...

Protected Attributes
Belle2::Particle *	m_particle
	pointer to framework type

const ParticleBase *	m_mother
	motherparticle

std::vector< ParticleBase * >	m_daughters
	daughter container

bool	m_isStronglyDecayingResonance
	decay length less than 1 micron

const ConstraintConfiguration *	m_config
	has all the constraint config

Static Private Member Functions
static bool	compTrkTransverseMomentum (const RecoTrack lhs, const RecoTrack rhs)
	compare transverse track momentum

Private Attributes
bool	m_massconstraint
	has mass constraint

bool	m_beamconstraint
	has beam constraint

bool	m_shares_vertex_with_mother
	shares vertex with mother, that means decay vertex = productionvertex

bool	m_geo_constraint
	use a geo metric constraint

bool	m_lifetimeconstraint
	has lifetime constraint

bool	m_isconversion
	is conversion

bool	m_automatic_vertex_constraining
	automatically figure out if mother and particle vertex should be the same and also add geometric constraints

int	m_index
	index

std::string	m_name
	name

Detailed Description

another unnecessary layer of abstraction

Definition at line 17 of file InternalParticle.h.

Member Function Documentation

◆ initMotherlessParticle()

ErrCode initMotherlessParticle ( FitParams & fitparams )

overridevirtual

init particle in case it has no mother

(0,0,0) is the best guess in any other case

Implements ParticleBase.

Reimplemented in Resonance.

Definition at line 89 of file InternalParticle.cc.

   {
     ErrCode status ;
     for (auto daughter : m_daughters) {
       status |= daughter->initMotherlessParticle(fitparams);
     }
  
     int posindex = posIndex();
     // FIXME update this and check if this position is already initialised
     // lower stream vertices might be better
     if (hasPosition()) {
       fitparams.getStateVector().segment(posindex, 3) = Eigen::Matrix<double, 3, 1>::Zero(3);
  
       std::vector<ParticleBase*> alldaughters;
       ParticleBase::collectVertexDaughters(alldaughters, posindex);
  
       std::vector<ParticleBase*> vtxdaughters;
  
       vector<RecoTrack*> trkdaughters;
       for (auto daughter : alldaughters) {
         if (daughter->type() == ParticleBase::TFParticleType::kRecoTrack) {
           trkdaughters.push_back(static_cast<RecoTrack*>(daughter));
         } else if (daughter->hasPosition()
                    && fitparams.getStateVector()(daughter->posIndex()) != 0) {
           vtxdaughters.push_back(daughter);
         }
       }
  
       if (trkdaughters.size() >= 2) {
  
         auto v0 = particle()->getV0();
         auto dummy_vertex = ROOT::Math::XYZVector(0, 0, 0);
  
         bool initWithV0 = false;
         if (v0 && v0->getFittedVertexPosition() != dummy_vertex) {
           auto part_dau1 = particle()->getDaughter(0);
           auto part_dau2 = particle()->getDaughter(1);
  
           auto recotrack_dau1 = std::find_if(trkdaughters.begin(), trkdaughters.end(),
           [&part_dau1](RecoTrack * rt) { return rt->particle()->getMdstArrayIndex() == part_dau1->getMdstArrayIndex(); });
           auto recotrack_dau2 = std::find_if(trkdaughters.begin(), trkdaughters.end(),
           [&part_dau2](RecoTrack * rt) { return rt->particle()->getMdstArrayIndex() == part_dau2->getMdstArrayIndex(); });
  
           if (recotrack_dau1 == trkdaughters.end() || recotrack_dau2 == trkdaughters.end()) {
             B2WARNING("V0 daughter particles do not match with RecoTracks.");
           } else {
             double X_V0(v0->getFittedVertexX()), Y_V0(v0->getFittedVertexY()), Z_V0(v0->getFittedVertexZ());
             fitparams.getStateVector()(posindex)     = X_V0;
             fitparams.getStateVector()(posindex + 1) = Y_V0;
             fitparams.getStateVector()(posindex + 2) = Z_V0;
  
             Belle2::Helix helix1 = v0->getTrackFitResults().first->getHelix();
             Belle2::Helix helix2 = v0->getTrackFitResults().second->getHelix();
  
             (*recotrack_dau1)->setFlightLength(helix1.getArcLength2DAtXY(X_V0, Y_V0));
             (*recotrack_dau2)->setFlightLength(helix2.getArcLength2DAtXY(X_V0, Y_V0));
  
             initWithV0 = true;
           }
         }
  
         if (!initWithV0) {
           std::sort(trkdaughters.begin(), trkdaughters.end(), compTrkTransverseMomentum);
  
           RecoTrack* dau1 = trkdaughters[0];
           RecoTrack* dau2 = trkdaughters[1];
  
           Belle2::Helix helix1 = dau1->particle()->getTrackFitResult()->getHelix();
           Belle2::Helix helix2 = dau2->particle()->getTrackFitResult()->getHelix();
  
           double flt1(0), flt2(0);
           Belle2::B2Vector3D v;
           HelixUtils::helixPoca(helix1, helix2, flt1, flt2, v, m_isconversion);
  
           fitparams.getStateVector()(posindex)     = v.X();
           fitparams.getStateVector()(posindex + 1) = v.Y();
           fitparams.getStateVector()(posindex + 2) = v.Z();
  
           dau1->setFlightLength(flt1);
           dau2->setFlightLength(flt2);
         }
  
       } else if (false && trkdaughters.size() + vtxdaughters.size() >= 2)  {
         // TODO switched off waiting for refactoring of init1 and init2 functions (does not affect performance)
       } else if (mother() && mother()->posIndex() >= 0) {
         const int posindexmother = mother()->posIndex();
         const int dim = m_config->m_originDimension; //TODO access mother
         fitparams.getStateVector().segment(posindex, dim) = fitparams.getStateVector().segment(posindexmother, dim);
       } else {
         fitparams.getStateVector().segment(posindex, 3) = Eigen::Matrix<double, 1, 3>::Zero(3);
       }
     }
  
     for (auto daughter :  m_daughters) {
       daughter->initParticleWithMother(fitparams);
     }
  
     initMomentum(fitparams);
     return status;
   }

◆ isAResonance()

bool isAResonance ( Belle2::Particle * particle )

staticprotectedinherited

controls if a particle is treated as a resonance(lifetime=0) or a particle that has a finite lifetime.

A finite life time means it will register a geo constraint for this particle

Definition at line 134 of file ParticleBase.cc.

◆ momIndex()

int momIndex ( ) const

overridevirtual

momentum index in fit params depending on whether it has a mother

indexing in { x, y, z, tau, px, py, pz, E } but tau is not existing for all InternalParticles

Reimplemented from ParticleBase.

Reimplemented in Resonance.

Definition at line 353 of file InternalParticle.cc.

◆ projectGeoConstraint()

ErrCode projectGeoConstraint	(	const FitParams &	fitparams,
		Projection &	p
	)		const

virtualinherited

project geometrical constraint

the direction of the momentum is very well known from the kinematic constraints that is why we do not extract the distance as a vector here

Definition at line 249 of file ParticleBase.cc.

◆ projectMassConstraintDaughters()

ErrCode projectMassConstraintDaughters	(	const FitParams &	fitparams,
		Projection &	p
	)		const

virtualinherited

project mass constraint using the parameters of the daughters

be aware that the signs here are important E-|p|-m extracts a negative mass and messes with the momentum !

Definition at line 314 of file ParticleBase.cc.

◆ projectMassConstraintParticle()

ErrCode projectMassConstraintParticle	(	const FitParams &	fitparams,
		Projection &	p
	)		const

virtualinherited

project mass constraint using the particles parameters

be aware that the signs here are important E-|p|-m extracts a negative mass and messes with the momentum !

Definition at line 375 of file ParticleBase.cc.

◆ tauIndex()

int tauIndex ( ) const

overridevirtual

tau index in fit params only if it has a mother

only exists if particle is geo constraint and has a mother

Reimplemented from ParticleBase.

Reimplemented in Resonance.

Definition at line 347 of file InternalParticle.cc.

◆ updateIndex()

void updateIndex ( int & offset )

virtualinherited

this sets the index for momentum, position, etc.

in the statevector

Definition at line 84 of file ParticleBase.cc.

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

analysis/VertexFitting/TreeFitter/include/InternalParticle.h
analysis/VertexFitting/TreeFitter/src/InternalParticle.cc

Public Types

Public Member Functions

Static Public Member Functions

Protected Types

Protected Member Functions

Static Protected Member Functions

Protected Attributes

Static Private Member Functions

Private Attributes

Detailed Description

Member Function Documentation

◆ initMotherlessParticle()

◆ isAResonance()

◆ momIndex()

◆ projectGeoConstraint()

◆ projectMassConstraintDaughters()

◆ projectMassConstraintParticle()

◆ tauIndex()

◆ updateIndex()