this class does a lot of stuff: Build decaytree structure allowing to index particles and handle the filtering of constraints across the tree More...

#include <DecayChain.h>

Collaboration diagram for DecayChain:

[legend]

Public Member Functions
	DecayChain ()
	empty constructor

	DecayChain (Belle2::Particle *bc, const ConstraintConfiguration &config, bool forceFitAll=false)
	constructor

	~DecayChain ()
	destructor

ErrCode	initialize (FitParams &par)
	initialize the chain

ErrCode	filter (FitParams &par)
	filter down the chain

ErrCode	filterWithReference (FitParams &par, const FitParams &ref)
	filter with respect to a previous iteration for better stability

int	dim () const
	get dimension

void	initConstraintList ()
	init contraintlist

void	removeConstraintFromList ()
	remove constraints from list

double	chiSquare (const FitParams &par) const
	get the chi2 for the head of the chain

ParticleBase *	mother ()
	get mother

const ParticleBase *	cand ()
	get candidate

const ParticleBase *	mother () const
	get mother

const ParticleBase *	locate (Belle2::Particle *bc) const
	convert Belle2::particle into particle base(fitter base particle)

int	index (Belle2::Particle *bc) const
	get the particle index

int	posIndex (Belle2::Particle *bc) const
	get the vertex index of the particle in state vector

int	momIndex (Belle2::Particle *bc) const
	get momentum index of the particle in the state vector

