1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
//
// Copyright 2020 Ettus Research, a National Instruments Brand
//
// SPDX-License-Identifier: GPL-3.0-or-later
//
#include "property_tree_python.hpp"
#include <uhd/property_tree.hpp>
#include <uhd/types/device_addr.hpp>
#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
#include <string>
namespace py = pybind11;
template <typename T>
void export_property(py::module& m, const std::string& type_str)
{
const std::string classname = std::string("property__") + type_str;
py::class_<uhd::property<T>>(m, classname.c_str())
.def("get", &uhd::property<T>::get)
.def("get_desired", &uhd::property<T>::get_desired)
.def("set", &uhd::property<T>::set)
.def("set_coerced", &uhd::property<T>::set_coerced);
}
void export_property_tree(py::module& m)
{
using property_tree = uhd::property_tree;
using fs_path = uhd::fs_path;
py::class_<fs_path>(m, "fs_path")
// Constructors
.def(py::init<>())
.def(py::init<std::string>());
py::implicitly_convertible<std::string, fs_path>();
// Per type we want to expose, add one line here, and one accessor below
export_property<int>(m, "int");
export_property<double>(m, "double");
export_property<std::string>(m, "str");
export_property<bool>(m, "bool");
export_property<uhd::device_addr_t>(m, "device_addr");
py::class_<property_tree>(m, "property_tree")
.def("subtree", &property_tree::subtree, py::arg("path"))
.def("exists", &property_tree::exists, py::arg("path"))
.def("list", &property_tree::list, py::arg("path"))
// One line per type
.def(
"access_int", &property_tree::access<int>, py::return_value_policy::reference)
.def("access_double",
&property_tree::access<double>,
py::return_value_policy::reference)
.def("access_str",
&property_tree::access<std::string>,
py::return_value_policy::reference)
.def("access_bool",
&property_tree::access<bool>,
py::return_value_policy::reference)
.def("access_device_addr",
&property_tree::access<uhd::device_addr_t>,
py::return_value_policy::reference)
// End of types
;
}
|
