From 9dfb917c200eb97223bdfcda93c08e509fe1362f Mon Sep 17 00:00:00 2001 From: Kate Tsuyu Date: Fri, 28 Aug 2020 08:58:51 -0400 Subject: std: Add std.math.divCeil --- lib/std/math.zig | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) (limited to 'lib/std/math.zig') diff --git a/lib/std/math.zig b/lib/std/math.zig index a76f0a391d..733d6b74b4 100644 --- a/lib/std/math.zig +++ b/lib/std/math.zig @@ -621,6 +621,26 @@ fn testDivFloor() void { testing.expect((divFloor(f32, -5.0, 3.0) catch unreachable) == -2.0); } +pub fn divCeil(comptime T: type, numerator: T, denominator: T) !T { + @setRuntimeSafety(false); + if (numerator <= 0) return divTrunc(T, numerator, denominator); + return (try divFloor(T, numerator - 1, denominator)) + 1; +} + +test "math.divCeil" { + testDivCeil(); + comptime testDivCeil(); +} +fn testDivCeil() void { + testing.expect((divCeil(i32, 5, 3) catch unreachable) == 2); + testing.expect((divCeil(i32, -5, 3) catch unreachable) == -1); + testing.expectError(error.DivisionByZero, divCeil(i8, -5, 0)); + testing.expectError(error.Overflow, divCeil(i8, -128, -1)); + + testing.expect((divCeil(f32, 5.0, 3.0) catch unreachable) == 2.0); + testing.expect((divCeil(f32, -5.0, 3.0) catch unreachable) == -1.0); +} + pub fn divExact(comptime T: type, numerator: T, denominator: T) !T { @setRuntimeSafety(false); if (denominator == 0) return error.DivisionByZero; -- cgit v1.2.3