from solutions import BaseSolution


class Solution(BaseSolution):
    input_file = '11.txt'
    furthest_away = 0
    # https://www.redblobgames.com/grids/hexagons/#coordinates
    DIRECTIONS = {
        'n': lambda x, y, z: (x, y + 1, z - 1),
        'ne': lambda x, y, z: (x + 1, y, z - 1),
        'se': lambda x, y, z: (x + 1, y - 1, z),
        's': lambda x, y, z: (x, y - 1, z + 1),
        'sw': lambda x, y, z: (x - 1, y, z + 1),
        'nw': lambda x, y, z: (x - 1, y + 1, z),
    }

    def __str__(self):
        return 'Day 11: Hex Ed'

    def _get_end(self, steps):
        x = 0
        y = 0
        z = 0
        self.furthest_away = 0
        for step in steps:
            x, y, z = self.DIRECTIONS[step](x, y, z)
            self.furthest_away = max(self.furthest_away, abs(x), abs(y), abs(z))
        return abs(x), abs(y), abs(z)

    def solve(self, puzzle_input):
        x, y, z = self._get_end(puzzle_input.split(','))
        return max(x, y, z)

    def solve_again(self, puzzle_input):
        _, *__ = self._get_end(puzzle_input.split(','))
        return self.furthest_away


if __name__ == '__main__':
    solution = Solution()
    solution.show_results()