98 lines
2.7 KiB
C++
98 lines
2.7 KiB
C++
/*
|
|
* Copyright 2017, Jérôme Duval.
|
|
* Copyright 2014, Ingo Weinhold, ingo_weinhold@gmx.de.
|
|
* Distributed under the terms of the MIT License.
|
|
*/
|
|
#ifndef _ZSTD_COMPRESSION_ALGORITHM_H_
|
|
#define _ZSTD_COMPRESSION_ALGORITHM_H_
|
|
|
|
|
|
#include <CompressionAlgorithm.h>
|
|
|
|
|
|
// compression level
|
|
enum {
|
|
B_ZSTD_COMPRESSION_NONE = 0,
|
|
B_ZSTD_COMPRESSION_FASTEST = 1,
|
|
B_ZSTD_COMPRESSION_BEST = 19,
|
|
B_ZSTD_COMPRESSION_DEFAULT = 2,
|
|
};
|
|
|
|
|
|
class BZstdCompressionParameters : public BCompressionParameters {
|
|
public:
|
|
BZstdCompressionParameters(
|
|
int compressionLevel
|
|
= B_ZSTD_COMPRESSION_DEFAULT);
|
|
virtual ~BZstdCompressionParameters();
|
|
|
|
int32 CompressionLevel() const;
|
|
void SetCompressionLevel(int32 level);
|
|
|
|
size_t BufferSize() const;
|
|
void SetBufferSize(size_t size);
|
|
|
|
private:
|
|
int32 fCompressionLevel;
|
|
size_t fBufferSize;
|
|
};
|
|
|
|
|
|
class BZstdDecompressionParameters : public BDecompressionParameters {
|
|
public:
|
|
BZstdDecompressionParameters();
|
|
virtual ~BZstdDecompressionParameters();
|
|
|
|
size_t BufferSize() const;
|
|
void SetBufferSize(size_t size);
|
|
|
|
private:
|
|
size_t fBufferSize;
|
|
};
|
|
|
|
|
|
class BZstdCompressionAlgorithm : public BCompressionAlgorithm {
|
|
public:
|
|
BZstdCompressionAlgorithm();
|
|
virtual ~BZstdCompressionAlgorithm();
|
|
|
|
virtual status_t CreateCompressingInputStream(BDataIO* input,
|
|
const BCompressionParameters* parameters,
|
|
BDataIO*& _stream);
|
|
virtual status_t CreateCompressingOutputStream(BDataIO* output,
|
|
const BCompressionParameters* parameters,
|
|
BDataIO*& _stream);
|
|
virtual status_t CreateDecompressingInputStream(BDataIO* input,
|
|
const BDecompressionParameters* parameters,
|
|
BDataIO*& _stream);
|
|
virtual status_t CreateDecompressingOutputStream(BDataIO* output,
|
|
const BDecompressionParameters* parameters,
|
|
BDataIO*& _stream);
|
|
|
|
virtual status_t CompressBuffer(const void* input,
|
|
size_t inputSize, void* output,
|
|
size_t outputSize, size_t& _compressedSize,
|
|
const BCompressionParameters* parameters
|
|
= NULL);
|
|
virtual status_t DecompressBuffer(const void* input,
|
|
size_t inputSize, void* output,
|
|
size_t outputSize,
|
|
size_t& _uncompressedSize,
|
|
const BDecompressionParameters* parameters
|
|
= NULL);
|
|
|
|
private:
|
|
struct CompressionStrategy;
|
|
struct DecompressionStrategy;
|
|
|
|
template<typename BaseClass, typename Strategy, typename StreamType> struct Stream;
|
|
template<typename BaseClass, typename Strategy, typename StreamType>
|
|
friend struct Stream;
|
|
|
|
private:
|
|
static status_t _TranslateZstdError(size_t error);
|
|
};
|
|
|
|
|
|
#endif // _ZSTD_COMPRESSION_ALGORITHM_H_
|