advent-of-code/2022/day10_cathode.py

def main(filename: str, expected_part_1: int = None, expected_part_2: int = None):
    print(f"\n+ Running on {filename}")
    with open(filename) as f:
        data = f.read().strip().split("\n")

    data = parse_data(data)
    solution_part_1 = solve_part_1(data)

    print(f"1. Found {solution_part_1}")
    if expected_part_1:
        assert expected_part_1 == solution_part_1

    solution_part_2 = solve_part_2(data)
    print(f"2. Found {solution_part_2}")
    if expected_part_2:
        assert expected_part_2 == solution_part_2


DataType = list[str]


def parse_data(data: list[str]) -> DataType:
    return data


def solve_part_1(data: DataType) -> int:
    total = 0
    register = 1
    clock = 0
    expected_cycles = [20, 60, 100, 140, 180, 220]
    for instruction in data:
        match instruction.split():
            case ["noop"]:
                clock += 1
                if clock in expected_cycles:
                    print(f"{clock=}, {register=}")
                    total += register * clock
            case ["addx", value]:
                clock += 1
                if clock in expected_cycles:
                    print(f"{clock=}, {register=}")
                    total += register * clock
                clock += 1
                if clock in expected_cycles:
                    print(f"{clock=}, {register=}")
                    total += register * clock
                register += int(value)
    return total


def solve_part_2(data: DataType) -> None:
    register = 1
    clock = 0
    for instruction in data:
        match instruction.split():
            case ["noop"]:
                clock += 1
                end_of_cycle(clock, register)
            case ["addx", value]:
                clock += 1
                end_of_cycle(clock, register)
                clock += 1
                end_of_cycle(clock, register)
                register += int(value)


def end_of_cycle(clock: int, register: int) -> None:
    if (clock % 40 - 1) in sprite_indices(register):
        char = "#"
    else:
        char = "."
    if clock % 40 == 0:
        end = "\n"
    else:
        end = ""
    print(char, end=end)


def sprite_indices(register: int) -> tuple[int, int, int]:
    return register - 1, register, register + 1


if __name__ == "__main__":
    main("inputs/day10-test1", expected_part_1=13140)
    main("inputs/day10", expected_part_1=12560)