renaming

marl_factory_grid/modules/clean_up/entitites.py

@@ -0,0 +1,35 @@
+from numpy import random
+
+from marl_factory_grid.environment.entity.entity import Entity
+from marl_factory_grid.utils.render import RenderEntity
+from marl_factory_grid.modules.clean_up import constants as d
+
+
+class DirtPile(Entity):
+
+    @property
+    def amount(self):
+        return self._amount
+
+    @property
+    def encoding(self):
+        # Edit this if you want items to be drawn in the ops differntly
+        return self._amount
+
+    def __init__(self, *args, max_local_amount=5, initial_amount=2, spawn_variation=0.05, **kwargs):
+        super(DirtPile, self).__init__(*args, **kwargs)
+        self._amount = abs(initial_amount + (
+                random.normal(loc=0, scale=spawn_variation, size=1).item() * initial_amount)
+                           )
+        self.max_local_amount = max_local_amount
+
+    def set_new_amount(self, amount):
+        self._amount = min(amount, self.max_local_amount)
+
+    def summarize_state(self):
+        state_dict = super().summarize_state()
+        state_dict.update(amount=float(self.amount))
+        return state_dict
+
+    def render(self):
+        return RenderEntity(d.DIRT, self.tile.pos, min(0.15 + self.amount, 1.5), 'scale')