compressed_file_writer.h

Go to the documentation of this file.

#ifndef META_COMPRESSED_FILE_WRITER_H_
#define META_COMPRESSED_FILE_WRITER_H_

#include <functional>
#include <stdexcept>
#include <string>

namespace meta
{
namespace io
{

class compressed_file_writer
{
  public:
    compressed_file_writer(const std::string& filename,
                           std::function<uint64_t(uint64_t)> mapping);

    ~compressed_file_writer();

    uint64_t bit_location() const;

    void write(uint64_t value);

    void write(const std::string& str);

    void close();

  private:
    void write_bit(bool bit);

    void write_buffer() const;

    FILE* outfile_;

    uint64_t char_cursor_;

    uint64_t bit_cursor_;

    uint64_t buffer_size_;

    unsigned char* buffer_;

    std::function<uint64_t(uint64_t)> mapping_;

    uint64_t bit_location_;

    bool closed_;

  public:
    class compressed_file_writer_exception : public std::runtime_error
    {
      public:
        using std::runtime_error::runtime_error;
    };
};

uint64_t default_compression_writer_func(uint64_t key);
}
}

#endif