97 lines
2.2 KiB
Python
Executable File
97 lines
2.2 KiB
Python
Executable File
import enum
|
|
|
|
from mech_factory.components.items import Gyroskope, Item, ItemStatus
|
|
|
|
|
|
class ZoneNames(enum.Enum):
|
|
HEAD = 'Kopf'
|
|
MIDDLE_TORSO_FRONT = 'Torso Mitte Vorn'
|
|
MIDDLE_TORSO_REAR = 'Torso Mitte Rücken'
|
|
RIGHT_TORSO_FRONT = 'Rechter Torso Vorn'
|
|
RIGHT_TORSO_REAR = 'Rechter Torso Rücken'
|
|
LEFT_TORSO_FRONT = 'Linker Torso Vorn'
|
|
LEFT_TORSO_REAR = 'Linker Torso Rücken'
|
|
RIGHT_LEG = 'Rechtes Bein'
|
|
RIGHT_ARM = 'Rechter Arm'
|
|
LEFT_LEG = 'Linkes Bein'
|
|
LEFT_ARM = 'Linker Arm'
|
|
|
|
|
|
class AreaNames(enum.Enum):
|
|
HEAD = 'Kopf'
|
|
MIDDLE_TORSO = 'Torso Mitte'
|
|
RIGHT_TORSO = 'Rechter Torso'
|
|
LEFT_TORSO = 'Linker Torso'
|
|
RIGHT_LEG = 'Rechtes Bein'
|
|
RIGHT_ARM = 'Rechter Arm'
|
|
LEFT_LEG = 'Linkes Bein'
|
|
LEFT_ARM = 'Linker Arm'
|
|
|
|
|
|
class AreaStatus(enum.Flag):
|
|
FINE = 0
|
|
DAMAGED = 1
|
|
DESTROYED = 2
|
|
|
|
|
|
class Zone:
|
|
|
|
def __init__(self):
|
|
self.__dict__['_armor'] = 0
|
|
self.__dict__['_parent'] = None
|
|
self.__dict__['_state'] = AreaStatus.FINE
|
|
|
|
def __setattr__(self, key, value):
|
|
self.__dict__['_' + key] = value
|
|
|
|
def __getattr__(self, item):
|
|
return self.__dict__['_' + item]
|
|
|
|
|
|
class Area(Zone):
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.__dict__['_outer_amor'] = list()
|
|
self.__dict__['_children'] = list()
|
|
self.__dict__['_items'] = dict()
|
|
|
|
|
|
class OuterAmor(Zone):
|
|
|
|
def __init__(self, area: Area):
|
|
super().__init__()
|
|
self.__dict__['_parent'] = area
|
|
self.__dict__['_area'] = area
|
|
|
|
|
|
class InnerStructure:
|
|
|
|
def __init__(self, tonnage: float, gyroskope: Gyroskope):
|
|
self._gyroskope = gyroskope
|
|
self._tonnage = tonnage
|
|
self._areas = dict()
|
|
self._zones = dict()
|
|
|
|
@property
|
|
def tonnage(self) -> float:
|
|
return self._tonnage
|
|
|
|
@property
|
|
def gyroskope(self) -> Gyroskope:
|
|
return self._gyroskope
|
|
|
|
@property
|
|
def areas(self) -> dict[AreaNames, Area]:
|
|
return self._areas
|
|
|
|
def add_area(self, index: AreaNames, area: Area):
|
|
self._areas[index] = area
|
|
|
|
@property
|
|
def zones(self) -> dict[ZoneNames, OuterAmor]:
|
|
return self._zones
|
|
|
|
def add_zone(self, index: ZoneNames, zone: OuterAmor):
|
|
self._zones[index] = zone
|