2019 day 7 part 1

Author: Peter554

2019/python/data/day07/input.txt

@@ -0,0 +1 @@
+3,8,1001,8,10,8,105,1,0,0,21,38,55,64,81,106,187,268,349,430,99999,3,9,101,2,9,9,1002,9,2,9,101,5,9,9,4,9,99,3,9,102,2,9,9,101,3,9,9,1002,9,4,9,4,9,99,3,9,102,2,9,9,4,9,99,3,9,1002,9,5,9,1001,9,4,9,102,4,9,9,4,9,99,3,9,102,2,9,9,1001,9,5,9,102,3,9,9,1001,9,4,9,102,5,9,9,4,9,99,3,9,1002,9,2,9,4,9,3,9,101,2,9,9,4,9,3,9,1002,9,2,9,4,9,3,9,1001,9,2,9,4,9,3,9,1001,9,2,9,4,9,3,9,101,1,9,9,4,9,3,9,1001,9,1,9,4,9,3,9,1001,9,2,9,4,9,3,9,101,1,9,9,4,9,3,9,1001,9,1,9,4,9,99,3,9,1002,9,2,9,4,9,3,9,101,2,9,9,4,9,3,9,1001,9,1,9,4,9,3,9,101,1,9,9,4,9,3,9,101,2,9,9,4,9,3,9,101,2,9,9,4,9,3,9,1001,9,1,9,4,9,3,9,101,1,9,9,4,9,3,9,102,2,9,9,4,9,3,9,101,2,9,9,4,9,99,3,9,1002,9,2,9,4,9,3,9,101,2,9,9,4,9,3,9,102,2,9,9,4,9,3,9,101,2,9,9,4,9,3,9,1001,9,2,9,4,9,3,9,1002,9,2,9,4,9,3,9,1002,9,2,9,4,9,3,9,101,2,9,9,4,9,3,9,1001,9,2,9,4,9,3,9,101,1,9,9,4,9,99,3,9,102,2,9,9,4,9,3,9,1001,9,2,9,4,9,3,9,1002,9,2,9,4,9,3,9,102,2,9,9,4,9,3,9,102,2,9,9,4,9,3,9,101,2,9,9,4,9,3,9,101,1,9,9,4,9,3,9,101,1,9,9,4,9,3,9,1001,9,1,9,4,9,3,9,102,2,9,9,4,9,99,3,9,101,1,9,9,4,9,3,9,1002,9,2,9,4,9,3,9,102,2,9,9,4,9,3,9,1002,9,2,9,4,9,3,9,101,1,9,9,4,9,3,9,102,2,9,9,4,9,3,9,1002,9,2,9,4,9,3,9,1002,9,2,9,4,9,3,9,101,1,9,9,4,9,3,9,102,2,9,9,4,9,99

2019/python/src/aoc2019/day07.py

