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         m_buffer.insert( m_buffer.end(), buffer, buffer + length );
  35         return length;
  36 }
  37 const char* data() const {
  38         return &( *m_buffer.begin() );
  39 }
  40 std::size_t size() const {
  41         return m_buffer.size();
  42 }
  43 void clear(){
  44         std::vector<char> empty;
  45         std::swap( empty, m_buffer );
  46 }
  47 };
  48
  49 template<typename T>
  50 inline BufferOutputStream& operator<<( BufferOutputStream& ostream, const T& t ){
  51         return ostream_write( ostream, t );
  52 }
  53
  54
  55 class BufferInputStream : public TextInputStream
  56 {
  57 const char* m_read;
  58 const char* m_end;
  59 public:
  60 BufferInputStream( const char* buffer, std::size_t length )
  61         : m_read( buffer ), m_end( buffer + length ){
  62 }
  63 std::size_t read( char* buffer, std::size_t length ){
  64         std::size_t count = std::min( std::size_t( m_end - m_read ), length );
  65         const char* end = m_read + count;
  66         while ( m_read != end )
  67         {
  68                 *buffer++ = *m_read++;
  69         }
  70         return count;
  71 }
  72 };
  73
  74 #endif