# Copyright 2023 highstreet technologies GmbH
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
# inspired by http://www.redblobgames.com/grids/hexagons/

# !/usr/bin/python
from network_generation.model.python.hexagon import Hex


class Cube:
    @staticmethod
    def direction_vectors() -> list[Hex]:
        q, r, s = 1, 0, -1
        return [
            Hex(q, r, s),
            Hex(-s, -q, -r),
            Hex(r, s, q),
            Hex(-q, -r, -s),
            Hex(s, q, r),
            Hex(-r, -s, -q),
        ]

    @staticmethod
    def direction(direction: int) -> Hex:
        if direction < 0 or direction > 5:
            raise ValueError(
                "Invalid direction. The direction value must be"
                + " in the range of [0..5]."
            )
        return Cube.direction_vectors()[direction]

    @staticmethod
    def add(hex: Hex, vec: Hex) -> Hex:
        return Hex(hex.q + vec.q, hex.r + vec.r, hex.s + vec.s)

    @staticmethod
    def neighbor(cube: Hex, direction: int) -> Hex:
        return Cube.add(cube, Cube.direction(direction))

    @staticmethod
    def scale(hex: Hex, factor: int) -> Hex:
        return Hex(hex.q * factor, hex.r * factor, hex.s * factor)

    @staticmethod
    def ring(center: Hex, radius: int) -> list[Hex]:
        if not (radius > 0):
            raise ValueError(
                "Invalid radius. The radius around the hex center must"
                + " be greater than 0 rings."
            )
        results: list[Hex] = []
        hex: Hex = Cube.add(center, Cube.scale(Cube.direction(4), radius))
        for i in range(6):
            for j in range(radius):
                results.append(hex)
                hex = Cube.neighbor(hex, i)
        return results

    @staticmethod
    def spiral(center: Hex, radius: int) -> list[Hex]:
        result: list[Hex] = [center]
        for k in range(1, radius + 1):
            result.extend(Cube.ring(center, k))
        return result