All files / complex-js/functions square.ts

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
import { IComplex, IComplexConstructor } from '../internal/complex';
import mask from '../internal/mask';
 
export default function square<T extends IComplex> (Complex: IComplexConstructor<T>, z: IComplex | number, i = 0): T {
  let zReal: number;
  let zImag: number;
  let zAbs: number;
  let zArg: number;
  let zMask: mask;
 
  if (typeof z === 'number') {
    zReal = z;
    zImag = i;
    zAbs = NaN;
    zArg = NaN;
    zMask = mask.HAS_CARTESIAN;
  } else {
    zReal = z._real;
    zImag = z._imag;
    zAbs = z._abs;
    zArg = z._arg;
    zMask = z._mask;
  }
 
  if ((zMask & mask.HAS_CARTESIAN) !== mask.HAS_CARTESIAN) {
    return new Complex(NaN, NaN, zAbs * zAbs, 2 * zArg, mask.HAS_POLAR);
  }
 
  const real2 = zReal * zReal;
  const imag2 = zImag * zImag;
 
  return new Complex(
    real2 - imag2,
    2 * zReal * zImag,
    real2 + imag2,
    2 * zArg,
    zMask | mask.HAS_ABS
  );
}