multi_file_reader.h

Go to the documentation of this file.

 
#pragma once
#include <adtf_file/reader.h>
#include <unordered_map>
 
namespace adtfdat_processing
{
 
class MultiFileReader : public adtf_file::Reader,
                        public adtf_file::ExplicitReader
{
public:
    MultiFileReader();
 
    std::string getReaderIdentifier() const override;
    void open(const std::string& file_name,
              std::shared_ptr<adtf_file::SampleFactory> sample_factory,
              std::shared_ptr<adtf_file::StreamTypeFactory> stream_type_factory) override;
 
    std::vector<adtf_file::Stream> getStreams() const override;
    adtf_file::FileItem getNextItem() override;
    std::optional<uint64_t> getItemCount() const override;
    std::optional<double> getProgress() const override;
 
protected:
    virtual adtf_file::ReaderFactories getFactories() const;
 
    std::shared_ptr<adtf_file::Reader> addFile(const std::string& file_name,
                                               std::shared_ptr<adtf_file::SampleFactory> sample_factory,
                                               std::shared_ptr<adtf_file::StreamTypeFactory> stream_type_factory);
    void build();
 
private:
 
    std::unordered_map<std::string, uint16_t> _stream_ids;
 
    struct File
    {
        std::shared_ptr<Reader> reader;
        std::unordered_map<uint16_t, uint16_t> stream_ids;
 
        adtf_file::FileItem getNext()
        {
            auto item = reader->getNextItem();
            item.stream_id = stream_ids[item.stream_id];
            return item;
        }
    };
    std::vector<File> _files;
    std::vector<adtf_file::Stream> _streams;
 
    struct QueueItem
    {
        File* file;
        adtf_file::FileItem file_item;
    };
    std::multimap<std::chrono::nanoseconds, QueueItem> _queue;
};
 
using MultiFileReaderFactory = adtf_file::ReaderFactoryImplementation<MultiFileReader>;
 
}