Belle II Software development
MassPointingVertexFitKFit Class Reference

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

#include <MassPointingVertexFitKFit.h>

Inheritance diagram for MassPointingVertexFitKFit:
KFitBase

Public Member Functions

 MassPointingVertexFitKFit (void)
 Construct an object with no argument.
 
 ~MassPointingVertexFitKFit (void)
 Destruct the object.
 
enum KFitError::ECode setInitialVertex (const HepPoint3D &v)
 Set an initial vertex point for the mass-vertex-pointing constraint fit.
 
enum KFitError::ECode setInvariantMass (const double m)
 Set an invariant mass for the mass-vertex-pointing constraint fit.
 
enum KFitError::ECode setProductionVertex (const HepPoint3D &v)
 Set the production vertex of the particle.
 
enum KFitError::ECode fixMass (void)
 Tell the object to fix the last added track property at the invariant mass.
 
enum KFitError::ECode unfixMass (void)
 Tell the object to unfix the last added track property at the invariant mass.
 
enum KFitError::ECode setCorrelation (const CLHEP::HepMatrix &m) override
 Set a correlation matrix.
 
enum KFitError::ECode setZeroCorrelation (void) override
 Indicate no correlation between tracks.
 
const HepPoint3D getVertex (const int flag=KFitConst::kAfterFit) const
 Get a vertex position.
 
const CLHEP::HepSymMatrix getVertexError (void) const
 Get a fitted vertex error matrix.
 
double getInvariantMass (void) const
 Get an invariant mass.
 
double getCHIsq (void) const override
 Get a chi-square of the fit.
 
const CLHEP::HepMatrix getTrackVertexError (const int id) const
 Get a vertex error matrix of the track.
 
double getTrackCHIsq (const int id) const override
 Get a chi-square of the track.
 
const CLHEP::HepMatrix getCorrelation (const int id1, const int id2, const int flag=KFitConst::kAfterFit) const override
 Get a correlation matrix between two tracks.
 
enum KFitError::ECode doFit (void)
 Perform a mass-vertex-pointing constraint fit.
 
enum KFitError::ECode updateMother (Particle *mother)
 Update mother particle.
 
enum KFitError::ECode addTrack (const KFitTrack &kp)
 Add a track to the fitter object.
 
enum KFitError::ECode addTrack (const CLHEP::HepLorentzVector &p, const HepPoint3D &x, const CLHEP::HepSymMatrix &e, const double q)
 Add a track to the fitter object with specifying its momentum, position, error matrix, and charge.
 
enum KFitError::ECode addParticle (const Particle *particle)
 Add a particle to the fitter.
 
enum KFitError::ECode setMagneticField (const double mf)
 Change a magnetic field from the default value KFitConst::kDefaultMagneticField.
 
enum KFitError::ECode getErrorCode (void) const
 Get a code of the last error.
 
int getTrackCount (void) const
 Get the number of added tracks.
 
virtual int getNDF (void) const
 Get an NDF of the fit.
 
double getMagneticField (void) const
 Get a magnetic field.
 
const CLHEP::HepLorentzVector getTrackMomentum (const int id) const
 Get a Lorentz vector of the track.
 
const HepPoint3D getTrackPosition (const int id) const
 Get a position of the track.
 
const CLHEP::HepSymMatrix getTrackError (const int id) const
 Get an error matrix of the track.
 
const KFitTrack getTrack (const int id) const
 Get a specified track object.
 
bool isFitted (void) const
 Return false if fit is not performed yet or performed fit is failed; otherwise true.
 

Protected Member Functions

const CLHEP::HepSymMatrix makeError1 (const CLHEP::HepLorentzVector &p, const CLHEP::HepMatrix &e) const
 Rebuild an error matrix from a Lorentz vector and an error matrix.
 
const CLHEP::HepMatrix makeError1 (const CLHEP::HepLorentzVector &p1, const CLHEP::HepLorentzVector &p2, const CLHEP::HepMatrix &e) const
 Rebuild an error matrix from a pair of Lorentz vectors and an error matrix.
 
const CLHEP::HepMatrix makeError2 (const CLHEP::HepLorentzVector &p, const CLHEP::HepMatrix &e) const
 Rebuild an error matrix from a Lorentz vector and an error matrix.
 
const CLHEP::HepSymMatrix makeError3 (const CLHEP::HepLorentzVector &p, const CLHEP::HepMatrix &e, const bool is_fix_mass) const
 Rebuild an error matrix from a Lorentz vector and an error matrix.
 
const CLHEP::HepMatrix makeError3 (const CLHEP::HepLorentzVector &p1, const CLHEP::HepLorentzVector &p2, const CLHEP::HepMatrix &e, const bool is_fix_mass1, const bool is_fix_mass2) const
 Rebuild an error matrix from a pair of Lorentz vectors and an error matrix.
 
const CLHEP::HepMatrix makeError4 (const CLHEP::HepLorentzVector &p, const CLHEP::HepMatrix &e) const
 Rebuild an error matrix from a Lorentz vector and an error matrix.
 
enum KFitError::ECode doFit1 (void)
 Perform a fit (used in MassFitKFit::doFit()).
 
enum KFitError::ECode doFit2 (void)
 Perform a fit (used in VertexFitKFit::doFit() and MassVertexFitKFit::doFit()).
 
bool isTrackIDInRange (const int id) const
 Check if the id is in the range.
 
bool isNonZeroEnergy (const CLHEP::HepLorentzVector &p) const
 Check if the energy is non-zero.
 

Protected Attributes

enum KFitError::ECode m_ErrorCode
 Error code.
 
bool m_FlagFitted
 Flag to indicate if the fit is performed and succeeded.
 
std::vector< KFitTrackm_Tracks
 Container of input tracks.
 
std::vector< CLHEP::HepMatrix > m_BeforeCorrelation
 Container of input correlation matrices.
 
CLHEP::HepSymMatrix m_V_al_0
 See J.Tanaka Ph.D (2001) p137 for definition.
 
CLHEP::HepMatrix m_al_0
 See J.Tanaka Ph.D (2001) p136 for definition.
 
CLHEP::HepMatrix m_al_1
 See J.Tanaka Ph.D (2001) p136 for definition.
 
CLHEP::HepMatrix m_al_a
 See J.Tanaka Ph.D (2001) p137 for definition.
 
CLHEP::HepMatrix m_property
 Container of charges and masses.
 
CLHEP::HepMatrix m_D
 See J.Tanaka Ph.D (2001) p137 for definition.
 
