convert_set.h

Go to the documentation of this file.

 
#pragma once
 
#include <knp/core/impexp.h>
#include <knp/core/messaging/spike_message.h>
 
#include <set>
#include <vector>
 
 
namespace knp::framework::io::output
{
 
class KNP_DECLSPEC ConvertToSet
{
public:
    explicit ConvertToSet(size_t output_size) : output_size_(output_size) {}
    std::set<core::messaging::SpikeIndex> operator()(const std::vector<core::messaging::SpikeMessage> &message_list)
    {
        std::set<core::messaging::SpikeIndex> result;
        for (auto &message : message_list)
        {
            result.insert(message.neuron_indexes_.cbegin(), message.neuron_indexes_.cend());
        }
 
        // Ignore extra neurons.
        auto iter = result.lower_bound(output_size_);
        result.erase(iter, result.end());
        return result;
    }
 
private:
    const size_t output_size_;
};
 
}  // namespace knp::framework::io::output