2022-12-23 10:27:39 +00:00
|
|
|
import { Config } from '../core/Config.js'
|
|
|
|
import type { Complex } from '../Complex/type.js'
|
2022-12-07 01:21:05 +00:00
|
|
|
|
2022-12-23 10:27:39 +00:00
|
|
|
export const add = (a: number, b: number): number => a + b
|
2022-12-22 05:14:58 +00:00
|
|
|
export const addReal = add
|
2022-12-23 10:27:39 +00:00
|
|
|
export const unaryMinus = (a: number): number => -a
|
|
|
|
export const conj = (a: number): number => a
|
|
|
|
export const subtract = (a: number, b: number): number => a - b
|
|
|
|
export const multiply = (a: number, b: number): number => a * b
|
|
|
|
export const absquare = (a: number): number => a * a
|
|
|
|
export const reciprocal = (a: number): number => 1 / a
|
|
|
|
export const divide = (a: number, b: number): number => a / b
|
|
|
|
export const divideByReal = divide
|
2022-12-22 05:14:58 +00:00
|
|
|
|
2022-12-23 10:27:39 +00:00
|
|
|
export const conservativeSqrt = (a: number): number => isNaN(a) ? NaN : Math.sqrt(a)
|
2022-12-22 05:14:58 +00:00
|
|
|
|
2022-12-07 01:21:05 +00:00
|
|
|
export const sqrt =
|
2022-12-23 10:27:39 +00:00
|
|
|
(dep: {
|
|
|
|
config: Config,
|
|
|
|
complex: (re: number, im: number) => Complex<number>
|
|
|
|
}): (a: number) => number | Complex<number> => {
|
|
|
|
if (dep.config.predictable || !dep.complex) return conservativeSqrt
|
|
|
|
return a => {
|
|
|
|
if (isNaN(a)) return NaN
|
|
|
|
if (a >= 0) return Math.sqrt(a)
|
|
|
|
return dep.complex(0, Math.sqrt(unaryMinus(a)))
|
|
|
|
}
|
|
|
|
}
|