diff options
| author | mormj <34754695+mormj@users.noreply.github.com> | 2020-11-03 16:51:44 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-11-03 16:51:44 -0500 |
| commit | ed95e8d45cfc6cd3e026298dadf4d2e8f9f1c3b8 (patch) | |
| tree | a15c82b2a2c2baa623434ad2c3a4c51416002082 /gr-fft/python/fft/bindings/fft_v_python.cc | |
| parent | 0f780e26d6545338b6c4a1b34405822a9a6ec706 (diff) | |
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>
Diffstat (limited to 'gr-fft/python/fft/bindings/fft_v_python.cc')
| -rw-r--r-- | gr-fft/python/fft/bindings/fft_v_python.cc | 57 |
1 files changed, 57 insertions, 0 deletions
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"); +} |