diff options
Diffstat (limited to 'host/lib/include/uhdlib')
| -rw-r--r-- | host/lib/include/uhdlib/rfnoc/rfnoc_device.hpp | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/host/lib/include/uhdlib/rfnoc/rfnoc_device.hpp b/host/lib/include/uhdlib/rfnoc/rfnoc_device.hpp index 7197bc9c4..1c2997f80 100644 --- a/host/lib/include/uhdlib/rfnoc/rfnoc_device.hpp +++ b/host/lib/include/uhdlib/rfnoc/rfnoc_device.hpp @@ -9,6 +9,7 @@ #include <uhd/device.hpp> #include <uhd/exception.hpp> +#include <uhd/rfnoc/mb_controller.hpp> #include <uhdlib/rfnoc/client_zero.hpp> #include <uhdlib/rfnoc/mb_iface.hpp> #include <boost/shared_ptr.hpp> @@ -41,6 +42,12 @@ public: return *_iface_registry.at(mb_idx); } + //! Return a reference to an MB controller + mb_controller::sptr get_mb_controller(const size_t mb_idx) const + { + return _mbc_registry.at(mb_idx); + } + /*! Return the number of motherboards in this device */ size_t get_num_mbs() @@ -73,8 +80,14 @@ protected: _iface_registry.emplace(mb_idx, std::move(mb_if)); } + void register_mb_controller(const size_t mb_idx, mb_controller::sptr mbc) + { + _mbc_registry.emplace(mb_idx, mbc); + } + private: std::unordered_map<size_t, mb_iface::uptr> _iface_registry; + std::unordered_map<size_t, mb_controller::sptr> _mbc_registry; }; // class rfnoc_device }}} // namespace uhd::rfnoc::detail |