417f4e1c9922a52ba33a9c63ef0db1d964747191
[opus.git] / 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 }