ByteStreams.h

Go to the documentation of this file.

// ========================================================================= //
// This file is part of MyraMath, copyright (c) 2014-2019 by Ryan A Chilton  //
// and distributed by MyraCore, LLC. See LICENSE.txt for license terms.      //
// ========================================================================= //

#ifndef MYRAMATH_IO_BYTESTREAMS_H
#define MYRAMATH_IO_BYTESTREAMS_H

#include <myramath/io/Streams.h>
#include <stdexcept>

namespace myra {

class ByteOutputStream : public OutputStream
  {
  public:

    ByteOutputStream(char* in_buffer, size_t in_N)
      : buffer(in_buffer), N(in_N) { }

    virtual void write_binary (const char* p, size_t n)
      {
      if (n > N)
        throw std::runtime_error("ByteOutputStream::write_binary(), buffer overrun.");
      N -= n;
      while (n--)
        *buffer++ = *p++;
      }

    size_t remaining() const
      { return N; }
      
  private:

    // Pointer to underlying storage (write-only)
    char* buffer;
    size_t N;
    
  };

class ByteInputStream : public InputStream
  {
  public:

    ByteInputStream(const char* in_buffer, size_t in_N)
      : buffer(in_buffer), N(in_N) { }

    virtual void read_binary (char* p, size_t n)
      {
      if (n > N)
        throw std::runtime_error("ByteInputStream::read_binary(), buffer overrun.");
      N -= n;
      while (n--)
        *p++ = *buffer++;
      }

    size_t remaining() const
      { return N; }
    
  private:

    // Pointer to underlying storage (read-only)
    const char* buffer;
    size_t N;
    
  };

} // namespace

#endif