matfile/matfileOutput.hh

Go to the documentation of this file.

#ifndef matfileoutput_hh
#define matfileoutput_hh

/*Matfile includes*/
#include "matfile_concepts/Output"
/*concepts includes*/
#include "basics.hh"
#include "sparseOWrapper.hh"
#include "denseOWrapper.hh"
#include "operator.hh"

#define Matfile_ending_test 0
using namespace matfile;

namespace concepts {

class MatfileOutput {

public:

  MatfileOutput() {
    _odevice = 0;
  }

  MatfileOutput(const std::string str) {
    //  DEBUGL(1,matfileEnding(str.c_str()));
    _odevice = new matfile::OutputDevice(matfileEnding(str.c_str()));
  }

  virtual ~MatfileOutput() {
    if (_odevice != 0)
      _odevice->close();
    delete (_odevice);
  }

  void openMatfile(const std::string str) {
    if (_odevice != 0)
      delete (_odevice);
    _odevice = new OutputDevice(matfileEnding(str.c_str()));
  }

  void closeMatfile() {
    delete (_odevice);
    _odevice = 0;
  }

  template<class T>
  void addSparse(concepts::SparseMatrix<T> *sparse, const std::string str) {
    if (_odevice == 0)
      _odevice = new OutputDevice(matfileEnding(str.c_str()));

    SparseOWrapper wrapper(*sparse);
//    if (wrapper.nnz() == 0) {
//      DenseMatrix<Real> zeroMat = 0;
//    } else
      _odevice->writeArray(str.c_str(), wrapper);
  }

  template<class T>
  void addDense(concepts::DenseMatrix<T> *dense, const std::string str) {
    if (_odevice == 0)
      _odevice = new OutputDevice(matfileEnding(str.c_str()));

    DenseOWrapper wrapper(*dense);
    _odevice->writeArray(str.c_str(), wrapper);
  }

private:

  std::string matfileEnding(const std::string& filename) {
    uint size = filename.size();
    if (size < 4 || std::string(filename, size - 4, 4) != ".mat")
      return (filename + ".mat");
    return filename;
  }

  matfile::OutputDevice *_odevice;

};/*class_end*/
}/*namespace_concepts_end*/
#endif