indexedfilewriter_v201_v301.h

Go to the documentation of this file.

 
#ifndef INDEXEDFILE_WRITER_V201_V301_CLASS_HEADER
#define INDEXEDFILE_WRITER_V201_V301_CLASS_HEADER
 
#include <ifhd/v201_v301/indexedfile_v201_v301.h>
#include <ifhd/v201_v301/indexwritetable_v201_v301.h>
 
namespace ifhd
{
namespace v201_v301
{

class OutputStream
{
public:
    virtual void write(const void* data, size_t data_size) = 0;
};

class IndexedFileWriter : public IndexedFile
{
private:
    class IndexedFileWriterImpl;
    a_util::memory::unique_ptr<IndexedFileWriterImpl> _d;
 
public:
    class ChunkDroppedCallback
    {
    public:
        virtual void onChunkDropped(uint64_t index, uint16_t stream_id, uint16_t flags, timestamp_t time) = 0;
    };
 
protected:
    bool _is_open;
    bool _catch_first_time;
    timestamp_t _time_offset;
 
    int64_t _ref_index;
    timestamp_t _last_chunk_time;
 
    IndexWriteTable _index_table;
 
    typedef struct TagStreamInfoAdd
    {
        uint8_t* data;
        int is_reference;
    } StreamInfoAdd;
 
    StreamInfoHeader _stream_info[max_indexed_streams];
    StreamInfoAdd _stream_info_add[max_indexed_streams];
 
    FilePos _file_pos_last_chunk;
 
    std::string _file_name;
    std::string _temp_file_name;
 
    std::string _prefix_of_temp_save_file_name;

    bool _use_prefix_temp_file_extension;
 
public:
    std::string getTempSaveFileName();

    void setPrefixTempFileExtension(bool use_mode);

    bool getPrefixTempFileExtensionMode() const;

    std::string getPrefix() const;

    std::string getNewFileNameWithPrefix(const std::string& filename) const;
 
private:
    void createAFileWithPrefixdAndAFileWithoutPrefix(const std::string& filename,
                                                     std::string& save_filename,
                                                     uint32_t open_flags);

    void renameTempSaveToFileName();

    void addChecksum(const FileRangeChecksumXxh3_64& checksum);
 
public:
    IndexedFileWriter();
    ~IndexedFileWriter();

    void setDateTime(const a_util::datetime::DateTime& date_time);

    void open(const std::string& filename,
              uint32_t flags = 0,
              uint64_t file_time_offset = 0,
              timestamp_t history = 0,
              utils5ext::FileSize history_size = 0,
              ChunkDroppedCallback* drop_callback = nullptr,
              timestamp_t index_delay = 1000000);

    void close();

    void writeChunk(uint16_t stream_id, const void* data, uint32_t data_size, timestamp_t time_stamp, uint32_t flags);

    void setAdditionalStreamInfo(uint16_t stream_id,
                                 const void* info_data,
                                 uint32_t info_data_size,
                                 bool use_as_reference = false);
    void setStreamName(uint16_t stream_id, const char* stream_name);

    void quitHistory();

    std::shared_ptr<OutputStream>
        getExtensionStream(const std::string& name, uint32_t user_id, uint32_t type_id, uint32_t file_version_id);
 
protected:
    void initialize();

    void writeFileHeader(bool store_checksum = false);

    void writeFileHeaderExt();

    void writeIndexTable();

    void updateChecksumsExtension();
};
 
} // namespace v201_v301
} // namespace ifhd
 
//*************************************************************************************************
#endif // _INDEXEDFILE_WRITER_V201_V301_CLASS_HEADER_