From ba9179c7991eeaf1b8d59a0db3975f049b2735b7 Mon Sep 17 00:00:00 2001
From: Roland Stigge <stigge@antcom.de>
Date: Sun, 27 Jan 2019 13:32:40 +0100
Subject: Added half FFT and magnitudes

---
 fft.h | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 fft.h

(limited to 'fft.h')

diff --git a/fft.h b/fft.h
new file mode 100644
index 0000000..f2b7038
--- /dev/null
+++ b/fft.h
@@ -0,0 +1,34 @@
+// FFT
+
+#pragma once
+
+#include <complex>
+#include <vector>
+
+namespace RIT {
+
+class FFT {
+private:
+	int mSize;
+	std::vector<int> order;
+	std::vector<std::complex<double>> expLUT;
+
+	bool mFlagHalfOnly;
+
+public:
+	FFT(int size, bool halfOnly = false);
+	std::vector<std::complex<double>> operator()(const std::vector<std::complex<double>> &v);
+
+	FFT& SetHalfOnly(bool enable);
+
+private:
+	int bitreverse(int i);
+
+	void reorder(const std::vector<std::complex<double>>& src, std::vector<std::complex<double>>& dst);
+	void fft_recursive(std::vector<std::complex<double>>::iterator X, int N);
+	void fft_half(std::vector<std::complex<double>>::iterator X, int N);
+}; // class FFT
+
+std::vector<double> magnitudes(std::vector<std::complex<double>>& v);
+
+} // namespace RIT
-- 
cgit v1.2.3