SingleGaussianState.h

Go to the documentation of this file.

#ifndef SingleGaussianState_H
#define SingleGaussianState_H

#define SMATRIX_USE_CONSTEXPR

#include "Math/SVector.h"
#include "Math/SMatrix.h"

template <unsigned int N> class SingleGaussianState {
public:
  using Vector = ROOT::Math::SVector<double, N>;
  using Matrix = ROOT::Math::SMatrix<double,N,N,ROOT::Math::MatRepSym<double,N>>;
  
public:
  SingleGaussianState() {}

  SingleGaussianState(const Vector& aMean,
              const Matrix& aCovariance, 
              double aWeight = 1.) : 
    theCovariance(aCovariance), theMean(aMean), theWeight(aWeight) {}  

  inline double weight() const {return theWeight;}
  inline const Vector & mean() const {return theMean;}
  inline const Matrix & covariance() const {return theCovariance;}
  const Matrix & weightMatrix() const;
  inline unsigned int dimension () const {return N;}
  void rescaleWeight (double scale) {theWeight *= scale;}

private:
  Matrix theCovariance;
  Vector theMean;
  double theWeight;

  mutable Matrix theWeightMatrix = ROOT::Math::SMatrixNoInit();
  mutable bool theHasWeightMatrix = false;
};

#include "TrackingTools/GsfTools/interface/SingleGaussianState.icc"

#endif