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