diff --git a/libcelt/entenc.c b/libcelt/entenc.c index 94201327532a13ac27730197b56aafcdebb65edf..9297b5f3c03c587eb51f83c9297605af4338307d 100644 --- a/libcelt/entenc.c +++ b/libcelt/entenc.c @@ -215,29 +215,27 @@ void ec_enc_bits(ec_enc *_this,ec_uint32 _fl,unsigned _bits){ _this->nbits_total+=_bits; } -int ec_enc_patch_initial_bits(ec_enc *_this,unsigned _val,int _nbits){ - int shift; - unsigned mask; - if(_nbits<0||_nbits>EC_SYM_BITS)return -1; - shift=EC_SYM_BITS-_nbits; - mask=(1<<_nbits)-1<<shift; - if(_this->offs>0){ - /*The first byte has been finalized.*/ - _this->buf[0]=(unsigned char)(_this->buf[0]&~mask|_val<<shift); - } - else if(_this->rem>=0){ - /*The first byte is still awaiting carry propagation.*/ - _this->rem=_this->rem&~mask|_val<<shift; - } - else if(_this->rng<=EC_CODE_TOP>>shift){ - /*The renormalization loop has never been run.*/ - _this->val=_this->val&~((ec_uint32)mask<<EC_CODE_SHIFT)| - (ec_uint32)_val<<EC_CODE_SHIFT+shift; - } - /*The encoder hasn't even encoded _nbits of data yet.*/ - else return -1; - /*Success!*/ - return 0; +void ec_enc_patch_initial_bits(ec_enc *_this,unsigned _val,unsigned _nbits){ + int shift; + unsigned mask; + celt_assert(_nbits<=EC_SYM_BITS); + shift=EC_SYM_BITS-_nbits; + mask=(1<<_nbits)-1<<shift; + if(_this->offs>0){ + /*The first byte has been finalized.*/ + _this->buf[0]=(unsigned char)(_this->buf[0]&~mask|_val<<shift); + } + else if(_this->rem>=0){ + /*The first byte is still awaiting carry propagation.*/ + _this->rem=_this->rem&~mask|_val<<shift; + } + else if(_this->rng<=EC_CODE_TOP>>shift){ + /*The renormalization loop has never been run.*/ + _this->val=_this->val&~((ec_uint32)mask<<EC_CODE_SHIFT)| + (ec_uint32)_val<<EC_CODE_SHIFT+shift; + } + /*The encoder hasn't even encoded _nbits of data yet.*/ + else _this->error=-1; } void ec_enc_shrink(ec_enc *_this,ec_uint32 _size){ diff --git a/libcelt/entenc.h b/libcelt/entenc.h index 51dbca9bde6f93c3efba8c7b048989df44cbde60..e64e88761ebd2fd28f20c6469572bb9512dfb43b 100644 --- a/libcelt/entenc.h +++ b/libcelt/entenc.h @@ -78,6 +78,22 @@ void ec_enc_uint(ec_enc *_this,ec_uint32 _fl,ec_uint32 _ft); This must be between 0 and 25, inclusive.*/ void ec_enc_bits(ec_enc *_this,ec_uint32 _fl,unsigned _ftb); +/*Overwrites a few bits at the very start of an existing stream, after they + have already been encoded. + This makes it possible to have a few flags up front, where it is easy for + decoders to access them without parsing the whole stream, even if their + values are not determined until late in the encoding process, without having + to buffer all the intermediate symbols in the encoder. + In order for this to work, at least _nbits bits must have already been + encoded using probabilities that are an exact power of two. + The encoder can verify the number of encoded bits is sufficient, but cannot + check this latter condition. + _val: The bits to encode (in the least _nbits significant bits). + They will be decoded in order from most-significant to least. + _nbits: The number of bits to overwrite. + This must be no more than 8.*/ +void ec_enc_patch_initial_bits(ec_enc *_this,unsigned _val,unsigned _nbits); + /*Compacts the data to fit in the target size. This moves up the raw bits at the end of the current buffer so they are at the end of the new buffer size.