From 5ab8a5088b729a9953b8dff1b2a985dc8fb2098b Mon Sep 17 00:00:00 2001 From: mzuenni Date: Mon, 27 Jun 2022 17:19:28 +0200 Subject: updated tcr --- math/transforms/fftMul.cpp | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 math/transforms/fftMul.cpp (limited to 'math/transforms/fftMul.cpp') diff --git a/math/transforms/fftMul.cpp b/math/transforms/fftMul.cpp new file mode 100644 index 0000000..dc19412 --- /dev/null +++ b/math/transforms/fftMul.cpp @@ -0,0 +1,14 @@ +vector mul(vector& a, vector& b) { + vector c(a.size()), d(a.size()); + for (int i = 0; i < b.size(); i++) { + c[i] = {real(a[i]), real(b[i])}; + } + c = fft(c); + for (int i = 0; i < b.size(); i++) { + int j = (a.size() - i) % a.size(); + cplx x = (c[i] + conj(c[j])) / cplx{2, 0}; //fft(a)[i]; + cplx y = (c[i] - conj(c[j])) / cplx{0, 2}; //fft(b)[i]; + d[i] = x * y; + } + return fft(d, true); +} -- cgit v1.2.3