Projection.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/Dense>
 
namespace TreeFitter {
 
  class Projection {
 
  public:
 
    Projection(int dimP, int dimC);
 
    virtual ~Projection() {};
 
    Eigen::Matrix < double, -1, -1, 0, 5, MAX_MATRIX_SIZE > & getH() {return m_H;}
 
    const Eigen::Matrix < double, -1, -1, 0, 5, MAX_MATRIX_SIZE > getH() const {return m_H;}
 
    Eigen::Matrix < double, -1, 1, 0, 5, 1 > & getResiduals() {return m_residual;}
 
    const Eigen::Matrix < double, -1, 1, 0, 5, 1 > getResiduals() const {return m_residual;}
 
    double getElementInResiduals(int row) { return m_residual(m_offset + row, 1); }
 
    Eigen::Matrix < double, -1, -1, 0, 5, 5 > & getV() {return m_V;}
 
    const Eigen::Matrix < double, -1, -1, 0, 5, 5 > & getV() const {return m_V;}
 
    void resetProjection();
 
    unsigned int getOffset() { 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, 5, MAX_MATRIX_SIZE > m_H;
 
    Eigen::Matrix < double, -1, -1, 0, 5, 5 > m_V;
 
    Eigen::Matrix < double, -1, 1, 0, 5, 1 > m_residual;
 
    int m_dimCov;
 
    int m_dimProj;
 
    unsigned int m_offset;
  };
}