cannam@154: /* Copyright (c) 2008-2011 Xiph.Org Foundation
cannam@154:    Written by Jean-Marc Valin */
cannam@154: /*
cannam@154:    Redistribution and use in source and binary forms, with or without
cannam@154:    modification, are permitted provided that the following conditions
cannam@154:    are met:
cannam@154: 
cannam@154:    - Redistributions of source code must retain the above copyright
cannam@154:    notice, this list of conditions and the following disclaimer.
cannam@154: 
cannam@154:    - Redistributions in binary form must reproduce the above copyright
cannam@154:    notice, this list of conditions and the following disclaimer in the
cannam@154:    documentation and/or other materials provided with the distribution.
cannam@154: 
cannam@154:    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
cannam@154:    ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
cannam@154:    LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
cannam@154:    A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
cannam@154:    OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
cannam@154:    EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
cannam@154:    PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
cannam@154:    PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
cannam@154:    LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
cannam@154:    NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
cannam@154:    SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
cannam@154: */
cannam@154: 
cannam@154: #ifdef HAVE_CONFIG_H
cannam@154: #include "config.h"
cannam@154: #endif
cannam@154: 
cannam@154: #ifndef CUSTOM_MODES
cannam@154: #define CUSTOM_MODES
cannam@154: #endif
cannam@154: 
cannam@154: #include <stdio.h>
cannam@154: #include <stdlib.h>
cannam@154: #include "vq.h"
cannam@154: #include "bands.h"
cannam@154: #include "stack_alloc.h"
cannam@154: #include <math.h>
cannam@154: 
cannam@154: 
cannam@154: #define MAX_SIZE 100
cannam@154: 
cannam@154: int ret=0;
cannam@154: void test_rotation(int N, int K)
cannam@154: {
cannam@154:    int i;
cannam@154:    double err = 0, ener = 0, snr, snr0;
cannam@154:    opus_val16 x0[MAX_SIZE];
cannam@154:    opus_val16 x1[MAX_SIZE];
cannam@154:    for (i=0;i<N;i++)
cannam@154:       x1[i] = x0[i] = rand()%32767-16384;
cannam@154:    exp_rotation(x1, N, 1, 1, K, SPREAD_NORMAL);
cannam@154:    for (i=0;i<N;i++)
cannam@154:    {
cannam@154:       err += (x0[i]-(double)x1[i])*(x0[i]-(double)x1[i]);
cannam@154:       ener += x0[i]*(double)x0[i];
cannam@154:    }
cannam@154:    snr0 = 20*log10(ener/err);
cannam@154:    err = ener = 0;
cannam@154:    exp_rotation(x1, N, -1, 1, K, SPREAD_NORMAL);
cannam@154:    for (i=0;i<N;i++)
cannam@154:    {
cannam@154:       err += (x0[i]-(double)x1[i])*(x0[i]-(double)x1[i]);
cannam@154:       ener += x0[i]*(double)x0[i];
cannam@154:    }
cannam@154:    snr = 20*log10(ener/err);
cannam@154:    printf ("SNR for size %d (%d pulses) is %f (was %f without inverse)\n", N, K, snr, snr0);
cannam@154:    if (snr < 60 || snr0 > 20)
cannam@154:    {
cannam@154:       fprintf(stderr, "FAIL!\n");
cannam@154:       ret = 1;
cannam@154:    }
cannam@154: }
cannam@154: 
cannam@154: int main(void)
cannam@154: {
cannam@154:    ALLOC_STACK;
cannam@154:    test_rotation(15, 3);
cannam@154:    test_rotation(23, 5);
cannam@154:    test_rotation(50, 3);
cannam@154:    test_rotation(80, 1);
cannam@154:    return ret;
cannam@154: }