cumsum.transform.js 1.5 KB

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