From d441db73e5b97654022487236065285ea4bb5abd Mon Sep 17 00:00:00 2001 From: Thibaud Date: Mon, 16 Dec 2019 00:23:01 +0100 Subject: [PATCH] wip wip wip --- day5/day5.py | 208 ++++++++++++++++++++++++++++++--------------------- 1 file changed, 124 insertions(+), 84 deletions(-) diff --git a/day5/day5.py b/day5/day5.py index 5f24de0..4b29bca 100644 --- a/day5/day5.py +++ b/day5/day5.py @@ -1,89 +1,122 @@ #! /usr/bin/env python3 from collections import namedtuple +from enum import Enum +import logging + +logging.basicConfig(format="%(levelname)s:%(message)s", level=logging.DEBUG) -def decode_opcode(inp): - opcode = inp % 100 # 2-digit opcode - a = (inp // 10000) % 10 # mode of 1st param - b = (inp // 1000) % 10 # mode of 2nd param - c = (inp // 100) % 10 # mode of 3rd param - # ABC DE - # ^^^ ^^ - # MODE OPCODE - return a, b, c, opcode +def get_nth_digit(n, number): + "Returns the nth digit of the input number" + return number // n ** 10 % 10 -def decode_parameters(input_prog, ip, modes): - """ - Returns the adress of the parameters according to the input modes. - if mode[n] == 1: direct access, else indirect access - """ - param1 = input_prog[ip + 1] if modes[2] == 0 else (ip + 1) - param2 = input_prog[ip + 2] if modes[1] == 0 else (ip + 2) - param3 = input_prog[ip + 3] if modes[0] == 0 else (ip + 3) - return param1, param2, param3 +class Operation(Enum): + ADDITION = 1 + MULTIPLICATION = 2 + INPUT = 3 + OUTPUT = 4 + JMP_IF_TRUE = 5 + JMP_IF_FALSE = 6 + LESS_THAN = 7 + EQUALS = 8 + TERMINATION = 99 +class Mode(Enum): + POSITION = 0 + IMMEDIATE = 1 -def interpret_intcode(input_prog, stdin=None): - # Instruction pointer: index of the next element to be executed + +class Instruction: + def __init__(self, opcode): + self.operation = Operation(opcode % 100) + self.modes = [Mode(get_nth_digit(n, opcode)) for n in range(2, 5)] + self.handler_name = f"handle_{self.operation.name.lower()}" + self.handler = getattr(self, self.handler_name, self.handle_termination) + + def __repr__(self): + return f"Instruction({self.operation}, {self.modes})" + + def handle(self, program, ip): + return self.handler(program, ip) + + def handle_addition(self, program, ip): + first, second = self._get_parameters(program, ip) + logging.debug(f"ADD {first} {second}") + result = first + second + # the last mode should *always* be POSITION + program[program[ip + 3]] = result + ip += 4 + return ip + + def handle_multiplication(self, program, ip): + first, second = self._get_parameters(program, ip) + logging.debug(f"MUL {first} {second}") + result = first * second + # the last mode should *always* be POSITION + program[program[ip + 3]] = result + ip += 4 + return ip + + def handle_input(self, program, ip): + ip += 2 + program[program[ip + 1]] = int(input("Enter ID > ")) + return ip + + def handle_output(self, program, ip): + print("OUT ", program[program[ip + 1]]) + ip += 2 + return ip + + def handle_jmp_if_true(self, program, ip): + first, second = self._get_parameters(program, ip) + logging.debug(f"JMPT {first} {second}") + return second if first != 0 else ip + 3 + + def handle_jmp_if_false(self, program, ip): + first, second = self._get_parameters(program, ip) + logging.debug(f"JMPF {first} {second}") + return second if first == 0 else ip + 3 + + def handle_less_than(self, program, ip): + first, second = self._get_parameters(program, ip) + logging.debug(f"LT {first} {second}") + program[program[ip + 3]] = int(first < second) + ip += 4 + return ip + + def handle_equals(self, program, ip): + first, second = self._get_parameters(program, ip) + logging.debug(f"EQ {first} {second}") + program[program[ip + 3]] = int(first == second) + ip += 4 + return ip + + def handle_termination(self, program, ip): + print("HALT") + return ip + + def _get_parameters(self, program, ip): + first = ( + program[ip + 1] + if self.modes[0] is Mode.IMMEDIATE + else program[program[ip + 1]] + ) + second = ( + program[ip + 2] + if self.modes[1] is Mode.IMMEDIATE + else program[program[ip + 2]] + ) + return first, second + + +def interpret_intcode(program, stdin=None): ip = 0 - while ip < len(input_prog): - instruction = input_prog[ip] - *modes, opcode = decode_opcode(instruction) - - if opcode == 99: - break - - parameters = decode_parameters(input_prog, ip, modes) - - if opcode == 1: # ADD - print("ADD", " ".join(str(input_prog[p]) for p in parameters)) - input_prog[parameters[2]] = input_prog[parameters[0]] + input_prog[parameters[1]] - ip += 4 - - elif opcode == 2: # MUL - print("MUL", " ".join(str(input_prog[p]) for p in parameters)) - input_prog[parameters[2]] = input_prog[parameters[0]] * input_prog[parameters[1]] - ip += 4 - - elif opcode == 3: # IN - input_prog[parameters[1]] = stdin - ip += 2 - - elif opcode == 4: # OUT - if input_prog[ip + 1] == 99: - print("End of execution") - break - - print("OUT ", input_prog[parameters[1]]) - ip += 2 - - elif opcode == 5: # JMP if true - if input_prog[parameters[0]] != 0: - print("JT", " ".join(str(input_prog[p]) for p in parameters)) - ip = input_prog[parameters[1]] - else: - ip += 3 - - elif opcode == 6: # JMP if false - if input_prog[parameters[0]] == 0: - print("JF", " ".join(str(input_prog[p]) for p in parameters)) - ip = input_prog[parameters[1]] - else: - ip += 3 - - elif opcode == 7: # LT - print("LT", " ".join(str(input_prog[p]) for p in parameters)) - res = int(input_prog[parameters[0]] < input_prog[parameters[1]]) - input_prog[parameters[2]] = res - ip += 4 - - elif opcode == 8: # EQ - print("EQ", " ".join(str(input_prog[p]) for p in parameters)) - res = int(input_prog[parameters[1]] == input_prog[parameters[2]]) - input_prog[parameters[2]] = res - ip += 4 + while program[ip] != 99: + opcode = program[ip] + instruction = Instruction(opcode) + ip = instruction.handle(program, ip) def tests(): @@ -92,12 +125,14 @@ def tests(): [2, 3, 0, 3, 99], # MUL 2 * 3 to location 3 [2, 4, 4, 5, 99, 0], [1, 1, 1, 4, 99, 5, 6, 0, 99], + [1101, 1, 1, 0, 99], # ADD 1 + 1 to location 0 (direct access) ) expected_outputs = ( [2, 0, 0, 0, 99], # 1 + 1 = 2 [2, 3, 0, 6, 99], # 3 * 2 = 6 [2, 4, 4, 5, 99, 9801], # 99 * 99 = 9801 [30, 1, 1, 4, 2, 5, 6, 0, 99], + [2, 1, 1, 0, 99], # 1 + 1 = 2 (direct access) ) for i, inp in enumerate(inputs): result = inp[:] @@ -108,16 +143,21 @@ def tests(): print("\nAll tests passed.\n") -if __name__ == "__main__": - tests() - # a = [3, 12, 6, 12, 15, 1, 13, 14, 13, 4, 13, 99, -1, 0, 1, 9] - # b = [3, 3, 1105, -1, 9, 1101, 0, 0, 12, 4, 12, 99, 1] - - # interpret_intcode(b, 0) - # interpret_intcode(a, 0) +def run_input_program(filename): + import os - with open("input.txt") as inp: + SCRIPT_DIR = os.path.dirname(os.path.realpath(__file__)) + with open(os.path.join(SCRIPT_DIR, filename)) as inp: print("Start of input program.") memory = [int(x) for x in inp.readline().strip().split(",")] interpret_intcode(memory, 5) + +if __name__ == "__main__": + # tests() + # a = [3, 12, 6, 12, 15, 1, 13, 14, 13, 4, 13, 99, -1, 0, 1, 9] + # b = [3, 3, 1105, -1, 9, 1101, 0, 0, 12, 4, 12, 99, 1] + + # interpret_intcode(b, 0) + # interpret_intcode(a, 0) + run_input_program("input.txt")