cumsum.transform.js 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. import { isBigNumber, isCollection, isNumber } from '../../utils/is.js';
  2. import { factory } from '../../utils/factory.js';
  3. import { errorTransform } from './utils/errorTransform.js';
  4. import { createCumSum } from '../../function/statistics/cumsum.js';
  5. /**
  6. * Attach a transform function to math.sum
  7. * Adds a property transform containing the transform function.
  8. *
  9. * This transform changed the last `dim` parameter of function sum
  10. * from one-based to zero based
  11. */
  12. var name = 'cumsum';
  13. var dependencies = ['typed', 'add', 'unaryPlus'];
  14. export var createCumSumTransform = /* #__PURE__ */factory(name, dependencies, _ref => {
  15. var {
  16. typed,
  17. add,
  18. unaryPlus
  19. } = _ref;
  20. var cumsum = createCumSum({
  21. typed,
  22. add,
  23. unaryPlus
  24. });
  25. return typed(name, {
  26. '...any': function any(args) {
  27. // change last argument dim from one-based to zero-based
  28. if (args.length === 2 && isCollection(args[0])) {
  29. var dim = args[1];
  30. if (isNumber(dim)) {
  31. args[1] = dim - 1;
  32. } else if (isBigNumber(dim)) {
  33. args[1] = dim.minus(1);
  34. }
  35. }
  36. try {
  37. return cumsum.apply(null, args);
  38. } catch (err) {
  39. throw errorTransform(err);
  40. }
  41. }
  42. });
  43. }, {
  44. isTransformFunction: true
  45. });