1 files changed, 33 insertions, 0 deletions

diff --git a/bench/src/zg_width.zig b/bench/src/zg_width.zig
new file mode 100644
index 0000000..c0ce57c
--- /dev/null
+++ b/bench/src/zg_width.zig
@@ -0,0 +1,33 @@
+const std = @import("std");
+
+const DisplayWidth = @import("DisplayWidth");
+
+pub fn main() !void {
+    var args_iter = std.process.args();
+    _ = args_iter.skip();
+    const in_path = args_iter.next() orelse return error.MissingArg;
+
+    var arena = std.heap.ArenaAllocator.init(std.heap.page_allocator);
+    defer arena.deinit();
+    const allocator = arena.allocator();
+
+    const input = try std.fs.cwd().readFileAlloc(
+        allocator,
+        in_path,
+        std.math.maxInt(u32),
+    );
+    defer allocator.free(input);
+
+    const width_data = try DisplayWidth.DisplayWidthData.init(allocator);
+    const display_width = DisplayWidth{ .data = &width_data };
+
+    var iter = std.mem.splitScalar(u8, input, '\n');
+    var result: usize = 0;
+    var timer = try std.time.Timer.start();
+
+    while (iter.next()) |line| {
+        const width = display_width.strWidth(line);
+        result += width;
+    }
+    std.debug.print("zg DisplayWidth.strWidth: result: {}, took: {}\n", .{ result, timer.lap() / std.time.ns_per_ms });
+}