MassPointingVertexFitKFit is a derived class from KFitBase It performs a kinematical fit with three constraints: the invariant mass is constrained to its PDG value, the tracks are forced to a common vertex, and the momentum vector is forced to point to an externally provided space point. More...

#include <MassPointingVertexFitKFit.h>

Inheritance diagram for MassPointingVertexFitKFit:

[legend]

Collaboration diagram for MassPointingVertexFitKFit:

[legend]

Public Member Functions
	MassPointingVertexFitKFit (void)
	Construct an object with no argument.

	~MassPointingVertexFitKFit (void)
	Destruct the object.

enum KFitError::ECode	setInitialVertex (const HepPoint3D &v)
	Set an initial vertex point for the 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.

enum KFitError::ECode	fixMass (void)
	Tell the object to fix the last added track property at the invariant mass.

enum KFitError::ECode	unfixMass (void)
	Tell the object to unfix the last added track property at the invariant mass.

enum KFitError::ECode	setCorrelation (const CLHEP::HepMatrix &m) override
	Set a correlation matrix.

enum KFitError::ECode	setZeroCorrelation (void) override
	Indicate no correlation between tracks.

const HepPoint3D	getVertex (const int flag=KFitConst::kAfterFit) const
	Get a vertex position.

const CLHEP::HepSymMatrix	getVertexError (void) const
	Get a fitted vertex error matrix.

double	getInvariantMass (void) const
	Get an invariant mass.

double	getCHIsq (void) const override
	Get a chi-square of the fit.

const CLHEP::HepMatrix	getTrackVertexError (const int id) const
	Get a vertex error matrix of the track.

double	getTrackCHIsq (const int id) const override
	Get a chi-square of the track.

const CLHEP::HepMatrix	getCorrelation (const int id1, const int id2, const int flag=KFitConst::kAfterFit) const override
	Get a correlation matrix between two tracks.

enum KFitError::ECode	doFit (void)
	Perform a mass-vertex-pointing constraint fit.

enum KFitError::ECode	updateMother (Particle *mother)
	Update mother particle.

enum KFitError::ECode	addTrack (const KFitTrack &kp)
	Add a track to the fitter object.

enum KFitError::ECode	addTrack (const CLHEP::HepLorentzVector &p, const HepPoint3D &x, const CLHEP::HepSymMatrix &e, const double q)
	Add a track to the fitter object with specifying its momentum, position, error matrix, and charge.

enum KFitError::ECode	addParticle (const Particle *particle)
	Add a particle to the fitter.

enum KFitError::ECode	setMagneticField (const double mf)
	Change a magnetic field from the default value KFitConst::kDefaultMagneticField.

enum KFitError::ECode	getErrorCode (void) const
	Get a code of the last error.

int	getTrackCount (void) const
	Get the number of added tracks.

virtual int	getNDF (void) const
	Get an NDF of the fit.

double	getMagneticField (void) const
	Get a magnetic field.

const CLHEP::HepLorentzVector	getTrackMomentum (const int id) const
	Get a Lorentz vector of the track.

const HepPoint3D	getTrackPosition (const int id) const
	Get a position of the track.

const CLHEP::HepSymMatrix	getTrackError (const int id) const
	Get an error matrix of the track.

const KFitTrack	getTrack (const int id) const
	Get a specified track object.

bool	isFitted (void) const
	Return false if fit is not performed yet or performed fit is failed; otherwise true.

Protected Member Functions
const CLHEP::HepSymMatrix	makeError1 (const CLHEP::HepLorentzVector &p, const CLHEP::HepMatrix &e) const
	Rebuild an error matrix from a Lorentz vector and an error matrix.

const CLHEP::HepMatrix	makeError1 (const CLHEP::HepLorentzVector &p1, const CLHEP::HepLorentzVector &p2, const CLHEP::HepMatrix &e) const
	Rebuild an error matrix from a pair of Lorentz vectors and an error matrix.

const CLHEP::HepMatrix	makeError2 (const CLHEP::HepLorentzVector &p, const CLHEP::HepMatrix &e) const
	Rebuild an error matrix from a Lorentz vector and an error matrix.

const CLHEP::HepSymMatrix	makeError3 (const CLHEP::HepLorentzVector &p, const CLHEP::HepMatrix &e, const bool is_fix_mass) const
	Rebuild an error matrix from a Lorentz vector and an error matrix.

const CLHEP::HepMatrix	makeError3 (const CLHEP::HepLorentzVector &p1, const CLHEP::HepLorentzVector &p2, const CLHEP::HepMatrix &e, const bool is_fix_mass1, const bool is_fix_mass2) const
	Rebuild an error matrix from a pair of Lorentz vectors and an error matrix.

const CLHEP::HepMatrix	makeError4 (const CLHEP::HepLorentzVector &p, const CLHEP::HepMatrix &e) const
	Rebuild an error matrix from a Lorentz vector and an error matrix.

enum KFitError::ECode	doFit1 (void)
	Perform a fit (used in MassFitKFit::doFit()).

enum KFitError::ECode	doFit2 (void)
	Perform a fit (used in VertexFitKFit::doFit() and MassVertexFitKFit::doFit()).

bool	isTrackIDInRange (const int id) const
	Check if the id is in the range.

bool	isNonZeroEnergy (const CLHEP::HepLorentzVector &p) const
	Check if the energy is non-zero.

Protected Attributes
enum KFitError::ECode	m_ErrorCode
	Error code.

bool	m_FlagFitted
	Flag to indicate if the fit is performed and succeeded.

std::vector< KFitTrack >	m_Tracks
	Container of input tracks.

std::vector< CLHEP::HepMatrix >	m_BeforeCorrelation
	Container of input correlation matrices.

CLHEP::HepSymMatrix	m_V_al_0
	See J.Tanaka Ph.D (2001) p137 for definition.

CLHEP::HepMatrix	m_al_0
	See J.Tanaka Ph.D (2001) p136 for definition.

CLHEP::HepMatrix	m_al_1
	See J.Tanaka Ph.D (2001) p136 for definition.

CLHEP::HepMatrix	m_al_a
	See J.Tanaka Ph.D (2001) p137 for definition.

CLHEP::HepMatrix	m_property
	Container of charges and masses.

CLHEP::HepMatrix	m_D
	See J.Tanaka Ph.D (2001) p137 for definition.

CLHEP::HepMatrix	m_d
	See J.Tanaka Ph.D (2001) p137 for definition.

CLHEP::HepMatrix	m_V_D
	See J.Tanaka Ph.D (2001) p138 for definition.

CLHEP::HepMatrix	m_V_al_1
	See J.Tanaka Ph.D (2001) p138 for definition.

CLHEP::HepMatrix	m_lam
	See J.Tanaka Ph.D (2001) p137 for definition.

CLHEP::HepMatrix	m_E
	See J.Tanaka Ph.D (2001) p137 for definition.

CLHEP::HepMatrix	m_V_E
	See J.Tanaka Ph.D (2001) p138 for definition.

CLHEP::HepMatrix	m_lam0
	See J.Tanaka Ph.D (2001) p138 for definition.

CLHEP::HepMatrix	m_v
	See J.Tanaka Ph.D (2001) p137 for definition.

CLHEP::HepMatrix	m_v_a
	See J.Tanaka Ph.D (2001) p137 for definition.

CLHEP::HepMatrix	m_V_Dt
	See J.Tanaka Ph.D (2001) p138 for definition.

CLHEP::HepMatrix	m_Cov_v_al_1
	See J.Tanaka Ph.D (2001) p137 for definition.

int	m_NDF
	NDF of the fit.

double	m_CHIsq
	chi-square of the fit.

int	m_TrackCount
	Number of tracks.

int	m_NecessaryTrackCount
	Number needed tracks to perform fit.

bool	m_FlagCorrelation
	Flag whether a correlation among tracks exists.

bool	m_FlagOverIteration
	Flag whether the iteration count exceeds the limit.

double	m_MagneticField
	Magnetic field.

Private Member Functions
enum KFitError::ECode	prepareInputMatrix (void) override
	Build grand matrices for minimum search from input-track properties.

enum KFitError::ECode	prepareInputSubMatrix (void) override
	Build sub-matrices for minimum search from input-track properties.

enum KFitError::ECode	prepareCorrelation (void) override
	Build a grand correlation matrix from input-track properties.

enum KFitError::ECode	prepareOutputMatrix (void) override
	Build an output error matrix.

enum KFitError::ECode	makeCoreMatrix (void) override
	Build matrices using the kinematical constraint.

enum KFitError::ECode	calculateNDF (void) override
	Calculate an NDF of the fit.

Private Attributes
HepPoint3D	m_BeforeVertex
	Vertex position before the fit.

HepPoint3D	m_AfterVertex
	Vertex position after the fit.

CLHEP::HepSymMatrix	m_AfterVertexError
	Vertex error matrix after the fit.

std::vector< CLHEP::HepMatrix >	m_AfterTrackVertexError
	array of vertex error matrices after the fit.

HepPoint3D	m_ProductionVertex
	Production vertex position.

double	m_InvariantMass
	Invariant mass.

std::vector< int >	m_IsFixMass
	Array of flags whether the track property is fixed at the mass.

Detailed Description

MassPointingVertexFitKFit is a derived class from KFitBase It performs a kinematical fit with three constraints: the invariant mass is constrained to its PDG value, the tracks are forced to a common vertex, and the momentum vector is forced to point to an externally provided space point.

Definition at line 30 of file MassPointingVertexFitKFit.h.

Constructor & Destructor Documentation

◆ MassPointingVertexFitKFit()

MassPointingVertexFitKFit ( void )

Construct an object with no argument.

Definition at line 21 of file MassPointingVertexFitKFit.cc.

                                                    :
  m_BeforeVertex(HepPoint3D(0., 0., 0.)),
  m_AfterVertexError(HepSymMatrix(3, 0)),
  m_ProductionVertex(HepPoint3D(0., 0., 0.))
{
  m_FlagFitted = false;
  m_NecessaryTrackCount = 2;
  m_V_E = HepMatrix(3, 3, 0);
  m_v   = HepMatrix(3, 1, 0);
  m_v_a = HepMatrix(3, 1, 0);
  m_InvariantMass = -1.0;
}

Member Function Documentation

◆ addParticle()

enum KFitError::ECode addParticle ( const Particle * particle )

inherited

Add a particle to the fitter.

The function gets track parameters from the Particle dataobject and calls addTrack().

Parameters

