Chris@32
|
1 /*
|
Chris@32
|
2 * Free FFT and convolution (C)
|
Chris@32
|
3 *
|
Chris@32
|
4 * Copyright (c) 2014 Project Nayuki
|
Chris@32
|
5 * http://www.nayuki.io/page/free-small-fft-in-multiple-languages
|
Chris@32
|
6 *
|
Chris@32
|
7 * (MIT License)
|
Chris@32
|
8 * Permission is hereby granted, free of charge, to any person obtaining a copy of
|
Chris@32
|
9 * this software and associated documentation files (the "Software"), to deal in
|
Chris@32
|
10 * the Software without restriction, including without limitation the rights to
|
Chris@32
|
11 * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
|
Chris@32
|
12 * the Software, and to permit persons to whom the Software is furnished to do so,
|
Chris@32
|
13 * subject to the following conditions:
|
Chris@32
|
14 * - The above copyright notice and this permission notice shall be included in
|
Chris@32
|
15 * all copies or substantial portions of the Software.
|
Chris@32
|
16 * - The Software is provided "as is", without warranty of any kind, express or
|
Chris@32
|
17 * implied, including but not limited to the warranties of merchantability,
|
Chris@32
|
18 * fitness for a particular purpose and noninfringement. In no event shall the
|
Chris@32
|
19 * authors or copyright holders be liable for any claim, damages or other
|
Chris@32
|
20 * liability, whether in an action of contract, tort or otherwise, arising from,
|
Chris@32
|
21 * out of or in connection with the Software or the use or other dealings in the
|
Chris@32
|
22 * Software.
|
Chris@32
|
23 */
|
Chris@32
|
24
|
Chris@32
|
25
|
Chris@32
|
26 /*
|
Chris@32
|
27 * Computes the discrete Fourier transform (DFT) of the given complex vector, storing the result back into the vector.
|
Chris@32
|
28 * The vector can have any length. This is a wrapper function. Returns 1 (true) if successful, 0 (false) otherwise (out of memory).
|
Chris@32
|
29 */
|
Chris@32
|
30 int transform(double real[], double imag[], size_t n);
|
Chris@32
|
31
|
Chris@32
|
32 /*
|
Chris@32
|
33 * Computes the inverse discrete Fourier transform (IDFT) of the given complex vector, storing the result back into the vector.
|
Chris@32
|
34 * The vector can have any length. This is a wrapper function. This transform does not perform scaling, so the inverse is not a true inverse.
|
Chris@32
|
35 * Returns 1 (true) if successful, 0 (false) otherwise (out of memory).
|
Chris@32
|
36 */
|
Chris@32
|
37 int inverse_transform(double real[], double imag[], size_t n);
|
Chris@32
|
38
|
Chris@32
|
39 /*
|
Chris@32
|
40 * Computes the discrete Fourier transform (DFT) of the given complex vector, storing the result back into the vector.
|
Chris@32
|
41 * The vector's length must be a power of 2. Uses the Cooley-Tukey decimation-in-time radix-2 algorithm.
|
Chris@32
|
42 * Returns 1 (true) if successful, 0 (false) otherwise (n is not a power of 2, or out of memory).
|
Chris@32
|
43 */
|
Chris@32
|
44 int transform_radix2(double real[], double imag[], size_t n);
|
Chris@32
|
45
|
Chris@33
|
46 /* Test versions with precalculated structures -- this API is
|
Chris@33
|
47 absolutely not for production use! */
|
Chris@32
|
48 typedef struct {
|
Chris@32
|
49 double *cos;
|
Chris@32
|
50 double *sin;
|
Chris@32
|
51 int levels;
|
Chris@32
|
52 } tables;
|
Chris@32
|
53
|
Chris@32
|
54 tables *precalc(size_t n);
|
Chris@32
|
55 void dispose(tables *);
|
Chris@32
|
56 void transform_radix2_precalc(double real[], double imag[], int n, tables *tables);
|
Chris@32
|
57
|
Chris@33
|
58 typedef struct {
|
Chris@33
|
59 float *cos;
|
Chris@33
|
60 float *sin;
|
Chris@33
|
61 int levels;
|
Chris@33
|
62 } tables_f;
|
Chris@33
|
63
|
Chris@33
|
64 tables_f *precalc_f(size_t n);
|
Chris@33
|
65 void dispose_f(tables_f *);
|
Chris@33
|
66 void transform_radix2_precalc_f(float real[], float imag[], int n, tables_f *tables);
|
Chris@33
|
67
|
Chris@32
|
68 /*
|
Chris@32
|
69 * Computes the discrete Fourier transform (DFT) of the given complex vector, storing the result back into the vector.
|
Chris@32
|
70 * The vector can have any length. This requires the convolution function, which in turn requires the radix-2 FFT function.
|
Chris@32
|
71 * Uses Bluestein's chirp z-transform algorithm. Returns 1 (true) if successful, 0 (false) otherwise (out of memory).
|
Chris@32
|
72 */
|
Chris@32
|
73 int transform_bluestein(double real[], double imag[], size_t n);
|
Chris@32
|
74
|
Chris@32
|
75 /*
|
Chris@32
|
76 * Computes the circular convolution of the given real vectors. Each vector's length must be the same.
|
Chris@32
|
77 * Returns 1 (true) if successful, 0 (false) otherwise (out of memory).
|
Chris@32
|
78 */
|
Chris@32
|
79 int convolve_real(const double x[], const double y[], double out[], size_t n);
|
Chris@32
|
80
|
Chris@32
|
81 /*
|
Chris@32
|
82 * Computes the circular convolution of the given complex vectors. Each vector's length must be the same.
|
Chris@32
|
83 * Returns 1 (true) if successful, 0 (false) otherwise (out of memory).
|
Chris@32
|
84 */
|
Chris@32
|
85 int convolve_complex(const double xreal[], const double ximag[], const double yreal[], const double yimag[], double outreal[], double outimag[], size_t n);
|
Chris@32
|
86
|