119 lines
3.7 KiB
C++
119 lines
3.7 KiB
C++
#pragma once
|
|
#include "ZlibBase.h"
|
|
#include <vector>
|
|
#include <string>
|
|
|
|
namespace ZlibCpp {
|
|
|
|
/**
|
|
* @brief zlib压缩器类
|
|
*/
|
|
class ZlibCompressor : public ZlibBase {
|
|
public:
|
|
/**
|
|
* @brief 构造函数
|
|
* @param level 压缩级别
|
|
* @param format 数据格式
|
|
* @param strategy 压缩策略
|
|
*/
|
|
ZlibCompressor(CompressionLevel level = CompressionLevel::DefaultCompression,
|
|
DataFormat format = DataFormat::Zlib,
|
|
CompressionStrategy strategy = CompressionStrategy::Default);
|
|
|
|
/**
|
|
* @brief 析构函数
|
|
*/
|
|
~ZlibCompressor() override;
|
|
|
|
/**
|
|
* @brief 初始化压缩器
|
|
* @param level 压缩级别
|
|
* @param format 数据格式
|
|
* @param strategy 压缩策略
|
|
*/
|
|
void Initialize(CompressionLevel level = CompressionLevel::DefaultCompression,
|
|
DataFormat format = DataFormat::Zlib,
|
|
CompressionStrategy strategy = CompressionStrategy::Default);
|
|
|
|
/**
|
|
* @brief 重置压缩器状态
|
|
*/
|
|
void Reset() override;
|
|
|
|
/**
|
|
* @brief 压缩数据
|
|
* @param input 输入数据
|
|
* @param flush 刷新模式
|
|
* @return 压缩后的数据
|
|
*/
|
|
std::vector<uint8_t> Compress(const std::vector<uint8_t>& input, bool flush = false);
|
|
|
|
/**
|
|
* @brief 压缩数据
|
|
* @param input 输入数据指针
|
|
* @param input_size 输入数据大小
|
|
* @param flush 刷新模式
|
|
* @return 压缩后的数据
|
|
*/
|
|
std::vector<uint8_t> Compress(const uint8_t* input, size_t input_size, bool flush = false);
|
|
|
|
/**
|
|
* @brief 完成压缩
|
|
* @return 剩余的压缩数据
|
|
*/
|
|
std::vector<uint8_t> Finish();
|
|
|
|
/**
|
|
* @brief 一次性压缩数据
|
|
* @param input 输入数据
|
|
* @param level 压缩级别
|
|
* @param format 数据格式
|
|
* @return 压缩后的数据
|
|
*/
|
|
static std::vector<uint8_t> CompressData(const std::vector<uint8_t>& input,
|
|
CompressionLevel level = CompressionLevel::DefaultCompression,
|
|
DataFormat format = DataFormat::Zlib);
|
|
|
|
/**
|
|
* @brief 一次性压缩字符串
|
|
* @param input 输入字符串
|
|
* @param level 压缩级别
|
|
* @param format 数据格式
|
|
* @return 压缩后的数据
|
|
*/
|
|
static std::vector<uint8_t> CompressString(const std::string& input,
|
|
CompressionLevel level = CompressionLevel::DefaultCompression,
|
|
DataFormat format = DataFormat::Zlib);
|
|
|
|
/**
|
|
* @brief 设置字典
|
|
* @param dictionary 字典数据
|
|
*/
|
|
void SetDictionary(const std::vector<uint8_t>& dictionary);
|
|
|
|
/**
|
|
* @brief 获取压缩参数
|
|
*/
|
|
void GetParams(CompressionLevel& level, CompressionStrategy& strategy) const;
|
|
|
|
/**
|
|
* @brief 设置压缩参数
|
|
*/
|
|
void SetParams(CompressionLevel level, CompressionStrategy strategy);
|
|
|
|
protected:
|
|
void Cleanup() override;
|
|
|
|
private:
|
|
CompressionLevel level_;
|
|
DataFormat format_;
|
|
CompressionStrategy strategy_;
|
|
bool finished_;
|
|
|
|
/**
|
|
* @brief 内部压缩实现
|
|
*/
|
|
std::vector<uint8_t> CompressInternal(const uint8_t* input, size_t input_size, int flush_mode);
|
|
};
|
|
|
|
} // namespace ZlibCpp
|