Nixyri/zig
1
0
Fork 0
mirror of https://github.com/ziglang/zig.git synced 2024-11-15 16:45:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update pow.zig

Browse Source
This commit is contained in:
expikr 2023-11-08 04:10:43 +08:00
parent 77bc8e7b67
commit 521153d1ef
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lib/std/math/complex/pow.zig
View File

@ -5,10 +5,8 @@ const cmath = math.complex;
const Complex = cmath.Complex; const Complex = cmath.Complex;
/// Returns z raised to the complex power of c. /// Returns z raised to the complex power of c.
pub fn pow(comptime T: type, z: T, c: T) T { pub fn pow(z: anytype, s: anytype) Complex(@TypeOf(z.re, z.im, s.re, s.im)) {
const p = cmath.log(z); return cmath.exp(cmath.log(z).mul(s));
const q = c.mul(p);
return cmath.exp(q);
} }
const epsilon = 0.0001; const epsilon = 0.0001;
@ -16,7 +14,7 @@ const epsilon = 0.0001;
test "complex.cpow" { test "complex.cpow" {
const a = Complex(f32).init(5, 3); const a = Complex(f32).init(5, 3);
const b = Complex(f32).init(2.3, -1.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.re, 58.049110, epsilon));
try testing.expect(math.approxEqAbs(f32, c.im, -101.003433, epsilon)); try testing.expect(math.approxEqAbs(f32, c.im, -101.003433, epsilon));