WIP: collection, entities, objects

marl_factory_grid/environment/groups/collection.py

@@ -0,0 +1,48 @@
+from marl_factory_grid.environment.groups.objects import Objects
+from marl_factory_grid.environment.entity.object import EnvObject
+
+
+class Collection(Objects):
+    _entity = EnvObject
+    var_is_blocking_light: bool = False
+    var_can_collide: bool = False
+    var_can_move: bool = False
+
+    var_has_position: bool = False  # alles was posmixin hat true
+    var_has_bound = False  # batteries, globalpos, inventories true
+    var_can_be_bound: bool = False  # == ^
+
+    @property
+    def encodings(self):
+        return [x.encoding for x in self]
+
+    def __init__(self, size, *args, **kwargs):
+        super(Collection, self).__init__(*args, **kwargs)
+        self.size = size
+
+    def add_item(self, item: EnvObject):
+        assert self.var_has_position or (len(self) <= self.size)
+        super(Collection, self).add_item(item)
+        return self
+
+    def delete_env_object(self, env_object: EnvObject):
+        del self[env_object.name]
+
+    def delete_env_object_by_name(self, name):
+        del self[name]
+
+    @property
+    def obs_pairs(self):
+        return [(x.name, x) for x in self]
+
+    def by_entity(self, entity):
+        try:
+            return next((x for x in self if x.belongs_to_entity(entity)))
+        except (StopIteration, AttributeError):
+            return None
+
+    def idx_by_entity(self, entity):
+        try:
+            return next((idx for idx, x in enumerate(self) if x.belongs_to_entity(entity)))
+        except (StopIteration, AttributeError):
+            return None