zig

commit 9dfb917c200eb97223bdfcda93c08e509fe1362f (tree)
parent 84d50c892d41850f666f05e05472c23ffabee434
Author: Kate Tsuyu <kate@kxt.io>
Date:   Fri, 28 Aug 2020 08:58:51 -0400

std: Add std.math.divCeil

Diffstat:
M
lib/std/math.zig
 | 
20
++++++++++++++++++++

1 file changed, 20 insertions(+), 0 deletions(-)
diff --git 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;