Belle II Software light-2607-kasei
Public Member Functions | Protected Member Functions | Static Protected Member Functions | Protected Attributes | Private Member Functions | Private Attributes | List of all members
MassFitKFit Class Reference

MassFitKFit is a derived class from KFitBase to perform mass-constraint kinematical fit. More...

#include <MassFitKFit.h>

Inheritance diagram for MassFitKFit:
Inheritance graph
[legend]
Collaboration diagram for MassFitKFit:
Collaboration graph
[legend]

Public Member Functions

 MassFitKFit (void)
 Construct an object with no argument.
 
 ~MassFitKFit (void) override
 Destruct the object.
 
enum KFitError::ECode setVertex (const HepPoint3D &v)
 Set an initial vertex position for the mass-constraint fit.
 
enum KFitError::ECode setVertexError (const CLHEP::HepSymMatrix &e)
 Set an initial vertex error matrix for the mass-constraint fit.
 
enum KFitError::ECode setInvariantMass (const double m)
 Set an invariant mass for the mass-constraint fit.
 
enum KFitError::ECode setFlagAtDecayPoint (const bool flag)
 Set a flag if to constraint at the decay point in the mass-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-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-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.
 
std::vector< int > m_IsFixMass
 Array of flags whether the track property is fixed at the mass.
 

Detailed Description

MassFitKFit is a derived class from KFitBase to perform mass-constraint kinematical fit.

Definition at line 33 of file MassFitKFit.h.

Constructor & Destructor Documentation

◆ MassFitKFit()

MassFitKFit ( void )

Construct an object with no argument.

Definition at line 26 of file MassFitKFit.cc.

27{
28 m_FlagFitted = false;
29 m_FlagTrackVertexError = false;
30 m_FlagFitIncludingVertex = false;
31 m_FlagAtDecayPoint = true;
32 m_NecessaryTrackCount = 2;
33 m_d = HepMatrix(1, 1, 0);
34 m_V_D = HepMatrix(1, 1, 0);
35 m_lam = HepMatrix(1, 1, 0);
36 m_AfterVertexError = HepSymMatrix(3, 0);
37 m_InvariantMass = -1.0;
38}
Belle2::analysis::KFitBase::m_NecessaryTrackCount
int m_NecessaryTrackCount
Number needed tracks to perform fit.
Definition KFitBase.h:303
Belle2::analysis::KFitBase::m_lam
CLHEP::HepMatrix m_lam
See J.Tanaka Ph.D (2001) p137 for definition.
Definition KFitBase.h:276
Belle2::analysis::KFitBase::m_d
CLHEP::HepMatrix m_d
See J.Tanaka Ph.D (2001) p137 for definition.
Definition KFitBase.h:268
Belle2::analysis::KFitBase::m_V_D
CLHEP::HepMatrix m_V_D
See J.Tanaka Ph.D (2001) p138 for definition.
Definition KFitBase.h:271
Belle2::analysis::KFitBase::m_FlagFitted
bool m_FlagFitted
Flag to indicate if the fit is performed and succeeded.
Definition KFitBase.h:245
Belle2::analysis::MassFitKFit::m_FlagTrackVertexError
bool m_FlagTrackVertexError
Flag to indicate if the vertex error matrix of the child particle is preset.
Definition MassFitKFit.h:153
Belle2::analysis::MassFitKFit::m_FlagAtDecayPoint
bool m_FlagAtDecayPoint
Flag controlled by setFlagAtDecayPoint().
Definition MassFitKFit.h:157
Belle2::analysis::MassFitKFit::m_FlagFitIncludingVertex
bool m_FlagFitIncludingVertex
Flag to indicate if the fit is allowed with moving the vertex position.
Definition MassFitKFit.h:155
Belle2::analysis::MassFitKFit::m_InvariantMass
double m_InvariantMass
Invariant mass.
Definition MassFitKFit.h:160
Belle2::analysis::MassFitKFit::m_AfterVertexError
CLHEP::HepSymMatrix m_AfterVertexError
Vertex error matrix after the fit.
Definition MassFitKFit.h:148

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]particleParticle.
Returns
error code (zero if success)

Definition at line 59 of file KFitBase.cc.

60{
61 return addTrack(
62 ROOTToCLHEP::getHepLorentzVector(particle->get4Vector()),
63 ROOTToCLHEP::getPoint3D(particle->getVertex()),
64 ROOTToCLHEP::getHepSymMatrix(particle->getMomentumVertexErrorMatrix()),
65 particle->getCharge());
66}
Belle2::Particle::getVertex
ROOT::Math::XYZVector getVertex() const
Returns vertex position (POCA for charged, IP for neutral FS particles)
Definition Particle.h:651
Belle2::Particle::getCharge
double getCharge(void) const
Returns particle charge.
Definition Particle.cc:653
Belle2::Particle::get4Vector
ROOT::Math::PxPyPzEVector get4Vector() const
Returns Lorentz vector.
Definition Particle.h:567
Belle2::Particle::getMomentumVertexErrorMatrix
TMatrixFSym getMomentumVertexErrorMatrix() const
Returns 7x7 error matrix.
Definition Particle.cc:451
Belle2::analysis::KFitBase::addTrack
enum KFitError::ECode addTrack(const KFitTrack &kp)
Add a track to the fitter object.
Definition KFitBase.cc:38

◆ 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
pLorentz vector of the track
xposition of the track
e(7x7) error matrix of the track
qcharge of the track
Returns
error code (zero if success)

Definition at line 47 of file KFitBase.cc.

47 {
48 if (e.num_row() != KFitConst::kNumber7)
49 {
50 m_ErrorCode = KFitError::kBadMatrixSize;
51 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
52 return m_ErrorCode;
53 }
54
55 return this->addTrack(KFitTrack(p, x, e, q));
56}
Belle2::analysis::KFitBase::m_ErrorCode
enum KFitError::ECode m_ErrorCode
Error code.
Definition KFitBase.h:243
Belle2::analysis::KFitError::displayError
static void displayError(const char *file, const int line, const char *func, const enum ECode code)
Display a description of error and its location.
Definition KFitError.h:71
Belle2::analysis::KFitError::kBadMatrixSize
@ kBadMatrixSize
Wrong correlation matrix size.
Definition KFitError.h:48
Belle2::analysis::KFitConst::kNumber7
static const int kNumber7
Constant 7 to check matrix size (internal use)
Definition KFitConst.h:30

◆ addTrack() [2/2]

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

Add a track to the fitter object.

Parameters
kpobject of the track
Returns
error code (zero if success)

Definition at line 38 of file KFitBase.cc.

38 {
39 m_Tracks.push_back(p);
40 m_TrackCount = m_Tracks.size();
41
42 return m_ErrorCode = KFitError::kNoError;
43}
Belle2::analysis::KFitBase::m_Tracks
std::vector< KFitTrack > m_Tracks
Container of input tracks.
Definition KFitBase.h:249
Belle2::analysis::KFitBase::m_TrackCount
int m_TrackCount
Number of tracks.
Definition KFitBase.h:301

◆ calculateNDF()

enum KFitError::ECode calculateNDF ( void )
overrideprivatevirtual

Calculate an NDF of the fit.

Returns
error code (zero if success)

Implements KFitBase.

Definition at line 674 of file MassFitKFit.cc.

674 {
675 m_NDF = 1;
676
677 return m_ErrorCode = KFitError::kNoError;
678}
Belle2::analysis::KFitBase::m_NDF
int m_NDF
NDF of the fit.
Definition KFitBase.h:295

◆ doFit()

enum KFitError::ECode doFit ( void )

Perform a mass-constraint fit.

Returns
error code (zero if success)

Definition at line 269 of file MassFitKFit.cc.

269 {
270 return KFitBase::doFit1();
271}
Belle2::analysis::KFitBase::doFit1
enum KFitError::ECode doFit1(void)
Perform a fit (used in MassFitKFit::doFit()).
Definition KFitBase.cc:504

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

