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 void reader_init_filereader(reader_t*r, int handle)
73 r->read = reader_fileread;
74 r->internal = (void*)handle;
75 r->type = READER_TYPE_FILE;
81 /* ---------------------------- mem reader ------------------------------- */
88 static int reader_memread(reader_t*reader, void* data, int len)
90 struct memread_t*mr = (struct memread_t*)reader->internal;
92 if(mr->length - reader->pos > len) {
93 memcpy(data, &mr->data[reader->pos], len);
97 memcpy(data, &mr->data[reader->pos], mr->length - reader->pos);
98 reader->pos += mr->length;
99 return mr->length - reader->pos;
102 static void reader_memread_dealloc(reader_t*reader)
105 free(reader->internal);
106 memset(reader, 0, sizeof(reader_t));
108 void reader_init_memreader(reader_t*r, void*newdata, int newlength)
110 struct memread_t*mr = (struct memread_t*)malloc(sizeof(struct memread_t));
111 mr->data = (unsigned char*)newdata;
112 mr->length = newlength;
113 r->read = reader_memread;
114 r->dealloc = reader_memread_dealloc;
115 r->internal = (void*)mr;
116 r->type = READER_TYPE_MEM;
122 /* ---------------------------- mem writer ------------------------------- */
130 static int writer_memwrite_write(writer_t*w, void* data, int len)
132 struct memwrite_t*mw = (struct memwrite_t*)w->internal;
133 if(mw->length - w->pos > len) {
134 memcpy(&mw->data[w->pos], data, len);
138 memcpy(&mw->data[w->pos], data, mw->length - w->pos);
140 return mw->length - w->pos;
143 static void writer_memwrite_finish(writer_t*w)
149 void writer_init_memwriter(writer_t*w, void*data, int len)
151 struct memwrite_t *mr;
152 mr = (struct memwrite_t*)malloc(sizeof(struct memwrite_t));
153 mr->data = (unsigned char *)data;
155 memset(w, 0, sizeof(writer_t));
156 w->write = writer_memwrite_write;
157 w->finish = writer_memwrite_finish;
158 w->internal = (void*)mr;
159 w->type = WRITER_TYPE_MEM;
165 /* ------------------------- growing mem writer ------------------------------- */
167 struct growmemwrite_t
173 static int writer_growmemwrite_write(writer_t*w, void* data, int len)
175 struct growmemwrite_t*mw = (struct growmemwrite_t*)w->internal;
177 fprintf(stderr, "Illegal write operation: data already given away");
180 if(mw->length - w->pos < len) {
181 int newlength = mw->length;
182 while(newlength - w->pos < len) {
183 newlength += mw->grow;
186 unsigned char*newmem = (unsigned char*)malloc(newlength);
187 memcpy(newmem, mw->data, mw->length);
191 mw->data = (unsigned char*)realloc(mw->data, newlength);
193 mw->length = newlength;
195 memcpy(&mw->data[w->pos], data, len);
199 static void writer_growmemwrite_finish(writer_t*w)
201 struct growmemwrite_t*mw = (struct growmemwrite_t*)w->internal;
206 free(w->internal);mw=0;
207 memset(w, 0, sizeof(writer_t));
209 void* writer_growmemwrite_memptr(writer_t*w, int*len)
211 struct growmemwrite_t*mw = (struct growmemwrite_t*)w->internal;
217 void* writer_growmemwrite_getmem(writer_t*w)
219 struct growmemwrite_t*mw = (struct growmemwrite_t*)w->internal;
221 /* remove own reference so that neither write() nor finish() can free it.
222 It's property of the caller now.
227 void writer_growmemwrite_reset(writer_t*w)
229 struct growmemwrite_t*mw = (struct growmemwrite_t*)w->internal;
234 void writer_init_growingmemwriter(writer_t*w, U32 grow)
236 struct growmemwrite_t *mr;
237 mr = (struct growmemwrite_t *)malloc(sizeof(struct growmemwrite_t));
239 mr->data = (unsigned char *)malloc(mr->length);
241 memset(w, 0, sizeof(writer_t));
242 w->write = writer_growmemwrite_write;
243 w->finish = writer_growmemwrite_finish;
244 w->internal = (void*)mr;
245 w->type = WRITER_TYPE_GROWING_MEM;
251 /* ---------------------------- file writer ------------------------------- */
259 static int writer_filewrite_write(writer_t*w, void* data, int len)
261 struct filewrite_t * fw= (struct filewrite_t*)w->internal;
262 return write(fw->handle, data, len);
264 static void writer_filewrite_finish(writer_t*w)
266 struct filewrite_t *mr = (struct filewrite_t*)w->internal;
270 memset(w, 0, sizeof(writer_t));
272 void writer_init_filewriter(writer_t*w, int handle)
274 struct filewrite_t *mr = (struct filewrite_t *)malloc(sizeof(struct filewrite_t));
277 memset(w, 0, sizeof(writer_t));
278 w->write = writer_filewrite_write;
279 w->finish = writer_filewrite_finish;
281 w->type = WRITER_TYPE_FILE;
286 void writer_init_filewriter2(writer_t*w, char*filename)
288 int fi = open("movie.swf",
292 O_WRONLY|O_CREAT|O_TRUNC, 0644);
293 writer_init_filewriter(w, fi);
294 ((struct filewrite_t*)w->internal)->free_handle = 1;
297 /* ---------------------------- null writer ------------------------------- */
299 static int writer_nullwrite_write(writer_t*w, void* data, int len)
304 static void writer_nullwrite_finish(writer_t*w)
306 memset(w, 0, sizeof(writer_t));
308 void writer_init_nullwriter(writer_t*w)
310 memset(w, 0, sizeof(writer_t));
311 w->write = writer_nullwrite_write;
312 w->finish = writer_nullwrite_finish;
314 w->type = WRITER_TYPE_NULL;
319 /* ---------------------------- zlibinflate reader -------------------------- */
326 unsigned char readbuffer[ZLIB_BUFFER_SIZE];
331 static void zlib_error(int ret, char* msg, z_stream*zs)
333 fprintf(stderr, "%s: zlib error (%d): last zlib error: %s\n",
336 zs->msg?zs->msg:"unknown");
342 static int reader_zlibinflate(reader_t*reader, void* data, int len)
345 struct zlibinflate_t*z = (struct zlibinflate_t*)reader->internal;
353 z->zs.next_out = (Bytef *)data;
354 z->zs.avail_out = len;
357 if(!z->zs.avail_in) {
358 z->zs.avail_in = z->input->read(z->input, z->readbuffer, ZLIB_BUFFER_SIZE);
359 z->zs.next_in = z->readbuffer;
362 ret = inflate(&z->zs, Z_NO_FLUSH);
364 ret = inflate(&z->zs, Z_FINISH);
367 ret != Z_STREAM_END) zlib_error(ret, "bitio:inflate_inflate", &z->zs);
369 if (ret == Z_STREAM_END) {
370 int pos = z->zs.next_out - (Bytef*)data;
371 ret = inflateEnd(&z->zs);
372 if (ret != Z_OK) zlib_error(ret, "bitio:inflate_end", &z->zs);
373 free(reader->internal);
374 reader->internal = 0;
378 if(!z->zs.avail_out) {
385 fprintf(stderr, "Error: swftools was compiled without zlib support");
389 static void reader_zlibinflate_dealloc(reader_t*reader)
392 struct zlibinflate_t*z = (struct zlibinflate_t*)reader->internal;
393 /* test whether read() already did basic deallocation */
394 if(reader->internal) {
396 free(reader->internal);
398 memset(reader, 0, sizeof(reader_t));
401 void reader_init_zlibinflate(reader_t*r, reader_t*input)
404 struct zlibinflate_t*z;
406 memset(r, 0, sizeof(reader_t));
407 z = (struct zlibinflate_t*)malloc(sizeof(struct zlibinflate_t));
408 memset(z, 0, sizeof(struct zlibinflate_t));
410 r->read = reader_zlibinflate;
411 r->dealloc = reader_zlibinflate_dealloc;
412 r->type = READER_TYPE_ZLIB;
415 memset(&z->zs,0,sizeof(z_stream));
416 z->zs.zalloc = Z_NULL;
417 z->zs.zfree = Z_NULL;
418 z->zs.opaque = Z_NULL;
419 ret = inflateInit(&z->zs);
420 if (ret != Z_OK) zlib_error(ret, "bitio:inflate_init", &z->zs);
423 fprintf(stderr, "Error: swftools was compiled without zlib support");
428 /* ---------------------------- zlibdeflate writer -------------------------- */
435 unsigned char writebuffer[ZLIB_BUFFER_SIZE];
439 static int writer_zlibdeflate_write(writer_t*writer, void* data, int len)
442 struct zlibdeflate_t*z = (struct zlibdeflate_t*)writer->internal;
444 if(writer->type != WRITER_TYPE_ZLIB) {
445 fprintf(stderr, "Wrong writer ID (writer not initialized?)\n");
449 fprintf(stderr, "zlib not initialized!\n");
455 z->zs.next_in = (Bytef *)data;
456 z->zs.avail_in = len;
459 ret = deflate(&z->zs, Z_NO_FLUSH);
461 if (ret != Z_OK) zlib_error(ret, "bitio:deflate_deflate", &z->zs);
463 if(z->zs.next_out != z->writebuffer) {
464 z->output->write(z->output, z->writebuffer, z->zs.next_out - (Bytef*)z->writebuffer);
465 z->zs.next_out = z->writebuffer;
466 z->zs.avail_out = ZLIB_BUFFER_SIZE;
469 if(!z->zs.avail_in) {
476 fprintf(stderr, "Error: swftools was compiled without zlib support");
480 static void writer_zlibdeflate_finish(writer_t*writer)
483 struct zlibdeflate_t*z = (struct zlibdeflate_t*)writer->internal;
486 if(writer->type != WRITER_TYPE_ZLIB) {
487 fprintf(stderr, "Wrong writer ID (writer not initialized?)\n");
494 ret = deflate(&z->zs, Z_FINISH);
496 ret != Z_STREAM_END) zlib_error(ret, "bitio:deflate_finish", &z->zs);
498 if(z->zs.next_out != z->writebuffer) {
499 z->output->write(z->output, z->writebuffer, z->zs.next_out - (Bytef*)z->writebuffer);
500 z->zs.next_out = z->writebuffer;
501 z->zs.avail_out = ZLIB_BUFFER_SIZE;
504 if (ret == Z_STREAM_END) {
509 ret = deflateEnd(&z->zs);
510 if (ret != Z_OK) zlib_error(ret, "bitio:deflate_end", &z->zs);
511 free(writer->internal);
512 memset(writer, 0, sizeof(writer_t));
513 //output->finish(output);
515 fprintf(stderr, "Error: swftools was compiled without zlib support");
519 void writer_init_zlibdeflate(writer_t*w, writer_t*output)
522 struct zlibdeflate_t*z;
524 memset(w, 0, sizeof(writer_t));
525 z = (struct zlibdeflate_t*)malloc(sizeof(struct zlibdeflate_t));
526 memset(z, 0, sizeof(struct zlibdeflate_t));
528 w->write = writer_zlibdeflate_write;
529 w->finish = writer_zlibdeflate_finish;
530 w->type = WRITER_TYPE_ZLIB;
533 memset(&z->zs,0,sizeof(z_stream));
534 z->zs.zalloc = Z_NULL;
535 z->zs.zfree = Z_NULL;
536 z->zs.opaque = Z_NULL;
537 ret = deflateInit(&z->zs, 9);
538 if (ret != Z_OK) zlib_error(ret, "bitio:deflate_init", &z->zs);
541 z->zs.next_out = z->writebuffer;
542 z->zs.avail_out = ZLIB_BUFFER_SIZE;
544 fprintf(stderr, "Error: swftools was compiled without zlib support");
549 /* ----------------------- bit handling routines -------------------------- */
551 void writer_writebit(writer_t*w, int bit)
555 w->write(w, &w->mybyte, 1);
560 w->mybyte |= 1 << (7 - w->bitpos);
563 void writer_writebits(writer_t*w, unsigned int data, int bits)
568 writer_writebit(w, (data >> (bits-t-1))&1);
571 void writer_resetbits(writer_t*w)
574 w->write(w, &w->mybyte, 1);
579 unsigned int reader_readbit(reader_t*r)
584 r->read(r, &r->mybyte, 1);
586 return (r->mybyte>>(7-r->bitpos++))&1;
588 unsigned int reader_readbits(reader_t*r, int num)
595 val|=reader_readbit(r);
599 void reader_resetbits(reader_t*r)
606 U8 reader_readU8(reader_t*r)
609 if(r->read(r, &b, 1)<1) {
610 fprintf(stderr, "bitio.c:reader_readU8: Read over end of memory region\n");
614 U16 reader_readU16(reader_t*r)
617 if(r->read(r, &b1, 1)<1) {
618 fprintf(stderr, "bitio.c:reader_readU16: Read over end of memory region\n");
620 if(r->read(r, &b2, 1)<1) {
621 fprintf(stderr, "bitio.c:reader_readU16: Read over end of memory region\n");
625 U32 reader_readU32(reader_t*r)
627 U8 b1=0,b2=0,b3=0,b4=0;
628 if(r->read(r, &b1, 1)<1)
629 fprintf(stderr, "bitio.c:reader_readU32: Read over end of memory region\n");
630 if(r->read(r, &b2, 1)<1)
631 fprintf(stderr, "bitio.c:reader_readU32: Read over end of memory region\n");
632 if(r->read(r, &b3, 1)<1)
633 fprintf(stderr, "bitio.c:reader_readU32: Read over end of memory region\n");
634 if(r->read(r, &b4, 1)<1)
635 fprintf(stderr, "bitio.c:reader_readU32: Read over end of memory region\n");
636 return b1|b2<<8|b3<<16|b4<<24;
638 float reader_readFloat(reader_t*r)
640 U8 b1=0,b2=0,b3=0,b4=0;
645 U32 w = (b1|b2<<8|b3<<16|b4<<24);
648 double reader_readDouble(reader_t*r)
656 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);
659 char*reader_readString(reader_t*r)
662 writer_init_growingmemwriter(&g, 16);
664 U8 b = reader_readU8(r);
665 writer_writeU8(&g, b);
669 char*string = (char*)writer_growmemwrite_getmem(&g);
670 writer_growmemwrite_finish(&g);
674 void writer_writeString(writer_t*w, const char*s)
678 w->write(w, (void*)s, l);
679 w->write(w, &zero, 1);
681 void writer_writeU8(writer_t*w, unsigned char b)
685 void writer_writeU16(writer_t*w, unsigned short v)
687 unsigned char b1 = v;
688 unsigned char b2 = v>>8;
692 void writer_writeU32(writer_t*w, unsigned long v)
694 unsigned char b1 = v;
695 unsigned char b2 = v>>8;
696 unsigned char b3 = v>>16;
697 unsigned char b4 = v>>24;
703 void writer_writeFloat(writer_t*w, float f)
705 unsigned long v = *(unsigned long*)&f;
706 unsigned char b1 = v;
707 unsigned char b2 = v>>8;
708 unsigned char b3 = v>>16;
709 unsigned char b4 = v>>24;
715 void writer_writeDouble(writer_t*w, double f)
720 unsigned long long v = *(unsigned long long*)&f;
721 unsigned char b1 = v;
722 unsigned char b2 = v>>8;
723 unsigned char b3 = v>>16;
724 unsigned char b4 = v>>24;
725 unsigned char b5 = v>>32;
726 unsigned char b6 = v>>40;
727 unsigned char b7 = v>>48;
728 unsigned char b8 = v>>56;