c@174: #include "kissfft.hh"
c@174: #include <iostream>
c@174: #include <cstdlib>
c@174: #include <typeinfo>
c@174: 
c@174: #include <sys/time.h>
c@174: static inline
c@174: double curtime(void)
c@174: {
c@174:     struct timeval tv;
c@174:     gettimeofday(&tv, NULL);
c@174:     return (double)tv.tv_sec + (double)tv.tv_usec*.000001;
c@174: }
c@174: 
c@174: using namespace std;
c@174: 
c@174: template <class T>
c@174: void dotest(int nfft)
c@174: {
c@174:     typedef kissfft<T> FFT;
c@174:     typedef std::complex<T> cpx_type;
c@174: 
c@174:     cout << "type:" << typeid(T).name() << " nfft:" << nfft;
c@174: 
c@174:     FFT fft(nfft,false);
c@174: 
c@174:     vector<cpx_type> inbuf(nfft);
c@174:     vector<cpx_type> outbuf(nfft);
c@174:     for (int k=0;k<nfft;++k)
c@174:         inbuf[k]= cpx_type( 
c@174:                 (T)(rand()/(double)RAND_MAX - .5),
c@174:                 (T)(rand()/(double)RAND_MAX - .5) );
c@174:     fft.transform( &inbuf[0] , &outbuf[0] );
c@174: 
c@174:     long double totalpower=0;
c@174:     long double difpower=0;
c@174:     for (int k0=0;k0<nfft;++k0) {
c@174:         complex<long double> acc = 0;
c@174:         long double phinc = 2*k0* M_PIl / nfft;
c@174:         for (int k1=0;k1<nfft;++k1) {
c@174:             complex<long double> x(inbuf[k1].real(),inbuf[k1].imag()); 
c@174:             acc += x * exp( complex<long double>(0,-k1*phinc) );
c@174:         }
c@174:         totalpower += norm(acc);
c@174:         complex<long double> x(outbuf[k0].real(),outbuf[k0].imag()); 
c@174:         complex<long double> dif = acc - x;
c@174:         difpower += norm(dif);
c@174:     }
c@174:     cout << " RMSE:" << sqrt(difpower/totalpower) << "\t";
c@174: 
c@174:     double t0 = curtime();
c@174:     int nits=20e6/nfft;
c@174:     for (int k=0;k<nits;++k) {
c@174:         fft.transform( &inbuf[0] , &outbuf[0] );
c@174:     }
c@174:     double t1 = curtime();
c@174:     cout << " MSPS:" << ( (nits*nfft)*1e-6/ (t1-t0) ) << endl;
c@174: }
c@174: 
c@174: int main(int argc,char ** argv)
c@174: {
c@174:     if (argc>1) {
c@174:         for (int k=1;k<argc;++k) {
c@174:             int nfft = atoi(argv[k]);
c@174:             dotest<float>(nfft); dotest<double>(nfft); dotest<long double>(nfft);
c@174:         }
c@174:     }else{
c@174:         dotest<float>(32); dotest<double>(32); dotest<long double>(32);
c@174:         dotest<float>(1024); dotest<double>(1024); dotest<long double>(1024);
c@174:         dotest<float>(840); dotest<double>(840); dotest<long double>(840);
c@174:     }
c@174:     return 0;
c@174: }