import { isAccessorNode, isArrayNode, isConstantNode, isFunctionNode, isIndexNode, isObjectNode, isOperatorNode } from '../../utils/is.js'; import { createUtil } from './simplify/util.js'; import { factory } from '../../utils/factory.js'; var name = 'simplifyCore'; var dependencies = ['equal', 'isZero', 'add', 'subtract', 'multiply', 'divide', 'pow', 'AccessorNode', 'ArrayNode', 'ConstantNode', 'FunctionNode', 'IndexNode', 'ObjectNode', 'OperatorNode', 'ParenthesisNode', 'SymbolNode']; export var createSimplifyCore = /* #__PURE__ */factory(name, dependencies, _ref => { var { equal, isZero, add, subtract, multiply, divide, pow, AccessorNode, ArrayNode, ConstantNode, FunctionNode, IndexNode, ObjectNode, OperatorNode, ParenthesisNode, SymbolNode } = _ref; var node0 = new ConstantNode(0); var node1 = new ConstantNode(1); var { hasProperty, isCommutative } = createUtil({ FunctionNode, OperatorNode, SymbolNode }); /** * simplifyCore() performs single pass simplification suitable for * applications requiring ultimate performance. In contrast, simplify() * extends simplifyCore() with additional passes to provide deeper * simplification. * * Syntax: * * simplifyCore(expr) * * Examples: * * const f = math.parse('2 * 1 * x ^ (2 - 1)') * math.simpifyCore(f) // Node {2 * x} * math.simplify('2 * 1 * x ^ (2 - 1)', [math.simplifyCore]) // Node {2 * x} * * See also: * * simplify, resolve, derivative * * @param {Node} node * The expression to be simplified * @param {Object} options * Simplification options, as per simplify() * @return {Node} Returns expression with basic simplifications applied */ function simplifyCore(node, options) { var context = options ? options.context : undefined; if (hasProperty(node, 'trivial', context)) { // This node does nothing if it has only one argument, so if so, // return that argument simplified if (isFunctionNode(node) && node.args.length === 1) { return simplifyCore(node.args[0], options); } // For other node types, we try the generic methods var simpChild = false; var childCount = 0; node.forEach(c => { ++childCount; if (childCount === 1) { simpChild = simplifyCore(c, options); } }); if (childCount === 1) { return simpChild; } } if (isOperatorNode(node) && node.isUnary()) { var a0 = simplifyCore(node.args[0], options); if (node.op === '-') { // unary minus if (isOperatorNode(a0)) { if (a0.isUnary() && a0.op === '-') { return a0.args[0]; } else if (a0.isBinary() && a0.fn === 'subtract') { return new OperatorNode('-', 'subtract', [a0.args[1], a0.args[0]]); } } return new OperatorNode(node.op, node.fn, [a0]); } } else if (isOperatorNode(node) && node.isBinary()) { var _a = simplifyCore(node.args[0], options); var a1 = simplifyCore(node.args[1], options); if (node.op === '+') { if (isConstantNode(_a)) { if (isZero(_a.value)) { return a1; } else if (isConstantNode(a1)) { return new ConstantNode(add(_a.value, a1.value)); } } if (isConstantNode(a1) && isZero(a1.value)) { return _a; } if (isOperatorNode(a1) && a1.isUnary() && a1.op === '-') { return new OperatorNode('-', 'subtract', [_a, a1.args[0]]); } return new OperatorNode(node.op, node.fn, a1 ? [_a, a1] : [_a]); } else if (node.op === '-') { if (isConstantNode(_a) && a1) { if (isConstantNode(a1)) { return new ConstantNode(subtract(_a.value, a1.value)); } else if (isZero(_a.value)) { return new OperatorNode('-', 'unaryMinus', [a1]); } } // if (node.fn === "subtract" && node.args.length === 2) { if (node.fn === 'subtract') { if (isConstantNode(a1) && isZero(a1.value)) { return _a; } if (isOperatorNode(a1) && a1.isUnary() && a1.op === '-') { return simplifyCore(new OperatorNode('+', 'add', [_a, a1.args[0]]), options); } return new OperatorNode(node.op, node.fn, [_a, a1]); } } else if (node.op === '*') { if (isConstantNode(_a)) { if (isZero(_a.value)) { return node0; } else if (equal(_a.value, 1)) { return a1; } else if (isConstantNode(a1)) { return new ConstantNode(multiply(_a.value, a1.value)); } } if (isConstantNode(a1)) { if (isZero(a1.value)) { return node0; } else if (equal(a1.value, 1)) { return _a; } else if (isOperatorNode(_a) && _a.isBinary() && _a.op === node.op && isCommutative(node, context)) { var a00 = _a.args[0]; if (isConstantNode(a00)) { var a00a1 = new ConstantNode(multiply(a00.value, a1.value)); return new OperatorNode(node.op, node.fn, [a00a1, _a.args[1]], node.implicit); // constants on left } } if (isCommutative(node, context)) { return new OperatorNode(node.op, node.fn, [a1, _a], node.implicit); // constants on left } else { return new OperatorNode(node.op, node.fn, [_a, a1], node.implicit); } } return new OperatorNode(node.op, node.fn, [_a, a1], node.implicit); } else if (node.op === '/') { if (isConstantNode(_a)) { if (isZero(_a.value)) { return node0; } else if (isConstantNode(a1) && (equal(a1.value, 1) || equal(a1.value, 2) || equal(a1.value, 4))) { return new ConstantNode(divide(_a.value, a1.value)); } } return new OperatorNode(node.op, node.fn, [_a, a1]); } else if (node.op === '^') { if (isConstantNode(a1)) { if (isZero(a1.value)) { return node1; } else if (equal(a1.value, 1)) { return _a; } else { if (isConstantNode(_a)) { // fold constant return new ConstantNode(pow(_a.value, a1.value)); } else if (isOperatorNode(_a) && _a.isBinary() && _a.op === '^') { var a01 = _a.args[1]; if (isConstantNode(a01)) { return new OperatorNode(node.op, node.fn, [_a.args[0], new ConstantNode(multiply(a01.value, a1.value))]); } } } } } return new OperatorNode(node.op, node.fn, [_a, a1]); } else if (isFunctionNode(node)) { return new FunctionNode(simplifyCore(node.fn), node.args.map(n => simplifyCore(n, options))); } else if (isArrayNode(node)) { return new ArrayNode(node.items.map(n => simplifyCore(n, options))); } else if (isAccessorNode(node)) { return new AccessorNode(simplifyCore(node.object, options), simplifyCore(node.index, options)); } else if (isIndexNode(node)) { return new IndexNode(node.dimensions.map(n => simplifyCore(n, options))); } else if (isObjectNode(node)) { var newProps = {}; for (var prop in node.properties) { newProps[prop] = simplifyCore(node.properties[prop], options); } return new ObjectNode(newProps); } else {// cannot simplify } return node; } return simplifyCore; });