| 1 | const std = @import("../../std.zig"); |
| 2 | const testing = std.testing; |
| 3 | const math = std.math; |
| 4 | const cmath = math.complex; |
| 5 | const Complex = cmath.Complex; |
| 6 | |
| 7 | /// Returns the projection of z onto the riemann sphere. |
| 8 | pub fn proj(z: anytype) Complex(@TypeOf(z.re, z.im)) { |
| 9 | const T = @TypeOf(z.re, z.im); |
| 10 | |
| 11 | if (math.isInf(z.re) or math.isInf(z.im)) { |
| 12 | return Complex(T).init(math.inf(T), math.copysign(@as(T, 0.0), z.re)); |
| 13 | } |
| 14 | |
| 15 | return Complex(T).init(z.re, z.im); |
| 16 | } |
| 17 | |
| 18 | test proj { |
| 19 | const a = Complex(f32).init(5, 3); |
| 20 | const c = proj(a); |
| 21 | |
| 22 | try testing.expectEqual(5, c.re); |
| 23 | try testing.expectEqual(3, c.im); |
| 24 | } |