Merge commit '48410b113dd2036e69dbf723a39ec9af02fc9b12'
[xonotic/netradiant.git] / 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