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