deflatestream.h
Go to the documentation of this file.
1 /*
2  * Copyright (C) 2005 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_DEFLATESTREAM_H
31 #define TNT_DEFLATESTREAM_H
32 
33 #include <iostream>
34 #include <stdexcept>
35 #include <zlib.h>
36 #include <vector>
37 
38 namespace tnt
39 {
40  class DeflateError : public std::runtime_error
41  {
42  int _zRet;
43 
44  public:
45  DeflateError(int zRet, const std::string& msg)
46  : std::runtime_error(msg),
47  _zRet(zRet)
48  { }
49 
50  int getRet() const { return _zRet; }
51  };
52 
53  class DeflateStreamBuf : public std::streambuf
54  {
55  z_stream _stream;
56  std::vector<char_type> _obuffer;
57  std::streambuf* _sink;
58 
59  public:
60  explicit DeflateStreamBuf(std::streambuf* sink_, int level = Z_DEFAULT_COMPRESSION, unsigned bufsize = 8192);
62 
64  int_type overflow(int_type c);
66  int_type underflow();
68  int sync();
69 
71  int end();
72  void reinitialize();
73  void setSink(std::streambuf* sink) { _sink = sink; }
74  uLong getAdler() const { return _stream.adler; }
75  };
76 
77  class DeflateStream : public std::ostream
78  {
79  DeflateStreamBuf _streambuf;
80 
81  public:
82  explicit DeflateStream(std::streambuf* sink, int level = Z_DEFAULT_COMPRESSION)
83  : std::ostream(0),
84  _streambuf(sink, level)
85  { init(&_streambuf); }
86  explicit DeflateStream(std::ostream& sink, int level = Z_DEFAULT_COMPRESSION)
87  : std::ostream(0),
88  _streambuf(sink.rdbuf(), level)
89  { init(&_streambuf); }
90 
91  void end();
92  void reinitialize()
93  { _streambuf.reinitialize(); }
94  void setSink(std::streambuf* sink) { _streambuf.setSink(sink); }
95  void setSink(std::ostream& sink) { _streambuf.setSink(sink.rdbuf()); }
96  uLong getAdler() const { return _streambuf.getAdler(); }
97  };
98 }
99 
100 #endif // TNT_DEFLATESTREAM_H
101