FitParams.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2018 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributor: Wouter Hulsbergen, Francesco Tenchini, Jo-Frederik Krohn  *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <analysis/VertexFitting/TreeFitter/EigenStackConfig.h>
#include <Eigen/Core>
#include <vector>
 
namespace TreeFitter {
 
  class ParticleBase;
 
  class FitParams {
 
  public:
 
    explicit FitParams(const int dim);
 
    ~FitParams() {};
 
    FitParams(const FitParams& toCopy)
      : m_dim(toCopy.m_dim),
        m_chiSquare(toCopy.m_chiSquare),
        m_nConstraints(toCopy.m_nConstraints),
        m_dimensionReduction(toCopy.m_dimensionReduction),
        m_globalState(Eigen::Matrix < double, -1, 1, 0, MAX_MATRIX_SIZE, 1 > (toCopy.m_globalState)),
        m_globalCovariance(Eigen::Matrix < double, -1, -1, 0, MAX_MATRIX_SIZE, MAX_MATRIX_SIZE > (toCopy.m_globalCovariance))
    { }
 
    FitParams& operator =(const FitParams& other)
    {
      m_dim = other.m_dim;
      m_chiSquare = other.m_chiSquare;
      m_nConstraints = other.m_nConstraints;
      m_dimensionReduction = other.m_dimensionReduction;
      m_globalState = other.m_globalState;
      m_globalCovariance = other.m_globalCovariance;
      return *this;
    }
 
    Eigen::Matrix < double, -1, -1, 0, MAX_MATRIX_SIZE, MAX_MATRIX_SIZE > & getCovariance()
    {
      return m_globalCovariance;
    }
 
    const Eigen::Matrix < double, -1, -1, 0, MAX_MATRIX_SIZE, MAX_MATRIX_SIZE > & getCovariance() const
    {
      return m_globalCovariance;
    }
 
    Eigen::Matrix < double, -1, 1, 0, MAX_MATRIX_SIZE, 1 > & getStateVector() {return m_globalState;}
 
    const Eigen::Matrix < double, -1, 1, 0, MAX_MATRIX_SIZE, 1 > & getStateVector() const
    {
      return m_globalState;
    }
 
    double& getRefToElementOfStateVec(int row) { return m_globalState(row, 0); }
 
    const double& getRefToElementOfStateVec(int row) const { return m_globalState(row, 0); }
 
    double getCovDiaElement(int counter) { return m_globalCovariance(counter, counter); }
 
    void resetStateVector();
 
    void resetCovariance();
 
    void resizeAndResetStateAndCov(int newdim);
 
    int getDimensionOfState() const {return m_dim;};
 
    int getNConstraints() {return m_nConstraints;}
 
    bool testCovariance() const;
 
    int& incrementNConstraintsVec(int row) { return m_nConstraintsVec[row];}
 
    int& nConstraintsVec(int row) { return m_nConstraintsVec[row - 1]; }
 
    int dim() const { return m_dim; }
 
    double chiSquare() const;
 
    int nConstraints() const { return m_nConstraints; }
 
    int nDof() const;
 
    void resize(int newdim);
 
    void resetPar();
 
    bool testCov() const;
 
    void addNConstraint(int value) { m_nConstraints += value; }
 
    void addChiSquare(double chisq, int nconstraints)
    {
      m_chiSquare += chisq;
      m_nConstraints += nconstraints;
    }
 
    void resetChiSquare()
    {
      m_chiSquare = 0;
      m_nConstraints = 0;
    }
 
    typedef std::vector< std::pair<const ParticleBase*, int> > indexmap;
 
  protected:
 
    FitParams() {}
 
  private:
 
    int m_dim;
 
    double m_chiSquare;
 
    int m_nConstraints;
 
    int m_dimensionReduction;
 
    std::vector<int> m_nConstraintsVec;
 
    Eigen::Matrix < double, -1, 1, 0, MAX_MATRIX_SIZE, 1 > m_globalState;
 
    Eigen::Matrix < double, -1, -1, 0, MAX_MATRIX_SIZE, MAX_MATRIX_SIZE > m_globalCovariance;
 
  };
}