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
 | |
| }
 |