#if !defined(INCLUDED_STREAM_MEMSTREAM_H)
#define INCLUDED_STREAM_MEMSTREAM_H

#include "itextstream.h"
#include <algorithm>
#include <vector>

class BufferOutputStream : public TextOutputStream
{
  std::vector<char> m_buffer;
public:
  std::size_t write(const char* buffer, std::size_t length)
  {
    m_buffer.insert(m_buffer.end(), buffer, buffer+length);
    return length;
  }
  const char* data() const
  {
    return &(*m_buffer.begin());
  }
  std::size_t size() const
  {
    return m_buffer.size();
  }
  void clear()
  {
    std::vector<char> empty;
    std::swap(empty, m_buffer);
  }
};

template<typename T>
inline BufferOutputStream& operator<<(BufferOutputStream& ostream, const T& t)
{
  return ostream_write(ostream, t);
}


class BufferInputStream : public TextInputStream
{
  const char* m_read;
  const char* m_end;
public:
  BufferInputStream(const char* buffer, std::size_t length)
    : m_read(buffer), m_end(buffer + length)
  {
  }
  std::size_t read(char* buffer, std::size_t length)
  {
    std::size_t count = std::min(std::size_t(m_end - m_read), length);
    const char* end = m_read + count;
    while(m_read != end)
    {
      *buffer++ = *m_read++;
    }
    return count;
  }
};

#endif