MassFourCFitKFit.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <analysis/VertexFitting/KFit/KFitConst.h>
#include <analysis/VertexFitting/KFit/KFitError.h>
#include <analysis/VertexFitting/KFit/KFitBase.h>
 
#include <Math/Vector4D.h>
 
#ifndef ENABLE_BACKWARDS_COMPATIBILITY
typedef HepGeom::Point3D<double> HepPoint3D;
#endif
 
namespace Belle2 {
 
  namespace analysis {

    class MassFourCFitKFit : public KFitBase {
    public:
      MassFourCFitKFit();
      ~MassFourCFitKFit(void);
 
 
    public:
 

      enum KFitError::ECode       addMassConstraint(const double m, std::vector<unsigned>& childTrackId);

      enum KFitError::ECode       setVertex(const HepPoint3D& v);
      enum KFitError::ECode       setVertexError(const CLHEP::HepSymMatrix& e);
      enum KFitError::ECode       setInvariantMass(const double m);
      enum KFitError::ECode       setFourMomentum(const ROOT::Math::PxPyPzEVector& m);
      enum KFitError::ECode       setFlagAtDecayPoint(const bool flag);
      enum KFitError::ECode       fixMass(void);
      enum KFitError::ECode       unfixMass(void);
      enum KFitError::ECode       setTrackVertexError(const CLHEP::HepMatrix& e);
      enum KFitError::ECode       setTrackZeroVertexError(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(const int flag = KFitConst::kAfterFit) const;
      double                      getInvariantMass(void) const;
      bool                        getFlagAtDecayPoint(void) const;
      bool                        getFlagFitWithVertex(void) const;
      double                      getCHIsq(void) const override;
      const CLHEP::HepMatrix             getTrackVertexError(const int id, const int flag = KFitConst::kAfterFit) 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;
      CLHEP::HepSymMatrix           m_BeforeVertexError;
      std::vector<CLHEP::HepMatrix> m_BeforeTrackVertexError;

      HepPoint3D             m_AfterVertex;
      CLHEP::HepSymMatrix           m_AfterVertexError;
      std::vector<CLHEP::HepMatrix> m_AfterTrackVertexError;

      bool m_FlagTrackVertexError;
      bool m_FlagFitIncludingVertex;
      bool m_FlagAtDecayPoint;

      double m_InvariantMass;

      ROOT::Math::PxPyPzEVector m_FourMomentum;

      std::vector<int> m_IsFixMass;

      int m_ConstraintMassCount;

      std::vector<double> m_ConstraintMassList;

      std::vector<std::pair<int, int>> m_ConstraintMassChildLists;
    };
 
  } // namespace analysis

} // namespace Belle2