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))