day13.py

#!/usr/bin/env python3

import sys
from lib.intcode import IntcodeVM

EMPTY, WALL, BLOCK, PADDLE, BALL = range(5)


# Open the first argument as input or use stdin if no arguments were given
fin = open(sys.argv[1]) if len(sys.argv) > 1 else sys.stdin

program = list(map(int, fin.read().split(',')))

vm = IntcodeVM(program)
out = vm.run()
blocks = set()

for i in range(0, len(out), 3):
	x, y, t = out[i:i + 3]
	if t == BLOCK:
		blocks.add((x, y))

total_blocks = len(blocks)
print('Part 1:', total_blocks)


vm.orig_code[0] = 2
vm.reset()

score    = 0
paddle_x = 0
inp      = []

while True:
	out = vm.run(inp, resume=True, n_out=3)
	if not out:
		break

	x, y, tile = out
	inp = [0]

	if (x, y) == (-1, 0):
		score = tile
		continue

	if tile == PADDLE:
		paddle_x = x
	elif tile == BALL:
		if x > paddle_x:
			inp[0] = 1
		elif x < paddle_x:
			inp[0] = -1

print('Part 2:', score)