@codehz/ecs 0.3.4 → 0.3.5

package/index.js

@@ -884,6 +884,7 @@ class World {
   commandBuffer = new CommandBuffer((entityId, commands) => this.executeEntityCommands(entityId, commands));
   hooks = new Map;
   exclusiveComponents = new Set;
+  cascadeDeleteComponents = new Set;
   constructor(snapshot) {
     if (snapshot && typeof snapshot === "object") {
       if (snapshot.entityManager) {
@@ -953,14 +954,29 @@ class World {
     return entityId;
   }
   destroyEntityImmediate(entityId) {
-    const archetype = this.entityToArchetype.get(entityId);
-    if (!archetype) {
-      return;
-    }
-    const componentReferences = this.getEntityReferences(entityId);
-    for (const [sourceEntityId, componentType] of componentReferences) {
-      const sourceArchetype = this.entityToArchetype.get(sourceEntityId);
-      if (sourceArchetype) {
+    const queue = [entityId];
+    const visited = new Set;
+    while (queue.length > 0) {
+      const cur = queue.shift();
+      if (visited.has(cur))
+        continue;
+      visited.add(cur);
+      const archetype = this.entityToArchetype.get(cur);
+      if (!archetype) {
+        continue;
+      }
+      const componentReferences = Array.from(this.getEntityReferences(cur));
+      for (const [sourceEntityId, componentType] of componentReferences) {
+        const sourceArchetype = this.entityToArchetype.get(sourceEntityId);
+        if (!sourceArchetype)
+          continue;
+        const detailedType = getDetailedIdType(componentType);
+        if (detailedType.type === "entity-relation" && this.cascadeDeleteComponents.has(detailedType.componentId)) {
+          if (!visited.has(sourceEntityId)) {
+            queue.push(sourceEntityId);
+          }
+          continue;
+        }
         const currentComponents = new Map;
         let removedComponent = sourceArchetype.get(sourceEntityId, componentType);
         for (const archetypeComponentType of sourceArchetype.componentTypes) {
@@ -976,17 +992,17 @@ class World {
         }
         newArchetype.addEntity(sourceEntityId, currentComponents);
         this.entityToArchetype.set(sourceEntityId, newArchetype);
-        this.untrackEntityReference(sourceEntityId, componentType, entityId);
+        this.untrackEntityReference(sourceEntityId, componentType, cur);
         this.triggerLifecycleHooks(sourceEntityId, new Map, new Map([[componentType, removedComponent]]));
       }
+      this.entityReferences.delete(cur);
+      archetype.removeEntity(cur);
+      if (archetype.getEntities().length === 0) {
+        this.cleanupEmptyArchetype(archetype);
+      }
+      this.entityToArchetype.delete(cur);
+      this.entityIdManager.deallocate(cur);
     }
-    this.entityReferences.delete(entityId);
-    archetype.removeEntity(entityId);
-    if (archetype.getEntities().length === 0) {
-      this.cleanupEmptyArchetype(archetype);
-    }
-    this.entityToArchetype.delete(entityId);
-    this.entityIdManager.deallocate(entityId);
   }
   exists(entityId) {
     return this.entityToArchetype.has(entityId);
@@ -1066,6 +1082,9 @@ class World {
   setExclusive(componentId) {
     this.exclusiveComponents.add(componentId);
   }
+  setCascadeDelete(componentId) {
+    this.cascadeDeleteComponents.add(componentId);
+  }
   update(...params) {
     const result = this.systemScheduler.update(...params);
     if (result instanceof Promise) {
@@ -1293,14 +1312,14 @@ class World {
   untrackEntityReference(sourceEntityId, componentType, targetEntityId) {
     const references = this.entityReferences.get(targetEntityId);
     if (references) {
-      references.get(sourceEntityId).delete(componentType);
+      references.remove(sourceEntityId, componentType);
       if (references.keyCount === 0) {
         this.entityReferences.delete(targetEntityId);
       }
     }
   }
   getEntityReferences(targetEntityId) {
-    return this.entityReferences.get(targetEntityId) ?? [];
+    return this.entityReferences.get(targetEntityId) ?? new MultiMap;
   }
   cleanupEmptyArchetype(archetype) {
     if (archetype.getEntities().length > 0) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@codehz/ecs",
-  "version": "0.3.4",
+  "version": "0.3.5",
   "type": "module",
   "main": "./index.js",
   "types": "./index.d.ts",

package/world.d.ts

@@ -35,6 +35,8 @@ export declare class World<UpdateParams extends any[] = []> {
     private hooks;
     /** Set of component IDs marked as exclusive relations */
     private exclusiveComponents;
+    /** Set of component IDs that will cascade delete when the relation target is deleted */
+    private cascadeDeleteComponents;
     /**
      * Create a new World.
      * If an optional snapshot object is provided (previously produced by `world.serialize()`),
@@ -108,6 +110,13 @@ export declare class World<UpdateParams extends any[] = []> {
      * For exclusive relations, an entity can have at most one relation per base component
      */
     setExclusive(componentId: EntityId): void;
+    /**
+     * Mark a component as cascade-delete relation
+     * For cascade relations, when the relation target entity is deleted,
+     * the referencing entity will also be deleted (cascade).
+     * Only applicable to entity-relation components
+     */
+    setCascadeDelete(componentId: EntityId): void;
     /**
      * Update the world (run all systems in dependency order)
      * This function is synchronous when all systems are synchronous,