504 {
505 if (m_ErrorCode != KFitError::kNoError) return m_ErrorCode;
506
507 if (m_TrackCount < m_NecessaryTrackCount)
508 {
509 m_ErrorCode = KFitError::kBadTrackSize;
510 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
511 return m_ErrorCode;
512 }
513
514 if (prepareInputMatrix() != KFitError::kNoError) return m_ErrorCode;
515 if (calculateNDF() != KFitError::kNoError) return m_ErrorCode;
516
517
518 double chisq = 0;
519 double tmp_chisq = KFitConst::kInitialCHIsq;
520 int err_inverse = 0;
521
522 HepMatrix tmp_al_1(m_al_1);
523 HepMatrix tmp_V_al_1(m_V_al_1);
524
525 m_al_a = m_al_0;
526 HepMatrix tmp_al_a(m_al_a);
527
528
529 for (int i = 0; i < KFitConst::kMaxIterationCount; i++)
530 {
531 if (makeCoreMatrix() != KFitError::kNoError) return m_ErrorCode;
532
533 m_V_D = (m_V_al_0.similarity(m_D)).inverse(err_inverse);
534 if (err_inverse != 0) {
535 m_ErrorCode = KFitError::kCannotGetMatrixInverse;
536 return m_ErrorCode;
537 }
538
539 m_lam = m_V_D * (m_D * (m_al_0 - m_al_1) + m_d);
540 chisq = ((m_lam.T()) * (m_D * (m_al_0 - m_al_1) + m_d))(1, 1);
541 m_al_1 = m_al_0 - m_V_al_0 * (m_D.T()) * m_lam;
542 m_V_al_1 = m_V_al_0 - m_V_al_0 * (m_D.T()) * m_V_D * m_D * m_V_al_0;
543
544 if (tmp_chisq <= chisq) {
545 if (i == 0) {
546 m_ErrorCode = KFitError::kBadInitialCHIsq;
547 return m_ErrorCode;
548 } else {
549 chisq = tmp_chisq;
550 m_al_1 = tmp_al_1;
551 m_al_a = tmp_al_a;
552 m_V_al_1 = tmp_V_al_1;
553 break;
554 }
555 } else {
556 tmp_chisq = chisq;
557 tmp_al_a = tmp_al_1;
558 tmp_al_1 = m_al_1;
559 tmp_V_al_1 = m_V_al_1;
560 if (i == KFitConst::kMaxIterationCount - 1) {
561 m_al_a = tmp_al_1;
562 m_FlagOverIteration = true;
563 }
564 }
565 }
566
567 if (m_ErrorCode != KFitError::kNoError) return m_ErrorCode;
568
569 if (prepareOutputMatrix() != KFitError::kNoError) return m_ErrorCode;
570
571 m_CHIsq = chisq;
572
573 m_FlagFitted = true;
574
575 return m_ErrorCode = KFitError::kNoError;
576}
Belle2::analysis::KFitBase::prepareInputMatrix
virtual enum KFitError::ECode prepareInputMatrix(void)=0
Build grand matrices for minimum search from input-track properties.
Belle2::analysis::KFitBase::prepareOutputMatrix
virtual enum KFitError::ECode prepareOutputMatrix(void)=0
Build an output error matrix.
Belle2::analysis::KFitBase::m_al_1
CLHEP::HepMatrix m_al_1
See J.Tanaka Ph.D (2001) p136 for definition.
Definition KFitBase.h:259
Belle2::analysis::KFitBase::m_FlagOverIteration
bool m_FlagOverIteration
Flag whether the iteration count exceeds the limit.
Definition KFitBase.h:308
Belle2::analysis::KFitBase::m_V_al_1
CLHEP::HepMatrix m_V_al_1
See J.Tanaka Ph.D (2001) p138 for definition.
Definition KFitBase.h:274
Belle2::analysis::KFitBase::m_al_a
CLHEP::HepMatrix m_al_a
See J.Tanaka Ph.D (2001) p137 for definition.
Definition KFitBase.h:261
Belle2::analysis::KFitBase::m_D
CLHEP::HepMatrix m_D
See J.Tanaka Ph.D (2001) p137 for definition.
Definition KFitBase.h:266
Belle2::analysis::KFitBase::m_V_al_0
CLHEP::HepSymMatrix m_V_al_0
See J.Tanaka Ph.D (2001) p137 for definition.
Definition KFitBase.h:255
Belle2::analysis::KFitBase::makeCoreMatrix
virtual enum KFitError::ECode makeCoreMatrix(void)=0
Build matrices using the kinematical constraint.
Belle2::analysis::KFitBase::m_CHIsq
double m_CHIsq
chi-square of the fit.
Definition KFitBase.h:297
Belle2::analysis::KFitBase::calculateNDF
virtual enum KFitError::ECode calculateNDF(void)=0
Calculate an NDF of the fit.
Belle2::analysis::KFitBase::m_al_0
CLHEP::HepMatrix m_al_0
See J.Tanaka Ph.D (2001) p136 for definition.
Definition KFitBase.h:257
Belle2::analysis::KFitError::kCannotGetMatrixInverse
@ kCannotGetMatrixInverse
Cannot calculate matrix inverse (bad track property or internal error)
Definition KFitError.h:57
Belle2::analysis::KFitError::kBadInitialCHIsq
@ kBadInitialCHIsq
Bad initial chi-square (internal error)
Definition KFitError.h:52
Belle2::analysis::KFitError::kBadTrackSize
@ kBadTrackSize
Track count too small to perform fit.
Definition KFitError.h:46
Belle2::analysis::KFitConst::kInitialCHIsq
static constexpr double kInitialCHIsq
Initial chi-square value (internal use)
Definition KFitConst.h:46
Belle2::analysis::KFitConst::kMaxIterationCount
static const int kMaxIterationCount
Maximum iteration step (internal use)
Definition KFitConst.h:43

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

580 {
581 if (m_ErrorCode != KFitError::kNoError) return m_ErrorCode;
582
583 if (m_TrackCount < m_NecessaryTrackCount)
584 {
585 m_ErrorCode = KFitError::kBadTrackSize;
586 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
587 return m_ErrorCode;
588 }
589
590 if (prepareInputMatrix() != KFitError::kNoError) return m_ErrorCode;
591 if (calculateNDF() != KFitError::kNoError) return m_ErrorCode;
592
593
594 double chisq = 0;
595 double tmp2_chisq = KFitConst::kInitialCHIsq;
596 int err_inverse = 0;
597
598 m_al_a = m_al_0;
599 HepMatrix tmp_al_a(m_al_a);
600
601 HepMatrix tmp_D(m_D), tmp_E(m_E);
602 HepMatrix tmp_V_D(m_V_D), tmp_V_E(m_V_E);
603 HepMatrix tmp_lam0(m_lam0), tmp_v_a(m_v_a);
604
605 HepMatrix tmp2_D(m_D), tmp2_E(m_E);
606 HepMatrix tmp2_V_D(m_V_D), tmp2_V_E(m_V_E);
607 HepMatrix tmp2_lam0(m_lam0), tmp2_v_a(m_v_a), tmp2_v(m_v_a);
608
609
610 for (int j = 0; j < KFitConst::kMaxIterationCount; j++) // j'th loop start
611 {
612
613 double tmp_chisq = KFitConst::kInitialCHIsq;
614
615 for (int i = 0; i < KFitConst::kMaxIterationCount; i++) { // i'th loop start
616
617 if (prepareInputSubMatrix() != KFitError::kNoError) return m_ErrorCode;
618 if (makeCoreMatrix() != KFitError::kNoError) return m_ErrorCode;
619
620 m_V_D = (m_V_al_0.similarity(m_D)).inverse(err_inverse);
621 if (err_inverse) {
622 m_ErrorCode = KFitError::kCannotGetMatrixInverse;
623 return m_ErrorCode;
624 }
625
626 m_V_E = ((m_E.T()) * m_V_D * m_E).inverse(err_inverse);
627 if (err_inverse) {
628 m_ErrorCode = KFitError::kCannotGetMatrixInverse;
629 return m_ErrorCode;
630 }
631 m_lam0 = m_V_D * (m_D * (m_al_0 - m_al_1) + m_d);
632 chisq = ((m_lam0.T()) * (m_D * (m_al_0 - m_al_1) + m_E * (m_v - m_v_a) + m_d))(1, 1);
633 m_v_a = m_v_a - m_V_E * (m_E.T()) * m_lam0;
634
635 if (tmp_chisq <= chisq) {
636 if (i == 0) {
637 m_ErrorCode = KFitError::kBadInitialCHIsq;
638 return m_ErrorCode;
639 } else {
640 chisq = tmp_chisq;
641 m_v_a = tmp_v_a;
642 m_V_E = tmp_V_E;
643 m_V_D = tmp_V_D;
644 m_lam0 = tmp_lam0;
645 m_E = tmp_E;
646 m_D = tmp_D;
647 break;
648 }
649 } else {
650 tmp_chisq = chisq;
651 tmp_v_a = m_v_a;
652 tmp_V_E = m_V_E;
653 tmp_V_D = m_V_D;
654 tmp_lam0 = m_lam0;
655 tmp_E = m_E;
656 tmp_D = m_D;
657 if (i == KFitConst::kMaxIterationCount - 1) {
658 m_FlagOverIteration = true;
659 }
660 }
661 } // i'th loop over
662
663
664 m_al_a = m_al_1;
665 m_lam = m_lam0 - m_V_D * m_E * m_V_E * (m_E.T()) * m_lam0;
666 m_al_1 = m_al_0 - m_V_al_0 * (m_D.T()) * m_lam;
667
668 if (j == 0) {
669
670 tmp2_chisq = chisq;
671 tmp2_v_a = m_v_a;
672 tmp2_v = m_v;
673 tmp2_V_E = m_V_E;
674 tmp2_V_D = m_V_D;
675 tmp2_lam0 = m_lam0;
676 tmp2_E = m_E;
677 tmp2_D = m_D;
678 tmp_al_a = m_al_a;
679
680 } else {
681
682 if (tmp2_chisq <= chisq) {
683 chisq = tmp2_chisq;
684 m_v_a = tmp2_v_a;
685 m_v = tmp2_v;
686 m_V_E = tmp2_V_E;
687 m_V_D = tmp2_V_D;
688 m_lam0 = tmp2_lam0;
689 m_E = tmp2_E;
690 m_D = tmp2_D;
691 m_al_a = tmp_al_a;
692 break;
693 } else {
694 tmp2_chisq = chisq;
695 tmp2_v_a = m_v_a;
696 tmp2_v = m_v;
697 tmp2_V_E = m_V_E;
698 tmp2_V_D = m_V_D;
699 tmp2_lam0 = m_lam0;
700 tmp2_E = m_E;
701 tmp2_D = m_D;
702 tmp_al_a = m_al_a;
703 if (j == KFitConst::kMaxIterationCount - 1) {
704 m_FlagOverIteration = true;
705 }
706 }
707 }
708 } // j'th loop over
709
710
711 if (m_ErrorCode != KFitError::kNoError) return m_ErrorCode;
712
713 m_lam = m_lam0 - m_V_D * m_E * m_V_E * (m_E.T()) * m_lam0;
714 m_al_1 = m_al_0 - m_V_al_0 * (m_D.T()) * m_lam;
715 m_V_Dt = m_V_D - m_V_D * m_E * m_V_E * (m_E.T()) * m_V_D;
716 m_V_al_1 = m_V_al_0 - m_V_al_0 * (m_D.T()) * m_V_Dt * m_D * m_V_al_0;
717 m_Cov_v_al_1 = -m_V_E * (m_E.T()) * m_V_D * m_D * m_V_al_0;
718
719 if (prepareOutputMatrix() != KFitError::kNoError) return m_ErrorCode;
720
721 m_CHIsq = chisq;
722
723 m_FlagFitted = true;
724
725 return m_ErrorCode = KFitError::kNoError;
726}
Belle2::analysis::KFitBase::m_V_Dt
CLHEP::HepMatrix m_V_Dt
See J.Tanaka Ph.D (2001) p138 for definition.
Definition KFitBase.h:289
Belle2::analysis::KFitBase::m_E
CLHEP::HepMatrix m_E
See J.Tanaka Ph.D (2001) p137 for definition.
Definition KFitBase.h:279
Belle2::analysis::KFitBase::prepareInputSubMatrix
virtual enum KFitError::ECode prepareInputSubMatrix(void)=0
Build sub-matrices for minimum search from input-track properties.
Belle2::analysis::KFitBase::m_lam0
CLHEP::HepMatrix m_lam0
See J.Tanaka Ph.D (2001) p138 for definition.
Definition KFitBase.h:283
Belle2::analysis::KFitBase::m_v_a
CLHEP::HepMatrix m_v_a
See J.Tanaka Ph.D (2001) p137 for definition.
Definition KFitBase.h:287
Belle2::analysis::KFitBase::m_V_E
CLHEP::HepMatrix m_V_E
See J.Tanaka Ph.D (2001) p138 for definition.
Definition KFitBase.h:281
Belle2::analysis::KFitBase::m_Cov_v_al_1
CLHEP::HepMatrix m_Cov_v_al_1
See J.Tanaka Ph.D (2001) p137 for definition.
Definition KFitBase.h:291
Belle2::analysis::KFitBase::m_v
CLHEP::HepMatrix m_v
See J.Tanaka Ph.D (2001) p137 for definition.
Definition KFitBase.h:285

