1 files changed, 60 insertions, 0 deletions
diff --git a/gr-trellis/python/trellis/bindings/encoder_python.cc b/gr-trellis/python/trellis/bindings/encoder_python.cc
new file mode 100644
index 0000000000..2a470be228
--- /dev/null
+++ b/gr-trellis/python/trellis/bindings/encoder_python.cc
@@ -0,0 +1,60 @@
+/*
+ * Copyright 2020 Free Software Foundation, Inc.
+ *
+ * This file is part of GNU Radio
+ *
+ * SPDX-License-Identifier: GPL-3.0-or-later
+ *
+ */
+
+/***********************************************************************************/
+/* This file is automatically generated using bindtool and can be manually edited  */
+/* The following lines can be configured to regenerate this file during cmake      */
+/* If manual edits are made, the following tags should be modified accordingly.    */
+/* BINDTOOL_GEN_AUTOMATIC(0)                                                       */
+/* BINDTOOL_USE_PYGCCXML(0)                                                        */
+/* BINDTOOL_HEADER_FILE(encoder.h)                                        */
+/* BINDTOOL_HEADER_FILE_HASH(50192d0209b88c4fc915b31237c0932e)                     */
+/***********************************************************************************/
+
+#include <pybind11/complex.h>
+#include <pybind11/pybind11.h>
+#include <pybind11/stl.h>
+
+namespace py = pybind11;
+
+#include <gnuradio/trellis/encoder.h>
+// pydoc.h is automatically generated in the build directory
+#include <encoder_pydoc.h>
+
+template <class IN_T, class OUT_T>
+void bind_encoder_template(py::module& m, const char* classname)
+{
+    using encoder = gr::trellis::encoder<IN_T, OUT_T>;
+
+    py::class_<encoder,
+               gr::sync_block,
+               gr::block,
+               gr::basic_block,
+               std::shared_ptr<encoder>>(m, classname)
+        .def(py::init(&gr::trellis::encoder<IN_T, OUT_T>::make),
+             py::arg("FSM"),
+             py::arg("ST"),
+             py::arg("K") = 0)
+        .def("FSM", &gr::trellis::encoder<IN_T, OUT_T>::FSM)
+        .def("ST", &gr::trellis::encoder<IN_T, OUT_T>::ST)
+        .def("K", &gr::trellis::encoder<IN_T, OUT_T>::K)
+        .def("set_FSM", &gr::trellis::encoder<IN_T, OUT_T>::set_FSM)
+        .def("set_ST", &gr::trellis::encoder<IN_T, OUT_T>::set_ST)
+        .def("set_K", &gr::trellis::encoder<IN_T, OUT_T>::set_K);
+}
+
+void bind_encoder(py::module& m)
+{
+    bind_encoder_template<std::uint8_t, std::uint8_t>(m, "encoder_bb");
+    bind_encoder_template<std::uint8_t, std::int16_t>(m, "encoder_bs");
+    bind_encoder_template<std::uint8_t, std::int32_t>(m, "encoder_bi");
+    bind_encoder_template<std::int16_t, std::int16_t>(m, "encoder_ss");
+    bind_encoder_template<std::int16_t, std::int32_t>(m, "encoder_si");
+    bind_encoder_template<std::int32_t, std::int32_t>(m, "encoder_ii");
+}