input_channel.h

Go to the documentation of this file.

 
#pragma once
 
#include <knp/core/core.h>
 
#include <utility>
#include <vector>
 
#include "input_converter.h"
 
 
namespace knp::framework::io::input
{
 
class KNP_DECLSPEC InputChannel
{
public:
    InputChannel(const core::UID &channel_uid, core::MessageEndpoint &&endpoint, const DataGenerator &generator)
        : base_{channel_uid}, endpoint_(std::move(endpoint)), generator_(generator)
    {
    }
 
    InputChannel(const core::UID &channel_uid, core::MessageEndpoint &&endpoint, DataGenerator &&generator)
        : base_{channel_uid}, endpoint_(std::move(endpoint)), generator_(std::move(generator))
    {
    }
 
    InputChannel(InputChannel &&) = default;
 
    virtual ~InputChannel() = default;
 
public:
    [[nodiscard]] const auto &get_uid() const { return base_.uid_; }
    [[nodiscard]] auto &get_tags() { return base_.tags_; }
 
public:
    virtual bool send(core::Step step) { return send_data(generator_(step), step); }
 
protected:
    bool send_data(const core::messaging::SpikeData &spikes, core::Step step)
    {
        if (spikes.empty())
        {
            return false;
        }
 
        core::messaging::SpikeMessage message{{get_uid(), step}, spikes};
        endpoint_.send_message(message);
        return true;
    }
 
private:
    core::BaseData base_;
 
    core::MessageEndpoint endpoint_;
 
    DataGenerator generator_;
};
 
 
inline void connect_input(
    const InputChannel &channel, core::MessageEndpoint &target_endpoint, const core::UID &receiver_uid)
{
    target_endpoint.subscribe<core::messaging::SpikeMessage>(receiver_uid, {channel.get_uid()});
}
 
}  // namespace knp::framework::io::input