◆ 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 85 of file MassFitKFit.cc.

85 {
86 m_IsFixMass.push_back(true);
87
88 return m_ErrorCode = KFitError::kNoError;
89}
Belle2::analysis::MassFitKFit::m_IsFixMass
std::vector< int > m_IsFixMass
Array of flags whether the track property is fixed at the mass.
Definition MassFitKFit.h:162

◆ 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 182 of file MassFitKFit.cc.

183{
184 return m_CHIsq;
185}

◆ 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
id1first track id
id2second track id
flagKFitConst::kBeforeFit or KFitConst::kAfterFit
Returns
(7x7) correlation matrix

Reimplemented from KFitBase.

Definition at line 242 of file MassFitKFit.cc.

243{
244 if (flag == KFitConst::kAfterFit && !isFitted()) return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
245 if (!isTrackIDInRange(id1)) return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
246 if (!isTrackIDInRange(id2)) return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
247
248 switch (flag) {
249 case KFitConst::kBeforeFit:
250 return KFitBase::getCorrelation(id1, id2, flag);
251
252 case KFitConst::kAfterFit:
253 return makeError3(
254 this->getTrackMomentum(id1),
255 this->getTrackMomentum(id2),
256 m_V_al_1.sub(KFitConst::kNumber7 * id1 + 1, KFitConst::kNumber7 * (id1 + 1), KFitConst::kNumber7 * id2 + 1,
257 KFitConst::kNumber7 * (id2 + 1)),
258 m_IsFixMass[id1],
259 m_IsFixMass[id2]);
260
261 default:
262 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kOutOfRange);
263 return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
264 }
265}
Belle2::analysis::KFitBase::makeError3
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.
Definition KFitBase.cc:322
Belle2::analysis::KFitBase::getTrackMomentum
const CLHEP::HepLorentzVector getTrackMomentum(const int id) const
Get a Lorentz vector of the track.
Definition KFitBase.cc:154
Belle2::analysis::KFitBase::isFitted
bool isFitted(void) const
Return false if fit is not performed yet or performed fit is failed; otherwise true.
Definition KFitBase.cc:730
Belle2::analysis::KFitBase::isTrackIDInRange
bool isTrackIDInRange(const int id) const
Check if the id is in the range.
Definition KFitBase.cc:741
Belle2::analysis::KFitBase::getCorrelation
virtual const CLHEP::HepMatrix getCorrelation(const int id1, const int id2, const int flag=KFitConst::kAfterFit) const
Get a correlation matrix between two tracks.
Definition KFitBase.cc:183
Belle2::analysis::KFitError::kOutOfRange
@ kOutOfRange
Specified track-id out of range.
Definition KFitError.h:41
Belle2::analysis::KFitConst::kAfterFit
static const int kAfterFit
Input parameter to specify after-fit when setting/getting a track attribute.
Definition KFitConst.h:35
Belle2::analysis::KFitConst::kBeforeFit
static const int kBeforeFit
Input parameter to specify before-fit when setting/getting a track attribute.
Definition KFitConst.h:33

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

101 {
102 return m_ErrorCode;
103}

◆ getFlagAtDecayPoint()

bool getFlagAtDecayPoint ( void ) const

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

Returns
flag value

Definition at line 168 of file MassFitKFit.cc.

169{
170 return m_FlagAtDecayPoint;
171}

◆ getFlagFitWithVertex()

bool getFlagFitWithVertex ( void ) const

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

Returns
flag value

Definition at line 175 of file MassFitKFit.cc.

176{
177 return m_FlagFitIncludingVertex;
178}

◆ getInvariantMass()

double getInvariantMass ( void ) const

Get an invariant mass.

Returns
invariant mass

Definition at line 161 of file MassFitKFit.cc.

162{
163 return m_InvariantMass;
164}

◆ getMagneticField()

double getMagneticField ( void ) const
inherited

Get a magnetic field.

Returns
magnetic field

Definition at line 128 of file KFitBase.cc.

129{
130 return m_MagneticField;
131}
Belle2::analysis::KFitBase::m_MagneticField
double m_MagneticField
Magnetic field.
Definition KFitBase.h:311

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

115{
116 return m_NDF;
117}

◆ getTrack()

const KFitTrack getTrack ( const int id) const
inherited

Get a specified track object.

Parameters
idtrack id
Returns
specified track object

Definition at line 175 of file KFitBase.cc.

176{
177 if (!isTrackIDInRange(id)) return KFitTrack();
178 return m_Tracks[id];
179}

◆ getTrackCHIsq()

double getTrackCHIsq ( const int id) const
overridevirtual

Get a chi-square of the track.

Parameters
idtrack id
Returns
chi-square of the track

Reimplemented from KFitBase.

Definition at line 206 of file MassFitKFit.cc.

207{
208 if (!isFitted()) return -1;
209 if (!isTrackIDInRange(id)) return -1;
210
211 if (m_IsFixMass[id]) {
212
213 HepMatrix da(m_Tracks[id].getFitParameter(KFitConst::kBeforeFit) - m_Tracks[id].getFitParameter(KFitConst::kAfterFit));
214 int err_inverse = 0;
215 const double chisq = (da.T() * (m_Tracks[id].getFitError(KFitConst::kBeforeFit).inverse(err_inverse)) * da)[0][0];
216
217 if (err_inverse) {
218 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kCannotGetMatrixInverse);
219 return -1;
220 }
221
222 return chisq;
223
224 } else {
225
226 HepMatrix da(m_Tracks[id].getMomPos(KFitConst::kBeforeFit) - m_Tracks[id].getMomPos(KFitConst::kAfterFit));
227 int err_inverse = 0;
228 const double chisq = (da.T() * (m_Tracks[id].getError(KFitConst::kBeforeFit).inverse(err_inverse)) * da)[0][0];
229
230 if (err_inverse) {
231 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kCannotGetMatrixInverse);
232 return -1;
233 }
234
235 return chisq;
236
237 }
238}

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

108{
109 return m_TrackCount;
110}

◆ getTrackError()

const HepSymMatrix getTrackError ( const int id) const
inherited

Get an error matrix of the track.

Parameters
idtrack id
Returns
error matrix of the track

Definition at line 168 of file KFitBase.cc.

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

◆ getTrackMomentum()

const HepLorentzVector getTrackMomentum ( const int id) const
inherited

Get a Lorentz vector of the track.

Parameters
idtrack id
Returns
Lorentz vector of the track

Definition at line 154 of file KFitBase.cc.

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

◆ getTrackPosition()

const HepPoint3D getTrackPosition ( const int id) const
inherited

Get a position of the track.

Parameters
idtrack id
Returns
position of the track

Definition at line 161 of file KFitBase.cc.

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

◆ getTrackVertexError()

const HepMatrix getTrackVertexError ( const int id,
const int flag = KFitConst::kAfterFit ) const

Get a vertex error matrix of the track.

Parameters
idtrack id
flagKFitConst::kBeforeFit or KFitConst::kAfterFit
Returns
vertex error matrix

Definition at line 189 of file MassFitKFit.cc.

190{
191 if (flag == KFitConst::kAfterFit && !isFitted()) return HepMatrix(3, KFitConst::kNumber7, 0);
192 if (!isTrackIDInRange(id)) return HepMatrix(3, KFitConst::kNumber7, 0);
193
194 if (flag == KFitConst::kBeforeFit)
195 return m_BeforeTrackVertexError[id];
196 else if (flag == KFitConst::kAfterFit && m_FlagFitIncludingVertex)
197 return m_AfterTrackVertexError[id];
198 else {
199 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kOutOfRange);
200 return HepMatrix(3, KFitConst::kNumber7, 0);
201 }
202}
Belle2::analysis::MassFitKFit::m_BeforeTrackVertexError
std::vector< CLHEP::HepMatrix > m_BeforeTrackVertexError
array of vertex error matrices before the fit.
Definition MassFitKFit.h:143
Belle2::analysis::MassFitKFit::m_AfterTrackVertexError
std::vector< CLHEP::HepMatrix > m_AfterTrackVertexError
array of vertex error matrices after the fit.
Definition MassFitKFit.h:150

◆ getVertex()

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

Get a vertex position.

Parameters
flagKFitConst::kBeforeFit or KFitConst::kAfterFit
Returns
vertex position

Definition at line 126 of file MassFitKFit.cc.

127{
128 if (flag == KFitConst::kAfterFit && !isFitted()) return HepPoint3D();
129
130 switch (flag) {
131 case KFitConst::kBeforeFit:
132 return m_BeforeVertex;
133
134 case KFitConst::kAfterFit:
135 return m_AfterVertex;
136
137 default:
138 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kOutOfRange);
139 return HepPoint3D();
140 }
141}
Belle2::analysis::MassFitKFit::m_AfterVertex
HepPoint3D m_AfterVertex
Vertex position after the fit.
Definition MassFitKFit.h:146
Belle2::analysis::MassFitKFit::m_BeforeVertex
HepPoint3D m_BeforeVertex
Vertex position before the fit.
Definition MassFitKFit.h:139

◆ getVertexError()

const HepSymMatrix getVertexError ( const int flag = KFitConst::kAfterFit) const

Get a vertex error matrix.

Parameters
flagKFitConst::kBeforeFit or KFitConst::kAfterFit
Returns
vertex error matrix

Definition at line 145 of file MassFitKFit.cc.

146{
147 if (flag == KFitConst::kAfterFit && !isFitted()) return HepSymMatrix(3, 0);
148
149 if (flag == KFitConst::kBeforeFit)
150 return m_BeforeVertexError;
151 else if (flag == KFitConst::kAfterFit && m_FlagFitIncludingVertex)
152 return m_AfterVertexError;
153 else {
154 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kOutOfRange);
155 return HepSymMatrix(3, 0);
156 }
157}
Belle2::analysis::MassFitKFit::m_BeforeVertexError
CLHEP::HepSymMatrix m_BeforeVertexError
Vertex error matrix before the fit.
Definition MassFitKFit.h:141

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

