zig

commit 224cd8a01eff193e7683362c7743b939b392c75a (tree)
parent 768af66977df40529e7639cd531deb49f6e500ae
Author: Andrew Kelley <superjoe30@gmail.com>
Date:   Tue, 26 Sep 2017 19:40:51 -0400

add fmt.parseInt

Diffstat:
M
std/fmt/index.zig
 | 
20
++++++++++++++++++++

1 file changed, 20 insertions(+), 0 deletions(-)
diff --git a/std/fmt/index.zig b/std/fmt/index.zig
@@ -364,6 +364,26 @@ fn formatIntCallback(context: &FormatIntBuf, bytes: []const u8) -> bool {
     return true;
 }
 
+pub fn parseInt(comptime T: type, buf: []const u8, radix: u8) -> %T {
+    if (!T.is_signed)
+        return parseUnsigned(T, buf, radix);
+    if (buf.len == 0)
+        return T(0);
+    if (buf[0] == '-') {
+        return math.negate(%return parseUnsigned(T, buf[1..], radix));
+    } else if (buf[0] == '+') {
+        return parseUnsigned(T, buf[1..], radix);
+    } else {
+        return parseUnsigned(T, buf, radix);
+    }
+}
+
+test "fmt.parseInt" {
+    assert(%%parseInt(i32, "-10", 10) == -10);
+    assert(%%parseInt(i32, "+10", 10) == 10);
+    assert(if (parseInt(i32, " 10", 10)) |_| false else |err| err == error.InvalidChar);
+}
+
 pub fn parseUnsigned(comptime T: type, buf: []const u8, radix: u8) -> %T {
     var x: T = 0;