diff options
Diffstat (limited to 'codegen/canon.zig')
| -rw-r--r-- | codegen/canon.zig | 27 |
1 files changed, 13 insertions, 14 deletions
diff --git a/codegen/canon.zig b/codegen/canon.zig index 28b7f28..d95a905 100644 --- a/codegen/canon.zig +++ b/codegen/canon.zig | |||
| @@ -1,32 +1,27 @@ | |||
| 1 | const std = @import("std"); | 1 | const std = @import("std"); |
| 2 | const builtin = @import("builtin"); | 2 | const builtin = @import("builtin"); |
| 3 | 3 | ||
| 4 | pub fn main() !void { | 4 | pub fn main() anyerror!void { |
| 5 | var arena = std.heap.ArenaAllocator.init(std.heap.page_allocator); | 5 | var arena = std.heap.ArenaAllocator.init(std.heap.page_allocator); |
| 6 | defer arena.deinit(); | 6 | defer arena.deinit(); |
| 7 | const allocator = arena.allocator(); | 7 | const allocator = arena.allocator(); |
| 8 | 8 | ||
| 9 | var write_buf: [4096]u8 = undefined; | ||
| 9 | // Process UnicodeData.txt | 10 | // Process UnicodeData.txt |
| 10 | var in_file = try std.fs.cwd().openFile("data/unicode/UnicodeData.txt", .{}); | 11 | var in_reader = std.io.Reader.fixed(@embedFile("UnicodeData.txt")); |
| 11 | defer in_file.close(); | ||
| 12 | var in_buf = std.io.bufferedReader(in_file.reader()); | ||
| 13 | const in_reader = in_buf.reader(); | ||
| 14 | |||
| 15 | var args_iter = try std.process.argsWithAllocator(allocator); | 12 | var args_iter = try std.process.argsWithAllocator(allocator); |
| 16 | defer args_iter.deinit(); | 13 | defer args_iter.deinit(); |
| 17 | _ = args_iter.skip(); | 14 | _ = args_iter.skip(); |
| 18 | const output_path = args_iter.next() orelse @panic("No output file arg!"); | 15 | const output_path = args_iter.next() orelse @panic("No output file arg!"); |
| 19 | 16 | ||
| 20 | const compressor = std.compress.flate.deflate.compressor; | ||
| 21 | var out_file = try std.fs.cwd().createFile(output_path, .{}); | 17 | var out_file = try std.fs.cwd().createFile(output_path, .{}); |
| 22 | defer out_file.close(); | 18 | defer out_file.close(); |
| 23 | var out_comp = try compressor(.raw, out_file.writer(), .{ .level = .best }); | 19 | var file_writer = out_file.writer(&write_buf); |
| 24 | const writer = out_comp.writer(); | 20 | var writer = &file_writer.interface; |
| 25 | |||
| 26 | const endian = builtin.cpu.arch.endian(); | 21 | const endian = builtin.cpu.arch.endian(); |
| 27 | var line_buf: [4096]u8 = undefined; | ||
| 28 | 22 | ||
| 29 | lines: while (try in_reader.readUntilDelimiterOrEof(&line_buf, '\n')) |line| { | 23 | lines: while (in_reader.takeDelimiterInclusive('\n')) |took| { |
| 24 | const line = std.mem.trimRight(u8, took, "\n"); | ||
| 30 | if (line.len == 0) continue; | 25 | if (line.len == 0) continue; |
| 31 | 26 | ||
| 32 | var field_iter = std.mem.splitScalar(u8, line, ';'); | 27 | var field_iter = std.mem.splitScalar(u8, line, ';'); |
| @@ -60,8 +55,12 @@ pub fn main() !void { | |||
| 60 | 55 | ||
| 61 | try writer.writeInt(u8, @intCast(len), endian); | 56 | try writer.writeInt(u8, @intCast(len), endian); |
| 62 | for (cps[0..len]) |cp| try writer.writeInt(u24, cp, endian); | 57 | for (cps[0..len]) |cp| try writer.writeInt(u24, cp, endian); |
| 58 | } else |err| switch (err) { | ||
| 59 | error.EndOfStream => {}, | ||
| 60 | else => { | ||
| 61 | return err; | ||
| 62 | }, | ||
| 63 | } | 63 | } |
| 64 | |||
| 65 | try writer.writeInt(u16, 0, endian); | 64 | try writer.writeInt(u16, 0, endian); |
| 66 | try out_comp.flush(); | 65 | try writer.flush(); |
| 67 | } | 66 | } |