MakeMotherKFit.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 <vector>
 
#include <CLHEP/Matrix/Matrix.h>
#include <CLHEP/Matrix/SymMatrix.h>
#include <CLHEP/Vector/ThreeVector.h>
#include <CLHEP/Vector/LorentzVector.h>
#include <CLHEP/Geometry/Point3D.h>
 
#include <analysis/VertexFitting/KFit/KFitConst.h>
#include <analysis/VertexFitting/KFit/KFitError.h>
#include <analysis/VertexFitting/KFit/KFitTrack.h>
 
#ifndef ENABLE_BACKWARDS_COMPATIBILITY
typedef HepGeom::Point3D<double> HepPoint3D;
#endif
 
 
namespace Belle2 {
  namespace analysis {
 
    class MakeMotherKFit {
    public:
      MakeMotherKFit(void);
      ~MakeMotherKFit(void);
 
      enum KFitError::ECode       addTrack(const KFitTrack& kp);
      enum KFitError::ECode       addTrack(const CLHEP::HepLorentzVector& p, const HepPoint3D& x, const CLHEP::HepSymMatrix& e,
                                           const double q, const int flag = KFitConst::kAfterFit);
      enum KFitError::ECode       setMagneticField(const double mf);
      enum KFitError::ECode       setVertex(const HepPoint3D& v);
      enum KFitError::ECode       setVertexError(const CLHEP::HepSymMatrix& e);
      enum KFitError::ECode       setTrackVertexError(const CLHEP::HepMatrix& e);
      enum KFitError::ECode       setTrackZeroVertexError(void);
      enum KFitError::ECode       setCorrelation(const CLHEP::HepMatrix& e);
      enum KFitError::ECode       setZeroCorrelation(void);
      enum KFitError::ECode       setFlagAtDecayPoint(const bool flag);
      enum KFitError::ECode       setFlagBeforeAfter(const int flag);
 
      enum KFitError::ECode       getErrorCode(void) const;
      const KFitTrack             getMother(void) const;
      const CLHEP::HepLorentzVector      getMotherMomentum(void) const;
      const HepPoint3D            getMotherPosition(void) const;
      const CLHEP::HepSymMatrix          getMotherError(void) const;
 
      enum KFitError::ECode doMake(void);
 
    private:
      void calculateError(CLHEP::HepSymMatrix* Ec) const;
      void calculateDELMDELC(CLHEP::HepMatrix* e) const;
 
 
    private:
      enum KFitError::ECode m_ErrorCode;
 
      bool     m_FlagAtDecayPoint;
      int      m_FlagBeforeAfter;
      bool     m_FlagVertexError;
      bool     m_FlagCorrelation;
      bool     m_FlagTrackVertexError;
 
      double   m_MagneticField;
      int      m_TrackCount;
 
 
      std::vector<KFitTrack> m_Tracks;
      std::vector<CLHEP::HepMatrix> m_TrackVertexError;
      std::vector<CLHEP::HepMatrix> m_Correlation;
 
 
      CLHEP::Hep3Vector   m_Vertex;
      CLHEP::HepSymMatrix m_VertexError;
 
 
      double    m_Charge;
      KFitTrack m_Mother;
    };
 
  } // namespace analysis
 
} // namespace Belle2