Chris@184: #include "kiss_fft.h"
Chris@184: 
Chris@184: 
Chris@184: void check(kiss_fft_cpx  * in,kiss_fft_cpx  * out,int nfft,int isinverse)
Chris@184: {
Chris@184:     int bin,k;
Chris@184:     double errpow=0,sigpow=0;
Chris@184:     
Chris@184:     for (bin=0;bin<nfft;++bin) {
Chris@184:         double ansr = 0;
Chris@184:         double ansi = 0;
Chris@184:         double difr;
Chris@184:         double difi;
Chris@184: 
Chris@184:         for (k=0;k<nfft;++k) {
Chris@184:             double phase = -2*M_PI*bin*k/nfft;
Chris@184:             double re = cos(phase);
Chris@184:             double im = sin(phase);
Chris@184:             if (isinverse)
Chris@184:                 im = -im;
Chris@184: 
Chris@184: #ifdef FIXED_POINT
Chris@184:             re /= nfft;
Chris@184:             im /= nfft;
Chris@184: #endif            
Chris@184: 
Chris@184:             ansr += in[k].r * re - in[k].i * im;
Chris@184:             ansi += in[k].r * im + in[k].i * re;
Chris@184:         }
Chris@184:         difr = ansr - out[bin].r;
Chris@184:         difi = ansi - out[bin].i;
Chris@184:         errpow += difr*difr + difi*difi;
Chris@184:         sigpow += ansr*ansr+ansi*ansi;
Chris@184:     }
Chris@184:     printf("nfft=%d inverse=%d,snr = %f\n",nfft,isinverse,10*log10(sigpow/errpow) );
Chris@184: }
Chris@184: 
Chris@184: void test1d(int nfft,int isinverse)
Chris@184: {
Chris@184:     size_t buflen = sizeof(kiss_fft_cpx)*nfft;
Chris@184: 
Chris@184:     kiss_fft_cpx  * in = (kiss_fft_cpx*)malloc(buflen);
Chris@184:     kiss_fft_cpx  * out= (kiss_fft_cpx*)malloc(buflen);
Chris@184:     kiss_fft_cfg  cfg = kiss_fft_alloc(nfft,isinverse,0,0);
Chris@184:     int k;
Chris@184: 
Chris@184:     for (k=0;k<nfft;++k) {
Chris@184:         in[k].r = (rand() % 65536) - 32768;
Chris@184:         in[k].i = (rand() % 65536) - 32768;
Chris@184:     }
Chris@184: 
Chris@184:     kiss_fft(cfg,in,out);
Chris@184: 
Chris@184:     check(in,out,nfft,isinverse);
Chris@184: 
Chris@184:     free(in);
Chris@184:     free(out);
Chris@184:     free(cfg);
Chris@184: }
Chris@184: 
Chris@184: int main(int argc,char ** argv)
Chris@184: {
Chris@184:     if (argc>1) {
Chris@184:         int k;
Chris@184:         for (k=1;k<argc;++k) {
Chris@184:             test1d(atoi(argv[k]),0);
Chris@184:             test1d(atoi(argv[k]),1);
Chris@184:         }
Chris@184:     }else{
Chris@184:         test1d(32,0);
Chris@184:         test1d(32,1);
Chris@184:     }
Chris@184:     return 0;
Chris@184: }