diff.js 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.createDiff = void 0;
  6. var _factory = require("../../utils/factory.js");
  7. var _number = require("../../utils/number.js");
  8. var _is = require("../../utils/is.js");
  9. var name = 'diff';
  10. var dependencies = ['typed', 'matrix', 'subtract', 'number'];
  11. var createDiff = /* #__PURE__ */(0, _factory.factory)(name, dependencies, function (_ref) {
  12. var typed = _ref.typed,
  13. matrix = _ref.matrix,
  14. subtract = _ref.subtract,
  15. number = _ref.number;
  16. /**
  17. * Create a new matrix or array of the difference between elements of the given array
  18. * The optional dim parameter lets you specify the dimension to evaluate the difference of
  19. * If no dimension parameter is passed it is assumed as dimension 0
  20. *
  21. * Dimension is zero-based in javascript and one-based in the parser and can be a number or bignumber
  22. * Arrays must be 'rectangular' meaning arrays like [1, 2]
  23. * If something is passed as a matrix it will be returned as a matrix but other than that all matrices are converted to arrays
  24. *
  25. * Syntax:
  26. *
  27. * math.diff(arr)
  28. * math.diff(arr, dim)
  29. *
  30. * Examples:
  31. *
  32. * const arr = [1, 2, 4, 7, 0]
  33. * math.diff(arr) // returns [1, 2, 3, -7] (no dimension passed so 0 is assumed)
  34. * math.diff(math.matrix(arr)) // returns Matrix [1, 2, 3, -7]
  35. *
  36. * const arr = [[1, 2, 3, 4, 5], [1, 2, 3, 4, 5], [9, 8, 7, 6, 4]]
  37. * math.diff(arr) // returns [[0, 0, 0, 0, 0], [8, 6, 4, 2, -1]]
  38. * math.diff(arr, 0) // returns [[0, 0, 0, 0, 0], [8, 6, 4, 2, -1]]
  39. * math.diff(arr, 1) // returns [[1, 1, 1, 1], [1, 1, 1, 1], [-1, -1, -1, -2]]
  40. * math.diff(arr, math.bignumber(1)) // returns [[1, 1, 1, 1], [1, 1, 1, 1], [-1, -1, -1, -2]]
  41. *
  42. * math.diff(arr, 2) // throws RangeError as arr is 2 dimensional not 3
  43. * math.diff(arr, -1) // throws RangeError as negative dimensions are not allowed
  44. *
  45. * // These will all produce the same result
  46. * math.diff([[1, 2], [3, 4]])
  47. * math.diff([math.matrix([1, 2]), math.matrix([3, 4])])
  48. * math.diff([[1, 2], math.matrix([3, 4])])
  49. * math.diff([math.matrix([1, 2]), [3, 4]])
  50. * // They do not produce the same result as math.diff(math.matrix([[1, 2], [3, 4]])) as this returns a matrix
  51. *
  52. * See Also:
  53. *
  54. * sum
  55. * subtract
  56. * partitionSelect
  57. *
  58. * @param {Array | Matrix} arr An array or matrix
  59. * @param {number} dim Dimension
  60. * @return {Array | Matrix} Difference between array elements in given dimension
  61. */
  62. return typed(name, {
  63. 'Array | Matrix': function ArrayMatrix(arr) {
  64. // No dimension specified => assume dimension 0
  65. if ((0, _is.isMatrix)(arr)) {
  66. return matrix(_diff(arr.toArray()));
  67. } else {
  68. return _diff(arr);
  69. }
  70. },
  71. 'Array | Matrix, number': function ArrayMatrixNumber(arr, dim) {
  72. if (!(0, _number.isInteger)(dim)) throw new RangeError('Dimension must be a whole number');
  73. if ((0, _is.isMatrix)(arr)) {
  74. return matrix(_recursive(arr.toArray(), dim));
  75. } else {
  76. return _recursive(arr, dim);
  77. }
  78. },
  79. 'Array, BigNumber': typed.referTo('Array,number', function (selfAn) {
  80. return function (arr, dim) {
  81. return selfAn(arr, number(dim));
  82. };
  83. }),
  84. 'Matrix, BigNumber': typed.referTo('Matrix,number', function (selfMn) {
  85. return function (arr, dim) {
  86. return selfMn(arr, number(dim));
  87. };
  88. })
  89. });
  90. /**
  91. * Recursively find the correct dimension in the array/matrix
  92. * Then Apply _diff to that dimension
  93. *
  94. * @param {Array} arr The array
  95. * @param {number} dim Dimension
  96. * @return {Array} resulting array
  97. */
  98. function _recursive(arr, dim) {
  99. if ((0, _is.isMatrix)(arr)) {
  100. arr = arr.toArray(); // Makes sure arrays like [ matrix([0, 1]), matrix([1, 0]) ] are processed properly
  101. }
  102. if (!Array.isArray(arr)) {
  103. throw RangeError('Array/Matrix does not have that many dimensions');
  104. }
  105. if (dim > 0) {
  106. var result = [];
  107. arr.forEach(function (element) {
  108. result.push(_recursive(element, dim - 1));
  109. });
  110. return result;
  111. } else if (dim === 0) {
  112. return _diff(arr);
  113. } else {
  114. throw RangeError('Cannot have negative dimension');
  115. }
  116. }
  117. /**
  118. * Difference between elements in the array
  119. *
  120. * @param {Array} arr An array
  121. * @return {Array} resulting array
  122. */
  123. function _diff(arr) {
  124. var result = [];
  125. var size = arr.length;
  126. for (var i = 1; i < size; i++) {
  127. result.push(_ElementDiff(arr[i - 1], arr[i]));
  128. }
  129. return result;
  130. }
  131. /**
  132. * Difference between 2 objects
  133. *
  134. * @param {Object} obj1 First object
  135. * @param {Object} obj2 Second object
  136. * @return {Array} resulting array
  137. */
  138. function _ElementDiff(obj1, obj2) {
  139. // Convert matrices to arrays
  140. if ((0, _is.isMatrix)(obj1)) obj1 = obj1.toArray();
  141. if ((0, _is.isMatrix)(obj2)) obj2 = obj2.toArray();
  142. var obj1IsArray = Array.isArray(obj1);
  143. var obj2IsArray = Array.isArray(obj2);
  144. if (obj1IsArray && obj2IsArray) {
  145. return _ArrayDiff(obj1, obj2);
  146. }
  147. if (!obj1IsArray && !obj2IsArray) {
  148. return subtract(obj2, obj1); // Difference is (second - first) NOT (first - second)
  149. }
  150. throw TypeError('Cannot calculate difference between 1 array and 1 non-array');
  151. }
  152. /**
  153. * Difference of elements in 2 arrays
  154. *
  155. * @param {Array} arr1 Array 1
  156. * @param {Array} arr2 Array 2
  157. * @return {Array} resulting array
  158. */
  159. function _ArrayDiff(arr1, arr2) {
  160. if (arr1.length !== arr2.length) {
  161. throw RangeError('Not all sub-arrays have the same length');
  162. }
  163. var result = [];
  164. var size = arr1.length;
  165. for (var i = 0; i < size; i++) {
  166. result.push(_ElementDiff(arr1[i], arr2[i]));
  167. }
  168. return result;
  169. }
  170. });
  171. exports.createDiff = createDiff;