212 lines
7.4 KiB
JavaScript
212 lines
7.4 KiB
JavaScript
import { getPrecedence } from '../operators.js';
|
|
import { escape } from '../../utils/string.js';
|
|
import { getSafeProperty } from '../../utils/customs.js';
|
|
import { latexOperators } from '../../utils/latex.js';
|
|
import { factory } from '../../utils/factory.js';
|
|
var name = 'RelationalNode';
|
|
var dependencies = ['Node'];
|
|
export var createRelationalNode = /* #__PURE__ */factory(name, dependencies, _ref => {
|
|
var {
|
|
Node
|
|
} = _ref;
|
|
|
|
/**
|
|
* A node representing a chained conditional expression, such as 'x > y > z'
|
|
*
|
|
* @param {String[]} conditionals An array of conditional operators used to compare the parameters
|
|
* @param {Node[]} params The parameters that will be compared
|
|
*
|
|
* @constructor RelationalNode
|
|
* @extends {Node}
|
|
*/
|
|
function RelationalNode(conditionals, params) {
|
|
if (!(this instanceof RelationalNode)) {
|
|
throw new SyntaxError('Constructor must be called with the new operator');
|
|
}
|
|
|
|
if (!Array.isArray(conditionals)) throw new TypeError('Parameter conditionals must be an array');
|
|
if (!Array.isArray(params)) throw new TypeError('Parameter params must be an array');
|
|
if (conditionals.length !== params.length - 1) throw new TypeError('Parameter params must contain exactly one more element than parameter conditionals');
|
|
this.conditionals = conditionals;
|
|
this.params = params;
|
|
}
|
|
|
|
RelationalNode.prototype = new Node();
|
|
RelationalNode.prototype.type = 'RelationalNode';
|
|
RelationalNode.prototype.isRelationalNode = true;
|
|
/**
|
|
* Compile a node into a JavaScript function.
|
|
* This basically pre-calculates as much as possible and only leaves open
|
|
* calculations which depend on a dynamic scope with variables.
|
|
* @param {Object} math Math.js namespace with functions and constants.
|
|
* @param {Object} argNames An object with argument names as key and `true`
|
|
* as value. Used in the SymbolNode to optimize
|
|
* for arguments from user assigned functions
|
|
* (see FunctionAssignmentNode) or special symbols
|
|
* like `end` (see IndexNode).
|
|
* @return {function} Returns a function which can be called like:
|
|
* evalNode(scope: Object, args: Object, context: *)
|
|
*/
|
|
|
|
RelationalNode.prototype._compile = function (math, argNames) {
|
|
var self = this;
|
|
var compiled = this.params.map(p => p._compile(math, argNames));
|
|
return function evalRelationalNode(scope, args, context) {
|
|
var evalLhs;
|
|
var evalRhs = compiled[0](scope, args, context);
|
|
|
|
for (var i = 0; i < self.conditionals.length; i++) {
|
|
evalLhs = evalRhs;
|
|
evalRhs = compiled[i + 1](scope, args, context);
|
|
var condFn = getSafeProperty(math, self.conditionals[i]);
|
|
|
|
if (!condFn(evalLhs, evalRhs)) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
};
|
|
};
|
|
/**
|
|
* Execute a callback for each of the child nodes of this node
|
|
* @param {function(child: Node, path: string, parent: Node)} callback
|
|
*/
|
|
|
|
|
|
RelationalNode.prototype.forEach = function (callback) {
|
|
this.params.forEach((n, i) => callback(n, 'params[' + i + ']', this), this);
|
|
};
|
|
/**
|
|
* Create a new RelationalNode having its childs be the results of calling
|
|
* the provided callback function for each of the childs of the original node.
|
|
* @param {function(child: Node, path: string, parent: Node): Node} callback
|
|
* @returns {RelationalNode} Returns a transformed copy of the node
|
|
*/
|
|
|
|
|
|
RelationalNode.prototype.map = function (callback) {
|
|
return new RelationalNode(this.conditionals.slice(), this.params.map((n, i) => this._ifNode(callback(n, 'params[' + i + ']', this)), this));
|
|
};
|
|
/**
|
|
* Create a clone of this node, a shallow copy
|
|
* @return {RelationalNode}
|
|
*/
|
|
|
|
|
|
RelationalNode.prototype.clone = function () {
|
|
return new RelationalNode(this.conditionals, this.params);
|
|
};
|
|
/**
|
|
* Get string representation.
|
|
* @param {Object} options
|
|
* @return {string} str
|
|
*/
|
|
|
|
|
|
RelationalNode.prototype._toString = function (options) {
|
|
var parenthesis = options && options.parenthesis ? options.parenthesis : 'keep';
|
|
var precedence = getPrecedence(this, parenthesis);
|
|
var paramStrings = this.params.map(function (p, index) {
|
|
var paramPrecedence = getPrecedence(p, parenthesis);
|
|
return parenthesis === 'all' || paramPrecedence !== null && paramPrecedence <= precedence ? '(' + p.toString(options) + ')' : p.toString(options);
|
|
});
|
|
var operatorMap = {
|
|
equal: '==',
|
|
unequal: '!=',
|
|
smaller: '<',
|
|
larger: '>',
|
|
smallerEq: '<=',
|
|
largerEq: '>='
|
|
};
|
|
var ret = paramStrings[0];
|
|
|
|
for (var i = 0; i < this.conditionals.length; i++) {
|
|
ret += ' ' + operatorMap[this.conditionals[i]] + ' ' + paramStrings[i + 1];
|
|
}
|
|
|
|
return ret;
|
|
};
|
|
/**
|
|
* Get a JSON representation of the node
|
|
* @returns {Object}
|
|
*/
|
|
|
|
|
|
RelationalNode.prototype.toJSON = function () {
|
|
return {
|
|
mathjs: 'RelationalNode',
|
|
conditionals: this.conditionals,
|
|
params: this.params
|
|
};
|
|
};
|
|
/**
|
|
* Instantiate a RelationalNode from its JSON representation
|
|
* @param {Object} json An object structured like
|
|
* `{"mathjs": "RelationalNode", "condition": ..., "trueExpr": ..., "falseExpr": ...}`,
|
|
* where mathjs is optional
|
|
* @returns {RelationalNode}
|
|
*/
|
|
|
|
|
|
RelationalNode.fromJSON = function (json) {
|
|
return new RelationalNode(json.conditionals, json.params);
|
|
};
|
|
/**
|
|
* Get HTML representation
|
|
* @param {Object} options
|
|
* @return {string} str
|
|
*/
|
|
|
|
|
|
RelationalNode.prototype.toHTML = function (options) {
|
|
var parenthesis = options && options.parenthesis ? options.parenthesis : 'keep';
|
|
var precedence = getPrecedence(this, parenthesis);
|
|
var paramStrings = this.params.map(function (p, index) {
|
|
var paramPrecedence = getPrecedence(p, parenthesis);
|
|
return parenthesis === 'all' || paramPrecedence !== null && paramPrecedence <= precedence ? '<span class="math-parenthesis math-round-parenthesis">(</span>' + p.toHTML(options) + '<span class="math-parenthesis math-round-parenthesis">)</span>' : p.toHTML(options);
|
|
});
|
|
var operatorMap = {
|
|
equal: '==',
|
|
unequal: '!=',
|
|
smaller: '<',
|
|
larger: '>',
|
|
smallerEq: '<=',
|
|
largerEq: '>='
|
|
};
|
|
var ret = paramStrings[0];
|
|
|
|
for (var i = 0; i < this.conditionals.length; i++) {
|
|
ret += '<span class="math-operator math-binary-operator math-explicit-binary-operator">' + escape(operatorMap[this.conditionals[i]]) + '</span>' + paramStrings[i + 1];
|
|
}
|
|
|
|
return ret;
|
|
};
|
|
/**
|
|
* Get LaTeX representation
|
|
* @param {Object} options
|
|
* @return {string} str
|
|
*/
|
|
|
|
|
|
RelationalNode.prototype._toTex = function (options) {
|
|
var parenthesis = options && options.parenthesis ? options.parenthesis : 'keep';
|
|
var precedence = getPrecedence(this, parenthesis);
|
|
var paramStrings = this.params.map(function (p, index) {
|
|
var paramPrecedence = getPrecedence(p, parenthesis);
|
|
return parenthesis === 'all' || paramPrecedence !== null && paramPrecedence <= precedence ? '\\left(' + p.toTex(options) + '\right)' : p.toTex(options);
|
|
});
|
|
var ret = paramStrings[0];
|
|
|
|
for (var i = 0; i < this.conditionals.length; i++) {
|
|
ret += latexOperators[this.conditionals[i]] + paramStrings[i + 1];
|
|
}
|
|
|
|
return ret;
|
|
};
|
|
|
|
return RelationalNode;
|
|
}, {
|
|
isClass: true,
|
|
isNode: true
|
|
}); |