123

CrySystem/CompressedFile.cpp

@@ -0,0 +1,97 @@
+#include "stdafx.h"
+#include <ILog.h>
+#include <Stream.h>
+#include "System.h"
+#include "zlib/zlib.h"
+#if defined(LINUX)
+#	include <sys/io.h>
+#else
+#	include <io.h>
+#endif
+#if defined(LINUX)
+#include "ILog.h"
+#endif
+
+#include <fcntl.h>
+
+#define USE_COMPRESSION
+
+bool CSystem::WriteCompressedFile(char *filename, void *data, unsigned int bitlen)
+{
+	FILE *pFile = fxopen(filename, "wb+");
+	if (!pFile)
+		return false;
+
+#ifdef USE_COMPRESSION
+  gzFile f = gzdopen(fileno(pFile), "wb9");
+	gzwrite(f, &bitlen, sizeof(int));
+	gzwrite(f, data, BITS2BYTES(bitlen));
+	gzclose(f);
+#else
+	fwrite(&bitlen, sizeof(int), 1, pFile);
+	fwrite(data, BITS2BYTES(bitlen), 1, pFile);
+	fclose(pFile);
+#endif
+    
+	return true;
+};
+
+unsigned int CSystem::GetCompressedFileSize(char *filename)
+{
+	FILE *pFile=fxopen(filename, "rb");
+	if (!pFile) 
+		return 0;
+	
+	fseek(pFile,0,SEEK_END);
+	long nLen=ftell(pFile);
+	fseek(pFile,0,SEEK_SET);
+	if (nLen<=0)
+	{
+		// gzread works incorrectly if the filesize is 0
+		fclose(pFile);
+		return (0);
+	}	
+
+	fclose(pFile);
+	pFile=fxopen(filename, "rb");
+	if (!pFile) 
+		return 0;
+
+	unsigned int bitlen;
+	gzFile f = gzdopen(fileno(pFile), "rb9");
+	gzread(f, &bitlen, sizeof(int));
+	gzclose(f);
+	fclose(pFile);
+	return bitlen;
+}
+
+unsigned int CSystem::ReadCompressedFile(char *filename, void *data, unsigned int maxbitlen)
+{
+	FILE * pFile=fxopen(filename, "rb");
+	if (!pFile) 
+		return 0;
+
+	/*
+	fseek(pFile,0,SEEK_END);
+	long nLen=ftell(pFile);
+	fseek(pFile,0,SEEK_SET);
+	*/ 
+	
+	unsigned int bitlen;
+	
+#ifdef USE_COMPRESSION
+  gzFile f = gzdopen(fileno(pFile), "rb9");
+	gzread(f, &bitlen, sizeof(int));
+	assert(bitlen<=maxbitlen);  // FIXME: nicer if caller doesn't need to know buffer size in advance
+	gzread(f, data, BITS2BYTES(bitlen));
+	gzclose(f);
+	fclose(pFile);
+#else	
+	fread(pFile, &bitlen, sizeof(int));
+	assert(bitlen<=maxbitlen);  // FIXME: nicer if caller doesn't need to know buffer size in advance
+	fread(pFile, data, BITS2BYTES(bitlen));
+	fclose(pFile);
+#endif
+
+	return bitlen;
+};