[in] particle Particle.

Returns: error code (zero if success)

Definition at line 59 of file KFitBase.cc.

{
  return addTrack(
           ROOTToCLHEP::getHepLorentzVector(particle->get4Vector()),
           ROOTToCLHEP::getPoint3D(particle->getVertex()),
           ROOTToCLHEP::getHepSymMatrix(particle->getMomentumVertexErrorMatrix()),
           particle->getCharge());
}

◆ addTrack() [1/2]

enum KFitError::ECode addTrack	(	const CLHEP::HepLorentzVector &	p,
		const HepPoint3D &	x,
		const CLHEP::HepSymMatrix &	e,
		const double	q
	)

inherited

Add a track to the fitter object with specifying its momentum, position, error matrix, and charge.

This function internally calls addTrack(const KFitTrack &kp).

Parameters

p	Lorentz vector of the track
x	position of the track
e	(7x7) error matrix of the track
q	charge of the track

Returns: error code (zero if success)

Definition at line 47 of file KFitBase.cc.

                                                                                                                  {
  if (e.num_row() != KFitConst::kNumber7)
  {
    m_ErrorCode = KFitError::kBadMatrixSize;
    KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
    return m_ErrorCode;
  }
 
  return this->addTrack(KFitTrack(p, x, e, q));
}

◆ addTrack() [2/2]

enum KFitError::ECode addTrack ( const KFitTrack & kp )

inherited

Add a track to the fitter object.

Parameters

kp	object of the track

Returns: error code (zero if success)

Definition at line 38 of file KFitBase.cc.

                                     {
  m_Tracks.push_back(p);
  m_TrackCount = m_Tracks.size();
 
  return m_ErrorCode = KFitError::kNoError;
}

◆ calculateNDF()

enum KFitError::ECode calculateNDF ( void )

overrideprivatevirtual

Calculate an NDF of the fit.

Returns: error code (zero if success)

Implements KFitBase.

Definition at line 597 of file MassPointingVertexFitKFit.cc.

                                        {
  m_NDF = 2 * m_TrackCount - 3 + 3;
 
  return m_ErrorCode = KFitError::kNoError;
}

◆ doFit()

enum KFitError::ECode doFit ( void )

Perform a mass-vertex-pointing constraint fit.

Returns: error code (zero if success)

Definition at line 202 of file MassPointingVertexFitKFit.cc.

                                 {
  return KFitBase::doFit2();
}

◆ doFit1()

enum KFitError::ECode doFit1 ( void )

protectedinherited

Perform a fit (used in MassFitKFit::doFit()).

Returns: error code (zero if success)

Definition at line 502 of file KFitBase.cc.

                 {
  if (m_ErrorCode != KFitError::kNoError) return m_ErrorCode;
 
  if (m_TrackCount < m_NecessaryTrackCount)
  {
    m_ErrorCode = KFitError::kBadTrackSize;
    KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
    return m_ErrorCode;
  }
 
  if (prepareInputMatrix() != KFitError::kNoError) return m_ErrorCode;
  if (calculateNDF() != KFitError::kNoError)       return m_ErrorCode;
 
 
  double chisq = 0;
  double tmp_chisq = KFitConst::kInitialCHIsq;
  int err_inverse = 0;
 
  HepMatrix tmp_al_1(m_al_1);
  HepMatrix tmp_V_al_1(m_V_al_1);
 
  m_al_a = m_al_0;
  HepMatrix tmp_al_a(m_al_a);
 
 
  for (int i = 0; i < KFitConst::kMaxIterationCount; i++)
  {
    if (makeCoreMatrix() != KFitError::kNoError) return m_ErrorCode;
 
    m_V_D = (m_V_al_0.similarity(m_D)).inverse(err_inverse);
    if (err_inverse != 0) {
      m_ErrorCode = KFitError::kCannotGetMatrixInverse;
      return m_ErrorCode;
    }
 
    m_lam    = m_V_D * (m_D * (m_al_0 - m_al_1) + m_d);
    chisq    = ((m_lam.T()) * (m_D * (m_al_0 - m_al_1) + m_d))(1, 1);
    m_al_1   = m_al_0 - m_V_al_0 * (m_D.T()) * m_lam;
    m_V_al_1 = m_V_al_0 - m_V_al_0 * (m_D.T()) * m_V_D * m_D * m_V_al_0;
 
    if (tmp_chisq <= chisq) {
      if (i == 0) {
        m_ErrorCode = KFitError::kBadInitialCHIsq;
        return m_ErrorCode;
      } else {
        chisq    = tmp_chisq;
        m_al_1   = tmp_al_1;
        m_al_a   = tmp_al_a;
        m_V_al_1 = tmp_V_al_1;
        break;
      }
    } else {
      tmp_chisq  = chisq;
      tmp_al_a   = tmp_al_1;
      tmp_al_1   = m_al_1;
      tmp_V_al_1 = m_V_al_1;
      if (i == KFitConst::kMaxIterationCount - 1) {
        m_al_a = tmp_al_1;
        m_FlagOverIteration = true;
      }
    }
  }
 
  if (m_ErrorCode != KFitError::kNoError) return m_ErrorCode;
 
  if (prepareOutputMatrix() != KFitError::kNoError) return m_ErrorCode;
 
  m_CHIsq = chisq;
 
  m_FlagFitted = true;
 
  return m_ErrorCode = KFitError::kNoError;
}

◆ doFit2()

enum KFitError::ECode doFit2 ( void )

protectedinherited

Perform a fit (used in VertexFitKFit::doFit() and MassVertexFitKFit::doFit()).

Returns: error code (zero if success)

Definition at line 578 of file KFitBase.cc.

                 {
  if (m_ErrorCode != KFitError::kNoError) return m_ErrorCode;
 
  if (m_TrackCount < m_NecessaryTrackCount)
  {
    m_ErrorCode = KFitError::kBadTrackSize;
    KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
    return m_ErrorCode;
  }
 
  if (prepareInputMatrix() != KFitError::kNoError) return m_ErrorCode;
  if (calculateNDF() != KFitError::kNoError)       return m_ErrorCode;
 
 
  double chisq = 0;
  double tmp2_chisq = KFitConst::kInitialCHIsq;
  int err_inverse = 0;
 
  m_al_a = m_al_0;
  HepMatrix tmp_al_a(m_al_a);
 
  HepMatrix tmp_D(m_D), tmp_E(m_E);
  HepMatrix tmp_V_D(m_V_D), tmp_V_E(m_V_E);
  HepMatrix tmp_lam0(m_lam0), tmp_v_a(m_v_a);
 
  HepMatrix tmp2_D(m_D), tmp2_E(m_E);
  HepMatrix tmp2_V_D(m_V_D), tmp2_V_E(m_V_E);
  HepMatrix tmp2_lam0(m_lam0), tmp2_v_a(m_v_a), tmp2_v(m_v_a);
 
 
  for (int j = 0; j < KFitConst::kMaxIterationCount; j++)   // j'th loop start
  {
 
    double tmp_chisq = KFitConst::kInitialCHIsq;
 
    for (int i = 0; i < KFitConst::kMaxIterationCount; i++) { // i'th loop start
 
      if (prepareInputSubMatrix() != KFitError::kNoError) return m_ErrorCode;
      if (makeCoreMatrix() != KFitError::kNoError)        return m_ErrorCode;
 
      m_V_D = (m_V_al_0.similarity(m_D)).inverse(err_inverse);
      if (err_inverse) {
        m_ErrorCode = KFitError::kCannotGetMatrixInverse;
        return m_ErrorCode;
      }
 
      m_V_E = ((m_E.T()) * m_V_D * m_E).inverse(err_inverse);
      if (err_inverse) {
        m_ErrorCode = KFitError::kCannotGetMatrixInverse;
        return m_ErrorCode;
      }
      m_lam0 = m_V_D * (m_D * (m_al_0 - m_al_1) + m_d);
      chisq  = ((m_lam0.T()) * (m_D * (m_al_0 - m_al_1) + m_E * (m_v - m_v_a) + m_d))(1, 1);
      m_v_a  = m_v_a - m_V_E * (m_E.T()) * m_lam0;
 
      if (tmp_chisq <= chisq) {
        if (i == 0) {
          m_ErrorCode = KFitError::kBadInitialCHIsq;
          return m_ErrorCode;
        } else {
          chisq   = tmp_chisq;
          m_v_a   = tmp_v_a;
          m_V_E   = tmp_V_E;
          m_V_D   = tmp_V_D;
          m_lam0  = tmp_lam0;
          m_E     = tmp_E;
          m_D     = tmp_D;
          break;
        }
      } else {
        tmp_chisq = chisq;
        tmp_v_a   = m_v_a;
        tmp_V_E   = m_V_E;
        tmp_V_D   = m_V_D;
        tmp_lam0  = m_lam0;
        tmp_E     = m_E;
        tmp_D     = m_D;
        if (i == KFitConst::kMaxIterationCount - 1) {
          m_FlagOverIteration = true;
        }
      }
    } // i'th loop over
 
 
    m_al_a   = m_al_1;
    m_lam    = m_lam0 - m_V_D * m_E * m_V_E * (m_E.T()) * m_lam0;
    m_al_1   = m_al_0 - m_V_al_0 * (m_D.T()) * m_lam;
 
    if (j == 0) {
 
      tmp2_chisq = chisq;
      tmp2_v_a   = m_v_a;
      tmp2_v     = m_v;
      tmp2_V_E   = m_V_E;
      tmp2_V_D   = m_V_D;
      tmp2_lam0  = m_lam0;
      tmp2_E     = m_E;
      tmp2_D     = m_D;
      tmp_al_a   = m_al_a;
 
    } else {
 
      if (tmp2_chisq <= chisq) {
        chisq   = tmp2_chisq;
        m_v_a   = tmp2_v_a;
        m_v     = tmp2_v;
        m_V_E   = tmp2_V_E;
        m_V_D   = tmp2_V_D;
        m_lam0  = tmp2_lam0;
        m_E     = tmp2_E;
        m_D     = tmp2_D;
        m_al_a  = tmp_al_a;
        break;
      } else {
        tmp2_chisq = chisq;
        tmp2_v_a   = m_v_a;
        tmp2_v     = m_v;
        tmp2_V_E   = m_V_E;
        tmp2_V_D   = m_V_D;
        tmp2_lam0  = m_lam0;
        tmp2_E     = m_E;
        tmp2_D     = m_D;
        tmp_al_a   = m_al_a;
        if (j == KFitConst::kMaxIterationCount - 1) {
          m_FlagOverIteration = true;
        }
      }
    }
  } // j'th loop over
 
 
  if (m_ErrorCode != KFitError::kNoError) return m_ErrorCode;
 
  m_lam    = m_lam0 - m_V_D * m_E * m_V_E * (m_E.T()) * m_lam0;
  m_al_1   = m_al_0 - m_V_al_0 * (m_D.T()) * m_lam;
  m_V_Dt   = m_V_D  - m_V_D * m_E * m_V_E * (m_E.T()) * m_V_D;
  m_V_al_1 = m_V_al_0 - m_V_al_0 * (m_D.T()) * m_V_Dt * m_D * m_V_al_0;
  m_Cov_v_al_1 = -m_V_E * (m_E.T()) * m_V_D * m_D * m_V_al_0;
 
  if (prepareOutputMatrix() != KFitError::kNoError) return m_ErrorCode;
 
  m_CHIsq = chisq;
 
  m_FlagFitted = true;
 
  return m_ErrorCode = KFitError::kNoError;
}

