c@409: #include <stdio.h>
c@409: #include <stdlib.h>
c@409: #include <sys/times.h>
c@409: #include <unistd.h>
c@409: #include "kiss_fft.h"
c@409: #include "kiss_fftr.h"
c@409: #include "kiss_fftnd.h"
c@409: #include "kiss_fftndr.h"
c@409: 
c@409: #include "pstats.h"
c@409: 
c@409: static
c@409: int getdims(int * dims, char * arg)
c@409: {
c@409:     char *s;
c@409:     int ndims=0;
c@409:     while ( (s=strtok( arg , ",") ) ) {
c@409:         dims[ndims++] = atoi(s);
c@409:         //printf("%s=%d\n",s,dims[ndims-1]);
c@409:         arg=NULL;
c@409:     }
c@409:     return ndims;
c@409: }
c@409: 
c@409: int main(int argc,char ** argv)
c@409: {
c@409:     int k;
c@409:     int nfft[32];
c@409:     int ndims = 1;
c@409:     int isinverse=0;
c@409:     int numffts=1000,i;
c@409:     kiss_fft_cpx * buf;
c@409:     kiss_fft_cpx * bufout;
c@409:     int real = 0;
c@409: 
c@409:     nfft[0] = 1024;// default
c@409: 
c@409:     while (1) {
c@409:         int c = getopt (argc, argv, "n:ix:r");
c@409:         if (c == -1)
c@409:             break;
c@409:         switch (c) {
c@409:             case 'r':
c@409:                 real = 1;
c@409:                 break;
c@409:             case 'n':
c@409:                 ndims = getdims(nfft, optarg );
c@409:                 if (nfft[0] != kiss_fft_next_fast_size(nfft[0]) ) {
c@409:                     int ng = kiss_fft_next_fast_size(nfft[0]);
c@409:                     fprintf(stderr,"warning: %d might be a better choice for speed than %d\n",ng,nfft[0]);
c@409:                 }
c@409:                 break;
c@409:             case 'x':
c@409:                 numffts = atoi (optarg);
c@409:                 break;
c@409:             case 'i':
c@409:                 isinverse = 1;
c@409:                 break;
c@409:         }
c@409:     }
c@409:     int nbytes = sizeof(kiss_fft_cpx);
c@409:     for (k=0;k<ndims;++k)
c@409:         nbytes *= nfft[k];
c@409: 
c@409: #ifdef USE_SIMD        
c@409:     numffts /= 4;
c@409:     fprintf(stderr,"since SIMD implementation does 4 ffts at a time, numffts is being reduced to %d\n",numffts);
c@409: #endif
c@409: 
c@409:     buf=(kiss_fft_cpx*)KISS_FFT_MALLOC(nbytes);
c@409:     bufout=(kiss_fft_cpx*)KISS_FFT_MALLOC(nbytes);
c@409:     memset(buf,0,nbytes);
c@409: 
c@409:     pstats_init();
c@409: 
c@409:     if (ndims==1) {
c@409:         if (real) {
c@409:             kiss_fftr_cfg st = kiss_fftr_alloc( nfft[0] ,isinverse ,0,0);
c@409:             if (isinverse)
c@409:                 for (i=0;i<numffts;++i)
c@409:                     kiss_fftri( st ,(kiss_fft_cpx*)buf,(kiss_fft_scalar*)bufout );
c@409:             else
c@409:                 for (i=0;i<numffts;++i)
c@409:                     kiss_fftr( st ,(kiss_fft_scalar*)buf,(kiss_fft_cpx*)bufout );
c@409:             free(st);
c@409:         }else{
c@409:             kiss_fft_cfg st = kiss_fft_alloc( nfft[0] ,isinverse ,0,0);
c@409:             for (i=0;i<numffts;++i)
c@409:                 kiss_fft( st ,buf,bufout );
c@409:             free(st);
c@409:         }
c@409:     }else{
c@409:         if (real) {
c@409:             kiss_fftndr_cfg st = kiss_fftndr_alloc( nfft,ndims ,isinverse ,0,0);
c@409:             if (isinverse)
c@409:                 for (i=0;i<numffts;++i)
c@409:                     kiss_fftndri( st ,(kiss_fft_cpx*)buf,(kiss_fft_scalar*)bufout );
c@409:             else
c@409:                 for (i=0;i<numffts;++i)
c@409:                     kiss_fftndr( st ,(kiss_fft_scalar*)buf,(kiss_fft_cpx*)bufout );
c@409:             free(st);
c@409:         }else{
c@409:             kiss_fftnd_cfg st= kiss_fftnd_alloc(nfft,ndims,isinverse ,0,0);
c@409:             for (i=0;i<numffts;++i)
c@409:                 kiss_fftnd( st ,buf,bufout );
c@409:             free(st);
c@409:         }
c@409:     }
c@409: 
c@409:     free(buf); free(bufout);
c@409: 
c@409:     fprintf(stderr,"KISS\tnfft=");
c@409:     for (k=0;k<ndims;++k)
c@409:         fprintf(stderr, "%d,",nfft[k]);
c@409:     fprintf(stderr,"\tnumffts=%d\n" ,numffts);
c@409:     pstats_report();
c@409: 
c@409:     kiss_fft_cleanup();
c@409: 
c@409:     return 0;
c@409: }
c@409: