From 0cf8a60a4e88d55de04cde1feb85eaadfdcf15d7 Mon Sep 17 00:00:00 2001
From: mattprost <matt.prost@ni.com>
Date: Mon, 14 Mar 2022 17:20:35 -0500
Subject: types: filters: Add python support for digital filters

Signed-off-by: mattprost <matt.prost@ni.com>
---
 host/lib/types/filters_python.hpp | 49 +++++++++++++++++++++++++++++++++++----
 host/python/uhd/filters.py        |  2 ++
 2 files changed, 46 insertions(+), 5 deletions(-)

(limited to 'host')

diff --git a/host/lib/types/filters_python.hpp b/host/lib/types/filters_python.hpp
index 7210faa21..15fea5f50 100644
--- a/host/lib/types/filters_python.hpp
+++ b/host/lib/types/filters_python.hpp
@@ -12,10 +12,12 @@
 
 void export_filters(py::module& m)
 {
-    using filter_info_base   = uhd::filter_info_base;
-    using filter_info_type   = filter_info_base::filter_type;
-    using analog_filter_base = uhd::analog_filter_base;
-    using analog_filter_lp   = uhd::analog_filter_lp;
+    using filter_info_base        = uhd::filter_info_base;
+    using filter_info_type        = filter_info_base::filter_type;
+    using analog_filter_base      = uhd::analog_filter_base;
+    using analog_filter_lp        = uhd::analog_filter_lp;
+    using digital_filter_base_i16 = uhd::digital_filter_base<int16_t>;
+    using digital_filter_fir_i16  = uhd::digital_filter_fir<int16_t>;
 
     py::enum_<filter_info_type>(m, "filter_type")
         .value("analog_low_pass", filter_info_base::ANALOG_LOW_PASS)
@@ -38,7 +40,8 @@ void export_filters(py::module& m)
         // Methods
         .def("get_analog_type", &analog_filter_base::get_analog_type);
 
-    py::class_<analog_filter_lp, analog_filter_base, std::shared_ptr<analog_filter_lp>>(m, "analog_filter_lp")
+    py::class_<analog_filter_lp, analog_filter_base, std::shared_ptr<analog_filter_lp>>(
+        m, "analog_filter_lp")
         .def(
             py::init<filter_info_type, bool, size_t, const std::string, double, double>())
 
@@ -46,6 +49,42 @@ void export_filters(py::module& m)
         .def("get_cutoff", &analog_filter_lp::get_cutoff)
         .def("get_rolloff", &analog_filter_lp::get_rolloff)
         .def("set_cutoff", &analog_filter_lp::set_cutoff);
+
+    py::class_<digital_filter_base_i16, filter_info_base, digital_filter_base_i16::sptr>(
+        m, "digital_filter_base_i16")
+        .def(py::init<filter_info_type,
+            bool,
+            size_t,
+            double,
+            size_t,
+            size_t,
+            double,
+            size_t,
+            std::vector<int16_t>>())
+
+        // Methods
+        .def("get_output_rate", &digital_filter_base_i16::get_output_rate)
+        .def("get_input_rate", &digital_filter_base_i16::get_input_rate)
+        .def("get_interpolation", &digital_filter_base_i16::get_interpolation)
+        .def("get_decimation", &digital_filter_base_i16::get_decimation)
+        .def("get_tap_full_scale", &digital_filter_base_i16::get_tap_full_scale)
+        .def("get_taps", &digital_filter_base_i16::get_taps);
+
+    py::class_<digital_filter_fir_i16,
+        digital_filter_base_i16,
+        std::shared_ptr<digital_filter_fir_i16>>(m, "digital_filter_fir_i16")
+        .def(py::init<filter_info_type,
+            bool,
+            size_t,
+            double,
+            size_t,
+            size_t,
+            size_t,
+            size_t,
+            std::vector<int16_t>>())
+
+        // Methods
+        .def("set_taps", &digital_filter_fir_i16::set_taps);
 }
 
 #endif /* INCLUDED_UHD_FILTERS_PYTHON_HPP */
diff --git a/host/python/uhd/filters.py b/host/python/uhd/filters.py
index 8479b56e1..2862e2945 100644
--- a/host/python/uhd/filters.py
+++ b/host/python/uhd/filters.py
@@ -14,3 +14,5 @@ FilterType = lib.filters.filter_type
 FilterInfoBase = lib.filters.filter_info_base
 AnalogFilterBase = lib.filters.analog_filter_base
 AnalogFilterLP = lib.filters.analog_filter_lp
+DigitalFilterBaseI16 = lib.filters.digital_filter_base_i16
+DigitalFilterFIRI16 = lib.filters.digital_filter_fir_i16
-- 
cgit v1.2.3