TwoTrackMassKinematicConstraint.h

Go to the documentation of this file.

#ifndef TwoTrackMassKinematicConstraint_H
#define TwoTrackMassKinematicConstraint_H

#include "RecoVertex/KinematicFitPrimitives/interface/MultiTrackKinematicConstraint.h"
#include "RecoVertex/KinematicFitPrimitives/interface/KinematicState.h"
#include "DataFormats/CLHEP/interface/AlgebraicObjects.h"
#include "RecoVertex/KinematicFitPrimitives/interface/ParticleMass.h"

class TwoTrackMassKinematicConstraint : public MultiTrackKinematicConstraint {
public:
  TwoTrackMassKinematicConstraint(ParticleMass& ms) : mass(ms) {}

  AlgebraicVector value(const std::vector<KinematicState>& states, const GlobalPoint& point) const override;

  AlgebraicMatrix parametersDerivative(const std::vector<KinematicState>& states,
                                       const GlobalPoint& point) const override;

  AlgebraicMatrix positionDerivative(const std::vector<KinematicState>& states,
                                     const GlobalPoint& point) const override;

  int numberOfEquations() const override;

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

private:
  ParticleMass mass;
};
#endif