◆ fixMass()

enum KFitError::ECode fixMass ( void )

Tell the object to fix the last added track property at the invariant mass.

Not intended for end user's use.

Returns: error code (zero if success)

Definition at line 63 of file MassPointingVertexFitKFit.cc.

                                   {
  m_IsFixMass.push_back(true);
 
  return m_ErrorCode = KFitError::kNoError;
}

◆ getCHIsq()

double getCHIsq ( void ) const

overridevirtual

Get a chi-square of the fit.

Returns: chi-square of the fit

Reimplemented from KFitBase.

Definition at line 124 of file MassPointingVertexFitKFit.cc.

{
  return m_CHIsq;
}

◆ getCorrelation()

const HepMatrix getCorrelation	(	const int	id1,
		const int	id2,
		const int	flag = `KFitConst::kAfterFit`
	)		const

overridevirtual

Get a correlation matrix between two tracks.

Parameters

id1	first track id
id2	second track id
flag	KFitConst::kBeforeFit or KFitConst::kAfterFit

Returns: (7x7) correlation matrix

Reimplemented from KFitBase.

Definition at line 175 of file MassPointingVertexFitKFit.cc.

{
  if (flag == KFitConst::kAfterFit && !isFitted()) return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
  if (!isTrackIDInRange(id1)) return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
  if (!isTrackIDInRange(id2)) return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
 
  switch (flag) {
    case KFitConst::kBeforeFit:
      return KFitBase::getCorrelation(id1, id2, flag);
 
    case KFitConst::kAfterFit:
      return makeError3(
               this->getTrackMomentum(id1),
               this->getTrackMomentum(id2),
               m_V_al_1.sub(KFitConst::kNumber7 * id1 + 1, KFitConst::kNumber7 * (id1 + 1), KFitConst::kNumber7 * id2 + 1,
                            KFitConst::kNumber7 * (id2 + 1)),
               m_IsFixMass[id1],
               m_IsFixMass[id2]);
 
    default:
      KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kOutOfRange);
      return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
  }
}

◆ getErrorCode()

enum KFitError::ECode getErrorCode ( void ) const

inherited

Get a code of the last error.

Returns: the last error code

Definition at line 101 of file KFitBase.cc.

                             {
  return m_ErrorCode;
}

◆ getInvariantMass()

double getInvariantMass ( void ) const

Get an invariant mass.

Returns: invariant mass

Definition at line 117 of file MassPointingVertexFitKFit.cc.

{
  return m_InvariantMass;
}

◆ getMagneticField()

double getMagneticField ( void ) const

inherited

Get a magnetic field.

Returns: magnetic field

Definition at line 128 of file KFitBase.cc.

{
  return m_MagneticField;
}

◆ getNDF()

int getNDF ( void ) const

virtualinherited

Get an NDF of the fit.

Returns: NDF of the fit

Definition at line 114 of file KFitBase.cc.

{
  return m_NDF;
}

◆ getTrack()

const KFitTrack getTrack ( const int id ) const

inherited

Get a specified track object.

Parameters

id track id

Returns: specified track object

Definition at line 175 of file KFitBase.cc.

{
  if (!isTrackIDInRange(id)) return KFitTrack();
  return m_Tracks[id];
}

◆ getTrackCHIsq()

double getTrackCHIsq ( const int id ) const

overridevirtual

Get a chi-square of the track.

Parameters

id track id

Returns: chi-square of the track

Reimplemented from KFitBase.

Definition at line 140 of file MassPointingVertexFitKFit.cc.

{
  if (!isFitted()) return -1;
  if (!isTrackIDInRange(id)) return -1;
 
  if (m_IsFixMass[id]) {
 
    HepMatrix da(m_Tracks[id].getFitParameter(KFitConst::kBeforeFit) - m_Tracks[id].getFitParameter(KFitConst::kAfterFit));
    int err_inverse = 0;
    const double chisq = (da.T() * (m_Tracks[id].getFitError(KFitConst::kBeforeFit).inverse(err_inverse)) * da)[0][0];
 
    if (err_inverse) {
      KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kCannotGetMatrixInverse);
      return -1;
    }
 
    return chisq;
 
  } else {
 
    HepMatrix da(m_Tracks[id].getMomPos(KFitConst::kBeforeFit) - m_Tracks[id].getMomPos(KFitConst::kAfterFit));
    int err_inverse = 0;
    const double chisq = (da.T() * (m_Tracks[id].getError(KFitConst::kBeforeFit).inverse(err_inverse)) * da)[0][0];
 
    if (err_inverse) {
      KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kCannotGetMatrixInverse);
      return -1;
    }
 
    return chisq;
  }
}

◆ getTrackCount()

int getTrackCount ( void ) const

inherited

Get the number of added tracks.

Returns: the number of added tracks

Definition at line 107 of file KFitBase.cc.

{
  return m_TrackCount;
}

◆ getTrackError()

const HepSymMatrix getTrackError ( const int id ) const

inherited

Get an error matrix of the track.

Parameters

id track id

Returns: error matrix of the track

Definition at line 168 of file KFitBase.cc.

{
  if (!isTrackIDInRange(id)) return HepSymMatrix(KFitConst::kNumber7, 0);
  return m_Tracks[id].getError();
}

◆ getTrackMomentum()

const HepLorentzVector getTrackMomentum ( const int id ) const

inherited

Get a Lorentz vector of the track.

Parameters

id track id

Returns: Lorentz vector of the track

Definition at line 154 of file KFitBase.cc.

{
  if (!isTrackIDInRange(id)) return HepLorentzVector();
  return m_Tracks[id].getMomentum();
}

◆ getTrackPosition()

const HepPoint3D getTrackPosition ( const int id ) const

inherited

Get a position of the track.

Parameters

id track id

Returns: position of the track

Definition at line 161 of file KFitBase.cc.

{
  if (!isTrackIDInRange(id)) return HepPoint3D();
  return m_Tracks[id].getPosition();
}

◆ getTrackVertexError()

const HepMatrix getTrackVertexError ( const int id ) const

Get a vertex error matrix of the track.

Parameters

id track id

Returns: vertex error matrix

Definition at line 131 of file MassPointingVertexFitKFit.cc.

{
  if (!isTrackIDInRange(id)) return HepMatrix(3, KFitConst::kNumber7, 0);
 
  return m_AfterTrackVertexError[id];
}

◆ getVertex()

const HepPoint3D getVertex ( const int flag = KFitConst::kAfterFit ) const

Get a vertex position.

Parameters

flag	KFitConst::kBeforeFit or KFitConst::kAfterFit

Returns: vertex position

Definition at line 91 of file MassPointingVertexFitKFit.cc.

{
  if (flag == KFitConst::kAfterFit && !isFitted()) return HepPoint3D();
 
  switch (flag) {
    case KFitConst::kBeforeFit:
      return m_BeforeVertex;
 
    case KFitConst::kAfterFit:
      return m_AfterVertex;
 
    default:
      KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kOutOfRange);
      return HepPoint3D();
  }
}

◆ getVertexError()

const HepSymMatrix getVertexError ( void ) const

Get a fitted vertex error matrix.

Returns: vertex error matrix

Definition at line 110 of file MassPointingVertexFitKFit.cc.

{
  return m_AfterVertexError;
}

◆ isFitted()

bool isFitted ( void ) const

inherited

Return false if fit is not performed yet or performed fit is failed; otherwise true.

Returns: see description

Definition at line 728 of file KFitBase.cc.

{
  if (m_FlagFitted) return true;
 
  KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kNotFittedYet);
 
  return false;
}

◆ isNonZeroEnergy()

bool isNonZeroEnergy ( const CLHEP::HepLorentzVector & p ) const

protectedinherited

Check if the energy is non-zero.

Parameters

p	Lorentz vector

Returns: true for non-zero energy, false for otherwise

Definition at line 750 of file KFitBase.cc.

{
  if (p.t() != 0) return true;
 
  KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kDivisionByZero);
 
  return false;
}

◆ isTrackIDInRange()

bool isTrackIDInRange ( const int id ) const

protectedinherited

Check if the id is in the range.

Parameters

id track id

Returns: true if the id is in the range, false otherwise

Definition at line 739 of file KFitBase.cc.

{
  if (0 <= id && id < m_TrackCount) return true;
 
  KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kOutOfRange);
 
  return false;
}

◆ makeCoreMatrix()

enum KFitError::ECode makeCoreMatrix ( void )

overrideprivatevirtual

Build matrices using the kinematical constraint.

Returns: error code (zero if success)

Implements KFitBase.