CLHEP::HepMatrix m_d
 See J.Tanaka Ph.D (2001) p137 for definition.
 
CLHEP::HepMatrix m_V_D
 See J.Tanaka Ph.D (2001) p138 for definition.
 
CLHEP::HepMatrix m_V_al_1
 See J.Tanaka Ph.D (2001) p138 for definition.
 
CLHEP::HepMatrix m_lam
 See J.Tanaka Ph.D (2001) p137 for definition.
 
CLHEP::HepMatrix m_E
 See J.Tanaka Ph.D (2001) p137 for definition.
 
CLHEP::HepMatrix m_V_E
 See J.Tanaka Ph.D (2001) p138 for definition.
 
CLHEP::HepMatrix m_lam0
 See J.Tanaka Ph.D (2001) p138 for definition.
 
CLHEP::HepMatrix m_v
 See J.Tanaka Ph.D (2001) p137 for definition.
 
CLHEP::HepMatrix m_v_a
 See J.Tanaka Ph.D (2001) p137 for definition.
 
CLHEP::HepMatrix m_V_Dt
 See J.Tanaka Ph.D (2001) p138 for definition.
 
CLHEP::HepMatrix m_Cov_v_al_1
 See J.Tanaka Ph.D (2001) p137 for definition.
 
int m_NDF
 NDF of the fit.
 
double m_CHIsq
 chi-square of the fit.
 
int m_TrackCount
 Number of tracks.
 
int m_NecessaryTrackCount
 Number needed tracks to perform fit.
 
bool m_FlagCorrelation
 Flag whether a correlation among tracks exists.
 
bool m_FlagOverIteration
 Flag whether the iteration count exceeds the limit.
 
double m_MagneticField
 Magnetic field.
 

Private Member Functions

enum KFitError::ECode prepareInputMatrix (void) override
 Build grand matrices for minimum search from input-track properties.
 
enum KFitError::ECode prepareInputSubMatrix (void) override
 Build sub-matrices for minimum search from input-track properties.
 
enum KFitError::ECode prepareCorrelation (void) override
 Build a grand correlation matrix from input-track properties.
 
enum KFitError::ECode prepareOutputMatrix (void) override
 Build an output error matrix.
 
enum KFitError::ECode makeCoreMatrix (void) override
 Build matrices using the kinematical constraint.
 
enum KFitError::ECode calculateNDF (void) override
 Calculate an NDF of the fit.
 

Private Attributes

HepPoint3D m_BeforeVertex
 Vertex position before the fit.
 
HepPoint3D m_AfterVertex
 Vertex position after the fit.
 
CLHEP::HepSymMatrix m_AfterVertexError
 Vertex error matrix after the fit.
 
std::vector< CLHEP::HepMatrix > m_AfterTrackVertexError
 array of vertex error matrices after the fit.
 
HepPoint3D m_ProductionVertex
 Production vertex position.
 
double m_InvariantMass
 Invariant mass.
 
std::vector< int > m_IsFixMass
 Array of flags whether the track property is fixed at the mass.
 

Detailed Description

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

Definition at line 30 of file MassPointingVertexFitKFit.h.

Constructor & Destructor Documentation

◆ MassPointingVertexFitKFit()

Construct an object with no argument.

Definition at line 23 of file MassPointingVertexFitKFit.cc.

23 :
24 m_BeforeVertex(HepPoint3D(0., 0., 0.)),
25 m_AfterVertexError(HepSymMatrix(3, 0)),
26 m_ProductionVertex(HepPoint3D(0., 0., 0.))
27{
28 m_FlagFitted = false;
30 m_V_E = HepMatrix(3, 3, 0);
31 m_v = HepMatrix(3, 1, 0);
32 m_v_a = HepMatrix(3, 1, 0);
33 m_InvariantMass = -1.0;
34}
int m_NecessaryTrackCount
Number needed tracks to perform fit.
Definition KFitBase.h:303
CLHEP::HepMatrix m_v_a
See J.Tanaka Ph.D (2001) p137 for definition.
Definition KFitBase.h:287
CLHEP::HepMatrix m_V_E
See J.Tanaka Ph.D (2001) p138 for definition.
Definition KFitBase.h:281
bool m_FlagFitted
Flag to indicate if the fit is performed and succeeded.
Definition KFitBase.h:245
CLHEP::HepMatrix m_v
See J.Tanaka Ph.D (2001) p137 for definition.
Definition KFitBase.h:285
HepPoint3D m_ProductionVertex
Production vertex position.
CLHEP::HepSymMatrix m_AfterVertexError
Vertex error matrix after the fit.
HepPoint3D m_BeforeVertex
Vertex position before the fit.

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}
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 {
51 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
52 return m_ErrorCode;
53 }
54
55 return this->addTrack(KFitTrack(p, x, e, q));
56}
enum KFitError::ECode m_ErrorCode
Error code.
Definition KFitBase.h:243
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
@ kBadMatrixSize
Wrong correlation matrix size.
Definition KFitError.h:48
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
43}
std::vector< KFitTrack > m_Tracks
Container of input tracks.
Definition KFitBase.h:249
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 599 of file MassPointingVertexFitKFit.cc.

599 {
600 m_NDF = 2 * m_TrackCount - 3 + 3;
601
603}
int m_NDF
NDF of the fit.
Definition KFitBase.h:295

◆ doFit()

enum KFitError::ECode doFit ( void )

Perform a mass-vertex-pointing constraint fit.

Returns
error code (zero if success)

Definition at line 204 of file MassPointingVertexFitKFit.cc.

204 {
205 return KFitBase::doFit2();
206}
enum KFitError::ECode doFit2(void)
Perform a fit (used in VertexFitKFit::doFit() and MassVertexFitKFit::doFit()).
Definition KFitBase.cc:578

◆ doFit1()

enum KFitError::ECode doFit1 ( void )
protectedinherited

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

Returns
error code (zero if success)

Definition at line 502 of file KFitBase.cc.

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

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

◆ 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 65 of file MassPointingVertexFitKFit.cc.

65 {
66 m_IsFixMass.push_back(true);
67
69}
std::vector< int > m_IsFixMass
Array of flags whether the track property is fixed at the mass.

◆ 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 126 of file MassPointingVertexFitKFit.cc.

127{
128 return m_CHIsq;
129}

◆ 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 177 of file MassPointingVertexFitKFit.cc.

