libs/stream/memstream.h

   1 /*
   2 Copyright (C) 2001-2006, William Joseph.
   3 All Rights Reserved.
   4
   5 This file is part of GtkRadiant.
   6
   7 GtkRadiant is free software; you can redistribute it and/or modify
   8 it under the terms of the GNU General Public License as published by
   9 the Free Software Foundation; either version 2 of the License, or
  10 (at your option) any later version.
  11
  12 GtkRadiant is distributed in the hope that it will be useful,
  13 but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 GNU General Public License for more details.
  16
  17 You should have received a copy of the GNU General Public License
  18 along with GtkRadiant; if not, write to the Free Software
  19 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  20 */
  21
  22 #if !defined(INCLUDED_STREAM_MEMSTREAM_H)
  23 #define INCLUDED_STREAM_MEMSTREAM_H
  24
  25 #include "itextstream.h"
  26 #include <algorithm>
  27 #include <vector>
  28
  29 class BufferOutputStream : public TextOutputStream
  30 {
  31   std::vector<char> m_buffer;
  32 public:
  33   std::size_t write(const char* buffer, std::size_t length)
  34   {
  35     m_buffer.insert(m_buffer.end(), buffer, buffer+length);
  36     return length;
  37   }
  38   const char* data() const
  39   {
  40     return &(*m_buffer.begin());
  41   }
  42   std::size_t size() const
  43   {
  44     return m_buffer.size();
  45   }
  46   void clear()
  47   {
  48     std::vector<char> empty;
  49     std::swap(empty, m_buffer);
  50   }
  51 };
  52
  53 template<typename T>
  54 inline BufferOutputStream& operator<<(BufferOutputStream& ostream, const T& t)
  55 {
  56   return ostream_write(ostream, t);
  57 }
  58
  59
  60 class BufferInputStream : public TextInputStream
  61 {
  62   const char* m_read;
  63   const char* m_end;
  64 public:
  65   BufferInputStream(const char* buffer, std::size_t length)
  66     : m_read(buffer), m_end(buffer + length)
  67   {
  68   }
  69   std::size_t read(char* buffer, std::size_t length)
  70   {
  71     std::size_t count = std::min(std::size_t(m_end - m_read), length);
  72     const char* end = m_read + count;
  73     while(m_read != end)
  74     {
  75       *buffer++ = *m_read++;
  76     }
  77     return count;
  78   }
  79 };
  80
  81 #endif