MassFourCFitKFit Class Reference

MassFourCFitKFit is a derived class from KFitBase to perform mass and 4 momentum-constraint kinematical fit. More...

#include <MassFourCFitKFit.h>

Inheritance diagram for MassFourCFitKFit:

Collaboration diagram for MassFourCFitKFit:

Public Member Functions
	MassFourCFitKFit ()
	Construct an object with no argument.
	~MassFourCFitKFit (void) override
	Destruct the object.
enum KFitError::ECode	addMassConstraint (const double m, std::vector< unsigned > &childTrackId)
	Set an invariant mass of daughter particle for the mass-four-momentum-constraint fit.
enum KFitError::ECode	setVertex (const HepPoint3D &v)
	Set an initial vertex position for the mass-four-momentum-constraint fit.
enum KFitError::ECode	setVertexError (const CLHEP::HepSymMatrix &e)
	Set an initial vertex error matrix for the mass-four-momentum-constraint fit.
enum KFitError::ECode	setInvariantMass (const double m)
	Set an invariant mass for the mass-four-momentum-constraint fit.
enum KFitError::ECode	setFourMomentum (const ROOT::Math::PxPyPzEVector &m)
	Set an 4 Momentum for the mass-four-constraint fit.
enum KFitError::ECode	setFlagAtDecayPoint (const bool flag)
	Set a flag if to constraint at the decay point in the mass-four-momentum-constraint fit.
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	setTrackVertexError (const CLHEP::HepMatrix &e)
	Set a vertex error matrix of the child particle in the addTrack'ed order.
enum KFitError::ECode	setTrackZeroVertexError (void)
	Indicate no vertex uncertainty in the child particle in the addTrack'ed order.
const HepPoint3D	getVertex (const int flag=KFitConst::kAfterFit) const
	Get a vertex position.
const CLHEP::HepSymMatrix	getVertexError (const int flag=KFitConst::kAfterFit) const
	Get a vertex error matrix.
double	getInvariantMass (void) const
	Get an invariant mass.
bool	getFlagAtDecayPoint (void) const
	Get a flag if to constraint at the decay point in the mass-four-momentum-constraint fit.
bool	getFlagFitWithVertex (void) const
	Get a flag if the fit is allowed with moving the vertex position.
double	getCHIsq (void) const override
	Get a chi-square of the fit.
const CLHEP::HepMatrix	getTrackVertexError (const int id, const int flag=KFitConst::kAfterFit) 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-four-momentum-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.
virtual enum KFitError::ECode	setCorrelation (const CLHEP::HepMatrix &c)
	Set a correlation matrix.
virtual enum KFitError::ECode	setZeroCorrelation (void)
	Indicate no correlation between tracks.
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
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.

Static Protected Member Functions
static CLHEP::HepSymMatrix	makeError1 (const CLHEP::HepLorentzVector &p, const CLHEP::HepMatrix &e)
	Rebuild an error matrix from a Lorentz vector and an error matrix.
static CLHEP::HepMatrix	makeError1 (const CLHEP::HepLorentzVector &p1, const CLHEP::HepLorentzVector &p2, const CLHEP::HepMatrix &e)
	Rebuild an error matrix from a pair of Lorentz vectors and an error matrix.
static CLHEP::HepMatrix	makeError2 (const CLHEP::HepLorentzVector &p, const CLHEP::HepMatrix &e)
	Rebuild an error matrix from a Lorentz vector and an error matrix.
static CLHEP::HepSymMatrix	makeError3 (const CLHEP::HepLorentzVector &p, const CLHEP::HepMatrix &e, const bool is_fix_mass)
	Rebuild an error matrix from a Lorentz vector and an error matrix.
static 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)
	Rebuild an error matrix from a pair of Lorentz vectors and an error matrix.
static CLHEP::HepMatrix	makeError4 (const CLHEP::HepLorentzVector &p, const CLHEP::HepMatrix &e)
	Rebuild an error matrix from a Lorentz vector and an error matrix.
static bool	isNonZeroEnergy (const CLHEP::HepLorentzVector &p)
	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.
CLHEP::HepSymMatrix	m_BeforeVertexError
	Vertex error matrix before the fit.
std::vector< CLHEP::HepMatrix >	m_BeforeTrackVertexError
	array of vertex error matrices 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.
