3 implementation of bitio.h.
5 Copyright (C) 2003 Matthias Kramm <kramm@quiss.org>
7 This program 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.
12 This program 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.
17 You should have received a copy of the GNU General Public License
18 along with this program; if not, write to the Free Software
19 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */
33 #include "../config.h"
37 #define ZLIB_BUFFER_SIZE 16384
41 /* ---------------------------- null reader ------------------------------- */
43 static int reader_nullread(reader_t*r, void* data, int len)
48 static void reader_nullread_dealloc(reader_t*r)
50 memset(r, 0, sizeof(reader_t));
52 void reader_init_nullreader(reader_t*r)
54 r->read = reader_nullread;
55 r->dealloc = reader_nullread_dealloc;
57 r->type = READER_TYPE_NULL;
62 /* ---------------------------- file reader ------------------------------- */
64 static int reader_fileread(reader_t*reader, void* data, int len)
66 int ret = read((int)reader->internal, data, len);
71 static void reader_fileread_dealloc(reader_t*r)
73 memset(r, 0, sizeof(reader_t));
75 void reader_init_filereader(reader_t*r, int handle)
77 r->read = reader_fileread;
78 r->dealloc = reader_fileread_dealloc;
79 r->internal = (void*)handle;
80 r->type = READER_TYPE_FILE;
86 /* ---------------------------- mem reader ------------------------------- */
88 typedef struct _memread
94 static int reader_memread(reader_t*reader, void* data, int len)
96 memread_t*mr = (memread_t*)reader->internal;
98 if(mr->length - reader->pos < len) {
99 len = mr->length - reader->pos;
101 memcpy(data, &mr->data[reader->pos], len);
105 static void reader_memread_dealloc(reader_t*reader)
108 free(reader->internal);
109 memset(reader, 0, sizeof(reader_t));
111 void reader_init_memreader(reader_t*r, void*newdata, int newlength)
113 memread_t*mr = (memread_t*)malloc(sizeof(memread_t));
114 mr->data = (unsigned char*)newdata;
115 mr->length = newlength;
116 r->read = reader_memread;
117 r->dealloc = reader_memread_dealloc;
118 r->internal = (void*)mr;
119 r->type = READER_TYPE_MEM;
125 /* ---------------------------- mem writer ------------------------------- */
127 typedef struct _memwrite
133 static int writer_memwrite_write(writer_t*w, void* data, int len)
135 memwrite_t*mw = (memwrite_t*)w->internal;
136 if(mw->length - w->pos > len) {
137 memcpy(&mw->data[w->pos], data, len);
141 memcpy(&mw->data[w->pos], data, mw->length - w->pos);
143 return mw->length - w->pos;
146 static void writer_memwrite_finish(writer_t*w)
152 static void dummy_flush(writer_t*w)
155 void writer_init_memwriter(writer_t*w, void*data, int len)
158 mr = (memwrite_t*)malloc(sizeof(memwrite_t));
159 mr->data = (unsigned char *)data;
161 memset(w, 0, sizeof(writer_t));
162 w->write = writer_memwrite_write;
163 w->flush = dummy_flush;
164 w->finish = writer_memwrite_finish;
165 w->internal = (void*)mr;
166 w->type = WRITER_TYPE_MEM;
172 /* ------------------------- growing mem writer ------------------------------- */
174 typedef struct _growmemwrite
180 static int writer_growmemwrite_write(writer_t*w, void* data, int len)
182 growmemwrite_t*mw = (growmemwrite_t*)w->internal;
184 fprintf(stderr, "Illegal write operation: data already given away");
187 if(mw->length - w->pos < len) {
188 int newlength = mw->length;
189 while(newlength - w->pos < len) {
190 newlength += mw->grow;
193 unsigned char*newmem = (unsigned char*)malloc(newlength);
194 memcpy(newmem, mw->data, mw->length);
198 mw->data = (unsigned char*)realloc(mw->data, newlength);
200 mw->length = newlength;
202 memcpy(&mw->data[w->pos], data, len);
206 static void writer_growmemwrite_finish(writer_t*w)
208 growmemwrite_t*mw = (growmemwrite_t*)w->internal;
210 free(mw->data);mw->data = 0;
213 free(w->internal);mw=0;
214 memset(w, 0, sizeof(writer_t));
216 void* writer_growmemwrite_memptr(writer_t*w, int*len)
218 growmemwrite_t*mw = (growmemwrite_t*)w->internal;
224 void* writer_growmemwrite_getmem(writer_t*w)
226 growmemwrite_t*mw = (growmemwrite_t*)w->internal;
228 /* remove own reference so that neither write() nor finish() can free it.
229 It's property of the caller now.
234 void writer_growmemwrite_reset(writer_t*w)
236 growmemwrite_t*mw = (growmemwrite_t*)w->internal;
241 void writer_init_growingmemwriter(writer_t*w, U32 grow)
243 growmemwrite_t *mw = (growmemwrite_t *)malloc(sizeof(growmemwrite_t));
245 mw->data = (unsigned char *)malloc(mw->length);
247 memset(w, 0, sizeof(writer_t));
248 w->write = writer_growmemwrite_write;
249 w->flush = dummy_flush;
250 w->finish = writer_growmemwrite_finish;
251 w->internal = (void*)mw;
252 w->type = WRITER_TYPE_GROWING_MEM;
258 /* ---------------------------- file writer ------------------------------- */
260 typedef struct _filewrite
266 static int writer_filewrite_write(writer_t*w, void* data, int len)
268 filewrite_t * fw= (filewrite_t*)w->internal;
270 return write(fw->handle, data, len);
272 static void writer_filewrite_finish(writer_t*w)
274 filewrite_t *mr = (filewrite_t*)w->internal;
278 memset(w, 0, sizeof(writer_t));
280 void writer_init_filewriter(writer_t*w, int handle)
282 filewrite_t *mr = (filewrite_t *)malloc(sizeof(filewrite_t));
285 memset(w, 0, sizeof(writer_t));
286 w->write = writer_filewrite_write;
287 w->finish = writer_filewrite_finish;
289 w->type = WRITER_TYPE_FILE;
294 void writer_init_filewriter2(writer_t*w, char*filename)
296 int fi = open("movie.swf",
300 O_WRONLY|O_CREAT|O_TRUNC, 0644);
301 writer_init_filewriter(w, fi);
302 ((filewrite_t*)w->internal)->free_handle = 1;
305 /* ---------------------------- null writer ------------------------------- */
307 static int writer_nullwrite_write(writer_t*w, void* data, int len)
312 static void writer_nullwrite_finish(writer_t*w)
314 memset(w, 0, sizeof(writer_t));
316 void writer_init_nullwriter(writer_t*w)
318 memset(w, 0, sizeof(writer_t));
319 w->write = writer_nullwrite_write;
320 w->flush = dummy_flush;
321 w->finish = writer_nullwrite_finish;
323 w->type = WRITER_TYPE_NULL;
328 /* ---------------------------- zlibinflate reader -------------------------- */
330 typedef struct _zlibinflate
335 unsigned char readbuffer[ZLIB_BUFFER_SIZE];
340 static void zlib_error(int ret, char* msg, z_stream*zs)
342 fprintf(stderr, "%s: zlib error (%d): last zlib error: %s\n",
345 zs->msg?zs->msg:"unknown");
351 static int reader_zlibinflate(reader_t*reader, void* data, int len)
354 zlibinflate_t*z = (zlibinflate_t*)reader->internal;
362 z->zs.next_out = (Bytef *)data;
363 z->zs.avail_out = len;
366 if(!z->zs.avail_in) {
367 z->zs.avail_in = z->input->read(z->input, z->readbuffer, ZLIB_BUFFER_SIZE);
368 z->zs.next_in = z->readbuffer;
371 ret = inflate(&z->zs, Z_NO_FLUSH);
373 ret = inflate(&z->zs, Z_FINISH);
376 ret != Z_STREAM_END) zlib_error(ret, "bitio:inflate_inflate", &z->zs);
378 if (ret == Z_STREAM_END) {
379 int pos = z->zs.next_out - (Bytef*)data;
380 ret = inflateEnd(&z->zs);
381 if (ret != Z_OK) zlib_error(ret, "bitio:inflate_end", &z->zs);
382 free(reader->internal);
383 reader->internal = 0;
387 if(!z->zs.avail_out) {
394 fprintf(stderr, "Error: swftools was compiled without zlib support");
398 static void reader_zlibinflate_dealloc(reader_t*reader)
401 zlibinflate_t*z = (zlibinflate_t*)reader->internal;
402 /* test whether read() already did basic deallocation */
403 if(reader->internal) {
405 free(reader->internal);
407 memset(reader, 0, sizeof(reader_t));
410 void reader_init_zlibinflate(reader_t*r, reader_t*input)
413 zlibinflate_t*z = (zlibinflate_t*)malloc(sizeof(zlibinflate_t));
414 memset(z, 0, sizeof(zlibinflate_t));
416 memset(r, 0, sizeof(reader_t));
418 r->read = reader_zlibinflate;
419 r->dealloc = reader_zlibinflate_dealloc;
420 r->type = READER_TYPE_ZLIB;
423 memset(&z->zs,0,sizeof(z_stream));
424 z->zs.zalloc = Z_NULL;
425 z->zs.zfree = Z_NULL;
426 z->zs.opaque = Z_NULL;
427 ret = inflateInit(&z->zs);
428 if (ret != Z_OK) zlib_error(ret, "bitio:inflate_init", &z->zs);
431 fprintf(stderr, "Error: swftools was compiled without zlib support");
436 /* ---------------------------- zlibdeflate writer -------------------------- */
438 typedef struct _zlibdeflate
443 unsigned char writebuffer[ZLIB_BUFFER_SIZE];
447 static int writer_zlibdeflate_write(writer_t*writer, void* data, int len)
450 zlibdeflate_t*z = (zlibdeflate_t*)writer->internal;
452 if(writer->type != WRITER_TYPE_ZLIB) {
453 fprintf(stderr, "Wrong writer ID (writer not initialized?)\n");
457 fprintf(stderr, "zlib not initialized!\n");
463 z->zs.next_in = (Bytef *)data;
464 z->zs.avail_in = len;
467 ret = deflate(&z->zs, Z_NO_FLUSH);
469 if (ret != Z_OK) zlib_error(ret, "bitio:deflate_deflate", &z->zs);
471 if(z->zs.next_out != z->writebuffer) {
472 writer->pos += z->zs.next_out - (Bytef*)z->writebuffer;
473 z->output->write(z->output, z->writebuffer, z->zs.next_out - (Bytef*)z->writebuffer);
474 z->zs.next_out = z->writebuffer;
475 z->zs.avail_out = ZLIB_BUFFER_SIZE;
478 if(!z->zs.avail_in) {
484 fprintf(stderr, "Error: swftools was compiled without zlib support");
489 void writer_zlibdeflate_flush(writer_t*writer)
492 zlibdeflate_t*z = (zlibdeflate_t*)writer->internal;
494 if(writer->type != WRITER_TYPE_ZLIB) {
495 fprintf(stderr, "Wrong writer ID (writer not initialized?)\n");
499 fprintf(stderr, "zlib not initialized!\n");
506 ret = deflate(&z->zs, Z_SYNC_FLUSH);
507 if (ret != Z_OK) zlib_error(ret, "bitio:deflate_flush", &z->zs);
508 if(z->zs.next_out != z->writebuffer) {
509 writer->pos += z->zs.next_out - (Bytef*)z->writebuffer;
510 z->output->write(z->output, z->writebuffer, z->zs.next_out - (Bytef*)z->writebuffer);
511 z->zs.next_out = z->writebuffer;
512 z->zs.avail_out = ZLIB_BUFFER_SIZE;
514 /* TODO: how will zlib let us know it needs more buffer space? */
519 fprintf(stderr, "Error: swftools was compiled without zlib support");
524 static void writer_zlibdeflate_finish(writer_t*writer)
527 zlibdeflate_t*z = (zlibdeflate_t*)writer->internal;
530 if(writer->type != WRITER_TYPE_ZLIB) {
531 fprintf(stderr, "Wrong writer ID (writer not initialized?)\n");
538 ret = deflate(&z->zs, Z_FINISH);
540 ret != Z_STREAM_END) zlib_error(ret, "bitio:deflate_finish", &z->zs);
542 if(z->zs.next_out != z->writebuffer) {
543 writer->pos += z->zs.next_out - (Bytef*)z->writebuffer;
544 z->output->write(z->output, z->writebuffer, z->zs.next_out - (Bytef*)z->writebuffer);
545 z->zs.next_out = z->writebuffer;
546 z->zs.avail_out = ZLIB_BUFFER_SIZE;
549 if (ret == Z_STREAM_END) {
554 ret = deflateEnd(&z->zs);
555 if (ret != Z_OK) zlib_error(ret, "bitio:deflate_end", &z->zs);
556 free(writer->internal);
557 memset(writer, 0, sizeof(writer_t));
558 //output->finish(output);
560 fprintf(stderr, "Error: swftools was compiled without zlib support");
564 void writer_init_zlibdeflate(writer_t*w, writer_t*output)
569 memset(w, 0, sizeof(writer_t));
570 z = (zlibdeflate_t*)malloc(sizeof(zlibdeflate_t));
571 memset(z, 0, sizeof(zlibdeflate_t));
573 w->write = writer_zlibdeflate_write;
574 w->flush = writer_zlibdeflate_flush;
575 w->finish = writer_zlibdeflate_finish;
576 w->type = WRITER_TYPE_ZLIB;
579 memset(&z->zs,0,sizeof(z_stream));
580 z->zs.zalloc = Z_NULL;
581 z->zs.zfree = Z_NULL;
582 z->zs.opaque = Z_NULL;
583 ret = deflateInit(&z->zs, 9);
584 if (ret != Z_OK) zlib_error(ret, "bitio:deflate_init", &z->zs);
587 z->zs.next_out = z->writebuffer;
588 z->zs.avail_out = ZLIB_BUFFER_SIZE;
590 fprintf(stderr, "Error: swftools was compiled without zlib support");
595 /* ----------------------- bit handling routines -------------------------- */
597 void writer_writebit(writer_t*w, int bit)
601 w->write(w, &w->mybyte, 1);
606 w->mybyte |= 1 << (7 - w->bitpos);
609 void writer_writebits(writer_t*w, unsigned int data, int bits)
614 writer_writebit(w, (data >> (bits-t-1))&1);
617 void writer_resetbits(writer_t*w)
620 w->write(w, &w->mybyte, 1);
625 unsigned int reader_readbit(reader_t*r)
630 r->read(r, &r->mybyte, 1);
632 return (r->mybyte>>(7-r->bitpos++))&1;
634 unsigned int reader_readbits(reader_t*r, int num)
641 val|=reader_readbit(r);
645 void reader_resetbits(reader_t*r)
652 U8 reader_readU8(reader_t*r)
655 if(r->read(r, &b, 1)<1) {
656 fprintf(stderr, "bitio.c:reader_readU8: Read over end of memory region\n");
660 U16 reader_readU16(reader_t*r)
663 if(r->read(r, &b1, 1)<1) {
664 fprintf(stderr, "bitio.c:reader_readU16: Read over end of memory region\n");
666 if(r->read(r, &b2, 1)<1) {
667 fprintf(stderr, "bitio.c:reader_readU16: Read over end of memory region\n");
671 U32 reader_readU32(reader_t*r)
673 U8 b1=0,b2=0,b3=0,b4=0;
674 if(r->read(r, &b1, 1)<1)
675 fprintf(stderr, "bitio.c:reader_readU32: Read over end of memory region\n");
676 if(r->read(r, &b2, 1)<1)
677 fprintf(stderr, "bitio.c:reader_readU32: Read over end of memory region\n");
678 if(r->read(r, &b3, 1)<1)
679 fprintf(stderr, "bitio.c:reader_readU32: Read over end of memory region\n");
680 if(r->read(r, &b4, 1)<1)
681 fprintf(stderr, "bitio.c:reader_readU32: Read over end of memory region\n");
682 return b1|b2<<8|b3<<16|b4<<24;
684 float reader_readFloat(reader_t*r)
690 U8 b1=0,b2=0,b3=0,b4=0;
695 U32 w = (b1|b2<<8|b3<<16|b4<<24);
698 double reader_readDouble(reader_t*r)
706 U64 w = ((U64)b[0]|(U64)b[1]<<8|(U64)b[2]<<16|(U64)b[3]<<24|(U64)b[4]<<32|(U64)b[5]<<40|(U64)b[6]<<48|(U64)b[7]<<56);
709 char*reader_readString(reader_t*r)
712 writer_init_growingmemwriter(&g, 16);
714 U8 b = reader_readU8(r);
715 writer_writeU8(&g, b);
719 char*string = (char*)writer_growmemwrite_getmem(&g);
724 void writer_writeString(writer_t*w, const char*s)
728 w->write(w, (void*)s, l);
729 w->write(w, &zero, 1);
731 void writer_writeU8(writer_t*w, unsigned char b)
735 void writer_writeU16(writer_t*w, unsigned short v)
737 unsigned char b1 = v;
738 unsigned char b2 = v>>8;
742 void writer_writeU32(writer_t*w, unsigned long v)
744 unsigned char b1 = v;
745 unsigned char b2 = v>>8;
746 unsigned char b3 = v>>16;
747 unsigned char b4 = v>>24;
753 void writer_writeFloat(writer_t*w, float f)
758 unsigned v = *(unsigned*)&f;
759 unsigned char b1 = v;
760 unsigned char b2 = v>>8;
761 unsigned char b3 = v>>16;
762 unsigned char b4 = v>>24;
768 void writer_writeDouble(writer_t*w, double f)
773 unsigned long long v = *(unsigned long long*)&f;
774 unsigned char b1 = v;
775 unsigned char b2 = v>>8;
776 unsigned char b3 = v>>16;
777 unsigned char b4 = v>>24;
778 unsigned char b5 = v>>32;
779 unsigned char b6 = v>>40;
780 unsigned char b7 = v>>48;
781 unsigned char b8 = v>>56;