731{
732 if (m_FlagFitted) return true;
733
734 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kNotFittedYet);
735
736 return false;
737}
Belle2::analysis::KFitError::kNotFittedYet
@ kNotFittedYet
Not fitted yet.
Definition KFitError.h:38

◆ isNonZeroEnergy()

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

Check if the energy is non-zero.

Parameters
pLorentz vector
Returns
true for non-zero energy, false for otherwise

Definition at line 752 of file KFitBase.cc.

753{
754 if (p.t() != 0) return true;
755
756 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kDivisionByZero);
757
758 return false;
759}
Belle2::analysis::KFitError::kDivisionByZero
@ kDivisionByZero
Division by zero (bad track property or internal error)
Definition KFitError.h:55

◆ isTrackIDInRange()

bool isTrackIDInRange ( const int id) const
protectedinherited

Check if the id is in the range.

Parameters
idtrack id
Returns
true if the id is in the range, false otherwise

Definition at line 741 of file KFitBase.cc.

742{
743 if (0 <= id && id < m_TrackCount) return true;
744
745 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kOutOfRange);
746
747 return false;
748}

◆ makeCoreMatrix()

enum KFitError::ECode makeCoreMatrix ( void )
overrideprivatevirtual

Build matrices using the kinematical constraint.

Returns
error code (zero if success)

Implements KFitBase.

Definition at line 526 of file MassFitKFit.cc.

526 {
527 if (!m_FlagFitIncludingVertex)
528 {
529
530 HepMatrix al_1_prime(m_al_1);
531 HepMatrix Sum_al_1(4, 1, 0);
532 std::vector<double> energy(m_TrackCount);
533 double a;
534
535 for (int i = 0; i < m_TrackCount; i++) {
536 a = m_property[i][2];
537 if (!m_FlagAtDecayPoint) a = 0.;
538 al_1_prime[i * KFitConst::kNumber7 + 0][0] -= a * (m_BeforeVertex.y() - al_1_prime[i * KFitConst::kNumber7 + 5][0]);
539 al_1_prime[i * KFitConst::kNumber7 + 1][0] += a * (m_BeforeVertex.x() - al_1_prime[i * KFitConst::kNumber7 + 4][0]);
540 energy[i] = sqrt(al_1_prime[i * KFitConst::kNumber7 + 0][0] * al_1_prime[i * KFitConst::kNumber7 + 0][0] +
541 al_1_prime[i * KFitConst::kNumber7 + 1][0] * al_1_prime[i * KFitConst::kNumber7 + 1][0] +
542 al_1_prime[i * KFitConst::kNumber7 + 2][0] * al_1_prime[i * KFitConst::kNumber7 + 2][0] +
543 m_property[i][1] * m_property[i][1]);
544 if (m_IsFixMass[i])
545 Sum_al_1[3][0] += energy[i];
546 else
547 Sum_al_1[3][0] += al_1_prime[i * KFitConst::kNumber7 + 3][0];
548 }
549
550 for (int i = 0; i < m_TrackCount; i++) {
551 for (int j = 0; j < 3; j++) Sum_al_1[j][0] += al_1_prime[i * KFitConst::kNumber7 + j][0];
552 }
553
554 m_d[0][0] =
555 + Sum_al_1[3][0] * Sum_al_1[3][0] - Sum_al_1[0][0] * Sum_al_1[0][0]
556 - Sum_al_1[1][0] * Sum_al_1[1][0] - Sum_al_1[2][0] * Sum_al_1[2][0]
557 - m_InvariantMass * m_InvariantMass;
558
559 for (int i = 0; i < m_TrackCount; i++) {
560 if (energy[i] == 0) {
561 m_ErrorCode = KFitError::kDivisionByZero;
562 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
563 break;
564 }
565
566 a = m_property[i][2];
567 if (!m_FlagAtDecayPoint) a = 0.;
568
569 if (m_IsFixMass[i]) {
570 double invE = 1. / energy[i];
571 m_D[0][i * KFitConst::kNumber7 + 0] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 0][0] * invE - Sum_al_1[0][0]);
572 m_D[0][i * KFitConst::kNumber7 + 1] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 1][0] * invE - Sum_al_1[1][0]);
573 m_D[0][i * KFitConst::kNumber7 + 2] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 2][0] * invE - Sum_al_1[2][0]);
574 m_D[0][i * KFitConst::kNumber7 + 3] = 0.;
575 m_D[0][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;
576 m_D[0][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;
577 m_D[0][i * KFitConst::kNumber7 + 6] = 0.;
578 } else {
579 m_D[0][i * KFitConst::kNumber7 + 0] = -2.*Sum_al_1[0][0];
580 m_D[0][i * KFitConst::kNumber7 + 1] = -2.*Sum_al_1[1][0];
581 m_D[0][i * KFitConst::kNumber7 + 2] = -2.*Sum_al_1[2][0];
582 m_D[0][i * KFitConst::kNumber7 + 3] = 2.*Sum_al_1[3][0];
583 m_D[0][i * KFitConst::kNumber7 + 4] = 2.*Sum_al_1[1][0] * a;
584 m_D[0][i * KFitConst::kNumber7 + 5] = -2.*Sum_al_1[0][0] * a;
585 m_D[0][i * KFitConst::kNumber7 + 6] = 0.;
586 }
587 }
588
589 } else
590 {
591
592 // m_FlagFitIncludingVertex == true
593 HepMatrix al_1_prime(m_al_1);
594 HepMatrix Sum_al_1(7, 1, 0);
595 std::vector<double> energy(m_TrackCount);
596 double a;
597
598 for (int i = 0; i < m_TrackCount; i++) {
599 a = m_property[i][2];
600 al_1_prime[i * KFitConst::kNumber7 + 0][0] -= a * (al_1_prime[KFitConst::kNumber7 * m_TrackCount + 1][0] - al_1_prime[i *
601 KFitConst::kNumber7 + 5][0]);
602 al_1_prime[i * KFitConst::kNumber7 + 1][0] += a * (al_1_prime[KFitConst::kNumber7 * m_TrackCount + 0][0] - al_1_prime[i *
603 KFitConst::kNumber7 + 4][0]);
604 energy[i] = sqrt(al_1_prime[i * KFitConst::kNumber7 + 0][0] * al_1_prime[i * KFitConst::kNumber7 + 0][0] +
605 al_1_prime[i * KFitConst::kNumber7 + 1][0] * al_1_prime[i * KFitConst::kNumber7 + 1][0] +
606 al_1_prime[i * KFitConst::kNumber7 + 2][0] * al_1_prime[i * KFitConst::kNumber7 + 2][0] +
607 m_property[i][1] * m_property[i][1]);
608 Sum_al_1[6][0] = + a;
609 }
610
611 for (int i = 0; i < m_TrackCount; i++) {
612 if (energy[i] == 0) {
613 m_ErrorCode = KFitError::kDivisionByZero;
614 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
615 break;
616 }
617
618 if (m_IsFixMass[i]) {
619 double invE = 1. / energy[i];
620 Sum_al_1[3][0] += energy[i];
621 Sum_al_1[4][0] += al_1_prime[i * KFitConst::kNumber7 + 1][0] * m_property[i][2] * invE;
622 Sum_al_1[5][0] += al_1_prime[i * KFitConst::kNumber7 + 0][0] * m_property[i][2] * invE;
623 } else {
624 Sum_al_1[3][0] += al_1_prime[i * KFitConst::kNumber7 + 3][0];
625 }
626
627 for (int j = 0; j < 3; j++) Sum_al_1[j][0] += al_1_prime[i * KFitConst::kNumber7 + j][0];
628 }
629
630 m_d[0][0] =
631 + Sum_al_1[3][0] * Sum_al_1[3][0] - Sum_al_1[0][0] * Sum_al_1[0][0]
632 - Sum_al_1[1][0] * Sum_al_1[1][0] - Sum_al_1[2][0] * Sum_al_1[2][0]
633 - m_InvariantMass * m_InvariantMass;
634
635 for (int i = 0; i < m_TrackCount; i++) {
636 if (energy[i] == 0) {
637 m_ErrorCode = KFitError::kDivisionByZero;
638 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
639 break;
640 }
641
642 a = m_property[i][2];
643
644 if (m_IsFixMass[i]) {
645 double invE = 1. / energy[i];
646 m_D[0][i * KFitConst::kNumber7 + 0] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 0][0] * invE - Sum_al_1[0][0]);
647 m_D[0][i * KFitConst::kNumber7 + 1] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 1][0] * invE - Sum_al_1[1][0]);
648 m_D[0][i * KFitConst::kNumber7 + 2] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 2][0] * invE - Sum_al_1[2][0]);
649 m_D[0][i * KFitConst::kNumber7 + 3] = 0.;
650 m_D[0][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;
651 m_D[0][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;
652 m_D[0][i * KFitConst::kNumber7 + 6] = 0.;
653 } else {
654 m_D[0][i * KFitConst::kNumber7 + 0] = -2.*Sum_al_1[0][0];
655 m_D[0][i * KFitConst::kNumber7 + 1] = -2.*Sum_al_1[1][0];
656 m_D[0][i * KFitConst::kNumber7 + 2] = -2.*Sum_al_1[2][0];
657 m_D[0][i * KFitConst::kNumber7 + 3] = 2.*Sum_al_1[3][0];
658 m_D[0][i * KFitConst::kNumber7 + 4] = 2.*Sum_al_1[1][0] * a;
659 m_D[0][i * KFitConst::kNumber7 + 5] = -2.*Sum_al_1[0][0] * a;
660 m_D[0][i * KFitConst::kNumber7 + 6] = 0.;
661 }
662 }
663
664 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]);
665 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]);
666 m_D[0][KFitConst::kNumber7 * m_TrackCount + 2] = 0.;
667 }
668
669 return m_ErrorCode = KFitError::kNoError;
670}
Belle2::analysis::KFitBase::m_property
CLHEP::HepMatrix m_property
Container of charges and masses.
Definition KFitBase.h:263

◆ 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
pLorentz vector
e(6x6) error matrix
Returns
(7x7) error matrix

Definition at line 223 of file KFitBase.cc.

