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
|
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#define E_PI 3.1415926535897932384626433832795028841971693993751058209749445923078164062
#define N_DEF 8
typedef struct {
float r;
float i;
} Complex;
float x[] = {0, 0, 2, 3, 4, 0, 0, 0};
void dft(int N, Complex *X) {
int i, n, k;
Complex *x = malloc(N * sizeof(X[0]));
for(i = 0; i < N; ++i) {
x[i] = X[i];
}
for(k = 0; k < N; ++k) {
X[k].r = 0.f;
X[k].i = 0.f;
for(n = 0; n < N; ++n) {
X[k].r += x[n].r * cos(2 * E_PI * k * n / N);
X[k].i -= x[n].r * sin(2 * E_PI * k * n / N);
}
}
free(x);
}
void idft(int N, Complex *X) {
int i, n, k;
Complex *x = malloc(N * sizeof(X[0]));
for(i = 0; i < N; ++i) {
x[i] = X[i];
}
for(n = 0; n < N; ++n) {
X[n].i = 0.f;
X[n].r = 0.f;
for(k = 0; k < N; ++k) {
X[n].r += x[k].r * cos(2 * E_PI * k * n / N) - x[k].i * sin(2 * E_PI * k * n / N);
}
X[n].r /= N;
}
free(x);
}
int main() {
int k;
Complex X_C[8];
for(k = 0; k < N_DEF; ++k) {
X_C[k].r = x[k];
X_C[k].i = 0;
}
dft(N_DEF, X_C);
for(k = 0; k < N_DEF; ++k) {
printf("x[%d] = %.2f, X[%d] = %.2f + %.2fi\n", k, x[k], k, X_C[k].r, X_C[k].i);
}
idft(N_DEF, X_C);
for(k = 0; k < N_DEF; ++k) {
printf("x[%d] = %.2f, X[%d] = %.2f + %.2fi\n", k, x[k], k, X_C[k].r, X_C[k].i);
}
}
|