mirror of
https://github.com/fluencelabs/assemblyscript
synced 2025-06-12 14:31:28 +00:00
Static operator overloads
This commit is contained in:
@ -1,10 +1,12 @@
|
||||
// singleton empty string
|
||||
const EMPTY: String = changetype<String>("");
|
||||
|
||||
// number of bytes preceeding string data
|
||||
const HEAD: usize = 4;
|
||||
|
||||
function allocate(length: i32): String {
|
||||
assert(<u32>length >= 1);
|
||||
var ptr = allocate_memory(HEAD + length * 2);
|
||||
assert(length > 0); // 0 -> EMPTY
|
||||
var ptr = allocate_memory(HEAD + (<usize>length << 1));
|
||||
store<i32>(ptr, length);
|
||||
return changetype<String>(ptr);
|
||||
}
|
||||
@ -15,45 +17,75 @@ export class String {
|
||||
|
||||
@operator("[]")
|
||||
charAt(pos: i32): String {
|
||||
assert(this != null);
|
||||
|
||||
if (<u32>pos >= this.length)
|
||||
return EMPTY;
|
||||
|
||||
var out = allocate(1);
|
||||
store<u16>(
|
||||
changetype<usize>(out),
|
||||
load<u16>(changetype<usize>(this) + (<usize>pos << 1), HEAD),
|
||||
load<u16>(
|
||||
changetype<usize>(this) + (<usize>pos << 1),
|
||||
HEAD
|
||||
),
|
||||
HEAD
|
||||
);
|
||||
return out;
|
||||
}
|
||||
|
||||
charCodeAt(pos: i32): i32 {
|
||||
assert(this != null);
|
||||
|
||||
if (<u32>pos >= this.length)
|
||||
return -1; // NaN
|
||||
return load<u16>(changetype<usize>(this) + (<usize>pos << 1), HEAD);
|
||||
return -1; // (NaN)
|
||||
|
||||
return load<u16>(
|
||||
changetype<usize>(this) + (<usize>pos << 1),
|
||||
HEAD
|
||||
);
|
||||
}
|
||||
|
||||
codePointAt(pos: i32): i32 {
|
||||
assert(this != null);
|
||||
|
||||
if (<u32>pos >= this.length)
|
||||
return -1; // undefined
|
||||
var first = <i32>load<u16>(changetype<usize>(this) + (<usize>pos << 1), HEAD);
|
||||
return -1; // (undefined)
|
||||
var first = <i32>load<u16>(
|
||||
changetype<usize>(this) + (<usize>pos << 1),
|
||||
HEAD
|
||||
);
|
||||
if (first < 0xD800 || first > 0xDBFF || pos + 1 == this.length)
|
||||
return first;
|
||||
var second = <i32>load<u16>(changetype<usize>(this) + ((<usize>pos + 1) << 1), HEAD);
|
||||
var second = <i32>load<u16>(
|
||||
changetype<usize>(this) + ((<usize>pos + 1) << 1),
|
||||
HEAD
|
||||
);
|
||||
if (second < 0xDC00 || second > 0xDFFF)
|
||||
return first;
|
||||
return ((first - 0xD800) << 10) + (second - 0xDC00) + 0x10000;
|
||||
}
|
||||
|
||||
@operator("+")
|
||||
private static __concat(left: String, right: String): String {
|
||||
if (left == null)
|
||||
left = changetype<String>("null");
|
||||
return left.concat(right);
|
||||
}
|
||||
|
||||
concat(other: String): String {
|
||||
assert(this != null);
|
||||
assert(other != null);
|
||||
|
||||
if (other == null)
|
||||
other = changetype<String>("null");
|
||||
|
||||
var thisLen: isize = this.length;
|
||||
var otherLen: isize = other.length;
|
||||
var len: usize = thisLen + otherLen;
|
||||
if (len == 0)
|
||||
var outLen: usize = thisLen + otherLen;
|
||||
if (outLen == 0)
|
||||
return EMPTY;
|
||||
var out = allocate(len);
|
||||
|
||||
var out = allocate(outLen);
|
||||
move_memory(
|
||||
changetype<usize>(out) + HEAD,
|
||||
changetype<usize>(this) + HEAD,
|
||||
@ -68,12 +100,17 @@ export class String {
|
||||
}
|
||||
|
||||
endsWith(searchString: String, endPosition: i32 = 0x7fffffff): bool {
|
||||
assert(searchString != null);
|
||||
var end: isize = <isize>min<i32>(max<i32>(endPosition, 0), this.length);
|
||||
assert(this != null);
|
||||
|
||||
if (searchString == null)
|
||||
return false;
|
||||
|
||||
var end: isize = <isize>min(max(endPosition, 0), this.length);
|
||||
var searchLength: isize = searchString.length;
|
||||
var start: isize = end - searchLength;
|
||||
if (start < 0)
|
||||
return false;
|
||||
|
||||
return !compare_memory(
|
||||
changetype<usize>(this) + HEAD + (start << 1),
|
||||
changetype<usize>(searchString) + HEAD,
|
||||
@ -82,17 +119,20 @@ export class String {
|
||||
}
|
||||
|
||||
@operator("==")
|
||||
private __eq(other: String): bool {
|
||||
if (this == null)
|
||||
return other == null;
|
||||
else if (other == null)
|
||||
private static __eq(left: String, right: String): bool {
|
||||
if (left == null)
|
||||
return right == null;
|
||||
else if (right == null)
|
||||
return false;
|
||||
if (this.length != other.length)
|
||||
|
||||
var leftLength = left.length;
|
||||
if (leftLength != right.length)
|
||||
return false;
|
||||
|
||||
return !compare_memory(
|
||||
changetype<usize>(this) + HEAD,
|
||||
changetype<usize>(other) + HEAD,
|
||||
<usize>(this.length << 1)
|
||||
changetype<usize>(left) + HEAD,
|
||||
changetype<usize>(right) + HEAD,
|
||||
(<usize>leftLength << 1)
|
||||
);
|
||||
}
|
||||
|
||||
@ -101,11 +141,17 @@ export class String {
|
||||
}
|
||||
|
||||
indexOf(searchString: String, position: i32 = 0): i32 {
|
||||
assert(searchString != null);
|
||||
assert(this != null);
|
||||
|
||||
if (searchString == null)
|
||||
searchString = changetype<String>("null");
|
||||
|
||||
var pos: isize = position;
|
||||
var len: isize = this.length;
|
||||
var start: isize = min<isize>(max<isize>(pos, 0), len);
|
||||
var searchLen: isize = searchString.length;
|
||||
var searchLen: isize = <isize>searchString.length;
|
||||
|
||||
// TODO: two-way, multiple char codes
|
||||
for (var k: usize = start; <isize>k + searchLen <= len; ++k)
|
||||
if (!compare_memory(
|
||||
changetype<usize>(this) + HEAD + (k << 1),
|
||||
@ -118,13 +164,17 @@ export class String {
|
||||
|
||||
startsWith(searchString: String, position: i32 = 0): bool {
|
||||
assert(this != null);
|
||||
assert(searchString != null);
|
||||
|
||||
if (searchString == null)
|
||||
searchString = changetype<String>("null");
|
||||
|
||||
var pos: isize = position;
|
||||
var len: isize = this.length;
|
||||
var start: isize = min<isize>(max<isize>(position, 0), len);
|
||||
var searchLength: isize = searchString.length;
|
||||
var searchLength: isize = <isize>searchString.length;
|
||||
if (searchLength + start > len)
|
||||
return false;
|
||||
|
||||
return !compare_memory(
|
||||
changetype<usize>(this) + HEAD + (start << 1),
|
||||
changetype<usize>(searchString) + HEAD,
|
||||
@ -134,14 +184,17 @@ export class String {
|
||||
|
||||
substr(start: i32, length: i32 = i32.MAX_VALUE): String {
|
||||
assert(this != null);
|
||||
|
||||
var intStart: isize = start;
|
||||
var end: isize = length;
|
||||
var size: isize = this.length;
|
||||
if (intStart < 0)
|
||||
intStart = max<isize>(size + intStart, 0);
|
||||
|
||||
var resultLength: isize = min<isize>(max<isize>(end, 0), size - intStart);
|
||||
if (resultLength <= 0)
|
||||
return EMPTY;
|
||||
|
||||
var out = allocate(resultLength);
|
||||
move_memory(
|
||||
changetype<usize>(out) + HEAD,
|
||||
@ -153,6 +206,7 @@ export class String {
|
||||
|
||||
substring(start: i32, end: i32 = i32.MAX_VALUE): String {
|
||||
assert(this != null);
|
||||
|
||||
var len = this.length;
|
||||
var finalStart = min<i32>(max<i32>(start, 0), len);
|
||||
var finalEnd = min<i32>(max<i32>(end, 0), len);
|
||||
@ -161,8 +215,10 @@ export class String {
|
||||
len = to - from;
|
||||
if (!len)
|
||||
return EMPTY;
|
||||
|
||||
if (!from && to == this.length)
|
||||
return this;
|
||||
|
||||
var out = allocate(len);
|
||||
move_memory(
|
||||
changetype<usize>(out) + HEAD,
|
||||
@ -174,17 +230,21 @@ export class String {
|
||||
|
||||
trim(): String {
|
||||
assert(this != null);
|
||||
|
||||
var length: usize = this.length;
|
||||
while (length && isWhiteSpaceOrLineTerminator(load<u16>(changetype<usize>(this) + (length << 1), HEAD)))
|
||||
--length;
|
||||
|
||||
var start: usize = 0;
|
||||
while (start < length && isWhiteSpaceOrLineTerminator(load<u16>(changetype<usize>(this) + (start << 1), HEAD))) {
|
||||
++start; --length;
|
||||
}
|
||||
while (start < length && isWhiteSpaceOrLineTerminator(load<u16>(changetype<usize>(this) + (start << 1), HEAD)))
|
||||
++start, --length;
|
||||
|
||||
if (!length)
|
||||
return EMPTY;
|
||||
|
||||
if (!start && length == this.length)
|
||||
return this;
|
||||
|
||||
var out = allocate(length);
|
||||
move_memory(
|
||||
changetype<usize>(out) + HEAD,
|
||||
@ -196,15 +256,19 @@ export class String {
|
||||
|
||||
trimLeft(): String {
|
||||
assert(this != null);
|
||||
|
||||
var start: isize = 0;
|
||||
var len: isize = this.length;
|
||||
while (start < len && isWhiteSpaceOrLineTerminator(load<u16>(changetype<usize>(this) + (start << 1), HEAD)))
|
||||
++start;
|
||||
|
||||
if (!start)
|
||||
return this;
|
||||
|
||||
var outLen = len - start;
|
||||
if (!outLen)
|
||||
return EMPTY;
|
||||
|
||||
var out = allocate(outLen);
|
||||
move_memory(
|
||||
changetype<usize>(out) + HEAD,
|
||||
@ -216,13 +280,17 @@ export class String {
|
||||
|
||||
trimRight(): String {
|
||||
assert(this != null);
|
||||
|
||||
var len: isize = this.length;
|
||||
while (len > 0 && isWhiteSpaceOrLineTerminator(load<u16>(changetype<usize>(this) + (len << 1), HEAD)))
|
||||
--len;
|
||||
|
||||
if (len <= 0)
|
||||
return EMPTY;
|
||||
|
||||
if (<i32>len == this.length)
|
||||
return this;
|
||||
|
||||
var out = allocate(len);
|
||||
move_memory(
|
||||
changetype<usize>(out) + HEAD,
|
||||
|
Reference in New Issue
Block a user