diff --git a/robot_controller.py b/robot_controller.py index 4b6855c..b84138e 100644 --- a/robot_controller.py +++ b/robot_controller.py @@ -3,7 +3,7 @@ import time class RobotController(object): - def __init__(self, fire_ext_max_capacity = 5, collision_damage = 1, k_sof = 1, k_cas = 1, k_asset = 1, k_hp = 1): + def __init__(self, fire_ext_max_capacity = 5, collision_damage = 1, k_sof = 1, k_cas = 1, k_asset = 1, k_hp = 1, fall_damage = 10): # STATE VARIABLES self._fire_ext_max_capacity = fire_ext_max_capacity @@ -13,11 +13,13 @@ class RobotController(object): self._hit_points = self._start_hp self._collision_damage = collision_damage self._charge_points = 100 + self._fall_damage = fall_damage self._collisions = [] self._exted_sofs = [] self._found_casualty = [] self._found_assets = [] + self._falls = [] self._k_sof = k_sof self._k_cas = k_cas @@ -53,7 +55,7 @@ class RobotController(object): return score def get_str_status(self): - status = f"{type(self).__name__}\n - Total score {self.get_score()}\n - Hit points: {self._hit_points}/{self._start_hp}\n - Extingushed sources of fire {len(self._exted_sofs)}\n - Fire extinguiher capacity {self._fire_ext_capacity}/{self._fire_ext_max_capacity}" + status = f"{type(self).__name__}\n - Total score {self.get_score()}\n - Hit points: {self._hit_points}/{self._start_hp}\n - Extingushed sources of fire {len(self._exted_sofs)}\n - Fire extinguiher capacity {self._fire_ext_capacity}/{self._fire_ext_max_capacity}\n - Collisions {len(self._collisions)}\n - Falls {len(self._falls)}" return status ''' EVENTS ''' @@ -74,4 +76,8 @@ class RobotController(object): self._found_assets.append((time_stamp, copy.deepcopy(asset_params))) + def _register_fall(self, time_stamp): + self._falls.append(time_stamp) + +