matfile/matfile_concepts/src/Devices/OutputDevice.hh

Go to the documentation of this file.

/*
    This file is part of matfile.

    Copyright (C) 2010-2011 Andrea Arteaga <arteagaa@student.ethz.ch>

    matfile is free software: you can redistribute it and/or modify
    it under the terms of the GNU Lesser General Public License as published
    by the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    matfile is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public License
    along with matfile.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef OUTPUTDEVICE_HPP_
#define OUTPUTDEVICE_HPP_

class OutputDevice
{
  //TODO: Implement a buffer and the compression of data with zlib.

public:

  OutputDevice(const std::string& filename)
    : fname_(filename), fs_(filename.c_str(), std::ofstream::out | std::ofstream::binary),
      fileIsOpen_(true), descriptionIsSet_(false)
  { }

  ~OutputDevice() {
    if(fileIsOpen_)
      fs_.close();
  }

  void close() {
    fs_.close();
    fileIsOpen_ = false;
  }

  template<typename charT>
  void setDescription(const std::basic_string<charT>& descr);

  template<class WrapperClass>
  void writeArray(const std::string& name, const WrapperClass& wrapper) {
    setDescription();
    detail::Writer<WrapperClass>::write(fs_, name, wrapper);
  }

private:
  std::string fname_;
  std::ofstream fs_;
  bool fileIsOpen_;
  bool descriptionIsSet_;

  inline void setDescription();
};


void OutputDevice::setDescription()
{
  if (descriptionIsSet_) return;

  std::string description = "MATfile++ MAT-file, Created on: ";
  
  time_t rawtime; 
  description.append(ctime(&rawtime));
  description.append(116 - description.length(), ' ');
  setDescription(description);
}

template<typename charT>
void OutputDevice::setDescription(const std::basic_string<charT>& descr)
{
  if (descriptionIsSet_) return;

  std::string description;

  if (descr.length() > 116)
    description = descr.substr(0, 116);
  else if (descr.length() < 116)
    (description = descr).append(116 - descr.length(), ' ');
  else
    description = descr;

  fs_.write(description.c_str(), 116);

  const char subsys[8] = {0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20 };
  fs_.write(subsys, 8);

  miINT16_t version = 0x0100;
  fs_.write(reinterpret_cast<char*>(&version), 2);

  miINT16_t endian = 19785;
  fs_.write(reinterpret_cast<char*>(&endian), 2);

  descriptionIsSet_ = true;
}

#endif /* OUTPUTDEVICE_HPP_ */