with open('day13-input', 'r') as file:
  data = [l.strip('\n') for l in file]
import numpy as np

grid = np.zeros((len(data[0]), len(data)), dtype=np.int64)
minecarts = []

codes = {
  ' ': 0,
  '-': 1,
  '|': 2,
  '+': 3,
  '/': 4,
  '\\': 5,
}
carts = {
  '>': (0, 1),
  'v': (1, 2),
  '<': (2, 1),
  '^': (3, 2),
}

directions = np.array([[1,0], [0,1], [-1,0], [0,-1]])
turns = [-1, 0, 1]

for y, line in enumerate(data):
  for x, char in enumerate(line):
    if char in codes:
      grid[x,y] = codes[char]
    else:
      grid[x,y] = carts[char][1]
      minecarts.append([x, y, carts[char][0], 0])

while len(minecarts) > 1:
  minecarts = sorted(minecarts, key=lambda c: c[1]*200+c[0])
  i = 0
  while i < len(minecarts):
    mc = minecarts[i]
    next_index = list(mc[0:2]+directions[mc[2]])
    g = grid[tuple(next_index)]
    mc[0:2] = next_index
    if g == 1 or g == 2:
      pass
    elif g == 4:
      if mc[2] == 0:
        mc[2] = 3
      elif mc[2] == 1:
        mc[2] = 2
      elif mc[2] == 2:
        mc[2] = 1
      elif mc[2] == 3:
        mc[2] = 0
    elif g == 5:
      if mc[2] == 0:
        mc[2] = 1
      elif mc[2] == 1:
        mc[2] = 0
      elif mc[2] == 2:
        mc[2] = 3
      elif mc[2] == 3:
        mc[2] = 2
    elif g == 3:
      mc[2] = (mc[2]+turns[mc[3]])%4
      mc[3] = (mc[3]+1) % 3
    for j, mc2 in enumerate(minecarts):
      if mc2 == mc:
        continue
      if np.array_equal(mc[0:2], mc2[0:2]):
        print('Crash at', mc[0:2])
        if j > i:
          minecarts.pop(j)
          minecarts.pop(i)
          i -= 1
        else:
          minecarts.pop(i)
          minecarts.pop(j)
          i -= 2
        break
    i += 1
print('Last cart', minecarts[0])