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: #include <stdio.h>
cannam@154: 
cannam@154: #include "mdct.h"
cannam@154: #include "stack_alloc.h"
cannam@154: #include "kiss_fft.h"
cannam@154: #include "mdct.h"
cannam@154: #include "modes.h"
cannam@154: 
cannam@154: #ifndef M_PI
cannam@154: #define M_PI 3.141592653
cannam@154: #endif
cannam@154: 
cannam@154: int ret = 0;
cannam@154: void check(kiss_fft_scalar  * in,kiss_fft_scalar  * out,int nfft,int isinverse)
cannam@154: {
cannam@154:     int bin,k;
cannam@154:     double errpow=0,sigpow=0;
cannam@154:     double snr;
cannam@154:     for (bin=0;bin<nfft/2;++bin) {
cannam@154:         double ansr = 0;
cannam@154:         double difr;
cannam@154: 
cannam@154:         for (k=0;k<nfft;++k) {
cannam@154:            double phase = 2*M_PI*(k+.5+.25*nfft)*(bin+.5)/nfft;
cannam@154:            double re = cos(phase);
cannam@154: 
cannam@154:            re /= nfft/4;
cannam@154: 
cannam@154:            ansr += in[k] * re;
cannam@154:         }
cannam@154:         /*printf ("%f %f\n", ansr, out[bin]);*/
cannam@154:         difr = ansr - out[bin];
cannam@154:         errpow += difr*difr;
cannam@154:         sigpow += ansr*ansr;
cannam@154:     }
cannam@154:     snr = 10*log10(sigpow/errpow);
cannam@154:     printf("nfft=%d inverse=%d,snr = %f\n",nfft,isinverse,snr );
cannam@154:     if (snr<60) {
cannam@154:        printf( "** poor snr: %f **\n", snr);
cannam@154:        ret = 1;
cannam@154:     }
cannam@154: }
cannam@154: 
cannam@154: void check_inv(kiss_fft_scalar  * in,kiss_fft_scalar  * out,int nfft,int isinverse)
cannam@154: {
cannam@154:    int bin,k;
cannam@154:    double errpow=0,sigpow=0;
cannam@154:    double snr;
cannam@154:    for (bin=0;bin<nfft;++bin) {
cannam@154:       double ansr = 0;
cannam@154:       double difr;
cannam@154: 
cannam@154:       for (k=0;k<nfft/2;++k) {
cannam@154:          double phase = 2*M_PI*(bin+.5+.25*nfft)*(k+.5)/nfft;
cannam@154:          double re = cos(phase);
cannam@154: 
cannam@154:          /*re *= 2;*/
cannam@154: 
cannam@154:          ansr += in[k] * re;
cannam@154:       }
cannam@154:       /*printf ("%f %f\n", ansr, out[bin]);*/
cannam@154:       difr = ansr - out[bin];
cannam@154:       errpow += difr*difr;
cannam@154:       sigpow += ansr*ansr;
cannam@154:    }
cannam@154:    snr = 10*log10(sigpow/errpow);
cannam@154:    printf("nfft=%d inverse=%d,snr = %f\n",nfft,isinverse,snr );
cannam@154:    if (snr<60) {
cannam@154:       printf( "** poor snr: %f **\n", snr);
cannam@154:       ret = 1;
cannam@154:    }
cannam@154: }
cannam@154: 
cannam@154: 
cannam@154: void test1d(int nfft,int isinverse,int arch)
cannam@154: {
cannam@154:     size_t buflen = sizeof(kiss_fft_scalar)*nfft;
cannam@154:     kiss_fft_scalar *in;
cannam@154:     kiss_fft_scalar *in_copy;
cannam@154:     kiss_fft_scalar *out;
cannam@154:     opus_val16 *window;
cannam@154:     int k;
cannam@154: 
cannam@154: #ifdef CUSTOM_MODES
cannam@154:     int shift = 0;
cannam@154:     const mdct_lookup *cfg;
cannam@154:     mdct_lookup _cfg;
cannam@154:     clt_mdct_init(&_cfg, nfft, 0, arch);
cannam@154:     cfg = &_cfg;
cannam@154: #else
cannam@154:     int shift;
cannam@154:     const mdct_lookup *cfg;
cannam@154:     CELTMode *mode = opus_custom_mode_create(48000, 960, NULL);
cannam@154:     if (nfft == 1920) shift = 0;
cannam@154:     else if (nfft == 960) shift = 1;
cannam@154:     else if (nfft == 480) shift = 2;
cannam@154:     else if (nfft == 240) shift = 3;
cannam@154:     else return;
cannam@154:     cfg = &mode->mdct;
cannam@154: #endif
cannam@154: 
cannam@154:     in = (kiss_fft_scalar*)malloc(buflen);
cannam@154:     in_copy = (kiss_fft_scalar*)malloc(buflen);
cannam@154:     out = (kiss_fft_scalar*)malloc(buflen);
cannam@154:     window = (opus_val16*)malloc(sizeof(opus_val16)*nfft/2);
cannam@154: 
cannam@154:     for (k=0;k<nfft;++k) {
cannam@154:         in[k] = (rand() % 32768) - 16384;
cannam@154:     }
cannam@154: 
cannam@154:     for (k=0;k<nfft/2;++k) {
cannam@154:        window[k] = Q15ONE;
cannam@154:     }
cannam@154:     for (k=0;k<nfft;++k) {
cannam@154:        in[k] *= 32768;
cannam@154:     }
cannam@154: 
cannam@154:     if (isinverse)
cannam@154:     {
cannam@154:        for (k=0;k<nfft;++k) {
cannam@154:           in[k] /= nfft;
cannam@154:        }
cannam@154:     }
cannam@154: 
cannam@154:     for (k=0;k<nfft;++k)
cannam@154:        in_copy[k] = in[k];
cannam@154:     /*for (k=0;k<nfft;++k) printf("%d %d ", in[k].r, in[k].i);printf("\n");*/
cannam@154: 
cannam@154:     if (isinverse)
cannam@154:     {
cannam@154:        for (k=0;k<nfft;++k)
cannam@154:           out[k] = 0;
cannam@154:        clt_mdct_backward(cfg,in,out, window, nfft/2, shift, 1, arch);
cannam@154:        /* apply TDAC because clt_mdct_backward() no longer does that */
cannam@154:        for (k=0;k<nfft/4;++k)
cannam@154:           out[nfft-k-1] = out[nfft/2+k];
cannam@154:        check_inv(in,out,nfft,isinverse);
cannam@154:     } else {
cannam@154:        clt_mdct_forward(cfg,in,out,window, nfft/2, shift, 1, arch);
cannam@154:        check(in_copy,out,nfft,isinverse);
cannam@154:     }
cannam@154:     /*for (k=0;k<nfft;++k) printf("%d %d ", out[k].r, out[k].i);printf("\n");*/
cannam@154: 
cannam@154: 
cannam@154:     free(in);
cannam@154:     free(in_copy);
cannam@154:     free(out);
cannam@154:     free(window);
cannam@154: #ifdef CUSTOM_MODES
cannam@154:     clt_mdct_clear(&_cfg, arch);
cannam@154: #endif
cannam@154: }
cannam@154: 
cannam@154: int main(int argc,char ** argv)
cannam@154: {
cannam@154:     ALLOC_STACK;
cannam@154:     int arch = opus_select_arch();
cannam@154: 
cannam@154:     if (argc>1) {
cannam@154:         int k;
cannam@154:         for (k=1;k<argc;++k) {
cannam@154:             test1d(atoi(argv[k]),0,arch);
cannam@154:             test1d(atoi(argv[k]),1,arch);
cannam@154:         }
cannam@154:     }else{
cannam@154:         test1d(32,0,arch);
cannam@154:         test1d(32,1,arch);
cannam@154:         test1d(256,0,arch);
cannam@154:         test1d(256,1,arch);
cannam@154:         test1d(512,0,arch);
cannam@154:         test1d(512,1,arch);
cannam@154:         test1d(1024,0,arch);
cannam@154:         test1d(1024,1,arch);
cannam@154:         test1d(2048,0,arch);
cannam@154:         test1d(2048,1,arch);
cannam@154: #ifndef RADIX_TWO_ONLY
cannam@154:         test1d(36,0,arch);
cannam@154:         test1d(36,1,arch);
cannam@154:         test1d(40,0,arch);
cannam@154:         test1d(40,1,arch);
cannam@154:         test1d(60,0,arch);
cannam@154:         test1d(60,1,arch);
cannam@154:         test1d(120,0,arch);
cannam@154:         test1d(120,1,arch);
cannam@154:         test1d(240,0,arch);
cannam@154:         test1d(240,1,arch);
cannam@154:         test1d(480,0,arch);
cannam@154:         test1d(480,1,arch);
cannam@154:         test1d(960,0,arch);
cannam@154:         test1d(960,1,arch);
cannam@154:         test1d(1920,0,arch);
cannam@154:         test1d(1920,1,arch);
cannam@154: #endif
cannam@154:     }
cannam@154:     return ret;
cannam@154: }