c@409: #include "kiss_fftr.h"
c@409: #include "_kiss_fft_guts.h"
c@409: #include <sys/times.h>
c@409: #include <time.h>
c@409: #include <unistd.h>
c@409: 
c@409: static double cputime(void)
c@409: {
c@409:     struct tms t;
c@409:     times(&t);
c@409:     return (double)(t.tms_utime + t.tms_stime)/  sysconf(_SC_CLK_TCK) ;
c@409: }
c@409: 
c@409: static
c@409: kiss_fft_scalar rand_scalar(void) 
c@409: {
c@409: #ifdef USE_SIMD
c@409:     return _mm_set1_ps(rand()-RAND_MAX/2);
c@409: #else
c@409:     kiss_fft_scalar s = (kiss_fft_scalar)(rand() -RAND_MAX/2);
c@409:     return s/2;
c@409: #endif
c@409: }
c@409: 
c@409: static
c@409: double snr_compare( kiss_fft_cpx * vec1,kiss_fft_cpx * vec2, int n)
c@409: {
c@409:     int k;
c@409:     double sigpow=1e-10,noisepow=1e-10,err,snr,scale=0;
c@409: 
c@409: #ifdef USE_SIMD
c@409:     float *fv1 = (float*)vec1;
c@409:     float *fv2 = (float*)vec2;
c@409:     for (k=0;k<8*n;++k) {
c@409:         sigpow += *fv1 * *fv1;
c@409:         err = *fv1 - *fv2;
c@409:         noisepow += err*err;
c@409:         ++fv1;
c@409:         ++fv2;
c@409:     }
c@409: #else
c@409:     for (k=0;k<n;++k) {
c@409:         sigpow += (double)vec1[k].r * (double)vec1[k].r + 
c@409:                   (double)vec1[k].i * (double)vec1[k].i;
c@409:         err = (double)vec1[k].r - (double)vec2[k].r;
c@409:         noisepow += err * err;
c@409:         err = (double)vec1[k].i - (double)vec2[k].i;
c@409:         noisepow += err * err;
c@409: 
c@409:         if (vec1[k].r)
c@409:             scale +=(double) vec2[k].r / (double)vec1[k].r;
c@409:     }
c@409: #endif    
c@409:     snr = 10*log10( sigpow / noisepow );
c@409:     scale /= n;
c@409:     if (snr<10) {
c@409:         printf( "\npoor snr, try a scaling factor %f\n" , scale );
c@409:         exit(1);
c@409:     }
c@409:     return snr;
c@409: }
c@409: 
c@409: #ifndef NUMFFTS
c@409: #define NUMFFTS 10000
c@409: #endif
c@409: 
c@409: 
c@409: int main(int argc,char ** argv)
c@409: {
c@409:     int nfft = 8*3*5;
c@409:     double ts,tfft,trfft;
c@409:     int i;
c@409:     if (argc>1)
c@409:         nfft = atoi(argv[1]);
c@409:     kiss_fft_cpx cin[nfft];
c@409:     kiss_fft_cpx cout[nfft];
c@409:     kiss_fft_cpx sout[nfft];
c@409:     kiss_fft_cfg  kiss_fft_state;
c@409:     kiss_fftr_cfg  kiss_fftr_state;
c@409: 
c@409:     kiss_fft_scalar rin[nfft+2];
c@409:     kiss_fft_scalar rout[nfft+2];
c@409:     kiss_fft_scalar zero;
c@409:     memset(&zero,0,sizeof(zero) ); // ugly way of setting short,int,float,double, or __m128 to zero
c@409: 
c@409:     srand(time(0));
c@409: 
c@409:     for (i=0;i<nfft;++i) {
c@409:         rin[i] = rand_scalar();
c@409:         cin[i].r = rin[i];
c@409:         cin[i].i = zero;
c@409:     }
c@409: 
c@409:     kiss_fft_state = kiss_fft_alloc(nfft,0,0,0);
c@409:     kiss_fftr_state = kiss_fftr_alloc(nfft,0,0,0);
c@409:     kiss_fft(kiss_fft_state,cin,cout);
c@409:     kiss_fftr(kiss_fftr_state,rin,sout);
c@409:     /*
c@409:     printf(" results from kiss_fft : (%f,%f), (%f,%f), (%f,%f) ...\n "
c@409:             , (float)cout[0].r , (float)cout[0].i
c@409:             , (float)cout[1].r , (float)cout[1].i
c@409:             , (float)cout[2].r , (float)cout[2].i); 
c@409:     printf(" results from kiss_fftr: (%f,%f), (%f,%f), (%f,%f) ...\n "
c@409:             , (float)sout[0].r , (float)sout[0].i
c@409:             , (float)sout[1].r , (float)sout[1].i
c@409:             , (float)sout[2].r , (float)sout[2].i); 
c@409:     */
c@409:         
c@409:     printf( "nfft=%d, inverse=%d, snr=%g\n",
c@409:             nfft,0, snr_compare(cout,sout,(nfft/2)+1) );
c@409:     ts = cputime();
c@409:     for (i=0;i<NUMFFTS;++i) {
c@409:         kiss_fft(kiss_fft_state,cin,cout);
c@409:     }
c@409:     tfft = cputime() - ts;
c@409:     
c@409:     ts = cputime();
c@409:     for (i=0;i<NUMFFTS;++i) {
c@409:         kiss_fftr( kiss_fftr_state, rin, cout );
c@409:         /* kiss_fftri(kiss_fftr_state,cout,rin); */
c@409:     }
c@409:     trfft = cputime() - ts;
c@409: 
c@409:     printf("%d complex ffts took %gs, real took %gs\n",NUMFFTS,tfft,trfft);
c@409: 
c@409:     free(kiss_fft_state);
c@409:     free(kiss_fftr_state);
c@409: 
c@409:     kiss_fft_state = kiss_fft_alloc(nfft,1,0,0);
c@409:     kiss_fftr_state = kiss_fftr_alloc(nfft,1,0,0);
c@409: 
c@409:     memset(cin,0,sizeof(cin));
c@409: #if 1
c@409:     for (i=1;i< nfft/2;++i) {
c@409:         //cin[i].r = (kiss_fft_scalar)(rand()-RAND_MAX/2);
c@409:         cin[i].r = rand_scalar();
c@409:         cin[i].i = rand_scalar();
c@409:     }
c@409: #else
c@409:     cin[0].r = 12000;
c@409:     cin[3].r = 12000;
c@409:     cin[nfft/2].r = 12000;
c@409: #endif
c@409: 
c@409:     // conjugate symmetry of real signal 
c@409:     for (i=1;i< nfft/2;++i) {
c@409:         cin[nfft-i].r = cin[i].r;
c@409:         cin[nfft-i].i = - cin[i].i;
c@409:     }
c@409: 
c@409:     kiss_fft(kiss_fft_state,cin,cout);
c@409:     kiss_fftri(kiss_fftr_state,cin,rout);
c@409:     /*
c@409:     printf(" results from inverse kiss_fft : (%f,%f), (%f,%f), (%f,%f), (%f,%f), (%f,%f) ...\n "
c@409:             , (float)cout[0].r , (float)cout[0].i , (float)cout[1].r , (float)cout[1].i , (float)cout[2].r , (float)cout[2].i , (float)cout[3].r , (float)cout[3].i , (float)cout[4].r , (float)cout[4].i
c@409:             ); 
c@409: 
c@409:     printf(" results from inverse kiss_fftr: %f,%f,%f,%f,%f ... \n"
c@409:             ,(float)rout[0] ,(float)rout[1] ,(float)rout[2] ,(float)rout[3] ,(float)rout[4]);
c@409: */
c@409:     for (i=0;i<nfft;++i) {
c@409:         sout[i].r = rout[i];
c@409:         sout[i].i = zero;
c@409:     }
c@409: 
c@409:     printf( "nfft=%d, inverse=%d, snr=%g\n",
c@409:             nfft,1, snr_compare(cout,sout,nfft/2) );
c@409:     free(kiss_fft_state);
c@409:     free(kiss_fftr_state);
c@409: 
c@409:     return 0;
c@409: }