224{
225 // self track
226 // Error(6x6,e) ==> Error(7x7,output(hsm)) using Momentum(p).
227
228 if (!isNonZeroEnergy(p)) return HepSymMatrix(KFitConst::kNumber7, 0);
229
230 HepSymMatrix hsm(KFitConst::kNumber7, 0);
231
232 for (int i = 0; i < 3; i++) for (int j = i; j < 3; j++) {
233 hsm[i][j] = e[i][j];
234 hsm[4 + i][4 + j] = e[3 + i][3 + j];
235 }
236 for (int i = 0; i < 3; i++) for (int j = 0; j < 3; j++) {
237 hsm[i][4 + j] = e[i][3 + j];
238 }
239
240 const double invE = 1 / p.t();
241 hsm[0][3] = (p.x() * hsm[0][0] + p.y() * hsm[0][1] + p.z() * hsm[0][2]) * invE;
242 hsm[1][3] = (p.x() * hsm[0][1] + p.y() * hsm[1][1] + p.z() * hsm[1][2]) * invE;
243 hsm[2][3] = (p.x() * hsm[0][2] + p.y() * hsm[1][2] + p.z() * hsm[2][2]) * invE;
244 hsm[3][3] = (p.x() * p.x() * hsm[0][0] + p.y() * p.y() * hsm[1][1] + p.z() * p.z() * hsm[2][2]
245 + 2.0 * p.x() * p.y() * hsm[0][1]
246 + 2.0 * p.x() * p.z() * hsm[0][2]
247 + 2.0 * p.y() * p.z() * hsm[1][2]) * invE * invE;
248 hsm[3][4] = (p.x() * hsm[0][4] + p.y() * hsm[1][4] + p.z() * hsm[2][4]) * invE;
249 hsm[3][5] = (p.x() * hsm[0][5] + p.y() * hsm[1][5] + p.z() * hsm[2][5]) * invE;
250 hsm[3][6] = (p.x() * hsm[0][6] + p.y() * hsm[1][6] + p.z() * hsm[2][6]) * invE;
251
252 return hsm;
253}
Belle2::analysis::KFitBase::isNonZeroEnergy
static bool isNonZeroEnergy(const CLHEP::HepLorentzVector &p)
Check if the energy is non-zero.
Definition KFitBase.cc:752

◆ 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
p1first Lorentz vector
p2second Lorentz vector
e(6x6) error matrix
Returns
(7x7) error matrix

Definition at line 257 of file KFitBase.cc.

258{
259 // track and track
260 // Error(6x6,e) ==> Error(7x7,output(hm)) using Momentum(p1&p2).
261
262 if (!isNonZeroEnergy(p1)) return HepSymMatrix(KFitConst::kNumber7, 0);
263 if (!isNonZeroEnergy(p2)) return HepSymMatrix(KFitConst::kNumber7, 0);
264
265 HepMatrix hm(KFitConst::kNumber7, KFitConst::kNumber7, 0);
266
267 for (int i = 0; i < 3; i++) for (int j = 0; j < 3; j++) {
268 hm[i][j] = e[i][j];
269 hm[4 + i][4 + j] = e[3 + i][3 + j];
270 hm[4 + i][j] = e[3 + i][j];
271 hm[i][4 + j] = e[i][3 + j];
272 }
273
274 const double invE1 = 1 / p1.t();
275 const double invE2 = 1 / p2.t();
276 hm[0][3] = (p2.x() * hm[0][0] + p2.y() * hm[0][1] + p2.z() * hm[0][2]) * invE2;
277 hm[1][3] = (p2.x() * hm[1][0] + p2.y() * hm[1][1] + p2.z() * hm[1][2]) * invE2;
278 hm[2][3] = (p2.x() * hm[2][0] + p2.y() * hm[2][1] + p2.z() * hm[2][2]) * invE2;
279 hm[4][3] = (p2.x() * hm[4][0] + p2.y() * hm[4][1] + p2.z() * hm[4][2]) * invE2;
280 hm[5][3] = (p2.x() * hm[5][0] + p2.y() * hm[5][1] + p2.z() * hm[5][2]) * invE2;
281 hm[6][3] = (p2.x() * hm[6][0] + p2.y() * hm[6][1] + p2.z() * hm[6][2]) * invE2;
282 hm[3][3] = (p1.x() * p2.x() * hm[0][0] + p1.y() * p2.y() * hm[1][1] + p1.z() * p2.z() * hm[2][2] +
283 p1.x() * p2.y() * hm[0][1] + p2.x() * p1.y() * hm[1][0] +
284 p1.x() * p2.z() * hm[0][2] + p2.x() * p1.z() * hm[2][0] +
285 p1.y() * p2.z() * hm[1][2] + p2.y() * p1.z() * hm[2][1]) * invE1 * invE2;
286 hm[3][0] = (p1.x() * hm[0][0] + p1.y() * hm[1][0] + p1.z() * hm[2][0]) * invE1;
287 hm[3][1] = (p1.x() * hm[0][1] + p1.y() * hm[1][1] + p1.z() * hm[2][1]) * invE1;
288 hm[3][2] = (p1.x() * hm[0][2] + p1.y() * hm[1][2] + p1.z() * hm[2][2]) * invE1;
289 hm[3][4] = (p1.x() * hm[0][4] + p1.y() * hm[1][4] + p1.z() * hm[2][4]) * invE1;
290 hm[3][5] = (p1.x() * hm[0][5] + p1.y() * hm[1][5] + p1.z() * hm[2][5]) * invE1;
291 hm[3][6] = (p1.x() * hm[0][6] + p1.y() * hm[1][6] + p1.z() * hm[2][6]) * invE1;
292
293 return hm;
294}

◆ 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
pLorentz vector
e(3x6) error matrix
Returns
(3x7) error matrix

Definition at line 298 of file KFitBase.cc.

299{
300 // vertex and track
301 // Error(3x6,e) ==> Error(3x7,output(hm)) using Momentum(p).
302
303 if (!isNonZeroEnergy(p)) return HepSymMatrix(KFitConst::kNumber7, 0);
304
305 HepMatrix hm(3, KFitConst::kNumber7, 0);
306
307 for (int i = 0; i < 3; i++) for (int j = 0; j < 3; j++) {
308 hm[i][j] = e[i][j];
309 hm[i][4 + j] = e[i][3 + j];
310 }
311
312 const double invE = 1 / p.t();
313 hm[0][3] = (p.x() * hm[0][0] + p.y() * hm[0][1] + p.z() * hm[0][2]) * invE;
314 hm[1][3] = (p.x() * hm[1][0] + p.y() * hm[1][1] + p.z() * hm[1][2]) * invE;
315 hm[2][3] = (p.x() * hm[2][0] + p.y() * hm[2][1] + p.z() * hm[2][2]) * invE;
316
317 return hm;
318}

◆ 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
pLorentz vector
e(7x7) error matrix
is_fix_masstrue to recalculate energy term from other parameters, false to do nothing
Returns
(7x7) error matrix

Definition at line 322 of file KFitBase.cc.

323{
324 // self track
325 // Error(7x7,e) ==> Error(7x7,output(hsm)) using Momentum(p).
326 // is_fix_mass = 1 : Energy term is recalculated from the other parameters.
327 // is_fix_mass = 0 : hsm = e.
328
329 if (!isNonZeroEnergy(p)) return HepSymMatrix(KFitConst::kNumber7, 0);
330
331 if (!is_fix_mass) {
332 HepSymMatrix hsm(KFitConst::kNumber7, 0);
333 for (int i = 0; i < 7; i++) for (int j = i; j < 7; j++) {
334 hsm[i][j] = e[i][j];
335 }
336 return hsm;
337 }
338
339 HepSymMatrix hsm(KFitConst::kNumber7, 0);
340
341 for (int i = 0; i < 7; i++) {
342 if (i != 3)
343 for (int j = i; j < 7; j++) hsm[i][j] = e[i][j];
344 }
345
346 double invE = 1 / p.t();
347 hsm[0][3] = (p.x() * hsm[0][0] + p.y() * hsm[0][1] + p.z() * hsm[0][2]) * invE;
348 hsm[1][3] = (p.x() * hsm[0][1] + p.y() * hsm[1][1] + p.z() * hsm[1][2]) * invE;
349 hsm[2][3] = (p.x() * hsm[0][2] + p.y() * hsm[1][2] + p.z() * hsm[2][2]) * invE;
350 hsm[3][3] = (p.x() * p.x() * hsm[0][0] + p.y() * p.y() * hsm[1][1] + p.z() * p.z() * hsm[2][2]
351 + 2.0 * p.x() * p.y() * hsm[0][1]
352 + 2.0 * p.x() * p.z() * hsm[0][2]
353 + 2.0 * p.y() * p.z() * hsm[1][2]) * invE * invE;
354 hsm[3][4] = (p.x() * hsm[0][4] + p.y() * hsm[1][4] + p.z() * hsm[2][4]) * invE;
355 hsm[3][5] = (p.x() * hsm[0][5] + p.y() * hsm[1][5] + p.z() * hsm[2][5]) * invE;
356 hsm[3][6] = (p.x() * hsm[0][6] + p.y() * hsm[1][6] + p.z() * hsm[2][6]) * invE;
357
358 return hsm;
359}

◆ 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
p1first Lorentz vector
p2second Lorentz vector
e(7x7) error matrix
is_fix_mass1true to recalculate energy term from other parameters, false to do nothing
is_fix_mass2true to recalculate energy term from other parameters, false to do nothing
Returns
(7x7) error matrix

Definition at line 363 of file KFitBase.cc.

