ColinearityKinematicConstraint.h

Go to the documentation of this file.

#ifndef ColinearityKinematicConstraint_H
#define ColinearityKinematicConstraint_H

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

class ColinearityKinematicConstraint : public MultiTrackKinematicConstraint {
public:
  enum ConstraintDim { Phi, PhiTheta };

  ColinearityKinematicConstraint(ConstraintDim dim = Phi);

  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 size; }

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

private:
  ConstraintDim dimension;
  unsigned int size;
};
#endif