Made a second version of ec_{en|de}code optimised for encoding bits (no div
[opus.git] / libcelt / entdec.c
1 #ifdef HAVE_CONFIG_H
2 #include "config.h"
3 #endif
4
5 #include <stddef.h>
6 #include "entdec.h"
7
8
9
10 void ec_byte_readinit(ec_byte_buffer *_b,unsigned char *_buf,long _bytes){
11   _b->buf=_b->ptr=_buf;
12   _b->storage=_bytes;
13 }
14
15 int ec_byte_look1(ec_byte_buffer *_b){
16   ptrdiff_t endbyte;
17   endbyte=_b->ptr-_b->buf;
18   if(endbyte>=_b->storage)return -1;
19   else return _b->ptr[0];
20 }
21
22 int ec_byte_look4(ec_byte_buffer *_b,ec_uint32 *_val){
23   ptrdiff_t endbyte;
24   endbyte=_b->ptr-_b->buf;
25   if(endbyte+4>_b->storage){
26     if(endbyte<_b->storage){
27       *_val=_b->ptr[0];
28       endbyte++;
29       if(endbyte<_b->storage){
30         *_val|=(ec_uint32)_b->ptr[1]<<8;
31         endbyte++;
32         if(endbyte<_b->storage)*_val|=(ec_uint32)_b->ptr[2]<<16;
33       }
34     }
35     return -1;
36   }
37   else{
38     *_val=_b->ptr[0];
39     *_val|=(ec_uint32)_b->ptr[1]<<8;
40     *_val|=(ec_uint32)_b->ptr[2]<<16;
41     *_val|=(ec_uint32)_b->ptr[3]<<24;
42   }
43   return 0;
44 }
45
46 void ec_byte_adv1(ec_byte_buffer *_b){
47   _b->ptr++;
48 }
49
50 void ec_byte_adv4(ec_byte_buffer *_b){
51   _b->ptr+=4;
52 }
53
54 int ec_byte_read1(ec_byte_buffer *_b){
55   ptrdiff_t endbyte;
56   endbyte=_b->ptr-_b->buf;
57   if(endbyte>=_b->storage)return -1;
58   else return *(_b->ptr++);
59 }
60
61 int ec_byte_read4(ec_byte_buffer *_b,ec_uint32 *_val){
62   unsigned char *end;
63   end=_b->buf+_b->storage;
64   if(_b->ptr+4>end){
65     if(_b->ptr<end){
66       *_val=*(_b->ptr++);
67       if(_b->ptr<end){
68         *_val|=(ec_uint32)*(_b->ptr++)<<8;
69         if(_b->ptr<end)*_val|=(ec_uint32)*(_b->ptr++)<<16;
70       }
71     }
72     return -1;
73   }
74   else{
75     *_val=(*_b->ptr++);
76     *_val|=(ec_uint32)*(_b->ptr++)<<8;
77     *_val|=(ec_uint32)*(_b->ptr++)<<16;
78     *_val|=(ec_uint32)*(_b->ptr++)<<24;
79   }
80   return 0;
81 }
82
83
84
85 ec_uint32 ec_dec_bits(ec_dec *_this,int _ftb){
86   ec_uint32 t;
87   unsigned  s;
88   unsigned  ft;
89   t=0;
90   while(_ftb>EC_UNIT_BITS){
91     s=ec_decode_bin(_this,EC_UNIT_BITS);
92     ec_dec_update(_this,s,s+1,EC_UNIT_MASK+1);
93     t=t<<EC_UNIT_BITS|s;
94     _ftb-=EC_UNIT_BITS;
95   }
96   ft=1U<<_ftb;
97   s=ec_decode_bin(_this,_ftb);
98   ec_dec_update(_this,s,s+1,ft);
99   t=t<<_ftb|s;
100   return t;
101 }
102
103 ec_uint64 ec_dec_bits64(ec_dec *_this,int _ftb){
104   ec_uint32 t;
105   if(_ftb>32){
106     t=ec_dec_bits(_this,_ftb-32);
107     _ftb=32;
108   }
109   else t=0;
110   return (ec_uint64)t<<32|ec_dec_bits(_this,_ftb);
111 }
112
113 ec_uint32 ec_dec_uint(ec_dec *_this,ec_uint32 _ft){
114   ec_uint32 t;
115   unsigned  ft;
116   unsigned  s;
117   int       ftb;
118   t=0;
119   _ft--;
120   ftb=EC_ILOG(_ft);
121   if(ftb>EC_UNIT_BITS){
122     ftb-=EC_UNIT_BITS;
123     ft=(unsigned)(_ft>>ftb)+1;
124     s=ec_decode(_this,ft);
125     ec_dec_update(_this,s,s+1,ft);
126     t=t<<EC_UNIT_BITS|s;
127     return t<<ftb|ec_dec_bits(_this,ftb);
128   } else {
129     _ft++;
130     s=ec_decode(_this,(unsigned)_ft);
131     ec_dec_update(_this,s,s+1,(unsigned)_ft);
132     t=t<<ftb|s;
133     return t;
134   }
135 }
136
137 ec_uint64 ec_dec_uint64(ec_dec *_this,ec_uint64 _ft){
138   ec_uint64 t;
139   unsigned  ft;
140   unsigned  s;
141   int       ftb;
142   t=0;
143   _ft--;
144   ftb=EC_ILOG64(_ft);
145   if(ftb>EC_UNIT_BITS){
146     ftb-=EC_UNIT_BITS;
147     ft=(unsigned)(_ft>>ftb)+1;
148     s=ec_decode(_this,ft);
149     ec_dec_update(_this,s,s+1,ft);
150     t=t<<EC_UNIT_BITS|s;
151     return t<<ftb|ec_dec_bits64(_this,ftb);
152   } else {
153     _ft++;
154     s=ec_decode(_this,(unsigned)_ft);
155     ec_dec_update(_this,s,s+1,(unsigned)_ft);
156     t=t<<ftb|s;
157     return t;
158   }
159 }