Fixes for the tests in pseudo-stack mode.
[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
19 int main(void)
20 {
21    int i;
22    int ret = 0;
23    ec_enc enc;
24    ec_dec dec;
25    ec_byte_buffer buf;
26    int val[10000], decay[10000];
27    ALLOC_STACK;
28    ec_byte_writeinit(&buf);
29    ec_enc_init(&enc,&buf);
30    
31    val[0] = 3; decay[0] = 6000;
32    val[1] = 0; decay[1] = 5800;
33    val[2] = -1; decay[2] = 5600;
34    for (i=3;i<10000;i++)
35    {
36       val[i] = rand()%15-7;
37       decay[i] = rand()%11000+5000;
38    }
39    for (i=0;i<10000;i++)
40       ec_laplace_encode(&enc, &val[i], decay[i]);      
41       
42    ec_enc_done(&enc);
43
44    ec_byte_readinit(&buf,ec_byte_get_buffer(&buf),ec_byte_bytes(&buf));
45    ec_dec_init(&dec,&buf);
46
47    for (i=0;i<10000;i++)
48    {
49       int d = ec_laplace_decode(&dec, decay[i]);
50       if (d != val[i])
51       {
52          fprintf (stderr, "Got %d instead of %d\n", d, val[i]);
53          ret = 1;
54       }
55    }
56    
57    ec_byte_writeclear(&buf);
58    return ret;
59 }