FourMomentumKinematicConstraint.h

Go to the documentation of this file.

#ifndef FourMomentumKinematicConstraint_H
#define FourMomentumKinematicConstraint_H

#include "RecoVertex/KinematicFitPrimitives/interface/KinematicConstraint.h"
#include "RecoVertex/KinematicFitPrimitives/interface/ParticleMass.h"

class FourMomentumKinematicConstraint : public KinematicConstraint {
public:
  FourMomentumKinematicConstraint(const AlgebraicVector& momentum, const AlgebraicVector& deviation);

  std::pair<AlgebraicVector, AlgebraicVector> value(const AlgebraicVector& exPoint) const override;

  std::pair<AlgebraicMatrix, AlgebraicVector> derivative(const AlgebraicVector& exPoint) const override;

  std::pair<AlgebraicMatrix, AlgebraicVector> derivative(
      const std::vector<RefCountedKinematicParticle>& par) const override;

  std::pair<AlgebraicVector, AlgebraicVector> value(const std::vector<RefCountedKinematicParticle>& par) const override;

  int numberOfEquations() const override;

  AlgebraicVector deviations(int nStates) const override;

  FourMomentumKinematicConstraint* clone() const override { return new FourMomentumKinematicConstraint(*this); }

private:
  AlgebraicVector mm;
  AlgebraicVector dd;
};
#endif