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