bool	m_FlagTrackVertexError
	Flag to indicate if the vertex error matrix of the child particle is preset.
bool	m_FlagFitIncludingVertex
	Flag to indicate if the fit is allowed with moving the vertex position.
bool	m_FlagAtDecayPoint
	Flag controlled by setFlagAtDecayPoint().
double	m_InvariantMass
	Invariant mass.
ROOT::Math::PxPyPzEVector	m_FourMomentum
	Four Momentum.
std::vector< int >	m_IsFixMass
	Array of flags whether the track property is fixed at the mass.
int	m_ConstraintMassCount
std::vector< double >	m_ConstraintMassList
	constrained mass
std::vector< std::pair< int, int > >	m_ConstraintMassChildLists
	Daughter track id of constrained particle.

Detailed Description

MassFourCFitKFit is a derived class from KFitBase to perform mass and 4 momentum-constraint kinematical fit.

Definition at line 32 of file MassFourCFitKFit.h.

Constructor & Destructor Documentation

MassFourCFitKFit()

MassFourCFitKFit

(

)

Construct an object with no argument.

Definition at line 26 of file MassFourCFitKFit.cc.

                                   : m_AfterVertexError(HepSymMatrix(3, 0)),
  m_FourMomentum(PxPyPzEVector())
{
  m_FlagFitted = false;
  m_FlagTrackVertexError = false;
  m_FlagFitIncludingVertex = false;
  m_FlagAtDecayPoint = true;
  m_NecessaryTrackCount = 2;
  m_InvariantMass = -1.0;
  m_ConstraintMassCount = 0;
  m_ConstraintMassList.reserve(KFitConst::kMaxTrackCount2);
  m_ConstraintMassChildLists.reserve(KFitConst::kMaxTrackCount2);
}

Member Function Documentation

addMassConstraint()

enum KFitError::ECode addMassConstraint	(	const double	m,
		std::vector< unsigned > &	childTrackId )

Set an invariant mass of daughter particle for the mass-four-momentum-constraint fit.

Parameters

m	invariant mass
childTrackId	vector of daughter track ids

Returns

error code (zero if success)

Definition at line 44 of file MassFourCFitKFit.cc.

                                                                                     {
  if ((childTrackId.back() - childTrackId.front()) != childTrackId.size() - 1)
  {
    return m_ErrorCode = KFitError::kOutOfRange;
  }
  m_ConstraintMassList.push_back(m);
  m_ConstraintMassChildLists.push_back(std::make_pair(childTrackId.front(), childTrackId.back()));
  return m_ErrorCode = KFitError::kNoError;
}

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 736 of file MassFourCFitKFit.cc.

                               {
  m_NDF = 4 + m_ConstraintMassCount;
 
  return m_ErrorCode = KFitError::kNoError;
}

doFit()

enum KFitError::ECode doFit

(

void

)

Perform a mass-four-momentum-constraint fit.

Returns

error code (zero if success)

Definition at line 287 of file MassFourCFitKFit.cc.

                        {
  return KFitBase::doFit1();
}

doFit1()

enum KFitError::ECode doFit1 ( void )

protectedinherited

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

Returns

error code (zero if success)

Definition at line 504 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 580 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 103 of file MassFourCFitKFit.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 200 of file MassFourCFitKFit.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 260 of file MassFourCFitKFit.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;
}

getFlagAtDecayPoint()

bool getFlagAtDecayPoint

(

void

)

const

Get a flag if to constraint at the decay point in the mass-four-momentum-constraint fit.

Returns

flag value

Definition at line 186 of file MassFourCFitKFit.cc.

{
  return m_FlagAtDecayPoint;
}

getFlagFitWithVertex()

bool getFlagFitWithVertex

(

void

)

const

Get a flag if the fit is allowed with moving the vertex position.

Returns

flag value

Definition at line 193 of file MassFourCFitKFit.cc.

{
  return m_FlagFitIncludingVertex;
}

getInvariantMass()

double getInvariantMass

(

void

)

const

Get an invariant mass.

Returns

invariant mass

Definition at line 179 of file MassFourCFitKFit.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 224 of file MassFourCFitKFit.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 int	flag = KFitConst::kAfterFit ) const

Get a vertex error matrix of the track.

