ecs.py

@@ -6,6 +6,8 @@ class World:
         self._to_apply: set[Entity] = set()
         self._entities: set[Entity] = set()
         self._mapping: dict[type, set[Entity]] = {}
+        self._to_apply_resources: list[tuple[type, object]] = []
+        self._resources: dict[type, object] = {}
     
     def create_entity(self, *components):
         return Entity(self, *components)
@@ -13,6 +15,15 @@ class World:
     def remove_entity(self, entity: 'Entity'):
         entity._deleted = True
         self._to_apply.add(entity)
+    
+    def set(self, *resources):
+        for resource in resources:
+            self._to_apply_resources.append((type(resource), resource))
+    
+    def remove(self, *resource_types):
+        for resource_type in resource_types:
+            if resource_type in self._resources:
+                self._to_apply_resources.append((resource_type, None))
 
     def apply(self):
         for entity in self._to_apply:
@@ -30,6 +41,11 @@ class World:
                         entity._components[component_type] = component
                         self._mapping.setdefault(component_type, set()).add(entity)
                 entity._to_apply.clear()
+        self._to_apply.clear()
+        for resource_type, resource in self._to_apply_resources:
+            if resource is None: del self._resources[resource_type]
+            else: self._resources[resource_type] = resource
+        self._to_apply_resources.clear()
     
     def query(self, *needed: type, without: tuple[type] = []) -> Iterator['Entity']:
         if not needed:
@@ -41,6 +57,12 @@ class World:
                 if all(entity in self._mapping.get(component_type, set()) for component_type in needed[1:]) and \
                         all(without_type not in entity for without_type in without):
                     yield entity
+    
+    def __getitem__(self, resource_type: type) -> object:
+        return self._resources[resource_type]
+    
+    def __contains__(self, resource_type: type) -> bool:
+        return resource_type in self._resources
 
 class Entity:
     def __init__(self, world: World, *components):

main.py

@@ -50,3 +50,33 @@ for entity in world.query():
         print(entity[Age], end=" ")
     print()
 
+# Création d'une ressource pouvant être ajoutée a un monde
+class Gravity(float): pass
+
+# On peut aussi ajouter des ressources globales
+world.set(Gravity(9.81))
+
+print("On vérifie que la ressource Gravity existe")
+print(Gravity in world)
+
+# On applique les moddifications
+world.apply()
+
+print("On vérifie que la ressource Gravity existe après l'application")
+print(Gravity in world)
+
+print("Récupération de la ressource Gravity")
+print(world[Gravity])
+
+# On supprime la ressource Gravity
+world.remove(Gravity)
+
+print("On vérifie que la ressource Gravity n'existe plus")
+print(Gravity in world)
+
+# On applique les moddifications
+world.apply()
+
+print("On vérifie que la ressource Gravity n'existe plus")
+print(Gravity in world)
+