nuclear@0: #ifndef KISS_FFT_H nuclear@0: #define KISS_FFT_H nuclear@0: nuclear@0: #include nuclear@0: #include nuclear@0: #include nuclear@0: #include nuclear@0: nuclear@0: #ifdef __cplusplus nuclear@0: extern "C" { nuclear@0: #endif nuclear@0: nuclear@0: /* nuclear@0: ATTENTION! nuclear@0: If you would like a : nuclear@0: -- a utility that will handle the caching of fft objects nuclear@0: -- real-only (no imaginary time component ) FFT nuclear@0: -- a multi-dimensional FFT nuclear@0: -- a command-line utility to perform ffts nuclear@0: -- a command-line utility to perform fast-convolution filtering nuclear@0: nuclear@0: Then see kfc.h kiss_fftr.h kiss_fftnd.h fftutil.c kiss_fastfir.c nuclear@0: in the tools/ directory. nuclear@0: */ nuclear@0: nuclear@0: #ifdef USE_SIMD nuclear@0: # include nuclear@0: # define kiss_fft_scalar __m128 nuclear@0: #define KISS_FFT_MALLOC(nbytes) _mm_malloc(nbytes,16) nuclear@0: #define KISS_FFT_FREE _mm_free nuclear@0: #else nuclear@0: #define KISS_FFT_MALLOC malloc nuclear@0: #define KISS_FFT_FREE free nuclear@0: #endif nuclear@0: nuclear@0: nuclear@0: #ifdef FIXED_POINT nuclear@0: #include nuclear@0: # if (FIXED_POINT == 32) nuclear@0: # define kiss_fft_scalar int32_t nuclear@0: # else nuclear@0: # define kiss_fft_scalar int16_t nuclear@0: # endif nuclear@0: #else nuclear@0: # ifndef kiss_fft_scalar nuclear@0: /* default is float */ nuclear@0: # define kiss_fft_scalar float nuclear@0: # endif nuclear@0: #endif nuclear@0: nuclear@0: typedef struct { nuclear@0: kiss_fft_scalar r; nuclear@0: kiss_fft_scalar i; nuclear@0: }kiss_fft_cpx; nuclear@0: nuclear@0: typedef struct kiss_fft_state* kiss_fft_cfg; nuclear@0: nuclear@0: /* nuclear@0: * kiss_fft_alloc nuclear@0: * nuclear@0: * Initialize a FFT (or IFFT) algorithm's cfg/state buffer. nuclear@0: * nuclear@0: * typical usage: kiss_fft_cfg mycfg=kiss_fft_alloc(1024,0,NULL,NULL); nuclear@0: * nuclear@0: * The return value from fft_alloc is a cfg buffer used internally nuclear@0: * by the fft routine or NULL. nuclear@0: * nuclear@0: * If lenmem is NULL, then kiss_fft_alloc will allocate a cfg buffer using malloc. nuclear@0: * The returned value should be free()d when done to avoid memory leaks. nuclear@0: * nuclear@0: * The state can be placed in a user supplied buffer 'mem': nuclear@0: * If lenmem is not NULL and mem is not NULL and *lenmem is large enough, nuclear@0: * then the function places the cfg in mem and the size used in *lenmem nuclear@0: * and returns mem. nuclear@0: * nuclear@0: * If lenmem is not NULL and ( mem is NULL or *lenmem is not large enough), nuclear@0: * then the function returns NULL and places the minimum cfg nuclear@0: * buffer size in *lenmem. nuclear@0: * */ nuclear@0: nuclear@0: kiss_fft_cfg kiss_fft_alloc(int nfft,int inverse_fft,void * mem,size_t * lenmem); nuclear@0: nuclear@0: /* nuclear@0: * kiss_fft(cfg,in_out_buf) nuclear@0: * nuclear@0: * Perform an FFT on a complex input buffer. nuclear@0: * for a forward FFT, nuclear@0: * fin should be f[0] , f[1] , ... ,f[nfft-1] nuclear@0: * fout will be F[0] , F[1] , ... ,F[nfft-1] nuclear@0: * Note that each element is complex and can be accessed like nuclear@0: f[k].r and f[k].i nuclear@0: * */ nuclear@0: void kiss_fft(kiss_fft_cfg cfg,const kiss_fft_cpx *fin,kiss_fft_cpx *fout); nuclear@0: nuclear@0: /* nuclear@0: A more generic version of the above function. It reads its input from every Nth sample. nuclear@0: * */ nuclear@0: void kiss_fft_stride(kiss_fft_cfg cfg,const kiss_fft_cpx *fin,kiss_fft_cpx *fout,int fin_stride); nuclear@0: nuclear@0: /* If kiss_fft_alloc allocated a buffer, it is one contiguous nuclear@0: buffer and can be simply free()d when no longer needed*/ nuclear@0: #define kiss_fft_free free nuclear@0: nuclear@0: /* nuclear@0: Cleans up some memory that gets managed internally. Not necessary to call, but it might clean up nuclear@0: your compiler output to call this before you exit. nuclear@0: */ nuclear@0: void kiss_fft_cleanup(void); nuclear@0: nuclear@0: nuclear@0: /* nuclear@0: * Returns the smallest integer k, such that k>=n and k has only "fast" factors (2,3,5) nuclear@0: */ nuclear@0: int kiss_fft_next_fast_size(int n); nuclear@0: nuclear@0: /* for real ffts, we need an even size */ nuclear@0: #define kiss_fftr_next_fast_size_real(n) \ nuclear@0: (kiss_fft_next_fast_size( ((n)+1)>>1)<<1) nuclear@0: nuclear@0: #ifdef __cplusplus nuclear@0: } nuclear@0: #endif nuclear@0: nuclear@0: #endif