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