NthEvent_policy.cc

#define TRACE_NAME "NthEventPolicy"
#include "artdaq/RoutingPolicies/RoutingMasterPolicy.hh"
#include "artdaq/RoutingPolicies/PolicyMacros.hh"
#include "fhiclcpp/ParameterSet.h"
#include "cetlib_except/exception.h"

namespace artdaq
{
    class NthEventPolicy : public RoutingMasterPolicy
    {
    public:
        explicit NthEventPolicy(fhicl::ParameterSet ps);

        virtual ~NthEventPolicy() { }

        detail::RoutingPacket GetCurrentTable() override;
    private:
        size_t nth_;
        int nth_rank_;
    };

    NthEventPolicy::NthEventPolicy(fhicl::ParameterSet ps)
        : RoutingMasterPolicy(ps)
        , nth_(ps.get<size_t>("nth_event"))
        , nth_rank_(ps.get<int>("target_receiver"))
    {
        if (nth_ == 0) throw cet::exception("NthEvent_policy") << "nth_event must be greater than 0!";
    }

    detail::RoutingPacket NthEventPolicy::GetCurrentTable()
    {
        auto tokens = getTokensSnapshot();
        std::map<int, int> table;
        for (auto token : *tokens.get())
        {
            table[token]++;
        }
        if (table.count(nth_rank_) == 0) table[nth_rank_] = 0;
        tokens->clear();

        detail::RoutingPacket output;
        TLOG(5) << "table[nth_rank_]=" << (table[nth_rank_])
            << ", Next nth=" << (((next_sequence_id_ / nth_) + 1) * nth_)
            << ", max seq=" << (next_sequence_id_ + table.size() - 1) ;
        auto endCondition = table.size() < GetReceiverCount() || (table[nth_rank_] <= 0 && (next_sequence_id_ % nth_ == 0 || ((next_sequence_id_ / nth_) + 1) * nth_ < next_sequence_id_ + table.size() - 1));
        while (!endCondition)
        {
            for (auto r : table)
            {
                TLOG(5) << "nth_=" << nth_
                    << ", nth_rank=" << nth_rank_
                    << ", r=" << r.first
                    << ", next_sequence_id=" << next_sequence_id_;
                if (next_sequence_id_ % nth_ == 0)
                {
                    TLOG(5) << "Diverting event " << next_sequence_id_ << " to EVB " << nth_rank_ ;
                    output.emplace_back(detail::RoutingPacketEntry(next_sequence_id_++, nth_rank_));
                    table[nth_rank_]--;
                }
                if (r.first != nth_rank_) {
                    TLOG(5) << "Sending event " << next_sequence_id_ << " to EVB " << r.first ;
                    output.emplace_back(detail::RoutingPacketEntry(next_sequence_id_++, r.first));
                    if (!endCondition) endCondition = r.second == 1;
                    table[r.first]--;
                }
            }
            TLOG(5) << "table[nth_rank_]=" << table[nth_rank_]
                << ", Next nth=" << (((next_sequence_id_ / nth_) + 1) * nth_)
                << ", max seq=" << (next_sequence_id_ + table.size() - 1) ;
            endCondition = endCondition || (table[nth_rank_] <= 0 && (next_sequence_id_ % nth_ == 0 || (next_sequence_id_ / nth_) * nth_ + nth_ < next_sequence_id_ + table.size() - 1));
        }

        for (auto r : table)
        {
            for (auto i = 0; i < r.second; ++i)
            {
                tokens->push_back(r.first);
            }
        }
        addUnusedTokens(std::move(tokens));

        return output;
    }
}

DEFINE_ARTDAQ_ROUTING_POLICY(artdaq::NthEventPolicy)