From 2a575bf9b5a4942f60e979161764b9e942699e1e Mon Sep 17 00:00:00 2001
From: Lars Amsel <lars.amsel@ni.com>
Date: Fri, 4 Jun 2021 08:27:50 +0200
Subject: uhd: Add support for the USRP X410
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Co-authored-by: Lars Amsel <lars.amsel@ni.com>
Co-authored-by: Michael Auchter <michael.auchter@ni.com>
Co-authored-by: Martin Braun <martin.braun@ettus.com>
Co-authored-by: Paul Butler <paul.butler@ni.com>
Co-authored-by: Cristina Fuentes <cristina.fuentes-curiel@ni.com>
Co-authored-by: Humberto Jimenez <humberto.jimenez@ni.com>
Co-authored-by: Virendra Kakade <virendra.kakade@ni.com>
Co-authored-by: Lane Kolbly <lane.kolbly@ni.com>
Co-authored-by: Max Köhler <max.koehler@ni.com>
Co-authored-by: Andrew Lynch <andrew.lynch@ni.com>
Co-authored-by: Grant Meyerhoff <grant.meyerhoff@ni.com>
Co-authored-by: Ciro Nishiguchi <ciro.nishiguchi@ni.com>
Co-authored-by: Thomas Vogel <thomas.vogel@ni.com>
---
 host/lib/cal/cal_python.hpp | 44 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)

(limited to 'host/lib/cal/cal_python.hpp')

diff --git a/host/lib/cal/cal_python.hpp b/host/lib/cal/cal_python.hpp
index fa38c568b..919f86e30 100644
--- a/host/lib/cal/cal_python.hpp
+++ b/host/lib/cal/cal_python.hpp
@@ -10,6 +10,7 @@
 #include <uhd/cal/database.hpp>
 #include <uhd/cal/iq_cal.hpp>
 #include <uhd/cal/pwr_cal.hpp>
+#include <uhd/cal/dsa_cal.hpp>
 #include <uhd/utils/interpolation.hpp>
 #include <uhd/utils/pybind_adaptors.hpp>
 #include <pybind11/stl.h>
@@ -124,6 +125,49 @@ void export_cal(py::module& m)
             py::arg("power_dbm"),
             py::arg("freq"),
             py::arg("temperature") = boost::optional<int>());
+
+    py::class_<zbx_tx_dsa_cal, container, zbx_tx_dsa_cal::sptr>(m, "zbx_tx_dsa_cal")
+        .def(py::init([](const std::string& name,
+                          const std::string& serial,
+                          const uint64_t timestamp) {
+            return zbx_tx_dsa_cal::make(name, serial, timestamp);
+        }))
+        .def(py::init([]() { return zbx_tx_dsa_cal::make(); }))
+        .def(py::init([](const py::bytes data) {
+            return container::make<zbx_tx_dsa_cal>(pybytes_to_vector(data));
+        }))
+        .def("add_frequency_band",
+            &zbx_tx_dsa_cal::add_frequency_band,
+            py::arg("max_freq"),
+            py::arg("name"),
+            py::arg("steps"))
+        .def("clear", &zbx_tx_dsa_cal::clear)
+        .def("get_dsa_setting",
+            &zbx_tx_dsa_cal::get_dsa_setting,
+            py::arg("freq"),
+            py::arg("gain_index"));
+
+    py::class_<zbx_rx_dsa_cal, container, zbx_rx_dsa_cal::sptr>(m, "zbx_rx_dsa_cal")
+        .def(py::init([](const std::string& name,
+                          const std::string& serial,
+                          const uint64_t timestamp) {
+            return zbx_rx_dsa_cal::make(name, serial, timestamp);
+        }))
+        .def(py::init([]() { return zbx_rx_dsa_cal::make(); }))
+        .def(py::init([](const py::bytes data) {
+            return container::make<zbx_rx_dsa_cal>(pybytes_to_vector(data));
+        }))
+        .def("add_frequency_band",
+            &zbx_rx_dsa_cal::add_frequency_band,
+            py::arg("max_freq"),
+            py::arg("name"),
+            py::arg("steps"))
+        .def("clear", &zbx_rx_dsa_cal::clear)
+        .def("get_dsa_setting",
+            &zbx_rx_dsa_cal::get_dsa_setting,
+            py::arg("freq"),
+            py::arg("gain_index"));
+
 }
 
 #endif /* INCLUDED_UHD_CAL_PYTHON_HPP */
-- 
cgit v1.2.3