Parameters

id	track id
flag	KFitConst::kBeforeFit or KFitConst::kAfterFit

Returns

vertex error matrix

Definition at line 207 of file MassFourCFitKFit.cc.

{
  if (flag == KFitConst::kAfterFit && !isFitted()) return HepMatrix(3, KFitConst::kNumber7, 0);
  if (!isTrackIDInRange(id)) return HepMatrix(3, KFitConst::kNumber7, 0);
 
  if (flag == KFitConst::kBeforeFit)
    return m_BeforeTrackVertexError[id];
  else if (flag == KFitConst::kAfterFit && m_FlagFitIncludingVertex)
    return m_AfterTrackVertexError[id];
  else {
    KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kOutOfRange);
    return HepMatrix(3, KFitConst::kNumber7, 0);
  }
}

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 144 of file MassFourCFitKFit.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

(

const int

flag = KFitConst::kAfterFit

)

const

Get a vertex error matrix.

Parameters

flag	KFitConst::kBeforeFit or KFitConst::kAfterFit

Returns

vertex error matrix

Definition at line 163 of file MassFourCFitKFit.cc.

{
  if (flag == KFitConst::kAfterFit && !isFitted()) return HepSymMatrix(3, 0);
 
  if (flag == KFitConst::kBeforeFit)
    return m_BeforeVertexError;
  else if (flag == KFitConst::kAfterFit && m_FlagFitIncludingVertex)
    return m_AfterVertexError;
  else {
    KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kOutOfRange);
    return HepSymMatrix(3, 0);
  }
}

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

staticprotectedinherited

Check if the energy is non-zero.

Parameters

p	Lorentz vector

Returns

true for non-zero energy, false for otherwise

