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