tests/laplace-test.c

   1 #ifdef HAVE_CONFIG_H
   2 #include "config.h"
   3 #endif
   4
   5 #include <stdio.h>
   6 #include <stdlib.h>
   7 #include "laplace.h"
   8 #define CELT_C
   9 #include "../libcelt/stack_alloc.h"
  10
  11 #include "../libcelt/rangeenc.c"
  12 #include "../libcelt/rangedec.c"
  13 #include "../libcelt/entenc.c"
  14 #include "../libcelt/entdec.c"
  15 #include "../libcelt/entcode.c"
  16 #include "../libcelt/laplace.c"
  17
  18 #define DATA_SIZE 40000
  19
  20 int main(void)
  21 {
  22    int i;
  23    int ret = 0;
  24    ec_enc enc;
  25    ec_dec dec;
  26    ec_byte_buffer buf;
  27    unsigned char *ptr;
  28    int val[10000], decay[10000];
  29    ALLOC_STACK;
  30    ptr = malloc(DATA_SIZE);
  31    ec_byte_writeinit_buffer(&buf, ptr, DATA_SIZE);
  32    //ec_byte_writeinit(&buf);
  33    ec_enc_init(&enc,&buf);
  34
  35    val[0] = 3; decay[0] = 6000;
  36    val[1] = 0; decay[1] = 5800;
  37    val[2] = -1; decay[2] = 5600;
  38    for (i=3;i<10000;i++)
  39    {
  40       val[i] = rand()%15-7;
  41       decay[i] = rand()%11000+5000;
  42    }
  43    for (i=0;i<10000;i++)
  44       ec_laplace_encode(&enc, &val[i], decay[i]);
  45
  46    ec_enc_done(&enc);
  47
  48    ec_byte_readinit(&buf,ec_byte_get_buffer(&buf),ec_byte_bytes(&buf));
  49    ec_dec_init(&dec,&buf);
  50
  51    for (i=0;i<10000;i++)
  52    {
  53       int d = ec_laplace_decode(&dec, decay[i]);
  54       if (d != val[i])
  55       {
  56          fprintf (stderr, "Got %d instead of %d\n", d, val[i]);
  57          ret = 1;
  58       }
  59    }
  60
  61    return ret;
  62 }