gzstream.h

浏览该文件的文档.

/************************************************************************
* @file gzstream.h
*
* @brief 压缩文件流相关类
*
* @details 压缩文件流相关类
*
* @author wukx
*
* @version  v1.0
*
* @date  2016.9.29
*
* @license 北京华科软科技有限公司
*
*************************************************************************/
 
#ifndef BIMHOMEBASE_GZSTREAM_H
#define BIMHOMEBASE_GZSTREAM_H 1
 
#include <sstream>
#include <zlib.h>
 
#ifdef _MSC_VER
using std::ostream;
using std::istream;
#endif
 
 
namespace Base
{
 
#define BUFFERSIZE 47+256
 
    // ----------------------------------------------------------------------------
    //  实现 gzstream 的内部类。有关用户类，请参见下文。
    // ----------------------------------------------------------------------------
    class BaseExport gzstreambuf : public std::streambuf
    {
    public:
        gzstreambuf() : file(nullptr), opened(0), mode(0)
        {
            setp(buffer, buffer + (bufferSize - 1)); // 设置输出缓冲区的起始和结束位置
            setg(buffer + 4,     // 输入缓冲区的回退区域起始位置
                buffer + 4,     // 输入缓冲区的读取位置
                buffer + 4);    // 输入缓冲区的结束位置      
        }
 
        int is_open() { return opened; }
 
        gzstreambuf* open(const char* name, int open_mode, int comp);
 
        gzstreambuf* close();
 
        ~gzstreambuf() { close(); }
 
        virtual int     overflow(int c = EOF);
 
        virtual int     underflow();
 
        virtual int     sync();
 
    private:
        int flush_buffer();
 
        static const int bufferSize;    
        //在 g++ 下，最后的 igzstream 总计 512 字节。
 
        gzFile file;               
        char buffer[BUFFERSIZE];   
        char opened;               
        int mode;                  
    };
 
    class BaseExport gzstreambase : virtual public std::ios
    {
    public:
        gzstreambase() { init(&buf); }
 
        gzstreambase(const char* name, int open_mode, int comp);
 
        ~gzstreambase();
 
        void open(const char* name, int open_mode, int comp);
 
        void close();
 
        gzstreambuf* rdbuf() { return &buf; }
 
    protected:
        gzstreambuf buf; 
    };
 
 
    // ----------------------------------------------------------------------------
    // 用户类。igzstream 和 ogzstream 分别类似于 ifstream 和 ofstream。它们基于 zlib 的 gz* 函数接口读取和写入文件。
    // 支持 gzip 压缩格式的文件。
    // ----------------------------------------------------------------------------
    class BaseExport igzstream : public gzstreambase, public std::istream
    {
    public:
        igzstream()
#ifdef _MSC_VER
            : istream(&buf) {}
#else
            : std::istream(&buf) {}
#endif
        igzstream(const char* name, int open_mode = std::ios_base::in, int comp = 1)
#ifdef _MSC_VER
            : gzstreambase(name, open_mode, comp), istream(&buf) {}
#else
            : gzstreambase(name, open_mode, comp), std::istream(&buf) {}
#endif
        gzstreambuf* rdbuf() { return gzstreambase::rdbuf(); }
 
        void open(const char* name, int open_mode = std::ios_base::in, int comp = 1)
        {
            gzstreambase::open(name, open_mode, comp);
        }
    };
 
    class BaseExport ogzstream : public gzstreambase, public std::ostream
    {
    public:
        ogzstream()
#ifdef _MSC_VER
            : ostream(&buf) {}
#else
            : std::ostream(&buf) {}
#endif
        ogzstream(const char* name, int mode = std::ios_base::out, int comp = 1)
#ifdef _MSC_VER
            : gzstreambase(name, mode, comp), ostream(&buf) {}
#else
            : gzstreambase(name, mode, comp), std::ostream(&buf) {}
#endif
        gzstreambuf* rdbuf() { return gzstreambase::rdbuf(); }
        void open(const char* name, int open_mode = std::ios_base::out, int comp = 1)
        {
            gzstreambase::open(name, open_mode, comp);
        }
    };
 
} // namespace BAse
 
#endif // BIMHOMEBASE_GZSTREAM_H
// ============================================================================
// EOF //