MatrixIO.h

Go to the documentation of this file.

#pragma once

#include <Eigen/Core>

#include <iostream>

namespace CortidQCT {

template <class MatrixType> struct MartrixIOWrapper {

  inline MartrixIOWrapper(Eigen::MatrixBase<MatrixType> &matrix)
      : matrix_(matrix.derived()) {}

  friend inline std::ostream &operator<<(std::ostream &os,
                                         MartrixIOWrapper const &wrapper) {
    return os << wrapper.matrix_;
  }

  friend inline std::istream &operator>>(std::istream &is,
                                         MartrixIOWrapper &&wrapper) {

    for (auto i = 0; i < wrapper.matrix_.rows(); ++i)
      for (auto j = 0; j < wrapper.matrix_.cols(); ++j)
        is >> wrapper.matrix_(i, j);

    return is;
  }

private:
  MatrixType &matrix_;
};

template <class MatrixType>
inline MartrixIOWrapper<MatrixType> io(Eigen::MatrixBase<MatrixType> &matrix) {
  return MartrixIOWrapper<MatrixType>(matrix);
}

template <class MatrixType>
inline MartrixIOWrapper<MatrixType>
io(Eigen::MatrixBase<MatrixType> const &matrix) {
  return MartrixIOWrapper<MatrixType const>(matrix);
}

} // namespace CortidQCT