htswanalysis/MACS/lib/gsl/gsl-1.11/doc/examples/fft.c

   1 #include <stdio.h>
   2 #include <math.h>
   3 #include <gsl/gsl_errno.h>
   4 #include <gsl/gsl_fft_complex.h>
   5
   6 #define REAL(z,i) ((z)[2*(i)])
   7 #define IMAG(z,i) ((z)[2*(i)+1])
   8
   9 int
  10 main (void)
  11 {
  12   int i; double data[2*128];
  13
  14   for (i = 0; i < 128; i++)
  15     {
  16        REAL(data,i) = 0.0; IMAG(data,i) = 0.0;
  17     }
  18
  19   REAL(data,0) = 1.0;
  20
  21   for (i = 1; i <= 10; i++)
  22     {
  23        REAL(data,i) = REAL(data,128-i) = 1.0;
  24     }
  25
  26   for (i = 0; i < 128; i++)
  27     {
  28       printf ("%d %e %e\n", i,
  29               REAL(data,i), IMAG(data,i));
  30     }
  31   printf ("\n");
  32
  33   gsl_fft_complex_radix2_forward (data, 1, 128);
  34
  35   for (i = 0; i < 128; i++)
  36     {
  37       printf ("%d %e %e\n", i,
  38               REAL(data,i)/sqrt(128),
  39               IMAG(data,i)/sqrt(128));
  40     }
  41
  42   return 0;
  43 }