eo_matrix.h

 
#ifndef EO_MATRIX_H
#define EO_MATRIX_H
 
#include <algorithm>
#include <Eigen/Dense>
#include <stdlib.h>
 
typedef Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic, Eigen::ColMajor> dMat; // declares a column-major non-sparse matrix type of double
typedef Eigen::Matrix<double, Eigen::Dynamic, 1, Eigen::ColMajor> dVec; // declares a vector of double.
 
#include <unsupported/Eigen/MatrixFunctions>
 
inline void removeRow(dMat& matrix, unsigned int rowToRemove)
{
  unsigned int numRows = matrix.rows()-1;
  unsigned int numCols = matrix.cols();
 
  if( rowToRemove < numRows )
    matrix.block(rowToRemove,0,numRows-rowToRemove,numCols) = matrix.block(rowToRemove+1,0,numRows-rowToRemove,numCols);
 
  matrix.conservativeResize(numRows,numCols);
}
 
inline void removeColumn(dMat& matrix, unsigned int colToRemove)
{
  unsigned int numRows = matrix.rows();
  unsigned int numCols = matrix.cols()-1;
 
  if( colToRemove < numCols )
    matrix.block(0,colToRemove,numRows,numCols-colToRemove) = matrix.block(0,colToRemove+1,numRows,numCols-colToRemove);
 
  matrix.conservativeResize(numRows,numCols);
}
 
inline void removeElement(dVec &vec, unsigned int k)
{
  if (k >= vec.size())
    return;
  std::copy(vec.data()+k+1,vec.data()+vec.size(),vec.data()+k);
  vec.conservativeResize(vec.size()-1);
}
 
inline void addElement(dVec &vec, unsigned int k, const double &xk)
{
  if (k >= vec.size()+1)
    return;
  vec.conservativeResize(vec.size()+1);
  std::copy(vec.data()+k,vec.data()+vec.size()-1,vec.data()+k+1);
  vec[k] = xk;
}
 
#endif