366{
367 // track and track
368 // Error(7x7,e) ==> Error(7x7,output(hm)) using Momentum(p1&p2).
369 // is_fix_mass = 1 : Energy term is recalculated from the other parameters.
370 // is_fix_mass = 0 : not.
371
372 if (is_fix_mass1 && is_fix_mass2) {
373 if (!isNonZeroEnergy(p1)) return HepSymMatrix(KFitConst::kNumber7, 0);
374 if (!isNonZeroEnergy(p2)) return HepSymMatrix(KFitConst::kNumber7, 0);
375
376 HepMatrix hm(e);
377
378 const double invE1 = 1 / p1.t();
379 const double invE2 = 1 / p2.t();
380 hm[0][3] = (p2.x() * hm[0][0] + p2.y() * hm[0][1] + p2.z() * hm[0][2]) * invE2;
381 hm[1][3] = (p2.x() * hm[1][0] + p2.y() * hm[1][1] + p2.z() * hm[1][2]) * invE2;
382 hm[2][3] = (p2.x() * hm[2][0] + p2.y() * hm[2][1] + p2.z() * hm[2][2]) * invE2;
383 hm[4][3] = (p2.x() * hm[4][0] + p2.y() * hm[4][1] + p2.z() * hm[4][2]) * invE2;
384 hm[5][3] = (p2.x() * hm[5][0] + p2.y() * hm[5][1] + p2.z() * hm[5][2]) * invE2;
385 hm[6][3] = (p2.x() * hm[6][0] + p2.y() * hm[6][1] + p2.z() * hm[6][2]) * invE2;
386 hm[3][0] = (p1.x() * hm[0][0] + p1.y() * hm[1][0] + p1.z() * hm[2][0]) * invE1;
387 hm[3][1] = (p1.x() * hm[0][1] + p1.y() * hm[1][1] + p1.z() * hm[2][1]) * invE1;
388 hm[3][2] = (p1.x() * hm[0][2] + p1.y() * hm[1][2] + p1.z() * hm[2][2]) * invE1;
389 hm[3][3] = (p1.x() * p2.x() * hm[0][0] + p1.y() * p2.y() * hm[1][1] + p1.z() * p2.z() * hm[2][2] +
390 p1.x() * p2.y() * hm[0][1] + p2.x() * p1.y() * hm[1][0] +
391 p1.x() * p2.z() * hm[0][2] + p2.x() * p1.z() * hm[2][0] +
392 p1.y() * p2.z() * hm[1][2] + p2.y() * p1.z() * hm[2][1]) * invE1 * invE2;
393 hm[3][4] = (p1.x() * hm[0][4] + p1.y() * hm[1][4] + p1.z() * hm[2][4]) * invE1;
394 hm[3][5] = (p1.x() * hm[0][5] + p1.y() * hm[1][5] + p1.z() * hm[2][5]) * invE1;
395 hm[3][6] = (p1.x() * hm[0][6] + p1.y() * hm[1][6] + p1.z() * hm[2][6]) * invE1;
396
397 return hm;
398 }
399
400
401 if (is_fix_mass1 && !is_fix_mass2) {
402 if (!isNonZeroEnergy(p1)) return HepSymMatrix(KFitConst::kNumber7, 0);
403
404 HepMatrix hm(e);
405
406 const double invE1 = 1 / p1.t();
407 hm[3][0] = (p1.x() * hm[0][0] + p1.y() * hm[1][0] + p1.z() * hm[2][0]) * invE1;
408 hm[3][1] = (p1.x() * hm[0][1] + p1.y() * hm[1][1] + p1.z() * hm[2][1]) * invE1;
409 hm[3][2] = (p1.x() * hm[0][2] + p1.y() * hm[1][2] + p1.z() * hm[2][2]) * invE1;
410 hm[3][3] = (p1.x() * hm[0][3] + p1.y() * hm[1][3] + p1.z() * hm[2][3]) * invE1;
411 hm[3][4] = (p1.x() * hm[0][4] + p1.y() * hm[1][4] + p1.z() * hm[2][4]) * invE1;
412 hm[3][5] = (p1.x() * hm[0][5] + p1.y() * hm[1][5] + p1.z() * hm[2][5]) * invE1;
413 hm[3][6] = (p1.x() * hm[0][6] + p1.y() * hm[1][6] + p1.z() * hm[2][6]) * invE1;
414
415 return hm;
416 }
417
418
419 if (!is_fix_mass1 && is_fix_mass2) {
420 if (!isNonZeroEnergy(p2)) return HepSymMatrix(KFitConst::kNumber7, 0);
421
422 HepMatrix hm(e);
423
424 const double invE2 = 1 / p2.t();
425 hm[0][3] = (p2.x() * hm[0][0] + p2.y() * hm[0][1] + p2.z() * hm[0][2]) * invE2;
426 hm[1][3] = (p2.x() * hm[1][0] + p2.y() * hm[1][1] + p2.z() * hm[1][2]) * invE2;
427 hm[2][3] = (p2.x() * hm[2][0] + p2.y() * hm[2][1] + p2.z() * hm[2][2]) * invE2;
428 hm[3][3] = (p2.x() * hm[3][0] + p2.y() * hm[3][1] + p2.z() * hm[3][2]) * invE2;
429 hm[4][3] = (p2.x() * hm[4][0] + p2.y() * hm[4][1] + p2.z() * hm[4][2]) * invE2;
430 hm[5][3] = (p2.x() * hm[5][0] + p2.y() * hm[5][1] + p2.z() * hm[5][2]) * invE2;
431 hm[6][3] = (p2.x() * hm[6][0] + p2.y() * hm[6][1] + p2.z() * hm[6][2]) * invE2;
432
433 return hm;
434 }
435
436 return e;
437}

◆ 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
pLorentz vector
e(3x7) error matrix
Returns
(7x7) error matrix

Definition at line 441 of file KFitBase.cc.

442{
443 // vertex and track
444 // Error(3x7,e) ==> Error(3x7,output(hm)) using Momentum(p).
445 // Energy term is recalculated from the other parameters.
446
447 if (!isNonZeroEnergy(p)) return HepSymMatrix(KFitConst::kNumber7, 0);
448
449 HepMatrix hm(e);
450
451 const double invE = 1 / p.t();
452 hm[0][3] = (p.x() * hm[0][0] + p.y() * hm[0][1] + p.z() * hm[0][2]) * invE;
453 hm[1][3] = (p.x() * hm[1][0] + p.y() * hm[1][1] + p.z() * hm[1][2]) * invE;
454 hm[2][3] = (p.x() * hm[2][0] + p.y() * hm[2][1] + p.z() * hm[2][2]) * invE;
455
456 return hm;
457}

◆ 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 405 of file MassFitKFit.cc.

405 {
406 if (m_BeforeCorrelation.size() != static_cast<unsigned int>(m_TrackCount * (m_TrackCount - 1) / 2))
407 {
408 m_ErrorCode = KFitError::kBadCorrelationSize;
409 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
410 return m_ErrorCode;
411 }
412
413 int row = 0, col = 0;
414
415 for (const auto& hm : m_BeforeCorrelation)
416 {
417 // counter
418 row++;
419 if (row == m_TrackCount) {
420 col++;
421 row = col + 1;
422 }
423
424 int ii = 0, jj = 0;
425 for (int i = KFitConst::kNumber7 * row; i < KFitConst::kNumber7 * (row + 1); i++) {
426 for (int j = KFitConst::kNumber7 * col; j < KFitConst::kNumber7 * (col + 1); j++) {
427 m_V_al_0[i][j] = hm[ii][jj];
428 jj++;
429 }
430 jj = 0;
431 ii++;
432 }
433 }
434
435 if (m_FlagFitIncludingVertex)
436 {
437 // ...error of vertex
438 m_V_al_0.sub(KFitConst::kNumber7 * m_TrackCount + 1, m_BeforeVertexError);
439
440 // ...error matrix between vertex and tracks
441 if (m_FlagTrackVertexError) {
442 if (m_BeforeTrackVertexError.size() != (unsigned int)m_TrackCount) {
443 m_ErrorCode = KFitError::kBadCorrelationSize;
444 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
445 return m_ErrorCode;
446 }
447
448 int i = 0;
449 for (const auto& hm : m_BeforeTrackVertexError) {
450 for (int j = 0; j < 3; j++) for (int k = 0; k < KFitConst::kNumber7; k++) {
451 m_V_al_0[j + KFitConst::kNumber7 * m_TrackCount][k + i * KFitConst::kNumber7] = hm[j][k];
452 }
453 i++;
454 }
455 }
456 }
457
458 return m_ErrorCode = KFitError::kNoError;
459}
Belle2::analysis::KFitBase::m_BeforeCorrelation
std::vector< CLHEP::HepMatrix > m_BeforeCorrelation
Container of input correlation matrices.
Definition KFitBase.h:251
Belle2::analysis::KFitError::kBadCorrelationSize
@ kBadCorrelationSize
Wrong correlation matrix size (internal error)
Definition KFitError.h:50

◆ 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 275 of file MassFitKFit.cc.

