0ad/source/tools/atlas/DatafileIO/Stream/Memory.h

#include "Stream.h"

namespace DatafileIO
{

	class SeekableInputStream_mem : public SeekableInputStream
	{
	public:
		// 'Release' provides common buffer-release mechanisms, called
		// when this stream is destroyed
		struct Releaser {
			virtual void release(char* buffer) = 0;
		};
		struct Releaser_None : public Releaser {
			virtual void release(char* /*buffer*/) {};
		};
		struct Releaser_StreamBuffer : public Releaser {
			Releaser_StreamBuffer(InputStream* s) : stream(s) {}
			InputStream* stream;
			virtual void release(char* buffer) {
				stream->ReleaseBuffer(buffer);
			}
		};
		struct Releaser_NewArray : public Releaser {
			virtual void release(char* buffer) {
				delete[] buffer;
			}
		};

		SeekableInputStream_mem(char* data, size_t size, Releaser* releaser);
		~SeekableInputStream_mem();
		virtual off_t Tell() const;
		virtual bool IsOk() const;
		virtual void Seek(off_t pos, Stream::whence mode);
		virtual size_t Read(void* buffer, size_t size);
		virtual bool AcquireBuffer(void*& buffer, size_t& size, size_t max_size);
		virtual void ReleaseBuffer(void* buffer);
	private:
		Releaser* m_Releaser;
		char* m_Data;
		size_t m_Size;
		off_t m_Cursor;
	};


	// Magically decompresses l33t-compressed files.
	class Maybel33tInputStream : public SeekableInputStream
	{
	public:
		// Take ownership of a SeekableInputStream
		Maybel33tInputStream(SeekableInputStream* stream);
		~Maybel33tInputStream();

		virtual off_t Tell() const;
		virtual bool IsOk() const;
		virtual void Seek(off_t pos, Stream::whence mode);
		virtual size_t Read(void* buffer, size_t size);
		virtual bool AcquireBuffer(void*& buffer, size_t& size, size_t max_size);
		virtual void ReleaseBuffer(void* buffer);
	private:
		SeekableInputStream* m_Stream;
	};
}