diff options
| author | Josh Morman <mormjb@gmail.com> | 2020-05-04 13:43:16 -0400 |
|---|---|---|
| committer | Josh Morman <mormjb@gmail.com> | 2020-06-04 10:05:48 -0400 |
| commit | e1d49194882e70cc1637ddeb491a41a86b58b8da (patch) | |
| tree | 5ae583bc2481233391a704b826e94ace1571ffec /gr-trellis/python | |
| parent | 5cdb091ee899b9bfd500d036279409b707dd3cd6 (diff) | |
trellis: add pybind bindings for template classes
Diffstat (limited to 'gr-trellis/python')
| -rw-r--r-- | gr-trellis/python/trellis/bindings/encoder_python.cc | 32 | ||||
| -rw-r--r-- | gr-trellis/python/trellis/bindings/viterbi_python.cc | 29 |
2 files changed, 59 insertions, 2 deletions
diff --git a/gr-trellis/python/trellis/bindings/encoder_python.cc b/gr-trellis/python/trellis/bindings/encoder_python.cc index bfa1c5a639..fef8ca881a 100644 --- a/gr-trellis/python/trellis/bindings/encoder_python.cc +++ b/gr-trellis/python/trellis/bindings/encoder_python.cc @@ -19,4 +19,34 @@ namespace py = pybind11; // pydoc.h is automatically generated in the build directory #include <encoder_pydoc.h> -void bind_encoder(py::module& m) {} +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"); +} diff --git a/gr-trellis/python/trellis/bindings/viterbi_python.cc b/gr-trellis/python/trellis/bindings/viterbi_python.cc index 32ba3f3f29..3729ed1257 100644 --- a/gr-trellis/python/trellis/bindings/viterbi_python.cc +++ b/gr-trellis/python/trellis/bindings/viterbi_python.cc @@ -19,4 +19,31 @@ namespace py = pybind11; // pydoc.h is automatically generated in the build directory #include <viterbi_pydoc.h> -void bind_viterbi(py::module& m) {} +template <typename T> +void bind_viterbi_template(py::module& m, const char* classname) +{ + using viterbi = gr::trellis::viterbi<T>; + + py::class_<viterbi, gr::block, gr::basic_block, std::shared_ptr<viterbi>>(m, + classname) + .def(py::init(&gr::trellis::viterbi<T>::make), + py::arg("FSM"), + py::arg("K"), + py::arg("S0"), + py::arg("SK")) + .def("FSM", &viterbi::FSM) + .def("K", &viterbi::K) + .def("S0", &viterbi::S0) + .def("SK", &viterbi::SK) + .def("set_FSM", &viterbi::set_FSM) + .def("set_K", &viterbi::set_K) + .def("set_S0", &viterbi::set_S0) + .def("set_SK", &viterbi::set_SK); +} + +void bind_viterbi(py::module& m) +{ + bind_viterbi_template<std::uint8_t>(m, "viterbi_b"); + bind_viterbi_template<std::int16_t>(m, "viterbi_s"); + bind_viterbi_template<std::int32_t>(m, "viterbi_i"); +} |