From ed95e8d45cfc6cd3e026298dadf4d2e8f9f1c3b8 Mon Sep 17 00:00:00 2001
From: mormj <34754695+mormj@users.noreply.github.com>
Date: Tue, 3 Nov 2020 16:51:44 -0500
Subject: Move FFT functions and blocks to templates (#3903)
* fft: refactor fft classes as templates
In order to reduce code duplication construct fft objects and blocks
with template paramters.
Separate forward and reverse as a template
parameter
Maintain the real fft block forward as r2c and reverse as r2c as well
Co-authored-by: Andrej Rode <mail@andrejro.de>
---
gr-fft/python/fft/bindings/fft_v_python.cc | 57 ++++++++++++++++++++++++++++++
1 file changed, 57 insertions(+)
create mode 100644 gr-fft/python/fft/bindings/fft_v_python.cc
(limited to 'gr-fft/python/fft/bindings/fft_v_python.cc')
diff --git a/gr-fft/python/fft/bindings/fft_v_python.cc b/gr-fft/python/fft/bindings/fft_v_python.cc
new file mode 100644
index 0000000000..720c484863
--- /dev/null
+++ b/gr-fft/python/fft/bindings/fft_v_python.cc
@@ -0,0 +1,57 @@
+/*
+ * Copyright 2020 Free Software Foundation, Inc.
+ *
+ * This file is part of GNU Radio
+ *
+ * SPDX-License-Identifier: GPL-3.0-or-later
+ *
+ */
+
+/***********************************************************************************/
+/* This file is automatically generated using bindtool and can be manually edited */
+/* The following lines can be configured to regenerate this file during cmake */
+/* If manual edits are made, the following tags should be modified accordingly. */
+/* BINDTOOL_GEN_AUTOMATIC(0) */
+/* BINDTOOL_USE_PYGCCXML(0) */
+/* BINDTOOL_HEADER_FILE(fft_v.h) */
+/* BINDTOOL_HEADER_FILE_HASH(ef3398e963b1b296e309b81721bef8cb) */
+/***********************************************************************************/
+
+#include <pybind11/complex.h>
+#include <pybind11/pybind11.h>
+#include <pybind11/stl.h>
+
+namespace py = pybind11;
+
+#include <gnuradio/fft/fft_v.h>
+// pydoc.h is automatically generated in the build directory
+#include <fft_v_pydoc.h>
+
+
+template <class T, bool forward>
+void bind_fft_v_template(py::module& m, const char* classname)
+{
+ using fft_v_blk = gr::fft::fft_v<T, forward>;
+
+ py::class_<fft_v_blk,
+ gr::sync_block,
+ gr::block,
+ gr::basic_block,
+ std::shared_ptr<fft_v_blk>>(m, classname)
+ .def(py::init(&gr::fft::fft_v<T, forward>::make),
+ py::arg("fft_size"),
+ py::arg("window"),
+ py::arg("shift") = false,
+ py::arg("nthreads") = 1)
+ .def("set_nthreads", &fft_v_blk::set_nthreads, py::arg("n"))
+ .def("nthreads", &fft_v_blk::nthreads)
+ .def("set_window", &fft_v_blk::set_window, py::arg("window"));
+}
+
+void bind_fft_v(py::module& m)
+{
+ bind_fft_v_template<gr_complex, true>(m, "fft_vcc_fwd");
+ bind_fft_v_template<gr_complex, false>(m, "fft_vcc_rev");
+ bind_fft_v_template<float, true>(m, "fft_vfc_fwd");
+ bind_fft_v_template<float, false>(m, "fft_vfc_rev");
+}
--
cgit v1.2.3