AdventOfCode/2019/day13.py

50 lines
1.2 KiB
Python
Raw Normal View History

2019-12-13 13:30:34 +01:00
from intcode import Intcode
def main():
with open("day13.txt", "rt") as input_file:
text = input_file.readlines()[0].rstrip()
codes = Intcode.parse_input(text)
game = Game(codes, debug=False)
game.run()
print("Num block tiles:", sum(sum(v == Game.TILE_BLOCK for v in row) for row in game.tiles))
class Game(Intcode):
TILES_DIM = 64
TILE_EMPTY = 0
TILE_WALL = 1
TILE_BLOCK = 2
TILE_HPAD = 3
TILE_BALL = 4
OUT_X = 0
OUT_Y = 1
OUT_T = 2
def __init__(self, codes, debug=False):
super().__init__(codes, debug=debug)
self.tiles = [[Game.TILE_EMPTY] * Game.TILES_DIM for _ in range(Game.TILES_DIM)]
self.output_type = Game.OUT_X
self.output_pos = [0, 0]
def output_data(self, data):
if self.output_type in [self.OUT_X, self.OUT_Y]:
self.output_pos[self.output_type] = data
elif self.output_type == self.OUT_T:
self.tiles[self.output_pos[1]][self.output_pos[0]] = data
self.output_type = (self.output_type + 1) % 3
def print_tiles(self):
for row in self.tiles:
for value in row:
print(str(value), end="")
print()
if __name__ == "__main__":
main()