Definition at line 752 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 741 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 554 of file MassFourCFitKFit.cc.

                                 {
  if (!m_FlagFitIncludingVertex)
  {
 
    HepMatrix al_1_prime(m_al_1);
    HepMatrix Sum_al_1(4, 1, 0);
    HepMatrix Sum_child_al_1(4 * m_ConstraintMassCount, 1, 0);
    std::vector<double> energy(m_TrackCount);
    double a;
 
    for (int i = 0; i < m_TrackCount; i++) {
      a = m_property[i][2];
      if (!m_FlagAtDecayPoint) a = 0.;
      al_1_prime[i * KFitConst::kNumber7 + 0][0] -= a * (m_BeforeVertex.y() - al_1_prime[i * KFitConst::kNumber7 + 5][0]);
      al_1_prime[i * KFitConst::kNumber7 + 1][0] += a * (m_BeforeVertex.x() - 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];
    }
 
    for (int i = 0; i < m_ConstraintMassCount; i++) {
      for (int k = m_ConstraintMassChildLists[i].first; k <= m_ConstraintMassChildLists[i].second; k++) {
        if (m_IsFixMass[k])
          Sum_child_al_1[i * 4 + 3][0] += energy[k];
        else
          Sum_child_al_1[i * 4 + 3][0] += al_1_prime[k * KFitConst::kNumber7 + 3][0];
        for (int j = 0; j < 3; j++) Sum_child_al_1[i * 4 + j][0] += al_1_prime[k * KFitConst::kNumber7 + j][0];
      }
    }
 
    m_d[0][0] = Sum_al_1[0][0]  - m_FourMomentum.Px();
    m_d[1][0] = Sum_al_1[1][0]  - m_FourMomentum.Py();
    m_d[2][0] = Sum_al_1[2][0]  - m_FourMomentum.Pz();
    m_d[3][0] = Sum_al_1[3][0]  - m_FourMomentum.E();
 
    for (int i = 0; i < m_ConstraintMassCount; i++) {
      m_d[4 + i][0] =
        + Sum_child_al_1[i * 4 + 3][0] * Sum_child_al_1[i * 4 + 3][0] - Sum_child_al_1[i * 4 + 0][0] * Sum_child_al_1[i * 4 + 0][0]
        - Sum_child_al_1[i * 4 + 1][0] * Sum_child_al_1[i * 4 + 1][0] - Sum_child_al_1[i * 4 + 2][0] * Sum_child_al_1[i * 4 + 2][0]
        - m_ConstraintMassList[i] * m_ConstraintMassList[i];
    }
 
    for (int i = 0; i < m_TrackCount; i++) {
      if (energy[i] == 0) {
        m_ErrorCode = KFitError::kDivisionByZero;
        KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
        break;
      }
 
      a = m_property[i][2];
      if (!m_FlagAtDecayPoint) a = 0.;
 
      // four-momentum conservation constraint
      for (int l = 0; l < 4; l++) {
        for (int n = 0; n < 6; n++) {
          m_D[l][i * KFitConst::kNumber7 + n] = 0;
        }
      }
      if (m_IsFixMass[i]) {
        double invE = 1. / energy[i];
        m_D[0][i * KFitConst::kNumber7 + 0] = 1;
        m_D[0][i * KFitConst::kNumber7 + 5] = -a;
        m_D[1][i * KFitConst::kNumber7 + 1] = 1;
        m_D[1][i * KFitConst::kNumber7 + 4] = a;
        m_D[2][i * KFitConst::kNumber7 + 2] = 1;
        m_D[3][i * KFitConst::kNumber7 + 0] = al_1_prime[i * KFitConst::kNumber7 + 0][0] * invE;
        m_D[3][i * KFitConst::kNumber7 + 1] = al_1_prime[i * KFitConst::kNumber7 + 1][0] * invE;
        m_D[3][i * KFitConst::kNumber7 + 2] = al_1_prime[i * KFitConst::kNumber7 + 2][0] * invE;
        m_D[3][i * KFitConst::kNumber7 + 4] = -al_1_prime[i * KFitConst::kNumber7 + 1][0] * invE * a;
        m_D[3][i * KFitConst::kNumber7 + 5] = al_1_prime[i * KFitConst::kNumber7 + 0][0] * invE * a;
      } else {
        m_D[0][i * KFitConst::kNumber7 + 0] = 1;
        m_D[1][i * KFitConst::kNumber7 + 1] = 1;
        m_D[2][i * KFitConst::kNumber7 + 2] = 1;
        m_D[3][i * KFitConst::kNumber7 + 3] = 1;
      }
 
      // invariant mass constraint
      for (int l = 0; l < m_ConstraintMassCount; l++) {
        if (i >= m_ConstraintMassChildLists[l].first && i <= m_ConstraintMassChildLists[l].second) {
          double invE = 1. / energy[i];
          m_D[4 + l][i * KFitConst::kNumber7 + 0] = 2.*(Sum_child_al_1[l * 4 + 3][0] * al_1_prime[i * KFitConst::kNumber7 + 0][0] * invE -
                                                        Sum_child_al_1[l * 4 + 0][0]);
          m_D[4 + l][i * KFitConst::kNumber7 + 1] = 2.*(Sum_child_al_1[l * 4 + 3][0] * al_1_prime[i * KFitConst::kNumber7 + 1][0] * invE -
                                                        Sum_child_al_1[l * 4 + 1][0]);
          m_D[4 + l][i * KFitConst::kNumber7 + 2] = 2.*(Sum_child_al_1[l * 4 + 3][0] * al_1_prime[i * KFitConst::kNumber7 + 2][0] * invE -
                                                        Sum_child_al_1[l * 4 + 2][0]);
          m_D[4 + l][i * KFitConst::kNumber7 + 3] = 0.;
          m_D[4 + l][i * KFitConst::kNumber7 + 4] = -2.*(Sum_child_al_1[l * 4 + 3][0] * al_1_prime[i * KFitConst::kNumber7 + 1][0] * invE -
                                                         Sum_child_al_1[l * 4 + 1][0]) * a;
          m_D[4 + l][i * KFitConst::kNumber7 + 5] =  2.*(Sum_child_al_1[l * 4 + 3][0] * al_1_prime[i * KFitConst::kNumber7 + 0][0] * invE -
                                                         Sum_child_al_1[l * 4 + 0][0]) * a;
          m_D[4 + l][i * KFitConst::kNumber7 + 6] = 0.;
        } else {
          for (int n = 0; n < 6; n++) {
            m_D[4 + l][i * KFitConst::kNumber7 + n] = 0;
          }
        }
      }
    }
 
  } else
  {
    //TODO: Not Implemented
    return m_ErrorCode = KFitError::kUnimplemented;
 
    // m_FlagFitIncludingVertex == true
    HepMatrix al_1_prime(m_al_1);
    HepMatrix Sum_al_1(7, 1, 0);
    double energy[KFitConst::kMaxTrackCount2];
 
    for (int i = 0; i < m_TrackCount; i++) {
      const double a = m_property[i][2];
      al_1_prime[i * KFitConst::kNumber7 + 0][0] -= a * (al_1_prime[KFitConst::kNumber7 * m_TrackCount + 1][0] - al_1_prime[i *
                                                         KFitConst::kNumber7 + 5][0]);
      al_1_prime[i * KFitConst::kNumber7 + 1][0] += a * (al_1_prime[KFitConst::kNumber7 * m_TrackCount + 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]);
      Sum_al_1[6][0] = + a;
    }
 
    for (int i = 0; i < m_TrackCount; i++) {
      if (energy[i] == 0) {
        m_ErrorCode = KFitError::kDivisionByZero;
        KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
        break;
      }
 
      if (m_IsFixMass[i]) {
        double invE = 1. / energy[i];
        Sum_al_1[3][0] += energy[i];
        Sum_al_1[4][0] += al_1_prime[i * KFitConst::kNumber7 + 1][0] * m_property[i][2] * invE;
        Sum_al_1[5][0] += al_1_prime[i * KFitConst::kNumber7 + 0][0] * m_property[i][2] * invE;
      } else {
        Sum_al_1[3][0] += al_1_prime[i * KFitConst::kNumber7 + 3][0];
      }
 
      for (int j = 0; j < 3; j++) Sum_al_1[j][0] += al_1_prime[i * KFitConst::kNumber7 + j][0];
    }
 
    m_d[0][0] = Sum_al_1[0][0]  - m_FourMomentum.Px();
    m_d[1][0] = Sum_al_1[1][0]  - m_FourMomentum.Py();
    m_d[2][0] = Sum_al_1[2][0]  - m_FourMomentum.Pz();
    m_d[3][0] = Sum_al_1[3][0]  - m_FourMomentum.E();
 
    for (int i = 0; i < m_TrackCount; i++) {
      if (energy[i] == 0) {
        m_ErrorCode = KFitError::kDivisionByZero;
        KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
        break;
      }
 
      for (int l = 0; l < 4; l++) {
        for (int n = 0; n < 6; n++) {
          if (l == n) m_D[l][i * KFitConst::kNumber7 + n] = 1;
          else m_D[l][i * KFitConst::kNumber7 + n] = 0;
        }
      }
    }
 
    m_D[0][KFitConst::kNumber7 * m_TrackCount + 0] = 2.*(Sum_al_1[3][0] * Sum_al_1[4][0] - Sum_al_1[1][0] * Sum_al_1[6][0]);
    m_D[0][KFitConst::kNumber7 * m_TrackCount + 1] = -2.*(Sum_al_1[3][0] * Sum_al_1[5][0] - Sum_al_1[0][0] * Sum_al_1[6][0]);
    m_D[0][KFitConst::kNumber7 * m_TrackCount + 2] = 0.;
  }
 
  return m_ErrorCode = KFitError::kNoError;
}

