sequence_converter.h

Go to the documentation of this file.

 
#pragma once
 
#include <knp/core/message_endpoint.h>
#include <knp/core/messaging/spike_message.h>
#include <knp/core/uid.h>
 
#include <spdlog/spdlog.h>
 
#include <functional>
#include <memory>
#include <sstream>
#include <string>
#include <type_traits>
#include <utility>
#include <vector>
 
 
namespace knp::framework::io::input
{
template <class ValueType>
class SequenceConverter
{
public:
    SequenceConverter(
        std::unique_ptr<std::istream> &&stream, std::function<bool(ValueType)> interpret, size_t data_size)
        : stream_(std::move(stream)), interpret_(std::move(interpret)), data_size_(data_size)
    {
    }
 
    core::messaging::SpikeData operator()(core::Step step = 0)
    {
        SPDLOG_TRACE("Getting message from a stream using sequence converter...");
 
        core::messaging::SpikeData message_data;
        for (size_t i = 0; i < data_size_; ++i)
        {
            ValueType value;
            *stream_ >> value;
            if (interpret_(value))
            {
                message_data.push_back(i);
            }
        }
        SPDLOG_TRACE("Finished loading a message, it contains {} spikes.", message_data.size());
        return message_data;
    }
 
public:
    [[nodiscard]] std::istream &get_stream() { return *stream_; }
 
    void set_size(size_t size) { data_size_ = size; }
 
private:
    std::unique_ptr<std::istream> stream_;
    std::function<bool(ValueType)> interpret_;
 
    size_t data_size_;
};
 
}  // namespace knp::framework::io::input