#pragma once
#include "FileHeader.h"


class Compression
{
public:
	// Enum maps directly some external tools
	enum ECompressionTypes
	{
		eCompressionType_None = 0,
		eCompressionType_RLE = 1,
		eCompressionType_LZXRLE = 2,
		eCompressionType_ZLIBRLE = 3,
		eCompressionType_PS3ZLIB = 4
	};

private:
	// 4J added so we can have separate contexts and rleBuf for different threads
	class ThreadStorage
	{
	public:
		Compression *compression;
		ThreadStorage();
		~ThreadStorage();
	};
	static DWORD tlsIdx;
	static ThreadStorage *tlsDefault;
public:
	// Each new thread that needs to use Compression will need to call one of the following 2 functions, to either create its own
	// local storage, or share the default storage already allocated by the main thread
	static void CreateNewThreadStorage();
	static void UseDefaultThreadStorage();
	static void ReleaseThreadStorage();

	static Compression *getCompression();

public:
	HRESULT Compress(void *pDestination, unsigned int *pDestSize, void *pSource, unsigned int SrcSize);
	HRESULT Decompress(void *pDestination, unsigned int *pDestSize, void *pSource, unsigned int SrcSize);
	HRESULT CompressLZXRLE(void *pDestination, unsigned int *pDestSize, void *pSource, unsigned int SrcSize);
	HRESULT DecompressLZXRLE(void *pDestination, unsigned int *pDestSize, void *pSource, unsigned int SrcSize);
	HRESULT CompressRLE(void *pDestination, unsigned int *pDestSize, void *pSource, unsigned int SrcSize);
	HRESULT DecompressRLE(void *pDestination, unsigned int *pDestSize, void *pSource, unsigned int SrcSize);
	static VOID VitaVirtualDecompress(void *pDestination, unsigned int *pDestSize, void *pSource, unsigned int SrcSize);

	void SetDecompressionType(ECompressionTypes type) { m_decompressType = type; }  // for loading a save from a different platform (Sony cloud storage cross play)
	ECompressionTypes GetDecompressionType() { return m_decompressType; }  
	void SetDecompressionType(ESavePlatform platform);

	Compression();
	~Compression();
private:

	HRESULT DecompressWithType(void *pDestination, unsigned int *pDestSize, void *pSource, unsigned int SrcSize);

#if defined __PS3__
#else
	XMEMCOMPRESSION_CONTEXT compressionContext;
	XMEMDECOMPRESSION_CONTEXT decompressionContext;
#endif
	CRITICAL_SECTION rleCompressLock;
	CRITICAL_SECTION rleDecompressLock;

	unsigned char rleCompressBuf[1024*100];
	static const unsigned int staticRleSize = 1024*200;
	unsigned char rleDecompressBuf[staticRleSize];
	ECompressionTypes m_decompressType;
	ECompressionTypes m_localDecompressType;

};

//extern Compression gCompression;

#if defined _WIN64
#define APPROPRIATE_COMPRESSION_TYPE Compression::eCompressionType_ZLIBRLE
#elif defined __PS3__
#define APPROPRIATE_COMPRESSION_TYPE Compression::eCompressionType_PS3ZLIB
#else
#define APPROPRIATE_COMPRESSION_TYPE Compression::eCompressionType_LZXRLE
#endif