@@ -0,0 +1,152 @@
+import dataclasses
+import itertools
+from pathlib import Path
+from typing import Self
+
+
+def part1(input: Path) -> int:
+    code = [int(n) for n in input.read_text().split(",")]
+
+    largest_signal = 0
+    for phases in itertools.permutations(range(5)):
+        print(phases)
+        amps = [IntcodeVM(code.copy(), [phase]) for phase in phases]
+        amps[0].push_input(0)
+        amps[1].push_input(amps[0].run_until_halted().outputs[-1])
+        amps[2].push_input(amps[1].run_until_halted().outputs[-1])
+        amps[3].push_input(amps[2].run_until_halted().outputs[-1])
+        amps[4].push_input(amps[3].run_until_halted().outputs[-1])
+        if (signal := amps[4].run_until_halted().outputs[-1]) > largest_signal:
+            largest_signal = signal
+    return largest_signal
+
+
+# def part2(input: Path) -> int:
+#     ...
+
+
+@dataclasses.dataclass(frozen=True)
+class InputRequired: ...
+
+
+@dataclasses.dataclass(frozen=True)
+class ProgramHalted: ...
+
+
+class InputRequiredError(Exception): ...
+
+
+class IntcodeVM:
+    def __init__(self, code: list[int], inputs: list[int]) -> None:
+        self._code = code.copy()
+        self._inputs = inputs.copy()
+        self._outputs: list[int] = []
+        self._ip: int = 0
+
+    def push_input(self, value: int) -> None:
+        self._inputs.append(value)
+
+    @property
+    def outputs(self) -> tuple[int, ...]:
+        return tuple(self._outputs)
+
+    def get_next_output(self) -> int | InputRequired | ProgramHalted:
+        while True:
+            opcode, parameter_modes = self._parse_opcode(self._code[self._ip])
+            match opcode:
+                case 1:
+                    self._set(
+                        self._ip + 3,
+                        parameter_modes[2],
+                        self._get(self._ip + 1, parameter_modes[0])
+                        + self._get(self._ip + 2, parameter_modes[1]),
+                    )
+                    self._ip += 4
+                case 2:
+                    self._set(
+                        self._ip + 3,
+                        parameter_modes[2],
+                        self._get(self._ip + 1, parameter_modes[0])
+                        * self._get(self._ip + 2, parameter_modes[1]),
+                    )
+                    self._ip += 4
+                case 3:
+                    if not self._inputs:
+                        return InputRequired()
+                    self._set(self._ip + 1, parameter_modes[0], self._inputs.pop(0))
+                    self._ip += 2
+                case 4:
+                    output = self._get(self._ip + 1, parameter_modes[0])
+                    self._outputs.append(output)
+                    self._ip += 2
+                    return output
+                case 5:
+                    jump = self._get(self._ip + 1, parameter_modes[0]) != 0
+                    if jump:
+                        self._ip = self._get(self._ip + 2, parameter_modes[1])
+                    else:
+                        self._ip += 3
+                case 6:
+                    jump = self._get(self._ip + 1, parameter_modes[0]) == 0
+                    if jump:
+                        self._ip = self._get(self._ip + 2, parameter_modes[1])
+                    else:
+                        self._ip += 3
+                case 7:
+                    self._set(
+                        self._ip + 3,
+                        parameter_modes[2],
+                        int(
+                            self._get(self._ip + 1, parameter_modes[0])
+                            < self._get(self._ip + 2, parameter_modes[1])
+                        ),
+                    )
+                    self._ip += 4
+                case 8:
+                    self._set(
+                        self._ip + 3,
+                        parameter_modes[2],
+                        int(
+                            self._get(self._ip + 1, parameter_modes[0])
+                            == self._get(self._ip + 2, parameter_modes[1])
+                        ),
+                    )
+                    self._ip += 4
+                case 99:
+                    return ProgramHalted()
+                case _:
+                    raise ValueError(f"Invalid opcode {opcode}")
+
+    def run_until_halted(self) -> Self:
+        while True:
+            output = self.get_next_output()
+            match output:
+                case InputRequired():
+                    raise InputRequiredError
+                case ProgramHalted():
+                    return self
+                case int():
+                    ...
+
+    @staticmethod
+    def _parse_opcode(opcode: int) -> tuple[int, tuple[int, ...]]:
+        s = str(opcode).zfill(5)
+        return int(s[3:]), (int(s[2]), int(s[1]), int(s[0]))
+
+    def _get(self, parameter_ip: int, parameter_mode: int) -> int:
+        match parameter_mode:
+            case 0:
+                return self._code[self._code[parameter_ip]]
+            case 1:
+                return self._code[parameter_ip]
+            case _:
+                raise ValueError(f"Invalid parameter mode {parameter_mode}")
+
+    def _set(self, parameter_ip: int, parameter_mode: int, value: int) -> None:
+        match parameter_mode:
+            case 0:
+                self._code[self._code[parameter_ip]] = value
+            case 1:
+                self._code[parameter_ip] = value
+            case _:
+                raise ValueError(f"Invalid parameter mode {parameter_mode}")

2019/python/tests/test_day07.py

@@ -0,0 +1,13 @@
+from pathlib import Path
+
+from aoc2019 import day07
+
+DATA_DIR = Path(__file__).parent.parent / "data/day07"
+
+
+def test_part1():
+    assert day07.part1(DATA_DIR / "input.txt") == 117312
+
+
+# def test_part2():
+#     assert day07.part2(DATA_DIR / "input.txt") == 1336480