operator/outputMatlab.hh

Go to the documentation of this file.

/* matlab output function for matrices
 */

#ifndef OperatorOutputMatlab_hh
#define OperatorOutputMatlab_hh

#include "basics/outputMatlab.hh"
#include "sparseMatrix.hh"

namespace concepts {

  // ********************************************** storeDenseMatrixToMatlab **

  static uint storeDenseMatrixMatlabCounter_ = 0;
  static uint storeSparseMatrixMatlabCounter_ = 0;

  template<class F>
  bool storeDenseMatrixToMatlab(F& matrix, const uint nofRows, 
        const uint nofCols, const std::string filename,
        std::string name = "", bool append = false) {

    // Mode, new or append
    std::ios_base::openmode mode = std::ofstream::out;
    if (append) mode = mode | std::ofstream::app;

    std::ofstream ofs(filename.c_str(), mode);
    if (!ofs.is_open()) {
      std::cerr << "Could not open file '" << filename << "'" << std::endl;
      return false;
    }

    storeDenseMatrixToMatlab(matrix, nofRows, nofCols, ofs, name);
    return true;
  }


  template<class F>
    void storeDenseMatrixToMatlab(F& matrix, const uint nofRows, 
        const uint nofCols, std::ostream& ofs,
        std::string name = "") {
      if (name.empty()) {
        std::stringstream s;
        s << "MatDense_" << storeDenseMatrixMatlabCounter_++;
        name = s.str();
      }
      std::string offset(name.size() + 4, ' ');
      ofs << name << " = [" << std::setprecision(16);

      for (uint i = 0; i < nofRows; ++i) {
        for (uint j = 0; j < nofCols; ++j) {
          ofs << OutputMatlab<typename F::value_type>(matrix(i,j));
          if (j == nofCols-1) {
            if (i < nofRows-1)
              ofs << std::endl << offset;
          } else ofs << ' ';
        }
      }
      ofs << "];" << std::endl;
    }


  template<class F>
  void storeSparseMatrixToOctave(SparseMatrix<F>& matrix, std::ostream& ofs,
                                 std::string name = "")
  {
    if (name.empty()) {
      std::stringstream s;
      s << "MatSparse_" << storeSparseMatrixMatlabCounter_++;
      name = s.str();
    }
    std::string offset(name.size() + 4, ' ');
    ofs << name << " = zeros(" << matrix.nofRows() << ", " << matrix.nofCols()
        << ");" << std::endl;
    ofs << "zzz = [" << std::endl << std::setprecision(16);
    matrix.m()->outputMatlab(ofs);
    ofs << "];" << std::endl;
    ofs << "for i__=1:size(zzz,1)" << std::endl
        << "  " << name << "(zzz(i__,1), zzz(i__,2)) = zzz(i__,3);"<< std::endl
        << "end" << std::endl
        << "clear zzz" << std::endl;
  }

} // namespace concepts

#endif // OperatorOutputMatlab_hh