mirror of
https://github.com/thib8956/advent-of-code.git
synced 2025-08-23 16:01:59 +00:00
66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
# TODO replace PYTHONPATH hack with a proper solution, like making intcode an
|
|
# installed module https://stackoverflow.com/a/50194143
|
|
import sys
|
|
from pathlib import Path
|
|
sys.path.append(str(Path(__file__).absolute().parent.parent / "intcode"))
|
|
from intcode import interpret_intcode, Interpreter
|
|
|
|
|
|
def paint(program, initial_state):
|
|
interpreter = Interpreter(program)
|
|
pos = 0 + 0j
|
|
direction = 0 - 1j # initially facing up ^
|
|
colors = {}
|
|
colors[pos] = initial_state
|
|
while True:
|
|
interpreter.stdin.append(colors.get(pos, 0))
|
|
interpreter.interpret(break_on_output=True)
|
|
interpreter.interpret(break_on_output=True)
|
|
if interpreter.halted:
|
|
return colors
|
|
|
|
color = interpreter.stdout.pop(0)
|
|
colors[pos] = color
|
|
|
|
turn = interpreter.stdout.pop(0)
|
|
if turn == 0:
|
|
direction *= -1j # turn left
|
|
elif turn == 1:
|
|
direction *= 1j # turn right
|
|
else:
|
|
assert False
|
|
|
|
pos += direction
|
|
|
|
|
|
def main(data, part=1):
|
|
program = list(map(int, data.readline().rstrip().split(",")))
|
|
if part == 1:
|
|
colors = paint(program, initial_state=0)
|
|
print("Part 1: ", len(colors))
|
|
else:
|
|
colors = paint(program, initial_state=1)
|
|
part2(colors)
|
|
|
|
|
|
def part2(colors):
|
|
min_x = int(min(x.real for x in colors.keys()))
|
|
max_x = int(max(x.real for x in colors.keys()))
|
|
min_y = int(min(x.imag for x in colors.keys()))
|
|
max_y = int(max(x.imag for x in colors.keys()))
|
|
for y in range(min_y, max_y + 1):
|
|
l = []
|
|
for x in range(min_x, max_x + 1):
|
|
point = complex(x, y)
|
|
l.append("\u2588" if point in colors and colors[point] == 1 else " ")
|
|
print("".join(l))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import fileinput
|
|
with fileinput.input() as f:
|
|
main(f, part=1)
|
|
#with fileinput.input() as f:
|
|
#main(f, part=2) # FIXME unable to run both parts simultaneously
|
|
|