From 7ee106187e1d0991e7ae79abd905a79dc5760b5f Mon Sep 17 00:00:00 2001
From: Martin Braun <martin.braun@ettus.com>
Date: Mon, 12 Oct 2020 12:22:00 +0200
Subject: python: Add access to the property_tree from Python

Example:
>>> usrp = uhd.usrp.multi_usrp("")
>>> tree = usrp.get_tree()
>>> print(tree.access_int("/name").get())
---
 host/lib/property_tree_python.cpp   | 61 +++++++++++++++++++++++++++++++++++++
 host/lib/property_tree_python.hpp   | 11 +++++++
 host/lib/usrp/multi_usrp_python.cpp |  4 +++
 3 files changed, 76 insertions(+)
 create mode 100644 host/lib/property_tree_python.cpp
 create mode 100644 host/lib/property_tree_python.hpp

(limited to 'host/lib')

diff --git a/host/lib/property_tree_python.cpp b/host/lib/property_tree_python.cpp
new file mode 100644
index 000000000..0b7f18aa2
--- /dev/null
+++ b/host/lib/property_tree_python.cpp
@@ -0,0 +1,61 @@
+//
+// 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 <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");
+
+    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)
+        // End of types
+        ;
+}
diff --git a/host/lib/property_tree_python.hpp b/host/lib/property_tree_python.hpp
new file mode 100644
index 000000000..c16dacfe3
--- /dev/null
+++ b/host/lib/property_tree_python.hpp
@@ -0,0 +1,11 @@
+//
+// Copyright 2020 Ettus Research, a National Instruments Brand
+//
+// SPDX-License-Identifier: GPL-3.0-or-later
+//
+
+#pragma once
+
+#include <pybind11/pybind11.h>
+
+void export_property_tree(pybind11::module& m);
diff --git a/host/lib/usrp/multi_usrp_python.cpp b/host/lib/usrp/multi_usrp_python.cpp
index 602fc5178..c1ee17d0b 100644
--- a/host/lib/usrp/multi_usrp_python.cpp
+++ b/host/lib/usrp/multi_usrp_python.cpp
@@ -5,7 +5,9 @@
 // SPDX-License-Identifier: GPL-3.0-or-later
 //
 
+#include <pybind11/complex.h>
 #include <pybind11/pybind11.h>
+#include <pybind11/stl.h>
 
 namespace py = pybind11;
 
@@ -27,6 +29,8 @@ void export_multi_usrp(py::module& m)
         .def(py::init(&multi_usrp::make))
 
         // clang-format off
+        .def("get_tree"                , &multi_usrp::get_tree)
+
         // General USRP methods
         .def("get_rx_freq"             , &multi_usrp::get_rx_freq, py::arg("chan") = 0)
         .def("get_rx_num_channels"     , &multi_usrp::get_rx_num_channels)
-- 
cgit v1.2.3