archetype-ecs 1.2.0 → 1.4.0

package/src/{Profiler.js → Profiler.ts}

@@ -1,11 +1,24 @@
+export interface ProfilerEntry {
+  avg: number;
+}
+
+export interface Profiler {
+  readonly enabled: boolean;
+  setEnabled(value: boolean): void;
+  begin(): number;
+  end(name: string, t0: number): void;
+  record(name: string, ms: number): void;
+  getData(): Map<string, ProfilerEntry>;
+}
+
 const EMA_ALPHA = 0.1;
-const data = new Map();
+const data = new Map<string, ProfilerEntry>();
 let enabled = false;
 
-export const profiler = {
+export const profiler: Profiler = {
   get enabled() { return enabled; },
 
-  setEnabled(value) {
+  setEnabled(value: boolean) {
     enabled = value;
     if (!value) data.clear();
   },
@@ -14,7 +27,7 @@ export const profiler = {
     return enabled ? performance.now() : 0;
   },
 
-  end(name, t0) {
+  end(name: string, t0: number) {
     if (!enabled) return;
     const ms = performance.now() - t0;
     const entry = data.get(name);
@@ -25,7 +38,7 @@ export const profiler = {
     }
   },
 
-  record(name, ms) {
+  record(name: string, ms: number) {
     if (!enabled) return;
     const entry = data.get(name);
     if (entry) {

package/src/System.ts

@@ -0,0 +1,201 @@
+import type { ComponentDef } from './ComponentRegistry.js';
+import type { EntityId, EntityManager, ArchetypeView } from './EntityManager.js';
+
+// ── Decorators (TC39 Stage 3) ────────────────────────────
+
+export function OnAdded(...types: ComponentDef[]) {
+  return function (_method: Function, context: ClassMethodDecoratorContext) {
+    context.addInitializer(function () {
+      const self = this as unknown as System;
+      self._registerHook('add', types, (self as any)[context.name].bind(self));
+    });
+  };
+}
+
+export function OnRemoved(...types: ComponentDef[]) {
+  return function (_method: Function, context: ClassMethodDecoratorContext) {
+    context.addInitializer(function () {
+      const self = this as unknown as System;
+      self._registerHook('remove', types, (self as any)[context.name].bind(self));
+    });
+  };
+}
+
+// ── Base class ───────────────────────────────────────────
+
+interface Hook {
+  buffer: Set<EntityId>;
+  handler: (id: EntityId) => void;
+}
+
+export class System {
+  em: EntityManager;
+  _unsubs: (() => void)[] = [];
+  _hooks: Hook[] = [];
+
+  constructor(em: EntityManager) {
+    this.em = em;
+  }
+
+  _registerHook(kind: 'add' | 'remove', types: ComponentDef[], handler: (id: EntityId) => void): void {
+    const buffer = new Set<EntityId>();
+
+    if (kind === 'add') {
+      for (const comp of types) {
+        const unsub = this.em.onAdd(comp, (id: EntityId) => {
+          for (let i = 0; i < types.length; i++) {
+            if (!this.em.hasComponent(id, types[i])) return;
+          }
+          buffer.add(id);
+        });
+        this._unsubs.push(unsub);
+      }
+    } else {
+      for (const comp of types) {
+        const unsub = this.em.onRemove(comp, (id: EntityId) => buffer.add(id));
+        this._unsubs.push(unsub);
+      }
+    }
+
+    this._hooks.push({ buffer, handler });
+  }
+
+  forEach(types: ComponentDef[], callback: (view: ArchetypeView) => void, exclude?: ComponentDef[]): void {
+    this.em.forEach(types, callback, exclude);
+  }
+
+  tick?(): void;
+
+  run(): void {
+    for (const hook of this._hooks) {
+      for (const id of hook.buffer) hook.handler(id);
+      hook.buffer.clear();
+    }
+    if (this.tick) this.tick();
+  }
+
+  dispose(): void {
+    for (const unsub of this._unsubs) unsub();
+    this._unsubs.length = 0;
+    this._hooks.length = 0;
+  }
+}
+
+// ── Functional API ───────────────────────────────────────
+
+type HookCallback = (entityId: EntityId) => void;
+
+export interface SystemContext {
+  onAdded(...args: [...ComponentDef[], HookCallback]): void;
+  onRemoved(...args: [...ComponentDef[], HookCallback]): void;
+  forEach(types: ComponentDef[], callback: (view: ArchetypeView) => void, exclude?: ComponentDef[]): void;
+}
+
+export type FunctionalSystemConstructor = (sys: SystemContext) => (() => void) | void;
+
+interface FunctionalHook {
+  unsubs: (() => void)[];
+  buffer: Set<EntityId>;
+  callback: HookCallback;
+}
+
+export interface FunctionalSystem {
+  (): void;
+  dispose(): void;
+}
+
+export function createSystem(em: EntityManager, constructor: FunctionalSystemConstructor): FunctionalSystem {
+  const hooks: FunctionalHook[] = [];
+
+  const sys: SystemContext = {
+    onAdded(...args: any[]) {
+      const callback = args[args.length - 1] as HookCallback;
+      const types = args.slice(0, -1) as ComponentDef[];
+      const buffer = new Set<EntityId>();
+      const unsubs: (() => void)[] = [];
+
+      for (const comp of types) {
+        const unsub = em.onAdd(comp, (id: EntityId) => {
+          for (let i = 0; i < types.length; i++) {
+            if (!em.hasComponent(id, types[i])) return;
+          }
+          buffer.add(id);
+        });
+        unsubs.push(unsub);
+      }
+
+      hooks.push({ unsubs, buffer, callback });
+    },
+
+    onRemoved(...args: any[]) {
+      const callback = args[args.length - 1] as HookCallback;
+      const types = args.slice(0, -1) as ComponentDef[];
+      const buffer = new Set<EntityId>();
+      const unsubs: (() => void)[] = [];
+
+      for (const comp of types) {
+        const unsub = em.onRemove(comp, (id: EntityId) => {
+          buffer.add(id);
+        });
+        unsubs.push(unsub);
+      }
+
+      hooks.push({ unsubs, buffer, callback });
+    },
+
+    forEach(types: ComponentDef[], callback: (view: ArchetypeView) => void, exclude?: ComponentDef[]) {
+      em.forEach(types, callback, exclude);
+    },
+  };
+
+  const tick = constructor(sys);
+
+  function system() {
+    for (const hook of hooks) {
+      for (const id of hook.buffer) hook.callback(id);
+      hook.buffer.clear();
+    }
+    if (tick) tick();
+  }
+
+  system.dispose = function () {
+    for (const hook of hooks) {
+      for (const unsub of hook.unsubs) unsub();
+    }
+    hooks.length = 0;
+  };
+
+  return system;
+}
+
+// ── Activator ────────────────────────────────────────────
+
+interface Runnable {
+  run(): void;
+  dispose(): void;
+}
+
+export interface Pipeline {
+  (): void;
+  dispose(): void;
+}
+
+export function createSystems(em: EntityManager, entries: (FunctionalSystemConstructor | (new (em: EntityManager) => System))[]): Pipeline {
+  const systems: Runnable[] = entries.map(Entry => {
+    if ((Entry as any).prototype instanceof System) {
+      return new (Entry as new (em: EntityManager) => System)(em);
+    }
+    const sys = createSystem(em, Entry as FunctionalSystemConstructor);
+    return { run: sys, dispose: sys.dispose };
+  });
+
+  function pipeline() {
+    for (let i = 0; i < systems.length; i++) systems[i].run();
+  }
+
+  pipeline.dispose = function () {
+    for (let i = 0; i < systems.length; i++) systems[i].dispose();
+  };
+
+  return pipeline;
+}

package/src/index.ts

@@ -0,0 +1,38 @@
+export { createEntityManager } from './EntityManager.js';
+export type { EntityId, FieldRef, ArchetypeView, EntityManager, SerializedData } from './EntityManager.js';
+export { createSystem, createSystems, System, OnAdded, OnRemoved } from './System.js';
+export type { SystemContext, FunctionalSystemConstructor, FunctionalSystem, Pipeline } from './System.js';
+export { profiler } from './Profiler.js';
+export type { Profiler, ProfilerEntry } from './Profiler.js';
+export { TYPED, componentSchemas, parseTypeSpec } from './ComponentRegistry.js';
+export type { ComponentDef, TypeSpec } from './ComponentRegistry.js';
+
+import { parseTypeSpec, componentSchemas, type ComponentDef, type TypeSpec } from './ComponentRegistry.js';
+
+export function component(name: string, typeOrSchema?: string | Record<string, string>, fields?: string[]): ComponentDef {
+  const sym = Symbol(name);
+  const comp: any = { _sym: sym, _name: name };
+
+  let schema: Record<string, TypeSpec> | undefined;
+
+  if (typeof typeOrSchema === 'string' && Array.isArray(fields)) {
+    const spec = parseTypeSpec(typeOrSchema);
+    schema = {};
+    for (const f of fields) {
+      schema[f] = spec;
+      comp[f] = { _sym: sym, _field: f };
+    }
+  } else if (typeOrSchema && typeof typeOrSchema === 'object') {
+    schema = {};
+    for (const [field, type] of Object.entries(typeOrSchema)) {
+      schema[field] = parseTypeSpec(type);
+      comp[field] = { _sym: sym, _field: field };
+    }
+  }
+
+  if (schema) {
+    componentSchemas.set(sym, schema);
+  }
+
+  return comp;
+}