2012-01-10 4 views
19

Stavo cercando un'implementazione FFT in C. Tuttavia, non sto cercando un'enorme libreria (come FFTW) ma per un'implementazione di file C singola e facile da usare. Purtroppo non sono stato in grado di trovare nulla di simile.FFT in un singolo file C

Qualcuno può consigliare un'implementazione semplice?

+0

Prova [alla ricerca di 'FFT' su github] (https://github.com/search?langOverride=c&q=fft&repo=&start_value=1&type=Repositories). Ma cosa non è facile da usare su FFTW? Intendi facile capire la fonte? – Rup

+9

Scrivi il tuo. Sarà un buon esercizio. Internet è pieno di spiegazioni su come calcolare DFT e FFT. Usalo. –

+1

Le routine FFT [qui] (http://www.fit.vutbr.cz/research/prod/?id=510) hanno meno di cento righe di codice. La libreria implementa gli algoritmi di trasformata veloce di Fourier (FFT) in avanti e inversa usando sia la decimazione nel tempo (DIT) che la decimazione in frequenza (DIF). – DaBler

risposta

20

La soluzione migliore è KissFFT - come suggerisce il nome è simple, ma è comunque abbastanza rispettabilmente veloce e molto più leggero di FFTW. È anche gratuito, mentre FFTW richiede una pesante quota di licenza se si desidera includerlo in un prodotto commerciale.

+1

Solo se lo stai ridistribuendo e non rilasciando la fonte in GPL ... –

+1

Questo è esattamente quello che stavo cercando. Grazie mille. – mijc

7

Questo file funziona correttamente così come è: basta copiare e incollare nel computer. Navigando sul web ho trovato questa facile implementazione sulla pagina di wikipedia here. La pagina è in italiano, quindi ho riscritto il codice con alcune traduzioni. Here ci sono quasi le stesse informazioni ma in inglese. GODERE!

#include <iostream> 
#include <complex> 
#define MAX 200 

using namespace std; 

#define M_PI 3.1415926535897932384 

int log2(int N) /*function to calculate the log2(.) of int numbers*/ 
{ 
    int k = N, i = 0; 
    while(k) { 
    k >>= 1; 
    i++; 
    } 
    return i - 1; 
} 

int check(int n) //checking if the number of element is a power of 2 
{ 
    return n > 0 && (n & (n - 1)) == 0; 
} 

int reverse(int N, int n) //calculating revers number 
{ 
    int j, p = 0; 
    for(j = 1; j <= log2(N); j++) { 
    if(n & (1 << (log2(N) - j))) 
     p |= 1 << (j - 1); 
    } 
    return p; 
} 

void ordina(complex<double>* f1, int N) //using the reverse order in the array 
{ 
    complex<double> f2[MAX]; 
    for(int i = 0; i < N; i++) 
    f2[i] = f1[reverse(N, i)]; 
    for(int j = 0; j < N; j++) 
    f1[j] = f2[j]; 
} 

void transform(complex<double>* f, int N) // 
{ 
    ordina(f, N); //first: reverse order 
    complex<double> *W; 
    W = (complex<double> *)malloc(N/2 * sizeof(complex<double>)); 
    W[1] = polar(1., -2. * M_PI/N); 
    W[0] = 1; 
    for(int i = 2; i < N/2; i++) 
    W[i] = pow(W[1], i); 
    int n = 1; 
    int a = N/2; 
    for(int j = 0; j < log2(N); j++) { 
    for(int i = 0; i < N; i++) { 
     if(!(i & n)) { 
     complex<double> temp = f[i]; 
     complex<double> Temp = W[(i * a) % (n * a)] * f[i + n]; 
     f[i] = temp + Temp; 
     f[i + n] = temp - Temp; 
     } 
    } 
    n *= 2; 
    a = a/2; 
    } 
} 

void FFT(complex<double>* f, int N, double d) 
{ 
    transform(f, N); 
    for(int i = 0; i < N; i++) 
    f[i] *= d; //multiplying by step 
} 

int main() 
{ 
    int n; 
    do { 
    cout << "specify array dimension (MUST be power of 2)" << endl; 
    cin >> n; 
    } while(!check(n)); 
    double d; 
    cout << "specify sampling step" << endl; //just write 1 in order to have the same results of matlab fft(.) 
    cin >> d; 
    complex<double> vec[MAX]; 
    cout << "specify the array" << endl; 
    for(int i = 0; i < n; i++) { 
    cout << "specify element number: " << i << endl; 
    cin >> vec[i]; 
    } 
    FFT(vec, n, d); 
    cout << "...printing the FFT of the array specified" << endl; 
    for(int j = 0; j < n; j++) 
    cout << vec[j] << endl; 
    return 0; 
} 
+2

Questo è sorprendentemente succinto in realtà. – Vortico

+0

facile da capire e, cosa più importante, funziona correttamente. Inoltre puoi facilmente modificarlo per adattarlo alla tua missione – Leos313

+0

come fare la FFT inversa con questo codice? – alexandrecosta