chunkedostream.h
Go to the documentation of this file.
1 /*
2  * Copyright (C) 2014 Tommi Maekitalo
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * As a special exception, you may use this file as part of a free
10  * software library without restriction. Specifically, if other files
11  * instantiate templates or use macros or inline functions from this
12  * file, or you compile this file and link it with other files to
13  * produce an executable, this file does not by itself cause the
14  * resulting executable to be covered by the GNU General Public
15  * License. This exception does not however invalidate any other
16  * reasons why the executable file might be covered by the GNU Library
17  * General Public License.
18  *
19  * This library is distributed in the hope that it will be useful,
20  * but WITHOUT ANY WARRANTY; without even the implied warranty of
21  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
22  * Lesser General Public License for more details.
23  *
24  * You should have received a copy of the GNU Lesser General Public
25  * License along with this library; if not, write to the Free Software
26  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
27  */
28 
29 
30 #ifndef TNT_CHUNKEDOSTREAM_H
31 #define TNT_CHUNKEDOSTREAM_H
32 
33 #include <iostream>
34 
35 namespace tnt
36 {
37  class ChunkedWriter : public std::streambuf
38  {
39  std::streambuf* _obuf;
40  char* _buffer;
41  unsigned _bufsize;
42  unsigned _bytesWritten;
43 
44  public:
45  explicit ChunkedWriter(std::streambuf* obuf, unsigned bufsize = 8192)
46  : _obuf(obuf),
47  _buffer(0),
48  _bufsize(bufsize)
49  { }
50 
52  { delete _buffer; }
53 
54  virtual int sync();
55  virtual int_type overflow(int_type ch);
56  virtual int_type underflow();
57 
58  void finish();
59 
60  void setSink(std::streambuf* obuf)
61  { _obuf = obuf; _bytesWritten = 0; }
62 
63  unsigned bytesWritten() const
64  { return _bytesWritten; }
65  };
66 
67  class ChunkedOStream : public std::ostream
68  {
69  ChunkedWriter _streambuf;
70 
71  public:
72  explicit ChunkedOStream(std::ostream& sink)
73  : std::ostream(0),
74  _streambuf(sink.rdbuf())
75  { std::ostream::init(&_streambuf); }
76 
77  explicit ChunkedOStream(std::streambuf* obuf)
78  : std::ostream(0),
79  _streambuf(obuf)
80  { std::ostream::init(&_streambuf); }
81 
82  void finish()
83  { _streambuf.finish(); }
84 
85  void setSink(std::ostream& sink)
86  { _streambuf.setSink(sink.rdbuf()); }
87 
88  void setSink(std::streambuf* sink)
89  { _streambuf.setSink(sink); }
90 
91  unsigned bytesWritten() const
92  { return _streambuf.bytesWritten(); }
93  };
94 }
95 
96 #endif // TNT_CHUNKEDOSTREAM_H
97