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