diff options
Diffstat (limited to 'gr-fft/python/fft/bindings/python_bindings.cc')
-rw-r--r-- | gr-fft/python/fft/bindings/python_bindings.cc | 54 |
1 files changed, 54 insertions, 0 deletions
diff --git a/gr-fft/python/fft/bindings/python_bindings.cc b/gr-fft/python/fft/bindings/python_bindings.cc new file mode 100644 index 0000000000..a1c191ed3d --- /dev/null +++ b/gr-fft/python/fft/bindings/python_bindings.cc @@ -0,0 +1,54 @@ + +/* + * Copyright 2020 Free Software Foundation, Inc. + * + * This file is part of GNU Radio + * + * SPDX-License-Identifier: GPL-3.0-or-later + * + */ + +#include <pybind11/pybind11.h> + +#define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION +#include <numpy/arrayobject.h> + +namespace py = pybind11; + +// void bind_ctrlport_probe_psd(py::module&); +void bind_fft(py::module&); +void bind_fft_shift(py::module&); +void bind_fft_vcc(py::module&); +void bind_fft_vfc(py::module&); +void bind_goertzel(py::module&); +void bind_goertzel_fc(py::module&); +void bind_window(py::module&); + +// We need this hack because import_array() returns NULL +// for newer Python versions. +// This function is also necessary because it ensures access to the C API +// and removes a warning. +void* init_numpy() +{ + import_array(); + return NULL; +} + +PYBIND11_MODULE(fft_python, m) +{ + // Initialize the numpy C API + // (otherwise we will see segmentation faults) + init_numpy(); + + // Allow access to base block methods + py::module::import("gnuradio.gr"); + + // bind_ctrlport_probe_psd(m); + bind_fft(m); + bind_fft_shift(m); + bind_fft_vcc(m); + bind_fft_vfc(m); + bind_goertzel(m); + bind_goertzel_fc(m); + bind_window(m); +} |