Definition at line 390 of file MassPointingVertexFitKFit.cc.

                                          {
  // Mass Constraint
  HepMatrix al_1_prime(m_al_1);
  HepMatrix Sum_al_1(4, 1, 0);
  std::vector<double> energy(m_TrackCount);
  double a;
 
  for (int i = 0; i < m_TrackCount; i++)
  {
    a = m_property[i][2];
    al_1_prime[i * KFitConst::kNumber7 + 0][0] -= a * (m_v_a[1][0] - al_1_prime[i * KFitConst::kNumber7 + 5][0]);
    al_1_prime[i * KFitConst::kNumber7 + 1][0] += a * (m_v_a[0][0] - al_1_prime[i * KFitConst::kNumber7 + 4][0]);
    energy[i] = sqrt(al_1_prime[i * KFitConst::kNumber7 + 0][0] * al_1_prime[i * KFitConst::kNumber7 + 0][0] +
                     al_1_prime[i * KFitConst::kNumber7 + 1][0] * al_1_prime[i * KFitConst::kNumber7 + 1][0] +
                     al_1_prime[i * KFitConst::kNumber7 + 2][0] * al_1_prime[i * KFitConst::kNumber7 + 2][0] +
                     m_property[i][1] * m_property[i][1]);
    if (m_IsFixMass[i])
      Sum_al_1[3][0] += energy[i];
    else
      Sum_al_1[3][0] += al_1_prime[i * KFitConst::kNumber7 + 3][0];
  }
 
  for (int i = 0; i < m_TrackCount; i++)
  {
    for (int j = 0; j < 3; j++) Sum_al_1[j][0] += al_1_prime[i * KFitConst::kNumber7 + j][0];
  }
 
  double vtx = sqrt((m_v_a[0][0] - m_ProductionVertex.x()) * (m_v_a[0][0] - m_ProductionVertex.x())
                    + (m_v_a[1][0] - m_ProductionVertex.y()) * (m_v_a[1][0] - m_ProductionVertex.y())
                    + (m_v_a[2][0] - m_ProductionVertex.z()) * (m_v_a[2][0] - m_ProductionVertex.z()));
  double vtxpt2 = (m_ProductionVertex.x() - m_v_a[0][0]) * (m_ProductionVertex.x() - m_v_a[0][0]) + (m_ProductionVertex.y() - m_v_a[1][0]) * (m_ProductionVertex.y() - m_v_a[1][0]);
  double mom = sqrt(Sum_al_1[0][0] * Sum_al_1[0][0] + Sum_al_1[1][0] * Sum_al_1[1][0] + Sum_al_1[2][0] * Sum_al_1[2][0]);
  double Pt2 = Sum_al_1[0][0] * Sum_al_1[0][0] + Sum_al_1[1][0] * Sum_al_1[1][0];
 
  m_d[2 * m_TrackCount][0] =
  + Sum_al_1[3][0] * Sum_al_1[3][0] - Sum_al_1[0][0] * Sum_al_1[0][0]
  - Sum_al_1[1][0] * Sum_al_1[1][0] - Sum_al_1[2][0] * Sum_al_1[2][0]
  - m_InvariantMass * m_InvariantMass;
 
  double phiPointingConstraint = atan2(m_v_a[1][0] - m_ProductionVertex.y(), m_v_a[0][0] - m_ProductionVertex.x()) - atan2(Sum_al_1[1][0], Sum_al_1[0][0]);
  phiPointingConstraint = std::fmod(phiPointingConstraint + TMath::Pi(), TMath::TwoPi());
  if (phiPointingConstraint < 0) phiPointingConstraint += TMath::TwoPi();
  phiPointingConstraint -= TMath::Pi();
 
  m_d[2 * m_TrackCount + 1][0] = phiPointingConstraint;
  m_d[2 * m_TrackCount + 2][0] = acos((m_v_a[2][0] - m_ProductionVertex.z()) / vtx) - acos(Sum_al_1[2][0] / mom);
 
  double Sum_a = 0., Sum_tmpx = 0., Sum_tmpy = 0.;
  for (int i = 0; i < m_TrackCount; i++)
  {
    if (energy[i] == 0) {
      m_ErrorCode = KFitError::kDivisionByZero;
      KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
      return m_ErrorCode;
    }
 
    a = m_property[i][2];
 
    if (m_IsFixMass[i]) {
      double invE = 1. / energy[i];
      m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 0] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 0][0] * invE -
                                                               Sum_al_1[0][0]);
      m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 1] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 1][0] * invE -
                                                               Sum_al_1[1][0]);
      m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 2] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 2][0] * invE -
                                                               Sum_al_1[2][0]);
      m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 3] = 0.;
      m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 4] = -2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 1][0] * invE -
                                                                Sum_al_1[1][0]) * a;
      m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 5] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 0][0] * invE -
                                                               Sum_al_1[0][0]) * a;
      m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 6] = 0.;
      Sum_tmpx += al_1_prime[i * KFitConst::kNumber7 + 0][0] * invE * a;
      Sum_tmpy += al_1_prime[i * KFitConst::kNumber7 + 1][0] * invE * a;
    } else {
      m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 0] = -2.*Sum_al_1[0][0];
      m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 1] = -2.*Sum_al_1[1][0];
      m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 2] = -2.*Sum_al_1[2][0];
      m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 3] =  2.*Sum_al_1[3][0];
      m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 4] =  2.*Sum_al_1[1][0] * a;
      m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 5] = -2.*Sum_al_1[0][0] * a;
      m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 6] =  0.;
    }
    m_D[2 * m_TrackCount + 1][i * KFitConst::kNumber7 + 0] =   Sum_al_1[1][0] / Pt2;
    m_D[2 * m_TrackCount + 1][i * KFitConst::kNumber7 + 1] = - Sum_al_1[0][0] / Pt2;
    m_D[2 * m_TrackCount + 1][i * KFitConst::kNumber7 + 2] = 0.;
    m_D[2 * m_TrackCount + 1][i * KFitConst::kNumber7 + 3] = 0.;
    m_D[2 * m_TrackCount + 1][i * KFitConst::kNumber7 + 4] = a * Sum_al_1[0][0] / Pt2;
    m_D[2 * m_TrackCount + 1][i * KFitConst::kNumber7 + 5] = a * Sum_al_1[1][0] / Pt2;
    m_D[2 * m_TrackCount + 1][i * KFitConst::kNumber7 + 6] = 0.;
    m_D[2 * m_TrackCount + 2][i * KFitConst::kNumber7 + 0] = - Sum_al_1[0][0] * Sum_al_1[2][0] / (sqrt(Pt2) * mom * mom);
    m_D[2 * m_TrackCount + 2][i * KFitConst::kNumber7 + 1] = - Sum_al_1[1][0] * Sum_al_1[2][0] / (sqrt(Pt2) * mom * mom);
    m_D[2 * m_TrackCount + 2][i * KFitConst::kNumber7 + 2] = sqrt(Pt2) / (mom * mom);
    m_D[2 * m_TrackCount + 2][i * KFitConst::kNumber7 + 3] = 0.;
    m_D[2 * m_TrackCount + 2][i * KFitConst::kNumber7 + 4] =   a * Sum_al_1[1][0] * Sum_al_1[2][0] / (sqrt(Pt2) * mom * mom);
    m_D[2 * m_TrackCount + 2][i * KFitConst::kNumber7 + 5] = - a * Sum_al_1[0][0] * Sum_al_1[2][0] / (sqrt(Pt2) * mom * mom);
    m_D[2 * m_TrackCount + 2][i * KFitConst::kNumber7 + 6] = 0.;
    Sum_a    += a;
  }
 
  // m_E
  m_E[2 * m_TrackCount][0] = -2.*Sum_al_1[1][0] * Sum_a + 2.*Sum_al_1[3][0] * Sum_tmpy;
  m_E[2 * m_TrackCount][1] =  2.*Sum_al_1[0][0] * Sum_a - 2.*Sum_al_1[3][0] * Sum_tmpx;
  m_E[2 * m_TrackCount][2] =  0.;
  m_E[2 * m_TrackCount + 1][0] = (m_ProductionVertex.y() - m_v_a[1][0]) / vtxpt2 - Sum_a * Sum_al_1[0][0] / Pt2;
  m_E[2 * m_TrackCount + 1][1] = (m_v_a[0][0] - m_ProductionVertex.x()) / vtxpt2 - Sum_a * Sum_al_1[1][0] / Pt2;
  m_E[2 * m_TrackCount + 1][2] = 0.;
  m_E[2 * m_TrackCount + 2][0] = (m_v_a[0][0] - m_ProductionVertex.x()) * (m_v_a[2][0] - m_ProductionVertex.z()) / (vtx* vtx * sqrt(vtxpt2)) - Sum_a * Sum_al_1[1][0] * Sum_al_1[2][0] / (sqrt(Pt2) * mom * mom);
  m_E[2 * m_TrackCount + 2][1] = (m_v_a[1][0] - m_ProductionVertex.y()) * (m_v_a[2][0] - m_ProductionVertex.z()) / (vtx* vtx * sqrt(vtxpt2)) + Sum_a * Sum_al_1[0][0] * Sum_al_1[2][0] / (sqrt(Pt2) * mom * mom);
  m_E[2 * m_TrackCount + 2][2] = - sqrt(vtxpt2) / (vtx * vtx);
 
  for (int i = 0; i < m_TrackCount; i++)
  {
    double S, U;
    double sininv;
 
    double px = m_al_1[i * KFitConst::kNumber7 + 0][0];
    double py = m_al_1[i * KFitConst::kNumber7 + 1][0];
    double pz = m_al_1[i * KFitConst::kNumber7 + 2][0];
    double x  = m_al_1[i * KFitConst::kNumber7 + 4][0];
    double y  = m_al_1[i * KFitConst::kNumber7 + 5][0];
    double z  = m_al_1[i * KFitConst::kNumber7 + 6][0];
    a  = m_property[i][2];
 
    double pt =  sqrt(px * px + py * py);
 
    if (pt == 0) {
      m_ErrorCode = KFitError::kDivisionByZero;
      KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
      return m_ErrorCode;
    }
 
    double invPt  =  1. / pt;
    double invPt2 =  invPt * invPt;
    double dlx    =  m_v_a[0][0] - x;
    double dly    =  m_v_a[1][0] - y;
    double dlz    =  m_v_a[2][0] - z;
    double a1     = -dlx * py + dly * px;
    double a2     =  dlx * px + dly * py;
    double r2d2   =  dlx * dlx + dly * dly;
    double Rx     =  dlx - 2.*px * a2 * invPt2;
    double Ry     =  dly - 2.*py * a2 * invPt2;
 
    if (a != 0.) { // charged
 
      double B = a * a2 * invPt2;
      if (fabs(B) > 1.) {
        m_ErrorCode = KFitError::kDivisionByZero;
        KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
        return m_ErrorCode;
      }
      // sin^(-1)(B)
      sininv = asin(B);
      double tmp0 = 1.0 - B * B;
      if (tmp0 == 0) {
        m_ErrorCode = KFitError::kDivisionByZero;
        KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
        return m_ErrorCode;
      }
      // 1/sqrt(1-B^2)
      double sqrtag = 1.0 / sqrt(tmp0);
      S = sqrtag * invPt2;
      U = dlz - pz * sininv / a;
 
    } else { // neutral
 
      sininv = 0.0;
      S = invPt2;
      U = dlz - pz * a2 * invPt2;
 
    }
 
    // d
    m_d[i * 2 + 0][0] = a1 - 0.5 * a * r2d2;
    m_d[i * 2 + 1][0] = U * pt;
 
    // D
    m_D[i * 2 + 0][i * KFitConst::kNumber7 + 0] =  dly;
    m_D[i * 2 + 0][i * KFitConst::kNumber7 + 1] = -dlx;
    m_D[i * 2 + 0][i * KFitConst::kNumber7 + 2] =  0.0;
    m_D[i * 2 + 0][i * KFitConst::kNumber7 + 4] =  py + a * dlx;
    m_D[i * 2 + 0][i * KFitConst::kNumber7 + 5] = -px + a * dly;
    m_D[i * 2 + 0][i * KFitConst::kNumber7 + 6] =  0.0;
    m_D[i * 2 + 1][i * KFitConst::kNumber7 + 0] = -pz * pt * S * Rx + U * px * invPt;
    m_D[i * 2 + 1][i * KFitConst::kNumber7 + 1] = -pz * pt * S * Ry + U * py * invPt;
    if (a != 0.)
      m_D[i * 2 + 1][i * KFitConst::kNumber7 + 2] = -sininv * pt / a;
    else
      m_D[i * 2 + 1][i * KFitConst::kNumber7 + 2] = -a2 * invPt;
    m_D[i * 2 + 1][i * KFitConst::kNumber7 + 4] =  px * pz * pt * S;
    m_D[i * 2 + 1][i * KFitConst::kNumber7 + 5] =  py * pz * pt * S;
    m_D[i * 2 + 1][i * KFitConst::kNumber7 + 6] = -pt;
 
    // E
    m_E[i * 2 + 0][0] = -py - a * dlx;
    m_E[i * 2 + 0][1] =  px - a * dly;
    m_E[i * 2 + 0][2] =  0.0;
    m_E[i * 2 + 1][0] = -px * pz * pt * S;
    m_E[i * 2 + 1][1] = -py * pz * pt * S;
    m_E[i * 2 + 1][2] =  pt;
  }
 
  return m_ErrorCode = KFitError::kNoError;
}

