VectorStream.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_VECTORSTREAM_H
#define MYRAMATH_IO_VECTORSTREAM_H

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

#include <vector>

namespace myra {

class VectorStream : public InputStream, public OutputStream
  {
  public:

    VectorStream()
      : i(0) {}

    virtual void write_binary (const char* p, size_t n)
      { while(n--) buffer.push_back(*p++); }

    virtual void read_binary (char* p, size_t n)
      { while(n--) *p++ = buffer.at(i++); }

    void swap(std::vector<char>& that)
      { that.swap(buffer); }
    
  private:

    // Internal buffer, populated via write().
    std::vector<char> buffer;

    // Internal pointer, advanced via read().
    size_t i;

  };

} // namespace

#endif