mlogjs 0.4.8 → 0.4.10

package/dist/index.mjs

@@ -146,7 +146,19 @@ const unaryOperators = [
 ];
 const updateOperators = ["++", "--"];
 const singleOperators = [...unaryOperators, ...updateOperators];
-const operators = [...leftRightOperators, ...singleOperators];
+const operators = [...leftRightOperators, ...singleOperators];
+const orderIndependentOperators = [
+    "!=",
+    "!==",
+    "&&",
+    "&",
+    "*",
+    "+",
+    "==",
+    "===",
+    "|",
+    "||",
+];
 
 class VoidValue {
     constructor() {
@@ -177,18 +189,18 @@ for (const key of operators) {
 }
 
 class InstructionBase {
-    constructor(...args) {
-        this.intent = EInstIntent.none;
-        this.alwaysRuns = true;
-        this._hidden = false;
-        this.args = args;
-    }
     get hidden() {
         return this._hidden;
     }
     set hidden(value) {
         this._hidden = value;
     }
+    constructor(...args) {
+        this.intent = EInstIntent.none;
+        this.alwaysRuns = true;
+        this._hidden = false;
+        this.args = args;
+    }
     resolve(_i) { }
     toString() {
         return this.args.filter(arg => arg).join(" ");
@@ -262,6 +274,11 @@ function camelToDashCase(name) {
 // eslint-disable-next-line @typescript-eslint/ban-types
 function assign(obj, props) {
     return Object.assign(obj, props);
+}
+/** Lazily computes and caches an expression */
+function lazy(fn) {
+    let value;
+    return () => (value ??= fn());
 }
 
 function appendSourceLocations(valueInst, node) {
@@ -401,14 +418,14 @@ class EndInstruction extends InstructionBase {
 }
 
 class AddressResolver extends InstructionBase {
-    constructor(...bonds) {
-        super();
-        this.bonds = bonds;
-    }
     get hidden() {
         return true;
     }
     set hidden(value) { }
+    constructor(...bonds) {
+        super();
+        this.bonds = bonds;
+    }
     resolve(i) {
         for (const literal of this.bonds)
             literal.data = i;
@@ -524,12 +541,16 @@ class BaseValue extends VoidValue {
             ],
         ];
     }
+    "!=="(scope, other, out) {
+        const [equal, equalInst] = this["==="](scope, other);
+        const [result, resultInst] = equal["!"](scope, out);
+        return [result, [...equalInst, ...resultInst]];
+    }
 }
 const operatorMap$1 = {
     "==": "equal",
     "===": "strictEqual",
     "!=": "notEqual",
-    "!==": "notEqual",
     "<": "lessThan",
     ">": "greaterThan",
     "<=": "lessThanEq",
@@ -616,6 +637,7 @@ class LiteralValue extends BaseValue {
         super();
         this.mutability = EMutability.constant;
         this.data = data;
+        this.name = this.toString();
     }
     eval(_scope) {
         return [this, []];
@@ -739,12 +761,40 @@ class StoreValue extends BaseValue {
         const [evalValue, evalInst] = value.eval(scope, this);
         return [this, [...evalInst, new SetInstruction(this, evalValue)]];
     }
+    "=="(scope, value, out) {
+        if (compareStores(this, value))
+            return [new LiteralValue(1), []];
+        return super["=="](scope, value, out);
+    }
+    "==="(scope, value, out) {
+        if (compareStores(this, value))
+            return [new LiteralValue(1), []];
+        return super["==="](scope, value, out);
+    }
+    "!="(scope, value, out) {
+        if (compareStores(this, value))
+            return [new LiteralValue(0), []];
+        return super["!="](scope, value, out);
+    }
+    "!=="(scope, value, out) {
+        if (compareStores(this, value))
+            return [new LiteralValue(0), []];
+        return super["!=="](scope, value, out);
+    }
     eval(_scope) {
         return [this, []];
     }
     toString() {
         return this.name;
     }
+}
+function compareStores(left, right) {
+    return (left === right ||
+        // prevents a store from being equal to a senseable value
+        (right instanceof left.constructor &&
+            right.mutability === EMutability.constant &&
+            left.mutability === EMutability.constant &&
+            right.name === left.name));
 }
 
 class ObjectValue extends VoidValue {
@@ -926,16 +976,18 @@ class LazyValue extends VoidValue {
 }
 
 class Scope {
-    constructor(values, parent = null, stacked = false, ntemp = 0, name = "", inst = []) {
+    constructor(values, parent = null, stacked = false, ntemp = 0, name = "", inst = [], operationCache = {}, cacheDependencies = {}) {
         this.parent = parent;
         this.stacked = stacked;
         this.ntemp = ntemp;
         this.name = name;
         this.inst = inst;
+        this.operationCache = operationCache;
+        this.cacheDependencies = cacheDependencies;
         this.data = values;
     }
     copy() {
-        const scope = new Scope({ ...this.data }, this.parent, this.stacked, this.ntemp, this.name, this.inst);
+        const scope = new Scope({ ...this.data }, this.parent, this.stacked, this.ntemp, this.name, this.inst, { ...this.operationCache }, { ...this.cacheDependencies });
         scope.break = this.break;
         scope.continue = this.continue;
         scope.function = this.function;
@@ -944,6 +996,8 @@ class Scope {
     createScope() {
         const scope = this.copy();
         scope.data = {};
+        scope.operationCache = {};
+        scope.cacheDependencies = {};
         scope.parent = this;
         return scope;
     }
@@ -989,6 +1043,46 @@ class Scope {
         this.ntemp++;
         return result;
     }
+    addCachedOperation(op, result, left, right) {
+        if (!left.name || !result.name || (right && !right.name))
+            return;
+        const id = formatCacheId(op, left, right);
+        this.operationCache[id] = result;
+        this.clearDependentCache(result);
+        addCacheDependency(this, left.name, id);
+        // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+        if (right)
+            addCacheDependency(this, right.name, id);
+        addCacheDependency(this, result.name, id);
+    }
+    getCachedOperation(op, left, right) {
+        const id = formatCacheId(op, left, right);
+        const result = this.operationCache[id];
+        if (result)
+            return result;
+        if (this.parent)
+            return this.parent.getCachedOperation(op, left, right);
+    }
+    clearDependentCache(value) {
+        if (!value.name)
+            return;
+        const dependents = this.cacheDependencies[value.name];
+        if (dependents) {
+            for (const id of dependents) {
+                delete this.operationCache[id];
+            }
+        }
+        if (this.parent)
+            this.parent.clearDependentCache(value);
+    }
+}
+function formatCacheId(op, left, right) {
+    if (right)
+        return `{${op}}{${left.name}}{${right.name}}`;
+    return `{${op}}{${left.name}}`;
+}
+function addCacheDependency(scope, name, id) {
+    (scope.cacheDependencies[name] ??= []).push(id);
 }
 
 const Program = (c, scope, node) => {
@@ -1164,7 +1258,21 @@ const BooleanLiteral = (_c, scope, node) => [new LiteralValue(+node.value), []];
 const LRExpression = (c, scope, node, out) => {
     const [left, leftInst] = c.handleEval(scope, node.left);
     const [right, rightInst] = c.handleEval(scope, node.right);
+    if (!(left instanceof LiteralValue) || !(right instanceof LiteralValue)) {
+        const cachedResult = scope.getCachedOperation(node.operator, left, right);
+        if (cachedResult)
+            return [cachedResult, [...leftInst, ...rightInst]];
+        // because a * b is the same as b * a
+        if (orderIndependentOperators.includes(node.operator)) {
+            const cachedResult = scope.getCachedOperation(node.operator, right, left);
+            if (cachedResult)
+                return [cachedResult, [...leftInst, ...rightInst]];
+        }
+    }
     const [op, opInst] = left[node.operator](scope, right, out);
+    if (!(op instanceof LiteralValue)) {
+        scope.addCachedOperation(node.operator, op, left, right);
+    }
     return [op, [...leftInst, ...rightInst, ...opInst]];
 };
 const BinaryExpression = LRExpression;
@@ -1186,18 +1294,26 @@ const AssignmentExpression = (c, scope, node) => {
         ];
     // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
     const [op, opInst] = left[node.operator](scope, right);
+    if (left)
+        scope.clearDependentCache(left);
     return [op, [...leftInst, ...rightInst, ...opInst]];
 };
 const UnaryExpression = (c, scope, { argument, operator }, out) => {
     const [arg, argInst] = c.handleEval(scope, argument);
+    const cachedResult = scope.getCachedOperation(operator, arg);
+    if (cachedResult)
+        return [cachedResult, argInst];
     const operatorId = operator == "+" || operator == "-" ? `u${operator}` : operator;
     if (operatorId === "throw")
         throw new CompilerError("throw operator is not supported");
     const [op, opInst] = arg[operatorId](scope, out);
+    scope.addCachedOperation(operator, op, arg);
     return [op, [...argInst, ...opInst]];
 };
 const UpdateExpression = (c, scope, { argument, operator, prefix }, out) => {
     const [arg, argInst] = c.handle(scope, argument);
+    if (arg)
+        scope.clearDependentCache(arg);
     // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
     const [op, opInst] = arg[operator](scope, prefix, out);
     return [op, [...argInst, ...opInst]];
@@ -1228,6 +1344,17 @@ const ConditionalExpression = (c, scope, node, out) => {
             new AddressResolver(endExpressionAdress),
         ],
     ];
+};
+const SequenceExpression = (c, scope, node, out) => {
+    const { expressions } = node;
+    const inst = [];
+    // compute every expression except the last one
+    for (let i = 0; i < expressions.length - 1; i++) {
+        const [, exprInst] = c.handleEval(scope, expressions[i], discardedName);
+        inst.push(...exprInst);
+    }
+    const [value, valueInst] = c.handleEval(scope, expressions[expressions.length - 1], out);
+    return [value, [...inst, ...valueInst]];
 };
 
 const Identifier = (c, scope, node) => {
@@ -1259,6 +1386,7 @@ const ReturnStatement = (c, scope, node, out) => {
     const [ret, retInst] = scope.function.return(scope, arg, out);
     return [ret, [...argInst, ...retInst]];
 };
+const EmptyStatement = () => [null, []];
 const LabeledStatement = (c, scope, node) => {
     const inner = scope.createScope();
     inner.label = node.label.name;
@@ -1288,9 +1416,12 @@ const IfStatement = (c, scope, node) => {
         return [null, inst];
     }
     const endIfAddr = new LiteralValue(null);
-    inst.push(...testInst, new JumpInstruction(endIfAddr, EJumpKind.Equal, test, new LiteralValue(0)), ...withAlwaysRuns(c.handle(scope, node.consequent), false)[1], new AddressResolver(endIfAddr));
-    const endElseAddr = new LiteralValue(null);
-    if (node.alternate) {
+    inst.push(...testInst, new JumpInstruction(endIfAddr, EJumpKind.Equal, test, new LiteralValue(0)), ...withAlwaysRuns(c.handle(scope, node.consequent), false)[1]);
+    if (!node.alternate) {
+        inst.push(new AddressResolver(endIfAddr));
+    }
+    else {
+        const endElseAddr = new LiteralValue(null);
         inst.push(new JumpInstruction(endElseAddr, EJumpKind.Always), new AddressResolver(endIfAddr), ...withAlwaysRuns(c.handle(scope, node.alternate), false)[1], new AddressResolver(endElseAddr));
     }
     return [null, inst];
@@ -1308,7 +1439,7 @@ const SwitchStatement = (c, scope, node) => {
     let constantCase = false;
     let onlyConstantTests = true;
     for (const scase of node.cases) {
-        const [, bodyInst] = c.handleMany(innerScope, scase.consequent);
+        const bodyInst = lazy(() => c.handleMany(innerScope, scase.consequent)[1]);
         const bodyAdress = new LiteralValue(null);
         const bodyLine = new AddressResolver(bodyAdress);
         const canFallInto = !endsWithoutFalltrough(inst, endLine);
@@ -1322,7 +1453,7 @@ const SwitchStatement = (c, scope, node) => {
                 ])[1];
             }
             if (includeBody)
-                inst.push(bodyLine, ...bodyInst);
+                inst.push(bodyLine, ...bodyInst());
             continue;
         }
         const [test, testInst] = c.handleEval(innerScope, scase.test);
@@ -1346,8 +1477,8 @@ const SwitchStatement = (c, scope, node) => {
                 // and all the other cases will never run
                 if (noFallInto &&
                     onlyConstantTests &&
-                    endsWithoutFalltrough(bodyInst, endLine)) {
-                    return [null, [...bodyInst, endLine]];
+                    endsWithoutFalltrough(bodyInst(), endLine)) {
+                    return [null, [...bodyInst(), endLine]];
                 }
                 jump = c.handle(innerScope, scase, () => [
                     null,
@@ -1364,7 +1495,7 @@ const SwitchStatement = (c, scope, node) => {
         if (includeJump)
             caseJumps.push(...testInst, ...jump);
         if (includeBody)
-            inst.push(bodyLine, ...bodyInst);
+            inst.push(bodyLine, ...bodyInst());
     }
     // ensures that the processor exits
     // the switch if no cases match
@@ -1842,7 +1973,9 @@ const ObjectPattern = (c, scope, node) => {
         });
     }
     return [new DestructuringValue(members), inst];
-};
+};
+// the mlog runtime already does this check for us
+const OptionalMemberExpression = MemberExpression;
 
 const TemplateLiteral = () => {
     throw new CompilerError("Template strings are currently unavailable. To inline mlog code use the asm function.");
@@ -1919,11 +2052,13 @@ var handlers = /*#__PURE__*/Object.freeze({
     UnaryExpression: UnaryExpression,
     UpdateExpression: UpdateExpression,
     ConditionalExpression: ConditionalExpression,
+    SequenceExpression: SequenceExpression,
     Identifier: Identifier,
     ExpressionStatement: ExpressionStatement,
     BreakStatement: BreakStatement,
     ContinueStatement: ContinueStatement,
     ReturnStatement: ReturnStatement,
+    EmptyStatement: EmptyStatement,
     LabeledStatement: LabeledStatement,
     IfStatement: IfStatement,
     SwitchStatement: SwitchStatement,
@@ -1942,6 +2077,7 @@ var handlers = /*#__PURE__*/Object.freeze({
     MemberExpression: MemberExpression,
     ArrayPattern: ArrayPattern,
     ObjectPattern: ObjectPattern,
+    OptionalMemberExpression: OptionalMemberExpression,
     TemplateLiteral: TemplateLiteral,
     BlockStatement: BlockStatement,
     File: File,
@@ -1989,6 +2125,7 @@ const mathOperations = {
     sqrt: a => Math.sqrt(a),
     rand: null,
 };
+const orderIndependentOperations = ["max", "min", "len"];
 function createMacroMathOperations() {
     const macroMathOperations = {
         PI: new StoreValue("@pi", EMutability.constant),
@@ -2005,7 +2142,17 @@ function createMacroMathOperations() {
                         throw new CompilerError("Cannot do math operation with non-numerical literals.");
                     return [new LiteralValue(fn(a.num, b.num)), []];
                 }
+                const cachedResult = scope.getCachedOperation(key, a, b);
+                if (cachedResult)
+                    return [cachedResult, []];
+                // max, min and len do not change if the arguments change in order
+                if (orderIndependentOperations.includes(key)) {
+                    const cachedResult = scope.getCachedOperation(key, b, a);
+                    if (cachedResult)
+                        return [cachedResult, []];
+                }
                 const temp = StoreValue.from(scope, out);
+                scope.addCachedOperation(key, temp, a, b);
                 return [temp, [new OperationInstruction(key, temp, a, b)]];
             }
             if (fn && a instanceof LiteralValue) {
@@ -2013,8 +2160,14 @@ function createMacroMathOperations() {
                     throw new CompilerError("Cannot do math operation with non-numerical literal.");
                 return [new LiteralValue(fn(a.num)), []];
             }
+            const cachedResult = scope.getCachedOperation(key, a);
+            if (cachedResult)
+                return [cachedResult, []];
             const temp = StoreValue.from(scope, out);
-            return [temp, [new OperationInstruction(key, temp, a, b)]];
+            // ensures that operations like `rand` are not cached
+            if (mathOperations[key])
+                scope.addCachedOperation(key, temp, a);
+            return [temp, [new OperationInstruction(key, temp, a, null)]];
         });
     }
     return macroMathOperations;