sample_timestamp_sanitizer_reader.h

Go to the documentation of this file.

 
#pragma once
#include <adtf_file/reader.h>
 
#include <chrono>
#include <cstdint>
#include <deque>
#include <memory>
#include <optional>
#include <string>
#include <utility>
#include <vector>
 
namespace adtfdat_processing
{
class SampleTimestampSanitizerReader : public adtf_file::SeekableReader, public adtf_file::ExplicitReader
{
public:
    SampleTimestampSanitizerReader();
 
    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:
    void setupReader(const std::string& file_name,
                     const std::shared_ptr<adtf_file::SampleFactory>& sample_factory,
                     const std::shared_ptr<adtf_file::StreamTypeFactory>& stream_type_factory);
    void resetTimestampTracking();
 
    void refillQueue();
    void sanitizeBlock();
    void sanitizeFinalBlock();
 
    void setSampleTimestamp(adtf_file::FileItem& file_item,
                            const std::shared_ptr<const adtf_file::Sample>& sample,
                            std::chrono::nanoseconds timestamp);
    void setFileItemTimestamp(adtf_file::FileItem& file_item, std::chrono::nanoseconds timestamp);
 
    std::shared_ptr<adtf_file::SampleFactory> _sample_factory{};
    std::shared_ptr<Reader> _reader{};
    std::vector<adtf_file::Stream> _streams{};
 
    std::optional<std::chrono::nanoseconds> _lower_timestamp_bound{};
    std::deque<adtf_file::FileItem> _item_queue;
};

using SampleTimestampSanitizerReaderFactory = adtf_file::ReaderFactoryImplementation<SampleTimestampSanitizerReader>;
 
} // namespace adtfdat_processing