@codehz/ecs 0.0.0 → 0.0.1

package/README.md

@@ -50,6 +50,33 @@ query.forEach([PositionId, VelocityId], (entity, position, velocity) => {
 });
 ```
 
+### 组件生命周期钩子
+
+ECS 支持在组件添加或移除时执行回调函数：
+
+```typescript
+// 注册组件生命周期钩子
+world.registerComponentLifecycleHook(PositionId, {
+  onAdded: (entityId, componentType, component) => {
+    console.log(`组件 ${componentType} 被添加到实体 ${entityId}`);
+  },
+  onRemoved: (entityId, componentType) => {
+    console.log(`组件 ${componentType} 被从实体 ${entityId} 移除`);
+  }
+});
+
+// 你也可以只注册其中一个钩子
+world.registerComponentLifecycleHook(VelocityId, {
+  onRemoved: (entityId, componentType) => {
+    console.log(`组件 ${componentType} 被从实体 ${entityId} 移除`);
+  }
+});
+
+// 添加组件时会触发钩子
+world.addComponent(entity, PositionId, { x: 0, y: 0 });
+world.flushCommands(); // 钩子在这里被调用
+```
+
 ### 运行示例
 
 ```bash
@@ -65,6 +92,8 @@ bun run examples/simple/demo.ts
 - `removeComponent(entity, componentId)`: 从实体移除组件
 - `createQuery(componentIds)`: 创建查询
 - `registerSystem(system)`: 注册系统
+- `registerComponentLifecycleHook(componentId, hook)`: 注册组件生命周期钩子
+- `unregisterComponentLifecycleHook(componentId, hook)`: 注销组件生命周期钩子
 - `update(deltaTime)`: 更新世界
 - `flushCommands()`: 应用命令缓冲区
 

package/index.js

