You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
697 lines
23 KiB
697 lines
23 KiB
"use strict";
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.or = exports.and = exports.not = exports.CodeGen = exports.operators = exports.varKinds = exports.ValueScopeName = exports.ValueScope = exports.Scope = exports.Name = exports.regexpCode = exports.stringify = exports.getProperty = exports.nil = exports.strConcat = exports.str = exports._ = void 0;
|
|
const code_1 = require("./code");
|
|
const scope_1 = require("./scope");
|
|
var code_2 = require("./code");
|
|
Object.defineProperty(exports, "_", { enumerable: true, get: function () { return code_2._; } });
|
|
Object.defineProperty(exports, "str", { enumerable: true, get: function () { return code_2.str; } });
|
|
Object.defineProperty(exports, "strConcat", { enumerable: true, get: function () { return code_2.strConcat; } });
|
|
Object.defineProperty(exports, "nil", { enumerable: true, get: function () { return code_2.nil; } });
|
|
Object.defineProperty(exports, "getProperty", { enumerable: true, get: function () { return code_2.getProperty; } });
|
|
Object.defineProperty(exports, "stringify", { enumerable: true, get: function () { return code_2.stringify; } });
|
|
Object.defineProperty(exports, "regexpCode", { enumerable: true, get: function () { return code_2.regexpCode; } });
|
|
Object.defineProperty(exports, "Name", { enumerable: true, get: function () { return code_2.Name; } });
|
|
var scope_2 = require("./scope");
|
|
Object.defineProperty(exports, "Scope", { enumerable: true, get: function () { return scope_2.Scope; } });
|
|
Object.defineProperty(exports, "ValueScope", { enumerable: true, get: function () { return scope_2.ValueScope; } });
|
|
Object.defineProperty(exports, "ValueScopeName", { enumerable: true, get: function () { return scope_2.ValueScopeName; } });
|
|
Object.defineProperty(exports, "varKinds", { enumerable: true, get: function () { return scope_2.varKinds; } });
|
|
exports.operators = {
|
|
GT: new code_1._Code(">"),
|
|
GTE: new code_1._Code(">="),
|
|
LT: new code_1._Code("<"),
|
|
LTE: new code_1._Code("<="),
|
|
EQ: new code_1._Code("==="),
|
|
NEQ: new code_1._Code("!=="),
|
|
NOT: new code_1._Code("!"),
|
|
OR: new code_1._Code("||"),
|
|
AND: new code_1._Code("&&"),
|
|
ADD: new code_1._Code("+"),
|
|
};
|
|
class Node {
|
|
optimizeNodes() {
|
|
return this;
|
|
}
|
|
optimizeNames(_names, _constants) {
|
|
return this;
|
|
}
|
|
}
|
|
class Def extends Node {
|
|
constructor(varKind, name, rhs) {
|
|
super();
|
|
this.varKind = varKind;
|
|
this.name = name;
|
|
this.rhs = rhs;
|
|
}
|
|
render({ es5, _n }) {
|
|
const varKind = es5 ? scope_1.varKinds.var : this.varKind;
|
|
const rhs = this.rhs === undefined ? "" : ` = ${this.rhs}`;
|
|
return `${varKind} ${this.name}${rhs};` + _n;
|
|
}
|
|
optimizeNames(names, constants) {
|
|
if (!names[this.name.str])
|
|
return;
|
|
if (this.rhs)
|
|
this.rhs = optimizeExpr(this.rhs, names, constants);
|
|
return this;
|
|
}
|
|
get names() {
|
|
return this.rhs instanceof code_1._CodeOrName ? this.rhs.names : {};
|
|
}
|
|
}
|
|
class Assign extends Node {
|
|
constructor(lhs, rhs, sideEffects) {
|
|
super();
|
|
this.lhs = lhs;
|
|
this.rhs = rhs;
|
|
this.sideEffects = sideEffects;
|
|
}
|
|
render({ _n }) {
|
|
return `${this.lhs} = ${this.rhs};` + _n;
|
|
}
|
|
optimizeNames(names, constants) {
|
|
if (this.lhs instanceof code_1.Name && !names[this.lhs.str] && !this.sideEffects)
|
|
return;
|
|
this.rhs = optimizeExpr(this.rhs, names, constants);
|
|
return this;
|
|
}
|
|
get names() {
|
|
const names = this.lhs instanceof code_1.Name ? {} : { ...this.lhs.names };
|
|
return addExprNames(names, this.rhs);
|
|
}
|
|
}
|
|
class AssignOp extends Assign {
|
|
constructor(lhs, op, rhs, sideEffects) {
|
|
super(lhs, rhs, sideEffects);
|
|
this.op = op;
|
|
}
|
|
render({ _n }) {
|
|
return `${this.lhs} ${this.op}= ${this.rhs};` + _n;
|
|
}
|
|
}
|
|
class Label extends Node {
|
|
constructor(label) {
|
|
super();
|
|
this.label = label;
|
|
this.names = {};
|
|
}
|
|
render({ _n }) {
|
|
return `${this.label}:` + _n;
|
|
}
|
|
}
|
|
class Break extends Node {
|
|
constructor(label) {
|
|
super();
|
|
this.label = label;
|
|
this.names = {};
|
|
}
|
|
render({ _n }) {
|
|
const label = this.label ? ` ${this.label}` : "";
|
|
return `break${label};` + _n;
|
|
}
|
|
}
|
|
class Throw extends Node {
|
|
constructor(error) {
|
|
super();
|
|
this.error = error;
|
|
}
|
|
render({ _n }) {
|
|
return `throw ${this.error};` + _n;
|
|
}
|
|
get names() {
|
|
return this.error.names;
|
|
}
|
|
}
|
|
class AnyCode extends Node {
|
|
constructor(code) {
|
|
super();
|
|
this.code = code;
|
|
}
|
|
render({ _n }) {
|
|
return `${this.code};` + _n;
|
|
}
|
|
optimizeNodes() {
|
|
return `${this.code}` ? this : undefined;
|
|
}
|
|
optimizeNames(names, constants) {
|
|
this.code = optimizeExpr(this.code, names, constants);
|
|
return this;
|
|
}
|
|
get names() {
|
|
return this.code instanceof code_1._CodeOrName ? this.code.names : {};
|
|
}
|
|
}
|
|
class ParentNode extends Node {
|
|
constructor(nodes = []) {
|
|
super();
|
|
this.nodes = nodes;
|
|
}
|
|
render(opts) {
|
|
return this.nodes.reduce((code, n) => code + n.render(opts), "");
|
|
}
|
|
optimizeNodes() {
|
|
const { nodes } = this;
|
|
let i = nodes.length;
|
|
while (i--) {
|
|
const n = nodes[i].optimizeNodes();
|
|
if (Array.isArray(n))
|
|
nodes.splice(i, 1, ...n);
|
|
else if (n)
|
|
nodes[i] = n;
|
|
else
|
|
nodes.splice(i, 1);
|
|
}
|
|
return nodes.length > 0 ? this : undefined;
|
|
}
|
|
optimizeNames(names, constants) {
|
|
const { nodes } = this;
|
|
let i = nodes.length;
|
|
while (i--) {
|
|
// iterating backwards improves 1-pass optimization
|
|
const n = nodes[i];
|
|
if (n.optimizeNames(names, constants))
|
|
continue;
|
|
subtractNames(names, n.names);
|
|
nodes.splice(i, 1);
|
|
}
|
|
return nodes.length > 0 ? this : undefined;
|
|
}
|
|
get names() {
|
|
return this.nodes.reduce((names, n) => addNames(names, n.names), {});
|
|
}
|
|
}
|
|
class BlockNode extends ParentNode {
|
|
render(opts) {
|
|
return "{" + opts._n + super.render(opts) + "}" + opts._n;
|
|
}
|
|
}
|
|
class Root extends ParentNode {
|
|
}
|
|
class Else extends BlockNode {
|
|
}
|
|
Else.kind = "else";
|
|
class If extends BlockNode {
|
|
constructor(condition, nodes) {
|
|
super(nodes);
|
|
this.condition = condition;
|
|
}
|
|
render(opts) {
|
|
let code = `if(${this.condition})` + super.render(opts);
|
|
if (this.else)
|
|
code += "else " + this.else.render(opts);
|
|
return code;
|
|
}
|
|
optimizeNodes() {
|
|
super.optimizeNodes();
|
|
const cond = this.condition;
|
|
if (cond === true)
|
|
return this.nodes; // else is ignored here
|
|
let e = this.else;
|
|
if (e) {
|
|
const ns = e.optimizeNodes();
|
|
e = this.else = Array.isArray(ns) ? new Else(ns) : ns;
|
|
}
|
|
if (e) {
|
|
if (cond === false)
|
|
return e instanceof If ? e : e.nodes;
|
|
if (this.nodes.length)
|
|
return this;
|
|
return new If(not(cond), e instanceof If ? [e] : e.nodes);
|
|
}
|
|
if (cond === false || !this.nodes.length)
|
|
return undefined;
|
|
return this;
|
|
}
|
|
optimizeNames(names, constants) {
|
|
var _a;
|
|
this.else = (_a = this.else) === null || _a === void 0 ? void 0 : _a.optimizeNames(names, constants);
|
|
if (!(super.optimizeNames(names, constants) || this.else))
|
|
return;
|
|
this.condition = optimizeExpr(this.condition, names, constants);
|
|
return this;
|
|
}
|
|
get names() {
|
|
const names = super.names;
|
|
addExprNames(names, this.condition);
|
|
if (this.else)
|
|
addNames(names, this.else.names);
|
|
return names;
|
|
}
|
|
}
|
|
If.kind = "if";
|
|
class For extends BlockNode {
|
|
}
|
|
For.kind = "for";
|
|
class ForLoop extends For {
|
|
constructor(iteration) {
|
|
super();
|
|
this.iteration = iteration;
|
|
}
|
|
render(opts) {
|
|
return `for(${this.iteration})` + super.render(opts);
|
|
}
|
|
optimizeNames(names, constants) {
|
|
if (!super.optimizeNames(names, constants))
|
|
return;
|
|
this.iteration = optimizeExpr(this.iteration, names, constants);
|
|
return this;
|
|
}
|
|
get names() {
|
|
return addNames(super.names, this.iteration.names);
|
|
}
|
|
}
|
|
class ForRange extends For {
|
|
constructor(varKind, name, from, to) {
|
|
super();
|
|
this.varKind = varKind;
|
|
this.name = name;
|
|
this.from = from;
|
|
this.to = to;
|
|
}
|
|
render(opts) {
|
|
const varKind = opts.es5 ? scope_1.varKinds.var : this.varKind;
|
|
const { name, from, to } = this;
|
|
return `for(${varKind} ${name}=${from}; ${name}<${to}; ${name}++)` + super.render(opts);
|
|
}
|
|
get names() {
|
|
const names = addExprNames(super.names, this.from);
|
|
return addExprNames(names, this.to);
|
|
}
|
|
}
|
|
class ForIter extends For {
|
|
constructor(loop, varKind, name, iterable) {
|
|
super();
|
|
this.loop = loop;
|
|
this.varKind = varKind;
|
|
this.name = name;
|
|
this.iterable = iterable;
|
|
}
|
|
render(opts) {
|
|
return `for(${this.varKind} ${this.name} ${this.loop} ${this.iterable})` + super.render(opts);
|
|
}
|
|
optimizeNames(names, constants) {
|
|
if (!super.optimizeNames(names, constants))
|
|
return;
|
|
this.iterable = optimizeExpr(this.iterable, names, constants);
|
|
return this;
|
|
}
|
|
get names() {
|
|
return addNames(super.names, this.iterable.names);
|
|
}
|
|
}
|
|
class Func extends BlockNode {
|
|
constructor(name, args, async) {
|
|
super();
|
|
this.name = name;
|
|
this.args = args;
|
|
this.async = async;
|
|
}
|
|
render(opts) {
|
|
const _async = this.async ? "async " : "";
|
|
return `${_async}function ${this.name}(${this.args})` + super.render(opts);
|
|
}
|
|
}
|
|
Func.kind = "func";
|
|
class Return extends ParentNode {
|
|
render(opts) {
|
|
return "return " + super.render(opts);
|
|
}
|
|
}
|
|
Return.kind = "return";
|
|
class Try extends BlockNode {
|
|
render(opts) {
|
|
let code = "try" + super.render(opts);
|
|
if (this.catch)
|
|
code += this.catch.render(opts);
|
|
if (this.finally)
|
|
code += this.finally.render(opts);
|
|
return code;
|
|
}
|
|
optimizeNodes() {
|
|
var _a, _b;
|
|
super.optimizeNodes();
|
|
(_a = this.catch) === null || _a === void 0 ? void 0 : _a.optimizeNodes();
|
|
(_b = this.finally) === null || _b === void 0 ? void 0 : _b.optimizeNodes();
|
|
return this;
|
|
}
|
|
optimizeNames(names, constants) {
|
|
var _a, _b;
|
|
super.optimizeNames(names, constants);
|
|
(_a = this.catch) === null || _a === void 0 ? void 0 : _a.optimizeNames(names, constants);
|
|
(_b = this.finally) === null || _b === void 0 ? void 0 : _b.optimizeNames(names, constants);
|
|
return this;
|
|
}
|
|
get names() {
|
|
const names = super.names;
|
|
if (this.catch)
|
|
addNames(names, this.catch.names);
|
|
if (this.finally)
|
|
addNames(names, this.finally.names);
|
|
return names;
|
|
}
|
|
}
|
|
class Catch extends BlockNode {
|
|
constructor(error) {
|
|
super();
|
|
this.error = error;
|
|
}
|
|
render(opts) {
|
|
return `catch(${this.error})` + super.render(opts);
|
|
}
|
|
}
|
|
Catch.kind = "catch";
|
|
class Finally extends BlockNode {
|
|
render(opts) {
|
|
return "finally" + super.render(opts);
|
|
}
|
|
}
|
|
Finally.kind = "finally";
|
|
class CodeGen {
|
|
constructor(extScope, opts = {}) {
|
|
this._values = {};
|
|
this._blockStarts = [];
|
|
this._constants = {};
|
|
this.opts = { ...opts, _n: opts.lines ? "\n" : "" };
|
|
this._extScope = extScope;
|
|
this._scope = new scope_1.Scope({ parent: extScope });
|
|
this._nodes = [new Root()];
|
|
}
|
|
toString() {
|
|
return this._root.render(this.opts);
|
|
}
|
|
// returns unique name in the internal scope
|
|
name(prefix) {
|
|
return this._scope.name(prefix);
|
|
}
|
|
// reserves unique name in the external scope
|
|
scopeName(prefix) {
|
|
return this._extScope.name(prefix);
|
|
}
|
|
// reserves unique name in the external scope and assigns value to it
|
|
scopeValue(prefixOrName, value) {
|
|
const name = this._extScope.value(prefixOrName, value);
|
|
const vs = this._values[name.prefix] || (this._values[name.prefix] = new Set());
|
|
vs.add(name);
|
|
return name;
|
|
}
|
|
getScopeValue(prefix, keyOrRef) {
|
|
return this._extScope.getValue(prefix, keyOrRef);
|
|
}
|
|
// return code that assigns values in the external scope to the names that are used internally
|
|
// (same names that were returned by gen.scopeName or gen.scopeValue)
|
|
scopeRefs(scopeName) {
|
|
return this._extScope.scopeRefs(scopeName, this._values);
|
|
}
|
|
scopeCode() {
|
|
return this._extScope.scopeCode(this._values);
|
|
}
|
|
_def(varKind, nameOrPrefix, rhs, constant) {
|
|
const name = this._scope.toName(nameOrPrefix);
|
|
if (rhs !== undefined && constant)
|
|
this._constants[name.str] = rhs;
|
|
this._leafNode(new Def(varKind, name, rhs));
|
|
return name;
|
|
}
|
|
// `const` declaration (`var` in es5 mode)
|
|
const(nameOrPrefix, rhs, _constant) {
|
|
return this._def(scope_1.varKinds.const, nameOrPrefix, rhs, _constant);
|
|
}
|
|
// `let` declaration with optional assignment (`var` in es5 mode)
|
|
let(nameOrPrefix, rhs, _constant) {
|
|
return this._def(scope_1.varKinds.let, nameOrPrefix, rhs, _constant);
|
|
}
|
|
// `var` declaration with optional assignment
|
|
var(nameOrPrefix, rhs, _constant) {
|
|
return this._def(scope_1.varKinds.var, nameOrPrefix, rhs, _constant);
|
|
}
|
|
// assignment code
|
|
assign(lhs, rhs, sideEffects) {
|
|
return this._leafNode(new Assign(lhs, rhs, sideEffects));
|
|
}
|
|
// `+=` code
|
|
add(lhs, rhs) {
|
|
return this._leafNode(new AssignOp(lhs, exports.operators.ADD, rhs));
|
|
}
|
|
// appends passed SafeExpr to code or executes Block
|
|
code(c) {
|
|
if (typeof c == "function")
|
|
c();
|
|
else if (c !== code_1.nil)
|
|
this._leafNode(new AnyCode(c));
|
|
return this;
|
|
}
|
|
// returns code for object literal for the passed argument list of key-value pairs
|
|
object(...keyValues) {
|
|
const code = ["{"];
|
|
for (const [key, value] of keyValues) {
|
|
if (code.length > 1)
|
|
code.push(",");
|
|
code.push(key);
|
|
if (key !== value || this.opts.es5) {
|
|
code.push(":");
|
|
(0, code_1.addCodeArg)(code, value);
|
|
}
|
|
}
|
|
code.push("}");
|
|
return new code_1._Code(code);
|
|
}
|
|
// `if` clause (or statement if `thenBody` and, optionally, `elseBody` are passed)
|
|
if(condition, thenBody, elseBody) {
|
|
this._blockNode(new If(condition));
|
|
if (thenBody && elseBody) {
|
|
this.code(thenBody).else().code(elseBody).endIf();
|
|
}
|
|
else if (thenBody) {
|
|
this.code(thenBody).endIf();
|
|
}
|
|
else if (elseBody) {
|
|
throw new Error('CodeGen: "else" body without "then" body');
|
|
}
|
|
return this;
|
|
}
|
|
// `else if` clause - invalid without `if` or after `else` clauses
|
|
elseIf(condition) {
|
|
return this._elseNode(new If(condition));
|
|
}
|
|
// `else` clause - only valid after `if` or `else if` clauses
|
|
else() {
|
|
return this._elseNode(new Else());
|
|
}
|
|
// end `if` statement (needed if gen.if was used only with condition)
|
|
endIf() {
|
|
return this._endBlockNode(If, Else);
|
|
}
|
|
_for(node, forBody) {
|
|
this._blockNode(node);
|
|
if (forBody)
|
|
this.code(forBody).endFor();
|
|
return this;
|
|
}
|
|
// a generic `for` clause (or statement if `forBody` is passed)
|
|
for(iteration, forBody) {
|
|
return this._for(new ForLoop(iteration), forBody);
|
|
}
|
|
// `for` statement for a range of values
|
|
forRange(nameOrPrefix, from, to, forBody, varKind = this.opts.es5 ? scope_1.varKinds.var : scope_1.varKinds.let) {
|
|
const name = this._scope.toName(nameOrPrefix);
|
|
return this._for(new ForRange(varKind, name, from, to), () => forBody(name));
|
|
}
|
|
// `for-of` statement (in es5 mode replace with a normal for loop)
|
|
forOf(nameOrPrefix, iterable, forBody, varKind = scope_1.varKinds.const) {
|
|
const name = this._scope.toName(nameOrPrefix);
|
|
if (this.opts.es5) {
|
|
const arr = iterable instanceof code_1.Name ? iterable : this.var("_arr", iterable);
|
|
return this.forRange("_i", 0, (0, code_1._) `${arr}.length`, (i) => {
|
|
this.var(name, (0, code_1._) `${arr}[${i}]`);
|
|
forBody(name);
|
|
});
|
|
}
|
|
return this._for(new ForIter("of", varKind, name, iterable), () => forBody(name));
|
|
}
|
|
// `for-in` statement.
|
|
// With option `ownProperties` replaced with a `for-of` loop for object keys
|
|
forIn(nameOrPrefix, obj, forBody, varKind = this.opts.es5 ? scope_1.varKinds.var : scope_1.varKinds.const) {
|
|
if (this.opts.ownProperties) {
|
|
return this.forOf(nameOrPrefix, (0, code_1._) `Object.keys(${obj})`, forBody);
|
|
}
|
|
const name = this._scope.toName(nameOrPrefix);
|
|
return this._for(new ForIter("in", varKind, name, obj), () => forBody(name));
|
|
}
|
|
// end `for` loop
|
|
endFor() {
|
|
return this._endBlockNode(For);
|
|
}
|
|
// `label` statement
|
|
label(label) {
|
|
return this._leafNode(new Label(label));
|
|
}
|
|
// `break` statement
|
|
break(label) {
|
|
return this._leafNode(new Break(label));
|
|
}
|
|
// `return` statement
|
|
return(value) {
|
|
const node = new Return();
|
|
this._blockNode(node);
|
|
this.code(value);
|
|
if (node.nodes.length !== 1)
|
|
throw new Error('CodeGen: "return" should have one node');
|
|
return this._endBlockNode(Return);
|
|
}
|
|
// `try` statement
|
|
try(tryBody, catchCode, finallyCode) {
|
|
if (!catchCode && !finallyCode)
|
|
throw new Error('CodeGen: "try" without "catch" and "finally"');
|
|
const node = new Try();
|
|
this._blockNode(node);
|
|
this.code(tryBody);
|
|
if (catchCode) {
|
|
const error = this.name("e");
|
|
this._currNode = node.catch = new Catch(error);
|
|
catchCode(error);
|
|
}
|
|
if (finallyCode) {
|
|
this._currNode = node.finally = new Finally();
|
|
this.code(finallyCode);
|
|
}
|
|
return this._endBlockNode(Catch, Finally);
|
|
}
|
|
// `throw` statement
|
|
throw(error) {
|
|
return this._leafNode(new Throw(error));
|
|
}
|
|
// start self-balancing block
|
|
block(body, nodeCount) {
|
|
this._blockStarts.push(this._nodes.length);
|
|
if (body)
|
|
this.code(body).endBlock(nodeCount);
|
|
return this;
|
|
}
|
|
// end the current self-balancing block
|
|
endBlock(nodeCount) {
|
|
const len = this._blockStarts.pop();
|
|
if (len === undefined)
|
|
throw new Error("CodeGen: not in self-balancing block");
|
|
const toClose = this._nodes.length - len;
|
|
if (toClose < 0 || (nodeCount !== undefined && toClose !== nodeCount)) {
|
|
throw new Error(`CodeGen: wrong number of nodes: ${toClose} vs ${nodeCount} expected`);
|
|
}
|
|
this._nodes.length = len;
|
|
return this;
|
|
}
|
|
// `function` heading (or definition if funcBody is passed)
|
|
func(name, args = code_1.nil, async, funcBody) {
|
|
this._blockNode(new Func(name, args, async));
|
|
if (funcBody)
|
|
this.code(funcBody).endFunc();
|
|
return this;
|
|
}
|
|
// end function definition
|
|
endFunc() {
|
|
return this._endBlockNode(Func);
|
|
}
|
|
optimize(n = 1) {
|
|
while (n-- > 0) {
|
|
this._root.optimizeNodes();
|
|
this._root.optimizeNames(this._root.names, this._constants);
|
|
}
|
|
}
|
|
_leafNode(node) {
|
|
this._currNode.nodes.push(node);
|
|
return this;
|
|
}
|
|
_blockNode(node) {
|
|
this._currNode.nodes.push(node);
|
|
this._nodes.push(node);
|
|
}
|
|
_endBlockNode(N1, N2) {
|
|
const n = this._currNode;
|
|
if (n instanceof N1 || (N2 && n instanceof N2)) {
|
|
this._nodes.pop();
|
|
return this;
|
|
}
|
|
throw new Error(`CodeGen: not in block "${N2 ? `${N1.kind}/${N2.kind}` : N1.kind}"`);
|
|
}
|
|
_elseNode(node) {
|
|
const n = this._currNode;
|
|
if (!(n instanceof If)) {
|
|
throw new Error('CodeGen: "else" without "if"');
|
|
}
|
|
this._currNode = n.else = node;
|
|
return this;
|
|
}
|
|
get _root() {
|
|
return this._nodes[0];
|
|
}
|
|
get _currNode() {
|
|
const ns = this._nodes;
|
|
return ns[ns.length - 1];
|
|
}
|
|
set _currNode(node) {
|
|
const ns = this._nodes;
|
|
ns[ns.length - 1] = node;
|
|
}
|
|
}
|
|
exports.CodeGen = CodeGen;
|
|
function addNames(names, from) {
|
|
for (const n in from)
|
|
names[n] = (names[n] || 0) + (from[n] || 0);
|
|
return names;
|
|
}
|
|
function addExprNames(names, from) {
|
|
return from instanceof code_1._CodeOrName ? addNames(names, from.names) : names;
|
|
}
|
|
function optimizeExpr(expr, names, constants) {
|
|
if (expr instanceof code_1.Name)
|
|
return replaceName(expr);
|
|
if (!canOptimize(expr))
|
|
return expr;
|
|
return new code_1._Code(expr._items.reduce((items, c) => {
|
|
if (c instanceof code_1.Name)
|
|
c = replaceName(c);
|
|
if (c instanceof code_1._Code)
|
|
items.push(...c._items);
|
|
else
|
|
items.push(c);
|
|
return items;
|
|
}, []));
|
|
function replaceName(n) {
|
|
const c = constants[n.str];
|
|
if (c === undefined || names[n.str] !== 1)
|
|
return n;
|
|
delete names[n.str];
|
|
return c;
|
|
}
|
|
function canOptimize(e) {
|
|
return (e instanceof code_1._Code &&
|
|
e._items.some((c) => c instanceof code_1.Name && names[c.str] === 1 && constants[c.str] !== undefined));
|
|
}
|
|
}
|
|
function subtractNames(names, from) {
|
|
for (const n in from)
|
|
names[n] = (names[n] || 0) - (from[n] || 0);
|
|
}
|
|
function not(x) {
|
|
return typeof x == "boolean" || typeof x == "number" || x === null ? !x : (0, code_1._) `!${par(x)}`;
|
|
}
|
|
exports.not = not;
|
|
const andCode = mappend(exports.operators.AND);
|
|
// boolean AND (&&) expression with the passed arguments
|
|
function and(...args) {
|
|
return args.reduce(andCode);
|
|
}
|
|
exports.and = and;
|
|
const orCode = mappend(exports.operators.OR);
|
|
// boolean OR (||) expression with the passed arguments
|
|
function or(...args) {
|
|
return args.reduce(orCode);
|
|
}
|
|
exports.or = or;
|
|
function mappend(op) {
|
|
return (x, y) => (x === code_1.nil ? y : y === code_1.nil ? x : (0, code_1._) `${par(x)} ${op} ${par(y)}`);
|
|
}
|
|
function par(x) {
|
|
return x instanceof code_1.Name ? x : (0, code_1._) `(${x})`;
|
|
}
|
|
//# sourceMappingURL=index.js.map
|