BinaryStreamWriter.cpp

Go to the documentation of this file.

// Mantid Repository : https://github.com/mantidproject/mantid
//
// Copyright © 2019 ISIS Rutherford Appleton Laboratory UKRI,
//   NScD Oak Ridge National Laboratory, European Spallation Source,
//   Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS
// SPDX - License - Identifier: GPL - 3.0 +
//------------------------------------------------------------------------------
// Includes
//------------------------------------------------------------------------------
#include "MantidKernel/BinaryStreamWriter.h"
#include "MantidKernel/Matrix.h"
 
#include <cassert>
#include <fstream>
#include <stdexcept>
 
namespace Mantid::Kernel {
 
//------------------------------------------------------------------------------
// Anonymous functions
//------------------------------------------------------------------------------
namespace {
 
template <typename T> inline void writeToStream(std::ostream &stream, const T &value) {
  stream.write(reinterpret_cast<const char *>(&value), sizeof(T));
}
 
template <typename T> inline void writeToStream(std::ostream &stream, const std::vector<T> &value, size_t nvals) {
  stream.write(reinterpret_cast<const char *>(value.data()), nvals * sizeof(T));
}
} // namespace
 
//------------------------------------------------------------------------------
// Public members
//------------------------------------------------------------------------------
 
BinaryStreamWriter::BinaryStreamWriter(std::ostream &ofstrm)
    : m_ofstrm(ofstrm), m_strLengthSize(static_cast<uint64_t>(sizeof(int32_t))) {
  if (ofstrm.fail()) {
    throw std::runtime_error("BinaryStreamWriter: Output stream is in a bad "
                             "state. Cannot continue.");
  }
}
 
BinaryStreamWriter &BinaryStreamWriter::operator<<(const int16_t &value) {
  writeToStream(m_ofstrm, value);
  return *this;
}
 
BinaryStreamWriter &BinaryStreamWriter::operator<<(const int32_t &value) {
  writeToStream(m_ofstrm, value);
  return *this;
}
 
BinaryStreamWriter &BinaryStreamWriter::operator<<(const int64_t &value) {
  writeToStream(m_ofstrm, value);
  return *this;
}
 
BinaryStreamWriter &BinaryStreamWriter::operator<<(const float &value) {
  writeToStream(m_ofstrm, value);
  return *this;
}
 
BinaryStreamWriter &BinaryStreamWriter::operator<<(const double &value) {
  writeToStream(m_ofstrm, value);
  return *this;
}
 
BinaryStreamWriter &BinaryStreamWriter::operator<<(const std::string &value) {
  // First write the size.
  decltype(m_strLengthSize) length(value.length());
  m_ofstrm.write(reinterpret_cast<char *>(&length), m_strLengthSize);
  // Now the value
  m_ofstrm.write(const_cast<char *>(value.data()), static_cast<size_t>(length));
  return *this;
}
 
BinaryStreamWriter &BinaryStreamWriter::operator<<(const uint16_t &value) {
  writeToStream(m_ofstrm, value);
  return *this;
}
 
BinaryStreamWriter &BinaryStreamWriter::operator<<(const uint32_t &value) {
  writeToStream(m_ofstrm, value);
  return *this;
}
 
BinaryStreamWriter &BinaryStreamWriter::write(const std::vector<int16_t> &value, const size_t nvals) {
  writeToStream(m_ofstrm, value, nvals);
  return *this;
}
 
BinaryStreamWriter &BinaryStreamWriter::write(const std::vector<int32_t> &value, const size_t nvals) {
  writeToStream(m_ofstrm, value, nvals);
  return *this;
}
 
BinaryStreamWriter &BinaryStreamWriter::write(const std::vector<int64_t> &value, const size_t nvals) {
  writeToStream(m_ofstrm, value, nvals);
  return *this;
}
 
BinaryStreamWriter &BinaryStreamWriter::write(const std::vector<float> &value, const size_t nvals) {
  writeToStream(m_ofstrm, value, nvals);
  return *this;
}
 
BinaryStreamWriter &BinaryStreamWriter::write(const std::vector<double> &value, const size_t nvals) {
  writeToStream(m_ofstrm, value, nvals);
  return *this;
}
 
BinaryStreamWriter &BinaryStreamWriter::write(const std::string &value, const size_t length) {
  m_ofstrm.write(const_cast<char *>(value.data()), length);
  return *this;
}
 
} // namespace Mantid::Kernel