Belle II Software  release-05-02-19
MassFitKFit.h
1 /**************************************************************************
2  * Copyright(C) 1997 - J. Tanaka *
3  * *
4  * Author: J. Tanaka *
5  * Contributor: J. Tanaka and *
6  * conversion to Belle II structure by T. Higuchi *
7  * *
8  * This software is provided "as is" without any warranty. *
9  **************************************************************************/
10 
11 #pragma once
12 
13 #include <analysis/VertexFitting/KFit/KFitConst.h>
14 #include <analysis/VertexFitting/KFit/KFitError.h>
15 #include <analysis/VertexFitting/KFit/KFitBase.h>
16 
17 
18 #ifndef ENABLE_BACKWARDS_COMPATIBILITY
20 #endif
21 
22 
23 namespace Belle2 {
29  namespace analysis {
30 
34  class MassFitKFit : public KFitBase {
35  public:
37  MassFitKFit(void);
39  ~MassFitKFit(void);
40 
41 
42  public:
47  enum KFitError::ECode setVertex(const HepPoint3D& v);
52  enum KFitError::ECode setVertexError(const CLHEP::HepSymMatrix& e);
57  enum KFitError::ECode setInvariantMass(const double m);
62  enum KFitError::ECode setFlagAtDecayPoint(const bool flag);
67  enum KFitError::ECode fixMass(void);
72  enum KFitError::ECode unfixMass(void);
77  enum KFitError::ECode setTrackVertexError(const CLHEP::HepMatrix& e);
82  enum KFitError::ECode setCorrelation(const CLHEP::HepMatrix& m) override;
83  enum KFitError::ECode setZeroCorrelation(void) override;
84 
85 
90  const HepPoint3D getVertex(const int flag = KFitConst::kAfterFit) const;
95  const CLHEP::HepSymMatrix getVertexError(const int flag = KFitConst::kAfterFit) const;
99  double getInvariantMass(void) const;
103  bool getFlagAtDecayPoint(void) const;
107  bool getFlagFitWithVertex(void) const;
108  double getCHIsq(void) const override;
114  const CLHEP::HepMatrix getTrackVertexError(const int id, const int flag = KFitConst::kAfterFit) const;
115  double getTrackCHIsq(const int id) const override;
116  const CLHEP::HepMatrix getCorrelation(const int id1, const int id2,
117  const int flag = KFitConst::kAfterFit) const override;
118 
119 
120  public:
124  enum KFitError::ECode doFit(void);
125 
130  enum KFitError::ECode updateMother(Particle* mother);
131 
132  private:
133  enum KFitError::ECode prepareInputMatrix(void) override;
134  enum KFitError::ECode prepareInputSubMatrix(void) override;
135  enum KFitError::ECode prepareCorrelation(void) override;
136  enum KFitError::ECode prepareOutputMatrix(void) override;
137  enum KFitError::ECode makeCoreMatrix(void) override;
138  enum KFitError::ECode calculateNDF(void) override;
139 
140 
141  private:
145  CLHEP::HepSymMatrix m_BeforeVertexError;
147  std::vector<CLHEP::HepMatrix> m_BeforeTrackVertexError;
148 
152  CLHEP::HepSymMatrix m_AfterVertexError;
154  std::vector<CLHEP::HepMatrix> m_AfterTrackVertexError;
155 
162 
166  std::vector<int> m_IsFixMass;
167  };
168 
169  } // namespace analysis
170 
172 } // namespace Belle2
Belle2::analysis::MassFitKFit::setZeroCorrelation
enum KFitError::ECode setZeroCorrelation(void) override
Indicate no correlation between tracks.
Definition: MassFitKFit.cc:131
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:166
Belle2::analysis::MassFitKFit::getVertex
const HepPoint3D getVertex(const int flag=KFitConst::kAfterFit) const
Get a vertex position.
Definition: MassFitKFit.cc:137
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:100
Belle2::analysis::MassFitKFit::prepareOutputMatrix
enum KFitError::ECode prepareOutputMatrix(void) override
Build an output error matrix.
Definition: MassFitKFit.cc:477
Belle2::analysis::MassFitKFit::m_BeforeVertexError
CLHEP::HepSymMatrix m_BeforeVertexError
Vertex error matrix before the fit.
Definition: MassFitKFit.h:145
Belle2::analysis::MassFitKFit::m_FlagAtDecayPoint
bool m_FlagAtDecayPoint
Flag controlled by setFlagAtDecayPoint().
Definition: MassFitKFit.h:161
Belle2::analysis::KFitBase
KFitBase is a base class for kinematical fitters.
Definition: KFitBase.h:38
Belle2::analysis::MassFitKFit::calculateNDF
enum KFitError::ECode calculateNDF(void) override
Calculate an NDF of the fit.
Definition: MassFitKFit.cc:690
Belle2::analysis::MassFitKFit
MassFitKFit is a derived class from KFitBase to perform mass-constraint kinematical fit.
Definition: MassFitKFit.h:34
Belle2::analysis::MassFitKFit::getFlagAtDecayPoint
bool getFlagAtDecayPoint(void) const
Get a flag if to constraint at the decay point in the mass-constraint fit.
Definition: MassFitKFit.cc:179
Belle2::analysis::MassFitKFit::getCHIsq
double getCHIsq(void) const override
Get a chi-square of the fit.
Definition: MassFitKFit.cc:193
Belle2::analysis::MassFitKFit::setTrackZeroVertexError
enum KFitError::ECode setTrackZeroVertexError(void)
Indicate no vertex uncertainty in the child particle in the addTrack'ed order.
Definition: MassFitKFit.cc:117
Belle2::analysis::MassFitKFit::setCorrelation
enum KFitError::ECode setCorrelation(const CLHEP::HepMatrix &m) override
Set a correlation matrix.
Definition: MassFitKFit.cc:125
Belle2::analysis::MassFitKFit::m_BeforeTrackVertexError
std::vector< CLHEP::HepMatrix > m_BeforeTrackVertexError
array of vertex error matrices before the fit.
Definition: MassFitKFit.h:147
Belle2::analysis::MassFitKFit::setFlagAtDecayPoint
enum KFitError::ECode setFlagAtDecayPoint(const bool flag)
Set a flag if to constraint at the decay point in the mass-constraint fit.
Definition: MassFitKFit.cc:76
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:200
Belle2::analysis::MassFitKFit::m_BeforeVertex
HepPoint3D m_BeforeVertex
Vertex position before the fit.
Definition: MassFitKFit.h:143
Belle2::analysis::MassFitKFit::MassFitKFit
MassFitKFit(void)
Construct an object with no argument.
Definition: MassFitKFit.cc:25
Belle2::analysis::MassFitKFit::getFlagFitWithVertex
bool getFlagFitWithVertex(void) const
Get a flag if the fit is allowed with moving the vertex position.
Definition: MassFitKFit.cc:186
Belle2::analysis::KFitConst::kAfterFit
static const int kAfterFit
Input parameter to specify after-fit when setting/getting a track attribute.
Definition: KFitConst.h:45
Belle2::analysis::MassFitKFit::setInvariantMass
enum KFitError::ECode setInvariantMass(const double m)
Set an invariant mass for the mass-constraint fit.
Definition: MassFitKFit.cc:68
Belle2::analysis::MassFitKFit::prepareCorrelation
enum KFitError::ECode prepareCorrelation(void) override
Build a grand correlation matrix from input-track properties.
Definition: MassFitKFit.cc:419
Belle2
Abstract base class for different kinds of events.
Definition: MillepedeAlgorithm.h:19
Belle2::analysis::MassFitKFit::~MassFitKFit
~MassFitKFit(void)
Destruct the object.
Belle2::analysis::MassFitKFit::m_InvariantMass
double m_InvariantMass
Invariant mass.
Definition: MassFitKFit.h:164
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:157
Belle2::analysis::MassFitKFit::setVertex
enum KFitError::ECode setVertex(const HepPoint3D &v)
Set an initial vertex position for the mass-constraint fit.
Definition: MassFitKFit.cc:44
Belle2::analysis::MassFitKFit::unfixMass
enum KFitError::ECode unfixMass(void)
Tell the object to unfix the last added track property at the invariant mass.
Definition: MassFitKFit.cc:92
Belle2::analysis::MassFitKFit::setVertexError
enum KFitError::ECode setVertexError(const CLHEP::HepSymMatrix &e)
Set an initial vertex error matrix for the mass-constraint fit.
Definition: MassFitKFit.cc:52
Belle2::analysis::MassFitKFit::updateMother
enum KFitError::ECode updateMother(Particle *mother)
Update mother particle.
Definition: MassFitKFit.cc:696
Belle2::analysis::MassFitKFit::m_FlagFitIncludingVertex
bool m_FlagFitIncludingVertex
Flag to indicate if the fit is allowed with moving the vertex position.
Definition: MassFitKFit.h:159
Belle2::analysis::MassFitKFit::prepareInputMatrix
enum KFitError::ECode prepareInputMatrix(void) override
Build grand matrices for minimum search from input-track properties.
Definition: MassFitKFit.cc:286
Belle2::analysis::MassFitKFit::getTrackCHIsq
double getTrackCHIsq(const int id) const override
Get a chi-square of the track.
Definition: MassFitKFit.cc:217
Belle2::analysis::MassFitKFit::makeCoreMatrix
enum KFitError::ECode makeCoreMatrix(void) override
Build matrices using the kinematical constraint.
Definition: MassFitKFit.cc:539
Belle2::Particle
Class to store reconstructed particles.
Definition: Particle.h:77
Belle2::analysis::MassFitKFit::doFit
enum KFitError::ECode doFit(void)
Perform a mass-constraint fit.
Definition: MassFitKFit.cc:280
Belle2::analysis::MassFitKFit::getVertexError
const CLHEP::HepSymMatrix getVertexError(const int flag=KFitConst::kAfterFit) const
Get a vertex error matrix.
Definition: MassFitKFit.cc:156
Belle2::analysis::MassFitKFit::getInvariantMass
double getInvariantMass(void) const
Get an invariant mass.
Definition: MassFitKFit.cc:172
Belle2::analysis::KFitError::ECode
ECode
ECode is a error code enumerate.
Definition: KFitError.h:43
HepGeom::Point3D< double >
Belle2::analysis::MassFitKFit::m_AfterVertex
HepPoint3D m_AfterVertex
Vertex position after the fit.
Definition: MassFitKFit.h:150
Belle2::analysis::MassFitKFit::prepareInputSubMatrix
enum KFitError::ECode prepareInputSubMatrix(void) override
Build sub-matrices for minimum search from input-track properties.
Definition: MassFitKFit.cc:408
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:84
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:253
Belle2::analysis::MassFitKFit::m_AfterVertexError
CLHEP::HepSymMatrix m_AfterVertexError
Vertex error matrix after the fit.
Definition: MassFitKFit.h:152
Belle2::analysis::MassFitKFit::m_AfterTrackVertexError
std::vector< CLHEP::HepMatrix > m_AfterTrackVertexError
array of vertex error matrices after the fit.
Definition: MassFitKFit.h:154