218 lines
		
	
	
		
			5.3 KiB
		
	
	
	
		
			Haxe
		
	
	
	
	
	
			
		
		
	
	
			218 lines
		
	
	
		
			5.3 KiB
		
	
	
	
		
			Haxe
		
	
	
	
	
	
/*
 | 
						|
 * Copyright (C)2005-2019 Haxe Foundation
 | 
						|
 *
 | 
						|
 * Permission is hereby granted, free of charge, to any person obtaining a
 | 
						|
 * copy of this software and associated documentation files (the "Software"),
 | 
						|
 * to deal in the Software without restriction, including without limitation
 | 
						|
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 | 
						|
 * and/or sell copies of the Software, and to permit persons to whom the
 | 
						|
 * Software is furnished to do so, subject to the following conditions:
 | 
						|
 *
 | 
						|
 * The above copyright notice and this permission notice shall be included in
 | 
						|
 * all copies or substantial portions of the Software.
 | 
						|
 *
 | 
						|
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 | 
						|
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 | 
						|
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 | 
						|
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 | 
						|
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 | 
						|
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 | 
						|
 * DEALINGS IN THE SOFTWARE.
 | 
						|
 */
 | 
						|
 | 
						|
package haxe.io;
 | 
						|
 | 
						|
class BytesInput extends Input {
 | 
						|
	var b:#if js js.lib.Uint8Array #elseif hl hl.Bytes #else BytesData #end;
 | 
						|
	#if !flash
 | 
						|
	var pos:Int;
 | 
						|
	var len:Int;
 | 
						|
	var totlen:Int;
 | 
						|
	#end
 | 
						|
 | 
						|
	/** The current position in the stream in bytes. */
 | 
						|
	public var position(get, set):Int;
 | 
						|
 | 
						|
	/** The length of the stream in bytes. */
 | 
						|
	public var length(get, never):Int;
 | 
						|
 | 
						|
	public function new(b:Bytes, ?pos:Int, ?len:Int) {
 | 
						|
		if (pos == null)
 | 
						|
			pos = 0;
 | 
						|
		if (len == null)
 | 
						|
			len = b.length - pos;
 | 
						|
		if (pos < 0 || len < 0 || pos + len > b.length)
 | 
						|
			throw Error.OutsideBounds;
 | 
						|
		#if flash
 | 
						|
		var ba = b.getData();
 | 
						|
		ba.position = pos;
 | 
						|
		if (len != ba.bytesAvailable) {
 | 
						|
			// truncate
 | 
						|
			this.b = new flash.utils.ByteArray();
 | 
						|
			ba.readBytes(this.b, 0, len);
 | 
						|
		} else
 | 
						|
			this.b = ba;
 | 
						|
		this.b.endian = flash.utils.Endian.LITTLE_ENDIAN;
 | 
						|
		#else
 | 
						|
		this.b = #if (js || hl) @:privateAccess b.b #else b.getData() #end;
 | 
						|
		this.pos = pos;
 | 
						|
		this.len = len;
 | 
						|
		this.totlen = len;
 | 
						|
		#end
 | 
						|
		#if python
 | 
						|
		bigEndian = false;
 | 
						|
		#end
 | 
						|
	}
 | 
						|
 | 
						|
	inline function get_position():Int {
 | 
						|
		#if flash
 | 
						|
		return b.position;
 | 
						|
		#else
 | 
						|
		return pos;
 | 
						|
		#end
 | 
						|
	}
 | 
						|
 | 
						|
	inline function get_length():Int {
 | 
						|
		#if flash
 | 
						|
		return b.length;
 | 
						|
		#else
 | 
						|
		return totlen;
 | 
						|
		#end
 | 
						|
	}
 | 
						|
 | 
						|
	function set_position(p:Int):Int {
 | 
						|
		if (p < 0)
 | 
						|
			p = 0;
 | 
						|
		else if (p > length)
 | 
						|
			p = length;
 | 
						|
		#if flash
 | 
						|
		return b.position = p;
 | 
						|
		#else
 | 
						|
		len = totlen - p;
 | 
						|
		return pos = p;
 | 
						|
		#end
 | 
						|
	}
 | 
						|
 | 
						|
	public override function readByte():Int {
 | 
						|
		#if flash
 | 
						|
		return try b.readUnsignedByte() catch (e:Dynamic) throw new Eof();
 | 
						|
		#else
 | 
						|
		if (this.len == 0)
 | 
						|
			throw new Eof();
 | 
						|
		len--;
 | 
						|
		#if neko
 | 
						|
		return untyped __dollar__sget(b, pos++);
 | 
						|
		#elseif cpp
 | 
						|
		return untyped b[pos++];
 | 
						|
		#elseif java
 | 
						|
		return untyped b[pos++] & 0xFF;
 | 
						|
		#elseif python // dodge https://github.com/HaxeFoundation/haxe/issues/5080
 | 
						|
		var b = b[pos];
 | 
						|
		pos++;
 | 
						|
		return b;
 | 
						|
		#else
 | 
						|
		return b[pos++];
 | 
						|
		#end
 | 
						|
		#end
 | 
						|
	}
 | 
						|
 | 
						|
	public override function readBytes(buf:Bytes, pos:Int, len:Int):Int {
 | 
						|
		#if !neko
 | 
						|
		if (pos < 0 || len < 0 || pos + len > buf.length)
 | 
						|
			throw Error.OutsideBounds;
 | 
						|
		#end
 | 
						|
		#if flash
 | 
						|
		var avail:Int = b.bytesAvailable;
 | 
						|
		if (len > avail && avail > 0)
 | 
						|
			len = avail;
 | 
						|
		try
 | 
						|
			b.readBytes(buf.getData(), pos, len)
 | 
						|
		catch (e:Dynamic)
 | 
						|
			throw new Eof();
 | 
						|
		#elseif java
 | 
						|
		var avail:Int = this.len;
 | 
						|
		if (len > avail)
 | 
						|
			len = avail;
 | 
						|
		if (len == 0)
 | 
						|
			throw new Eof();
 | 
						|
		java.lang.System.arraycopy(this.b, this.pos, buf.getData(), pos, len);
 | 
						|
		this.pos += len;
 | 
						|
		this.len -= len;
 | 
						|
		#elseif cs
 | 
						|
		var avail:Int = this.len;
 | 
						|
		if (len > avail)
 | 
						|
			len = avail;
 | 
						|
		if (len == 0)
 | 
						|
			throw new Eof();
 | 
						|
		cs.system.Array.Copy(this.b, this.pos, buf.getData(), pos, len);
 | 
						|
		this.pos += len;
 | 
						|
		this.len -= len;
 | 
						|
		#else
 | 
						|
		if (this.len == 0 && len > 0)
 | 
						|
			throw new Eof();
 | 
						|
		if (this.len < len)
 | 
						|
			len = this.len;
 | 
						|
		#if neko
 | 
						|
		try
 | 
						|
			untyped __dollar__sblit(buf.getData(), pos, b, this.pos, len)
 | 
						|
		catch (e:Dynamic)
 | 
						|
			throw Error.OutsideBounds;
 | 
						|
		#elseif hl
 | 
						|
		@:privateAccess buf.b.blit(pos, b, this.pos, len);
 | 
						|
		#else
 | 
						|
		var b1 = b;
 | 
						|
		var b2 = #if js @:privateAccess buf.b #else buf.getData() #end;
 | 
						|
		for (i in 0...len)
 | 
						|
			b2[pos + i] = b1[this.pos + i];
 | 
						|
		#end
 | 
						|
		this.pos += len;
 | 
						|
		this.len -= len;
 | 
						|
		#end
 | 
						|
		return len;
 | 
						|
	}
 | 
						|
 | 
						|
	#if flash
 | 
						|
	@:dox(hide)
 | 
						|
	override function set_bigEndian(e) {
 | 
						|
		bigEndian = e;
 | 
						|
		b.endian = e ? flash.utils.Endian.BIG_ENDIAN : flash.utils.Endian.LITTLE_ENDIAN;
 | 
						|
		return e;
 | 
						|
	}
 | 
						|
 | 
						|
	@:dox(hide)
 | 
						|
	override function readFloat() {
 | 
						|
		return try b.readFloat() catch (e:Dynamic) throw new Eof();
 | 
						|
	}
 | 
						|
 | 
						|
	@:dox(hide)
 | 
						|
	override function readDouble() {
 | 
						|
		return try b.readDouble() catch (e:Dynamic) throw new Eof();
 | 
						|
	}
 | 
						|
 | 
						|
	@:dox(hide)
 | 
						|
	override function readInt8() {
 | 
						|
		return try b.readByte() catch (e:Dynamic) throw new Eof();
 | 
						|
	}
 | 
						|
 | 
						|
	@:dox(hide)
 | 
						|
	override function readInt16() {
 | 
						|
		return try b.readShort() catch (e:Dynamic) throw new Eof();
 | 
						|
	}
 | 
						|
 | 
						|
	@:dox(hide)
 | 
						|
	override function readUInt16():Int {
 | 
						|
		return try b.readUnsignedShort() catch (e:Dynamic) throw new Eof();
 | 
						|
	}
 | 
						|
 | 
						|
	@:dox(hide)
 | 
						|
	override function readInt32():Int {
 | 
						|
		return try b.readInt() catch (e:Dynamic) throw new Eof();
 | 
						|
	}
 | 
						|
 | 
						|
	@:dox(hide)
 | 
						|
	override function readString(len:Int, ?encoding:Encoding) {
 | 
						|
		return try encoding == RawNative ? b.readMultiByte(len, "unicode") : b.readUTFBytes(len) catch (e:Dynamic) throw new Eof();
 | 
						|
	}
 | 
						|
	#end
 | 
						|
}
 |