1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70
   |  
#include <fftw3.h>
#include <stdlib.h>
 
#define N 1024
#define DEMI_PER 16
#define AMPLITUDE 5
 
void show_signal(double * in, unsigned long size)
{
  unsigned long i;
 
  for( i = 0 ; i < size ; i++ )
    {
      fprintf(stdout," %f \n",in[i] );
    }
}
 
void show_fftw_complex(fftw_complex * in, unsigned long size)
{
  unsigned long i;
 
  for( i = 0 ; i < size ; i++ )
    {
      fprintf(stdout,"%f \n",in[i][0]);
    }
}
 
int main ( int argc, char **argv)
{
  double *in;
  unsigned long i,j = 0;
  fftw_complex * out;
  fftw_plan p;
 
  in = malloc( N * sizeof(double));
  out = fftw_malloc( N * sizeof(fftw_complex));
 
  /* Initialisation du plan de calcul pour la fft */
  p = fftw_plan_dft_r2c_1d(N,in,out,FFTW_ESTIMATE);
 
  /* Création d'un signal pseudo rectangle */
  for( i = 0 ; i < N ; i++)
    {
      if( i % DEMI_PER == 0 )
	{
	  if ( j == AMPLITUDE )
	    j = 0;
	  else
	    j = AMPLITUDE;
	}
      in[i] = j;
    }
 
  /* Affiche le signal d'entrée */
  show_signal(in,N);
 
  /* Calcule la fft */
  fftw_execute(p);
 
  /* Affiche la TF du signal d'entrée */
  show_fftw_complex(out,N/2);
 
  fftw_destroy_plan(p);
  free(in);
  fftw_free(out);
 
  return EXIT_SUCCESS;
 
} |