Client/GameClient/Assets/YooAsset/Runtime/Utility/BufferWriter.cs

using System;
using System.Collections;
using System.Collections.Generic;
using System.Text;
using System.Diagnostics;
using System.IO;

namespace YooAsset
{
	/// <summary>
	/// 数据存储以小端字节序为标准
	/// </summary>
	internal class BufferWriter
	{
		private readonly byte[] _buffer;
		private int _index = 0;

		public BufferWriter(int capacity)
		{
			_buffer = new byte[capacity];
		}

		/// <summary>
		/// 缓冲区容量
		/// </summary>
		public int Capacity
		{
			get { return _buffer.Length; }
		}

		/// <summary>
		/// 清空缓冲区
		/// </summary>
		public void Clear()
		{
			_index = 0;
		}

		/// <summary>
		/// 将有效数据写入文件流
		/// </summary>
		public void WriteToStream(FileStream fileStream)
		{
			fileStream.Write(_buffer, 0, _index);
		}

		public void WriteBytes(byte[] data)
		{
			int count = data.Length;
			CheckWriterIndex(count);
			Buffer.BlockCopy(data, 0, _buffer, _index, count);
			_index += count;
		}
		public void WriteByte(byte value)
		{
			CheckWriterIndex(1);
			_buffer[_index++] = value;
		}

		public void WriteBool(bool value)
		{
			WriteByte((byte)(value ? 1 : 0));
		}
		public void WriteInt16(short value)
		{
			WriteUInt16((ushort)value);
		}
		public void WriteUInt16(ushort value)
		{
			CheckWriterIndex(2);
			_buffer[_index++] = (byte)value;
			_buffer[_index++] = (byte)(value >> 8);
		}
		public void WriteInt32(int value)
		{
			WriteUInt32((uint)value);
		}
		public void WriteUInt32(uint value)
		{
			CheckWriterIndex(4);
			_buffer[_index++] = (byte)value;
			_buffer[_index++] = (byte)(value >> 8);
			_buffer[_index++] = (byte)(value >> 16);
			_buffer[_index++] = (byte)(value >> 24);
		}
		public void WriteInt64(long value)
		{
			WriteUInt64((ulong)value);
		}
		public void WriteUInt64(ulong value)
		{
			CheckWriterIndex(8);
			_buffer[_index++] = (byte)value;
			_buffer[_index++] = (byte)(value >> 8);
			_buffer[_index++] = (byte)(value >> 16);
			_buffer[_index++] = (byte)(value >> 24);
			_buffer[_index++] = (byte)(value >> 32);
			_buffer[_index++] = (byte)(value >> 40);
			_buffer[_index++] = (byte)(value >> 48);
			_buffer[_index++] = (byte)(value >> 56);
		}

		public void WriteUTF8(string value)
		{
			if (string.IsNullOrEmpty(value))
			{
				WriteUInt16(0);
			}
			else
			{
				byte[] bytes = Encoding.UTF8.GetBytes(value);
				int count = bytes.Length;
				if (count > ushort.MaxValue)
					throw new FormatException($"Write string length cannot be greater than {ushort.MaxValue} !");

				WriteUInt16(Convert.ToUInt16(count));
				WriteBytes(bytes);
			}
		}
		public void WriteInt32Array(int[] values)
		{
			if (values == null)
			{
				WriteUInt16(0);
			}
			else
			{
				int count = values.Length;
				if (count > ushort.MaxValue)
					throw new FormatException($"Write array length cannot be greater than {ushort.MaxValue} !");

				WriteUInt16(Convert.ToUInt16(count));
				for (int i = 0; i < count; i++)
				{
					WriteInt32(values[i]);
				}
			}
		}
		public void WriteInt64Array(long[] values)
		{
			if (values == null)
			{
				WriteUInt16(0);
			}
			else
			{
				int count = values.Length;
				if (count > ushort.MaxValue)
					throw new FormatException($"Write array length cannot be greater than {ushort.MaxValue} !");

				WriteUInt16(Convert.ToUInt16(count));
				for (int i = 0; i < count; i++)
				{
					WriteInt64(values[i]);
				}
			}
		}
		public void WriteUTF8Array(string[] values)
		{
			if (values == null)
			{
				WriteUInt16(0);
			}
			else
			{
				int count = values.Length;
				if (count > ushort.MaxValue)
					throw new FormatException($"Write array length cannot be greater than {ushort.MaxValue} !");

				WriteUInt16(Convert.ToUInt16(count));
				for (int i = 0; i < count; i++)
				{
					WriteUTF8(values[i]);
				}
			}
		}

		[Conditional("DEBUG")]
		private void CheckWriterIndex(int length)
		{
			if (_index + length > Capacity)
			{
				throw new IndexOutOfRangeException();
			}
		}
	}
}