unity-flash-tools/Assets/FlashTools/Scripts/Editor/FTSwfTools/SwfStreamReader.cs

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

using Ionic.Zlib;

namespace FTSwfTools {
	public class SwfStreamReader {
		struct BitContext {
			public byte CachedByte;
			public byte BitIndex;
		}
		BitContext   _bitContext;
		BinaryReader _binaryReader;

		long Length {
			get { return _binaryReader.BaseStream.Length; }
		}

		long Position {
			get { return _binaryReader.BaseStream.Position; }
		}

		long BytesLeft {
			get { return Length - Position; }
		}

		// ---------------------------------------------------------------------
		//
		// Public
		//
		// ---------------------------------------------------------------------

		public SwfStreamReader(byte[] data) {
			var memory_stream = new MemoryStream(data);
			_binaryReader = new BinaryReader(memory_stream);
		}

		public SwfStreamReader(Stream stream) {
			_binaryReader = new BinaryReader(stream);
		}

		public bool IsEOF {
			get { return Position >= Length; }
		}

		public void AlignToByte() {
			_bitContext.BitIndex   = 0;
			_bitContext.CachedByte = 0;
		}

		public byte[] ReadRest() {
			return ReadBytes((int)BytesLeft);
		}

		public bool ReadBit() {
			var bit_index = _bitContext.BitIndex & 0x07;
			if ( bit_index == 0 ) {
				_bitContext.CachedByte = ReadByte();
			}
			++_bitContext.BitIndex;
			return ((_bitContext.CachedByte << bit_index) & 0x80) != 0;
		}

		public byte ReadByte() {
			return _binaryReader.ReadByte();
		}

		public byte[] ReadBytes(int count) {
			return count <= 0
				? new byte[0]
				: _binaryReader.ReadBytes(count);
		}

		public char ReadChar() {
			return _binaryReader.ReadChar();
		}

		public char[] ReadChars(int count) {
			return count <= 0
				? new char[0]
				: _binaryReader.ReadChars(count);
		}

		public short ReadInt16() {
			return _binaryReader.ReadInt16();
		}

		public int ReadInt32() {
			return _binaryReader.ReadInt32();
		}

		public ushort ReadUInt16() {
			return _binaryReader.ReadUInt16();
		}

		public uint ReadUInt32() {
			return _binaryReader.ReadUInt32();
		}

		public float ReadFloat32() {
			return _binaryReader.ReadSingle();
		}

		public double ReadDouble64() {
			return _binaryReader.ReadDouble();
		}

		public int ReadSignedBits(uint count) {
			if ( count == 0 ) {
				return 0;
			}
			bool sign = ReadBit();
			var res = sign ? uint.MaxValue : 0;
			--count;
			for ( var i = 0; i < count; ++i ) {
				var bit = ReadBit();
				res = (res << 1 | (bit ? 1u : 0u));
			}
			return (int)res;
		}

		public uint ReadUnsignedBits(uint count) {
			if ( count == 0 ) {
				return 0;
			}
			uint res = 0;
			for ( var i = 0; i < count; ++i ) {
				var bit = ReadBit();
				res = (res << 1 | (bit ? 1u : 0u));
			}
			return res;
		}

		public string ReadString() {
			var bytes = new List<byte>();
			while ( true ) {
				var bt = ReadByte();
				if ( bt == 0 ) {
					break;
				}
				bytes.Add(bt);
			}
			return Encoding.UTF8.GetString(bytes.ToArray());
		}

		public float ReadFixedPoint_8_8() {
			var value = ReadInt16();
			return value / 256.0f;
		}

		public float ReadFixedPoint_16_16() {
			var value = ReadInt32();
			return value / 65536.0f;
		}

		public float ReadFixedPoint16(uint bits) {
			var value = ReadSignedBits(bits);
			return value / 65536.0f;
		}

		public uint ReadEncodedU32() {
			uint val = 0;
			var bt = ReadByte();
			val |= bt & 0x7Fu;
			if ( (bt & 0x80) == 0 ) {
				return val;
			}
			bt = ReadByte();
			val |= (bt & 0x7Fu) << 7;
			if ( (bt & 0x80) == 0 ) {
				return val;
			}
			bt = ReadByte();
			val |= (bt & 0x7Fu) << 14;
			if ( (bt & 0x80) == 0 ) {
				return val;
			}
			bt = ReadByte();
			val |= (bt & 0x7Fu) << 21;
			if ( (bt & 0x80) == 0 ) {
				return val;
			}
			bt = ReadByte();
			val |= (bt & 0x7Fu) << 28;
			return val;
		}

		static public MemoryStream DecompressZBytes(byte[] compressed_bytes) {
			var target     = new MemoryStream();
			var zip_stream = new ZlibStream(target, CompressionMode.Decompress);
			zip_stream.Write(compressed_bytes, 0, compressed_bytes.Length);
			target.Position = 0;
			return target;
		}

		static public SwfStreamReader DecompressZBytesToReader(byte[] compressd_bytes) {
			return new SwfStreamReader(DecompressZBytes(compressd_bytes));
		}
	}
}