aboutsummaryrefslogtreecommitdiffstats
path: root/host/lib/rfnoc/sink_node_ctrl.cpp
blob: 53a26d7ed968e47845e28c74c0f986ce437b1acd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
//
// Copyright 2014 Ettus Research LLC
// Copyright 2018 Ettus Research, a National Instruments Company
//
// SPDX-License-Identifier: GPL-3.0-or-later
//

#include <uhd/rfnoc/sink_node_ctrl.hpp>
#include <uhd/rfnoc/source_node_ctrl.hpp>
#include <uhd/utils/log.hpp>
#include <uhdlib/rfnoc/utils.hpp>

using namespace uhd::rfnoc;

size_t sink_node_ctrl::connect_upstream(
    node_ctrl_base::sptr upstream_node, size_t port, const uhd::device_addr_t& args)
{
    boost::mutex::scoped_lock lock(_input_mutex);
    port = _request_input_port(port, args);
    _register_upstream_node(upstream_node, port);
    return port;
}

void sink_node_ctrl::set_tx_streamer(bool active, const size_t port)
{
    UHD_RFNOC_BLOCK_TRACE() << "sink_node_ctrl::set_tx_streamer() " << active << " "
                            << port;

    /* Enable all downstream connections:
    for(const node_ctrl_base::node_map_pair_t downstream_node:  list_downstream_nodes()) {
        sptr curr_downstream_block_ctrl =
            boost::dynamic_pointer_cast<sink_node_ctrl>(downstream_node.second.lock());
        if (curr_downstream_block_ctrl) {
            curr_downstream_block_ctrl->set_tx_streamer(
                    active,
                    get_downstream_port(downstream_node.first)
            );
        }
    }
    */

    // Only enable 1:1
    if (list_downstream_nodes().count(port)) {
        sink_node_ctrl::sptr this_downstream_block_ctrl =
            boost::dynamic_pointer_cast<sink_node_ctrl>(
                list_downstream_nodes().at(port).lock());
        if (this_downstream_block_ctrl) {
            this_downstream_block_ctrl->set_tx_streamer(
                active, get_downstream_port(port));
        }
    }

    _tx_streamer_active[port] = active;
}

size_t sink_node_ctrl::_request_input_port(
    const size_t suggested_port, const uhd::device_addr_t&) const
{
    return utils::node_map_find_first_free(_upstream_nodes, suggested_port);
}

void sink_node_ctrl::_register_upstream_node(
    node_ctrl_base::sptr upstream_node, size_t port)
{
    // Do all the checks:
    if (port == ANY_PORT) {
        throw uhd::type_error("Invalid input port number.");
    }
    if (_upstream_nodes.count(port) and not _upstream_nodes[port].expired()) {
        throw uhd::runtime_error(
            str(boost::format("On node %s, input port %d is already connected.")
                % unique_id() % port));
    }
    if (not boost::dynamic_pointer_cast<source_node_ctrl>(upstream_node)) {
        throw uhd::type_error("Attempting to register a non-source block as upstream.");
    }
    // Alles klar, Herr Kommissar :)

    _upstream_nodes[port] = boost::weak_ptr<node_ctrl_base>(upstream_node);
}