Advent_of_Code_2016/Day2.py at master · Casarella/Advent_of_Code_2016 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
# Advent of Code Day 2

# Keypad is 3x3 square, directions are in file
# "AoC_Day2_directions" - U goes up one key (ignoring
# subsequent retrievals for a direction at the edge)

# Ignoring of commands occurs in two places, where there are >2
# subsequent commands in the same direction OR when you reach an
# edge of the grid and try to keep moving OFF the grid

import numpy as np
import re

directions_path='AoC_Day2_directions'
#directions_list=[]
U = np.array([0,1])
R = np.array([1,0])
D = np.array([0,-1])
L = np.array([-1,0])
C = np.array([0,0])

direction_dict={'U':U,'R':R,'D':D,'L':L,'\n': C}
print(direction_dict)

#directions_list=list(list(open(directions_path,'r'))[2])


#print(directions_list)

key0=[0,0]

# Imagine this is simply a grid, with
# U adding [0,1]
# R adding [1,0]
# D adding [0,-1]
# L adding [-1,0]


# Clip directions at edge
#key_vector=[]
key_vector=np.array([0,0])
def take_step():
  return direction_dict[entry]

#for entry in directions_list:
#for entry in directions_list:
  #take_step()
  #print('step',key_vector)


def normalize_edge():
  """
  After each 'step', check to see if the vector coordinates are >+/-1, if
  they are, just change them to +/-1
  """
  if key_vector[0]>1:
    key_vector[0]=1
  elif key_vector[0]<-1:
    key_vector[0]=-1
  elif key_vector[1]>1:
    key_vector[1]=1
  elif key_vector[1]<-1:
    key_vector[1]=-1
  return key_vector


# Define final button press after directions
def button_press():
  if np.array_equal(key_vector,U+L):
    return '1'
  elif np.array_equal(key_vector,U):
    return '2'
  elif np.array_equal(key_vector,U+R):
    return '3'
  elif np.array_equal(key_vector,L):
    return '4'
  elif np.array_equal(key_vector,C):
    return '5'
  elif np.array_equal(key_vector,R):
    return '6'
  elif np.array_equal(key_vector,D+L):
    return '7'
  elif np.array_equal(key_vector,D):
    return '8'
  else:
    return '9'

#test section
#key_vector=key0+direction_dict['L']
#print(key_vector)
#key_vector+=direction_dict['L']
#print(key_vector)
#normalize_edge()
#print(key_vector)
#print('Button press:',button_press())

# order of operations:
# step, normalize, step until end, return button press
for key_number in range(len(list(open(directions_path,'r')))):
  directions_list=list(list(open(directions_path,'r'))[key_number])
  for entry in directions_list:
    key_vector+=take_step()
    normalize_edge()
  #print('Move:',key_vector)
  print(button_press())