275 {
276 if (m_TrackCount > KFitConst::kMaxTrackCount)
277 {
278 m_ErrorCode = KFitError::kBadTrackSize;
279 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
280 return m_ErrorCode;
281 }
282
283
284 if (m_IsFixMass.size() == 0)
285 {
286 // If no fix_mass flag at all,
287 // all tracks are considered to be fixed at mass.
288 for (int i = 0; i < m_TrackCount; i++) this->fixMass();
289 } else if (m_IsFixMass.size() != (unsigned int)m_TrackCount)
290 {
291 m_ErrorCode = KFitError::kBadTrackSize;
292 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
293 return m_ErrorCode;
294 }
295
296
297 if (!m_FlagFitIncludingVertex)
298 {
299 int index = 0;
300 m_al_0 = HepMatrix(KFitConst::kNumber7 * m_TrackCount, 1, 0);
301 m_property = HepMatrix(m_TrackCount, 3, 0);
302 m_V_al_0 = HepSymMatrix(KFitConst::kNumber7 * m_TrackCount, 0);
303
304 for (const auto& track : m_Tracks) {
305 // momentum x,y,z and position x,y,z
306 m_al_0[index * KFitConst::kNumber7 + 0][0] = track.getMomentum(KFitConst::kBeforeFit).x();
307 m_al_0[index * KFitConst::kNumber7 + 1][0] = track.getMomentum(KFitConst::kBeforeFit).y();
308 m_al_0[index * KFitConst::kNumber7 + 2][0] = track.getMomentum(KFitConst::kBeforeFit).z();
309 m_al_0[index * KFitConst::kNumber7 + 3][0] = track.getMomentum(KFitConst::kBeforeFit).t();
310 m_al_0[index * KFitConst::kNumber7 + 4][0] = track.getPosition(KFitConst::kBeforeFit).x();
311 m_al_0[index * KFitConst::kNumber7 + 5][0] = track.getPosition(KFitConst::kBeforeFit).y();
312 m_al_0[index * KFitConst::kNumber7 + 6][0] = track.getPosition(KFitConst::kBeforeFit).z();
313 // these error
314 m_V_al_0.sub(index * KFitConst::kNumber7 + 1, track.getError(KFitConst::kBeforeFit));
315 // charge, mass, a
316 m_property[index][0] = track.getCharge();
317 m_property[index][1] = track.getMass();
318 const double c = Const::speedOfLight * 1e-4;
319 m_property[index][2] = -c * m_MagneticField * track.getCharge();
320 index++;
321 }
322
323 // error between track and track
324 if (m_FlagCorrelation) {
325 this->prepareCorrelation();
326 if (m_ErrorCode != KFitError::kNoError) {
327 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
328 return m_ErrorCode;
329 }
330 }
331
332 // set member matrix
333 m_al_1 = m_al_0;
334
335 // define size of matrix
336 m_V_al_1 = HepMatrix(KFitConst::kNumber7 * m_TrackCount, KFitConst::kNumber7 * m_TrackCount, 0);
337 m_D = m_V_al_1.sub(1, 1, 1, KFitConst::kNumber7 * m_TrackCount);
338
339 } else
340 {
341 // m_FlagFitIncludingVertex == true
342 int index = 0;
343 m_al_0 = HepMatrix(KFitConst::kNumber7 * m_TrackCount + 3, 1, 0);
344 m_property = HepMatrix(m_TrackCount, 3, 0);
345 m_V_al_0 = HepSymMatrix(KFitConst::kNumber7 * m_TrackCount + 3, 0);
346
347 for (const auto& track : m_Tracks) {
348 // momentum x,y,z and position x,y,z
349 m_al_0[index * KFitConst::kNumber7 + 0][0] = track.getMomentum(KFitConst::kBeforeFit).x();
350 m_al_0[index * KFitConst::kNumber7 + 1][0] = track.getMomentum(KFitConst::kBeforeFit).y();
351 m_al_0[index * KFitConst::kNumber7 + 2][0] = track.getMomentum(KFitConst::kBeforeFit).z();
352 m_al_0[index * KFitConst::kNumber7 + 3][0] = track.getMomentum(KFitConst::kBeforeFit).t();
353 m_al_0[index * KFitConst::kNumber7 + 4][0] = track.getPosition(KFitConst::kBeforeFit).x();
354 m_al_0[index * KFitConst::kNumber7 + 5][0] = track.getPosition(KFitConst::kBeforeFit).y();
355 m_al_0[index * KFitConst::kNumber7 + 6][0] = track.getPosition(KFitConst::kBeforeFit).z();
356 // these error
357 m_V_al_0.sub(index * KFitConst::kNumber7 + 1, track.getError(KFitConst::kBeforeFit));
358 // charge, mass, a
359 m_property[index][0] = track.getCharge();
360 m_property[index][1] = track.getMass();
361 const double c = Const::speedOfLight * 1e-4;
362 m_property[index][2] = -c * m_MagneticField * track.getCharge();
363 index++;
364 }
365
366 // vertex
367 m_al_0[KFitConst::kNumber7 * m_TrackCount + 0][0] = m_BeforeVertex.x();
368 m_al_0[KFitConst::kNumber7 * m_TrackCount + 1][0] = m_BeforeVertex.y();
369 m_al_0[KFitConst::kNumber7 * m_TrackCount + 2][0] = m_BeforeVertex.z();
370 m_V_al_0.sub(KFitConst::kNumber7 * m_TrackCount + 1, m_BeforeVertexError);
371
372 // error between track and track
373 if (m_FlagCorrelation) {
374 this->prepareCorrelation();
375 if (m_ErrorCode != KFitError::kNoError) {
376 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
377 return m_ErrorCode;
378 }
379 }
380
381 // set member matrix
382 m_al_1 = m_al_0;
383
384 // define size of matrix
385 m_V_al_1 = HepMatrix(KFitConst::kNumber7 * m_TrackCount + 3, KFitConst::kNumber7 * m_TrackCount + 3, 0);
386 m_D = m_V_al_1.sub(1, 1, 1, KFitConst::kNumber7 * m_TrackCount + 3);
387 }
388
389 return m_ErrorCode = KFitError::kNoError;
390}
Belle2::Const::speedOfLight
static const double speedOfLight
[cm/ns]
Definition Const.h:696
Belle2::analysis::KFitBase::m_FlagCorrelation
bool m_FlagCorrelation
Flag whether a correlation among tracks exists.
Definition KFitBase.h:306
Belle2::analysis::MassFitKFit::prepareCorrelation
enum KFitError::ECode prepareCorrelation(void) override
Build a grand correlation matrix from input-track properties.
Definition MassFitKFit.cc:405
Belle2::analysis::MassFitKFit::fixMass
enum KFitError::ECode fixMass(void)
Tell the object to fix the last added track property at the invariant mass.
Definition MassFitKFit.cc:85
Belle2::analysis::KFitConst::kMaxTrackCount
static const int kMaxTrackCount
Maximum track size.
Definition KFitConst.h:38

◆ 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 394 of file MassFitKFit.cc.

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

◆ prepareOutputMatrix()

enum KFitError::ECode prepareOutputMatrix ( void )
overrideprivatevirtual

Build an output error matrix.

Returns
error code (zero if success)

Implements KFitBase.

Definition at line 463 of file MassFitKFit.cc.

463 {
464 Hep3Vector h3v;
465 int index = 0;
466 for (auto& pdata : m_Tracks)
467 {
468 // tracks
469 // momentum
470 h3v.setX(m_al_1[index * KFitConst::kNumber7 + 0][0]);
471 h3v.setY(m_al_1[index * KFitConst::kNumber7 + 1][0]);
472 h3v.setZ(m_al_1[index * KFitConst::kNumber7 + 2][0]);
473 if (m_IsFixMass[index])
474 pdata.setMomentum(HepLorentzVector(h3v, sqrt(h3v.mag2() + pdata.getMass()*pdata.getMass())), KFitConst::kAfterFit);
475 else
476 pdata.setMomentum(HepLorentzVector(h3v, m_al_1[index * KFitConst::kNumber7 + 3][0]), KFitConst::kAfterFit);
477 // position
478 pdata.setPosition(HepPoint3D(
479 m_al_1[index * KFitConst::kNumber7 + 4][0],
480 m_al_1[index * KFitConst::kNumber7 + 5][0],
481 m_al_1[index * KFitConst::kNumber7 + 6][0]), KFitConst::kAfterFit);
482 // error of the tracks
483 pdata.setError(this->makeError3(pdata.getMomentum(),
484 m_V_al_1.sub(
485 index * KFitConst::kNumber7 + 1,
486 (index + 1)*KFitConst::kNumber7,
487 index * KFitConst::kNumber7 + 1,
488 (index + 1)*KFitConst::kNumber7), m_IsFixMass[index]),
489 KFitConst::kAfterFit);
490 if (m_ErrorCode != KFitError::kNoError) break;
491 index++;
492 }
493
494 if (m_FlagFitIncludingVertex)
495 {
496 // vertex
497 m_AfterVertex.setX(m_al_1[KFitConst::kNumber7 * m_TrackCount + 0][0]);
498 m_AfterVertex.setY(m_al_1[KFitConst::kNumber7 * m_TrackCount + 1][0]);
499 m_AfterVertex.setZ(m_al_1[KFitConst::kNumber7 * m_TrackCount + 2][0]);
500 // error of the vertex
501 for (int i = 0; i < 3; i++) for (int j = i; j < 3; j++) {
502 m_AfterVertexError[i][j] = m_V_al_1[KFitConst::kNumber7 * m_TrackCount + i][KFitConst::kNumber7 * m_TrackCount + j];
503 }
504 // error between vertex and tracks
505 for (int i = 0; i < m_TrackCount; i++) {
506 HepMatrix hm(3, KFitConst::kNumber7, 0);
507 for (int j = 0; j < 3; j++) for (int k = 0; k < KFitConst::kNumber7; k++) {
508 hm[j][k] = m_V_al_1[KFitConst::kNumber7 * m_TrackCount + j][KFitConst::kNumber7 * i + k];
509 }
510 if (m_IsFixMass[i])
511 m_AfterTrackVertexError.push_back(this->makeError4(m_Tracks[i].getMomentum(), hm));
512 else
513 m_AfterTrackVertexError.push_back(hm);
514 }
515 } else
516 {
517 // not fit
518 m_AfterVertex = m_BeforeVertex;
519 }
520
521 return m_ErrorCode = KFitError::kNoError;
522}
Belle2::analysis::KFitBase::makeError4
static CLHEP::HepMatrix makeError4(const CLHEP::HepLorentzVector &p, const CLHEP::HepMatrix &e)
Rebuild an error matrix from a Lorentz vector and an error matrix.
Definition KFitBase.cc:441

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

70 {
71 if (e.num_row() != KFitConst::kNumber7)
72 {
73 m_ErrorCode = KFitError::kBadMatrixSize;
74 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
75 return m_ErrorCode;
76 }
77 m_BeforeCorrelation.push_back(e);
78 m_FlagCorrelation = true;
79
80 return m_ErrorCode = KFitError::kNoError;
81}

◆ setFlagAtDecayPoint()

enum KFitError::ECode setFlagAtDecayPoint ( const bool flag)

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

Parameters
flagtrue for constraint, false for not
Returns
error code (zero if success)

Definition at line 77 of file MassFitKFit.cc.

77 {
78 m_FlagAtDecayPoint = flag;
79
80 return m_ErrorCode = KFitError::kNoError;
81}

◆ setInvariantMass()

enum KFitError::ECode setInvariantMass ( const double m)

Set an invariant mass for the mass-constraint fit.

Parameters
minvariant mass
Returns
error code (zero if success)

Definition at line 69 of file MassFitKFit.cc.

69 {
70 m_InvariantMass = m;
71
72 return m_ErrorCode = KFitError::kNoError;
73}

◆ setMagneticField()

enum KFitError::ECode setMagneticField ( const double mf)
inherited

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

Parameters
mfmagnetic field to set
Returns
error code (zero if success)

Definition at line 93 of file KFitBase.cc.

93 {
94 m_MagneticField = mf;
95
96 return m_ErrorCode = KFitError::kNoError;
97}

◆ 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 101 of file MassFitKFit.cc.

101 {
102 if (e.num_row() != 3 || e.num_col() != KFitConst::kNumber7)
103 {
104 m_ErrorCode = KFitError::kBadMatrixSize;
105 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
106 return m_ErrorCode;
107 }
108
109 m_BeforeTrackVertexError.push_back(e);
110 m_FlagTrackVertexError = true;
111 m_FlagFitIncludingVertex = true;
112
113 return m_ErrorCode = KFitError::kNoError;
114}

