Projection.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 * External Contributor: Wouter Hulsbergen                                *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
#pragma once
 
#include <analysis/VertexFitting/TreeFitter/EigenStackConfig.h>
 
#include <Eigen/Dense>
 
namespace TreeFitter {
 
  class Projection {
 
  public:
 
    Projection(int dimP, int dimC);
 
    virtual ~Projection() {};
 
    Eigen::Matrix < double, -1, -1, 0, 7, MAX_MATRIX_SIZE > & getH() {return m_H;}
 
    const Eigen::Matrix < double, -1, -1, 0, 7, MAX_MATRIX_SIZE > getH() const {return m_H;}
 
    Eigen::Matrix < double, -1, 1, 0, 7, 1 > & getResiduals() {return m_residual;}
 
    const Eigen::Matrix < double, -1, 1, 0, 7, 1 > getResiduals() const {return m_residual;}
 
    double getElementInResiduals(int row) { return m_residual(m_offset + row, 1); }
 
    Eigen::Matrix < double, -1, -1, 0, 7, 7 > & getV() {return m_V;}
 
    const Eigen::Matrix < double, -1, -1, 0, 7, 7 > & getV() const {return m_V;}
 
    void resetProjection();
 
    unsigned int getOffset() const { return m_offset; }
 
    void incrementOffset(unsigned int i) { m_offset += i; }
 
    unsigned int offset() const { return m_offset; }
 
    int getDim() const { return m_dimProj; }
 
    int getDimCov() const { return m_dimCov; }
  private:
 
    Eigen::Matrix < double, -1, -1, 0, 7, MAX_MATRIX_SIZE > m_H;
 
    Eigen::Matrix < double, -1, -1, 0, 7, 7 > m_V;
 
    Eigen::Matrix < double, -1, 1, 0, 7, 1 > m_residual;
 
    int m_dimCov;
 
    int m_dimProj;
 
    unsigned int m_offset;
  };
}