Cleanup, de-inlining some math functions
[opus.git] / tests / ectest.c
index 947afae..4552c1b 100644 (file)
@@ -1,16 +1,32 @@
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
 #include <stdlib.h>
 #include <stdio.h>
 #include <math.h>
-#include "bitrenc.h"
+#include <time.h>       
 #include "entcode.h"
 #include "entenc.h"
 #include "entdec.h"
+#include <string.h>
+
+#include "../libcelt/rangeenc.c"
+#include "../libcelt/rangedec.c"
+#include "../libcelt/entenc.c"
+#include "../libcelt/entdec.c"
+#include "../libcelt/entcode.c"
+
+#ifndef M_LOG2E
+# define M_LOG2E    1.4426950408889634074
+#endif
+#define DATA_SIZE 10000000
+#define DATA_SIZE2 10000
 
 int main(int _argc,char **_argv){
   ec_byte_buffer buf;
   ec_enc         enc;
   ec_dec         dec;
-  ec_uint64      sym64;
   long           nbits;
   long           nbits2;
   double         entropy;
@@ -18,18 +34,28 @@ int main(int _argc,char **_argv){
   int            ftb;
   int            sym;
   int            sz;
-  int            s;
   int            i;
+  int            ret;
+  unsigned int   seed;
+  ret=0;
   entropy=0;
+  unsigned char *ptr;
+    if (_argc > 2) {
+       fprintf(stderr, "Usage: %s [<seed>]\n", _argv[0]);
+       return 1;
+    }
+    if (_argc > 1)
+       seed = atoi(_argv[1]);
+    else
+       seed = time(NULL);
   /*Testing encoding of raw bit values.*/
-  ec_byte_writeinit(&buf);
+  ptr = malloc(DATA_SIZE);
+  ec_byte_writeinit_buffer(&buf, ptr, DATA_SIZE);
   ec_enc_init(&enc,&buf);
-  for(ft=0;ft<1024;ft++){
+  for(ft=2;ft<1024;ft++){
     for(i=0;i<ft;i++){
       entropy+=log(ft)*M_LOG2E;
       ec_enc_uint(&enc,i,ft);
-      entropy+=log(ft)*M_LOG2E+30;
-      ec_enc_uint64(&enc,(ec_uint64)i<<30|i,(ec_uint64)ft<<30);
     }
   }
   /*Testing encoding of raw bit values.*/
@@ -42,14 +68,7 @@ int main(int _argc,char **_argv){
       if(nbits2-nbits!=ftb){
         fprintf(stderr,"Used %li bits to encode %i bits directly.\n",
          nbits2-nbits,ftb);
-      }
-      entropy+=ftb+30;
-      nbits=nbits2;
-      ec_enc_bits64(&enc,(ec_uint64)i<<30|i,ftb+30);
-      nbits2=ec_enc_tell(&enc,0);
-      if(nbits2-nbits!=ftb+30){
-        fprintf(stderr,"Used %li bits to encode %i bits directly.\n",
-         nbits2-nbits,ftb+30);
+        ret=-1;
       }
     }
   }
@@ -59,19 +78,14 @@ int main(int _argc,char **_argv){
    "Encoded %0.2lf bits of entropy to %0.2lf bits (%0.3lf%% wasted).\n",
    entropy,ldexp(nbits,-4),100*(nbits-ldexp(entropy,4))/nbits);
   fprintf(stderr,"Packed to %li bytes.\n",(long)(buf.ptr-buf.buf));
-  ec_byte_readinit(&buf,ec_byte_get_buffer(&buf),ec_byte_bytes(&buf));
+  ec_byte_readinit(&buf,ptr,DATA_SIZE);
   ec_dec_init(&dec,&buf);
-  for(ft=0;ft<1024;ft++){
+  for(ft=2;ft<1024;ft++){
     for(i=0;i<ft;i++){
       sym=ec_dec_uint(&dec,ft);
       if(sym!=i){
         fprintf(stderr,"Decoded %i instead of %i with ft of %i.\n",sym,i,ft);
-        return -1;
-      }
-      sym64=ec_dec_uint64(&dec,(ec_uint64)ft<<30);
-      if(sym64!=((ec_uint64)i<<30|i)){
-        fprintf(stderr,"Decoded %lli instead of %lli with ft of %lli.\n",sym64,
-         (ec_uint64)i<<30|i,(ec_uint64)ft<<30);
+        ret=-1;
       }
     }
   }
@@ -80,12 +94,7 @@ int main(int _argc,char **_argv){
       sym=ec_dec_bits(&dec,ftb);
       if(sym!=i){
         fprintf(stderr,"Decoded %i instead of %i with ftb of %i.\n",sym,i,ftb);
-        return -1;
-      }
-      sym64=ec_dec_bits64(&dec,ftb+30);
-      if(sym64!=((ec_uint64)i<<30|i)){
-        fprintf(stderr,"Decoded %lli instead of %lli with ftb of %i.\n",
-         sym64,(ec_uint64)i<<30|i,ftb+30);
+        ret=-1;
       }
     }
   }
@@ -94,19 +103,19 @@ int main(int _argc,char **_argv){
     fprintf(stderr,
      "Reported number of bits used was %0.2lf, should be %0.2lf.\n",
      ldexp(nbits2,-4),ldexp(nbits,-4));
+    ret=-1;
   }
-  ec_byte_writeclear(&buf);
-  fprintf(stderr,"Testing random streams...\n");
-  srand(0);
-  for(i=0;i<4096;i++){
+  srand(seed);
+  fprintf(stderr,"Testing random streams... Random seed: %u (%.4X)\n", seed, rand() % 65536);
+  for(i=0;i<409600;i++){
     unsigned *data;
     int       j;
     int tell_bits;
     int zeros;
-    ft=rand()/((RAND_MAX>>9)+1)+512;
-    sz=rand()/((RAND_MAX>>9)+1);
+    ft=rand()/((RAND_MAX>>(rand()%11))+1)+10;
+    sz=rand()/((RAND_MAX>>(rand()%9))+1);
     data=(unsigned *)malloc(sz*sizeof(*data));
-    ec_byte_writeinit(&buf);
+    ec_byte_writeinit_buffer(&buf, ptr, DATA_SIZE2);
     ec_enc_init(&enc,&buf);
     zeros = rand()%13==0;
     for(j=0;j<sz;j++){
@@ -116,31 +125,31 @@ int main(int _argc,char **_argv){
         data[j]=rand()%ft;
       ec_enc_uint(&enc,data[j],ft);
     }
+    if (rand()%2==0)
+      while(ec_enc_tell(&enc, 0)%8 != 0)
+        ec_enc_uint(&enc, rand()%2, 2);
     tell_bits = ec_enc_tell(&enc, 0);
     ec_enc_done(&enc);
     if ((tell_bits+7)/8 < ec_byte_bytes(&buf))
     {
-      fprintf (stderr, "tell() lied, there's %d bytes instead of %d\n", 
-               ec_byte_bytes(&buf), (tell_bits+7)/8);
+      fprintf (stderr, "tell() lied, there's %li bytes instead of %d (Random seed: %u)\n", 
+               ec_byte_bytes(&buf), (tell_bits+7)/8,seed);
+      ret=-1;
     }
     tell_bits -= 8*ec_byte_bytes(&buf);
-    if (tell_bits > 8)
-    {
-       printf ("tell() failed with %d bit offset\n", tell_bits);
-       return -1;
-    }
-    ec_byte_readinit(&buf,ec_byte_get_buffer(&buf),ec_byte_bytes(&buf));
+    ec_byte_readinit(&buf,ptr,DATA_SIZE2);
     ec_dec_init(&dec,&buf);
     for(j=0;j<sz;j++){
       sym=ec_dec_uint(&dec,ft);
       if(sym!=data[j]){
         fprintf(stderr,
-         "Decoded %i instead of %i with ft of %i at position %i of %i.\n",
-         sym,data[j],ft,j,sz);
+         "Decoded %i instead of %i with ft of %i at position %i of %i (Random seed: %u).\n",
+         sym,data[j],ft,j,sz,seed);
+        ret=-1;
       }
     }
-    ec_byte_writeclear(&buf);
     free(data);
   }
-  return 0;
+  free(ptr);
+  return ret;
 }