summaryrefslogtreecommitdiff
path: root/codegen
diff options
context:
space:
mode:
authorGravatar Sam Atman2025-07-08 12:15:32 -0400
committerGravatar Sam Atman2025-07-08 12:15:32 -0400
commit9427a9e53aaa29ee071f4dcb35b809a699d75aa9 (patch)
tree2607c185fd8053b84d60041fadc35c05a0225d34 /codegen
parentMerge pull request 'Fix benchmarks' (#56) from jacobsandlund/zg:benchmarks in... (diff)
parentAdd Words.zig example to README (diff)
downloadzg-master.tar.gz
zg-master.tar.xz
zg-master.zip
Merge branch 'develop-next'HEADv0.14.1master
Diffstat (limited to 'codegen')
-rw-r--r--codegen/wbp.zig146
1 files changed, 146 insertions, 0 deletions
diff --git a/codegen/wbp.zig b/codegen/wbp.zig
new file mode 100644
index 0000000..741103e
--- /dev/null
+++ b/codegen/wbp.zig
@@ -0,0 +1,146 @@
1const std = @import("std");
2const builtin = @import("builtin");
3
4const WordBreakProperty = enum(u5) {
5 none,
6 Double_Quote,
7 Single_Quote,
8 Hebrew_Letter,
9 CR,
10 LF,
11 Newline,
12 Extend,
13 Regional_Indicator,
14 Format,
15 Katakana,
16 ALetter,
17 MidLetter,
18 MidNum,
19 MidNumLet,
20 Numeric,
21 ExtendNumLet,
22 ZWJ,
23 WSegSpace,
24};
25
26const block_size = 256;
27const Block = [block_size]u5;
28
29const BlockMap = std.HashMap(
30 Block,
31 u16,
32 struct {
33 pub fn hash(_: @This(), k: Block) u64 {
34 var hasher = std.hash.Wyhash.init(0);
35 std.hash.autoHashStrat(&hasher, k, .DeepRecursive);
36 return hasher.final();
37 }
38
39 pub fn eql(_: @This(), a: Block, b: Block) bool {
40 return std.mem.eql(u5, &a, &b);
41 }
42 },
43 std.hash_map.default_max_load_percentage,
44);
45
46pub fn main() !void {
47 var arena = std.heap.ArenaAllocator.init(std.heap.page_allocator);
48 defer arena.deinit();
49 const allocator = arena.allocator();
50
51 var flat_map = std.AutoHashMap(u21, u5).init(allocator);
52 defer flat_map.deinit();
53
54 var line_buf: [4096]u8 = undefined;
55
56 // Process HangulSyllableType.txt
57 var in_file = try std.fs.cwd().openFile("data/unicode/auxiliary/WordBreakProperty.txt", .{});
58 defer in_file.close();
59 var in_buf = std.io.bufferedReader(in_file.reader());
60 const in_reader = in_buf.reader();
61
62 while (try in_reader.readUntilDelimiterOrEof(&line_buf, '\n')) |line| {
63 if (line.len == 0 or line[0] == '#') continue;
64
65 const no_comment = if (std.mem.indexOfScalar(u8, line, '#')) |octo| line[0..octo] else line;
66
67 var field_iter = std.mem.tokenizeAny(u8, no_comment, "; ");
68 var current_code: [2]u21 = undefined;
69
70 var i: usize = 0;
71 while (field_iter.next()) |field| : (i += 1) {
72 switch (i) {
73 0 => {
74 // Code point(s)
75 if (std.mem.indexOf(u8, field, "..")) |dots| {
76 current_code = .{
77 try std.fmt.parseInt(u21, field[0..dots], 16),
78 try std.fmt.parseInt(u21, field[dots + 2 ..], 16),
79 };
80 } else {
81 const code = try std.fmt.parseInt(u21, field, 16);
82 current_code = .{ code, code };
83 }
84 },
85 1 => {
86 // WordBreak type
87 const st: WordBreakProperty = std.meta.stringToEnum(WordBreakProperty, field) orelse .none;
88 for (current_code[0]..current_code[1] + 1) |cp| try flat_map.put(@intCast(cp), @intFromEnum(st));
89 },
90 else => {},
91 }
92 }
93 }
94
95 var blocks_map = BlockMap.init(allocator);
96 defer blocks_map.deinit();
97
98 var stage1 = std.ArrayList(u16).init(allocator);
99 defer stage1.deinit();
100
101 var stage2 = std.ArrayList(u5).init(allocator);
102 defer stage2.deinit();
103
104 var block: Block = [_]u5{0} ** block_size;
105 var block_len: u16 = 0;
106
107 for (0..0x110000) |i| {
108 const cp: u21 = @intCast(i);
109 const st = flat_map.get(cp) orelse 0;
110
111 // Process block
112 block[block_len] = st;
113 block_len += 1;
114
115 if (block_len < block_size and cp != 0x10ffff) continue;
116
117 const gop = try blocks_map.getOrPut(block);
118 if (!gop.found_existing) {
119 gop.value_ptr.* = @intCast(stage2.items.len);
120 try stage2.appendSlice(&block);
121 }
122
123 try stage1.append(gop.value_ptr.*);
124 block_len = 0;
125 }
126
127 var args_iter = try std.process.argsWithAllocator(allocator);
128 defer args_iter.deinit();
129 _ = args_iter.skip();
130 const output_path = args_iter.next() orelse @panic("No output file arg!");
131
132 const compressor = std.compress.flate.deflate.compressor;
133 var out_file = try std.fs.cwd().createFile(output_path, .{});
134 defer out_file.close();
135 var out_comp = try compressor(.raw, out_file.writer(), .{ .level = .best });
136 const writer = out_comp.writer();
137
138 const endian = builtin.cpu.arch.endian();
139 try writer.writeInt(u16, @intCast(stage1.items.len), endian);
140 for (stage1.items) |i| try writer.writeInt(u16, i, endian);
141
142 try writer.writeInt(u16, @intCast(stage2.items.len), endian);
143 for (stage2.items) |i| try writer.writeInt(u8, i, endian);
144
145 try out_comp.flush();
146}