MultiTrackKinematicConstraintT.h

Go to the documentation of this file.

#ifndef MultiTrackKinematicConstraintT_H
#define MultiTrackKinematicConstraintT_H
 
#include "RecoVertex/KinematicFitPrimitives/interface/RefCountedKinematicParticle.h"
#include "RecoVertex/KinematicFitPrimitives/interface/RefCountedKinematicVertex.h"
#include "DataFormats/CLHEP/interface/AlgebraicObjects.h"
 
class MultiTrackKinematicConstraintBaseT {
public:
  virtual ~MultiTrackKinematicConstraintBaseT() {}
 
  // initialize the constraint so it can precompute common qualtities to the three next call
  virtual void init(const std::vector<KinematicState>& states, const GlobalPoint& point, const GlobalVector& mf) = 0;
 
  virtual int numberOfEquations() const = 0;
 
  virtual MultiTrackKinematicConstraintBaseT* clone() const = 0;
};
 
template <int NTRK, int DIM>
class MultiTrackKinematicConstraintT : public MultiTrackKinematicConstraintBaseT {
public:
  enum { nTrk = NTRK, nDim = DIM };
 
  typedef MultiTrackKinematicConstraintT<NTRK, DIM> self;
 
  typedef ROOT::Math::SVector<double, DIM> valueType;
 
  typedef ROOT::Math::SMatrix<double, DIM, 7 * NTRK> parametersDerivativeType;
 
  typedef ROOT::Math::SMatrix<double, DIM, 3> positionDerivativeType;
 
  ~MultiTrackKinematicConstraintT() override {}
 
  valueType const& value() const {
    fillValue();
    return m_vl;
  }
 
  parametersDerivativeType const& parametersDerivative() const {
    fillParametersDerivative();
    return m_jac_d;
  };
 
  positionDerivativeType const& positionDerivative() const {
    fillPositionDerivative();
    return m_jac_e;
  }
 
private:
  virtual void fillValue() const = 0;
 
  virtual void fillParametersDerivative() const = 0;
 
  virtual void fillPositionDerivative() const = 0;
 
protected:
  valueType& vl() const { return m_vl; }
  parametersDerivativeType& jac_d() const { return m_jac_d; }
  positionDerivativeType& jac_e() const { return m_jac_e; }
 
  // self & me() const { return *const_cast<self*>(this); }
 
  double& vl(size_t i) const { return m_vl(i); }
  double& jac_d(size_t i, size_t j) const { return m_jac_d(i, j); }
  double& jac_e(size_t i, size_t j) const { return m_jac_e(i, j); }
 
private:
  mutable valueType m_vl;
  mutable parametersDerivativeType m_jac_d;
  mutable positionDerivativeType m_jac_e;
};
 
#endif