xref: /aosp_15_r20/external/libopus/celt/tests/test_unit_laplace.c (revision a58d3d2adb790c104798cd88c8a3aff4fa8b82cc)
1*a58d3d2aSXin Li /* Copyright (c) 2008-2011 Xiph.Org Foundation, Mozilla Corporation
2*a58d3d2aSXin Li    Written by Jean-Marc Valin and Timothy B. Terriberry */
3*a58d3d2aSXin Li /*
4*a58d3d2aSXin Li    Redistribution and use in source and binary forms, with or without
5*a58d3d2aSXin Li    modification, are permitted provided that the following conditions
6*a58d3d2aSXin Li    are met:
7*a58d3d2aSXin Li 
8*a58d3d2aSXin Li    - Redistributions of source code must retain the above copyright
9*a58d3d2aSXin Li    notice, this list of conditions and the following disclaimer.
10*a58d3d2aSXin Li 
11*a58d3d2aSXin Li    - Redistributions in binary form must reproduce the above copyright
12*a58d3d2aSXin Li    notice, this list of conditions and the following disclaimer in the
13*a58d3d2aSXin Li    documentation and/or other materials provided with the distribution.
14*a58d3d2aSXin Li 
15*a58d3d2aSXin Li    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16*a58d3d2aSXin Li    ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
17*a58d3d2aSXin Li    LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
18*a58d3d2aSXin Li    A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
19*a58d3d2aSXin Li    OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
20*a58d3d2aSXin Li    EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
21*a58d3d2aSXin Li    PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
22*a58d3d2aSXin Li    PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
23*a58d3d2aSXin Li    LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
24*a58d3d2aSXin Li    NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
25*a58d3d2aSXin Li    SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26*a58d3d2aSXin Li */
27*a58d3d2aSXin Li 
28*a58d3d2aSXin Li #ifdef HAVE_CONFIG_H
29*a58d3d2aSXin Li #include "config.h"
30*a58d3d2aSXin Li #endif
31*a58d3d2aSXin Li 
32*a58d3d2aSXin Li #include <stdio.h>
33*a58d3d2aSXin Li #include <stdlib.h>
34*a58d3d2aSXin Li #define CELT_C
35*a58d3d2aSXin Li #include "laplace.h"
36*a58d3d2aSXin Li #include "stack_alloc.h"
37*a58d3d2aSXin Li 
38*a58d3d2aSXin Li #include "entenc.c"
39*a58d3d2aSXin Li #include "entdec.c"
40*a58d3d2aSXin Li #include "entcode.c"
41*a58d3d2aSXin Li #include "laplace.c"
42*a58d3d2aSXin Li 
43*a58d3d2aSXin Li #define DATA_SIZE 40000
44*a58d3d2aSXin Li 
ec_laplace_get_start_freq(int decay)45*a58d3d2aSXin Li int ec_laplace_get_start_freq(int decay)
46*a58d3d2aSXin Li {
47*a58d3d2aSXin Li    opus_uint32 ft = 32768 - LAPLACE_MINP*(2*LAPLACE_NMIN+1);
48*a58d3d2aSXin Li    int fs = (ft*(16384-decay))/(16384+decay);
49*a58d3d2aSXin Li    return fs+LAPLACE_MINP;
50*a58d3d2aSXin Li }
51*a58d3d2aSXin Li 
main(void)52*a58d3d2aSXin Li int main(void)
53*a58d3d2aSXin Li {
54*a58d3d2aSXin Li    int i;
55*a58d3d2aSXin Li    int ret = 0;
56*a58d3d2aSXin Li    ec_enc enc;
57*a58d3d2aSXin Li    ec_dec dec;
58*a58d3d2aSXin Li    unsigned char *ptr;
59*a58d3d2aSXin Li    int val[10000], decay[10000];
60*a58d3d2aSXin Li    ALLOC_STACK;
61*a58d3d2aSXin Li    ptr = (unsigned char *)malloc(DATA_SIZE);
62*a58d3d2aSXin Li    ec_enc_init(&enc,ptr,DATA_SIZE);
63*a58d3d2aSXin Li 
64*a58d3d2aSXin Li    val[0] = 3; decay[0] = 6000;
65*a58d3d2aSXin Li    val[1] = 0; decay[1] = 5800;
66*a58d3d2aSXin Li    val[2] = -1; decay[2] = 5600;
67*a58d3d2aSXin Li    for (i=3;i<10000;i++)
68*a58d3d2aSXin Li    {
69*a58d3d2aSXin Li       val[i] = rand()%15-7;
70*a58d3d2aSXin Li       decay[i] = rand()%11000+5000;
71*a58d3d2aSXin Li    }
72*a58d3d2aSXin Li    for (i=0;i<10000;i++)
73*a58d3d2aSXin Li       ec_laplace_encode(&enc, &val[i],
74*a58d3d2aSXin Li             ec_laplace_get_start_freq(decay[i]), decay[i]);
75*a58d3d2aSXin Li 
76*a58d3d2aSXin Li    ec_enc_done(&enc);
77*a58d3d2aSXin Li 
78*a58d3d2aSXin Li    ec_dec_init(&dec,ec_get_buffer(&enc),ec_range_bytes(&enc));
79*a58d3d2aSXin Li 
80*a58d3d2aSXin Li    for (i=0;i<10000;i++)
81*a58d3d2aSXin Li    {
82*a58d3d2aSXin Li       int d = ec_laplace_decode(&dec,
83*a58d3d2aSXin Li             ec_laplace_get_start_freq(decay[i]), decay[i]);
84*a58d3d2aSXin Li       if (d != val[i])
85*a58d3d2aSXin Li       {
86*a58d3d2aSXin Li          fprintf (stderr, "Got %d instead of %d\n", d, val[i]);
87*a58d3d2aSXin Li          ret = 1;
88*a58d3d2aSXin Li       }
89*a58d3d2aSXin Li    }
90*a58d3d2aSXin Li 
91*a58d3d2aSXin Li    free(ptr);
92*a58d3d2aSXin Li    RESTORE_STACK;
93*a58d3d2aSXin Li    return ret;
94*a58d3d2aSXin Li }
95