1 files changed, 131 insertions, 0 deletions
diff --git a/NW4RTools/OutputStream.cs b/NW4RTools/OutputStream.cs
new file mode 100644
index 0000000..3f250a2
--- /dev/null
+++ b/NW4RTools/OutputStream.cs
@@ -0,0 +1,131 @@
+using System;
+using System.IO;
+
+namespace NW4RTools {
+	public class OutputStream {
+		// TODO: look into making this (and InputStream) inherit from Stream, or something
+		public readonly ByteEndian Endian;
+		private readonly bool MustReverseArrays;
+		private readonly System.IO.MemoryStream BaseStream;
+
+		public int Position {
+			get { return (int)BaseStream.Position; }
+		}
+
+		public byte[] GetBuffer() {
+			return BaseStream.ToArray();
+		}
+
+		public OutputStream() {
+			BaseStream = new MemoryStream();
+			Endian = ByteEndian.BigEndian;
+			MustReverseArrays = BitConverter.IsLittleEndian;
+		}
+
+		public OutputStream(ByteEndian endian) {
+			BaseStream = new MemoryStream();
+			Endian = endian;
+
+			if (Endian == ByteEndian.BigEndian)
+				MustReverseArrays = BitConverter.IsLittleEndian;
+			else
+				MustReverseArrays = !BitConverter.IsLittleEndian;
+		}
+
+		public void Seek(int pos) {
+			if (pos < 0 || pos > BaseStream.Length)
+				throw new ArgumentOutOfRangeException();
+			
+			BaseStream.Position = pos;
+		}
+
+		public void WriteBytes(byte[] data) {
+			BaseStream.Write(data, 0, data.Length);
+		}
+
+		private void WriteReversedBytes(byte[] data) {
+			// TODO: figure out if this modifies the array
+			if (MustReverseArrays)
+				Array.Reverse(data);
+			BaseStream.Write(data, 0, data.Length);
+		}
+
+		public void WriteByte(byte val) {
+			BaseStream.WriteByte(val);
+		}
+
+		public void WriteSByte(sbyte val) {
+			WriteByte(unchecked((byte)val));
+		}
+
+		public void WriteInt16(Int16 val) {
+			WriteReversedBytes(BitConverter.GetBytes(val));
+		}
+
+		public void WriteUInt16(UInt16 val) {
+			WriteReversedBytes(BitConverter.GetBytes(val));
+		}
+
+		public void WriteInt32(Int32 val) {
+			WriteReversedBytes(BitConverter.GetBytes(val));
+		}
+
+		public void WriteUInt32(UInt32 val) {
+			WriteReversedBytes(BitConverter.GetBytes(val));
+		}
+
+		public void WriteFloat(float val) {
+			WriteReversedBytes(BitConverter.GetBytes(val));
+		}
+
+		public void WriteDouble(double val) {
+			WriteReversedBytes(BitConverter.GetBytes(val));
+		}
+
+		public void WriteColor(Color col) {
+			WriteByte(col.r);
+			WriteByte(col.g);
+			WriteByte(col.b);
+			WriteByte(col.a);
+		}
+
+		public void WriteVec2(Vec2 vec) {
+			WriteFloat(vec.x);
+			WriteFloat(vec.y);
+		}
+
+		public void WriteVec3(Vec3 vec) {
+			WriteFloat(vec.x);
+			WriteFloat(vec.y);
+			WriteFloat(vec.z);
+		}
+
+		public void WriteMatrix(Matrix m) {
+			WriteFloat(m.v00);
+			WriteFloat(m.v01);
+			WriteFloat(m.v02);
+			WriteFloat(m.v03);
+			WriteFloat(m.v10);
+			WriteFloat(m.v11);
+			WriteFloat(m.v12);
+			WriteFloat(m.v13);
+			WriteFloat(m.v20);
+			WriteFloat(m.v21);
+			WriteFloat(m.v22);
+			WriteFloat(m.v23);
+		}
+
+		public void WriteName(string name) {
+			byte[] encoded = System.Text.Encoding.GetEncoding("Shift_JIS").GetBytes(name);
+			WriteInt32(encoded.Length);
+			WriteBytes(encoded);
+
+			if ((encoded.Length & 3) != 0) {
+				for (int i = 4 - (encoded.Length & 3); i > 0; i--) {
+					WriteByte(0);
+				}
+			}
+		}
+	}
+}
+