191 lines
4.5 KiB
C++
191 lines
4.5 KiB
C++
#ifndef __FILESTREAM_H_
|
||
#define __FILESTREAM_H_
|
||
|
||
#include <zlib.h>
|
||
#include "memory/base_allocator.hpp"
|
||
|
||
namespace stream
|
||
{
|
||
//基础抽象流 流的基础类，提供流最基本的虚函数接口声明
|
||
class BaseStream
|
||
{
|
||
public:
|
||
BaseStream(BaseAllocator* alloc)
|
||
{
|
||
alloc_ = alloc;
|
||
};
|
||
virtual ~BaseStream() {};
|
||
|
||
virtual int getSize();
|
||
virtual bool setSize(int)
|
||
{
|
||
return false;
|
||
}
|
||
inline int getPosition()
|
||
{
|
||
return seek(0, SEEK_CUR);
|
||
}
|
||
int setPosition(const int tPosition);
|
||
virtual int seek(const int tOffset, const int Origin) = 0;
|
||
virtual int read(void* lpBuffer, const int tSizeToRead) = 0;
|
||
virtual int write(const void* lpBuffer, const size_t tSizeToWrite) = 0;
|
||
virtual int copyFrom(BaseStream& stream, int tSizeToCopy = 0);
|
||
protected:
|
||
BaseAllocator* alloc_;
|
||
};
|
||
|
||
/* 文件流，
|
||
对文件系统中的文件读写操作的类
|
||
*/
|
||
class FileStream
|
||
: public BaseStream
|
||
{
|
||
protected:
|
||
#ifdef _MSC_VER
|
||
HANDLE hdl_;
|
||
#else
|
||
FILE* fp_;
|
||
#endif
|
||
public:
|
||
#ifdef _MSC_VER
|
||
void setHandle(HANDLE Handle);
|
||
#endif
|
||
//文件的打开方式
|
||
enum FileAccessType
|
||
{
|
||
faRead = 0x0001, //读取
|
||
faWrite = 0x0002, //写入
|
||
faCreate = 0x1000, //创建
|
||
|
||
faShareRead = 0x0100, //读共享
|
||
faShareWrite = 0x0200, //写共享
|
||
faShareDelete = 0x0400, //删除共享
|
||
};
|
||
//win32文件创建的模式
|
||
enum OpenDisposition
|
||
{
|
||
CreateIfNotExists = 1, //文件不存在则创建
|
||
AlwaysCreate, //总是创建文件，如果文件存在则将文件截断为0字节
|
||
OpenExistsOnly, //仅打开存在的文件
|
||
AlwaysOpen, //总是打开文件，若文件存在则直接打开，否则尝试创建文件并打开
|
||
TruncExistsOnly, //如果文件存在则打开文件并截断为0字节
|
||
};
|
||
private:
|
||
char fn_[256];
|
||
protected:
|
||
void setFileName(const char* lpFileName);
|
||
void construct(const char* lpFileName, unsigned int dwAccessType, unsigned int dwWin32CreationDisposition);
|
||
public:
|
||
FileStream(const char* lpFileName, unsigned int dwAccessType, BaseAllocator* alloc);
|
||
FileStream(const char* lpFileName, unsigned int dwAccessType, OpenDisposition eWin32CreateionDisposition, BaseAllocator* alloc);
|
||
~FileStream();
|
||
int seek(const int tOffset, const int Origin);
|
||
int read(void* lpBuffer, const int tSizeToRead);
|
||
int write(const void* lpBuffer, const size_t tSizeToWrite);
|
||
inline const char* getFileName()
|
||
{
|
||
return fn_;
|
||
}
|
||
};
|
||
|
||
|
||
/* 内存流
|
||
使用内存作为数据存储区的流对象类
|
||
*/
|
||
class MemoryStream
|
||
: public BaseStream
|
||
{
|
||
public:
|
||
typedef BaseStream Inherited;
|
||
|
||
private:
|
||
char* mem_; //内存块指针
|
||
char* ptr_; //流指针
|
||
char* end_ptr_; //文件尾指针
|
||
char* mem_end_; //内存块结束地址指针
|
||
int alloc_count_; // 每次扩展至少增加的数量
|
||
protected:
|
||
//如果需要处理内存申请和释放相关事宜，则在继承类中覆盖此函数即可
|
||
virtual char* AllocStream(char* ptr, const int size);
|
||
|
||
public:
|
||
MemoryStream(BaseAllocator* alloc);
|
||
~MemoryStream();
|
||
|
||
bool setSize(int tSize);
|
||
int seek(const int tOffset, const int Origin);
|
||
int read(void* lpBuffer, const int tSizeToRead);
|
||
int write(const void* lpBuffer, const size_t tSizeToWrite);
|
||
void setAllocSize(int size)
|
||
{
|
||
alloc_count_ = size;
|
||
}
|
||
//获取内存流的内存块指针
|
||
inline void* getMemory()
|
||
{
|
||
return mem_;
|
||
}
|
||
//从文件加流数据到内存
|
||
int loadFromFile(const char* lpFileName);
|
||
//将流数据保存到文件
|
||
int saveToFile(const char* lpFileName);
|
||
};
|
||
|
||
/* ZLib解压流
|
||
可对数据进行解压并写入到流中
|
||
*/
|
||
class ZDecompressionStream
|
||
: public BaseStream
|
||
{
|
||
public:
|
||
typedef BaseStream Inherited;
|
||
protected:
|
||
BaseStream* strm_; //zlib操作的目标流
|
||
int strm_pos_; //目标流的指针位置
|
||
z_stream z_strm_; //zlib操作的z_stream结构
|
||
Bytef buff_[0xffff]; //zlib操作缓冲区
|
||
int last_code_; //上一次操作的返回码
|
||
public:
|
||
ZDecompressionStream(BaseStream& source, BaseAllocator* alloc);
|
||
~ZDecompressionStream();
|
||
|
||
int seek(const int tOffset, const int Origin);
|
||
int read(void* lpBuffer, const int tSizeToRead);
|
||
int write(const void*, const size_t)
|
||
{
|
||
return -1;
|
||
}
|
||
int copyFrom(BaseStream&, int)
|
||
{
|
||
return 0;
|
||
}
|
||
private:
|
||
static void* zlibAllocMem(ZDecompressionStream& ZStream, uInt Items, uInt Size);
|
||
static void zlibFreeMem(ZDecompressionStream& ZStream, void* lpBlock);
|
||
|
||
protected:
|
||
//zlib操作内存申请函数，如果关心内存的申请和释放，则可在继承类中覆盖此函数
|
||
void* AllocStream(int size);
|
||
//zlib操作释放内存函数，如果关心内存的申请和释放，则可在继承类中覆盖此函数
|
||
void SFree(void* ptr);
|
||
//zlib操作进度通知函数，如果关心进度，则在子类中覆盖此函数即可
|
||
void DoProgress();
|
||
//设置上当前操作的返回码
|
||
inline void setLastCode(const int code)
|
||
{
|
||
last_code_ = code;
|
||
}
|
||
//获取上一次操作的返回码
|
||
inline int getLastCode()
|
||
{
|
||
return last_code_;
|
||
}
|
||
inline int succeeded()
|
||
{
|
||
return last_code_ >= 0;
|
||
}
|
||
};
|
||
};
|
||
|
||
#endif
|