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
|
params = {1:3, 2:3, 3:1, 4:1, 5:2, 6:2, 7:3, 8:3, 99:0}
ADD = 1
MULT = 2
IN = 3
OUT = 4
JNZ = 5
JEZ = 6
LET = 7
EQV = 8
HAL = 99
class Computer(object):
def __init__(self, program):
self.memory = program.copy()
self.pointer = 0
self.phase_read = False
self.input = None
self.output = None
def parse_op(self, op):
code = op % 100
return [code] + [(op // 10**(i+2)) % 10**(i+1) for i in range(params[code])]
def clear_flags(self):
self.input = None
self.output = None
def get_param(self, inst, num):
return self.memory[self.pointer + num] if inst[num] == True else\
self.memory[self.memory[self.pointer + num]]
def step(self):
inst = self.parse_op(self.memory[self.pointer])
if inst[0] == HAL:
return
elif inst[0] == ADD:
self.memory[self.memory[self.pointer+3]] = \
self.get_param(inst, 1) + self.get_param(inst, 2)
self.pointer += 4
elif inst[0] == MULT:
self.memory[self.memory[self.pointer+3]] = \
self.get_param(inst, 1) * self.get_param(inst, 2)
self.pointer += 4
elif inst[0] == IN:
self.memory[self.memory[self.pointer+1]] = self.input
self.input = None
self.pointer += 2
elif inst[0] == OUT:
self.output = self.get_param(inst, 1)
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.memory[self.memory[self.pointer+3]] = 1 if \
self.get_param(inst, 1) < self.get_param(inst, 2) \
else 0
self.pointer += 4
elif inst[0] == EQV:
self.memory[self.memory[self.pointer+3]] = 1 if \
self.get_param(inst, 1) == self.get_param(inst, 2) \
else 0
self.pointer += 4
else:
print(self.memory)
print(self.pointer)
print("invalid instruction", self.memory[self.pointer])
print(inst)
|
