New VQ search is now enabled by default after fixing the last remaining issues:
[opus.git] / tests / dft-test.c
index ac03624..4e9c0ab 100644 (file)
@@ -5,11 +5,16 @@
 #include <stdio.h>
 #include "kiss_fft.h"
 
+#ifndef M_PI
+#define M_PI 3.141592653
+#endif
+
+int ret = 0;
 
 void check(kiss_fft_cpx  * in,kiss_fft_cpx  * out,int nfft,int isinverse)
 {
     int bin,k;
-    double errpow=0,sigpow=0;
+    double errpow=0,sigpow=0, snr;
     
     for (bin=0;bin<nfft;++bin) {
         double ansr = 0;
@@ -24,13 +29,11 @@ void check(kiss_fft_cpx  * in,kiss_fft_cpx  * out,int nfft,int isinverse)
             if (isinverse)
                 im = -im;
 
-#ifdef FIXED_POINT
             if (!isinverse)
             {
                re /= nfft;
                im /= nfft;
             }
-#endif            
 
             ansr += in[k].r * re - in[k].i * im;
             ansi += in[k].r * im + in[k].i * re;
@@ -41,7 +44,12 @@ void check(kiss_fft_cpx  * in,kiss_fft_cpx  * out,int nfft,int isinverse)
         errpow += difr*difr + difi*difi;
         sigpow += ansr*ansr+ansi*ansi;
     }
-    printf("nfft=%d inverse=%d,snr = %f\n",nfft,isinverse,10*log10(sigpow/errpow) );
+    snr = 10*log10(sigpow/errpow);
+    printf("nfft=%d inverse=%d,snr = %f\n",nfft,isinverse,snr );
+    if (snr<60) {
+       printf( "** poor snr: %f ** \n", snr);
+       ret = 1;
+    }
 }
 
 void test1d(int nfft,int isinverse)
@@ -54,14 +62,14 @@ void test1d(int nfft,int isinverse)
     int k;
 
     for (k=0;k<nfft;++k) {
-        in[k].r = (rand() % 65536) - 32768;
-        in[k].i = (rand() % 65536) - 32768;
+        in[k].r = (rand() % 32767) - 16384;
+        in[k].i = (rand() % 32767) - 16384;
     }
 
 #ifdef DOUBLE_PRECISION
     for (k=0;k<nfft;++k) {
-       in[k].r *= 65536;
-       in[k].i *= 65536;
+       in[k].r *= 32768;
+       in[k].i *= 32768;
     }
 #endif
     
@@ -100,6 +108,11 @@ int main(int argc,char ** argv)
     }else{
         test1d(32,0);
         test1d(32,1);
+        test1d(128,0);
+        test1d(128,1);
+        test1d(256,0);
+        test1d(256,1);
+#ifndef RADIX_TWO_ONLY
         test1d(36,0);
         test1d(36,1);
         test1d(50,0);
@@ -108,6 +121,7 @@ int main(int argc,char ** argv)
         test1d(120,1);
         test1d(105,0);
         test1d(105,1);
+#endif
     }
-    return 0;
+    return ret;
 }