re-enable support for resizable buffers in the range coder
authorJean-Marc Valin <jean-marc.valin@usherbrooke.ca>
Sat, 18 Oct 2008 13:11:05 +0000 (09:11 -0400)
committerJean-Marc Valin <jean-marc.valin@usherbrooke.ca>
Sat, 18 Oct 2008 13:11:05 +0000 (09:11 -0400)
TODO
libcelt/entcode.h
libcelt/entenc.c

diff --git a/TODO b/TODO
index 05826ed..4c48482 100644 (file)
--- a/TODO
+++ b/TODO
 
 - Encode band shape (or just tilt)?
 - Make energy encoding more robust to losses?
+
+
+Misc:
+Detect uint decoding and flag them in the decoder directly
+If we attempt to write too many bits on the encoder side, set a flag instead of
+aborting
+Save "raw bytes" at the end of the stream
index 67c6df4..298d19b 100644 (file)
@@ -26,6 +26,7 @@ struct ec_byte_buffer{
   unsigned char *buf;
   unsigned char *ptr;
   long           storage;
+  int            resizable;
 };
 
 /*Encoding functions.*/
index 1a9f5ba..3da351e 100644 (file)
 void ec_byte_writeinit_buffer(ec_byte_buffer *_b, unsigned char *_buf, long _size){
   _b->ptr=_b->buf=_buf;
   _b->storage=_size;
+  _b->resizable = 0;
 }
 
 void ec_byte_writeinit(ec_byte_buffer *_b){
   _b->ptr=_b->buf=celt_alloc(EC_BUFFER_INCREMENT*sizeof(char));
   _b->storage=EC_BUFFER_INCREMENT;
+  _b->resizable = 1;
 }
 
 void ec_byte_writetrunc(ec_byte_buffer *_b,long _bytes){
@@ -27,7 +29,13 @@ void ec_byte_write1(ec_byte_buffer *_b,unsigned _value){
   ptrdiff_t endbyte;
   endbyte=_b->ptr-_b->buf;
   if(endbyte>=_b->storage){
-    celt_fatal("range encoder overflow\n");
+    if (_b->resizable){
+      _b->buf=celt_realloc(_b->buf,(_b->storage+EC_BUFFER_INCREMENT)*sizeof(char));
+      _b->storage+=EC_BUFFER_INCREMENT;
+      _b->ptr=_b->buf+endbyte;
+    } else {
+      celt_fatal("range encoder overflow\n");
+    }
   }
   *(_b->ptr++)=(unsigned char)_value;
 }
@@ -36,7 +44,13 @@ void ec_byte_write4(ec_byte_buffer *_b,ec_uint32 _value){
   ptrdiff_t endbyte;
   endbyte=_b->ptr-_b->buf;
   if(endbyte+4>_b->storage){
-    celt_fatal("range encoder overflow\n");
+    if (_b->resizable){
+      _b->buf=celt_realloc(_b->buf,(_b->storage+EC_BUFFER_INCREMENT)*sizeof(char));
+      _b->storage+=EC_BUFFER_INCREMENT;
+      _b->ptr=_b->buf+endbyte;
+    } else {
+      celt_fatal("range encoder overflow\n");
+    }
   }
   *(_b->ptr++)=(unsigned char)_value;
   _value>>=8;
@@ -51,7 +65,13 @@ void ec_byte_writecopy(ec_byte_buffer *_b,void *_source,long _bytes){
   ptrdiff_t endbyte;
   endbyte=_b->ptr-_b->buf;
   if(endbyte+_bytes>_b->storage){
-    celt_fatal("range encoder overflow\n");
+    if (_b->resizable){
+      _b->storage=endbyte+_bytes+EC_BUFFER_INCREMENT;
+      _b->buf=celt_realloc(_b->buf,_b->storage*sizeof(char));
+      _b->ptr=_b->buf+endbyte;
+    } else {
+      celt_fatal("range encoder overflow\n");
+    }
   }
   memmove(_b->ptr,_source,_bytes);
   _b->ptr+=_bytes;