day05.py

import logging
from collections import defaultdict

logging.basicConfig(
    level=logging.DEBUG,
    handlers=[logging.StreamHandler(), logging.FileHandler("log.log")],
)

log = logging.getLogger(__name__)


def parse_parameters(tape, register):
    instruction_pointer = register["instruction_pointer"]
    relative_base = register["relative_base"]
    input1 = tape[instruction_pointer + 1]
    input2 = tape[instruction_pointer + 2]
    output = tape[instruction_pointer + 3]

    if register["parameter1_mode"] == 0:
        input1 = tape[input1]
    elif register["parameter1_mode"] == 2:
        input1 = tape[input1 + relative_base]

    if register["parameter2_mode"] == 0:
        input2 = tape[input2]
    elif register["parameter2_mode"] == 2:
        input2 = tape[input2 + relative_base]

    if register["parameter3_mode"] == 1:
        output = tape[output]
    elif register["parameter3_mode"] == 2:
        output = output + relative_base

    return input1, input2, output


def add(tape, register):
    input1, input2, output = parse_parameters(tape, register)
    tape[output] = input1 + input2
    register["instruction_pointer"] += 4
    return tape, register


def mult(tape, register):
    input1, input2, output = parse_parameters(tape, register)
    tape[output] = input1 * input2
    register["instruction_pointer"] += 4
    return tape, register


def input_(tape, register):
    instruction_pointer = register["instruction_pointer"]
    relative_base = register["relative_base"]
    input1 = tape[instruction_pointer + 1]
    if register["parameter1_mode"] == 0:
        tape[input1] = register["input"]
    elif register["parameter1_mode"] == 2:
        tape[input1 + relative_base] = register["input"]
    # log.debug("Input: %s", register["input"])
    register["instruction_pointer"] += 2
    return tape, register


def output_(tape, register):

    output, _, _ = parse_parameters(tape, register)
    # instruction_pointer = register["instruction_pointer"]
    # output = tape[instruction_pointer + 1]
    # output = output if register["parameter1_mode"] else tape[output]
    register["output"] = output
    register["instruction_pointer"] += 2
    # log.debug("Output: %s", output)
    return tape, register


def jump_if_true(tape, register):
    input1, input2, _ = parse_parameters(tape, register)
    # instruction_pointer = register["instruction_pointer"]
    # input1 = tape[instruction_pointer + 1]
    # input1 = input1 if register["parameter1_mode"] else tape[input1]
    # input2 = tape[instruction_pointer + 2]
    # input2 = input2 if register["parameter2_mode"] else tape[input2]
    if input1:
        register["instruction_pointer"] = input2
    else:
        register["instruction_pointer"] += 3
    return tape, register


def jump_if_false(tape, register):
    input1, input2, _ = parse_parameters(tape, register)
    # instruction_pointer = register["instruction_pointer"]
    # input1 = tape[instruction_pointer + 1]
    # input1 = input1 if register["parameter1_mode"] else tape[input1]
    # input2 = tape[instruction_pointer + 2]
    # input2 = input2 if register["parameter2_mode"] else tape[input2]
    if not input1:
        register["instruction_pointer"] = input2
    else:
        register["instruction_pointer"] += 3
    return tape, register


def less_than(tape, register):
    input1, input2, output = parse_parameters(tape, register)
    if input1 < input2:
        tape[output] = 1
    else:
        tape[output] = 0
    register["instruction_pointer"] += 4
    return tape, register


def equals(tape, register):
    input1, input2, output = parse_parameters(tape, register)
    if input1 == input2:
        tape[output] = 1
    else:
        tape[output] = 0
    register["instruction_pointer"] += 4
    return tape, register


def parse_opcode(opcode_number, register):
    """Parse the opcode_number and update the register accordingly"""
    params, _, opcode = str(opcode_number).rpartition("0")
    register["opcode"] = int(str(opcode_number)[-2:])
    register["parameter1_mode"] = register["parameter2_mode"] = register[
        "parameter3_mode"
    ] = 0
    if params:
        register["parameter1_mode"] = int(params[-1])
        try:
            register["parameter2_mode"] = int(params[-2])
        except IndexError:
            return register
        try:
            register["parameter3_mode"] = int(params[-3])
        except IndexError:
            return register
    return register


operation = {
    1: add,
    2: mult,
    3: input_,
    4: output_,
    5: jump_if_true,
    6: jump_if_false,
    7: less_than,
    8: equals,
}


def process_tape(tape, input_value):
    register = dict(
        instruction_pointer=0,
        opcode=0,
        parameter1_mode=0,
        parameter2_mode=0,
        parameter3_mode=0,
        input=input_value,
        output=0,
        relative_base=0,
    )
    while True:

        parse_opcode(tape[register["instruction_pointer"]], register)
        opcode = register["opcode"]
        if opcode == 99:
            return tape, register
        else:
            tape, register = operation[opcode](tape, register)


if __name__ == "__main__":

    with open("day05_input.txt") as f:
        original_list = [int(x) for x in f.readline().split(",")]

    # convert tape from list to default dict
    original_tape = defaultdict(int)
    for k, v in enumerate(original_list):
        original_tape[k] = v

    tape = original_tape.copy()
    t, r = process_tape(tape, 1)
    log.info("Final output: %s", r["output"])  # 9961446
    tape = original_tape.copy()
    t, r = process_tape(tape, 5)
    log.info("Final output: %s", r["output"])  # 742621