UltimateZoneBuilder/Source/Core/IO/SerializerStream.cs

#region ================== Copyright (c) 2007 Pascal vd Heiden

/*
 * Copyright (c) 2007 Pascal vd Heiden, www.codeimp.com
 * This program is released under GNU General Public License
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 */

#endregion

#region ================== Namespaces

using System;
using System.Collections.Generic;
using CodeImp.DoomBuilder.Geometry;
using System.IO;

#endregion

namespace CodeImp.DoomBuilder.IO
{
	internal sealed class SerializerStream : IReadWriteStream, IDisposable
	{
		#region ================== Constants

		#endregion

		#region ================== Variables

		//private Stream stream;
		private readonly BinaryWriter writer;
		private readonly Dictionary<string, ushort> stringstable;
		private bool isdisposed; //mxd

		#endregion

		#region ================== Properties

		public bool IsWriting { get { return true; } }

		#endregion

		#region ================== Constructor / Destructor

		// Constructor
		public SerializerStream(Stream stream)
		{
			// Initialize
			//this.stream = stream;
			this.writer = new BinaryWriter(stream);
			this.stringstable = new Dictionary<string, ushort>(StringComparer.Ordinal);
		}

		//mxd
		public void Dispose()
		{
			// Not already disposed?
			if(!isdisposed)
			{
				if(writer != null) writer.Close();
				isdisposed = true;
			}
		}

		#endregion

		#region ================== Methods

		// Management
		public void Begin()
		{
			// First 4 bytes are reserved for the offset of the strings table
			const int offset = 0;
			writer.Write(offset);
		}

		public void End()
		{
			// Write the offset bytes
			int offset = (int)writer.BaseStream.Length;
			writer.Seek(0, SeekOrigin.Begin);
			writer.Write(offset);

			// Write the strings
			writer.Seek(0, SeekOrigin.End);
			foreach(KeyValuePair<string, ushort> str in stringstable)
				writer.Write(str.Key);
		}

		// Bidirectional
		public void rwInt(ref int v) { writer.Write(v); }

		public void rwByte(ref byte v) { writer.Write(v); }

		public void rwShort(ref short v) { writer.Write(v); }

		public void rwString(ref string v)
		{
			ushort index;
			if(stringstable.ContainsKey(v))
				index = stringstable[v];
			else
				index = stringstable[v] = (ushort)stringstable.Count;
			writer.Write(index);
		}

		public void rwLong(ref long v) { writer.Write(v); }

		public void rwUInt(ref uint v) { writer.Write(v); }

		public void rwUShort(ref ushort v) { writer.Write(v); }

		public void rwULong(ref ulong v) { writer.Write(v); }

		public void rwFloat(ref float v) { writer.Write(v); }

		public void rwDouble(ref double v) { writer.Write(v); }

		public void rwBool(ref bool v) { writer.Write(v); }

		public void rwVector2D(ref Vector2D v)
		{
			writer.Write(v.x);
			writer.Write(v.y);
		}

		public void rwVector3D(ref Vector3D v)
		{
			writer.Write(v.x);
			writer.Write(v.y);
			writer.Write(v.z);
		}

		// Write-only
		public void wInt(int v) { writer.Write(v); }

		public void wByte(byte v) { writer.Write(v); }

		public void wShort(short v) { writer.Write(v); }

		public void wString(string v)
		{
			ushort index;
			if(stringstable.ContainsKey(v))
				index = stringstable[v];
			else
				index = stringstable[v] = (ushort)stringstable.Count;
			writer.Write(index);
		}

		public void wLong(long v) { writer.Write(v); }

		public void wUInt(uint v) { writer.Write(v); }

		public void wUShort(ushort v) { writer.Write(v); }

		public void wULong(ulong v) { writer.Write(v); }

		public void wFloat(float v) { writer.Write(v); }

		public void wDouble(double v) { writer.Write(v); }

		public void wBool(bool v) { writer.Write(v); }

		public void wVector2D(Vector2D v)
		{
			writer.Write(v.x);
			writer.Write(v.y);
		}

		public void wVector3D(Vector3D v)
		{
			writer.Write(v.x);
			writer.Write(v.y);
			writer.Write(v.z);
		}

		// Read-only is not supported
		public void rInt(out int v) { v = 0; General.Fail("Read-only is not supported on serialization stream. Consider passing the element by reference for bidirectional support."); }

		public void rByte(out byte v) { v = 0; General.Fail("Read-only is not supported on serialization stream. Consider passing the element by reference for bidirectional support."); }

		public void rShort(out short v) { v = 0; General.Fail("Read-only is not supported on serialization stream. Consider passing the element by reference for bidirectional support."); }

		public void rString(out string v) { v = ""; General.Fail("Read-only is not supported on serialization stream. Consider passing the element by reference for bidirectional support."); }

		public void rLong(out long v) { v = 0; General.Fail("Read-only is not supported on serialization stream. Consider passing the element by reference for bidirectional support."); }

		public void rUInt(out uint v) { v = 0; General.Fail("Read-only is not supported on serialization stream. Consider passing the element by reference for bidirectional support."); }

		public void rUShort(out ushort v) { v = 0; General.Fail("Read-only is not supported on serialization stream. Consider passing the element by reference for bidirectional support."); }

		public void rULong(out ulong v) { v = 0; General.Fail("Read-only is not supported on serialization stream. Consider passing the element by reference for bidirectional support."); }

		public void rFloat(out float v) { v = 0; General.Fail("Read-only is not supported on serialization stream. Consider passing the element by reference for bidirectional support."); }

		public void rDouble(out double v) { v = 0; General.Fail("Read-only is not supported on serialization stream. Consider passing the element by reference for bidirectional support."); }

		public void rBool(out bool v) { v = false; General.Fail("Read-only is not supported on serialization stream. Consider passing the element by reference for bidirectional support."); }

		public void rVector2D(out Vector2D v)
		{
			v = new Vector2D();
			General.Fail("Read-only is not supported on serialization stream. Consider passing the element by reference for bidirectional support.");
		}

		public void rVector3D(out Vector3D v)
		{
			v = new Vector3D();
			General.Fail("Read-only is not supported on serialization stream. Consider passing the element by reference for bidirectional support.");
		}
		
		#endregion
	}
}