diff options
Diffstat (limited to 'solutions/py/d11.py')
| -rw-r--r-- | solutions/py/d11.py | 79 |
1 files changed, 79 insertions, 0 deletions
diff --git a/solutions/py/d11.py b/solutions/py/d11.py new file mode 100644 index 0000000..88d942f --- /dev/null +++ b/solutions/py/d11.py @@ -0,0 +1,79 @@ +import intcode +import time + +_input = open("../input/11","r").readlines() + +program = [int(x) for x in _input[0].split(",")] +x, y = 0, 0 +direction = 0 +# 0 is up +# 1 is left +# 2 is down +# 3 is right +# 4 is up (again) +# turning left is direction += 1 +# turning right is direction -= 1 +# direction is direction % 4 +painted = 0 +colors = {(0,0): 1} # (x,y): 1/0 (1 = white, 0 = black) + +got_color = False + +def draw(colors, ship, direction): + min_x, max_x, min_y, max_y = 0, 0, 0, 0 + ship_c = "" + if direction == 0: + ship_c = "^" + elif direction == 1: + ship_c = "<" + elif direction == 2: + ship_c = "v" + elif direction == 3: + ship_c = ">" + for color in colors: + min_x = min(min_x, color[0]) + max_x = max(max_x, color[0]) + min_y = min(min_y, color[1]) + max_y = max(max_y, color[1]) + for y in range(min_y-5, max_y+5): + for x in range(min_x-5, max_x+5): + if (x,y) == ship: + print(ship_c, end="") + elif (x,y) in colors: + c = colors[(x,y)] + print("\u2588" if c == 1 else ".", end="") + else: + print(" ", end="") + print("") + + +c = intcode.Computer(program) +while c.memory[c.pointer] != 99: + # input + #print(x, y) + #draw(colors, (x,y), direction % 4) + #input() + #time.sleep(0.1) + c.input = colors.get((x, y), 0) + c.step() + if c.output is not None: + if not got_color: + colors[(x, y)] = c.output + c.output = None + got_color = True + elif got_color: + direction += (1 if c.output == 0 else -1) + dir = direction % 4 + if dir == 0: + y -= 1 + elif dir == 1: + x -= 1 + elif dir == 2: + y += 1 + elif dir == 3: + x += 1 + got_color = False + c.output = None +draw(colors, (0,0), 0) +print(colors) +print(len(colors)) |