83 lines
3.1 KiB
TypeScript
83 lines
3.1 KiB
TypeScript
import type { IScene, ServiceContainer } from '@esengine/ecs-framework';
|
||
import { ComponentRegistry, Core } from '@esengine/ecs-framework';
|
||
import type { IRuntimeModule, IPlugin, ModuleManifest, SystemContext } from '@esengine/engine-core';
|
||
import type { AssetManager } from '@esengine/asset-system';
|
||
|
||
import { BehaviorTreeRuntimeComponent } from './execution/BehaviorTreeRuntimeComponent';
|
||
import { BehaviorTreeExecutionSystem } from './execution/BehaviorTreeExecutionSystem';
|
||
import { BehaviorTreeAssetManager } from './execution/BehaviorTreeAssetManager';
|
||
import { GlobalBlackboardService } from './Services/GlobalBlackboardService';
|
||
import { BehaviorTreeLoader } from './loaders/BehaviorTreeLoader';
|
||
import { BehaviorTreeAssetType } from './index';
|
||
|
||
export interface BehaviorTreeSystemContext extends SystemContext {
|
||
behaviorTreeSystem?: BehaviorTreeExecutionSystem;
|
||
assetManager?: AssetManager;
|
||
}
|
||
|
||
class BehaviorTreeRuntimeModule implements IRuntimeModule {
|
||
private _loaderRegistered = false;
|
||
|
||
registerComponents(registry: typeof ComponentRegistry): void {
|
||
registry.register(BehaviorTreeRuntimeComponent);
|
||
}
|
||
|
||
registerServices(services: ServiceContainer): void {
|
||
if (!services.isRegistered(GlobalBlackboardService)) {
|
||
services.registerSingleton(GlobalBlackboardService);
|
||
}
|
||
if (!services.isRegistered(BehaviorTreeAssetManager)) {
|
||
services.registerSingleton(BehaviorTreeAssetManager);
|
||
}
|
||
}
|
||
|
||
createSystems(scene: IScene, context: SystemContext): void {
|
||
const btContext = context as BehaviorTreeSystemContext;
|
||
|
||
if (!this._loaderRegistered && btContext.assetManager) {
|
||
btContext.assetManager.registerLoader(BehaviorTreeAssetType, new BehaviorTreeLoader());
|
||
this._loaderRegistered = true;
|
||
}
|
||
|
||
// 使用 context 中的 services，确保与调用方使用同一个 ServiceContainer 实例
|
||
// Use services from context to ensure same ServiceContainer instance as caller
|
||
const services = (btContext as any).services || Core.services;
|
||
const behaviorTreeSystem = new BehaviorTreeExecutionSystem(services);
|
||
|
||
if (btContext.assetManager) {
|
||
behaviorTreeSystem.setAssetManager(btContext.assetManager);
|
||
}
|
||
|
||
if (btContext.isEditor) {
|
||
behaviorTreeSystem.enabled = false;
|
||
}
|
||
|
||
scene.addSystem(behaviorTreeSystem);
|
||
btContext.behaviorTreeSystem = behaviorTreeSystem;
|
||
}
|
||
}
|
||
|
||
const manifest: ModuleManifest = {
|
||
id: 'behavior-tree',
|
||
name: '@esengine/behavior-tree',
|
||
displayName: 'Behavior Tree',
|
||
version: '1.0.0',
|
||
description: 'AI behavior tree system',
|
||
category: 'AI',
|
||
icon: 'GitBranch',
|
||
isCore: false,
|
||
defaultEnabled: false,
|
||
isEngineModule: true,
|
||
canContainContent: true,
|
||
dependencies: ['core'],
|
||
exports: { components: ['BehaviorTreeComponent'] },
|
||
editorPackage: '@esengine/behavior-tree-editor'
|
||
};
|
||
|
||
export const BehaviorTreePlugin: IPlugin = {
|
||
manifest,
|
||
runtimeModule: new BehaviorTreeRuntimeModule()
|
||
};
|
||
|
||
export { BehaviorTreeRuntimeModule };
|