120 lines
3.2 KiB
C++
120 lines
3.2 KiB
C++
#ifndef _WYL_ZLIB_STREAM_H_
|
||
#define _WYL_ZLIB_STREAM_H_
|
||
|
||
|
||
/******************************************************************
|
||
*
|
||
* wyLib库 2008 - 2010
|
||
*
|
||
* $ ZIP压缩、解压类 $
|
||
*
|
||
* - 主要功能 -
|
||
*
|
||
* 使用流的方式封装ZIP压缩和解压缩代码。
|
||
*
|
||
*****************************************************************/
|
||
|
||
|
||
namespace wylib
|
||
{
|
||
namespace zstream
|
||
{
|
||
|
||
/* ZLib流的基类
|
||
|
||
*/
|
||
|
||
class CBaseZStream
|
||
: public stream::CBaseStream
|
||
{
|
||
public:
|
||
typedef CBaseStream Inherited;
|
||
|
||
protected:
|
||
CBaseStream *m_pStrm; //zlib操作的目标流
|
||
LONGLONG m_nStrmPos; //目标流的指针位置
|
||
z_stream m_ZStrm; //zlib操作的z_stream结构
|
||
Bytef m_sBuffer[260]; //zlib操作缓冲区
|
||
int m_nLastCode; //上一次操作的返回码
|
||
|
||
private:
|
||
static void* zlibAllocMem(CBaseZStream& ZStream, uInt Items, uInt Size);
|
||
static void zlibFreeMem(CBaseZStream& ZStream, void *lpBlock);
|
||
|
||
protected:
|
||
//zlib操作内存申请函数，如果关心内存的申请和释放，则可在继承类中覆盖此函数
|
||
virtual void* Alloc(LONGLONG size);
|
||
//zlib操作释放内存函数，如果关心内存的申请和释放，则可在继承类中覆盖此函数
|
||
virtual void Free(void* ptr);
|
||
//zlib操作进度通知函数，如果关心进度，则在子类中覆盖此函数即可
|
||
virtual void DoProgress();
|
||
//设置上当前操作的返回码
|
||
inline void setLastCode(const int code){ m_nLastCode = code; }
|
||
public:
|
||
CBaseZStream(CBaseStream &stream);
|
||
virtual ~CBaseZStream();
|
||
|
||
//获取上一次操作的返回码
|
||
inline int getLastCode(){ return m_nLastCode; }
|
||
inline int succeeded(){ return m_nLastCode >= 0; }
|
||
};
|
||
|
||
|
||
/* ZLib压缩流
|
||
可对数据进行压缩并写入到流中
|
||
*/
|
||
class CZCompressionStream
|
||
: public CBaseZStream
|
||
{
|
||
public:
|
||
typedef CBaseZStream Inherited;
|
||
|
||
enum CompressionLevel
|
||
{
|
||
clNone = 0, //储存（未压缩）
|
||
clFastest, //最快压缩（速度快，节省时间但压缩比不高）
|
||
clDefault, //默认压缩等级（压缩速度和压缩比率均折中）
|
||
clMax, //最高压缩比（压缩速度慢，但压缩比较高）
|
||
};
|
||
|
||
public:
|
||
CZCompressionStream(CBaseStream& destStream, CompressionLevel CmprsLvl);
|
||
~CZCompressionStream();
|
||
|
||
LONGLONG seek(const LONGLONG tOffset, const int Origin);
|
||
LONGLONG read(LPVOID lpBuffer, const LONGLONG tSizeToRead){ return -1; };
|
||
LONGLONG write(LPCVOID lpBuffer, const LONGLONG tSizeToWrite);
|
||
//提交压缩缓冲中的数据到流中
|
||
void finish();
|
||
//获取压缩后的压缩比，返回的是一个浮点数
|
||
inline float getCompressionRate()
|
||
{
|
||
if ( m_ZStrm.total_in == 0 )
|
||
return 0;
|
||
else return (1.0f - ((float)m_ZStrm.total_out / m_ZStrm.total_in)) * 100.0f;
|
||
}
|
||
};
|
||
|
||
/* ZLib解压流
|
||
可对数据进行解压并写入到流中
|
||
*/
|
||
class CZDecompressionStream
|
||
: public CBaseZStream
|
||
{
|
||
public:
|
||
typedef CBaseZStream Inherited;
|
||
|
||
public:
|
||
CZDecompressionStream(CBaseStream &source);
|
||
~CZDecompressionStream();
|
||
|
||
LONGLONG seek(const LONGLONG tOffset, const int Origin);
|
||
LONGLONG read(LPVOID lpBuffer, const LONGLONG tSizeToRead);
|
||
LONGLONG write(LPCVOID lpBuffer, const LONGLONG tSizeToWrite){ return -1; };
|
||
};
|
||
};
|
||
}
|
||
|
||
#endif
|
||
|