xref: /haiku/headers/private/support/ZstdCompressionAlgorithm.h (revision 268f99dd7dc4bd7474a8bd2742d3f1ec1de6752a)

/*
 * 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		= 22,
	B_ZSTD_COMPRESSION_DEFAULT	= 4,
};


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_