makeError1() [1/2]

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

staticprotectedinherited

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 223 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]

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

staticprotectedinherited

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

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

staticprotectedinherited

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 298 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]

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

staticprotectedinherited

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 322 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]

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

staticprotectedinherited

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

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

staticprotectedinherited

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 441 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 428 of file MassFourCFitKFit.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 (const 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++;
    }
  }
 
  if (m_FlagFitIncludingVertex)
  {
    //TODO: Not Implemented
    return m_ErrorCode = KFitError::kUnimplemented;
 
    // ...error of vertex
    m_V_al_0.sub(KFitConst::kNumber7 * m_TrackCount + 1, m_BeforeVertexError);
 
    // ...error matrix between vertex and tracks
    if (m_FlagTrackVertexError) {
      if (m_BeforeTrackVertexError.size() != (unsigned int)m_TrackCount) {
        m_ErrorCode = KFitError::kBadCorrelationSize;
        KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
        return m_ErrorCode;
      }
 
      int i = 0;
      for (const auto& hm : m_BeforeTrackVertexError) {
        for (int j = 0; j < 3; j++) for (int k = 0; k < KFitConst::kNumber7; k++) {
            m_V_al_0[j + KFitConst::kNumber7 * m_TrackCount][k + i * KFitConst::kNumber7] = hm[j][k];
          }
        i++;
      }
    }
  }
 
  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 293 of file MassFourCFitKFit.cc.

                                     {
  m_ConstraintMassCount = m_ConstraintMassList.size();
  m_d   = HepMatrix(4 + m_ConstraintMassCount, 1, 0);
  m_V_D = HepMatrix(4 + m_ConstraintMassCount, 4 + m_ConstraintMassCount, 0);
  m_lam = HepMatrix(4 + m_ConstraintMassCount, 1, 0);
 
  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;
  }
 
  if (!m_FlagFitIncludingVertex)
  {
    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 (const auto& track : m_Tracks) {
      // momentum x,y,z 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 = 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;
      }
    }
 
    // set member matrix
    m_al_1   = m_al_0;
 
    // define size of matrix
    m_V_al_1 = HepMatrix(KFitConst::kNumber7 * m_TrackCount, KFitConst::kNumber7 * m_TrackCount, 0);
    m_D      = m_V_al_1.sub(1, 4 + m_ConstraintMassCount, 1, KFitConst::kNumber7 * m_TrackCount);
 
  } else
  {
    //TODO: Not Implemented
    return m_ErrorCode = KFitError::kUnimplemented;
    // m_FlagFitIncludingVertex == true
    int index = 0;
    m_al_0     = HepMatrix(KFitConst::kNumber7 * m_TrackCount + 3, 1, 0);
    m_property = HepMatrix(m_TrackCount, 3, 0);
    m_V_al_0   = HepSymMatrix(KFitConst::kNumber7 * m_TrackCount + 3, 0);
 
    for (const auto& track : m_Tracks) {
      // momentum x,y,z 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 = Const::speedOfLight * 1e-4;
      m_property[index][2] = -c * m_MagneticField * track.getCharge();
      index++;
    }
 
    // vertex
    m_al_0[KFitConst::kNumber7 * m_TrackCount + 0][0] = m_BeforeVertex.x();
    m_al_0[KFitConst::kNumber7 * m_TrackCount + 1][0] = m_BeforeVertex.y();
    m_al_0[KFitConst::kNumber7 * m_TrackCount + 2][0] = m_BeforeVertex.z();
    m_V_al_0.sub(KFitConst::kNumber7 * m_TrackCount + 1, m_BeforeVertexError);
 
    // 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;
      }
    }
 
    // set member matrix
    m_al_1   = m_al_0;
 
    // define size of matrix
    m_V_al_1 = HepMatrix(KFitConst::kNumber7 * m_TrackCount + 3, KFitConst::kNumber7 * m_TrackCount + 3, 0);
    m_D      = m_V_al_1.sub(1, 4, 1, KFitConst::kNumber7 * m_TrackCount + 3);
  }
 
  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 417 of file MassFourCFitKFit.cc.

                                        { // unused
  char buf[1024];
  sprintf(buf, "%s:%s(): internal error; this function should never be called", __FILE__, __func__);
  B2FATAL(buf);
 
  /* NEVER REACHEd HERE */
  return KFitError::kOutOfRange;
}