178{
179 if (flag == KFitConst::kAfterFit && !isFitted()) return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
180 if (!isTrackIDInRange(id1)) return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
181 if (!isTrackIDInRange(id2)) return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
182
183 switch (flag) {
185 return KFitBase::getCorrelation(id1, id2, flag);
186
188 return makeError3(
189 this->getTrackMomentum(id1),
190 this->getTrackMomentum(id2),
191 m_V_al_1.sub(KFitConst::kNumber7 * id1 + 1, KFitConst::kNumber7 * (id1 + 1), KFitConst::kNumber7 * id2 + 1,
192 KFitConst::kNumber7 * (id2 + 1)),
193 m_IsFixMass[id1],
194 m_IsFixMass[id2]);
195
196 default:
197 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kOutOfRange);
198 return HepMatrix(KFitConst::kNumber7, KFitConst::kNumber7, 0);
199 }
200}
const CLHEP::HepSymMatrix makeError3(const CLHEP::HepLorentzVector &p, const CLHEP::HepMatrix &e, const bool is_fix_mass) const
Rebuild an error matrix from a Lorentz vector and an error matrix.
Definition KFitBase.cc:320
const CLHEP::HepLorentzVector getTrackMomentum(const int id) const
Get a Lorentz vector of the track.
Definition KFitBase.cc:154
bool isFitted(void) const
Return false if fit is not performed yet or performed fit is failed; otherwise true.
Definition KFitBase.cc:728
bool isTrackIDInRange(const int id) const
Check if the id is in the range.
Definition KFitBase.cc:739
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
@ kOutOfRange
Specified track-id out of range.
Definition KFitError.h:41
static const int kAfterFit
Input parameter to specify after-fit when setting/getting a track attribute.
Definition KFitConst.h:35
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}

◆ getInvariantMass()

double getInvariantMass ( void ) const

Get an invariant mass.

Returns
invariant mass

Definition at line 119 of file MassPointingVertexFitKFit.cc.

120{
121 return m_InvariantMass;
122}

◆ 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}
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 142 of file MassPointingVertexFitKFit.cc.

143{
144 if (!isFitted()) return -1;
145 if (!isTrackIDInRange(id)) return -1;
146
147 if (m_IsFixMass[id]) {
148
149 HepMatrix da(m_Tracks[id].getFitParameter(KFitConst::kBeforeFit) - m_Tracks[id].getFitParameter(KFitConst::kAfterFit));
150 int err_inverse = 0;
151 const double chisq = (da.T() * (m_Tracks[id].getFitError(KFitConst::kBeforeFit).inverse(err_inverse)) * da)[0][0];
152
153 if (err_inverse) {
154 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kCannotGetMatrixInverse);
155 return -1;
156 }
157
158 return chisq;
159
160 } else {
161
162 HepMatrix da(m_Tracks[id].getMomPos(KFitConst::kBeforeFit) - m_Tracks[id].getMomPos(KFitConst::kAfterFit));
163 int err_inverse = 0;
164 const double chisq = (da.T() * (m_Tracks[id].getError(KFitConst::kBeforeFit).inverse(err_inverse)) * da)[0][0];
165
166 if (err_inverse) {
167 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kCannotGetMatrixInverse);
168 return -1;
169 }
170
171 return chisq;
172 }
173}

◆ 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

Get a vertex error matrix of the track.

Parameters
idtrack id
Returns
vertex error matrix

Definition at line 133 of file MassPointingVertexFitKFit.cc.

134{
135 if (!isTrackIDInRange(id)) return HepMatrix(3, KFitConst::kNumber7, 0);
136
137 return m_AfterTrackVertexError[id];
138}
std::vector< CLHEP::HepMatrix > m_AfterTrackVertexError
array of vertex error matrices after the fit.

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

94{
95 if (flag == KFitConst::kAfterFit && !isFitted()) return HepPoint3D();
96
97 switch (flag) {
99 return m_BeforeVertex;
100
102 return m_AfterVertex;
103
104 default:
105 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kOutOfRange);
106 return HepPoint3D();
107 }
108}
HepPoint3D m_AfterVertex
Vertex position after the fit.

◆ getVertexError()

const HepSymMatrix getVertexError ( void ) const

Get a fitted vertex error matrix.

Returns
vertex error matrix

Definition at line 112 of file MassPointingVertexFitKFit.cc.

113{
114 return m_AfterVertexError;
115}

◆ isFitted()

bool isFitted ( void ) const
inherited

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

Returns
see description

Definition at line 728 of file KFitBase.cc.

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

◆ isNonZeroEnergy()

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

Check if the energy is non-zero.

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

Definition at line 750 of file KFitBase.cc.

751{
752 if (p.t() != 0) return true;
753
754 KFitError::displayError(__FILE__, __LINE__, __func__, KFitError::kDivisionByZero);
755
756 return false;
757}
@ 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 739 of file KFitBase.cc.

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

◆ makeCoreMatrix()

enum KFitError::ECode makeCoreMatrix ( void )
overrideprivatevirtual

Build matrices using the kinematical constraint.

Returns
error code (zero if success)

Implements KFitBase.

Definition at line 392 of file MassPointingVertexFitKFit.cc.

