Added IBufferedFileStream - a faster variant of IFileStream designed …

cmake/headerlist.cmake

@@ -1,5 +1,6 @@
 set(headers
 	common/IArchive.h
+	common/IBufferedFileStream.h
 	common/IBufferStream.h
 	common/IConsole.h
 	common/ICriticalSection.h

cmake/sourcelist.cmake

@@ -1,5 +1,6 @@
 set(sources
 	common/IArchive.cpp
+	common/IBufferedFileStream.cpp
 	common/IBufferStream.cpp
 	common/IConsole.cpp
 	common/IDataStream.cpp

common/IBufferedFileStream.cpp

@@ -0,0 +1,84 @@
+#include "IBufferedFileStream.h"
+
+IBufferedFileStream::IBufferedFileStream() { }
+
+IBufferedFileStream::~IBufferedFileStream()
+{
+	Close();
+}
+
+bool IBufferedFileStream::Open(const char* name)
+{
+	bool success = streamFile.Open(name);
+
+	if (success)
+	{
+		SetLength(streamFile.GetLength());
+		SetOffset(0);
+
+		streamFile.ReadBuf(streamBuffer.data(), streamLength);
+
+		streamFile.SetOffset(0);
+	}
+
+	return success;
+}
+
+bool IBufferedFileStream::Create(const char* name)
+{
+	bool success = streamFile.Create(name);
+
+	if (success)
+	{
+		SetLength(0);
+		SetOffset(0);
+	}
+
+	return success;
+}
+
+void IBufferedFileStream::Close(void)
+{
+	if (streamBufferHasWrite)
+	{
+		streamFile.SetLength(streamLength);
+
+		streamFile.SetOffset(0);
+
+		streamFile.WriteBuf(streamBuffer.data(), streamLength);
+
+		streamBufferHasWrite = false;
+	}
+
+	SetLength(0);
+	SetOffset(0);
+
+	streamFile.Close();
+}
+
+void IBufferedFileStream::ReadBuf(void* buf, UInt32 inLength)
+{
+	ASSERT_STR(inLength <= GetRemain(), "IBufferedFileStream::ReadBuf: hit eof");
+	ASSERT_STR(streamBuffer.size() >= streamOffset + inLength, "IBufferedFileStream::ReadBuf: hit buffer eof");
+
+	memcpy(buf, &streamBuffer[streamOffset], inLength);
+	streamOffset += inLength;
+}
+
+void IBufferedFileStream::WriteBuf(const void* buf, UInt32 inLength)
+{
+	streamBufferHasWrite = true;
+
+	if (streamBuffer.size() < streamOffset + inLength)
+		SetLength(streamOffset + inLength);
+
+	memcpy(&streamBuffer[streamOffset], buf, inLength);
+	streamOffset += inLength;
+}
+
+void IBufferedFileStream::SetLength(UInt64 length)
+{
+	streamBuffer.resize(length);
+
+	streamLength = length;
+}

common/IBufferedFileStream.h

@@ -0,0 +1,31 @@
+#pragma once
+
+#include "common/IDataStream.h"
+#include "common/IFileStream.h"
+#include <vector>
+
+/**
+ *	An input/output file stream with buffering, write file on close
+ */
+class IBufferedFileStream : public IDataStream
+{
+public:
+	IBufferedFileStream();
+	virtual ~IBufferedFileStream();
+
+	bool	Open(const char* name);
+	bool	Create(const char* name);
+
+	void	Close(void);
+
+	virtual void	ReadBuf(void* buf, UInt32 inLength);
+	virtual void	WriteBuf(const void* buf, UInt32 inLength);
+
+	void	SetLength(UInt64 length);
+
+private:
+	std::vector<UInt8>	streamBuffer;
+	IFileStream			streamFile;
+
+	bool	streamBufferHasWrite = false;
+};

common/IFileStream.h

@@ -3,7 +3,7 @@
 #include "common/IDataStream.h"
 
 /**
- *	An input file stream
+ *	An input/output file stream
  */
 class IFileStream : public IDataStream
 {