mirror/zig
1
0
Fork 0
mirror of https://github.com/ziglang/zig.git synced 2025-12-06 14:23:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

std.math.complex: Add squared magnitude function (#21998)

Browse Source
This commit is contained in:
Chris Boesch 2024-11-26 14:03:48 +01:00 committed by GitHub
parent b0dcce93f7
commit 87863a834b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
lib/std/math/complex.zig
View File

@ -115,6 +115,10 @@ pub fn Complex(comptime T: type) type {
pub fn magnitude(self: Self) T { pub fn magnitude(self: Self) T {
return @sqrt(self.re * self.re + self.im * self.im); return @sqrt(self.re * self.re + self.im * self.im);
} }
pub fn squaredMagnitude(self: Self) T {
return self.re * self.re + self.im * self.im;
}
}; };
} }
@ -189,6 +193,13 @@ test "magnitude" {
try testing.expect(math.approxEqAbs(f32, c, 5.83095, epsilon)); try testing.expect(math.approxEqAbs(f32, c, 5.83095, epsilon));
} }
test "squaredMagnitude" {
const a = Complex(f32).init(5, 3);
const c = a.squaredMagnitude();
try testing.expect(math.approxEqAbs(f32, c, math.pow(f32, a.magnitude(), 2), epsilon));
}
test { test {
_ = @import("complex/abs.zig"); _ = @import("complex/abs.zig");
_ = @import("complex/acosh.zig"); _ = @import("complex/acosh.zig");