392 {
393 // Mass Constraint
394 HepMatrix al_1_prime(m_al_1);
395 HepMatrix Sum_al_1(4, 1, 0);
396 std::vector<double> energy(m_TrackCount);
397 double a;
398
399 for (int i = 0; i < m_TrackCount; i++)
400 {
401 a = m_property[i][2];
402 al_1_prime[i * KFitConst::kNumber7 + 0][0] -= a * (m_v_a[1][0] - al_1_prime[i * KFitConst::kNumber7 + 5][0]);
403 al_1_prime[i * KFitConst::kNumber7 + 1][0] += a * (m_v_a[0][0] - al_1_prime[i * KFitConst::kNumber7 + 4][0]);
404 energy[i] = sqrt(al_1_prime[i * KFitConst::kNumber7 + 0][0] * al_1_prime[i * KFitConst::kNumber7 + 0][0] +
405 al_1_prime[i * KFitConst::kNumber7 + 1][0] * al_1_prime[i * KFitConst::kNumber7 + 1][0] +
406 al_1_prime[i * KFitConst::kNumber7 + 2][0] * al_1_prime[i * KFitConst::kNumber7 + 2][0] +
407 m_property[i][1] * m_property[i][1]);
408 if (m_IsFixMass[i])
409 Sum_al_1[3][0] += energy[i];
410 else
411 Sum_al_1[3][0] += al_1_prime[i * KFitConst::kNumber7 + 3][0];
412 }
413
414 for (int i = 0; i < m_TrackCount; i++)
415 {
416 for (int j = 0; j < 3; j++) Sum_al_1[j][0] += al_1_prime[i * KFitConst::kNumber7 + j][0];
417 }
418
419 double vtx = sqrt((m_v_a[0][0] - m_ProductionVertex.x()) * (m_v_a[0][0] - m_ProductionVertex.x())
420 + (m_v_a[1][0] - m_ProductionVertex.y()) * (m_v_a[1][0] - m_ProductionVertex.y())
421 + (m_v_a[2][0] - m_ProductionVertex.z()) * (m_v_a[2][0] - m_ProductionVertex.z()));
422 double vtxpt2 = (m_ProductionVertex.x() - m_v_a[0][0]) * (m_ProductionVertex.x() - m_v_a[0][0]) + (m_ProductionVertex.y() - m_v_a[1][0]) * (m_ProductionVertex.y() - m_v_a[1][0]);
423 double mom = sqrt(Sum_al_1[0][0] * Sum_al_1[0][0] + Sum_al_1[1][0] * Sum_al_1[1][0] + Sum_al_1[2][0] * Sum_al_1[2][0]);
424 double Pt2 = Sum_al_1[0][0] * Sum_al_1[0][0] + Sum_al_1[1][0] * Sum_al_1[1][0];
425
426 m_d[2 * m_TrackCount][0] =
427 + Sum_al_1[3][0] * Sum_al_1[3][0] - Sum_al_1[0][0] * Sum_al_1[0][0]
428 - Sum_al_1[1][0] * Sum_al_1[1][0] - Sum_al_1[2][0] * Sum_al_1[2][0]
430
431 double phiPointingConstraint = atan2(m_v_a[1][0] - m_ProductionVertex.y(), m_v_a[0][0] - m_ProductionVertex.x()) - atan2(Sum_al_1[1][0], Sum_al_1[0][0]);
432 phiPointingConstraint = std::fmod(phiPointingConstraint + TMath::Pi(), TMath::TwoPi());
433 if (phiPointingConstraint < 0) phiPointingConstraint += TMath::TwoPi();
434 phiPointingConstraint -= TMath::Pi();
435
436 m_d[2 * m_TrackCount + 1][0] = phiPointingConstraint;
437 m_d[2 * m_TrackCount + 2][0] = acos((m_v_a[2][0] - m_ProductionVertex.z()) / vtx) - acos(Sum_al_1[2][0] / mom);
438
439 double Sum_a = 0., Sum_tmpx = 0., Sum_tmpy = 0.;
440 for (int i = 0; i < m_TrackCount; i++)
441 {
442 if (energy[i] == 0) {
444 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
445 return m_ErrorCode;
446 }
447
448 a = m_property[i][2];
449
450 if (m_IsFixMass[i]) {
451 double invE = 1. / energy[i];
452 m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 0] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 0][0] * invE -
453 Sum_al_1[0][0]);
454 m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 1] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 1][0] * invE -
455 Sum_al_1[1][0]);
456 m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 2] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 2][0] * invE -
457 Sum_al_1[2][0]);
458 m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 3] = 0.;
459 m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 4] = -2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 1][0] * invE -
460 Sum_al_1[1][0]) * a;
461 m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 5] = 2.*(Sum_al_1[3][0] * al_1_prime[i * KFitConst::kNumber7 + 0][0] * invE -
462 Sum_al_1[0][0]) * a;
463 m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 6] = 0.;
464 Sum_tmpx += al_1_prime[i * KFitConst::kNumber7 + 0][0] * invE * a;
465 Sum_tmpy += al_1_prime[i * KFitConst::kNumber7 + 1][0] * invE * a;
466 } else {
467 m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 0] = -2.*Sum_al_1[0][0];
468 m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 1] = -2.*Sum_al_1[1][0];
469 m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 2] = -2.*Sum_al_1[2][0];
470 m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 3] = 2.*Sum_al_1[3][0];
471 m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 4] = 2.*Sum_al_1[1][0] * a;
472 m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 5] = -2.*Sum_al_1[0][0] * a;
473 m_D[2 * m_TrackCount][i * KFitConst::kNumber7 + 6] = 0.;
474 }
475 m_D[2 * m_TrackCount + 1][i * KFitConst::kNumber7 + 0] = Sum_al_1[1][0] / Pt2;
476 m_D[2 * m_TrackCount + 1][i * KFitConst::kNumber7 + 1] = - Sum_al_1[0][0] / Pt2;
477 m_D[2 * m_TrackCount + 1][i * KFitConst::kNumber7 + 2] = 0.;
478 m_D[2 * m_TrackCount + 1][i * KFitConst::kNumber7 + 3] = 0.;
479 m_D[2 * m_TrackCount + 1][i * KFitConst::kNumber7 + 4] = a * Sum_al_1[0][0] / Pt2;
480 m_D[2 * m_TrackCount + 1][i * KFitConst::kNumber7 + 5] = a * Sum_al_1[1][0] / Pt2;
481 m_D[2 * m_TrackCount + 1][i * KFitConst::kNumber7 + 6] = 0.;
482 m_D[2 * m_TrackCount + 2][i * KFitConst::kNumber7 + 0] = - Sum_al_1[0][0] * Sum_al_1[2][0] / (sqrt(Pt2) * mom * mom);
483 m_D[2 * m_TrackCount + 2][i * KFitConst::kNumber7 + 1] = - Sum_al_1[1][0] * Sum_al_1[2][0] / (sqrt(Pt2) * mom * mom);
484 m_D[2 * m_TrackCount + 2][i * KFitConst::kNumber7 + 2] = sqrt(Pt2) / (mom * mom);
485 m_D[2 * m_TrackCount + 2][i * KFitConst::kNumber7 + 3] = 0.;
486 m_D[2 * m_TrackCount + 2][i * KFitConst::kNumber7 + 4] = a * Sum_al_1[1][0] * Sum_al_1[2][0] / (sqrt(Pt2) * mom * mom);
487 m_D[2 * m_TrackCount + 2][i * KFitConst::kNumber7 + 5] = - a * Sum_al_1[0][0] * Sum_al_1[2][0] / (sqrt(Pt2) * mom * mom);
488 m_D[2 * m_TrackCount + 2][i * KFitConst::kNumber7 + 6] = 0.;
489 Sum_a += a;
490 }
491
492 // m_E
493 m_E[2 * m_TrackCount][0] = -2.*Sum_al_1[1][0] * Sum_a + 2.*Sum_al_1[3][0] * Sum_tmpy;
494 m_E[2 * m_TrackCount][1] = 2.*Sum_al_1[0][0] * Sum_a - 2.*Sum_al_1[3][0] * Sum_tmpx;
495 m_E[2 * m_TrackCount][2] = 0.;
496 m_E[2 * m_TrackCount + 1][0] = (m_ProductionVertex.y() - m_v_a[1][0]) / vtxpt2 - Sum_a * Sum_al_1[0][0] / Pt2;
497 m_E[2 * m_TrackCount + 1][1] = (m_v_a[0][0] - m_ProductionVertex.x()) / vtxpt2 - Sum_a * Sum_al_1[1][0] / Pt2;
498 m_E[2 * m_TrackCount + 1][2] = 0.;
499 m_E[2 * m_TrackCount + 2][0] = (m_v_a[0][0] - m_ProductionVertex.x()) * (m_v_a[2][0] - m_ProductionVertex.z()) / (vtx* vtx * sqrt(vtxpt2)) - Sum_a * Sum_al_1[1][0] * Sum_al_1[2][0] / (sqrt(Pt2) * mom * mom);
500 m_E[2 * m_TrackCount + 2][1] = (m_v_a[1][0] - m_ProductionVertex.y()) * (m_v_a[2][0] - m_ProductionVertex.z()) / (vtx* vtx * sqrt(vtxpt2)) + Sum_a * Sum_al_1[0][0] * Sum_al_1[2][0] / (sqrt(Pt2) * mom * mom);
501 m_E[2 * m_TrackCount + 2][2] = - sqrt(vtxpt2) / (vtx * vtx);
502
503 for (int i = 0; i < m_TrackCount; i++)
504 {
505 double S, U;
506 double sininv;
507
508 double px = m_al_1[i * KFitConst::kNumber7 + 0][0];
509 double py = m_al_1[i * KFitConst::kNumber7 + 1][0];
510 double pz = m_al_1[i * KFitConst::kNumber7 + 2][0];
511 double x = m_al_1[i * KFitConst::kNumber7 + 4][0];
512 double y = m_al_1[i * KFitConst::kNumber7 + 5][0];
513 double z = m_al_1[i * KFitConst::kNumber7 + 6][0];
514 a = m_property[i][2];
515
516 double pt = sqrt(px * px + py * py);
517
518 if (pt == 0) {
520 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
521 return m_ErrorCode;
522 }
523
524 double invPt = 1. / pt;
525 double invPt2 = invPt * invPt;
526 double dlx = m_v_a[0][0] - x;
527 double dly = m_v_a[1][0] - y;
528 double dlz = m_v_a[2][0] - z;
529 double a1 = -dlx * py + dly * px;
530 double a2 = dlx * px + dly * py;
531 double r2d2 = dlx * dlx + dly * dly;
532 double Rx = dlx - 2.*px * a2 * invPt2;
533 double Ry = dly - 2.*py * a2 * invPt2;
534
535 if (a != 0.) { // charged
536
537 double B = a * a2 * invPt2;
538 if (fabs(B) > 1.) {
540 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
541 return m_ErrorCode;
542 }
543 // sin^(-1)(B)
544 sininv = asin(B);
545 double tmp0 = 1.0 - B * B;
546 if (tmp0 == 0) {
548 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
549 return m_ErrorCode;
550 }
551 // 1/sqrt(1-B^2)
552 double sqrtag = 1.0 / sqrt(tmp0);
553 S = sqrtag * invPt2;
554 U = dlz - pz * sininv / a;
555
556 } else { // neutral
557
558 sininv = 0.0;
559 S = invPt2;
560 U = dlz - pz * a2 * invPt2;
561
562 }
563
564 // d
565 m_d[i * 2 + 0][0] = a1 - 0.5 * a * r2d2;
566 m_d[i * 2 + 1][0] = U * pt;
567
568 // D
569 m_D[i * 2 + 0][i * KFitConst::kNumber7 + 0] = dly;
570 m_D[i * 2 + 0][i * KFitConst::kNumber7 + 1] = -dlx;
571 m_D[i * 2 + 0][i * KFitConst::kNumber7 + 2] = 0.0;
572 m_D[i * 2 + 0][i * KFitConst::kNumber7 + 4] = py + a * dlx;
573 m_D[i * 2 + 0][i * KFitConst::kNumber7 + 5] = -px + a * dly;
574 m_D[i * 2 + 0][i * KFitConst::kNumber7 + 6] = 0.0;
575 m_D[i * 2 + 1][i * KFitConst::kNumber7 + 0] = -pz * pt * S * Rx + U * px * invPt;
576 m_D[i * 2 + 1][i * KFitConst::kNumber7 + 1] = -pz * pt * S * Ry + U * py * invPt;
577 if (a != 0.)
578 m_D[i * 2 + 1][i * KFitConst::kNumber7 + 2] = -sininv * pt / a;
579 else
580 m_D[i * 2 + 1][i * KFitConst::kNumber7 + 2] = -a2 * invPt;
581 m_D[i * 2 + 1][i * KFitConst::kNumber7 + 4] = px * pz * pt * S;
582 m_D[i * 2 + 1][i * KFitConst::kNumber7 + 5] = py * pz * pt * S;
583 m_D[i * 2 + 1][i * KFitConst::kNumber7 + 6] = -pt;
584
585 // E
586 m_E[i * 2 + 0][0] = -py - a * dlx;
587 m_E[i * 2 + 0][1] = px - a * dly;
588 m_E[i * 2 + 0][2] = 0.0;
589 m_E[i * 2 + 1][0] = -px * pz * pt * S;
590 m_E[i * 2 + 1][1] = -py * pz * pt * S;
591 m_E[i * 2 + 1][2] = pt;
592 }
593
595}
CLHEP::HepMatrix m_property
Container of charges and masses.
Definition KFitBase.h:263
double sqrt(double a)
sqrt for double
Definition beamHelpers.h:28