◆ makeError1() [1/2]

const HepSymMatrix makeError1	(	const CLHEP::HepLorentzVector &	p,
		const CLHEP::HepMatrix &	e
	)		const

protectedinherited

Rebuild an error matrix from a Lorentz vector and an error matrix.

Parameters

p	Lorentz vector
e	(6x6) error matrix

Returns: (7x7) error matrix

Definition at line 221 of file KFitBase.cc.

{
  // self track
  // Error(6x6,e) ==> Error(7x7,output(hsm)) using Momentum(p).
 
  if (!isNonZeroEnergy(p)) return HepSymMatrix(KFitConst::kNumber7, 0);
 
  HepSymMatrix hsm(KFitConst::kNumber7, 0);
 
  for (int i = 0; i < 3; i++) for (int j = i; j < 3; j++) {
      hsm[i][j]     = e[i][j];
      hsm[4 + i][4 + j] = e[3 + i][3 + j];
    }
  for (int i = 0; i < 3; i++) for (int j = 0; j < 3; j++) {
      hsm[i][4 + j] = e[i][3 + j];
    }
 
  const double invE = 1 / p.t();
  hsm[0][3] = (p.x() * hsm[0][0] + p.y() * hsm[0][1] + p.z() * hsm[0][2]) * invE;
  hsm[1][3] = (p.x() * hsm[0][1] + p.y() * hsm[1][1] + p.z() * hsm[1][2]) * invE;
  hsm[2][3] = (p.x() * hsm[0][2] + p.y() * hsm[1][2] + p.z() * hsm[2][2]) * invE;
  hsm[3][3] = (p.x() * p.x() * hsm[0][0] + p.y() * p.y() * hsm[1][1] + p.z() * p.z() * hsm[2][2]
               + 2.0 * p.x() * p.y() * hsm[0][1]
               + 2.0 * p.x() * p.z() * hsm[0][2]
               + 2.0 * p.y() * p.z() * hsm[1][2]) * invE * invE;
  hsm[3][4] = (p.x() * hsm[0][4] + p.y() * hsm[1][4] + p.z() * hsm[2][4]) * invE;
  hsm[3][5] = (p.x() * hsm[0][5] + p.y() * hsm[1][5] + p.z() * hsm[2][5]) * invE;
  hsm[3][6] = (p.x() * hsm[0][6] + p.y() * hsm[1][6] + p.z() * hsm[2][6]) * invE;
 
  return hsm;
}

◆ makeError1() [2/2]

const HepMatrix makeError1	(	const CLHEP::HepLorentzVector &	p1,
		const CLHEP::HepLorentzVector &	p2,
		const CLHEP::HepMatrix &	e
	)		const

protectedinherited

Rebuild an error matrix from a pair of Lorentz vectors and an error matrix.

Parameters

p1	first Lorentz vector
p2	second Lorentz vector
e	(6x6) error matrix

Returns: (7x7) error matrix

Definition at line 255 of file KFitBase.cc.

{
  // track and track
  // Error(6x6,e) ==> Error(7x7,output(hm)) using Momentum(p1&p2).
 
  if (!isNonZeroEnergy(p1)) return HepSymMatrix(KFitConst::kNumber7, 0);
  if (!isNonZeroEnergy(p2)) return HepSymMatrix(KFitConst::kNumber7, 0);
 
  HepMatrix hm(KFitConst::kNumber7, KFitConst::kNumber7, 0);
 
  for (int i = 0; i < 3; i++) for (int j = 0; j < 3; j++) {
      hm[i][j]     = e[i][j];
      hm[4 + i][4 + j] = e[3 + i][3 + j];
      hm[4 + i][j]   = e[3 + i][j];
      hm[i][4 + j]   = e[i][3 + j];
    }
 
  const double invE1 = 1 / p1.t();
  const double invE2 = 1 / p2.t();
  hm[0][3] = (p2.x() * hm[0][0] + p2.y() * hm[0][1] + p2.z() * hm[0][2]) * invE2;
  hm[1][3] = (p2.x() * hm[1][0] + p2.y() * hm[1][1] + p2.z() * hm[1][2]) * invE2;
  hm[2][3] = (p2.x() * hm[2][0] + p2.y() * hm[2][1] + p2.z() * hm[2][2]) * invE2;
  hm[4][3] = (p2.x() * hm[4][0] + p2.y() * hm[4][1] + p2.z() * hm[4][2]) * invE2;
  hm[5][3] = (p2.x() * hm[5][0] + p2.y() * hm[5][1] + p2.z() * hm[5][2]) * invE2;
  hm[6][3] = (p2.x() * hm[6][0] + p2.y() * hm[6][1] + p2.z() * hm[6][2]) * invE2;
  hm[3][3] = (p1.x() * p2.x() * hm[0][0] + p1.y() * p2.y() * hm[1][1] + p1.z() * p2.z() * hm[2][2] +
              p1.x() * p2.y() * hm[0][1] + p2.x() * p1.y() * hm[1][0] +
              p1.x() * p2.z() * hm[0][2] + p2.x() * p1.z() * hm[2][0] +
              p1.y() * p2.z() * hm[1][2] + p2.y() * p1.z() * hm[2][1]) * invE1 * invE2;
  hm[3][0] = (p1.x() * hm[0][0] + p1.y() * hm[1][0] + p1.z() * hm[2][0]) * invE1;
  hm[3][1] = (p1.x() * hm[0][1] + p1.y() * hm[1][1] + p1.z() * hm[2][1]) * invE1;
  hm[3][2] = (p1.x() * hm[0][2] + p1.y() * hm[1][2] + p1.z() * hm[2][2]) * invE1;
  hm[3][4] = (p1.x() * hm[0][4] + p1.y() * hm[1][4] + p1.z() * hm[2][4]) * invE1;
  hm[3][5] = (p1.x() * hm[0][5] + p1.y() * hm[1][5] + p1.z() * hm[2][5]) * invE1;
  hm[3][6] = (p1.x() * hm[0][6] + p1.y() * hm[1][6] + p1.z() * hm[2][6]) * invE1;
 
  return hm;
}

◆ makeError2()

const HepMatrix makeError2	(	const CLHEP::HepLorentzVector &	p,
		const CLHEP::HepMatrix &	e
	)		const

protectedinherited

Rebuild an error matrix from a Lorentz vector and an error matrix.

Parameters

p	Lorentz vector
e	(3x6) error matrix

Returns: (3x7) error matrix

Definition at line 296 of file KFitBase.cc.

{
  // vertex and track
  // Error(3x6,e) ==> Error(3x7,output(hm)) using Momentum(p).
 
  if (!isNonZeroEnergy(p)) return HepSymMatrix(KFitConst::kNumber7, 0);
 
  HepMatrix hm(3, KFitConst::kNumber7, 0);
 
  for (int i = 0; i < 3; i++) for (int j = 0; j < 3; j++) {
      hm[i][j]   = e[i][j];
      hm[i][4 + j] = e[i][3 + j];
    }
 
  const double invE = 1 / p.t();
  hm[0][3] = (p.x() * hm[0][0] + p.y() * hm[0][1] + p.z() * hm[0][2]) * invE;
  hm[1][3] = (p.x() * hm[1][0] + p.y() * hm[1][1] + p.z() * hm[1][2]) * invE;
  hm[2][3] = (p.x() * hm[2][0] + p.y() * hm[2][1] + p.z() * hm[2][2]) * invE;
 
  return hm;
}

◆ makeError3() [1/2]

const HepSymMatrix makeError3	(	const CLHEP::HepLorentzVector &	p,
		const CLHEP::HepMatrix &	e,
		const bool	is_fix_mass
	)		const

protectedinherited

Rebuild an error matrix from a Lorentz vector and an error matrix.

Parameters

p	Lorentz vector
e	(7x7) error matrix
is_fix_mass	true to recalculate energy term from other parameters, false to do nothing

Returns: (7x7) error matrix

Definition at line 320 of file KFitBase.cc.

