jyapayne/Nim
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Implement complex sgn (#20087)

Browse source 
Co-authored-by: Clay Sweetser <Varriount@users.noreply.github.com>
This commit is contained in:
Dan Rose 2022-08-28 21:17:18 -05:00 • committed by GitHub
commit cc81866da1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
lib/pure/complex.nim
View file

@ -82,6 +82,13 @@ func abs2*[T](z: Complex[T]): T =
## This is more efficient than `abs(z) ^ 2`. ## This is more efficient than `abs(z) ^ 2`.
result = z.re * z.re + z.im * z.im result = z.re * z.re + z.im * z.im
func sgn*[T](z: Complex[T]): Complex[T] =
## Returns the phase of `z` as a unit complex number,
## or 0 if `z` is 0.
let a = abs(z)
if a != 0:
result = z / a
func conjugate*[T](z: Complex[T]): Complex[T] = func conjugate*[T](z: Complex[T]): Complex[T] =
## Returns the complex conjugate of `z` (`complex(z.re, -z.im)`). ## Returns the complex conjugate of `z` (`complex(z.re, -z.im)`).
result.re = z.re result.re = z.re