◆ makeError1() [1/2]

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

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

Parameters
pLorentz vector
e(6x6) error matrix
Returns
(7x7) error matrix

Definition at line 221 of file KFitBase.cc.

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

◆ makeError1() [2/2]

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

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

Parameters
p1first Lorentz vector
p2second Lorentz vector
e(6x6) error matrix
Returns
(7x7) error matrix

Definition at line 255 of file KFitBase.cc.

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

◆ makeError2()

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

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

Parameters
pLorentz vector
e(3x6) error matrix
Returns
(3x7) error matrix

Definition at line 296 of file KFitBase.cc.

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

◆ makeError3() [1/2]

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

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

Parameters
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 320 of file KFitBase.cc.

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

◆ makeError3() [2/2]

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

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

Parameters
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 361 of file KFitBase.cc.

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

◆ makeError4()

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

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

Parameters
pLorentz vector
e(3x7) error matrix
Returns
(7x7) error matrix

Definition at line 439 of file KFitBase.cc.

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

◆ 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 299 of file MassPointingVertexFitKFit.cc.

299 {
300 if (m_BeforeCorrelation.size() != static_cast<unsigned int>(m_TrackCount * (m_TrackCount - 1) / 2))
301 {
303 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
304 return m_ErrorCode;
305 }
306
307 int row = 0, col = 0;
308
309 for (auto& hm : m_BeforeCorrelation)
310 {
311 // counter
312 row++;
313 if (row == m_TrackCount) {
314 col++;
315 row = col + 1;
316 }
317
318 int ii = 0, jj = 0;
319 for (int i = KFitConst::kNumber7 * row; i < KFitConst::kNumber7 * (row + 1); i++) {
320 for (int j = KFitConst::kNumber7 * col; j < KFitConst::kNumber7 * (col + 1); j++) {
321 m_V_al_0[i][j] = hm[ii][jj];
322 jj++;
323 }
324 jj = 0;
325 ii++;
326 }
327 }
328
330}
std::vector< CLHEP::HepMatrix > m_BeforeCorrelation
Container of input correlation matrices.
Definition KFitBase.h:251
@ 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 210 of file MassPointingVertexFitKFit.cc.

