Making VBR slightly exceed the budget rather than fail
[opus.git] / tests / real-fft-test.c
index e0e5a1d..f4bb20b 100644 (file)
@@ -7,17 +7,20 @@
 #include <stdio.h>
 #include <string.h>
 
+#define CELT_C 
+#include "../libcelt/stack_alloc.h"
+#include "../libcelt/kiss_fft.c"
+#include "../libcelt/kiss_fftr.c"
+
+#ifdef FIXED_DEBUG
+long long celt_mips=0;
+#endif
 int ret=0;
 
 static
 kiss_fft_scalar rand_scalar(void) 
 {
-#ifdef USE_SIMD
-    return _mm_set1_ps(rand()-RAND_MAX/2);
-#else
-    kiss_fft_scalar s = (kiss_fft_scalar)(rand() -RAND_MAX/2);
-    return s/2;
-#endif
+    return (rand()%32767)-16384;
 }
 
 static
@@ -26,12 +29,15 @@ double snr_compare( kiss_fft_cpx * vec1,kiss_fft_scalar * vec2, int n)
     int k;
     double sigpow=1e-10, noisepow=1e-10, err,snr;
 
-    for (k=1;k<n;++k) {
+    vec1[0].i = vec1[n].r;
+    for (k=0;k<n;++k) {
         sigpow += (double)vec1[k].r * (double)vec1[k].r + 
                   (double)vec1[k].i * (double)vec1[k].i;
         err = (double)vec1[k].r - (double)vec2[2*k];
+        /*printf ("%f %f\n", (double)vec1[k].r, (double)vec2[2*k]);*/
         noisepow += err * err;
         err = (double)vec1[k].i - (double)vec2[2*k+1];
+        /*printf ("%f %f\n", (double)vec1[k].i, (double)vec2[2*k+1]);*/
         noisepow += err * err;
 
     }
@@ -49,7 +55,7 @@ double snr_compare_scal( kiss_fft_scalar * vec1,kiss_fft_scalar * vec2, int n)
     int k;
     double sigpow=1e-10, noisepow=1e-10, err,snr;
 
-    for (k=1;k<n;++k) {
+    for (k=0;k<n;++k) {
         sigpow += (double)vec1[k] * (double)vec1[k];
         err = (double)vec1[k] - (double)vec2[k];
         noisepow += err * err;
@@ -61,7 +67,11 @@ double snr_compare_scal( kiss_fft_scalar * vec1,kiss_fft_scalar * vec2, int n)
     }
     return snr;
 }
+#ifdef RADIX_TWO_ONLY
+#define NFFT 1024
+#else
 #define NFFT 8*3*5
+#endif
 
 #ifndef NUMFFTS
 #define NUMFFTS 10000
@@ -81,12 +91,14 @@ int main(void)
     kiss_fft_scalar rin[NFFT+2];
     kiss_fft_scalar rout[NFFT+2];
     kiss_fft_scalar zero;
+    ALLOC_STACK;
     memset(&zero,0,sizeof(zero) ); // ugly way of setting short,int,float,double, or __m128 to zero
 
-    srand(time(0));
-
     for (i=0;i<NFFT;++i) {
         rin[i] = rand_scalar();
+#if defined(FIXED_POINT) && defined(DOUBLE_PRECISION)
+        rin[i] *= 32768;
+#endif
         cin[i].r = rin[i];
         cin[i].i = zero;
     }
@@ -100,7 +112,6 @@ int main(void)
             NFFT,0, snr_compare(cout,sout,(NFFT/2)) );
 
     memset(cin,0,sizeof(cin));
-#if 1
     cin[0].r = rand_scalar();
     cin[NFFT/2].r = rand_scalar();
     for (i=1;i< NFFT/2;++i) {
@@ -108,11 +119,6 @@ int main(void)
         cin[i].r = rand_scalar();
         cin[i].i = rand_scalar();
     }
-#else
-    cin[0].r = 12000;
-    cin[3].r = 12000;
-    cin[NFFT/2].r = 12000;
-#endif
 
     // conjugate symmetry of real signal 
     for (i=1;i< NFFT/2;++i) {