aboutsummaryrefslogtreecommitdiff
path: root/lib/std/math/complex
diff options
context:
space:
mode:
authorexpikr <77922942+expikr@users.noreply.github.com>2023-11-08 04:10:43 +0800
committerAndrew Kelley <andrew@ziglang.org>2024-01-14 18:09:26 -0800
commitff23efe9fa46f6ae568ae6d335fc0fd058064aa1 (patch)
tree02068ed3eb80f0c864e3b80fd43ffa5772efd513 /lib/std/math/complex
parent0c70d9c71455122c6d67777c216e47133b8f8416 (diff)
downloadzig-ff23efe9fa46f6ae568ae6d335fc0fd058064aa1.tar.gz
zig-ff23efe9fa46f6ae568ae6d335fc0fd058064aa1.zip
Update pow.zig
Diffstat (limited to 'lib/std/math/complex')
-rw-r--r--lib/std/math/complex/pow.zig8
1 files changed, 3 insertions, 5 deletions
diff --git a/lib/std/math/complex/pow.zig b/lib/std/math/complex/pow.zig
index 36c6796431..af93c2f43c 100644
--- a/lib/std/math/complex/pow.zig
+++ b/lib/std/math/complex/pow.zig
@@ -5,10 +5,8 @@ const cmath = math.complex;
const Complex = cmath.Complex;
/// Returns z raised to the complex power of c.
-pub fn pow(comptime T: type, z: T, c: T) T {
- const p = cmath.log(z);
- const q = c.mul(p);
- return cmath.exp(q);
+pub fn pow(z: anytype, s: anytype) Complex(@TypeOf(z.re, z.im, s.re, s.im)) {
+ return cmath.exp(cmath.log(z).mul(s));
}
const epsilon = 0.0001;
@@ -16,7 +14,7 @@ const epsilon = 0.0001;
test "complex.cpow" {
const a = Complex(f32).init(5, 3);
const b = Complex(f32).init(2.3, -1.3);
- const c = pow(Complex(f32), a, b);
+ const c = pow(a, b);
try testing.expect(math.approxEqAbs(f32, c.re, 58.049110, epsilon));
try testing.expect(math.approxEqAbs(f32, c.im, -101.003433, epsilon));