MassConstraint.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TrkVertexKinematicFitterUtils_MassConstraint_h
#define TrkVertexKinematicFitterUtils_MassConstraint_h
 
// This object defines a MASS constraint to be used with the constrained vertex
// fitter
 
#include "TrkVertexFitterInterfaces/IKinematicConstraint.h"
 
namespace Trk {
 
class MassConstraint : public IKinematicConstraint
{
public:
  MassConstraint() = default;

  MassConstraint(double mass);
 
  virtual ~MassConstraint() = default;

  virtual Amg::VectorX vectorOfValues(
    std::vector<Amg::VectorX>& cart_coordList,
    std::vector<int>& charges,
    Amg::Vector3D refPoint,
    double b_fieldTesla) const override final;

  virtual Amg::MatrixX matrixOfDerivatives(
    std::vector<Amg::VectorX>& cart_coordList,
    std::vector<int>& charges,
    Amg::Vector3D refPoint,
    double b_fieldTesla) const override final;

  virtual int numberOfEquations() const override final { return s_eqno; }

  double mass() { return m_mass; }
 
private:
  double m_mass = 0.;
  static constexpr int s_eqno = 1;
};
}
 
#endif // TrkVertexKinematicFitterUtils_MassConstraint_h