celt_encoder_create() now defaults to Opus standard 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 #define DATA_SIZE 40000
19
20 int ec_laplace_get_start_freq(int decay)
21 {
22    celt_uint32 ft = 32768 - LAPLACE_MINP*(2*LAPLACE_NMIN+1);
23    int fs = (ft*(16384-decay))/(16384+decay);
24    return fs+LAPLACE_MINP;
25 }
26
27 int main(void)
28 {
29    int i;
30    int ret = 0;
31    ec_enc enc;
32    ec_dec dec;
33    ec_byte_buffer buf;
34    unsigned char *ptr;
35    int val[10000], decay[10000];
36    ALLOC_STACK;
37    ptr = malloc(DATA_SIZE);
38    ec_byte_writeinit_buffer(&buf, ptr, DATA_SIZE);
39    //ec_byte_writeinit(&buf);
40    ec_enc_init(&enc,&buf);
41    
42    val[0] = 3; decay[0] = 6000;
43    val[1] = 0; decay[1] = 5800;
44    val[2] = -1; decay[2] = 5600;
45    for (i=3;i<10000;i++)
46    {
47       val[i] = rand()%15-7;
48       decay[i] = rand()%11000+5000;
49    }
50    for (i=0;i<10000;i++)
51       ec_laplace_encode(&enc, &val[i],
52             ec_laplace_get_start_freq(decay[i]), decay[i]);
53
54    ec_enc_done(&enc);
55
56    ec_byte_readinit(&buf,ec_byte_get_buffer(&buf),ec_byte_bytes(&buf));
57    ec_dec_init(&dec,&buf);
58
59    for (i=0;i<10000;i++)
60    {
61       int d = ec_laplace_decode(&dec,
62             ec_laplace_get_start_freq(decay[i]), decay[i]);
63       if (d != val[i])
64       {
65          fprintf (stderr, "Got %d instead of %d\n", d, val[i]);
66          ret = 1;
67       }
68    }
69    
70    return ret;
71 }