{
  // self track
  // Error(7x7,e) ==> Error(7x7,output(hsm)) using Momentum(p).
  // is_fix_mass = 1 : Energy term is recalculated from the other parameters.
  // is_fix_mass = 0 : hsm = e.
 
  if (!isNonZeroEnergy(p)) return HepSymMatrix(KFitConst::kNumber7, 0);
 
  if (!is_fix_mass) {
    HepSymMatrix hsm(KFitConst::kNumber7, 0);
    for (int i = 0; i < 7; i++) for (int j = i; j < 7; j++) {
        hsm[i][j] = e[i][j];
      }
    return hsm;
  }
 
  HepSymMatrix hsm(KFitConst::kNumber7, 0);
 
  for (int i = 0; i < 7; i++) {
    if (i != 3)
      for (int j = i; j < 7; j++) hsm[i][j] = e[i][j];
  }
 
  double invE = 1 / p.t();
  hsm[0][3] = (p.x() * hsm[0][0] + p.y() * hsm[0][1] + p.z() * hsm[0][2]) * invE;
  hsm[1][3] = (p.x() * hsm[0][1] + p.y() * hsm[1][1] + p.z() * hsm[1][2]) * invE;
  hsm[2][3] = (p.x() * hsm[0][2] + p.y() * hsm[1][2] + p.z() * hsm[2][2]) * invE;
  hsm[3][3] = (p.x() * p.x() * hsm[0][0] + p.y() * p.y() * hsm[1][1] + p.z() * p.z() * hsm[2][2]
               + 2.0 * p.x() * p.y() * hsm[0][1]
               + 2.0 * p.x() * p.z() * hsm[0][2]
               + 2.0 * p.y() * p.z() * hsm[1][2]) * invE * invE;
  hsm[3][4] = (p.x() * hsm[0][4] + p.y() * hsm[1][4] + p.z() * hsm[2][4]) * invE;
  hsm[3][5] = (p.x() * hsm[0][5] + p.y() * hsm[1][5] + p.z() * hsm[2][5]) * invE;
  hsm[3][6] = (p.x() * hsm[0][6] + p.y() * hsm[1][6] + p.z() * hsm[2][6]) * invE;
 
  return hsm;
}

◆ makeError3() [2/2]

const HepMatrix makeError3	(	const CLHEP::HepLorentzVector &	p1,
		const CLHEP::HepLorentzVector &	p2,
		const CLHEP::HepMatrix &	e,
		const bool	is_fix_mass1,
		const bool	is_fix_mass2
	)		const

protectedinherited

Rebuild an error matrix from a pair of Lorentz vectors and an error matrix.

Parameters

p1	first Lorentz vector
p2	second Lorentz vector
e	(7x7) error matrix
is_fix_mass1	true to recalculate energy term from other parameters, false to do nothing
is_fix_mass2	true to recalculate energy term from other parameters, false to do nothing

Returns: (7x7) error matrix

Definition at line 361 of file KFitBase.cc.

{
  // track and track
  // Error(7x7,e) ==> Error(7x7,output(hm)) using Momentum(p1&p2).
  // is_fix_mass = 1 : Energy term is recalculated from the other parameters.
  // is_fix_mass = 0 : not.
 
  if (is_fix_mass1 && is_fix_mass2) {
    if (!isNonZeroEnergy(p1)) return HepSymMatrix(KFitConst::kNumber7, 0);
    if (!isNonZeroEnergy(p2)) return HepSymMatrix(KFitConst::kNumber7, 0);
 
    HepMatrix hm(e);
 
    const double invE1 = 1 / p1.t();
    const double invE2 = 1 / p2.t();
    hm[0][3] = (p2.x() * hm[0][0] + p2.y() * hm[0][1] + p2.z() * hm[0][2]) * invE2;
    hm[1][3] = (p2.x() * hm[1][0] + p2.y() * hm[1][1] + p2.z() * hm[1][2]) * invE2;
    hm[2][3] = (p2.x() * hm[2][0] + p2.y() * hm[2][1] + p2.z() * hm[2][2]) * invE2;
    hm[4][3] = (p2.x() * hm[4][0] + p2.y() * hm[4][1] + p2.z() * hm[4][2]) * invE2;
    hm[5][3] = (p2.x() * hm[5][0] + p2.y() * hm[5][1] + p2.z() * hm[5][2]) * invE2;
    hm[6][3] = (p2.x() * hm[6][0] + p2.y() * hm[6][1] + p2.z() * hm[6][2]) * invE2;
    hm[3][0] = (p1.x() * hm[0][0] + p1.y() * hm[1][0] + p1.z() * hm[2][0]) * invE1;
    hm[3][1] = (p1.x() * hm[0][1] + p1.y() * hm[1][1] + p1.z() * hm[2][1]) * invE1;
    hm[3][2] = (p1.x() * hm[0][2] + p1.y() * hm[1][2] + p1.z() * hm[2][2]) * invE1;
    hm[3][3] = (p1.x() * p2.x() * hm[0][0] + p1.y() * p2.y() * hm[1][1] + p1.z() * p2.z() * hm[2][2] +
                p1.x() * p2.y() * hm[0][1] + p2.x() * p1.y() * hm[1][0] +
                p1.x() * p2.z() * hm[0][2] + p2.x() * p1.z() * hm[2][0] +
                p1.y() * p2.z() * hm[1][2] + p2.y() * p1.z() * hm[2][1]) * invE1 * invE2;
    hm[3][4] = (p1.x() * hm[0][4] + p1.y() * hm[1][4] + p1.z() * hm[2][4]) * invE1;
    hm[3][5] = (p1.x() * hm[0][5] + p1.y() * hm[1][5] + p1.z() * hm[2][5]) * invE1;
    hm[3][6] = (p1.x() * hm[0][6] + p1.y() * hm[1][6] + p1.z() * hm[2][6]) * invE1;
 
    return hm;
  }
 
 
  if (is_fix_mass1 && !is_fix_mass2) {
    if (!isNonZeroEnergy(p1)) return HepSymMatrix(KFitConst::kNumber7, 0);
 
    HepMatrix hm(e);
 
    const double invE1 = 1 / p1.t();
    hm[3][0] = (p1.x() * hm[0][0] + p1.y() * hm[1][0] + p1.z() * hm[2][0]) * invE1;
    hm[3][1] = (p1.x() * hm[0][1] + p1.y() * hm[1][1] + p1.z() * hm[2][1]) * invE1;
    hm[3][2] = (p1.x() * hm[0][2] + p1.y() * hm[1][2] + p1.z() * hm[2][2]) * invE1;
    hm[3][3] = (p1.x() * hm[0][3] + p1.y() * hm[1][3] + p1.z() * hm[2][3]) * invE1;
    hm[3][4] = (p1.x() * hm[0][4] + p1.y() * hm[1][4] + p1.z() * hm[2][4]) * invE1;
    hm[3][5] = (p1.x() * hm[0][5] + p1.y() * hm[1][5] + p1.z() * hm[2][5]) * invE1;
    hm[3][6] = (p1.x() * hm[0][6] + p1.y() * hm[1][6] + p1.z() * hm[2][6]) * invE1;
 
    return hm;
  }
 
 
  if (!is_fix_mass1 &&  is_fix_mass2) {
    if (!isNonZeroEnergy(p2)) return HepSymMatrix(KFitConst::kNumber7, 0);
 
    HepMatrix hm(e);
 
    const double invE2 = 1 / p2.t();
    hm[0][3] = (p2.x() * hm[0][0] + p2.y() * hm[0][1] + p2.z() * hm[0][2]) * invE2;
    hm[1][3] = (p2.x() * hm[1][0] + p2.y() * hm[1][1] + p2.z() * hm[1][2]) * invE2;
    hm[2][3] = (p2.x() * hm[2][0] + p2.y() * hm[2][1] + p2.z() * hm[2][2]) * invE2;
    hm[3][3] = (p2.x() * hm[3][0] + p2.y() * hm[3][1] + p2.z() * hm[3][2]) * invE2;
    hm[4][3] = (p2.x() * hm[4][0] + p2.y() * hm[4][1] + p2.z() * hm[4][2]) * invE2;
    hm[5][3] = (p2.x() * hm[5][0] + p2.y() * hm[5][1] + p2.z() * hm[5][2]) * invE2;
    hm[6][3] = (p2.x() * hm[6][0] + p2.y() * hm[6][1] + p2.z() * hm[6][2]) * invE2;
 
    return hm;
  }
 
  return e;
}

◆ makeError4()

const HepMatrix makeError4	(	const CLHEP::HepLorentzVector &	p,
		const CLHEP::HepMatrix &	e
	)		const

protectedinherited

Rebuild an error matrix from a Lorentz vector and an error matrix.

Parameters

p	Lorentz vector
e	(3x7) error matrix

Returns: (7x7) error matrix

Definition at line 439 of file KFitBase.cc.

{
  // vertex and track
  // Error(3x7,e) ==> Error(3x7,output(hm)) using Momentum(p).
  // Energy term is recalculated from the other parameters.
 
  if (!isNonZeroEnergy(p)) return HepSymMatrix(KFitConst::kNumber7, 0);
 
  HepMatrix hm(e);
 
  const double invE = 1 / p.t();
  hm[0][3] = (p.x() * hm[0][0] + p.y() * hm[0][1] + p.z() * hm[0][2]) * invE;
  hm[1][3] = (p.x() * hm[1][0] + p.y() * hm[1][1] + p.z() * hm[1][2]) * invE;
  hm[2][3] = (p.x() * hm[2][0] + p.y() * hm[2][1] + p.z() * hm[2][2]) * invE;
 
  return hm;
}

◆ prepareCorrelation()

enum KFitError::ECode prepareCorrelation ( void )

overrideprivatevirtual

Build a grand correlation matrix from input-track properties.

Returns: error code (zero if success)

Reimplemented from KFitBase.

Definition at line 297 of file MassPointingVertexFitKFit.cc.

                                              {
  if (m_BeforeCorrelation.size() != static_cast<unsigned int>(m_TrackCount * (m_TrackCount - 1) / 2))
  {
    m_ErrorCode = KFitError::kBadCorrelationSize;
    KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
    return m_ErrorCode;
  }
 
  int row = 0, col = 0;
 
  for (auto& hm : m_BeforeCorrelation)
  {
    // counter
    row++;
    if (row == m_TrackCount) {
      col++;
      row = col + 1;
    }
 
    int ii = 0, jj = 0;
    for (int i = KFitConst::kNumber7 * row; i < KFitConst::kNumber7 * (row + 1); i++) {
      for (int j = KFitConst::kNumber7 * col; j < KFitConst::kNumber7 * (col + 1); j++) {
        m_V_al_0[i][j] = hm[ii][jj];
        jj++;
      }
      jj = 0;
      ii++;
    }
  }
 
  return m_ErrorCode = KFitError::kNoError;
}

