LatencyTestConsumer.hxx

Go to the documentation of this file.

 
#ifndef _OPENLCB_LATENCYTESTCONSUMER_HXX_
#define _OPENLCB_LATENCYTESTCONSUMER_HXX_
 
#include "openlcb/EventHandlerTemplates.hxx"
 
namespace openlcb
{
 
class LatencyTestConsumer : public SimpleEventHandler
{
public:
    using HookFn = std::function<void(Notifiable *)>;
 
    LatencyTestConsumer(Node *node, HookFn hook = nullptr)
        : node_(node)
        , hook_(hook)
    {
        EventRegistry::instance()->register_handler(
            EventRegistryEntry(this, EVENT_BASE), 32);
    }
 
    void handle_identify_global(const EventRegistryEntry &registry_entry,
        EventReport *event, BarrierNotifiable *done) override
    {
        AutoNotify an(done);
        if (event->dst_node && event->dst_node != node_)
        {
            return;
        }
        event->event_write_helper<1>()->WriteAsync(node_,
            Defs::MTI_CONSUMER_IDENTIFIED_RANGE, WriteHelper::global(),
            eventid_to_buffer(EncodeRange(EVENT_BASE, 0xffffffff)),
            done->new_child());
    }
 
    void handle_identify_consumer(const EventRegistryEntry &entry,
        EventReport *event, BarrierNotifiable *done) override
    {
        event_ = event;
        done_ = done;
        if (hook_)
        {
            hook_(new TempNotifiable([this]() { reply(); }));
        }
        else
        {
            reply();
        }
    }
 
private:
    void reply()
    {
        AutoNotify an(done_);
        event_->event_write_helper<1>()->WriteAsync(node_,
            Defs::MTI_CONSUMER_IDENTIFIED_UNKNOWN, WriteHelper::global(),
            eventid_to_buffer(event_->event), done_->new_child());
    }
 
    static constexpr EventId EVENT_BASE = 0x0900013900000000;
 
    Node *node_;
 
    HookFn hook_;
 
    EventReport *event_;
 
    BarrierNotifiable *done_ {nullptr};
};
 
} // namespace openlcb
 
#endif // _OPENLCB_LATENCYTESTCONSUMER_HXX_