import { factory } from '../../utils/factory.js'

import { createStd } from '../../function/statistics/std.js'

import { isBigNumber, isNumber, isCollection } from '../../utils/is.js'

import { errorTransform } from './utils/errorTransform.js'

const name = 'std'

const dependencies = ['typed', 'sqrt', 'variance']

/**

* Attach a transform function to math.std

* Adds a property transform containing the transform function.

*

* This transform changed the `dim` parameter of function std

* from one-based to zero based

*/

export const createStdTransform = /* #__PURE__ */ factory(name, dependencies, ({ typed, sqrt, variance }) => {

const std = createStd({ typed, sqrt, variance })

return typed('std', {

'...any': function (args) {

// change last argument dim from one-based to zero-based

if (args.length >= 2 && isCollection(args[0])) {

const dim = args[1]

if (isNumber(dim)) {

args[1] = dim - 1

} else if (isBigNumber(dim)) {

args[1] = dim.minus(1)

}

}

try {

return std.apply(null, args)

} catch (err) {

throw errorTransform(err)

}

}

})

}, { isTransformFunction: true })