Day10 [WIP]

1 files changed, 85 insertions, 0 deletions

diff --git a/day10/__init__.py b/day10/__init__.py
new file mode 100644
index 0000000..e6d385a
--- /dev/null
+++ b/day10/__init__.py
@@ -0,0 +1,85 @@
+# -*- coding: utf-8 -*-
+from abc import ABC
+from functools import lru_cache
+from typing import Iterator, List, Dict, Any
+
+from aoc import BaseAssignment
+
+
+class Assignment(BaseAssignment, ABC):
+    def __init__(self, path):
+        super().__init__(path)
+        self.x_history = {1: 1}
+
+    def addx(self, value: str) -> Iterator[int]:
+        yield 0
+        yield int(value)
+
+    def noop(self) -> Iterator:
+        yield 0
+
+    def x_after(self, cycles: int, instructions: List[str]):
+        try:
+            return self.x_history[cycles]
+        except KeyError:
+            pass
+
+        cycle = 1
+        x = 1
+
+        while cycle <= cycles and len(instructions) > 0:
+            instruction, *instructions = instructions
+            instruction, *args = instruction.split(" ")
+
+            for val in getattr(self, instruction)(*args):
+                x += val
+                cycle += 1
+                self.x_history[cycle] = x
+
+        return self.x_history[cycles]
+
+
+class AssignmentOne(Assignment):
+    example_result = 13140
+
+    def run(self, input: Iterator) -> int:
+        instructions = list(input)
+
+        return sum(
+            [
+                cycles * self.x_after(cycles, instructions)
+                for cycles in [220, 180, 140, 100, 60, 20]
+            ]
+        )
+
+
+class AssignmentTwo(Assignment):
+    example_result = """##..##..##..##..##..##..##..##..##..##..
+###...###...###...###...###...###...###.
+####....####....####....####....####....
+#####.....#####.....#####.....#####.....
+######......######......######......####
+#######.......#######.......#######....."""
+
+    def render_pixel(self, row: int, col: int, instructions: List[str]):
+        cycle = row * 40 + col
+        x_register = self.x_after(cycle, instructions)
+
+        if col in [x_register - 1, x_register, x_register + 1]:
+            return "#"
+        return "."
+
+    def run(self, input: Iterator) -> str:
+        instructions = list(input)
+
+        screen = [
+            [
+                self.render_pixel(row, col, instructions)
+                for col in reversed(range(1, 41))
+            ]
+            for row in reversed(range(6))
+        ]
+
+        return "\n".join(
+            ["".join(reversed([str(i) for i in row])) for row in reversed(screen)]
+        )