210 {
212 {
214 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
215 return m_ErrorCode;
216 }
217
218
219 if (m_IsFixMass.size() == 0)
220 {
221 // If no fix_mass flag at all,
222 // all tracks are considered to be fixed at mass.
223 for (int i = 0; i < m_TrackCount; i++) this->fixMass();
224 } else if (m_IsFixMass.size() != (unsigned int)m_TrackCount)
225 {
227 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
228 return m_ErrorCode;
229 }
230
231
232 int index = 0;
233 m_al_0 = HepMatrix(KFitConst::kNumber7 * m_TrackCount, 1, 0);
234 m_property = HepMatrix(m_TrackCount, 3, 0);
235 m_V_al_0 = HepSymMatrix(KFitConst::kNumber7 * m_TrackCount, 0);
236
237 for (auto& track : m_Tracks)
238 {
239 // momentum x,y,z,E and position x,y,z
240 m_al_0[index * KFitConst::kNumber7 + 0][0] = track.getMomentum(KFitConst::kBeforeFit).x();
241 m_al_0[index * KFitConst::kNumber7 + 1][0] = track.getMomentum(KFitConst::kBeforeFit).y();
242 m_al_0[index * KFitConst::kNumber7 + 2][0] = track.getMomentum(KFitConst::kBeforeFit).z();
243 m_al_0[index * KFitConst::kNumber7 + 3][0] = track.getMomentum(KFitConst::kBeforeFit).t();
244 m_al_0[index * KFitConst::kNumber7 + 4][0] = track.getPosition(KFitConst::kBeforeFit).x();
245 m_al_0[index * KFitConst::kNumber7 + 5][0] = track.getPosition(KFitConst::kBeforeFit).y();
246 m_al_0[index * KFitConst::kNumber7 + 6][0] = track.getPosition(KFitConst::kBeforeFit).z();
247 // these error
248 m_V_al_0.sub(index * KFitConst::kNumber7 + 1, track.getError(KFitConst::kBeforeFit));
249 // charge, mass, a
250 m_property[index][0] = track.getCharge();
251 m_property[index][1] = track.getMass();
252 const double c = Const::speedOfLight * 1e-4;
253 m_property[index][2] = -c * m_MagneticField * track.getCharge();
254 index++;
255 }
256
257 // error between track and track
259 {
260 this->prepareCorrelation();
262 KFitError::displayError(__FILE__, __LINE__, __func__, m_ErrorCode);
263 return m_ErrorCode;
264 }
265 }
266
267 // vertex
268 m_v_a[0][0] = m_BeforeVertex.x();
269 m_v_a[1][0] = m_BeforeVertex.y();
270 m_v_a[2][0] = m_BeforeVertex.z();
271
272 // set member matrix
273 m_al_1 = m_al_0;
274
277 m_E = m_V_al_1.sub(1, m_TrackCount * 2 + 3, 1, 3);
278 m_d = m_V_al_1.sub(1, m_TrackCount * 2 + 3, 1, 1);
279 m_V_D = m_V_al_1.sub(1, m_TrackCount * 2 + 3, 1, m_TrackCount * 2 + 3);
280 m_lam = m_V_al_1.sub(1, m_TrackCount * 2 + 3, 1, 1);
281 m_lam0 = m_V_al_1.sub(1, m_TrackCount * 2 + 3, 1, 1);
282 m_V_Dt = m_V_al_1.sub(1, m_TrackCount * 2 + 3, 1, m_TrackCount * 2 + 3);
284
286}
static const double speedOfLight
[cm/ns]
Definition Const.h:695
bool m_FlagCorrelation
Flag whether a correlation among tracks exists.
Definition KFitBase.h:306
enum KFitError::ECode prepareCorrelation(void) override
Build a grand correlation matrix from input-track properties.
enum KFitError::ECode fixMass(void)
Tell the object to fix the last added track property at the invariant mass.
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 290 of file MassPointingVertexFitKFit.cc.

290 {
291 // vertex
292 for (int i = 0; i < 3; i++) m_v[i][0] = m_v_a[i][0];
293
295}

◆ prepareOutputMatrix()

enum KFitError::ECode prepareOutputMatrix ( void )
overrideprivatevirtual

Build an output error matrix.

Returns
error code (zero if success)

Implements KFitBase.

Definition at line 334 of file MassPointingVertexFitKFit.cc.

