another unneccessary layer of abstraction More...

#include <InternalParticle.h>

Inheritance diagram for InternalParticle:

Collaboration diagram for InternalParticle:

[legend]

Public Types
enum	TFParticleType { kInteractionPoint, kOrigin, kRecoComposite, kRecoResonance, kInternalParticle, kRecoTrack, kResonance, kRecoPhoton, kRecoKlong, kMissingParticle, kFeedthroughParticle, kInternalTrack }
	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

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

double	phidomain (const double)
	rotate in positive phi domain

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

const std::string &	name () const
	get name of the particle

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

double	pdgMass () const
	get pdg mass

double	pdgWidth () const
	get pdg width

double	pdgLifeTime () const
	get pdg lifetime

double	pdgTime () const
	get Tau

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

void	setParticle (Belle2::Particle *particle)
	set the relation to basf2 particle type

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 double	pdgLifeTime (Belle2::Particle *particle)
	get pdg lifetime

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

static double	bFieldOverC ()
	Bz/c

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 cosntraint

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

const double	m_pdgMass
	pdg mass

double	m_pdgWidth
	particle width

const double	m_pdgLifeTime
	lifetime in cm

int	m_charge
	charge

std::string	m_name
	name

Detailed Description

another unneccessary layer of abstraction

Definition at line 26 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 86 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);
  
       if (fitparams.getStateVector()(posindex)  == 0 &&
           fitparams.getStateVector()(posindex + 1) == 0 &&
           fitparams.getStateVector()(posindex + 2) == 0) {
  
         TVector3 vtx = particle()->getVertex();
         if (vtx.Mag()) { //if it's not zero
           fitparams.getStateVector()(posindex) = vtx.X();
           fitparams.getStateVector()(posindex + 1) = vtx.Y();
           fitparams.getStateVector()(posindex + 2) = vtx.Z();
         } else {
  
           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);
             }
           }
  
           TVector3 v;
  
           if (trkdaughters.size() >= 2) {
             std::sort(trkdaughters.begin(), trkdaughters.end(), compTrkTransverseMomentum);
  
             RecoTrack* dau1 = trkdaughters[0];
             RecoTrack* dau2 = trkdaughters[1];
  
             Belle2::Helix helix1 = dau1->particle()->getTrack()->getTrackFitResultWithClosestMass(Belle2::Const::ChargedStable(std::abs(
                                      dau1->particle()->getPDGCode())))->getHelix();
             Belle2::Helix helix2 = dau2->particle()->getTrack()->getTrackFitResultWithClosestMass(Belle2::Const::ChargedStable(std::abs(
                                      dau2->particle()->getPDGCode())))->getHelix();
  
             double flt1(0), flt2(0);
             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 acess 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 235 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 313 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 359 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 435 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 496 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 cosntraint and has a mother

Reimplemented from ParticleBase.

Reimplemented in Resonance.

Definition at line 307 of file InternalParticle.cc.

◆ updateIndex()

void updateIndex ( int & offset )

virtualinherited

this sets the index for momentum, position, etc.

in the statevector

Definition at line 138 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()