diff options
| author | 2025-07-08 12:15:32 -0400 | |
|---|---|---|
| committer | 2025-07-08 12:15:32 -0400 | |
| commit | 9427a9e53aaa29ee071f4dcb35b809a699d75aa9 (patch) | |
| tree | 2607c185fd8053b84d60041fadc35c05a0225d34 /codegen | |
| parent | Merge pull request 'Fix benchmarks' (#56) from jacobsandlund/zg:benchmarks in... (diff) | |
| parent | Add Words.zig example to README (diff) | |
| download | zg-master.tar.gz zg-master.tar.xz zg-master.zip | |
Diffstat (limited to 'codegen')
| -rw-r--r-- | codegen/wbp.zig | 146 |
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 @@ | |||
| 1 | const std = @import("std"); | ||
| 2 | const builtin = @import("builtin"); | ||
| 3 | |||
| 4 | const 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 | |||
| 26 | const block_size = 256; | ||
| 27 | const Block = [block_size]u5; | ||
| 28 | |||
| 29 | const 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 | |||
| 46 | pub 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 | } | ||