From b864b45734e8ba68eb3c7b0a60a4a5b98e6b4571 Mon Sep 17 00:00:00 2001 From: Yann Herklotz Date: Mon, 5 Mar 2018 11:13:04 +0000 Subject: Adding dft --- Project/RTDSP/dft.c | 43 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 43 insertions(+) create mode 100644 Project/RTDSP/dft.c (limited to 'Project/RTDSP') diff --git a/Project/RTDSP/dft.c b/Project/RTDSP/dft.c new file mode 100644 index 0000000..4d97085 --- /dev/null +++ b/Project/RTDSP/dft.c @@ -0,0 +1,43 @@ +#include +#include + +#define E_PI 3.1415926535897932384626433832795028841971693993751058209749445923078164062 +#define N 8 + +typedef struct { + float r; + float i; +} Complex; + +float x[] = {0, 0, 2, 3, 4, 0, 0, 0}; +Complex X[N]; +float x_out[N]; + +void dft() { + for(int k = 0; k < N; ++k) { + X[k].r = 0.f; + X[k].i = 0.f; + for(int n = 0; n < N; ++n) { + X[k].r += x[n] * cos(2 * E_PI * k * n / N); + X[k].i -= x[n] * sin(2 * E_PI * k * n / N); + } + } +} + +void idft() { + for(int n = 0; n < N; ++n) { + x_out[n] = 0.f; + for(int k = 0; k < N; ++k) { + x_out[n] += X[k].r * cos(2 * E_PI * k * n / N) - X[k].i * sin(2 * E_PI * k * n / N); + } + x_out[n] /= N; + } +} + +int main() { + dft(); + idft(); + for(int k = 0; k < N; ++k) { + printf("x[%d] = %.2f, X[%d] = %.2f + %.2fi, x_out[%d] = %.2f\n", k, x[k], k, X[k].r, X[k].i, k, x_out[k]); + } +} -- cgit