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