MultiTrackMassKinematicConstraint.h

Go to the documentation of this file.

#ifndef MultiTrackMassKinematicConstraint_H
#define MultiTrackMassKinematicConstraint_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 MultiTrackMassKinematicConstraint : public MultiTrackKinematicConstraint {
public:
  MultiTrackMassKinematicConstraint(const ParticleMass& theMass, const unsigned int nbrParticles)
      : mass(theMass), nPart(nbrParticles) {}

  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 { return 1; }

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

private:
  const ParticleMass mass;
  const unsigned int nPart;
};
#endif