xref: /aosp_15_r20/external/libopus/celt/tests/test_unit_rotation.c (revision a58d3d2adb790c104798cd88c8a3aff4fa8b82cc)
1*a58d3d2aSXin Li /* Copyright (c) 2008-2011 Xiph.Org Foundation
2*a58d3d2aSXin Li    Written by Jean-Marc Valin */
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 #ifndef CUSTOM_MODES
33*a58d3d2aSXin Li #define CUSTOM_MODES
34*a58d3d2aSXin Li #endif
35*a58d3d2aSXin Li 
36*a58d3d2aSXin Li #include <stdio.h>
37*a58d3d2aSXin Li #include <stdlib.h>
38*a58d3d2aSXin Li #include "vq.h"
39*a58d3d2aSXin Li #include "bands.h"
40*a58d3d2aSXin Li #include "stack_alloc.h"
41*a58d3d2aSXin Li #include <math.h>
42*a58d3d2aSXin Li 
43*a58d3d2aSXin Li 
44*a58d3d2aSXin Li #define MAX_SIZE 100
45*a58d3d2aSXin Li 
46*a58d3d2aSXin Li int ret=0;
test_rotation(int N,int K)47*a58d3d2aSXin Li void test_rotation(int N, int K)
48*a58d3d2aSXin Li {
49*a58d3d2aSXin Li    int i;
50*a58d3d2aSXin Li    double err = 0, ener = 0, snr, snr0;
51*a58d3d2aSXin Li    opus_val16 x0[MAX_SIZE];
52*a58d3d2aSXin Li    opus_val16 x1[MAX_SIZE];
53*a58d3d2aSXin Li    for (i=0;i<N;i++)
54*a58d3d2aSXin Li       x1[i] = x0[i] = rand()%32767-16384;
55*a58d3d2aSXin Li    exp_rotation(x1, N, 1, 1, K, SPREAD_NORMAL);
56*a58d3d2aSXin Li    for (i=0;i<N;i++)
57*a58d3d2aSXin Li    {
58*a58d3d2aSXin Li       err += (x0[i]-(double)x1[i])*(x0[i]-(double)x1[i]);
59*a58d3d2aSXin Li       ener += x0[i]*(double)x0[i];
60*a58d3d2aSXin Li    }
61*a58d3d2aSXin Li    snr0 = 20*log10(ener/err);
62*a58d3d2aSXin Li    err = ener = 0;
63*a58d3d2aSXin Li    exp_rotation(x1, N, -1, 1, K, SPREAD_NORMAL);
64*a58d3d2aSXin Li    for (i=0;i<N;i++)
65*a58d3d2aSXin Li    {
66*a58d3d2aSXin Li       err += (x0[i]-(double)x1[i])*(x0[i]-(double)x1[i]);
67*a58d3d2aSXin Li       ener += x0[i]*(double)x0[i];
68*a58d3d2aSXin Li    }
69*a58d3d2aSXin Li    snr = 20*log10(ener/err);
70*a58d3d2aSXin Li    printf ("SNR for size %d (%d pulses) is %f (was %f without inverse)\n", N, K, snr, snr0);
71*a58d3d2aSXin Li    if (snr < 60 || snr0 > 20)
72*a58d3d2aSXin Li    {
73*a58d3d2aSXin Li       fprintf(stderr, "FAIL!\n");
74*a58d3d2aSXin Li       ret = 1;
75*a58d3d2aSXin Li    }
76*a58d3d2aSXin Li }
77*a58d3d2aSXin Li 
main(void)78*a58d3d2aSXin Li int main(void)
79*a58d3d2aSXin Li {
80*a58d3d2aSXin Li    ALLOC_STACK;
81*a58d3d2aSXin Li    test_rotation(15, 3);
82*a58d3d2aSXin Li    test_rotation(23, 5);
83*a58d3d2aSXin Li    test_rotation(50, 3);
84*a58d3d2aSXin Li    test_rotation(80, 1);
85*a58d3d2aSXin Li    RESTORE_STACK;
86*a58d3d2aSXin Li    return ret;
87*a58d3d2aSXin Li }
88