zstream.hpp

Go to the documentation of this file.

 
#ifndef __JOIN_ZSTREAM_HPP__
#define __JOIN_ZSTREAM_HPP__
 
// libjoin.
#include <join/streambuf.hpp>
 
// Libraries.
#include <zlib.h>
 
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;
 
        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