387 lines
		
	
	
		
			8.1 KiB
		
	
	
	
		
			Haxe
		
	
	
	
	
	
		
		
			
		
	
	
			387 lines
		
	
	
		
			8.1 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. | ||
|  |  */ | ||
|  | 
 | ||
|  | import haxe.iterators.ArrayKeyValueIterator; | ||
|  | 
 | ||
|  | @:coreApi final class Array<T> { | ||
|  | 	private var __a:neko.NativeArray<T>; | ||
|  | 
 | ||
|  | 	public var length(default, null):Int; | ||
|  | 
 | ||
|  | 	public function new():Void { | ||
|  | 		this.__a = neko.NativeArray.alloc(0); | ||
|  | 		this.length = 0; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	private static function new1<T>(a:neko.NativeArray<T>, l:Int):Array<T> { | ||
|  | 		var inst = new Array<T>(); | ||
|  | 		inst.__a = a; | ||
|  | 		inst.length = l; | ||
|  | 		return inst; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public function concat(a:Array<T>):Array<T> { | ||
|  | 		var a1 = this.__a; | ||
|  | 		var a2 = a.__a; | ||
|  | 		var s1 = this.length; | ||
|  | 		var s2 = a.length; | ||
|  | 		var a = neko.NativeArray.alloc(s1 + s2); | ||
|  | 		neko.NativeArray.blit(a, 0, a1, 0, s1); | ||
|  | 		neko.NativeArray.blit(a, s1, a2, 0, s2); | ||
|  | 		return new1(a, s1 + s2); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public function copy():Array<T> { | ||
|  | 		return new1(neko.NativeArray.sub(this.__a, 0, this.length), this.length); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public inline function iterator():haxe.iterators.ArrayIterator<T> { | ||
|  | 		return new haxe.iterators.ArrayIterator(this); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public inline function keyValueIterator():ArrayKeyValueIterator<T> { | ||
|  | 		return new ArrayKeyValueIterator(this); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public function insert(pos:Int, x:T):Void { | ||
|  | 		var l = this.length; | ||
|  | 		if (pos < 0) { | ||
|  | 			pos = l + pos; | ||
|  | 			if (pos < 0) | ||
|  | 				pos = 0; | ||
|  | 		} | ||
|  | 		if (pos > l) | ||
|  | 			pos = l; | ||
|  | 		this.__grow(l + 1); | ||
|  | 		var a = this.__a; | ||
|  | 		neko.NativeArray.blit(a, pos + 1, a, pos, l - pos); | ||
|  | 		a[pos] = x; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public function join(sep:String):String { | ||
|  | 		var s = new StringBuf(); | ||
|  | 		var a = this.__a; | ||
|  | 		var max = this.length - 1; | ||
|  | 		for (p in 0...this.length) { | ||
|  | 			s.add(a[p]); | ||
|  | 			if (p != max) | ||
|  | 				s.add(sep); | ||
|  | 		} | ||
|  | 		return s.toString(); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	static var __hx_toString_depth = 0; | ||
|  | 
 | ||
|  | 	public function toString():String { | ||
|  | 		if (__hx_toString_depth >= 5) { | ||
|  | 			return "..."; | ||
|  | 		} | ||
|  | 		var s = new StringBuf(); | ||
|  | 		s.add("["); | ||
|  | 		var it = iterator(); | ||
|  | 		__hx_toString_depth++; | ||
|  | 		try { | ||
|  | 			for (i in it) { | ||
|  | 				s.add(i); | ||
|  | 				if (it.hasNext()) | ||
|  | 					s.addChar(",".code); | ||
|  | 			} | ||
|  | 			__hx_toString_depth--; | ||
|  | 		} catch (e:Dynamic) { | ||
|  | 			__hx_toString_depth--; | ||
|  | 			neko.Lib.rethrow(e); | ||
|  | 		} | ||
|  | 		s.add("]"); | ||
|  | 		return s.toString(); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public function pop():Null<T> { | ||
|  | 		if (this.length == 0) | ||
|  | 			return null; | ||
|  | 		this.length -= 1; | ||
|  | 		var x = this.__a[this.length]; | ||
|  | 		this.__a[this.length] = null; | ||
|  | 		return x; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public function push(x:T):Int { | ||
|  | 		var l = this.length; | ||
|  | 		this.__grow(l + 1); | ||
|  | 		this.__a[l] = x; | ||
|  | 		return l + 1; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public function unshift(x:T):Void { | ||
|  | 		var l = this.length; | ||
|  | 		this.__grow(l + 1); | ||
|  | 		var a = this.__a; | ||
|  | 		neko.NativeArray.blit(a, 1, a, 0, l); | ||
|  | 		a[0] = x; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public function remove(x:T):Bool { | ||
|  | 		var i = 0; | ||
|  | 		var l = this.length; | ||
|  | 		var a = this.__a; | ||
|  | 		while (i < l) { | ||
|  | 			if (a[i] == x) { | ||
|  | 				neko.NativeArray.blit(a, i, a, i + 1, l - i - 1); | ||
|  | 				l -= 1; | ||
|  | 				this.length = l; | ||
|  | 				a[l] = null; | ||
|  | 				return true; | ||
|  | 			} | ||
|  | 			i += 1; | ||
|  | 		} | ||
|  | 		return false; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public function contains(x:T):Bool { | ||
|  | 		var i = 0; | ||
|  | 		var l = this.length; | ||
|  | 		var a = this.__a; | ||
|  | 		while (i < l) { | ||
|  | 			if (a[i] == x) { | ||
|  | 				return true; | ||
|  | 			} | ||
|  | 			i += 1; | ||
|  | 		} | ||
|  | 		return false; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public function indexOf(x:T, ?fromIndex:Int):Int { | ||
|  | 		var len = length; | ||
|  | 		var i:Int = (fromIndex != null) ? fromIndex : 0; | ||
|  | 		var a = __a; | ||
|  | 		if (i < 0) { | ||
|  | 			i += len; | ||
|  | 			if (i < 0) | ||
|  | 				i = 0; | ||
|  | 		} | ||
|  | 		while (i < len) { | ||
|  | 			if (a[i] == x) | ||
|  | 				return i; | ||
|  | 			i++; | ||
|  | 		} | ||
|  | 		return -1; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public function lastIndexOf(x:T, ?fromIndex:Int):Int { | ||
|  | 		var len = length; | ||
|  | 		var i:Int = (fromIndex != null) ? fromIndex : len - 1; | ||
|  | 		var a = __a; | ||
|  | 		if (i >= len) | ||
|  | 			i = len - 1; | ||
|  | 		else if (i < 0) | ||
|  | 			i += len; | ||
|  | 		while (i >= 0) { | ||
|  | 			if (a[i] == x) | ||
|  | 				return i; | ||
|  | 			i--; | ||
|  | 		} | ||
|  | 		return -1; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public function reverse():Void { | ||
|  | 		var i = 0; | ||
|  | 		var l = this.length; | ||
|  | 		var a = this.__a; | ||
|  | 		var half = l >> 1; | ||
|  | 		l -= 1; | ||
|  | 		while (i < half) { | ||
|  | 			var tmp = a[i]; | ||
|  | 			a[i] = a[l - i]; | ||
|  | 			a[l - i] = tmp; | ||
|  | 			i += 1; | ||
|  | 		} | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public function shift():Null<T> { | ||
|  | 		var l = this.length; | ||
|  | 		if (l == 0) | ||
|  | 			return null; | ||
|  | 		var a = this.__a; | ||
|  | 		var x = a[0]; | ||
|  | 		l -= 1; | ||
|  | 		neko.NativeArray.blit(a, 0, a, 1, l); | ||
|  | 		a[l] = null; | ||
|  | 		this.length = l; | ||
|  | 		return x; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public function slice(pos:Int, ?end:Int):Array<T> { | ||
|  | 		if (pos < 0) { | ||
|  | 			pos = this.length + pos; | ||
|  | 			if (pos < 0) | ||
|  | 				pos = 0; | ||
|  | 		} | ||
|  | 		if (end == null) | ||
|  | 			end = this.length; | ||
|  | 		else if (end < 0) | ||
|  | 			end = this.length + end; | ||
|  | 		if (end > this.length) | ||
|  | 			end = this.length; | ||
|  | 		var len = end - pos; | ||
|  | 		if (len < 0) | ||
|  | 			return new Array(); | ||
|  | 		return new1(neko.NativeArray.sub(this.__a, pos, len), len); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public function sort(f:T->T->Int):Void { | ||
|  | 		var a = this.__a; | ||
|  | 		var i = 0; | ||
|  | 		var l = this.length; | ||
|  | 		while (i < l) { | ||
|  | 			var swap = false; | ||
|  | 			var j = 0; | ||
|  | 			var max = l - i - 1; | ||
|  | 			while (j < max) { | ||
|  | 				if (f(a[j], a[j + 1]) > 0) { | ||
|  | 					var tmp = a[j + 1]; | ||
|  | 					a[j + 1] = a[j]; | ||
|  | 					a[j] = tmp; | ||
|  | 					swap = true; | ||
|  | 				} | ||
|  | 				j += 1; | ||
|  | 			} | ||
|  | 			if (!swap) | ||
|  | 				break; | ||
|  | 			i += 1; | ||
|  | 		} | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public function splice(pos:Int, len:Int):Array<T> { | ||
|  | 		if (len < 0) | ||
|  | 			return new Array(); | ||
|  | 		if (pos < 0) { | ||
|  | 			pos = this.length + pos; | ||
|  | 			if (pos < 0) | ||
|  | 				pos = 0; | ||
|  | 		} | ||
|  | 		if (pos > this.length) { | ||
|  | 			pos = 0; | ||
|  | 			len = 0; | ||
|  | 		} else if (pos + len > this.length) { | ||
|  | 			len = this.length - pos; | ||
|  | 			if (len < 0) | ||
|  | 				len = 0; | ||
|  | 		} | ||
|  | 		var a = this.__a; | ||
|  | 		var ret = new1(neko.NativeArray.sub(a, pos, len), len); | ||
|  | 		var end = pos + len; | ||
|  | 		neko.NativeArray.blit(a, pos, a, end, this.length - end); | ||
|  | 		this.length -= len; | ||
|  | 		while (--len >= 0) | ||
|  | 			a[this.length + len] = null; | ||
|  | 		return ret; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public inline function map<S>(f:T->S):Array<S> { | ||
|  | 		var l = length; | ||
|  | 		var ret = new1(neko.NativeArray.alloc(l), l); | ||
|  | 		for (i in 0...l) { | ||
|  | 			ret[i] = f(this[i]); | ||
|  | 		} | ||
|  | 		return ret; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public inline function filter(f:T->Bool):Array<T> { | ||
|  | 		var ret = []; | ||
|  | 		for (elt in this) | ||
|  | 			if (f(elt)) | ||
|  | 				ret.push(elt); | ||
|  | 		return ret; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	public function resize(len:Int):Void { | ||
|  | 		if (length < len) { | ||
|  | 			__set(len - 1, null); | ||
|  | 		} else if (length > len) { | ||
|  | 			for (i in len...length) { | ||
|  | 				__a[i] = null; | ||
|  | 			} | ||
|  | 			this.length = len; | ||
|  | 		} | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/* NEKO INTERNAL */ | ||
|  | 	private function __get(pos:Int):T { | ||
|  | 		return this.__a[pos]; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	private function __set(pos:Int, v:T):T { | ||
|  | 		var a = this.__a; | ||
|  | 		if (this.length <= pos) { | ||
|  | 			var l = pos + 1; | ||
|  | 			var dlen = l - neko.NativeArray.length(a); | ||
|  | 			if (dlen > 0) { | ||
|  | 				if (dlen == 1) { | ||
|  | 					this.__grow(l); | ||
|  | 					a = this.__a; | ||
|  | 				} else { | ||
|  | 					a = neko.NativeArray.alloc(l); | ||
|  | 					neko.NativeArray.blit(a, 0, this.__a, 0, this.length); | ||
|  | 					this.__a = a; | ||
|  | 				} | ||
|  | 			} | ||
|  | 			this.length = l; | ||
|  | 		} | ||
|  | 		a[pos] = v; | ||
|  | 		return v; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	private function __grow(l:Int):Void { | ||
|  | 		var a = this.__a; | ||
|  | 		var sz = neko.NativeArray.length(a); | ||
|  | 		if (sz >= l) { | ||
|  | 			this.length = l; | ||
|  | 			return; | ||
|  | 		} | ||
|  | 		var big = (sz * 3) >> 1; | ||
|  | 		if (big < l) | ||
|  | 			big = l; | ||
|  | 		var a2 = neko.NativeArray.alloc(big); | ||
|  | 		neko.NativeArray.blit(a2, 0, a, 0, this.length); | ||
|  | 		this.__a = a2; | ||
|  | 		this.length = l; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	private function __neko():neko.NativeArray<T> { | ||
|  | 		var a = this.__a; | ||
|  | 		var sz = neko.NativeArray.length(a); | ||
|  | 		if (sz != this.length) { | ||
|  | 			a = neko.NativeArray.sub(a, 0, this.length); | ||
|  | 			this.__a = a; | ||
|  | 		} | ||
|  | 		return a; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	#if !(macro || interp) | ||
|  | 	static function __init__():Void { | ||
|  | 		try { | ||
|  | 			var msort:Dynamic = neko.Lib.load("std", "merge_sort", 3); | ||
|  | 			untyped Array.prototype.sort = function(cmp) msort(__this__.__a, __this__.length, cmp); | ||
|  | 		} catch (e:Dynamic) {} | ||
|  | 	} | ||
|  | 	#end | ||
|  | } |