V0ValidationVertex.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 <mdst/dataobjects/V0.h>
 
#include <Math/Vector3D.h>
 
namespace Belle2 {

  class V0ValidationVertex : public V0 {
  public:

    V0ValidationVertex() :
      V0(),
      m_fittedMomentum(0),
      m_fittedInvariantMass(0),
      m_chi2(0)
    {}

    V0ValidationVertex(const std::pair<const Belle2::Track*, const Belle2::TrackFitResult*>& trackPairPositive,
                       const std::pair<const Belle2::Track*, const Belle2::TrackFitResult*>& trackPairNegative,
                       double vertexX, double vertexY, double vertexZ) :
      V0(trackPairPositive, trackPairNegative, vertexX, vertexY, vertexZ),
      m_fittedMomentum(0),
      m_fittedInvariantMass(0),
      m_chi2(0)
    {}

    V0ValidationVertex(const std::pair<const Belle2::Track*, const Belle2::TrackFitResult*>& trackPairPositive,
                       const std::pair<const Belle2::Track*, const Belle2::TrackFitResult*>& trackPairNegative,
                       const ROOT::Math::XYZVector& fittedVertexPosition,
                       const TMatrixDSym& fittedVertexPositionCovariance,
                       const double& momentum,
                       const double& invariantMass,
                       const double& chi2) :
      V0(trackPairPositive, trackPairNegative, fittedVertexPosition.X(), fittedVertexPosition.Y(), fittedVertexPosition.Z()),
      m_fittedVertexPositionCovariance(fittedVertexPositionCovariance),
      m_fittedMomentum(momentum),
      m_fittedInvariantMass(invariantMass),
      m_chi2(chi2)
    {}

    ROOT::Math::XYZVector getVertexPosition() const { return getFittedVertexPosition(); }

    TMatrixDSym getVertexPositionCovariance() const { return m_fittedVertexPositionCovariance; }

    float getFittedMomentum() const { return m_fittedMomentum; }

    float getFittedInvariantMass() const { return m_fittedInvariantMass; }

    float getVertexChi2() const { return m_chi2; }
 
  private:
    TMatrixDSym m_fittedVertexPositionCovariance;  
    float m_fittedMomentum; 
    float m_fittedInvariantMass; 
    float m_chi2; 
 
    ClassDef(V0ValidationVertex, 4);  
  };

}