MassPointingVertexFitKFit.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2019 - Belle II Collaboration                             *
 * Contributors: Frank Meier                                              *
 *                                                                        *
 * 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>
 
namespace Belle2 {
  namespace analysis {
 
    class MassPointingVertexFitKFit : public KFitBase {
    public:
      MassPointingVertexFitKFit(void);
      ~MassPointingVertexFitKFit(void);
 
 
    public:
      enum KFitError::ECode       setInitialVertex(const HepPoint3D& v);
      enum KFitError::ECode       setInvariantMass(const double m);
      enum KFitError::ECode       setProductionVertex(const HepPoint3D& v);
      enum KFitError::ECode       fixMass(void);
      enum KFitError::ECode       unfixMass(void);
      enum KFitError::ECode       setCorrelation(const CLHEP::HepMatrix& m) override;
      enum KFitError::ECode       setZeroCorrelation(void) override;
 
      const HepPoint3D            getVertex(const int flag = KFitConst::kAfterFit) const;
      const CLHEP::HepSymMatrix   getVertexError(void) const;
      double                      getInvariantMass(void) const;
      double                      getCHIsq(void) const override;
      const CLHEP::HepMatrix      getTrackVertexError(const int id) const;
      double                      getTrackCHIsq(const int id) const override;
      const CLHEP::HepMatrix      getCorrelation(const int id1, const int id2,
                                                 const int flag = KFitConst::kAfterFit) const override;
 
 
    public:
      enum KFitError::ECode doFit(void);
 
      enum KFitError::ECode updateMother(Particle* mother);
 
 
    private:
      enum KFitError::ECode prepareInputMatrix(void) override;
      enum KFitError::ECode prepareInputSubMatrix(void) override;
      enum KFitError::ECode prepareCorrelation(void) override;
      enum KFitError::ECode prepareOutputMatrix(void) override;
      enum KFitError::ECode makeCoreMatrix(void) override;
      enum KFitError::ECode calculateNDF(void) override;
 
 
    private:
      HepPoint3D             m_BeforeVertex;
      HepPoint3D             m_AfterVertex;
      CLHEP::HepSymMatrix           m_AfterVertexError;
      std::vector<CLHEP::HepMatrix> m_AfterTrackVertexError;
      HepPoint3D m_ProductionVertex;
      double m_InvariantMass;
      std::vector<int> m_IsFixMass;
    };
 
  } // namespace analysis
 
} // namespace Belle2