48 lines
1.2 KiB
JavaScript
48 lines
1.2 KiB
JavaScript
|
import { isBigNumber, isCollection, isNumber } from '../../utils/is.js';
|
||
|
|
import { factory } from '../../utils/factory.js';
|
||
|
|
import { errorTransform } from './utils/errorTransform.js';
|
||
|
|
import { createCumSum } from '../../function/statistics/cumsum.js';
|
||
|
|
/**
|
||
|
|
* Attach a transform function to math.sum
|
||
|
|
* Adds a property transform containing the transform function.
|
||
|
|
*
|
||
|
|
* This transform changed the last `dim` parameter of function sum
|
||
|
|
* from one-based to zero based
|
||
|
|
*/
|
||
|
|
|
||
|
|
var name = 'cumsum';
|
||
|
|
var dependencies = ['typed', 'add', 'unaryPlus'];
|
||
|
|
export var createCumSumTransform = /* #__PURE__ */factory(name, dependencies, _ref => {
|
||
|
|
var {
|
||
|
|
typed,
|
||
|
|
add,
|
||
|
|
unaryPlus
|
||
|
|
} = _ref;
|
||
|
|
var cumsum = createCumSum({
|
||
|
|
typed,
|
||
|
|
add,
|
||
|
|
unaryPlus
|
||
|
|
});
|
||
|
|
return typed(name, {
|
||
|
|
'...any': function any(args) {
|
||
|
|
// change last argument dim from one-based to zero-based
|
||
|
|
if (args.length === 2 && isCollection(args[0])) {
|
||
|
|
var dim = args[1];
|
||
|
|
|
||
|
|
if (isNumber(dim)) {
|
||
|
|
args[1] = dim - 1;
|
||
|
|
} else if (isBigNumber(dim)) {
|
||
|
|
args[1] = dim.minus(1);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
try {
|
||
|
|
return cumsum.apply(null, args);
|
||
|
|
} catch (err) {
|
||
|
|
throw errorTransform(err);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}, {
|
||
|
|
isTransformFunction: true
|
||
|
|
});
|