@@ -374,17 +374,26 @@ class CommandBuffer {
     this.commands.push({ type: "destroyEntity", entityId });
   }
   execute() {
-    const entityCommands = new Map;
-    for (const cmd of this.commands) {
-      if (!entityCommands.has(cmd.entityId)) {
-        entityCommands.set(cmd.entityId, []);
+    const MAX_ITERATIONS = 100;
+    let iterations = 0;
+    while (this.commands.length > 0) {
+      if (iterations >= MAX_ITERATIONS) {
+        throw new Error("Command execution exceeded maximum iterations, possible infinite loop");
+      }
+      iterations++;
+      const currentCommands = [...this.commands];
+      this.commands = [];
+      const entityCommands = new Map;
+      for (const cmd of currentCommands) {
+        if (!entityCommands.has(cmd.entityId)) {
+          entityCommands.set(cmd.entityId, []);
+        }
+        entityCommands.get(cmd.entityId).push(cmd);
+      }
+      for (const [entityId, commands] of entityCommands) {
+        this.executeEntityCommands(entityId, commands);
       }
-      entityCommands.get(cmd.entityId).push(cmd);
-    }
-    for (const [entityId, commands] of entityCommands) {
-      this.executeEntityCommands(entityId, commands);
     }
-    this.commands = [];
   }
   getCommands() {
     return [...this.commands];
@@ -512,6 +521,9 @@ class World {
   queries = [];
   commandBuffer;
   componentToArchetypes = new Map;
+  componentLifecycleHooks = new Map;
+  wildcardRelationLifecycleHooks = new Map;
+  entityReverseIndex = new Map;
   constructor() {
     this.commandBuffer = new CommandBuffer((entityId, commands) => this.executeEntityCommands(entityId, commands));
   }
@@ -530,7 +542,36 @@ class World {
     if (!archetype) {
       return;
     }
+    const componentReferences = this.getComponentReferences(entityId);
+    for (const { sourceEntityId, componentType } of componentReferences) {
+      const sourceArchetype = this.entityToArchetype.get(sourceEntityId);
+      if (sourceArchetype) {
+        const currentComponents = new Map;
+        for (const compType of sourceArchetype.componentTypes) {
+          if (compType !== componentType) {
+            const data = sourceArchetype.getComponent(sourceEntityId, compType);
+            if (data !== undefined) {
+              currentComponents.set(compType, data);
+            }
+          }
+        }
+        const newComponentTypes = Array.from(currentComponents.keys()).sort((a, b) => a - b);
+        const newArchetype = this.getOrCreateArchetype(newComponentTypes);
+        sourceArchetype.removeEntity(sourceEntityId);
+        if (sourceArchetype.getEntities().length === 0) {
+          this.removeEmptyArchetype(sourceArchetype);
+        }
+        newArchetype.addEntity(sourceEntityId, currentComponents);
+        this.entityToArchetype.set(sourceEntityId, newArchetype);
+        this.removeComponentReference(sourceEntityId, componentType, entityId);
+        this.executeComponentLifecycleHooks(sourceEntityId, new Map, new Set([componentType]));
+      }
+    }
+    this.entityReverseIndex.delete(entityId);
     archetype.removeEntity(entityId);
+    if (archetype.getEntities().length === 0) {
+      this.removeEmptyArchetype(archetype);
+    }
     this.entityToArchetype.delete(entityId);
     this.entityIdManager.deallocate(entityId);
   }
@@ -569,6 +610,36 @@ class World {
       this.systems.splice(index, 1);
     }
   }
+  registerComponentLifecycleHook(componentType, hook) {
+    if (!this.componentLifecycleHooks.has(componentType)) {
+      this.componentLifecycleHooks.set(componentType, new Set);
+    }
+    this.componentLifecycleHooks.get(componentType).add(hook);
+  }
+  unregisterComponentLifecycleHook(componentType, hook) {
+    const hooks = this.componentLifecycleHooks.get(componentType);
+    if (hooks) {
+      hooks.delete(hook);
+      if (hooks.size === 0) {
+        this.componentLifecycleHooks.delete(componentType);
+      }
+    }
+  }
+  registerWildcardRelationLifecycleHook(baseComponentType, hook) {
+    if (!this.wildcardRelationLifecycleHooks.has(baseComponentType)) {
+      this.wildcardRelationLifecycleHooks.set(baseComponentType, new Set);
+    }
+    this.wildcardRelationLifecycleHooks.get(baseComponentType).add(hook);
+  }
+  unregisterWildcardRelationLifecycleHook(baseComponentType, hook) {
+    const hooks = this.wildcardRelationLifecycleHooks.get(baseComponentType);
+    if (hooks) {
+      hooks.delete(hook);
+      if (hooks.size === 0) {
+        this.wildcardRelationLifecycleHooks.delete(baseComponentType);
+      }
+    }
+  }
   update(...params) {
     for (const system of this.systems) {
       system.update(this, ...params);
@@ -713,6 +784,25 @@ class World {
         currentArchetype.setComponent(entityId, componentType, component);
       }
     }
+    for (const componentType of removes) {
+      const detailedType = getDetailedIdType(componentType);
+      if (detailedType.type === "entity-relation") {
+        const targetEntityId = detailedType.targetId;
+        this.removeComponentReference(entityId, componentType, targetEntityId);
+      } else if (detailedType.type === "entity") {
+        this.removeComponentReference(entityId, componentType, componentType);
+      }
+    }
+    for (const [componentType, component] of adds) {
+      const detailedType = getDetailedIdType(componentType);
+      if (detailedType.type === "entity-relation") {
+        const targetEntityId = detailedType.targetId;
+        this.addComponentReference(entityId, componentType, targetEntityId);
+      } else if (detailedType.type === "entity") {
+        this.addComponentReference(entityId, componentType, componentType);
+      }
+    }
+    this.executeComponentLifecycleHooks(entityId, adds, removes);
   }
   getOrCreateArchetype(componentTypes) {
     const sortedTypes = [...componentTypes].sort((a, b) => a - b);
@@ -731,6 +821,96 @@ class World {
       return newArchetype;
     });
   }
+  addComponentReference(sourceEntityId, componentType, targetEntityId) {
+    if (!this.entityReverseIndex.has(targetEntityId)) {
+      this.entityReverseIndex.set(targetEntityId, new Set);
+    }
+    this.entityReverseIndex.get(targetEntityId).add({ sourceEntityId, componentType });
+  }
+  removeComponentReference(sourceEntityId, componentType, targetEntityId) {
+    const references = this.entityReverseIndex.get(targetEntityId);
+    if (references) {
+      references.forEach((ref) => {
+        if (ref.sourceEntityId === sourceEntityId && ref.componentType === componentType) {
+          references.delete(ref);
+        }
+      });
+      if (references.size === 0) {
+        this.entityReverseIndex.delete(targetEntityId);
+      }
+    }
+  }
+  getComponentReferences(targetEntityId) {
+    const references = this.entityReverseIndex.get(targetEntityId);
+    return references ? Array.from(references) : [];
+  }
+  removeEmptyArchetype(archetype) {
+    if (archetype.getEntities().length > 0) {
+      return;
+    }
+    const index = this.archetypes.indexOf(archetype);
+    if (index !== -1) {
+      this.archetypes.splice(index, 1);
+    }
+    const hashKey = this.getComponentTypesHash(archetype.componentTypes);
+    this.archetypeMap.delete(hashKey);
+    for (const componentType of archetype.componentTypes) {
+      const archetypes = this.componentToArchetypes.get(componentType);
+      if (archetypes) {
+        const compIndex = archetypes.indexOf(archetype);
+        if (compIndex !== -1) {
+          archetypes.splice(compIndex, 1);
+          if (archetypes.length === 0) {
+            this.componentToArchetypes.delete(componentType);
+          }
+        }
+      }
+    }
+  }
+  executeComponentLifecycleHooks(entityId, addedComponents, removedComponents) {
+    for (const [componentType, component] of addedComponents) {
+      const hooks = this.componentLifecycleHooks.get(componentType);
+      if (hooks) {
+        for (const hook of hooks) {
+          if (hook.onAdded) {
+            hook.onAdded(entityId, componentType, component);
+          }
+        }
+      }
+      const detailedType = getDetailedIdType(componentType);
+      if (detailedType.type === "entity-relation" || detailedType.type === "component-relation" || detailedType.type === "wildcard-relation") {
+        const wildcardHooks = this.wildcardRelationLifecycleHooks.get(detailedType.componentId);
+        if (wildcardHooks) {
+          for (const hook of wildcardHooks) {
+            if (hook.onAdded) {
+              hook.onAdded(entityId, componentType, component);
+            }
+          }
+        }
+      }
+    }
+    for (const componentType of removedComponents) {
+      const hooks = this.componentLifecycleHooks.get(componentType);
+      if (hooks) {
+        for (const hook of hooks) {
+          if (hook.onRemoved) {
+            hook.onRemoved(entityId, componentType);
+          }
+        }
+      }
+      const detailedType = getDetailedIdType(componentType);
+      if (detailedType.type === "entity-relation" || detailedType.type === "component-relation" || detailedType.type === "wildcard-relation") {
+        const wildcardHooks = this.wildcardRelationLifecycleHooks.get(detailedType.componentId);
+        if (wildcardHooks) {
+          for (const hook of wildcardHooks) {
+            if (hook.onRemoved) {
+              hook.onRemoved(entityId, componentType);
+            }
+          }
+        }
+      }
+    }
+  }
 }
 export {
   isRelationId,

package/package.json

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

package/world.d.ts

@@ -5,6 +5,33 @@ import { Query } from "./query";
 import type { QueryFilter } from "./query-filter";
 import type { ComponentTuple } from "./types";
 import type { System } from "./system";
+/**
+ * Hook types for component lifecycle events
+ */
+export interface ComponentLifecycleHook<T> {
+    /**
+     * Called when a component is added to an entity
+     */
+    onAdded?: (entityId: EntityId, componentType: EntityId<T>, component: T) => void;
+    /**
+     * Called when a component is removed from an entity
+     */
+    onRemoved?: (entityId: EntityId, componentType: EntityId<T>) => void;
+}
+/**
+ * Hook types for wildcard relation lifecycle events
+ * These hooks are triggered for any component that matches a wildcard relation pattern
+ */
+export interface WildcardRelationLifecycleHook<T = unknown> {
+    /**
+     * Called when any component matching the wildcard relation pattern is added to an entity
+     */
+    onAdded?: (entityId: EntityId, componentType: EntityId<T>, component: T) => void;
+    /**
+     * Called when any component matching the wildcard relation pattern is removed from an entity
+     */
+    onRemoved?: (entityId: EntityId, componentType: EntityId<T>) => void;
+}
 /**
  * World class for ECS architecture
  * Manages entities, components, and systems
@@ -18,6 +45,21 @@ export declare class World<ExtraParams extends any[] = [deltaTime: number]> {
     private queries;
     private commandBuffer;
     private componentToArchetypes;
+    /**
+     * Hook storage for component lifecycle events
+     */
+    private componentLifecycleHooks;
+    /**
+     * Hook storage for wildcard relation lifecycle events
+     * Maps base component type to set of wildcard relation hooks
+     */
+    private wildcardRelationLifecycleHooks;
+    /**
+     * Reverse index tracking which entities use each entity as a component type
+     * Maps entity ID to set of {sourceEntityId, componentType} pairs where componentType uses this entity
+     * This includes both relation components and direct usage of entities as component types
+     */
+    private entityReverseIndex;
     constructor();
     /**
      * Generate a hash key for component types array
@@ -63,6 +105,23 @@ export declare class World<ExtraParams extends any[] = [deltaTime: number]> {
      * Unregister a system
      */
     unregisterSystem(system: System<ExtraParams>): void;
+    /**
+     * Register a lifecycle hook for component events
+     */
+    registerComponentLifecycleHook<T>(componentType: EntityId<T>, hook: ComponentLifecycleHook<T>): void;
+    /**
+     * Unregister a lifecycle hook for component events
+     */
+    unregisterComponentLifecycleHook<T>(componentType: EntityId<T>, hook: ComponentLifecycleHook<T>): void;
+    /**
+     * Register a lifecycle hook for wildcard relation events
+     * The hook will be triggered for any component that matches the wildcard relation pattern
+     */
+    registerWildcardRelationLifecycleHook<T>(baseComponentType: EntityId<T>, hook: WildcardRelationLifecycleHook<T>): void;
+    /**
+     * Unregister a lifecycle hook for wildcard relation events
+     */
+    unregisterWildcardRelationLifecycleHook<T>(baseComponentType: EntityId<T>, hook: WildcardRelationLifecycleHook<T>): void;
     /**
      * Update the world (run all systems)
      */
@@ -103,4 +162,32 @@ export declare class World<ExtraParams extends any[] = [deltaTime: number]> {
      * Get or create an archetype for the given component types
      */
     private getOrCreateArchetype;
+    /**
+     * Add a component reference to the reverse index when an entity is used as a component type
+     * @param sourceEntityId The entity that has the component
+     * @param componentType The component type (which may be an entity ID used as component type)
+     * @param targetEntityId The entity being used as component type
+     */
+    private addComponentReference;
+    /**
+     * Remove a component reference from the reverse index
+     * @param sourceEntityId The entity that has the component
+     * @param componentType The component type
+     * @param targetEntityId The entity being used as component type
+     */
+    private removeComponentReference;
+    /**
+     * Get all component references where a target entity is used as a component type
+     * @param targetEntityId The target entity
+     * @returns Array of {sourceEntityId, componentType} pairs
+     */
+    private getComponentReferences;
+    /**
+     * Remove an empty archetype from all internal data structures
+     */
+    private removeEmptyArchetype;
+    /**
+     * Execute component lifecycle hooks for added and removed components
+     */
+    private executeComponentLifecycleHooks;
 }