◆ 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 118 of file MassFitKFit.cc.

118 {
119 HepMatrix zero(3, KFitConst::kNumber7, 0);
120
121 return this->setTrackVertexError(zero);
122}
Belle2::analysis::MassFitKFit::setTrackVertexError
enum KFitError::ECode setTrackVertexError(const CLHEP::HepMatrix &e)
Set a vertex error matrix of the child particle in the addTrack'ed order.
Definition MassFitKFit.cc:101

◆ setVertex()

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

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

Parameters
vvertex position
Returns
error code (zero if success)

Definition at line 45 of file MassFitKFit.cc.

45 {
46 m_BeforeVertex = v;
47
48 return m_ErrorCode = KFitError::kNoError;
49}

◆ setVertexError()

enum KFitError::ECode setVertexError ( const CLHEP::HepSymMatrix & e)

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

Parameters
evertex error matrix
Returns
error code (zero if success)

Definition at line 53 of file MassFitKFit.cc.

53 {
54 if (e.num_row() != 3)
55 {
56 m_ErrorCode = KFitError::kBadMatrixSize;
57 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
58 return m_ErrorCode;
59 }
60
61 m_BeforeVertexError = e;
62 m_FlagFitIncludingVertex = true;
63
64 return m_ErrorCode = KFitError::kNoError;
65}

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

85 {
86 HepMatrix zero(KFitConst::kNumber7, KFitConst::kNumber7, 0);
87
88 return this->setCorrelation(zero);
89}
Belle2::analysis::KFitBase::setCorrelation
virtual enum KFitError::ECode setCorrelation(const CLHEP::HepMatrix &c)
Set a correlation matrix.
Definition KFitBase.cc:70

◆ 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 93 of file MassFitKFit.cc.

93 {
94 m_IsFixMass.push_back(false);
95
96 return m_ErrorCode = KFitError::kNoError;
97}

◆ updateMother()

enum KFitError::ECode updateMother ( Particle * mother)

Update mother particle.

Parameters
[in]motherMother particle.

Definition at line 680 of file MassFitKFit.cc.

681{
682 MakeMotherKFit kmm;
683 kmm.setMagneticField(m_MagneticField);
684 unsigned n = getTrackCount();
685 for (unsigned i = 0; i < n; ++i) {
686 kmm.addTrack(getTrackMomentum(i), getTrackPosition(i), getTrackError(i),
687 getTrack(i).getCharge());
688 if (getFlagFitWithVertex())
689 kmm.setTrackVertexError(getTrackVertexError(i));
690 for (unsigned j = i + 1; j < n; ++j) {
691 kmm.setCorrelation(getCorrelation(i, j));
692 }
693 }
694 kmm.setVertex(getVertex());
695 if (getFlagFitWithVertex())
696 kmm.setVertexError(getVertexError());
697 m_ErrorCode = kmm.doMake();
698 if (m_ErrorCode != KFitError::kNoError)
699 return m_ErrorCode;
700 double chi2 = getCHIsq();
701 int ndf = getNDF();
702 double prob = TMath::Prob(chi2, ndf);
703 //
704 bool haschi2 = mother->hasExtraInfo("chiSquared");
705 if (haschi2) {
706 mother->setExtraInfo("chiSquared", chi2);
707 mother->setExtraInfo("ndf", ndf);
708 } else {
709 mother->addExtraInfo("chiSquared", chi2);
710 mother->addExtraInfo("ndf", ndf);
711 }
712
713 mother->updateMomentum(
714 CLHEPToROOT::getLorentzVector(kmm.getMotherMomentum()),
715 CLHEPToROOT::getXYZVector(kmm.getMotherPosition()),
716 CLHEPToROOT::getTMatrixFSym(kmm.getMotherError()),
717 prob);
718 m_ErrorCode = KFitError::kNoError;
719 return m_ErrorCode;
720}
Belle2::Particle::setExtraInfo
void setExtraInfo(const std::string &name, double value)
Sets the user-defined data of given name to the given value.
Definition Particle.cc:1402
Belle2::Particle::hasExtraInfo
bool hasExtraInfo(const std::string &name) const
Return whether the extra info with the given name is set.
Definition Particle.cc:1351
Belle2::Particle::addExtraInfo
void addExtraInfo(const std::string &name, double value)
Sets the user-defined data of given name to the given value.
Definition Particle.cc:1421
Belle2::Particle::updateMomentum
void updateMomentum(const ROOT::Math::PxPyPzEVector &p4, const ROOT::Math::XYZVector &vertex, const TMatrixFSym &errMatrix, double pValue)
Sets Lorentz vector, position, 7x7 error matrix and p-value.
Definition Particle.h:397
Belle2::analysis::KFitBase::getTrackError
const CLHEP::HepSymMatrix getTrackError(const int id) const
Get an error matrix of the track.
Definition KFitBase.cc:168
Belle2::analysis::KFitBase::getTrackPosition
const HepPoint3D getTrackPosition(const int id) const
Get a position of the track.
Definition KFitBase.cc:161
Belle2::analysis::KFitBase::getNDF
virtual int getNDF(void) const
Get an NDF of the fit.
Definition KFitBase.cc:114
Belle2::analysis::KFitBase::getTrack
const KFitTrack getTrack(const int id) const
Get a specified track object.
Definition KFitBase.cc:175
Belle2::analysis::KFitBase::getTrackCount
int getTrackCount(void) const
Get the number of added tracks.
Definition KFitBase.cc:107
Belle2::analysis::MakeMotherKFit::setVertex
enum KFitError::ECode setVertex(const HepPoint3D &v)
Set a vertex position of the mother particle.
Definition MakeMotherKFit.cc:71
Belle2::analysis::MakeMotherKFit::addTrack
enum KFitError::ECode addTrack(const KFitTrack &kp)
Add a track to the make-mother object.
Definition MakeMotherKFit.cc:39
Belle2::analysis::MakeMotherKFit::doMake
enum KFitError::ECode doMake(void)
Perform a reconstruction of mother particle.
Definition MakeMotherKFit.cc:192
Belle2::analysis::MakeMotherKFit::getMotherError
const CLHEP::HepSymMatrix getMotherError(void) const
Get an error matrix of the mother particle.
Definition MakeMotherKFit.cc:185
Belle2::analysis::MakeMotherKFit::setCorrelation
enum KFitError::ECode setCorrelation(const CLHEP::HepMatrix &e)
Set a correlation matrix.
Definition MakeMotherKFit.cc:119
Belle2::analysis::MakeMotherKFit::getMotherPosition
const HepPoint3D getMotherPosition(void) const
Get a position of the mother particle.
Definition MakeMotherKFit.cc:178
Belle2::analysis::MakeMotherKFit::setVertexError
enum KFitError::ECode setVertexError(const CLHEP::HepSymMatrix &e)
Set a vertex error matrix of the mother particle.
Definition MakeMotherKFit.cc:79
Belle2::analysis::MakeMotherKFit::setTrackVertexError
enum KFitError::ECode setTrackVertexError(const CLHEP::HepMatrix &e)
Set a vertex error matrix of the child particle in the addTrack'ed order.
Definition MakeMotherKFit.cc:95
Belle2::analysis::MakeMotherKFit::getMotherMomentum
const CLHEP::HepLorentzVector getMotherMomentum(void) const
Get a Lorentz vector of the mother particle.
Definition MakeMotherKFit.cc:171
Belle2::analysis::MakeMotherKFit::setMagneticField
enum KFitError::ECode setMagneticField(const double mf)
Change a magnetic field from the default value KFitConst::kDefaultMagneticField.
Definition MakeMotherKFit.cc:63
Belle2::analysis::MassFitKFit::getFlagFitWithVertex
bool getFlagFitWithVertex(void) const
Get a flag if the fit is allowed with moving the vertex position.
Definition MassFitKFit.cc:175
Belle2::analysis::MassFitKFit::getCHIsq
double getCHIsq(void) const override
Get a chi-square of the fit.
Definition MassFitKFit.cc:182
Belle2::analysis::MassFitKFit::getCorrelation
const CLHEP::HepMatrix getCorrelation(const int id1, const int id2, const int flag=KFitConst::kAfterFit) const override
Get a correlation matrix between two tracks.
Definition MassFitKFit.cc:242
Belle2::analysis::MassFitKFit::getVertexError
const CLHEP::HepSymMatrix getVertexError(const int flag=KFitConst::kAfterFit) const
Get a vertex error matrix.
Definition MassFitKFit.cc:145
Belle2::analysis::MassFitKFit::getVertex
const HepPoint3D getVertex(const int flag=KFitConst::kAfterFit) const
Get a vertex position.
Definition MassFitKFit.cc:126
Belle2::analysis::MassFitKFit::getTrackVertexError
const CLHEP::HepMatrix getTrackVertexError(const int id, const int flag=KFitConst::kAfterFit) const
Get a vertex error matrix of the track.
Definition MassFitKFit.cc:189

Member Data Documentation

◆ m_AfterTrackVertexError

std::vector<CLHEP::HepMatrix> m_AfterTrackVertexError
private

array of vertex error matrices after the fit.

Definition at line 150 of file MassFitKFit.h.

◆ m_AfterVertex

HepPoint3D m_AfterVertex
private

Vertex position after the fit.

Definition at line 146 of file MassFitKFit.h.

◆ m_AfterVertexError

CLHEP::HepSymMatrix m_AfterVertexError
private

Vertex error matrix after the fit.

Definition at line 148 of file MassFitKFit.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 143 of file MassFitKFit.h.

◆ m_BeforeVertex

HepPoint3D m_BeforeVertex
private

Vertex position before the fit.

Definition at line 139 of file MassFitKFit.h.

◆ m_BeforeVertexError

CLHEP::HepSymMatrix m_BeforeVertexError
private

Vertex error matrix before the fit.

Definition at line 141 of file MassFitKFit.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_FlagAtDecayPoint

bool m_FlagAtDecayPoint
private

Flag controlled by setFlagAtDecayPoint().

Definition at line 157 of file MassFitKFit.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 155 of file MassFitKFit.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 153 of file MassFitKFit.h.

◆ m_InvariantMass

double m_InvariantMass
private

Invariant mass.

Definition at line 160 of file MassFitKFit.h.

◆ m_IsFixMass

std::vector<int> m_IsFixMass
private

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

Definition at line 162 of file MassFitKFit.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: