shithub: opus

ref: 86cb7e860064a384aeb0166bc4a0450d6b0c54ab
dir: /libcelt/entenc.c/

View raw version
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <stdlib.h>
#include <string.h>
#include "entenc.h"



#define EC_BUFFER_INCREMENT (256)

void ec_byte_writeinit(ec_byte_buffer *_b){
  _b->ptr=_b->buf=malloc(EC_BUFFER_INCREMENT*sizeof(char));
  _b->storage=EC_BUFFER_INCREMENT;
}

void ec_byte_writetrunc(ec_byte_buffer *_b,long _bytes){
  _b->ptr=_b->buf+_bytes;
}

void ec_byte_write1(ec_byte_buffer *_b,unsigned _value){
  ptrdiff_t endbyte;
  endbyte=_b->ptr-_b->buf;
  if(endbyte>=_b->storage){
    _b->buf=realloc(_b->buf,(_b->storage+EC_BUFFER_INCREMENT)*sizeof(char));
    _b->storage+=EC_BUFFER_INCREMENT;
    _b->ptr=_b->buf+endbyte;
  }
  *(_b->ptr++)=(unsigned char)_value;
}

void ec_byte_write4(ec_byte_buffer *_b,ec_uint32 _value){
  ptrdiff_t endbyte;
  endbyte=_b->ptr-_b->buf;
  if(endbyte+4>_b->storage){
    _b->buf=realloc(_b->buf,(_b->storage+EC_BUFFER_INCREMENT)*sizeof(char));
    _b->storage+=EC_BUFFER_INCREMENT;
    _b->ptr=_b->buf+endbyte;
  }
  *(_b->ptr++)=(unsigned char)_value;
  _value>>=8;
  *(_b->ptr++)=(unsigned char)_value;
  _value>>=8;
  *(_b->ptr++)=(unsigned char)_value;
  _value>>=8;
  *(_b->ptr++)=(unsigned char)_value;
}

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){
    _b->storage=endbyte+_bytes+EC_BUFFER_INCREMENT;
    _b->buf=realloc(_b->buf,_b->storage*sizeof(char));
    _b->ptr=_b->buf+endbyte;
  }
  memmove(_b->ptr,_source,_bytes);
  _b->ptr+=_bytes;
}

void ec_byte_writeclear(ec_byte_buffer *_b){
  free(_b->buf);
}



void ec_enc_bits(ec_enc *_this,ec_uint32 _fl,int _ftb){
  unsigned fl;
  unsigned ft;
  while(_ftb>EC_UNIT_BITS){
    _ftb-=EC_UNIT_BITS;
    fl=(unsigned)(_fl>>_ftb)&EC_UNIT_MASK;
    ec_encode(_this,fl,fl+1,EC_UNIT_MASK+1);
  }
  ft=1<<_ftb;
  fl=(unsigned)_fl&ft-1;
  ec_encode(_this,fl,fl+1,ft);
}

void ec_enc_bits64(ec_enc *_this,ec_uint64 _fl,int _ftb){
  if(_ftb>32){
    ec_enc_bits(_this,(ec_uint32)(_fl>>32),_ftb-32);
    _ftb=32;
    _fl&=0xFFFFFFFF;
  }
  ec_enc_bits(_this,(ec_uint32)_fl,_ftb);
}

void ec_enc_uint(ec_enc *_this,ec_uint32 _fl,ec_uint32 _ft){
  ec_uint32 mask;
  unsigned  ft;
  unsigned  fl;
  int       ftb;
  _ft--;
  ftb=EC_ILOG(_ft)&-!!_ft;
  while(ftb>EC_UNIT_BITS){
    ftb-=EC_UNIT_BITS;
    ft=(_ft>>ftb)+1;
    fl=(unsigned)(_fl>>ftb);
    ec_encode(_this,fl,fl+1,ft);
    if(fl<ft-1){
      ec_enc_bits(_this,_fl,ftb);
      return;
    }
    mask=((ec_uint32)1<<ftb)-1;
    _fl=_fl&mask;
    _ft=_ft&mask;
  }
  ec_encode(_this,_fl,_fl+1,_ft+1);
}

void ec_enc_uint64(ec_enc *_this,ec_uint64 _fl,ec_uint64 _ft){
  ec_uint64 mask;
  unsigned  ft;
  unsigned  fl;
  int       ftb;
  _ft--;
  ftb=EC_ILOG64(_ft)&-!!_ft;
  while(ftb>EC_UNIT_BITS){
    ftb-=EC_UNIT_BITS;
    ft=(unsigned)(_ft>>ftb)+1;
    fl=(unsigned)(_fl>>ftb);
    ec_encode(_this,fl,fl+1,ft);
    if(fl<ft-1){
      ec_enc_bits64(_this,_fl,ftb);
      return;
    }
    mask=((ec_uint64)1<<ftb)-1;
    _fl=_fl&mask;
    _ft=_ft&mask;
  }
  ec_encode(_this,_fl,_fl+1,_ft+1);
}