zstream.hpp

Go to the documentation of this file.

 
#ifndef JOIN_DATA_ZSTREAM_HPP
#define JOIN_DATA_ZSTREAM_HPP
 
// libjoin.
#include <join/streambuf.hpp>
 
// Libraries.
#include <zlib.h>
 
// C++.
#include <iostream>
#include <memory>
 
namespace join
{
    class Zstreambuf : public StreambufDecorator
    {
    public:
        Zstreambuf (std::streambuf* streambuf, int format, bool own = false);
 
        Zstreambuf (const Zstreambuf& other) = delete;
 
        Zstreambuf& operator= (const Zstreambuf& other) = delete;
 
        Zstreambuf (Zstreambuf&& other);
 
        Zstreambuf& operator= (Zstreambuf&& other);
 
        virtual ~Zstreambuf ();
 
    protected:
        virtual int_type underflow () override;
 
        virtual int_type overflow (int_type c = traits_type::eof ()) override;
 
        virtual int_type sync () override;
 
        void release ();
 
        static const std::streamsize _bufsize = 16384;
 
        std::unique_ptr<z_stream> _inflate;
 
        int _instate = Z_OK;
 
        std::unique_ptr<z_stream> _deflate;
 
        std::unique_ptr<char[]> _buf;
    };
 
    class Zstream : public std::iostream
    {
    public:
        enum Format
        {
            Deflate = -MAX_WBITS,  
            Zlib = MAX_WBITS,      
            Gzip = MAX_WBITS + 16, 
        };
 
        Zstream (std::iostream& stream, Format format = Zlib);
 
        Zstream (const Zstream& other) = delete;
 
        Zstream& operator= (const Zstream& other) = delete;
 
        Zstream (Zstream&& other);
 
        Zstream& operator= (Zstream&& other);
 
        virtual ~Zstream () = default;
 
    protected:
        Zstreambuf _zbuf;
    };
}
 
#endif