◆ prepareInputMatrix()

enum KFitError::ECode prepareInputMatrix ( void )

overrideprivatevirtual

Build grand matrices for minimum search from input-track properties.

Returns: error code (zero if success)

Implements KFitBase.

Definition at line 208 of file MassPointingVertexFitKFit.cc.

                                              {
  if (m_TrackCount > KFitConst::kMaxTrackCount)
  {
    m_ErrorCode = KFitError::kBadTrackSize;
    KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
    return m_ErrorCode;
  }
 
 
  if (m_IsFixMass.size() == 0)
  {
    // If no fix_mass flag at all,
    // all tracks are considered to be fixed at mass.
    for (int i = 0; i < m_TrackCount; i++) this->fixMass();
  } else if (m_IsFixMass.size() != (unsigned int)m_TrackCount)
  {
    m_ErrorCode = KFitError::kBadTrackSize;
    KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
    return m_ErrorCode;
  }
 
 
  int index = 0;
  m_al_0     = HepMatrix(KFitConst::kNumber7 * m_TrackCount, 1, 0);
  m_property = HepMatrix(m_TrackCount, 3, 0);
  m_V_al_0   = HepSymMatrix(KFitConst::kNumber7 * m_TrackCount, 0);
 
  for (auto& track : m_Tracks)
  {
    // momentum x,y,z,E and position x,y,z
    m_al_0[index * KFitConst::kNumber7 + 0][0] = track.getMomentum(KFitConst::kBeforeFit).x();
    m_al_0[index * KFitConst::kNumber7 + 1][0] = track.getMomentum(KFitConst::kBeforeFit).y();
    m_al_0[index * KFitConst::kNumber7 + 2][0] = track.getMomentum(KFitConst::kBeforeFit).z();
    m_al_0[index * KFitConst::kNumber7 + 3][0] = track.getMomentum(KFitConst::kBeforeFit).t();
    m_al_0[index * KFitConst::kNumber7 + 4][0] = track.getPosition(KFitConst::kBeforeFit).x();
    m_al_0[index * KFitConst::kNumber7 + 5][0] = track.getPosition(KFitConst::kBeforeFit).y();
    m_al_0[index * KFitConst::kNumber7 + 6][0] = track.getPosition(KFitConst::kBeforeFit).z();
    // these error
    m_V_al_0.sub(index * KFitConst::kNumber7 + 1, track.getError(KFitConst::kBeforeFit));
    // charge, mass, a
    m_property[index][0] =  track.getCharge();
    m_property[index][1] =  track.getMass();
    const double c = Belle2::Const::speedOfLight * 1e-4;
    m_property[index][2] = -c * m_MagneticField * track.getCharge();
    index++;
  }
 
  // error between track and track
  if (m_FlagCorrelation)
  {
    this->prepareCorrelation();
    if (m_ErrorCode != KFitError::kNoError) {
      KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
      return m_ErrorCode;
    }
  }
 
  // vertex
  m_v_a[0][0] = m_BeforeVertex.x();
  m_v_a[1][0] = m_BeforeVertex.y();
  m_v_a[2][0] = m_BeforeVertex.z();
 
  // set member matrix
  m_al_1     = m_al_0;
 
  m_V_al_1     = HepMatrix(KFitConst::kNumber7 * m_TrackCount, KFitConst::kNumber7 * m_TrackCount, 0);
  m_D          = m_V_al_1.sub(1, m_TrackCount * 2 + 3, 1, KFitConst::kNumber7 * m_TrackCount);
  m_E          = m_V_al_1.sub(1, m_TrackCount * 2 + 3, 1, 3);
  m_d          = m_V_al_1.sub(1, m_TrackCount * 2 + 3, 1, 1);
  m_V_D        = m_V_al_1.sub(1, m_TrackCount * 2 + 3, 1, m_TrackCount * 2 + 3);
  m_lam        = m_V_al_1.sub(1, m_TrackCount * 2 + 3, 1, 1);
  m_lam0       = m_V_al_1.sub(1, m_TrackCount * 2 + 3, 1, 1);
  m_V_Dt       = m_V_al_1.sub(1, m_TrackCount * 2 + 3, 1, m_TrackCount * 2 + 3);
  m_Cov_v_al_1 = m_V_al_1.sub(1, 3, 1, KFitConst::kNumber7 * m_TrackCount);
 
  return m_ErrorCode = KFitError::kNoError;
}

◆ prepareInputSubMatrix()

enum KFitError::ECode prepareInputSubMatrix ( void )

overrideprivatevirtual

Build sub-matrices for minimum search from input-track properties.

Returns: error code (zero if success)

Implements KFitBase.

Definition at line 288 of file MassPointingVertexFitKFit.cc.

                                                 {
  // vertex
  for (int i = 0; i < 3; i++) m_v[i][0] = m_v_a[i][0];
 
  return m_ErrorCode = KFitError::kNoError;
}

◆ prepareOutputMatrix()

enum KFitError::ECode prepareOutputMatrix ( void )

overrideprivatevirtual

Build an output error matrix.

Returns: error code (zero if success)

Implements KFitBase.

Definition at line 332 of file MassPointingVertexFitKFit.cc.

                                               {
  Hep3Vector h3v;
  int index = 0;
  for (auto& pdata : m_Tracks)
  {
    // tracks
    // momentum
    h3v.setX(m_al_1[index * KFitConst::kNumber7 + 0][0]);
    h3v.setY(m_al_1[index * KFitConst::kNumber7 + 1][0]);
    h3v.setZ(m_al_1[index * KFitConst::kNumber7 + 2][0]);
    if (m_IsFixMass[index])
      pdata.setMomentum(HepLorentzVector(h3v, sqrt(h3v.mag2() + pdata.getMass()*pdata.getMass())), KFitConst::kAfterFit);
    else
      pdata.setMomentum(HepLorentzVector(h3v, m_al_1[index * KFitConst::kNumber7 + 3][0]), KFitConst::kAfterFit);
    // position
    pdata.setPosition(HepPoint3D(
                        m_al_1[index * KFitConst::kNumber7 + 4][0],
                        m_al_1[index * KFitConst::kNumber7 + 5][0],
                        m_al_1[index * KFitConst::kNumber7 + 6][0]), KFitConst::kAfterFit);
    // error of the tracks
    pdata.setError(this->makeError3(pdata.getMomentum(),
                                    m_V_al_1.sub(
                                      index    * KFitConst::kNumber7 + 1,
                                      (index + 1)*KFitConst::kNumber7,
                                      index    * KFitConst::kNumber7 + 1,
                                      (index + 1)*KFitConst::kNumber7), m_IsFixMass[index]),
                   KFitConst::kAfterFit);
    if (m_ErrorCode != KFitError::kNoError) break;
    index++;
  }
 
  // vertex
  m_AfterVertex.setX(m_v_a[0][0]);
  m_AfterVertex.setY(m_v_a[1][0]);
  m_AfterVertex.setZ(m_v_a[2][0]);
  // error of the vertex
  for (int i = 0; i < 3; i++) for (int j = i; j < 3; j++)
    {
      m_AfterVertexError[i][j] = m_V_E[i][j];
    }
  // error between vertex and tracks
  for (int i = 0; i < m_TrackCount; i++)
  {
    HepMatrix hm(3, KFitConst::kNumber7, 0);
    for (int j = 0; j < 3; j++)  for (int k = 0; k < KFitConst::kNumber7; k++) {
        hm[j][k] = m_Cov_v_al_1[j][KFitConst::kNumber7 * i + k];
      }
    if (m_IsFixMass[i])
      m_AfterTrackVertexError.push_back(this->makeError4(m_Tracks[i].getMomentum(), hm));
    else
      m_AfterTrackVertexError.push_back(hm);
  }
 
  return m_ErrorCode = KFitError::kNoError;
}

◆ setCorrelation()

enum KFitError::ECode setCorrelation ( const CLHEP::HepMatrix & c )

overridevirtual

Set a correlation matrix.

Not intended for end user's use.

Parameters

c	(7x7) correlation matrix

Returns: error code (zero if success)

Reimplemented from KFitBase.

Definition at line 79 of file MassPointingVertexFitKFit.cc.

                                                            {
  return KFitBase::setCorrelation(m);
}

◆ setInitialVertex()

enum KFitError::ECode setInitialVertex ( const HepPoint3D & v )

Set an initial vertex point for the mass-vertex-pointing constraint fit.

Parameters

v	initial vertex point

Returns: error code (zero if success)

Definition at line 39 of file MassPointingVertexFitKFit.cc.

                                                               {
  m_BeforeVertex = v;
 
  return m_ErrorCode = KFitError::kNoError;
}

◆ setInvariantMass()

enum KFitError::ECode setInvariantMass ( const double m )

Set an invariant mass for the mass-vertex-pointing constraint fit.

Parameters

m	invariant mass

Returns: error code (zero if success)

Definition at line 47 of file MassPointingVertexFitKFit.cc.

                                                          {
  m_InvariantMass = m;
 
  return m_ErrorCode = KFitError::kNoError;
}

◆ setMagneticField()

enum KFitError::ECode setMagneticField ( const double mf )

inherited

Change a magnetic field from the default value KFitConst::kDefaultMagneticField.

Parameters

mf	magnetic field to set

Returns: error code (zero if success)

Definition at line 93 of file KFitBase.cc.

                                          {
  m_MagneticField = mf;
 
  return m_ErrorCode = KFitError::kNoError;
}

◆ setProductionVertex()

enum KFitError::ECode setProductionVertex ( const HepPoint3D & v )

Set the production vertex of the particle.

Parameters

v	production vertex

Returns: error code (zero if success)

Definition at line 55 of file MassPointingVertexFitKFit.cc.

                                                                  {
  m_ProductionVertex = v;
 
  return m_ErrorCode = KFitError::kNoError;
}

◆ setZeroCorrelation()

enum KFitError::ECode setZeroCorrelation ( void )

overridevirtual

Indicate no correlation between tracks.

Not intended for end user's use.

Returns: error code (zero if success)

Reimplemented from KFitBase.

Definition at line 85 of file MassPointingVertexFitKFit.cc.

                                              {
  return KFitBase::setZeroCorrelation();
}

◆ unfixMass()

enum KFitError::ECode unfixMass ( void )

Tell the object to unfix the last added track property at the invariant mass.

