from tools.aoc import AOCDay
from typing import Any

class Day(AOCDay):
    inputs = [
        [
            ("11111000111110000", "input16"),
        ],
        [
            ("10111100110110100", "input16"),
        ]
    ]

    def get_checksum(self, disk_size: int) -> str:
        state = self.getInput()
        while len(state) < disk_size:
            state = state + "0" + "".join("1" if x == "0" else "0" for x in reversed(state))

        state = state[:disk_size]
        while len(state) % 2 == 0:
            state = "".join("1" if state[i] == state[i + 1] else "0" for i in range(0, len(state) - 1, 2))

        return state

    def part1(self) -> Any:
        disk_size = 272
        return self.get_checksum(disk_size)

    def part2(self) -> Any:
        disk_size = 35651584
        return self.get_checksum(disk_size)


if __name__ == '__main__':
    day = Day(2016, 16)
    day.run(verbose=True)