diff --git a/.gitignore b/.gitignore index e4f3289..c3667ea 100644 --- a/.gitignore +++ b/.gitignore @@ -1,4 +1,5 @@ */.zig-cache */zig-out */src/test +*/src/test2 */src/input diff --git a/03/build.zig b/03/build.zig new file mode 100644 index 0000000..3aa09e9 --- /dev/null +++ b/03/build.zig @@ -0,0 +1,66 @@ +const std = @import("std"); + +// Although this function looks imperative, note that its job is to +// declaratively construct a build graph that will be executed by an external +// runner. +pub fn build(b: *std.Build) void { + // Standard target options allows the person running `zig build` to choose + // what target to build for. Here we do not override the defaults, which + // means any target is allowed, and the default is native. Other options + // for restricting supported target set are available. + const target = b.standardTargetOptions(.{}); + + // Standard optimization options allow the person running `zig build` to select + // between Debug, ReleaseSafe, ReleaseFast, and ReleaseSmall. Here we do not + // set a preferred release mode, allowing the user to decide how to optimize. + const optimize = b.standardOptimizeOption(.{}); + + const exe = b.addExecutable(.{ + .name = "03", + .root_source_file = b.path("src/main.zig"), + .target = target, + .optimize = optimize, + }); + + // This declares intent for the executable to be installed into the + // standard location when the user invokes the "install" step (the default + // step when running `zig build`). + b.installArtifact(exe); + + // This *creates* a Run step in the build graph, to be executed when another + // step is evaluated that depends on it. The next line below will establish + // such a dependency. + const run_cmd = b.addRunArtifact(exe); + + // By making the run step depend on the install step, it will be run from the + // installation directory rather than directly from within the cache directory. + // This is not necessary, however, if the application depends on other installed + // files, this ensures they will be present and in the expected location. + run_cmd.step.dependOn(b.getInstallStep()); + + // This allows the user to pass arguments to the application in the build + // command itself, like this: `zig build run -- arg1 arg2 etc` + if (b.args) |args| { + run_cmd.addArgs(args); + } + + // This creates a build step. It will be visible in the `zig build --help` menu, + // and can be selected like this: `zig build run` + // This will evaluate the `run` step rather than the default, which is "install". + const run_step = b.step("run", "Run the app"); + run_step.dependOn(&run_cmd.step); + + const exe_unit_tests = b.addTest(.{ + .root_source_file = b.path("src/main.zig"), + .target = target, + .optimize = optimize, + }); + + const run_exe_unit_tests = b.addRunArtifact(exe_unit_tests); + + // Similar to creating the run step earlier, this exposes a `test` step to + // the `zig build --help` menu, providing a way for the user to request + // running the unit tests. + const test_step = b.step("test", "Run unit tests"); + test_step.dependOn(&run_exe_unit_tests.step); +} diff --git a/03/build.zig.zon b/03/build.zig.zon new file mode 100644 index 0000000..14769b3 --- /dev/null +++ b/03/build.zig.zon @@ -0,0 +1,72 @@ +.{ + // This is the default name used by packages depending on this one. For + // example, when a user runs `zig fetch --save `, this field is used + // as the key in the `dependencies` table. Although the user can choose a + // different name, most users will stick with this provided value. + // + // It is redundant to include "zig" in this name because it is already + // within the Zig package namespace. + .name = "03", + + // This is a [Semantic Version](https://semver.org/). + // In a future version of Zig it will be used for package deduplication. + .version = "0.0.0", + + // This field is optional. + // This is currently advisory only; Zig does not yet do anything + // with this value. + //.minimum_zig_version = "0.11.0", + + // This field is optional. + // Each dependency must either provide a `url` and `hash`, or a `path`. + // `zig build --fetch` can be used to fetch all dependencies of a package, recursively. + // Once all dependencies are fetched, `zig build` no longer requires + // internet connectivity. + .dependencies = .{ + // See `zig fetch --save ` for a command-line interface for adding dependencies. + //.example = .{ + // // When updating this field to a new URL, be sure to delete the corresponding + // // `hash`, otherwise you are communicating that you expect to find the old hash at + // // the new URL. + // .url = "https://example.com/foo.tar.gz", + // + // // This is computed from the file contents of the directory of files that is + // // obtained after fetching `url` and applying the inclusion rules given by + // // `paths`. + // // + // // This field is the source of truth; packages do not come from a `url`; they + // // come from a `hash`. `url` is just one of many possible mirrors for how to + // // obtain a package matching this `hash`. + // // + // // Uses the [multihash](https://multiformats.io/multihash/) format. + // .hash = "...", + // + // // When this is provided, the package is found in a directory relative to the + // // build root. In this case the package's hash is irrelevant and therefore not + // // computed. This field and `url` are mutually exclusive. + // .path = "foo", + + // // When this is set to `true`, a package is declared to be lazily + // // fetched. This makes the dependency only get fetched if it is + // // actually used. + // .lazy = false, + //}, + }, + + // Specifies the set of files and directories that are included in this package. + // Only files and directories listed here are included in the `hash` that + // is computed for this package. Only files listed here will remain on disk + // when using the zig package manager. As a rule of thumb, one should list + // files required for compilation plus any license(s). + // Paths are relative to the build root. Use the empty string (`""`) to refer to + // the build root itself. + // A directory listed here means that all files within, recursively, are included. + .paths = .{ + "build.zig", + "build.zig.zon", + "src", + // For example... + //"LICENSE", + //"README.md", + }, +} diff --git a/03/src/main.zig b/03/src/main.zig new file mode 100644 index 0000000..dbf130f --- /dev/null +++ b/03/src/main.zig @@ -0,0 +1,231 @@ +// Day 3! I chose the worst way to do this and then stuck with it +// Also zig is hell I want rust... + +const std = @import("std"); + +var arena = std.heap.ArenaAllocator.init(std.heap.page_allocator); +const alloc = arena.allocator(); + +const test_data = @embedFile("test"); +const test_data2 = @embedFile("test2"); +const input_data = @embedFile("input"); + +pub fn main() !void { + const stdout = std.io.getStdOut().writer(); + + const value1 = try run1(input_data); + try stdout.print("{d}\n", .{value1}); + + const value2 = try run2(input_data); + try stdout.print("{d}\n", .{value2}); +} + +test "test input part 1" { + const value = try run1(test_data); + std.debug.print("{d}\n", .{value}); + try std.testing.expectEqual(value, 161); +} + +test "test input part 2" { + const value = try run2(test_data2); + std.debug.print("{d}\n", .{value}); + try std.testing.expectEqual(value, 48); +} + +const STATE = enum { waiting, m, u, l, open, d, o, n, open_do, close_do, apo, t, open_dont, close_dont, rewind }; + +fn integerUntilDelimiter(reader: anytype, writer: anytype, delimiter: u8, optional_max_size: ?usize) anyerror!void { + if (optional_max_size) |max_size| { + for (0..max_size) |_| { + const byte: u8 = try reader.readByte(); + if (byte == delimiter) return; + if (byte < '0' or byte > '9') return error.BadCharacter; + try writer.writeByte(byte); + } + return error.StreamTooLong; + } else { + while (true) { + const byte: u8 = try reader.readByte(); + if (byte == delimiter) return; + try writer.writeByte(byte); + } + // Can not throw `error.StreamTooLong` since there are no boundary. + } +} + +fn run2(s: []const u8) !u64 { + // zig doesn't have regex + // sobs + + var s_fbs = std.io.fixedBufferStream(s); + var reader = s_fbs.reader(); + var state = STATE.waiting; + var buf: [1024]u8 = undefined; + var total: u64 = 0; + var do_mul: bool = true; + + // i tried a state machine. + // god i don't know zig at all + // this is so genuinely awful + read: while (true) { + // state transition + state = sw: { + break :sw switch (state) { + STATE.rewind => { + try s_fbs.seekBy(-1); + break :sw STATE.waiting; + }, + STATE.waiting => { + switch (reader.readByte() catch { + break :read; + }) { + 'm' => { + break :sw STATE.m; + }, + 'd' => { + break :sw STATE.d; + }, + else => { + break :sw STATE.waiting; + }, + } + }, + STATE.m => if (reader.readByte() catch { + break :read; + } == 'u') STATE.u else STATE.rewind, + STATE.u => if (reader.readByte() catch { + break :read; + } == 'l') STATE.l else STATE.rewind, + STATE.l => if (reader.readByte() catch { + break :read; + } == '(') STATE.open else STATE.rewind, + + STATE.d => if (reader.readByte() catch { + break :read; + } == 'o') STATE.o else STATE.rewind, + STATE.o => switch (reader.readByte() catch { + break :read; + }) { + 'n' => STATE.n, + '(' => STATE.open_do, + else => STATE.rewind, + }, + STATE.open_do => if (reader.readByte() catch { + break :read; + } == ')') STATE.close_do else STATE.rewind, + STATE.n => if (reader.readByte() catch { + break :read; + } == '\'') STATE.apo else STATE.rewind, + STATE.apo => if (reader.readByte() catch { + break :read; + } == 't') STATE.t else STATE.rewind, + STATE.t => if (reader.readByte() catch { + break :read; + } == '(') STATE.open_dont else STATE.rewind, + STATE.open_dont => if (reader.readByte() catch { + break :read; + } == ')') STATE.close_dont else STATE.rewind, + + STATE.close_do => { + do_mul = true; + break :sw STATE.waiting; + }, + STATE.close_dont => { + do_mul = false; + break :sw STATE.waiting; + }, + + STATE.open => { + var fbs = std.io.fixedBufferStream(&buf); + integerUntilDelimiter(reader, fbs.writer(), ',', 1024) catch { + state = STATE.waiting; + continue :read; + }; + const num0 = std.fmt.parseInt(u64, fbs.getWritten(), 10) catch { + state = STATE.waiting; + continue :read; + }; + fbs = std.io.fixedBufferStream(&buf); + integerUntilDelimiter(reader, fbs.writer(), ')', 1024) catch { + state = STATE.waiting; + continue :read; + }; + const num1 = std.fmt.parseInt(u64, fbs.getWritten(), 10) catch { + state = STATE.waiting; + continue :read; + }; + if (do_mul) { + std.debug.print("{d} {d}\n", .{ num0, num1 }); + total += num0 * num1; + } + + break :sw STATE.waiting; + }, + }; + }; + } + + return total; +} + +fn run1(s: []const u8) !u64 { + // zig doesn't have regex + // sobs + + var s_fbs = std.io.fixedBufferStream(s); + var reader = s_fbs.reader(); + var state = STATE.waiting; + var buf: [1024]u8 = undefined; + var total: u64 = 0; + + // i tried a state machine. + // god i don't know zig at all + // this is so genuinely awful + read: while (true) { + // state transition + state = sw: { + break :sw switch (state) { + STATE.waiting => if (reader.readByte() catch { + break :read; + } == 'm') STATE.m else STATE.waiting, + STATE.m => if (reader.readByte() catch { + break :read; + } == 'u') STATE.u else STATE.waiting, + STATE.u => if (reader.readByte() catch { + break :read; + } == 'l') STATE.l else STATE.waiting, + STATE.l => if (reader.readByte() catch { + break :read; + } == '(') STATE.open else STATE.waiting, + + STATE.open => { + var fbs = std.io.fixedBufferStream(&buf); + integerUntilDelimiter(reader, fbs.writer(), ',', 1024) catch { + state = STATE.waiting; + continue :read; + }; + const num0 = std.fmt.parseInt(u64, fbs.getWritten(), 10) catch { + state = STATE.waiting; + continue :read; + }; + fbs = std.io.fixedBufferStream(&buf); + integerUntilDelimiter(reader, fbs.writer(), ')', 1024) catch { + state = STATE.waiting; + continue :read; + }; + const num1 = std.fmt.parseInt(u64, fbs.getWritten(), 10) catch { + state = STATE.waiting; + continue :read; + }; + std.debug.print("{d} {d}\n", .{ num0, num1 }); + total += num0 * num1; + + break :sw STATE.waiting; + }, + else => STATE.waiting, + }; + }; + } + + return total; +}