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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
param_amount = {1:3, 2:3, 3:1, 4:1, 5:2, 6:2, 7:3, 8:3, 9:1, 99:0}
ADD = 1
MULT = 2
IN = 3
OUT = 4
JNZ = 5
JEZ = 6
LET = 7
EQV = 8
BAS = 9
HAL = 99
class Computer(object):
def __init__(self, program):
self.memory = program.copy()
self.memory_size = len(self.memory)
self.extra_memory = {}
self.instruction_cache = {}
self.pointer = 0
self.phase_read = False # for day 7
self.relative_base = 0
self.input = None
self.output = None
self.SIG_INPUT = False
self.SIG_OUTPUT = False
self.SIG_HALT = False
def parse_op(self, op):
if op in self.instruction_cache:
return self.instruction_cache[op]
code = op % 100
ops = str(op).zfill(param_amount[code]+2)
self.instruction_cache[op] = [code] + [int(x) for x in ops[:-2][::-1]]
return self.instruction_cache[op]
#return [code] + [(op // 10**(i+2)) % 10**(i+1) for i in range(param_amount[code])]
def clear_flags(self):
self.input = None
self.output = None
def write(self, addr, val):
if addr >= self.memory_size:
self.extra_memory[addr] = val
else:
self.memory[addr] = val
def get(self, addr):
if addr >= self.memory_size:
return self.extra_memory.get(addr, 0)
return self.memory[addr]
def get_param(self, inst, num):
if inst[num] == 0:
return self.get(self.get(self.pointer + num))
elif inst[num] == 1:
return self.get(self.pointer + num)
elif inst[num] == 2:
return self.get(self.relative_base + self.get(self.pointer + num))
def write_param(self, inst, num, val):
if inst[num] == 0:
self.write(self.get(self.pointer + num), val)
elif inst[num] == 1:
self.write(self.pointer + num, val)
elif inst[num] == 2:
self.write(self.relative_base + self.get(self.pointer + num), val)
def step(self):
if self.SIG_OUTPUT and self.output is None:
self.SIG_OUTPUT = False
inst = self.parse_op(self.memory[self.pointer])
if inst[0] == HAL:
self.SIG_HALT = True
return
elif inst[0] == ADD:
self.write_param(inst, 3, \
self.get_param(inst, 1) + self.get_param(inst, 2))
self.pointer += 4
elif inst[0] == MULT:
self.write_param(inst, 3, \
self.get_param(inst, 1) * self.get_param(inst, 2))
self.pointer += 4
elif inst[0] == IN:
if self.input is None:
self.SIG_INPUT = True
return
self.write_param(inst, 1, self.input)
self.input = None
self.SIG_INPUT = False
self.pointer += 2
elif inst[0] == OUT:
if self.SIG_OUTPUT:
return
self.output = self.get_param(inst, 1)
self.SIG_OUTPUT = True
self.pointer += 2
elif inst[0] == JNZ:
if self.get_param(inst, 1) != 0:
self.pointer = self.get_param(inst, 2)
else:
self.pointer += 3
elif inst[0] == JEZ:
if self.get_param(inst, 1) == 0:
self.pointer = self.get_param(inst, 2)
else:
self.pointer += 3
elif inst[0] == LET:
self.write_param(inst, 3, 1 if \
self.get_param(inst, 1) < self.get_param(inst, 2) \
else 0)
self.pointer += 4
elif inst[0] == EQV:
self.write_param(inst, 3, 1 if \
self.get_param(inst, 1) == self.get_param(inst, 2) \
else 0)
self.pointer += 4
elif inst[0] == BAS:
self.relative_base += self.get_param(inst, 1)
self.pointer += 2
else:
print(self.memory)
print(self.pointer)
print("invalid instruction", self.memory[self.pointer])
print(inst)
|
