cannam@127: cannam@127: cannam@127: cannam@127: cannam@127:
cannam@127:cannam@127: Next: Complex Multi-Dimensional DFTs, Previous: Tutorial, Up: Tutorial [Contents][Index]
cannam@127:cannam@127:cannam@127: cannam@127: cannam@127:Plan: To bother about the best method of accomplishing an accidental result. cannam@127: [Ambrose Bierce, The Enlarged Devil’s Dictionary.] cannam@127: cannam@127:
The basic usage of FFTW to compute a one-dimensional DFT of size
cannam@127: N
is simple, and it typically looks something like this code:
cannam@127:
#include <fftw3.h>
cannam@127: ...
cannam@127: {
cannam@127: fftw_complex *in, *out;
cannam@127: fftw_plan p;
cannam@127: ...
cannam@127: in = (fftw_complex*) fftw_malloc(sizeof(fftw_complex) * N);
cannam@127: out = (fftw_complex*) fftw_malloc(sizeof(fftw_complex) * N);
cannam@127: p = fftw_plan_dft_1d(N, in, out, FFTW_FORWARD, FFTW_ESTIMATE);
cannam@127: ...
cannam@127: fftw_execute(p); /* repeat as needed */
cannam@127: ...
cannam@127: fftw_destroy_plan(p);
cannam@127: fftw_free(in); fftw_free(out);
cannam@127: }
cannam@127:
You must link this code with the fftw3
library. On Unix systems,
cannam@127: link with -lfftw3 -lm
.
cannam@127:
The example code first allocates the input and output arrays. You can
cannam@127: allocate them in any way that you like, but we recommend using
cannam@127: fftw_malloc
, which behaves like
cannam@127:
cannam@127: malloc
except that it properly aligns the array when SIMD
cannam@127: instructions (such as SSE and Altivec) are available (see SIMD alignment and fftw_malloc). [Alternatively, we provide a convenient wrapper function fftw_alloc_complex(N)
which has the same effect.]
cannam@127:
cannam@127:
cannam@127:
The data is an array of type fftw_complex
, which is by default a
cannam@127: double[2]
composed of the real (in[i][0]
) and imaginary
cannam@127: (in[i][1]
) parts of a complex number.
cannam@127:
cannam@127:
The next step is to create a plan, which is an object cannam@127: cannam@127: that contains all the data that FFTW needs to compute the FFT. cannam@127: This function creates the plan: cannam@127:
cannam@127:fftw_plan fftw_plan_dft_1d(int n, fftw_complex *in, fftw_complex *out, cannam@127: int sign, unsigned flags); cannam@127:
The first argument, n
, is the size of the transform you are
cannam@127: trying to compute. The size n
can be any positive integer, but
cannam@127: sizes that are products of small factors are transformed most
cannam@127: efficiently (although prime sizes still use an O(n log n) algorithm).
cannam@127:
The next two arguments are pointers to the input and output arrays of cannam@127: the transform. These pointers can be equal, indicating an cannam@127: in-place transform. cannam@127: cannam@127:
cannam@127: cannam@127:The fourth argument, sign
, can be either FFTW_FORWARD
cannam@127: (-1
) or FFTW_BACKWARD
(+1
),
cannam@127:
cannam@127:
cannam@127: and indicates the direction of the transform you are interested in;
cannam@127: technically, it is the sign of the exponent in the transform.
cannam@127:
The flags
argument is usually either FFTW_MEASURE
or
cannam@127:
cannam@127: FFTW_ESTIMATE
. FFTW_MEASURE
instructs FFTW to run
cannam@127:
cannam@127: and measure the execution time of several FFTs in order to find the
cannam@127: best way to compute the transform of size n
. This process takes
cannam@127: some time (usually a few seconds), depending on your machine and on
cannam@127: the size of the transform. FFTW_ESTIMATE
, on the contrary,
cannam@127: does not run any computation and just builds a
cannam@127:
cannam@127: reasonable plan that is probably sub-optimal. In short, if your
cannam@127: program performs many transforms of the same size and initialization
cannam@127: time is not important, use FFTW_MEASURE
; otherwise use the
cannam@127: estimate.
cannam@127:
You must create the plan before initializing the input, because
cannam@127: FFTW_MEASURE
overwrites the in
/out
arrays.
cannam@127: (Technically, FFTW_ESTIMATE
does not touch your arrays, but you
cannam@127: should always create plans first just to be sure.)
cannam@127:
Once the plan has been created, you can use it as many times as you
cannam@127: like for transforms on the specified in
/out
arrays,
cannam@127: computing the actual transforms via fftw_execute(plan)
:
cannam@127:
void fftw_execute(const fftw_plan plan); cannam@127:
The DFT results are stored in-order in the array out
, with the
cannam@127: zero-frequency (DC) component in out[0]
.
cannam@127:
cannam@127: If in != out
, the transform is out-of-place and the input
cannam@127: array in
is not modified. Otherwise, the input array is
cannam@127: overwritten with the transform.
cannam@127:
If you want to transform a different array of the same size, you
cannam@127: can create a new plan with fftw_plan_dft_1d
and FFTW
cannam@127: automatically reuses the information from the previous plan, if
cannam@127: possible. Alternatively, with the “guru” interface you can apply a
cannam@127: given plan to a different array, if you are careful.
cannam@127: See FFTW Reference.
cannam@127:
When you are done with the plan, you deallocate it by calling
cannam@127: fftw_destroy_plan(plan)
:
cannam@127:
void fftw_destroy_plan(fftw_plan plan); cannam@127:
If you allocate an array with fftw_malloc()
you must deallocate
cannam@127: it with fftw_free()
. Do not use free()
or, heaven
cannam@127: forbid, delete
.
cannam@127:
cannam@127:
FFTW computes an unnormalized DFT. Thus, computing a forward
cannam@127: followed by a backward transform (or vice versa) results in the original
cannam@127: array scaled by n
. For the definition of the DFT, see What FFTW Really Computes.
cannam@127:
cannam@127:
cannam@127:
If you have a C compiler, such as gcc
, that supports the
cannam@127: C99 standard, and you #include <complex.h>
before
cannam@127: <fftw3.h>
, then fftw_complex
is the native
cannam@127: double-precision complex type and you can manipulate it with ordinary
cannam@127: arithmetic. Otherwise, FFTW defines its own complex type, which is
cannam@127: bit-compatible with the C99 complex type. See Complex numbers.
cannam@127: (The C++ <complex>
template class may also be usable via a
cannam@127: typecast.)
cannam@127:
cannam@127:
To use single or long-double precision versions of FFTW, replace the
cannam@127: fftw_
prefix by fftwf_
or fftwl_
and link with
cannam@127: -lfftw3f
or -lfftw3l
, but use the same
cannam@127: <fftw3.h>
header file.
cannam@127:
cannam@127:
Many more flags exist besides FFTW_MEASURE
and
cannam@127: FFTW_ESTIMATE
. For example, use FFTW_PATIENT
if you’re
cannam@127: willing to wait even longer for a possibly even faster plan (see FFTW Reference).
cannam@127:
cannam@127: You can also save plans for future use, as described by Words of Wisdom-Saving Plans.
cannam@127:
cannam@127: Next: Complex Multi-Dimensional DFTs, Previous: Tutorial, Up: Tutorial [Contents][Index]
cannam@127: