renaming

marl_factory_grid/modules/machines/constants.py

@@ -0,0 +1,11 @@
+
+MACHINES = 'Machines'
+MACHINE  = 'Machine'
+
+STATE_WORK = 'working'
+STATE_IDLE = 'idling'
+STATE_MAINTAIN = 'maintenance'
+
+SYMBOL_WORK = 1
+SYMBOL_IDLE = 0.6
+SYMBOL_MAINTAIN = 0.3

marl_factory_grid/modules/machines/entitites.py

@@ -0,0 +1,53 @@
+from marl_factory_grid.environment.entity.entity import Entity
+from marl_factory_grid.utils.render import RenderEntity
+from marl_factory_grid.environment import constants as c
+from marl_factory_grid.utils.results import TickResult
+from marl_factory_grid.modules.machines import constants as m, rewards as r
+
+
+class Machine(Entity):
+
+    @property
+    def encoding(self):
+        return self._encodings[self.state]
+
+    def __init__(self, *args, work_interval: int = 10, pause_interval: int = 15, **kwargs):
+        super(Machine, self).__init__(*args, **kwargs)
+        self._intervals = dict({m.STATE_IDLE: pause_interval, m.STATE_WORK: work_interval})
+        self._encodings = dict({m.STATE_IDLE: pause_interval, m.STATE_WORK: work_interval})
+
+        self.state = m.STATE_IDLE
+        self.health = 100
+        self._counter = 0
+        self.__delattr__('move')
+
+    def maintain(self):
+        if self.state == m.STATE_WORK:
+            return c.NOT_VALID
+        if self.health <= 98:
+            self.health = 100
+            return c.VALID
+        else:
+            return c.NOT_VALID
+
+    def tick(self):
+        if self.state == m.STATE_MAINTAIN and any([c.AGENT in x.name for x in self.tile.guests]):
+            return TickResult(self.name, c.VALID, r.NONE, self)
+        elif self.state == m.STATE_MAINTAIN and not any([c.AGENT in x.name for x in self.tile.guests]):
+            self.state = m.STATE_WORK
+            self.reset_counter()
+            return None
+        elif self._counter:
+            self._counter -= 1
+            self.health -= 1
+            return None
+        else:
+            self.state = m.STATE_WORK if self.state == m.STATE_IDLE else m.STATE_IDLE
+            self.reset_counter()
+            return None
+
+    def reset_counter(self):
+        self._counter = self._intervals[self.state]
+
+    def render(self):
+        return RenderEntity(m.MACHINE, self.pos)

marl_factory_grid/modules/machines/groups.py

@@ -0,0 +1,13 @@
+from marl_factory_grid.environment.groups.env_objects import EnvObjects
+from marl_factory_grid.environment.groups.mixins import PositionMixin
+from marl_factory_grid.modules.machines.entitites import Machine
+
+
+class Machines(PositionMixin, EnvObjects):
+
+    _entity = Machine
+    is_blocking_light: bool = False
+    can_collide: bool = False
+
+    def __init__(self, *args, **kwargs):
+        super(Machines, self).__init__(*args, **kwargs)

marl_factory_grid/modules/machines/rewards.py

@@ -0,0 +1,5 @@
+MAINTAIN_VALID: float = 0.5
+MAINTAIN_FAIL: float = -0.1
+FAIL_MISSING_MAINTENANCE: float = -0.5
+
+NONE: float = 0

marl_factory_grid/modules/machines/rules.py

@@ -0,0 +1,29 @@
+from typing import List
+from marl_factory_grid.environment.rules import Rule
+from marl_factory_grid.utils.results import TickResult, DoneResult
+from marl_factory_grid.environment import constants as c
+from marl_factory_grid.modules.machines import constants as m
+from marl_factory_grid.modules.machines.entitites import Machine
+
+
+class MachineRule(Rule):
+
+    def __init__(self, n_machines: int = 2):
+        super(MachineRule, self).__init__()
+        self.n_machines = n_machines
+
+    def on_init(self, state):
+        empty_tiles = state[c.FLOOR].empty_tiles[:self.n_machines]
+        state[m.MACHINES].add_items(Machine(tile) for tile in empty_tiles)
+
+    def tick_pre_step(self, state) -> List[TickResult]:
+        pass
+
+    def tick_step(self, state) -> List[TickResult]:
+        pass
+
+    def tick_post_step(self, state) -> List[TickResult]:
+        pass
+
+    def on_check_done(self, state) -> List[DoneResult]:
+        pass