From d5e99cc05ea05d701adffce55defc512d75e10d1 Mon Sep 17 00:00:00 2001
From: Marc Tiehuis <marctiehuis@gmail.com>
Date: Tue, 24 Apr 2018 19:18:31 +1200
Subject: Add initial complex-number support

 - Library type instead of builtin
 - All C complex functions implemented

Partial WIP: Needs more tests for edge cases.
---
 std/math/complex/arg.zig | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)
 create mode 100644 std/math/complex/arg.zig

(limited to 'std/math/complex/arg.zig')

diff --git a/std/math/complex/arg.zig b/std/math/complex/arg.zig
new file mode 100644
index 0000000000..f24512ac73
--- /dev/null
+++ b/std/math/complex/arg.zig
@@ -0,0 +1,18 @@
+const std = @import("../../index.zig");
+const debug = std.debug;
+const math = std.math;
+const cmath = math.complex;
+const Complex = cmath.Complex;
+
+pub fn arg(z: var) @typeOf(z.re) {
+    const T = @typeOf(z.re);
+    return math.atan2(T, z.im, z.re);
+}
+
+const epsilon = 0.0001;
+
+test "complex.carg" {
+    const a = Complex(f32).new(5, 3);
+    const c = arg(a);
+    debug.assert(math.approxEq(f32, c, 0.540420, epsilon));
+}
-- 
cgit v1.2.3