334 {
335 Hep3Vector h3v;
336 int index = 0;
337 for (auto& pdata : m_Tracks)
338 {
339 // tracks
340 // momentum
341 h3v.setX(m_al_1[index * KFitConst::kNumber7 + 0][0]);
342 h3v.setY(m_al_1[index * KFitConst::kNumber7 + 1][0]);
343 h3v.setZ(m_al_1[index * KFitConst::kNumber7 + 2][0]);
344 if (m_IsFixMass[index])
345 pdata.setMomentum(HepLorentzVector(h3v, sqrt(h3v.mag2() + pdata.getMass()*pdata.getMass())), KFitConst::kAfterFit);
346 else
347 pdata.setMomentum(HepLorentzVector(h3v, m_al_1[index * KFitConst::kNumber7 + 3][0]), KFitConst::kAfterFit);
348 // position
349 pdata.setPosition(HepPoint3D(
350 m_al_1[index * KFitConst::kNumber7 + 4][0],
351 m_al_1[index * KFitConst::kNumber7 + 5][0],
353 // error of the tracks
354 pdata.setError(this->makeError3(pdata.getMomentum(),
355 m_V_al_1.sub(
356 index * KFitConst::kNumber7 + 1,
357 (index + 1)*KFitConst::kNumber7,
358 index * KFitConst::kNumber7 + 1,
359 (index + 1)*KFitConst::kNumber7), m_IsFixMass[index]),
361 if (m_ErrorCode != KFitError::kNoError) break;
362 index++;
363 }
364
365 // vertex
366 m_AfterVertex.setX(m_v_a[0][0]);
367 m_AfterVertex.setY(m_v_a[1][0]);
368 m_AfterVertex.setZ(m_v_a[2][0]);
369 // error of the vertex
370 for (int i = 0; i < 3; i++) for (int j = i; j < 3; j++)
371 {
372 m_AfterVertexError[i][j] = m_V_E[i][j];
373 }
374 // error between vertex and tracks
375 for (int i = 0; i < m_TrackCount; i++)
376 {
377 HepMatrix hm(3, KFitConst::kNumber7, 0);
378 for (int j = 0; j < 3; j++) for (int k = 0; k < KFitConst::kNumber7; k++) {
379 hm[j][k] = m_Cov_v_al_1[j][KFitConst::kNumber7 * i + k];
380 }
381 if (m_IsFixMass[i])
382 m_AfterTrackVertexError.push_back(this->makeError4(m_Tracks[i].getMomentum(), hm));
383 else
384 m_AfterTrackVertexError.push_back(hm);
385 }
386
388}
const CLHEP::HepMatrix makeError4(const CLHEP::HepLorentzVector &p, const CLHEP::HepMatrix &e) const
Rebuild an error matrix from a Lorentz vector and an error matrix.
Definition KFitBase.cc:439

◆ setCorrelation()

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

Set a correlation matrix.

Not intended for end user's use.

Parameters
c(7x7) correlation matrix
Returns
error code (zero if success)

Reimplemented from KFitBase.

Definition at line 81 of file MassPointingVertexFitKFit.cc.

81 {
83}
virtual enum KFitError::ECode setCorrelation(const CLHEP::HepMatrix &c)
Set a correlation matrix.
Definition KFitBase.cc:70

◆ setInitialVertex()

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

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

Parameters
vinitial vertex point
Returns
error code (zero if success)

Definition at line 41 of file MassPointingVertexFitKFit.cc.

41 {
43
45}

◆ setInvariantMass()

enum KFitError::ECode setInvariantMass ( const double m)

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

Parameters
minvariant mass
Returns
error code (zero if success)

Definition at line 49 of file MassPointingVertexFitKFit.cc.

49 {
51
53}

◆ 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
97}

◆ setProductionVertex()

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

Set the production vertex of the particle.

Parameters
vproduction vertex
Returns
error code (zero if success)

Definition at line 57 of file MassPointingVertexFitKFit.cc.

57 {
59
61}

◆ setZeroCorrelation()

enum KFitError::ECode setZeroCorrelation ( void )
overridevirtual

Indicate no correlation between tracks.

Not intended for end user's use.

Returns
error code (zero if success)

Reimplemented from KFitBase.

Definition at line 87 of file MassPointingVertexFitKFit.cc.

87 {
89}
virtual enum KFitError::ECode setZeroCorrelation(void)
Indicate no correlation between tracks.
Definition KFitBase.cc:85

◆ 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 73 of file MassPointingVertexFitKFit.cc.

73 {
74 m_IsFixMass.push_back(false);
75
77}

◆ updateMother()

enum KFitError::ECode updateMother ( Particle * mother)

Update mother particle.

Parameters
motherparticle

Definition at line 605 of file MassPointingVertexFitKFit.cc.

606{
607 MakeMotherKFit kmm;
609 unsigned n = getTrackCount();
610 for (unsigned i = 0; i < n; ++i) {
612 getTrack(i).getCharge());
614 for (unsigned j = i + 1; j < n; ++j) {
616 }
617 }
618 kmm.setVertex(getVertex());
620 m_ErrorCode = kmm.doMake();
622 B2DEBUG(2, "Error in doMake");
623 return m_ErrorCode;
624 }
625 double chi2 = getCHIsq();
626 int ndf = getNDF();
627 double prob = TMath::Prob(chi2, ndf);
628 mother->addExtraInfo("ndf", ndf);
629 mother->addExtraInfo("chiSquared", chi2);
630 mother->updateMomentum(
631 CLHEPToROOT::getLorentzVector(kmm.getMotherMomentum()),
632 CLHEPToROOT::getXYZVector(kmm.getMotherPosition()),
633 CLHEPToROOT::getTMatrixFSym(kmm.getMotherError()),
634 prob);
636 return m_ErrorCode;
637}
void addExtraInfo(const std::string &name, double value)
Sets the user-defined data of given name to the given value.
Definition Particle.cc:1421
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
const CLHEP::HepSymMatrix getTrackError(const int id) const
Get an error matrix of the track.
Definition KFitBase.cc:168
const HepPoint3D getTrackPosition(const int id) const
Get a position of the track.
Definition KFitBase.cc:161
virtual int getNDF(void) const
Get an NDF of the fit.
Definition KFitBase.cc:114
const KFitTrack getTrack(const int id) const
Get a specified track object.
Definition KFitBase.cc:175
int getTrackCount(void) const
Get the number of added tracks.
Definition KFitBase.cc:107
enum KFitError::ECode setVertex(const HepPoint3D &v)
Set a vertex position of the mother particle.
enum KFitError::ECode addTrack(const KFitTrack &kp)
Add a track to the make-mother object.
enum KFitError::ECode doMake(void)
Perform a reconstruction of mother particle.
const CLHEP::HepSymMatrix getMotherError(void) const
Get an error matrix of the mother particle.
enum KFitError::ECode setCorrelation(const CLHEP::HepMatrix &e)
Set a correlation matrix.
const HepPoint3D getMotherPosition(void) const
Get a position of the mother particle.
enum KFitError::ECode setVertexError(const CLHEP::HepSymMatrix &e)
Set a vertex error matrix of the mother particle.
enum KFitError::ECode setTrackVertexError(const CLHEP::HepMatrix &e)
Set a vertex error matrix of the child particle in the addTrack'ed order.
const CLHEP::HepLorentzVector getMotherMomentum(void) const
Get a Lorentz vector of the mother particle.
enum KFitError::ECode setMagneticField(const double mf)
Change a magnetic field from the default value KFitConst::kDefaultMagneticField.
double getCHIsq(void) const override
Get a chi-square of the fit.
const CLHEP::HepSymMatrix getVertexError(void) const
Get a fitted vertex error matrix.
const CLHEP::HepMatrix getCorrelation(const int id1, const int id2, const int flag=KFitConst::kAfterFit) const override
Get a correlation matrix between two tracks.
const HepPoint3D getVertex(const int flag=KFitConst::kAfterFit) const
Get a vertex position.
const CLHEP::HepMatrix getTrackVertexError(const int id) const
Get a vertex error matrix of the track.

