from MechFabrik.bauteile import Activator, Joint, Mech, InnerStructure, Gyroskope, Area, AreaNames, Cockpit, Reaktor, \
    Sensors, \
    Sustainment
from MechFabrik.reader import ComponentReader


class MechFabrik(object):

    def __init__(self, name: str, tonnage: float, movement_rate: int):
        self.__components = ComponentReader()
        self.__tonnage = tonnage
        self.__reactor_value = int(movement_rate * tonnage)
        self.__areas = {
                AreaNames.HEAD:         Area(),
                AreaNames.MIDDLE_TORSO: Area(),
                AreaNames.LEFT_TORSO:   Area(),
                AreaNames.RIGHT_TORSO:  Area(),
                AreaNames.LEFT_ARM:     Area(),
                AreaNames.RIGHT_ARM:    Area(),
                AreaNames.LEFT_LEG:     Area(),
                AreaNames.RIGHT_LEG:    Area(),
                }
        self.__build_inner_structure(tonnage, movement_rate)
        self._mech = Mech(name, tonnage, movement_rate, self._inner_structure)

    def __build_inner_structure(self, tonnage: float, movement_rate: int) -> None:
        reactor_value = int(movement_rate * tonnage)
        gyroscope = Gyroskope(float(round(reactor_value / 100)))
        self._inner_structure = InnerStructure(self.__components.inner_structure_from(tonnage), gyroscope)
        self.__build_head_area()
        self.__build_middle_torso_area(gyroscope)

    def __build_head_area(self) -> None:
        cockpit = Cockpit()
        self.__areas[AreaNames.HEAD].cockpit = cockpit
        self.__areas[AreaNames.HEAD].items = {
                (1, 1): Sustainment(),
                (1, 2): Sensors(),
                (1, 3): cockpit,
                (1, 4): None,
                (1, 5): Sensors(),
                (1, 6): Sustainment(),
                }
        self.__areas[AreaNames.HEAD].inner_armor = 3

    def __build_middle_torso_area(self, gyroscope: Gyroskope) -> None:
        self.__areas[AreaNames.LEFT_TORSO].parent = self.__areas[AreaNames.MIDDLE_TORSO]
        self.__areas[AreaNames.RIGHT_TORSO].parent = self.__areas[AreaNames.MIDDLE_TORSO]
        reactor = Reaktor(self.__reactor_value, self.__components.reactor_tonnage_from(self.__reactor_value))
        self.__areas[AreaNames.MIDDLE_TORSO].reactor = reactor
        self.__areas[AreaNames.MIDDLE_TORSO].children = [
                self.__areas[AreaNames.RIGHT_TORSO],
                self.__areas[AreaNames.LEFT_TORSO]
                ]
        self.__areas[AreaNames.MIDDLE_TORSO].items = {
                (1, 1): reactor,
                (1, 2): reactor,
                (1, 3): reactor,
                (1, 4): gyroscope,
                (1, 5): gyroscope,
                (1, 6): gyroscope,
                (2, 1): gyroscope,
                (2, 2): reactor,
                (2, 3): reactor,
                (2, 4): reactor,
                (2, 5): None,
                (2, 6): None,
                }
        self.__areas[AreaNames.MIDDLE_TORSO].inner_armor = self.__components.inner_amor_for_tonnage_and_area(
                self.__tonnage,
                AreaNames.MIDDLE_TORSO
                )
        self.__areas[AreaNames.LEFT_TORSO].inner_armor = self.__components.inner_amor_for_tonnage_and_area(
                self.__tonnage, AreaNames.LEFT_TORSO)
        self.__areas[AreaNames.RIGHT_TORSO].inner_armor = self.__components.inner_amor_for_tonnage_and_area(
                self.__tonnage,
                AreaNames.RIGHT_TORSO
                )
        self.__build_arm(self.__areas[AreaNames.RIGHT_TORSO], AreaNames.RIGHT_ARM)
        self.__build_leg(self.__areas[AreaNames.RIGHT_TORSO], AreaNames.RIGHT_LEG)
        self.__build_arm(self.__areas[AreaNames.LEFT_TORSO], AreaNames.LEFT_ARM)
        self.__build_leg(self.__areas[AreaNames.LEFT_TORSO], AreaNames.LEFT_LEG)

    def __build_arm(self, parent: Area, index: AreaNames):
        self.__areas[index].parent = parent
        self.__areas[index].items = {
                (1, 1): Joint(),
                (1, 2): Activator(),
                (1, 3): Activator(),
                (1, 4): Activator(),
                (1, 5): None,
                (1, 6): None,
                (2, 1): None,
                (2, 2): None,
                (2, 3): None,
                (2, 4): None,
                (2, 5): None,
                (2, 6): None,
                }
        self.__areas[index].inner_armor = self.__components.inner_amor_for_tonnage_and_area(self.__tonnage, index)
        parent.children.append(self.__areas[index])


    def __build_leg(self, parent: Area, index: AreaNames):
        self.__areas[index].parent = parent
        self.__areas[index].items = {
                (1, 1): Joint(),
                (1, 2): Activator(),
                (1, 3): Activator(),
                (1, 4): Activator(),
                (1, 5): None,
                (1, 6): None,
                }
        self.__areas[index].inner_armor = self.__components.inner_amor_for_tonnage_and_area(self.__tonnage, index)
        parent.children.append(self.__areas[index])

    def auslieferung(self) -> Mech:
        for (index, area) in self.__areas.items():
            self._inner_structure.add_area(index, area)
        return self._mech

    def set_head_armor(self, armor: int) -> None:
        self.__areas[AreaNames.HEAD].armor = armor

    def set_middle_torso_armor(self, armor: int) -> None:
        self.__areas[AreaNames.MIDDLE_TORSO].armor = armor

    def set_side_torso_armor(self, armor: int) -> None:
        self.__areas[AreaNames.RIGHT_TORSO].armor = armor
        self.__areas[AreaNames.LEFT_TORSO].armor = armor

    def set_arm_armor(self, armor: int) -> None:
        self.__areas[AreaNames.RIGHT_ARM].armor = armor
        self.__areas[AreaNames.LEFT_ARM].armor = armor

    def set_leg_armor(self, armor: int) -> None:
        self.__areas[AreaNames.RIGHT_LEG].armor = armor
        self.__areas[AreaNames.LEFT_LEG].armor = armor