JavaScript/Maths/ArithmeticGeometricMean.js at master · MMABSOUT/JavaScript · GitHub

28 lines (27 loc) · 1.26 KB

/**

* @function agm

* @description This finds the Arithmetic-Geometric Mean between any 2 numbers.

* @param {Number} a - 1st number, also used to store Arithmetic Mean.

* @param {Number} g - 2nd number, also used to store Geometric Mean.

* @return {Number} - AGM of both numbers.

* @see [AGM](https://en.wikipedia.org/wiki/Arithmetic%E2%80%93geometric_mean)

export const agm = (a, g) => {

if (a === Infinity && g === 0) return NaN

if (Object.is(a, -0) && !Object.is(g, -0)) return 0

if (a === g) return a // avoid rounding errors, and increase efficiency

let x // temp var

do {

;[a, g, x] = [(a + g) / 2, Math.sqrt(a * g), a]

} while (a !== x && !isNaN(a))

`x !== a` ensures the return value has full precision,

and prevents infinite loops caused by rounding differences between `div` and `sqrt` (no need for "epsilon").

If we were to compare `a` with `g`, some input combinations (not all) can cause an infinite loop,

because the rounding mode never changes at runtime.

Precision is not the same as accuracy, but they're related.

This function isn't always 100% accurate (round-errors), but at least is more than 95% accurate.

`!isNaN(x)` prevents infinite loops caused by invalid inputs like: negatives, NaNs and Infinities.

return a

}

Expand file tree