KFitTrack.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 * External Contributor: J. Tanaka                                        *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <CLHEP/Matrix/Matrix.h>
#include <CLHEP/Matrix/SymMatrix.h>
#include <CLHEP/Vector/LorentzVector.h>
#include <CLHEP/Geometry/Point3D.h>
 
#include <framework/logging/Logger.h>
 
#include <analysis/VertexFitting/KFit/KFitConst.h>
 
#ifndef ENABLE_BACKWARDS_COMPATIBILITY
typedef HepGeom::Point3D<double> HepPoint3D;
#endif
 
 
namespace Belle2 {
  namespace analysis {
 
    class KFitTrack {
    public:
      struct KFitPXE {
        CLHEP::HepLorentzVector   m_P;
        HepPoint3D         m_X;
        CLHEP::HepSymMatrix       m_E;
      };
 
      KFitTrack(void);
      KFitTrack(const KFitTrack& kp);
      KFitTrack(
        const CLHEP::HepLorentzVector& p,
        const HepPoint3D&       x,
        const CLHEP::HepSymMatrix&     e,
        const double           q,
        const int              flag = KFitConst::kBeforeFit
      );
 
      ~KFitTrack(void);
 
 
    public:
      KFitTrack& operator = (const KFitTrack&);
 
 
    public:
      void setMomentum(const CLHEP::HepLorentzVector& p, const int flag = KFitConst::kBeforeFit);
      void setPosition(const HepPoint3D& x,       const int flag = KFitConst::kBeforeFit);
      void setError(const CLHEP::HepSymMatrix& e,     const int flag = KFitConst::kBeforeFit);
      void setCharge(const double q);
      void setVertex(const HepPoint3D& v);
      void setVertexError(const CLHEP::HepSymMatrix& ve);
 
      const CLHEP::HepLorentzVector getMomentum(const int flag = KFitConst::kAfterFit) const;
      const HepPoint3D       getPosition(const int flag = KFitConst::kAfterFit) const;
      const CLHEP::HepSymMatrix     getError(const int flag = KFitConst::kAfterFit) const;
      double                 getCharge(void) const;
      double                 getMass(void) const;
      const HepPoint3D       getVertex(void) const;
      const CLHEP::HepSymMatrix     getVertexError(void) const;
 
      double                 getFitParameter(const int which, const int flag) const;
      const CLHEP::HepMatrix        getFitParameter(const int flag) const;
      const CLHEP::HepSymMatrix     getFitError(const int flag) const;
      const CLHEP::HepMatrix        getMomPos(const int flag) const;
 
 
    private:
      inline void checkFlag(const int flag) const
      {
        if (flag != KFitConst::kBeforeFit && flag != KFitConst::kAfterFit) B2FATAL("checkFlag");
      }
 
      inline void checkMatrixDimension(const CLHEP::HepSymMatrix& m, const int dim) const
      {
        if (m.num_row() != dim) B2FATAL("checkMatrixDimension");
      }
 
 
    private:
      struct KFitPXE m_PXEBefore;
 
      struct KFitPXE m_PXEAfter;
 
      double         m_Charge;
      double         m_Mass;
 
      HepPoint3D     m_Vertex;
      CLHEP::HepSymMatrix   m_VertexError;
    };
 
  } // namespace analysis
} // namespace Belle2