123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261 |
- import _defineProperty from "@babel/runtime/helpers/defineProperty";
- function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); enumerableOnly && (symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; })), keys.push.apply(keys, symbols); } return keys; }
- function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = null != arguments[i] ? arguments[i] : {}; i % 2 ? ownKeys(Object(source), !0).forEach(function (key) { _defineProperty(target, key, source[key]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)) : ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } return target; }
- import { isFunctionNode, isOperatorNode, isParenthesisNode } from '../../../utils/is.js';
- import { factory } from '../../../utils/factory.js';
- import { hasOwnProperty } from '../../../utils/object.js';
- var name = 'simplifyUtil';
- var dependencies = ['FunctionNode', 'OperatorNode', 'SymbolNode'];
- export var createUtil = /* #__PURE__ */factory(name, dependencies, _ref => {
- var {
- FunctionNode,
- OperatorNode,
- SymbolNode
- } = _ref;
- // TODO commutative/associative properties rely on the arguments
- // e.g. multiply is not commutative for matrices
- // The properties should be calculated from an argument to simplify, or possibly something in math.config
- // the other option is for typed() to specify a return type so that we can evaluate the type of arguments
- /* So that properties of an operator fit on one line: */
- var T = true;
- var F = false;
- var defaultName = 'defaultF';
- var defaultContext = {
- /* */add: {
- trivial: T,
- total: T,
- commutative: T,
- associative: T
- },
- /**/unaryPlus: {
- trivial: T,
- total: T,
- commutative: T,
- associative: T
- },
- /* */subtract: {
- trivial: F,
- total: T,
- commutative: F,
- associative: F
- },
- /* */multiply: {
- trivial: T,
- total: T,
- commutative: T,
- associative: T
- },
- /* */divide: {
- trivial: F,
- total: T,
- commutative: F,
- associative: F
- },
- /* */paren: {
- trivial: T,
- total: T,
- commutative: T,
- associative: F
- },
- /* */defaultF: {
- trivial: F,
- total: T,
- commutative: F,
- associative: F
- }
- };
- var realContext = {
- divide: {
- total: F
- },
- log: {
- total: F
- }
- };
- var positiveContext = {
- subtract: {
- total: F
- },
- abs: {
- trivial: T
- },
- log: {
- total: T
- }
- };
- function hasProperty(nodeOrName, property) {
- var context = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : defaultContext;
- var name = defaultName;
- if (typeof nodeOrName === 'string') {
- name = nodeOrName;
- } else if (isOperatorNode(nodeOrName)) {
- name = nodeOrName.fn.toString();
- } else if (isFunctionNode(nodeOrName)) {
- name = nodeOrName.name;
- } else if (isParenthesisNode(nodeOrName)) {
- name = 'paren';
- }
- if (hasOwnProperty(context, name)) {
- var properties = context[name];
- if (hasOwnProperty(properties, property)) {
- return properties[property];
- }
- if (hasOwnProperty(defaultContext, name)) {
- return defaultContext[name][property];
- }
- }
- if (hasOwnProperty(context, defaultName)) {
- var _properties = context[defaultName];
- if (hasOwnProperty(_properties, property)) {
- return _properties[property];
- }
- return defaultContext[defaultName][property];
- }
- /* name not found in context and context has no global default */
- /* So use default context. */
- if (hasOwnProperty(defaultContext, name)) {
- var _properties2 = defaultContext[name];
- if (hasOwnProperty(_properties2, property)) {
- return _properties2[property];
- }
- }
- return defaultContext[defaultName][property];
- }
- function isCommutative(node) {
- var context = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : defaultContext;
- return hasProperty(node, 'commutative', context);
- }
- function isAssociative(node) {
- var context = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : defaultContext;
- return hasProperty(node, 'associative', context);
- }
- /**
- * Merge the given contexts, with primary overriding secondary
- * wherever they might conflict
- */
- function mergeContext(primary, secondary) {
- var merged = _objectSpread({}, primary);
- for (var prop in secondary) {
- if (hasOwnProperty(primary, prop)) {
- merged[prop] = _objectSpread(_objectSpread({}, secondary[prop]), primary[prop]);
- } else {
- merged[prop] = secondary[prop];
- }
- }
- return merged;
- }
- /**
- * Flatten all associative operators in an expression tree.
- * Assumes parentheses have already been removed.
- */
- function flatten(node, context) {
- if (!node.args || node.args.length === 0) {
- return node;
- }
- node.args = allChildren(node, context);
- for (var i = 0; i < node.args.length; i++) {
- flatten(node.args[i], context);
- }
- }
- /**
- * Get the children of a node as if it has been flattened.
- * TODO implement for FunctionNodes
- */
- function allChildren(node, context) {
- var op;
- var children = [];
- var findChildren = function findChildren(node) {
- for (var i = 0; i < node.args.length; i++) {
- var child = node.args[i];
- if (isOperatorNode(child) && op === child.op) {
- findChildren(child);
- } else {
- children.push(child);
- }
- }
- };
- if (isAssociative(node, context)) {
- op = node.op;
- findChildren(node);
- return children;
- } else {
- return node.args;
- }
- }
- /**
- * Unflatten all flattened operators to a right-heavy binary tree.
- */
- function unflattenr(node, context) {
- if (!node.args || node.args.length === 0) {
- return;
- }
- var makeNode = createMakeNodeFunction(node);
- var l = node.args.length;
- for (var i = 0; i < l; i++) {
- unflattenr(node.args[i], context);
- }
- if (l > 2 && isAssociative(node, context)) {
- var curnode = node.args.pop();
- while (node.args.length > 0) {
- curnode = makeNode([node.args.pop(), curnode]);
- }
- node.args = curnode.args;
- }
- }
- /**
- * Unflatten all flattened operators to a left-heavy binary tree.
- */
- function unflattenl(node, context) {
- if (!node.args || node.args.length === 0) {
- return;
- }
- var makeNode = createMakeNodeFunction(node);
- var l = node.args.length;
- for (var i = 0; i < l; i++) {
- unflattenl(node.args[i], context);
- }
- if (l > 2 && isAssociative(node, context)) {
- var curnode = node.args.shift();
- while (node.args.length > 0) {
- curnode = makeNode([curnode, node.args.shift()]);
- }
- node.args = curnode.args;
- }
- }
- function createMakeNodeFunction(node) {
- if (isOperatorNode(node)) {
- return function (args) {
- try {
- return new OperatorNode(node.op, node.fn, args, node.implicit);
- } catch (err) {
- console.error(err);
- return [];
- }
- };
- } else {
- return function (args) {
- return new FunctionNode(new SymbolNode(node.name), args);
- };
- }
- }
- return {
- createMakeNodeFunction,
- hasProperty,
- isCommutative,
- isAssociative,
- mergeContext,
- flatten,
- allChildren,
- unflattenr,
- unflattenl,
- defaultContext,
- realContext,
- positiveContext
- };
- });
|