Member Data Documentation

◆ m_AfterTrackVertexError

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

array of vertex error matrices after the fit.

Definition at line 120 of file MassPointingVertexFitKFit.h.

◆ m_AfterVertex

HepPoint3D m_AfterVertex
private

Vertex position after the fit.

Definition at line 116 of file MassPointingVertexFitKFit.h.

◆ m_AfterVertexError

CLHEP::HepSymMatrix m_AfterVertexError
private

Vertex error matrix after the fit.

Definition at line 118 of file MassPointingVertexFitKFit.h.

◆ m_al_0

CLHEP::HepMatrix m_al_0
protectedinherited

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

Definition at line 257 of file KFitBase.h.

◆ m_al_1

CLHEP::HepMatrix m_al_1
protectedinherited

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

Definition at line 259 of file KFitBase.h.

◆ m_al_a

CLHEP::HepMatrix m_al_a
protectedinherited

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

Definition at line 261 of file KFitBase.h.

◆ m_BeforeCorrelation

std::vector<CLHEP::HepMatrix> m_BeforeCorrelation
protectedinherited

Container of input correlation matrices.

Definition at line 251 of file KFitBase.h.

◆ m_BeforeVertex

HepPoint3D m_BeforeVertex
private

Vertex position before the fit.

Definition at line 114 of file MassPointingVertexFitKFit.h.

◆ m_CHIsq

double m_CHIsq
protectedinherited

chi-square of the fit.

Definition at line 297 of file KFitBase.h.

◆ m_Cov_v_al_1

CLHEP::HepMatrix m_Cov_v_al_1
protectedinherited

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

Definition at line 291 of file KFitBase.h.

◆ m_D

CLHEP::HepMatrix m_D
protectedinherited

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

Definition at line 266 of file KFitBase.h.

◆ m_d

CLHEP::HepMatrix m_d
protectedinherited

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

Definition at line 268 of file KFitBase.h.

◆ m_E

CLHEP::HepMatrix m_E
protectedinherited

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

Definition at line 279 of file KFitBase.h.

◆ m_ErrorCode

enum KFitError::ECode m_ErrorCode
protectedinherited

Error code.

Definition at line 243 of file KFitBase.h.

◆ m_FlagCorrelation

bool m_FlagCorrelation
protectedinherited

Flag whether a correlation among tracks exists.

Definition at line 306 of file KFitBase.h.

◆ m_FlagFitted

bool m_FlagFitted
protectedinherited

Flag to indicate if the fit is performed and succeeded.

Definition at line 245 of file KFitBase.h.

◆ m_FlagOverIteration

bool m_FlagOverIteration
protectedinherited

Flag whether the iteration count exceeds the limit.

Definition at line 308 of file KFitBase.h.

◆ m_InvariantMass

double m_InvariantMass
private

Invariant mass.

Definition at line 124 of file MassPointingVertexFitKFit.h.

◆ m_IsFixMass

std::vector<int> m_IsFixMass
private

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

Definition at line 126 of file MassPointingVertexFitKFit.h.

◆ m_lam

CLHEP::HepMatrix m_lam
protectedinherited

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

Definition at line 276 of file KFitBase.h.

◆ m_lam0

CLHEP::HepMatrix m_lam0
protectedinherited

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

Definition at line 283 of file KFitBase.h.

◆ m_MagneticField

double m_MagneticField
protectedinherited

Magnetic field.

Definition at line 311 of file KFitBase.h.

◆ m_NDF

int m_NDF
protectedinherited

NDF of the fit.

Definition at line 295 of file KFitBase.h.

◆ m_NecessaryTrackCount

int m_NecessaryTrackCount
protectedinherited

Number needed tracks to perform fit.

Definition at line 303 of file KFitBase.h.

◆ m_ProductionVertex

HepPoint3D m_ProductionVertex
private

Production vertex position.

Definition at line 122 of file MassPointingVertexFitKFit.h.

◆ m_property

CLHEP::HepMatrix m_property
protectedinherited

Container of charges and masses.

Definition at line 263 of file KFitBase.h.

◆ m_TrackCount

int m_TrackCount
protectedinherited

Number of tracks.

Definition at line 301 of file KFitBase.h.

◆ m_Tracks

std::vector<KFitTrack> m_Tracks
protectedinherited

Container of input tracks.

Definition at line 249 of file KFitBase.h.

◆ m_v

CLHEP::HepMatrix m_v
protectedinherited

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

Definition at line 285 of file KFitBase.h.

◆ m_v_a

CLHEP::HepMatrix m_v_a
protectedinherited

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

Definition at line 287 of file KFitBase.h.

◆ m_V_al_0

CLHEP::HepSymMatrix m_V_al_0
protectedinherited

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

Definition at line 255 of file KFitBase.h.

◆ m_V_al_1

CLHEP::HepMatrix m_V_al_1
protectedinherited

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

Definition at line 274 of file KFitBase.h.

◆ m_V_D

CLHEP::HepMatrix m_V_D
protectedinherited

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

Definition at line 271 of file KFitBase.h.

◆ m_V_Dt

CLHEP::HepMatrix m_V_Dt
protectedinherited

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

Definition at line 289 of file KFitBase.h.

◆ m_V_E

CLHEP::HepMatrix m_V_E
protectedinherited

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

Definition at line 281 of file KFitBase.h.


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