2 implementation of bitio.h.
4 Part of the swftools package.
6 Copyright (c) 2001 Matthias Kramm <kramm@quiss.org>
8 This file is distributed under the GPL, see file COPYING for details */
16 #include "../config.h"
20 #define ZLIB_BUFFER_SIZE 16384
24 /* ---------------------------- file reader ------------------------------- */
26 static int reader_fileread(struct reader_t*reader, void* data, int len)
28 int ret = read((int)reader->internal, data, len);
33 void reader_init_filereader(struct reader_t*r, int handle)
35 r->read = reader_fileread;
36 r->internal = (void*)handle;
37 r->type = READER_TYPE_FILE;
43 /* ---------------------------- mem reader ------------------------------- */
50 static int reader_memread(struct reader_t*reader, void* data, int len)
52 struct memread_t*mr = (struct memread_t*)reader->internal;
54 if(mr->length - reader->pos > len) {
55 memcpy(data, &mr->data[reader->pos], len);
59 memcpy(data, &mr->data[reader->pos], mr->length - reader->pos);
60 reader->pos = mr->length;
61 return mr->length - reader->pos;
64 void reader_init_memreader(struct reader_t*r, void*newdata, int newlength)
66 struct memread_t*mr = malloc(sizeof(struct memread_t));
68 mr->length = newlength;
69 r->read = reader_memread;
70 r->internal = (void*)mr;
71 r->type = READER_TYPE_MEM;
77 /* ---------------------------- mem writer ------------------------------- */
85 static int writer_memwrite_write(struct writer_t*w, void* data, int len)
87 struct memread_t*mw = (struct memread_t*)w->internal;
88 if(mw->length - w->pos > len) {
89 memcpy(&mw->data[w->pos], data, len);
93 memcpy(&mw->data[w->pos], data, mw->length - w->pos);
95 return mw->length - w->pos;
98 static void writer_memwrite_finish(struct writer_t*w)
102 void writer_init_memwriter(struct writer_t*w, void*data, int len)
104 struct memwrite_t *mr;
105 mr = malloc(sizeof(struct memwrite_t));
108 memset(w, 0, sizeof(struct writer_t));
109 w->write = writer_memwrite_write;
110 w->finish = writer_memwrite_finish;
111 w->internal = (void*)mr;
112 w->type = WRITER_TYPE_FILE;
118 /* ---------------------------- file writer ------------------------------- */
126 static int writer_filewrite_write(struct writer_t*w, void* data, int len)
128 struct filewrite_t * fw= (struct filewrite_t*)w->internal;
129 return write(fw->handle, data, len);
131 static void writer_filewrite_finish(struct writer_t*w)
133 struct filewrite_t *mr = (struct filewrite_t*)w->internal;
138 void writer_init_filewriter(struct writer_t*w, int handle)
140 struct filewrite_t *mr = malloc(sizeof(struct filewrite_t));
143 memset(w, 0, sizeof(struct writer_t));
144 w->write = writer_filewrite_write;
145 w->finish = writer_filewrite_finish;
147 w->type = WRITER_TYPE_FILE;
152 void writer_init_filewriter2(struct writer_t*w, char*filename)
154 int fi = open("movie.swf",
158 O_WRONLY|O_CREAT|O_TRUNC, 0644);
159 writer_init_filewriter(w, fi);
160 ((struct filewrite_t*)w->internal)->free_handle = 1;
163 /* ---------------------------- zlibinflate writer -------------------------- */
169 struct reader_t*input;
170 unsigned char readbuffer[ZLIB_BUFFER_SIZE];
175 static void zlib_error(int ret, char* msg, z_stream*zs)
177 fprintf(stderr, "%s: zlib error (%d): last zlib error: %s\n",
180 zs->msg?zs->msg:"unknown");
186 static int reader_zlibinflate(struct reader_t*reader, void* data, int len)
189 struct zlibinflate_t*z = (struct zlibinflate_t*)reader->internal;
192 fprintf(stderr, "zlib not initialized!\n");
198 z->zs.next_out = data;
199 z->zs.avail_out = len;
202 if(!z->zs.avail_in) {
203 z->zs.avail_in = z->input->read(z->input, z->readbuffer, ZLIB_BUFFER_SIZE);
204 z->zs.next_in = z->readbuffer;
207 ret = inflate(&z->zs, Z_NO_FLUSH);
209 ret = inflate(&z->zs, Z_FINISH);
212 ret != Z_STREAM_END) zlib_error(ret, "bitio:inflate_inflate", &z->zs);
214 if (ret == Z_STREAM_END) {
215 int pos = z->zs.next_out - (Bytef*)data;
216 ret = inflateEnd(&z->zs);
217 if (ret != Z_OK) zlib_error(ret, "bitio:inflate_end", &z->zs);
218 free(reader->internal);
219 reader->internal = 0;
223 if(!z->zs.avail_out) {
230 fprintf(stderr, "Error: swftools was compiled without zlib support");
234 void reader_init_zlibinflate(struct reader_t*r, struct reader_t*input)
237 struct zlibinflate_t*z;
239 memset(r, 0, sizeof(struct reader_t));
240 z = (struct zlibinflate_t*)malloc(sizeof(struct zlibinflate_t));
241 memset(z, 0, sizeof(struct zlibinflate_t));
243 r->read = reader_zlibinflate;
244 r->type = READER_TYPE_ZLIB;
247 memset(&z->zs,0,sizeof(z_stream));
248 z->zs.zalloc = Z_NULL;
249 z->zs.zfree = Z_NULL;
250 z->zs.opaque = Z_NULL;
251 ret = inflateInit(&z->zs);
252 if (ret != Z_OK) zlib_error(ret, "bitio:inflate_init", &z->zs);
255 fprintf(stderr, "Error: swftools was compiled without zlib support");
259 /* ---------------------------- zlibdeflate writer -------------------------- */
265 struct writer_t*output;
266 unsigned char writebuffer[ZLIB_BUFFER_SIZE];
270 static int writer_zlibdeflate_write(struct writer_t*writer, void* data, int len)
273 struct zlibdeflate_t*z = (struct zlibdeflate_t*)writer->internal;
276 fprintf(stderr, "zlib not initialized!\n");
282 z->zs.next_in = data;
283 z->zs.avail_in = len;
286 ret = deflate(&z->zs, Z_NO_FLUSH);
288 if (ret != Z_OK) zlib_error(ret, "bitio:deflate_deflate", &z->zs);
290 if(z->zs.next_out != z->writebuffer) {
291 z->output->write(z->output, z->writebuffer, z->zs.next_out - (Bytef*)z->writebuffer);
292 z->zs.next_out = z->writebuffer;
293 z->zs.avail_out = ZLIB_BUFFER_SIZE;
296 if(!z->zs.avail_in) {
303 fprintf(stderr, "Error: swftools was compiled without zlib support");
307 static void writer_zlibdeflate_finish(struct writer_t*writer)
310 struct zlibdeflate_t*z = (struct zlibdeflate_t*)writer->internal;
311 struct writer_t*output;
317 ret = deflate(&z->zs, Z_FINISH);
319 ret != Z_STREAM_END) zlib_error(ret, "bitio:deflate_finish", &z->zs);
321 if(z->zs.next_out != z->writebuffer) {
322 z->output->write(z->output, z->writebuffer, z->zs.next_out - (Bytef*)z->writebuffer);
323 z->zs.next_out = z->writebuffer;
324 z->zs.avail_out = ZLIB_BUFFER_SIZE;
327 if (ret == Z_STREAM_END) {
332 ret = deflateEnd(&z->zs);
333 if (ret != Z_OK) zlib_error(ret, "bitio:deflate_end", &z->zs);
334 free(writer->internal);
335 writer->internal = 0;
336 output->finish(output);
338 fprintf(stderr, "Error: swftools was compiled without zlib support");
342 void writer_init_zlibdeflate(struct writer_t*w, struct writer_t*output)
345 struct zlibdeflate_t*z;
347 memset(w, 0, sizeof(struct writer_t));
348 z = (struct zlibdeflate_t*)malloc(sizeof(struct zlibdeflate_t));
349 memset(z, 0, sizeof(struct zlibdeflate_t));
351 w->write = writer_zlibdeflate_write;
352 w->finish = writer_zlibdeflate_finish;
353 w->type = WRITER_TYPE_ZLIB;
356 memset(&z->zs,0,sizeof(z_stream));
357 z->zs.zalloc = Z_NULL;
358 z->zs.zfree = Z_NULL;
359 z->zs.opaque = Z_NULL;
360 ret = deflateInit(&z->zs, 9);
361 if (ret != Z_OK) zlib_error(ret, "bitio:deflate_init", &z->zs);
364 z->zs.next_out = z->writebuffer;
365 z->zs.avail_out = ZLIB_BUFFER_SIZE;
367 fprintf(stderr, "Error: swftools was compiled without zlib support");
372 /* ----------------------- bit handling routines -------------------------- */
374 void writer_writebit(struct writer_t*w, int bit)
378 w->write(w, &w->mybyte, 1);
383 w->mybyte |= 1 << (7 - w->bitpos);
386 void writer_writebits(struct writer_t*w, unsigned int data, int bits)
391 writer_writebit(w, (data >> (bits-t-1))&1);
394 void writer_resetbits(struct writer_t*w)
397 w->write(w, &w->mybyte, 1);
402 unsigned int reader_readbit(struct reader_t*r)
407 r->read(r, &r->mybyte, 1);
409 return (r->mybyte>>(7-r->bitpos++))&1;
411 unsigned int reader_readbits(struct reader_t*r, int num)
418 val|=reader_readbit(r);
422 void reader_resetbits(struct reader_t*r)