Not intended for end user's use.

Returns: error code (zero if success)

Definition at line 71 of file MassPointingVertexFitKFit.cc.

                                     {
  m_IsFixMass.push_back(false);
 
  return m_ErrorCode = KFitError::kNoError;
}

◆ updateMother()

enum KFitError::ECode updateMother ( Particle * mother )

Update mother particle.

Parameters

mother particle

Definition at line 603 of file MassPointingVertexFitKFit.cc.

{
  MakeMotherKFit kmm;
  kmm.setMagneticField(m_MagneticField);
  unsigned n = getTrackCount();
  for (unsigned i = 0; i < n; ++i) {
    kmm.addTrack(getTrackMomentum(i), getTrackPosition(i), getTrackError(i),
                 getTrack(i).getCharge());
    kmm.setTrackVertexError(getTrackVertexError(i));
    for (unsigned j = i + 1; j < n; ++j) {
      kmm.setCorrelation(getCorrelation(i, j));
    }
  }
  kmm.setVertex(getVertex());
  kmm.setVertexError(getVertexError());
  m_ErrorCode = kmm.doMake();
  if (m_ErrorCode != KFitError::kNoError) {
    B2DEBUG(2, "Error in doMake");
    return m_ErrorCode;
  }
  double chi2 = getCHIsq();
  int ndf = getNDF();
  double prob = TMath::Prob(chi2, ndf);
  mother->addExtraInfo("ndf", ndf);
  mother->addExtraInfo("chiSquared", chi2);
  mother->updateMomentum(
    CLHEPToROOT::getLorentzVector(kmm.getMotherMomentum()),
    CLHEPToROOT::getXYZVector(kmm.getMotherPosition()),
    CLHEPToROOT::getTMatrixFSym(kmm.getMotherError()),
    prob);
  m_ErrorCode = KFitError::kNoError;
  return m_ErrorCode;
}

Member Data Documentation

◆ m_AfterTrackVertexError

std::vector<CLHEP::HepMatrix> m_AfterTrackVertexError

private

array of vertex error matrices after the fit.

Definition at line 120 of file MassPointingVertexFitKFit.h.

◆ m_AfterVertex

HepPoint3D m_AfterVertex

private

Vertex position after the fit.

Definition at line 116 of file MassPointingVertexFitKFit.h.

◆ m_AfterVertexError

CLHEP::HepSymMatrix m_AfterVertexError

private

Vertex error matrix after the fit.

Definition at line 118 of file MassPointingVertexFitKFit.h.

◆ m_al_0

CLHEP::HepMatrix m_al_0

protectedinherited

See J.Tanaka Ph.D (2001) p136 for definition.

Definition at line 257 of file KFitBase.h.

◆ m_al_1

CLHEP::HepMatrix m_al_1

protectedinherited

See J.Tanaka Ph.D (2001) p136 for definition.

Definition at line 259 of file KFitBase.h.

◆ m_al_a

CLHEP::HepMatrix m_al_a

protectedinherited

See J.Tanaka Ph.D (2001) p137 for definition.

Definition at line 261 of file KFitBase.h.

◆ m_BeforeCorrelation

std::vector<CLHEP::HepMatrix> m_BeforeCorrelation

protectedinherited

Container of input correlation matrices.

Definition at line 251 of file KFitBase.h.

◆ m_BeforeVertex

HepPoint3D m_BeforeVertex

private

Vertex position before the fit.

Definition at line 114 of file MassPointingVertexFitKFit.h.

◆ m_CHIsq

double m_CHIsq

protectedinherited

chi-square of the fit.

Definition at line 297 of file KFitBase.h.

◆ m_Cov_v_al_1

CLHEP::HepMatrix m_Cov_v_al_1

protectedinherited

See J.Tanaka Ph.D (2001) p137 for definition.

Definition at line 291 of file KFitBase.h.

◆ m_D

CLHEP::HepMatrix m_D

protectedinherited

See J.Tanaka Ph.D (2001) p137 for definition.

Definition at line 266 of file KFitBase.h.

◆ m_d

CLHEP::HepMatrix m_d

protectedinherited

See J.Tanaka Ph.D (2001) p137 for definition.

Definition at line 268 of file KFitBase.h.

◆ m_E

CLHEP::HepMatrix m_E

protectedinherited

See J.Tanaka Ph.D (2001) p137 for definition.

Definition at line 279 of file KFitBase.h.

◆ m_ErrorCode

enum KFitError::ECode m_ErrorCode

protectedinherited

Error code.

Definition at line 243 of file KFitBase.h.

◆ m_FlagCorrelation

bool m_FlagCorrelation

protectedinherited

Flag whether a correlation among tracks exists.

Definition at line 306 of file KFitBase.h.

◆ m_FlagFitted

bool m_FlagFitted

protectedinherited

Flag to indicate if the fit is performed and succeeded.

Definition at line 245 of file KFitBase.h.

◆ m_FlagOverIteration

bool m_FlagOverIteration

protectedinherited

Flag whether the iteration count exceeds the limit.

Definition at line 308 of file KFitBase.h.

◆ m_InvariantMass

double m_InvariantMass

private

Invariant mass.

Definition at line 124 of file MassPointingVertexFitKFit.h.

◆ m_IsFixMass

std::vector<int> m_IsFixMass

private

Array of flags whether the track property is fixed at the mass.

Definition at line 126 of file MassPointingVertexFitKFit.h.

◆ m_lam

CLHEP::HepMatrix m_lam

protectedinherited

See J.Tanaka Ph.D (2001) p137 for definition.

Definition at line 276 of file KFitBase.h.

◆ m_lam0

CLHEP::HepMatrix m_lam0

protectedinherited

See J.Tanaka Ph.D (2001) p138 for definition.

Definition at line 283 of file KFitBase.h.

◆ m_MagneticField

double m_MagneticField

protectedinherited

Magnetic field.

Definition at line 311 of file KFitBase.h.

◆ m_NDF

int m_NDF

protectedinherited

NDF of the fit.

Definition at line 295 of file KFitBase.h.

◆ m_NecessaryTrackCount

int m_NecessaryTrackCount

protectedinherited

Number needed tracks to perform fit.

Definition at line 303 of file KFitBase.h.

◆ m_ProductionVertex

HepPoint3D m_ProductionVertex

private

Production vertex position.

Definition at line 122 of file MassPointingVertexFitKFit.h.

◆ m_property

CLHEP::HepMatrix m_property

protectedinherited

Container of charges and masses.

Definition at line 263 of file KFitBase.h.

◆ m_TrackCount

int m_TrackCount

protectedinherited

Number of tracks.

Definition at line 301 of file KFitBase.h.

◆ m_Tracks

std::vector<KFitTrack> m_Tracks

protectedinherited

Container of input tracks.

Definition at line 249 of file KFitBase.h.

◆ m_v

CLHEP::HepMatrix m_v

protectedinherited

See J.Tanaka Ph.D (2001) p137 for definition.

Definition at line 285 of file KFitBase.h.

◆ m_v_a

CLHEP::HepMatrix m_v_a

protectedinherited

See J.Tanaka Ph.D (2001) p137 for definition.

Definition at line 287 of file KFitBase.h.

◆ m_V_al_0

CLHEP::HepSymMatrix m_V_al_0

protectedinherited

See J.Tanaka Ph.D (2001) p137 for definition.

Definition at line 255 of file KFitBase.h.

◆ m_V_al_1

CLHEP::HepMatrix m_V_al_1

protectedinherited

See J.Tanaka Ph.D (2001) p138 for definition.

Definition at line 274 of file KFitBase.h.

◆ m_V_D

CLHEP::HepMatrix m_V_D

protectedinherited

See J.Tanaka Ph.D (2001) p138 for definition.

Definition at line 271 of file KFitBase.h.

◆ m_V_Dt

CLHEP::HepMatrix m_V_Dt

protectedinherited

See J.Tanaka Ph.D (2001) p138 for definition.

Definition at line 289 of file KFitBase.h.

◆ m_V_E

CLHEP::HepMatrix m_V_E

protectedinherited

See J.Tanaka Ph.D (2001) p138 for definition.

Definition at line 281 of file KFitBase.h.

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

analysis/VertexFitting/KFit/include/MassPointingVertexFitKFit.h
analysis/VertexFitting/KFit/src/MassPointingVertexFitKFit.cc

Public Member Functions

Protected Member Functions

Protected Attributes

Private Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ MassPointingVertexFitKFit()

Member Function Documentation

◆ addParticle()

◆ addTrack() [1/2]

◆ addTrack() [2/2]

◆ calculateNDF()

◆ doFit()

◆ doFit1()

◆ doFit2()

◆ fixMass()

◆ getCHIsq()

◆ getCorrelation()

◆ getErrorCode()

◆ getInvariantMass()

◆ getMagneticField()

◆ getNDF()

◆ getTrack()

◆ getTrackCHIsq()

◆ getTrackCount()

◆ getTrackError()

◆ getTrackMomentum()

◆ getTrackPosition()

◆ getTrackVertexError()

◆ getVertex()

◆ getVertexError()

◆ isFitted()

◆ isNonZeroEnergy()

◆ isTrackIDInRange()

◆ makeCoreMatrix()

◆ makeError1() [1/2]

◆ makeError1() [2/2]

◆ makeError2()

◆ makeError3() [1/2]

◆ makeError3() [2/2]

◆ makeError4()

◆ prepareCorrelation()

◆ prepareInputMatrix()

◆ prepareInputSubMatrix()

◆ prepareOutputMatrix()

◆ setCorrelation()

◆ setInitialVertex()

◆ setInvariantMass()

◆ setMagneticField()

◆ setProductionVertex()

◆ setZeroCorrelation()

◆ unfixMass()

◆ updateMother()

Member Data Documentation

◆ m_AfterTrackVertexError

◆ m_AfterVertex

◆ m_AfterVertexError

◆ m_al_0

◆ m_al_1

◆ m_al_a

◆ m_BeforeCorrelation

◆ m_BeforeVertex

◆ m_CHIsq

◆ m_Cov_v_al_1

◆ m_D

◆ m_d

◆ m_E

◆ m_ErrorCode

◆ m_FlagCorrelation

◆ m_FlagFitted

◆ m_FlagOverIteration

◆ m_InvariantMass

◆ m_IsFixMass

◆ m_lam

◆ m_lam0

◆ m_MagneticField

◆ m_NDF

◆ m_NecessaryTrackCount

◆ m_ProductionVertex