BufferedMatrix.h

Go to the documentation of this file.

 
#pragma once
 
#include <Eigen/Core>
 
namespace tvm
{
 
namespace utils
{
 
namespace internal
{
class BufferedMatrix
{
public:
  BufferedMatrix(Eigen::DenseIndex m, Eigen::DenseIndex n);
 
  Eigen::Map<const Eigen::MatrixXd, Eigen::Aligned> get() const;
  Eigen::Map<Eigen::MatrixXd, Eigen::Aligned> get();
 
  void resize(Eigen::DenseIndex m, Eigen::DenseIndex n);
 
  template<typename Derived>
  Eigen::Map<Eigen::MatrixXd, Eigen::Aligned> & operator=(const Eigen::EigenBase<Derived> & xpr);
 
private:
  Eigen::DenseIndex m_;    
  Eigen::DenseIndex n_;    
  Eigen::VectorXd buffer_; 
};
 
inline BufferedMatrix::BufferedMatrix(Eigen::DenseIndex m, Eigen::DenseIndex n) { resize(m, n); }
 
inline Eigen::Map<const Eigen::MatrixXd, Eigen::Aligned> BufferedMatrix::get() const
{ return Eigen::Map<const Eigen::MatrixXd, Eigen::Aligned>(buffer_.data(), m_, n_); }
 
inline Eigen::Map<Eigen::MatrixXd, Eigen::Aligned> BufferedMatrix::get()
{ return Eigen::Map<Eigen::MatrixXd, Eigen::Aligned>(buffer_.data(), m_, n_); }
 
inline void BufferedMatrix::resize(Eigen::DenseIndex m, Eigen::DenseIndex n)
{
  if(m * n > buffer_.size())
  {
    buffer_.resize(2 * m * n);
  }
  m_ = m;
  n_ = n;
}
 
template<typename Derived>
inline Eigen::Map<Eigen::MatrixXd, Eigen::Aligned> & BufferedMatrix::operator=(const Eigen::EigenBase<Derived> & xpr)
{
  resize(xpr.rows(), xpr.cols());
  return get() = xpr;
}
 
} // namespace internal
 
} // namespace utils
 
} // namespace tvm