prepareOutputMatrix()

enum KFitError::ECode prepareOutputMatrix ( void )

overrideprivatevirtual

Build an output error matrix.

Returns

error code (zero if success)

Implements KFitBase.

Definition at line 489 of file MassFourCFitKFit.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++;
  }
 
  if (m_FlagFitIncludingVertex)
  {
    //TODO: Not Implemented
    return m_ErrorCode = KFitError::kUnimplemented;
    // vertex
    m_AfterVertex.setX(m_al_1[KFitConst::kNumber7 * m_TrackCount + 0][0]);
    m_AfterVertex.setY(m_al_1[KFitConst::kNumber7 * m_TrackCount + 1][0]);
    m_AfterVertex.setZ(m_al_1[KFitConst::kNumber7 * m_TrackCount + 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_al_1[KFitConst::kNumber7 * m_TrackCount + i][KFitConst::kNumber7 * m_TrackCount + 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_V_al_1[KFitConst::kNumber7 * m_TrackCount + 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);
    }
  } else
  {
    // not fit
    m_AfterVertex = m_BeforeVertex;
  }
 
  return m_ErrorCode = KFitError::kNoError;
}

setCorrelation()

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

virtualinherited

Set a correlation matrix.

Not intended for end user's use.

Parameters

c	(7x7) correlation matrix

Returns

error code (zero if success)

Definition at line 70 of file KFitBase.cc.

                                           {
  if (e.num_row() != KFitConst::kNumber7)
  {
    m_ErrorCode = KFitError::kBadMatrixSize;
    KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
    return m_ErrorCode;
  }
  m_BeforeCorrelation.push_back(e);
  m_FlagCorrelation = true;
 
  return m_ErrorCode = KFitError::kNoError;
}

setFlagAtDecayPoint()

enum KFitError::ECode setFlagAtDecayPoint

(

const bool

flag

)

Set a flag if to constraint at the decay point in the mass-four-momentum-constraint fit.

Parameters

flag	true for constraint, false for not

Returns

error code (zero if success)

Definition at line 95 of file MassFourCFitKFit.cc.

                                                     {
  m_FlagAtDecayPoint = flag;
 
  return m_ErrorCode = KFitError::kNoError;
}

setFourMomentum()

enum KFitError::ECode setFourMomentum

(

const ROOT::Math::PxPyPzEVector &

m

)

Set an 4 Momentum for the mass-four-constraint fit.

Parameters

m	four momentum

Returns

error code (zero if success)

Definition at line 87 of file MassFourCFitKFit.cc.

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

setInvariantMass()

enum KFitError::ECode setInvariantMass

(

const double

m

)

Set an invariant mass for the mass-four-momentum-constraint fit.

Parameters

m	invariant mass

Returns

error code (zero if success)

Definition at line 79 of file MassFourCFitKFit.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;
}

