gzstream.h

Go to the documentation of this file.

#ifndef META_UTIL_GZSTREAM_H_
#define META_UTIL_GZSTREAM_H_

#include <zlib.h>

#include <istream>
#include <ostream>
#include <streambuf>
#include <vector>

namespace meta
{
namespace io
{

class gzstreambuf : public std::streambuf
{
  public:
    gzstreambuf(const char* filename, const char* openmode,
                size_t buffer_size = 512);

    ~gzstreambuf();

    int_type underflow() override;

    int_type overflow(int_type ch) override;

    int sync() override;

    bool is_open() const;

  private:
    std::vector<char> buffer_;
    gzFile file_;
};

class gzifstream : public std::istream
{
  public:
    explicit gzifstream(std::string name);

    gzstreambuf* rdbuf() const;

    void flush();

  private:
    gzstreambuf buffer_;
};

class gzofstream : public std::ostream
{
  public:
    explicit gzofstream(std::string name);

    gzstreambuf* rdbuf() const;

    void flush();

  private:
    gzstreambuf buffer_;
};
}
}
#endif