int	tauIndex (Belle2::Particle *bc) const
	get tau (i.e.

int	momIndex () const
	!NOT IMPLEMENTED

Private Types
typedef std::map< Belle2::Particle , const ParticleBase >	ParticleMap
	typedef for a map of a particle to a TreeFitter::ParticleBase

Private Attributes
int	m_dim
	the dimension of constraint

ParticleBase *	m_headOfChain
	head of decay tree

const ParticleBase *	m_cand
	fit candidate (not same to mother in case of bs/be constraint)

ParticleBase::constraintlist	m_constraintlist
	list of constraints

ParticleMap	m_particleMap
	the map from Belle2::Particles to TreeFitter::ParticleBase

const bool	m_isOwner
	internal class member to check if we own the chain

const ConstraintConfiguration	m_config
	config container

Detailed Description

this class does a lot of stuff: Build decaytree structure allowing to index particles and handle the filtering of constraints across the tree

Definition at line 23 of file DecayChain.h.

Member Typedef Documentation

◆ ParticleMap

typedef std::map<Belle2::Particle*, const ParticleBase*> ParticleMap

private

typedef for a map of a particle to a TreeFitter::ParticleBase

Definition at line 99 of file DecayChain.h.

Constructor & Destructor Documentation

◆ DecayChain() [1/2]

DecayChain ( )

inline

empty constructor

Definition at line 28 of file DecayChain.h.

28: m_dim(0), m_headOfChain(0), m_cand(0), m_isOwner(true) {} //Default constructor (actually never used)

TreeFitter::DecayChain::m_dim

int m_dim

the dimension of constraint

Definition: DecayChain.h:89

TreeFitter::DecayChain::m_headOfChain

ParticleBase * m_headOfChain

head of decay tree

Definition: DecayChain.h:91

TreeFitter::DecayChain::m_isOwner

const bool m_isOwner

internal class member to check if we own the chain

Definition: DecayChain.h:105

TreeFitter::DecayChain::m_cand

const ParticleBase * m_cand

fit candidate (not same to mother in case of bs/be constraint)

Definition: DecayChain.h:93

◆ DecayChain() [2/2]

DecayChain	(	Belle2::Particle *	bc,
		const ConstraintConfiguration &	config,
		bool	forceFitAll = `false`
	)

constructor

Definition at line 21 of file DecayChain.cc.

                          :
    m_dim(0),
    m_headOfChain(nullptr),
    m_isOwner(true),
    m_config(config)
  {
 
    if (config.m_ipConstraint && config.m_customOrigin) {
      B2FATAL("Setup error. Can't have both custom origin and ip constraint.");
    }
    config.m_headOfTreePDG = std::abs(particle->getPDGCode());
    if (config.m_ipConstraint || config.m_customOrigin) {
      m_headOfChain = ParticleBase::createOrigin(particle, config, forceFitAll);
    } else {
      m_headOfChain = ParticleBase::createParticle(particle, nullptr, config, forceFitAll);
    }
 
    m_headOfChain->updateIndex(m_dim);
 
    m_cand = locate(particle);
  }

◆ ~DecayChain()

~DecayChain ( )

destructor

Definition at line 46 of file DecayChain.cc.

  {
    if (m_headOfChain && m_isOwner) {
      delete m_headOfChain;
    }
  }

Member Function Documentation

◆ cand()

const ParticleBase * cand ( )

inline

get candidate

Definition at line 64 of file DecayChain.h.

64{ return m_cand ; }

◆ chiSquare()

double chiSquare ( const FitParams & par ) const

get the chi2 for the head of the chain

Definition at line 106 of file DecayChain.cc.

  {
    return m_headOfChain->chiSquare(par);
  }

◆ dim()

int dim ( ) const

inline

get dimension

Definition at line 49 of file DecayChain.h.

49{ return m_dim;}

◆ filter()

ErrCode filter ( FitParams & par )

filter down the chain

Definition at line 81 of file DecayChain.cc.

  {
    ErrCode status;
    par.resetCovariance();
    status |= m_headOfChain->initCovariance(par);
    par.resetChiSquare();
    for (auto constraint : m_constraintlist) {
      status |= constraint.filter(par);
    }
    return status;
  }

◆ filterWithReference()

ErrCode filterWithReference	(	FitParams &	par,
		const FitParams &	ref
	)

filter with respect to a previous iteration for better stability

Definition at line 93 of file DecayChain.cc.

  {
    ErrCode status;
    par.resetCovariance();
    status |= m_headOfChain->initCovariance(par);
    par.resetChiSquare();
    for (auto constraint : m_constraintlist) {
      status |= constraint.filterWithReference(par, ref);
    }
    return status;
  }

◆ index()

int index ( Belle2::Particle * bc ) const

get the particle index

Definition at line 130 of file DecayChain.cc.

  {
    const ParticleBase* base = locate(particle);
    if (base) {
      return base->index();
    }
    return -1;
  }

◆ initConstraintList()

void initConstraintList ( )

init contraintlist

Definition at line 53 of file DecayChain.cc.

  {
    m_constraintlist.clear();
    m_headOfChain->addToConstraintList(m_constraintlist, 0);
    removeConstraintFromList();
    std::sort(m_constraintlist.begin(), m_constraintlist.end());
  }

◆ initialize()

ErrCode initialize ( FitParams & par )

initialize the chain

Definition at line 70 of file DecayChain.cc.

  {
    ErrCode status;
    par.resetStateVector();
    status |= m_headOfChain->initMotherlessParticle(par);
    par.resetCovariance();
    status |= m_headOfChain->initCovariance(par);
    initConstraintList();
    return status;
  }

◆ locate()

const ParticleBase * locate ( Belle2::Particle * bc ) const

convert Belle2::particle into particle base(fitter base particle)

Definition at line 111 of file DecayChain.cc.

  {
    const ParticleBase* rc(nullptr);
    const auto mapRow = m_particleMap.find(particle) ;
 
    if (mapRow == m_particleMap.end()) {
      //take head of chain and recursively find particle in it
      rc = m_headOfChain->locate(particle);
 
      if (rc && rc->particle()) {
        const_cast<DecayChain*>(this)->m_particleMap[rc->particle()] = rc;
      }
    } else {
      //only used for "head of tree"
      rc = mapRow->second;// (B2::Particle, Particlebase)
    }
    return rc;
  }

◆ momIndex()

int momIndex ( Belle2::Particle * bc ) const

get momentum index of the particle in the state vector

Definition at line 148 of file DecayChain.cc.

  {
    const ParticleBase* base = locate(particle);
    if (base) {
      return base->momIndex();
    }
    return -1;
  }

◆ mother() [1/2]

ParticleBase * mother ( )

inline

get mother

Definition at line 61 of file DecayChain.h.

61{ return m_headOfChain ; }

◆ mother() [2/2]

const ParticleBase * mother ( ) const

inline

get mother

Definition at line 67 of file DecayChain.h.

67{ return m_headOfChain ; }

◆ posIndex()

int posIndex ( Belle2::Particle * bc ) const

get the vertex index of the particle in state vector

Definition at line 139 of file DecayChain.cc.

  {
    const ParticleBase* base = locate(particle);
    if (base) {
      return base->posIndex();
    }
    return -1;
  }

◆ removeConstraintFromList()

void removeConstraintFromList ( )

remove constraints from list

Definition at line 61 of file DecayChain.cc.

  {
    for (auto removeConstraint : m_config.m_removeConstraintList) {
      m_constraintlist.erase(std::remove_if(m_constraintlist.begin(), m_constraintlist.end(),
      [&](const Constraint & constraint) { return constraint.name() == removeConstraint ;}),
      m_constraintlist.end());
    }
  }

◆ tauIndex()

int tauIndex ( Belle2::Particle * bc ) const

get tau (i.e.

decay lifetime) index of the particle in the state vector

Definition at line 157 of file DecayChain.cc.

  {
    const ParticleBase* base = locate(particle);
    if (base) {
      return base->tauIndex();
    }
    return -1;
  }

Member Data Documentation

◆ m_cand

const ParticleBase* m_cand

private

fit candidate (not same to mother in case of bs/be constraint)

Definition at line 93 of file DecayChain.h.

◆ m_config

const ConstraintConfiguration m_config

private

config container

Definition at line 108 of file DecayChain.h.

◆ m_constraintlist

ParticleBase::constraintlist m_constraintlist

private

list of constraints

Definition at line 96 of file DecayChain.h.

◆ m_dim

int m_dim

mutableprivate

the dimension of constraint

Definition at line 89 of file DecayChain.h.

◆ m_headOfChain

ParticleBase* m_headOfChain

mutableprivate

head of decay tree

Definition at line 91 of file DecayChain.h.

◆ m_isOwner

const bool m_isOwner

private

internal class member to check if we own the chain

Definition at line 105 of file DecayChain.h.

◆ m_particleMap

ParticleMap m_particleMap

private

the map from Belle2::Particles to TreeFitter::ParticleBase

Definition at line 102 of file DecayChain.h.

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

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

Public Member Functions

Private Types

Private Attributes

Detailed Description

Member Typedef Documentation

◆ ParticleMap

Constructor & Destructor Documentation

◆ DecayChain() [1/2]

◆ DecayChain() [2/2]

◆ ~DecayChain()

Member Function Documentation

◆ cand()

◆ chiSquare()

◆ dim()

◆ filter()

◆ filterWithReference()

◆ index()

◆ initConstraintList()

◆ initialize()

◆ locate()

◆ momIndex()

◆ mother() [1/2]

◆ mother() [2/2]

◆ posIndex()

◆ removeConstraintFromList()

◆ tauIndex()

Member Data Documentation

◆ m_cand

◆ m_config

◆ m_constraintlist

◆ m_dim

◆ m_headOfChain

◆ m_isOwner

◆ m_particleMap