web-dev-qa-db-fra.com

FFT dans un seul fichier C

Je cherchais une implémentation FFT en C. Cependant, je ne cherche pas une énorme bibliothèque (comme FFTW) mais une implémentation simple de fichier C simple à utiliser. Malheureusement, je n'ai rien trouvé de tel.

Quelqu'un peut-il recommander une mise en œuvre simple?

cfft
28
mijc

Votre meilleur pari est KissFFT - comme son nom l'indique c'est simple , mais il est toujours assez respectable et beaucoup plus léger que FFTW. Il est également gratuit, alors que FFTW exige des frais de licence élevés si vous souhaitez l'inclure dans un produit commercial.

25
Paul R

Ce fichier fonctionne correctement tel qu'il est: copiez-collez simplement sur votre ordinateur. Surfer sur le web, j'ai trouvé cette implémentation facile sur la page wikipedia ici . La page est en italien, j'ai donc réécrit le code avec quelques traductions. Ici il y a presque les mêmes informations mais en anglais. PRENDRE PLAISIR!

#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;
  }
  free(W);
}

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;
}
15
Leos313

Vous pouvez commencer à convertir this Java snippet to C, l'auteur déclare qu'il l'a converti à partir de C sur la base du livre recettes numériques que vous trouvez en ligne! ici

6
stacker

Voici une bibliothèque C sous licence permissive avec une variété d'implémentations FFT différentes , dont chacune est dans son propre fichier C autonome.

3
Connor McKay