VertexFitKFit.h

/**************************************************************************
 * Copyright(C) 1997 - J. Tanaka                                          *
 *                                                                        *
 * Author: J. Tanaka                                                      *
 * Contributor: J. Tanaka and                                             *
 *              conversion to Belle II structure by T. Higuchi            *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <analysis/VertexFitting/KFit/KFitConst.h>
#include <analysis/VertexFitting/KFit/KFitError.h>
#include <analysis/VertexFitting/KFit/KFitBase.h>
 
#ifndef ENABLE_BACKWARDS_COMPATIBILITY
typedef HepGeom::Point3D<double> HepPoint3D;
#endif
 
 
namespace Belle2 {
  namespace analysis {
 
    class VertexFitKFit : public KFitBase {
    public:
      VertexFitKFit(void);
      ~VertexFitKFit(void);
 
 
    public:
      enum KFitError::ECode       setInitialVertex(const HepPoint3D& v);
      enum KFitError::ECode       setInitialVertex(const TVector3& v);
      enum KFitError::ECode       setIpProfile(const HepPoint3D& ip, const CLHEP::HepSymMatrix& ipe);
      enum KFitError::ECode       setIpTubeProfile(const CLHEP::HepLorentzVector& p, const HepPoint3D& x, const CLHEP::HepSymMatrix& e,
                                                   const double q);
      enum KFitError::ECode       setKnownVertex(const bool flag = true);
      enum KFitError::ECode       setCorrelationMode(const bool m);
 
 
    public:
      const HepPoint3D            getVertex(const int flag = KFitConst::kAfterFit) const;
      const CLHEP::HepSymMatrix          getVertexError(void) const;
      double                      getCHIsq(void) const override;
      double                      getCHIsqVertex(void) const;
      const CLHEP::HepMatrix             getTrackVertexError(const int id) const;
      double                      getTrackCHIsq(const int id) const override;
      double                      getTrackPartCHIsq(void) const;
      int                         getTrackPartNDF(void) const;
 
 
    public:
      enum KFitError::ECode doFit(void);
 
      enum KFitError::ECode updateMother(Particle* mother);
 
    private:
      enum KFitError::ECode doFit3(void);
      enum KFitError::ECode doFit4(void);
      enum KFitError::ECode doFit5(void);
 
 
    private:
      enum KFitError::ECode prepareInputMatrix(void) override;
      enum KFitError::ECode prepareInputSubMatrix(void) override;
      enum KFitError::ECode prepareOutputMatrix(void) override;
      enum KFitError::ECode makeCoreMatrix(void) override;
      enum KFitError::ECode calculateNDF(void) override;
 
 
    private:
      enum KFitError::ECode appendTube(void);
      enum KFitError::ECode deleteTube(void);
 
 
    private:
      bool   m_CorrelationMode;
      double m_EachCHIsq[KFitConst::kMaxTrackCount2];
      double m_CHIsqVertex;
 
      HepPoint3D             m_BeforeVertex;
 
      HepPoint3D             m_AfterVertex;
      CLHEP::HepSymMatrix           m_AfterVertexError;
      std::vector<CLHEP::HepMatrix> m_AfterTrackVertexError;
 
      bool         m_FlagBeam;
      CLHEP::HepSymMatrix m_BeamError;
 
      bool m_FlagKnownVertex;
 
      bool      m_FlagTube;
      int       m_iTrackTube;
      KFitTrack m_TubeTrack;
    };
 
  } // namespace analysis
 
} // namespace Belle2