setTrackVertexError()

enum KFitError::ECode setTrackVertexError

(

const CLHEP::HepMatrix &

e

)

Set a vertex error matrix of the child particle in the addTrack'ed order.

Parameters

e	(3x7) vertex error matrix

Returns

error code (zero if success)

Definition at line 119 of file MassFourCFitKFit.cc.

                                                        {
  if (e.num_row() != 3 || e.num_col() != KFitConst::kNumber7)
  {
    m_ErrorCode = KFitError::kBadMatrixSize;
    KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
    return m_ErrorCode;
  }
 
  m_BeforeTrackVertexError.push_back(e);
  m_FlagTrackVertexError = true;
  m_FlagFitIncludingVertex = true;
 
  return m_ErrorCode = KFitError::kNoError;
}

setTrackZeroVertexError()

enum KFitError::ECode setTrackZeroVertexError

(

void

)

Indicate no vertex uncertainty in the child particle in the addTrack'ed order.

Returns

error code (zero if success)

Definition at line 136 of file MassFourCFitKFit.cc.

                                          {
  HepMatrix zero(3, KFitConst::kNumber7, 0);
 
  return this->setTrackVertexError(zero);
}

setVertex()

enum KFitError::ECode setVertex

(

const HepPoint3D &

v

)

Set an initial vertex position for the mass-four-momentum-constraint fit.

Parameters

v	vertex position

Returns

error code (zero if success)

Definition at line 55 of file MassFourCFitKFit.cc.

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

setVertexError()

enum KFitError::ECode setVertexError

(

const CLHEP::HepSymMatrix &

e

)

Set an initial vertex error matrix for the mass-four-momentum-constraint fit.

Parameters

e	vertex error matrix

Returns

error code (zero if success)

Definition at line 63 of file MassFourCFitKFit.cc.

                                                      {
  if (e.num_row() != 3)
  {
    m_ErrorCode = KFitError::kBadMatrixSize;
    KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
    return m_ErrorCode;
  }
 
  m_BeforeVertexError = e;
  m_FlagFitIncludingVertex = true;
 
  return m_ErrorCode = KFitError::kNoError;
}

setZeroCorrelation()

enum KFitError::ECode setZeroCorrelation ( void )

virtualinherited

Indicate no correlation between tracks.

Not intended for end user's use.

Returns

error code (zero if success)

