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