Removing the 64-bit part of the range coder.
[opus.git] / tests / ectest.c
1 #ifdef HAVE_CONFIG_H
2 #include "config.h"
3 #endif
4
5 #include <stdlib.h>
6 #include <stdio.h>
7 #include <math.h>
8 #include "entcode.h"
9 #include "entenc.h"
10 #include "entdec.h"
11
12 #ifndef M_LOG2E
13 # define M_LOG2E    1.4426950408889634074
14 #endif
15
16 int main(int _argc,char **_argv){
17   ec_byte_buffer buf;
18   ec_enc         enc;
19   ec_dec         dec;
20   long           nbits;
21   long           nbits2;
22   double         entropy;
23   int            ft;
24   int            ftb;
25   int            sym;
26   int            sz;
27   int            i;
28   int            ret;
29   ret=0;
30   entropy=0;
31   /*Testing encoding of raw bit values.*/
32   ec_byte_writeinit(&buf);
33   ec_enc_init(&enc,&buf);
34   for(ft=0;ft<1024;ft++){
35     for(i=0;i<ft;i++){
36       entropy+=log(ft)*M_LOG2E;
37       ec_enc_uint(&enc,i,ft);
38     }
39   }
40   /*Testing encoding of raw bit values.*/
41   for(ftb=0;ftb<16;ftb++){
42     for(i=0;i<(1<<ftb);i++){
43       entropy+=ftb;
44       nbits=ec_enc_tell(&enc,0);
45       ec_enc_bits(&enc,i,ftb);
46       nbits2=ec_enc_tell(&enc,0);
47       if(nbits2-nbits!=ftb){
48         fprintf(stderr,"Used %li bits to encode %i bits directly.\n",
49          nbits2-nbits,ftb);
50         ret=-1;
51       }
52     }
53   }
54   nbits=ec_enc_tell(&enc,4);
55   ec_enc_done(&enc);
56   fprintf(stderr,
57    "Encoded %0.2lf bits of entropy to %0.2lf bits (%0.3lf%% wasted).\n",
58    entropy,ldexp(nbits,-4),100*(nbits-ldexp(entropy,4))/nbits);
59   fprintf(stderr,"Packed to %li bytes.\n",(long)(buf.ptr-buf.buf));
60   ec_byte_readinit(&buf,ec_byte_get_buffer(&buf),ec_byte_bytes(&buf));
61   ec_dec_init(&dec,&buf);
62   for(ft=0;ft<1024;ft++){
63     for(i=0;i<ft;i++){
64       sym=ec_dec_uint(&dec,ft);
65       if(sym!=i){
66         fprintf(stderr,"Decoded %i instead of %i with ft of %i.\n",sym,i,ft);
67         ret=-1;
68       }
69     }
70   }
71   for(ftb=0;ftb<16;ftb++){
72     for(i=0;i<(1<<ftb);i++){
73       sym=ec_dec_bits(&dec,ftb);
74       if(sym!=i){
75         fprintf(stderr,"Decoded %i instead of %i with ftb of %i.\n",sym,i,ftb);
76         ret=-1;
77       }
78     }
79   }
80   nbits2=ec_dec_tell(&dec,4);
81   if(nbits!=nbits2){
82     fprintf(stderr,
83      "Reported number of bits used was %0.2lf, should be %0.2lf.\n",
84      ldexp(nbits2,-4),ldexp(nbits,-4));
85     ret=-1;
86   }
87   ec_byte_writeclear(&buf);
88   fprintf(stderr,"Testing random streams...\n");
89   srand(0);
90   for(i=0;i<409600;i++){
91     unsigned *data;
92     int       j;
93     int tell_bits;
94     int zeros;
95     ft=rand()/((RAND_MAX>>(rand()%11))+1)+10;
96     sz=rand()/((RAND_MAX>>(rand()%9))+1);
97     data=(unsigned *)malloc(sz*sizeof(*data));
98     ec_byte_writeinit(&buf);
99     ec_enc_init(&enc,&buf);
100     zeros = rand()%13==0;
101     for(j=0;j<sz;j++){
102       if (zeros)
103         data[j]=0;
104       else
105         data[j]=rand()%ft;
106       ec_enc_uint(&enc,data[j],ft);
107     }
108     if (rand()%2==0)
109       while(ec_enc_tell(&enc, 0)%8 != 0)
110         ec_enc_uint(&enc, rand()%2, 2);
111     tell_bits = ec_enc_tell(&enc, 0);
112     ec_enc_done(&enc);
113     if ((tell_bits+7)/8 < ec_byte_bytes(&buf))
114     {
115       fprintf (stderr, "tell() lied, there's %li bytes instead of %d\n", 
116                ec_byte_bytes(&buf), (tell_bits+7)/8);
117       ret=-1;
118     }
119     tell_bits -= 8*ec_byte_bytes(&buf);
120     ec_byte_readinit(&buf,ec_byte_get_buffer(&buf),ec_byte_bytes(&buf));
121     ec_dec_init(&dec,&buf);
122     for(j=0;j<sz;j++){
123       sym=ec_dec_uint(&dec,ft);
124       if(sym!=data[j]){
125         fprintf(stderr,
126          "Decoded %i instead of %i with ft of %i at position %i of %i.\n",
127          sym,data[j],ft,j,sz);
128         ret=-1;
129       }
130     }
131     ec_byte_writeclear(&buf);
132     free(data);
133   }
134   return ret;
135 }