Definition at line 85 of file KFitBase.cc.

                             {
  HepMatrix zero(KFitConst::kNumber7, KFitConst::kNumber7, 0);
 
  return this->setCorrelation(zero);
}

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 111 of file MassFourCFitKFit.cc.

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

updateMother()

enum KFitError::ECode updateMother

(

Particle *

mother

)

Update mother particle.

Parameters

[in]

mother

Mother particle.

Definition at line 742 of file MassFourCFitKFit.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());
    if (getFlagFitWithVertex())
      kmm.setTrackVertexError(getTrackVertexError(i));
    for (unsigned j = i + 1; j < n; ++j) {
      kmm.setCorrelation(getCorrelation(i, j));
    }
  }
  kmm.setVertex(getVertex());
  if (getFlagFitWithVertex())
    kmm.setVertexError(getVertexError());
  m_ErrorCode = kmm.doMake();
  if (m_ErrorCode != KFitError::kNoError)
    return m_ErrorCode;
  double chi2 = getCHIsq();
  int ndf = getNDF();
  double prob = TMath::Prob(chi2, ndf);
  //
  bool haschi2 = mother->hasExtraInfo("chiSquared");
  if (haschi2) {
    mother->setExtraInfo("chiSquared", chi2);
    mother->setExtraInfo("ndf", ndf);
  } else {
    mother->addExtraInfo("chiSquared", chi2);
    mother->addExtraInfo("ndf", ndf);
  }
 
  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 162 of file MassFourCFitKFit.h.

m_AfterVertex

HepPoint3D m_AfterVertex

private

Vertex position after the fit.

Definition at line 158 of file MassFourCFitKFit.h.

m_AfterVertexError

CLHEP::HepSymMatrix m_AfterVertexError

private

Vertex error matrix after the fit.

Definition at line 160 of file MassFourCFitKFit.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_BeforeTrackVertexError

std::vector<CLHEP::HepMatrix> m_BeforeTrackVertexError

private

array of vertex error matrices before the fit.

Definition at line 155 of file MassFourCFitKFit.h.

m_BeforeVertex

HepPoint3D m_BeforeVertex

private

Vertex position before the fit.

Definition at line 151 of file MassFourCFitKFit.h.

m_BeforeVertexError

CLHEP::HepSymMatrix m_BeforeVertexError

private

Vertex error matrix before the fit.

Definition at line 153 of file MassFourCFitKFit.h.

m_CHIsq

double m_CHIsq

protectedinherited

chi-square of the fit.

Definition at line 297 of file KFitBase.h.

m_ConstraintMassChildLists

std::vector<std::pair<int, int> > m_ConstraintMassChildLists

private

Daughter track id of constrained particle.

Definition at line 187 of file MassFourCFitKFit.h.

m_ConstraintMassCount

int m_ConstraintMassCount

private

of constrained mass .

Definition at line 181 of file MassFourCFitKFit.h.

m_ConstraintMassList

std::vector<double> m_ConstraintMassList

private

constrained mass

Definition at line 184 of file MassFourCFitKFit.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_FlagAtDecayPoint

bool m_FlagAtDecayPoint

private

Flag controlled by setFlagAtDecayPoint().

Definition at line 169 of file MassFourCFitKFit.h.

m_FlagCorrelation

bool m_FlagCorrelation

protectedinherited

Flag whether a correlation among tracks exists.

Definition at line 306 of file KFitBase.h.

m_FlagFitIncludingVertex

bool m_FlagFitIncludingVertex

private

Flag to indicate if the fit is allowed with moving the vertex position.

Definition at line 167 of file MassFourCFitKFit.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_FlagTrackVertexError

bool m_FlagTrackVertexError

private

Flag to indicate if the vertex error matrix of the child particle is preset.

Definition at line 165 of file MassFourCFitKFit.h.

m_FourMomentum

ROOT::Math::PxPyPzEVector m_FourMomentum

private

Four Momentum.

Definition at line 175 of file MassFourCFitKFit.h.

m_InvariantMass

double m_InvariantMass

private

Invariant mass.

Definition at line 172 of file MassFourCFitKFit.h.

m_IsFixMass

std::vector<int> m_IsFixMass

private

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

Definition at line 178 of file MassFourCFitKFit.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_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/MassFourCFitKFit.h
• analysis/VertexFitting/KFit/src/MassFourCFitKFit.cc