You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
1.3 KiB
41 lines
1.3 KiB
class OpcodeComputer():
|
|
def parse_param(self, pc, mem):
|
|
i = str(mem[pc])
|
|
a, b, c = (0, 0, 0)
|
|
try:
|
|
a = mem[pc+1] if i[-3:-4:-1] == '1' else mem[mem[pc+1]]
|
|
b = mem[pc+2] if i[-4:-5:-1] == '1' else mem[mem[pc+2]]
|
|
c = mem[pc+3] if i[-5:-6:-1] == '1' else mem[mem[pc+3]]
|
|
except IndexError:
|
|
pass
|
|
return (int(i[-2:]), a, b, c)
|
|
|
|
def process_op(self, mem):
|
|
pc = 0
|
|
cmd = mem[pc]
|
|
while cmd != 99:
|
|
cmd, a, b, c = self.parse_param(pc, mem)
|
|
if cmd == 1:
|
|
mem[mem[pc+3]] = a + b
|
|
pc += 4
|
|
elif cmd == 2:
|
|
mem[mem[pc+3]] = a * b
|
|
pc += 4
|
|
elif cmd == 3:
|
|
mem[mem[pc+1]] = int(input())
|
|
pc += 2
|
|
elif cmd == 4:
|
|
print(str(a))
|
|
pc += 2
|
|
elif cmd == 5:
|
|
pc = b if a != 0 else pc + 3
|
|
elif cmd == 6:
|
|
pc = b if a == 0 else pc + 3
|
|
elif cmd == 7:
|
|
mem[mem[pc+3]] = 1 if a < b else 0
|
|
pc += 4
|
|
elif cmd == 8:
|
|
mem[mem[pc+3]] = 1 if a == b else 0
|
|
pc += 4
|
|
cmd = mem[pc]
|
|
return mem[0]
|
|
|