diff options
| author | Josh Morman <mormjb@gmail.com> | 2020-04-23 10:59:04 -0400 |
|---|---|---|
| committer | Josh Morman <mormjb@gmail.com> | 2020-06-04 10:05:48 -0400 |
| commit | 9f49495b36004af83372a31bb658f32fbb334599 (patch) | |
| tree | 3dcfd9a4549af722f1de320c9f6758aa5679e278 /gr-uhd/python/uhd/bindings/python_bindings.cc | |
| parent | 25d5e75363199413239e5c69dda2c41faca896ac (diff) | |
uhd: add pybind11 bindings
Diffstat (limited to 'gr-uhd/python/uhd/bindings/python_bindings.cc')
| -rw-r--r-- | gr-uhd/python/uhd/bindings/python_bindings.cc | 46 |
1 files changed, 46 insertions, 0 deletions
diff --git a/gr-uhd/python/uhd/bindings/python_bindings.cc b/gr-uhd/python/uhd/bindings/python_bindings.cc new file mode 100644 index 0000000000..eeb7c6e430 --- /dev/null +++ b/gr-uhd/python/uhd/bindings/python_bindings.cc @@ -0,0 +1,46 @@ + +/* + * 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_amsg_source(py::module&); +void bind_usrp_block(py::module&); +void bind_usrp_sink(py::module&); +void bind_usrp_source(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(uhd_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_amsg_source(m); + bind_usrp_block(m); + bind_usrp_sink(m); + bind_usrp_source(m); +} |