105 lines
2.8 KiB
C
105 lines
2.8 KiB
C
|
#pragma once
|
|||
|
|
#include "ZlibBase.h"
|
|||
|
|
#include <vector>
|
|||
|
|
#include <string>
|
|||
|
|
|
|||
|
|
namespace ZlibCpp {
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief zlib解压缩器类
|
|||
|
|
*/
|
|||
|
|
class ZlibDecompressor : public ZlibBase {
|
|||
|
|
public:
|
|||
|
|
/**
|
|||
|
|
* @brief 构造函数
|
|||
|
|
* @param format 数据格式
|
|||
|
|
*/
|
|||
|
|
explicit ZlibDecompressor(DataFormat format = DataFormat::Zlib);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 析构函数
|
|||
|
|
*/
|
|||
|
|
~ZlibDecompressor() override;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 初始化解压缩器
|
|||
|
|
* @param format 数据格式
|
|||
|
|
*/
|
|||
|
|
void Initialize(DataFormat format = DataFormat::Zlib);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 重置解压缩器状态
|
|||
|
|
*/
|
|||
|
|
void Reset() override;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 解压缩数据
|
|||
|
|
* @param input 输入数据
|
|||
|
|
* @param flush 刷新模式
|
|||
|
|
* @return 解压缩后的数据
|
|||
|
|
*/
|
|||
|
|
std::vector<uint8_t> Decompress(const std::vector<uint8_t>& input, bool flush = false);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 解压缩数据
|
|||
|
|
* @param input 输入数据指针
|
|||
|
|
* @param input_size 输入数据大小
|
|||
|
|
* @param flush 刷新模式
|
|||
|
|
* @return 解压缩后的数据
|
|||
|
|
*/
|
|||
|
|
std::vector<uint8_t> Decompress(const uint8_t* input, size_t input_size, bool flush = false);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 完成解压缩
|
|||
|
|
* @return 剩余的解压缩数据
|
|||
|
|
*/
|
|||
|
|
std::vector<uint8_t> Finish();
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 一次性解压缩数据
|
|||
|
|
* @param input 输入数据
|
|||
|
|
* @param format 数据格式
|
|||
|
|
* @return 解压缩后的数据
|
|||
|
|
*/
|
|||
|
|
static std::vector<uint8_t> DecompressData(const std::vector<uint8_t>& input,
|
|||
|
|
DataFormat format = DataFormat::Zlib);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 一次性解压缩到字符串
|
|||
|
|
* @param input 输入数据
|
|||
|
|
* @param format 数据格式
|
|||
|
|
* @return 解压缩后的字符串
|
|||
|
|
*/
|
|||
|
|
static std::string DecompressString(const std::vector<uint8_t>& input,
|
|||
|
|
DataFormat format = DataFormat::Zlib);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 设置字典
|
|||
|
|
* @param dictionary 字典数据
|
|||
|
|
*/
|
|||
|
|
void SetDictionary(const std::vector<uint8_t>& dictionary);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 检查是否完成解压缩
|
|||
|
|
*/
|
|||
|
|
bool IsFinished() const;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 获取数据格式
|
|||
|
|
*/
|
|||
|
|
DataFormat GetFormat() const;
|
|||
|
|
|
|||
|
|
protected:
|
|||
|
|
void Cleanup() override;
|
|||
|
|
|
|||
|
|
private:
|
|||
|
|
DataFormat format_;
|
|||
|
|
bool finished_;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 内部解压缩实现
|
|||
|
|
*/
|
|||
|
|
std::vector<uint8_t> DecompressInternal(const uint8_t* input, size_t input_size, int flush_mode);
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
} // namespace ZlibCpp
|