moved wxDash typedef to gdicmn.h
[wxWidgets.git] / src / common / zstream.cpp
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: zstream.cpp
3 // Purpose: Compressed stream classes
4 // Author: Guilhem Lavaux
5 // Modified by:
6 // Created: 11/07/98
7 // RCS-ID: $Id$
8 // Copyright: (c) Guilhem Lavaux
9 // Licence: wxWindows licence
10 /////////////////////////////////////////////////////////////////////////////
11
12 #ifdef __GNUG__
13 #pragma implementation "zstream.h"
14 #endif
15
16 // For compilers that support precompilation, includes "wx.h".
17 #include "wx/wxprec.h"
18
19 #ifdef __BORLANDC__
20 #pragma hdrstop
21 #endif
22
23 #if wxUSE_ZLIB && wxUSE_STREAMS
24
25 #include "wx/zstream.h"
26 #include "wx/utils.h"
27 #include "wx/intl.h"
28 #include "wx/log.h"
29
30 // When using configure, the path must be "zlib.h" I don't know
31 // what other ports (wxMac, wxMotif without configure) need here.
32 // If we are building with configure (defines __WX_SETUP_H__),
33 // we trust the zlib path is given as a -I option.
34 #if defined(__WXMSW__) && !defined(__WX_SETUP_H__)
35 #include "../zlib/zlib.h"
36 #else
37 #include "zlib.h"
38 #endif
39
40 #define ZSTREAM_BUFFER_SIZE 1024
41
42 //////////////////////
43 // wxZlibInputStream
44 //////////////////////
45
46 wxZlibInputStream::wxZlibInputStream(wxInputStream& stream)
47 : wxFilterInputStream(stream)
48 {
49 int err;
50
51 // I need a private stream buffer.
52 m_inflate = new z_stream_s;
53
54 m_inflate->zalloc = (alloc_func)0;
55 m_inflate->zfree = (free_func)0;
56 m_inflate->opaque = (voidpf)0;
57
58 err = inflateInit(m_inflate);
59 if (err != Z_OK) {
60 inflateEnd(m_inflate);
61 delete m_inflate;
62 return;
63 }
64
65 m_z_buffer = new unsigned char[ZSTREAM_BUFFER_SIZE];
66 m_z_size = ZSTREAM_BUFFER_SIZE;
67
68 m_inflate->avail_in = 0;
69 m_inflate->next_in = NULL;
70 }
71
72 wxZlibInputStream::~wxZlibInputStream()
73 {
74 inflateEnd(m_inflate);
75 delete m_inflate;
76 }
77
78 size_t wxZlibInputStream::OnSysRead(void *buffer, size_t size)
79 {
80 int err;
81
82 m_inflate->next_out = (unsigned char *)buffer;
83 m_inflate->avail_out = size;
84
85 while (m_inflate->avail_out > 0) {
86 if (m_inflate->avail_in == 0) {
87
88 m_parent_i_stream->Read(m_z_buffer, m_z_size);
89 m_inflate->next_in = m_z_buffer;
90 m_inflate->avail_in = m_parent_i_stream->LastRead();
91
92 if (m_parent_i_stream->LastError() != wxStream_NOERROR &&
93 m_parent_i_stream->LastError() != wxStream_EOF)
94 {
95 m_lasterror = m_parent_i_stream->LastError();
96 return 0; // failed to read anything
97 }
98 }
99 err = inflate(m_inflate, Z_FINISH);
100 if (err == Z_STREAM_END)
101 return (size - m_inflate->avail_out);
102 }
103
104 return size-m_inflate->avail_out;
105 }
106
107 //////////////////////
108 // wxZlibOutputStream
109 //////////////////////
110
111 wxZlibOutputStream::wxZlibOutputStream(wxOutputStream& stream, int level)
112 : wxFilterOutputStream(stream)
113 {
114 int err;
115
116 m_deflate = new z_stream_s;
117
118 m_deflate->zalloc = (alloc_func)0;
119 m_deflate->zfree = (free_func)0;
120 m_deflate->opaque = (voidpf)0;
121
122 if (level == -1) level = Z_DEFAULT_COMPRESSION;
123 wxASSERT_MSG(level >= 0 && level <= 9, wxT("wxZlibOutputStream compression level must be between 0 and 9!"));
124
125 err = deflateInit(m_deflate, level);
126 if (err != Z_OK) {
127 deflateEnd(m_deflate);
128 return;
129 }
130
131 m_z_buffer = new unsigned char[ZSTREAM_BUFFER_SIZE];
132 m_z_size = ZSTREAM_BUFFER_SIZE;
133
134 m_deflate->avail_in = 0;
135 m_deflate->next_out = m_z_buffer;
136 m_deflate->avail_out = m_z_size;
137 }
138
139 wxZlibOutputStream::~wxZlibOutputStream()
140 {
141 int err;
142
143 Sync();
144
145 err = deflate(m_deflate, Z_FINISH);
146 if (err != Z_STREAM_END)
147 {
148 wxLogDebug( wxT("wxZlibOutputStream: an error occured while closing the stream.\n") );
149 return;
150 }
151
152 deflateEnd(m_deflate);
153
154 delete[] m_z_buffer;
155 }
156
157 void wxZlibOutputStream::Sync()
158 {
159 int err;
160
161 m_parent_o_stream->Write(m_z_buffer, m_z_size-m_deflate->avail_out);
162 m_deflate->next_out = m_z_buffer;
163 m_deflate->avail_out = m_z_size;
164
165 err = deflate(m_deflate, Z_FULL_FLUSH);
166 if (err != Z_OK) {
167 return;
168 }
169
170 // Fixed by "Stefan Csomor" <csomor@advancedconcepts.ch>
171 while( m_deflate->avail_out == 0 )
172 {
173 m_parent_o_stream->Write(m_z_buffer, m_z_size );
174 m_deflate->next_out = m_z_buffer;
175 m_deflate->avail_out = m_z_size;
176 err = deflate(m_deflate, Z_FULL_FLUSH);
177 if (err != Z_OK) {
178 return;
179 }
180 }
181 // End
182
183 m_parent_o_stream->Write(m_z_buffer, m_z_size-m_deflate->avail_out);
184 m_deflate->next_out = m_z_buffer;
185 m_deflate->avail_out = m_z_size;
186 }
187
188 size_t wxZlibOutputStream::OnSysWrite(const void *buffer, size_t size)
189 {
190 int err;
191
192 m_deflate->next_in = (unsigned char *)buffer;
193 m_deflate->avail_in = size;
194
195 while (m_deflate->avail_in > 0) {
196
197 if (m_deflate->avail_out == 0) {
198 m_parent_o_stream->Write(m_z_buffer, m_z_size);
199 if (m_parent_o_stream->LastError() != wxStream_NOERROR)
200 return (size - m_deflate->avail_in);
201
202 m_deflate->next_out = m_z_buffer;
203 m_deflate->avail_out = m_z_size;
204 }
205
206 err = deflate(m_deflate, Z_NO_FLUSH);
207 if (err != Z_OK)
208 return (size - m_deflate->avail_in);
209 }
210 return size;
211 }
212
213 #endif
214 // wxUSE_ZLIB && wxUSE_STREAMS
215