stream_renaming_reader.h

Go to the documentation of this file.

 
#pragma once
#include <adtf_file/reader.h>
#include <adtf_file/configuration.h>
#include <adtf_file/stream_item.h>
 
#include <stdint.h>
#include <string>
#include <memory>
#include <optional>
#include <regex>
#include <vector>
#include <unordered_set>
 
namespace adtfdat_processing
{
class StreamRenamingReader : public adtf_file::SeekableReader, public adtf_file::ExplicitReader
{
public:
    StreamRenamingReader();
 
    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;
 
    uint32_t getFileVersion() const override;
    std::string getDescription() const override;
    std::vector<adtf_file::Stream> getStreams() const override;
    std::vector<adtf_file::Extension> getExtensions() const override;
    adtf_file::FileItem getNextItem() override;
    std::optional<uint64_t> getItemCount() const override;
    std::optional<double> getProgress() const override;
 
    uint64_t getItemIndexForTimeStamp(std::chrono::nanoseconds time_stamp) override;
    uint64_t getItemIndexForStreamItemIndex(uint16_t stream_id, uint64_t stream_item_index) override;
    std::shared_ptr<const adtf_file::StreamType> getStreamTypeBefore(uint64_t item_index, uint16_t stream_id) override;
    void seekTo(uint64_t item_index) override;
 
private:
    struct RegexRule
    {
        std::string new_name{};
        std::regex pattern{};
        bool matched{false};
    };
 
    std::shared_ptr<Reader> _reader{};
    std::vector<adtf_file::Stream> _streams{};
 
    void setupReader(const std::string& file_name,
                     std::shared_ptr<adtf_file::SampleFactory> sample_factory,
                     std::shared_ptr<adtf_file::StreamTypeFactory> stream_type_factory);
    std::vector<RegexRule> parseRegexRules() const;
    void renameStreams(std::vector<RegexRule> regex_rules);
};

using StreamRenamingReaderFactory = adtf_file::ReaderFactoryImplementation<StreamRenamingReader>;
 
} // namespace adtfdat_processing