317 lines
11 KiB
TypeScript
317 lines
11 KiB
TypeScript
|
import { EntitySystem, Entity, Matcher, Time } from '@esengine/ecs-framework';
|
|||
|
|
import { AIComponent, Transform, Health } from '../components';
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* AI系统 - 处理AI行为和状态机
|
|||
|
|
*/
|
|||
|
|
export class AISystem extends EntitySystem {
|
|||
|
|
|
|||
|
|
/** 系统处理的实体计数器 */
|
|||
|
|
private processedEntityCount: number = 0;
|
|||
|
|
|
|||
|
|
/** 状态转换计数器 */
|
|||
|
|
private stateTransitionCount: number = 0;
|
|||
|
|
|
|||
|
|
constructor() {
|
|||
|
|
// 处理具有AI组件的实体
|
|||
|
|
super(Matcher.empty().all(AIComponent));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 处理所有实体
|
|||
|
|
*/
|
|||
|
|
protected process(entities: Entity[]): void {
|
|||
|
|
const deltaTime = Time.deltaTime;
|
|||
|
|
const currentTime = Time.totalTime;
|
|||
|
|
|
|||
|
|
this.processedEntityCount = entities.length;
|
|||
|
|
|
|||
|
|
for (const entity of entities) {
|
|||
|
|
this.processEntity(entity, deltaTime, currentTime);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 批量处理AI间的交互
|
|||
|
|
this.processAIInteractions(entities);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 处理单个实体
|
|||
|
|
*/
|
|||
|
|
private processEntity(entity: Entity, deltaTime: number, currentTime: number): void {
|
|||
|
|
const ai = entity.getComponent(AIComponent);
|
|||
|
|
const transform = entity.getComponent(Transform);
|
|||
|
|
const health = entity.getComponent(Health);
|
|||
|
|
|
|||
|
|
if (!ai) return;
|
|||
|
|
|
|||
|
|
// 更新感知系统
|
|||
|
|
ai.updatePerception(deltaTime);
|
|||
|
|
|
|||
|
|
// 处理状态机
|
|||
|
|
this.updateStateMachine(ai, deltaTime);
|
|||
|
|
|
|||
|
|
// 更新行为树
|
|||
|
|
this.updateBehaviorTree(ai, deltaTime);
|
|||
|
|
|
|||
|
|
// 处理AI能力（如果有Transform和Health组件）
|
|||
|
|
if (transform && health) {
|
|||
|
|
this.updateAICapabilities(ai, transform, health, deltaTime);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 处理记忆衰减
|
|||
|
|
this.updateMemory(ai, deltaTime);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 更新状态机
|
|||
|
|
*/
|
|||
|
|
private updateStateMachine(ai: AIComponent, deltaTime: number): void {
|
|||
|
|
const currentStateName = ai.currentState;
|
|||
|
|
const transitions = ai.stateMachine.transitions.get(currentStateName);
|
|||
|
|
|
|||
|
|
if (transitions) {
|
|||
|
|
// 按优先级排序转换条件
|
|||
|
|
const sortedTransitions = transitions.sort((a, b) => b.priority - a.priority);
|
|||
|
|
|
|||
|
|
for (const transition of sortedTransitions) {
|
|||
|
|
if (transition.condition()) {
|
|||
|
|
// 执行状态转换
|
|||
|
|
const currentState = ai.stateMachine.states.get(currentStateName);
|
|||
|
|
const newState = ai.stateMachine.states.get(transition.targetState);
|
|||
|
|
|
|||
|
|
if (currentState && newState) {
|
|||
|
|
currentState.exit();
|
|||
|
|
ai.currentState = transition.targetState as any;
|
|||
|
|
newState.enter();
|
|||
|
|
this.stateTransitionCount++;
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 更新行为树
|
|||
|
|
*/
|
|||
|
|
private updateBehaviorTree(ai: AIComponent, deltaTime: number): void {
|
|||
|
|
const behaviorTree = ai.config.behaviorTree;
|
|||
|
|
const blackboard = behaviorTree.blackboard;
|
|||
|
|
|
|||
|
|
// 更新黑板数据
|
|||
|
|
blackboard.set('deltaTime', deltaTime);
|
|||
|
|
blackboard.set('currentTime', Date.now());
|
|||
|
|
|
|||
|
|
// 模拟行为树执行
|
|||
|
|
const executionResult = this.executeBehaviorNode(behaviorTree.rootNode, ai);
|
|||
|
|
|
|||
|
|
// 记录执行历史
|
|||
|
|
behaviorTree.executionHistory.push({
|
|||
|
|
nodeName: behaviorTree.rootNode.name,
|
|||
|
|
startTime: Date.now(),
|
|||
|
|
endTime: Date.now() + Math.random() * 10,
|
|||
|
|
result: executionResult,
|
|||
|
|
data: { deltaTime }
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// 保持历史记录在合理范围内
|
|||
|
|
if (behaviorTree.executionHistory.length > 50) {
|
|||
|
|
behaviorTree.executionHistory.shift();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 执行行为树节点（模拟）
|
|||
|
|
*/
|
|||
|
|
private executeBehaviorNode(node: any, ai: AIComponent): 'success' | 'failure' | 'running' {
|
|||
|
|
// 简单的行为树执行模拟
|
|||
|
|
switch (node.name) {
|
|||
|
|
case 'root':
|
|||
|
|
case 'selector':
|
|||
|
|
// 选择器节点：尝试执行子节点直到一个成功
|
|||
|
|
for (const child of node.children) {
|
|||
|
|
const result = this.executeBehaviorNode(child, ai);
|
|||
|
|
if (result === 'success' || result === 'running') {
|
|||
|
|
return result;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return 'failure';
|
|||
|
|
|
|||
|
|
case 'sequence':
|
|||
|
|
// 序列节点：按顺序执行所有子节点
|
|||
|
|
for (const child of node.children) {
|
|||
|
|
const result = this.executeBehaviorNode(child, ai);
|
|||
|
|
if (result === 'failure' || result === 'running') {
|
|||
|
|
return result;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return 'success';
|
|||
|
|
|
|||
|
|
case 'condition':
|
|||
|
|
// 条件节点：检查AI状态
|
|||
|
|
return ai.config.personality.intelligence > 0.5 ? 'success' : 'failure';
|
|||
|
|
|
|||
|
|
case 'action':
|
|||
|
|
// 动作节点：执行AI行为
|
|||
|
|
return Math.random() > 0.3 ? 'success' : 'running';
|
|||
|
|
|
|||
|
|
default:
|
|||
|
|
return 'failure';
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 更新AI能力
|
|||
|
|
*/
|
|||
|
|
private updateAICapabilities(ai: AIComponent, transform: Transform, health: Health, deltaTime: number): void {
|
|||
|
|
const capabilities = ai.config.capabilities;
|
|||
|
|
|
|||
|
|
// 根据健康状况调整能力
|
|||
|
|
const healthRatio = health.currentHealth / health.maxHealth;
|
|||
|
|
const effectiveSpeed = capabilities.movementSpeed * healthRatio;
|
|||
|
|
|
|||
|
|
// 更新移动速度
|
|||
|
|
transform.speed = effectiveSpeed;
|
|||
|
|
|
|||
|
|
// 根据个性调整行为
|
|||
|
|
if (ai.config.personality.aggression > 0.7 && healthRatio > 0.5) {
|
|||
|
|
// 高攻击性且健康状况良好时更主动
|
|||
|
|
ai.currentState = 'chase';
|
|||
|
|
} else if (healthRatio < 0.3) {
|
|||
|
|
// 生命值低时逃跑
|
|||
|
|
ai.currentState = 'flee';
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 更新记忆系统
|
|||
|
|
*/
|
|||
|
|
private updateMemory(ai: AIComponent, deltaTime: number): void {
|
|||
|
|
const memory = ai.config.memory;
|
|||
|
|
const currentTime = Date.now();
|
|||
|
|
|
|||
|
|
// 衰减已知位置的可信度
|
|||
|
|
memory.knownLocations.forEach(location => {
|
|||
|
|
const timeSinceVisit = currentTime - location.lastVisited;
|
|||
|
|
const decayFactor = Math.exp(-timeSinceVisit / 30000); // 30秒衰减率
|
|||
|
|
location.confidence *= decayFactor;
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// 移除可信度过低的位置
|
|||
|
|
memory.knownLocations = memory.knownLocations.filter(location => location.confidence > 0.1);
|
|||
|
|
|
|||
|
|
// 衰减关系信任度
|
|||
|
|
memory.relationships.forEach(relation => {
|
|||
|
|
const timeSinceInteraction = currentTime - relation.lastInteraction;
|
|||
|
|
if (timeSinceInteraction > 60000) { // 60秒没有交互
|
|||
|
|
relation.trustLevel *= 0.99; // 缓慢衰减
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 处理AI间的交互
|
|||
|
|
*/
|
|||
|
|
private processAIInteractions(entities: Entity[]): void {
|
|||
|
|
const aiEntities = entities.filter(e => e.getComponent(AIComponent));
|
|||
|
|
|
|||
|
|
for (let i = 0; i < aiEntities.length; i++) {
|
|||
|
|
for (let j = i + 1; j < aiEntities.length; j++) {
|
|||
|
|
this.processAIPair(aiEntities[i], aiEntities[j]);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 处理两个AI实体间的交互
|
|||
|
|
*/
|
|||
|
|
private processAIPair(entity1: Entity, entity2: Entity): void {
|
|||
|
|
const ai1 = entity1.getComponent(AIComponent);
|
|||
|
|
const ai2 = entity2.getComponent(AIComponent);
|
|||
|
|
const transform1 = entity1.getComponent(Transform);
|
|||
|
|
const transform2 = entity2.getComponent(Transform);
|
|||
|
|
|
|||
|
|
if (!ai1 || !ai2 || !transform1 || !transform2) return;
|
|||
|
|
|
|||
|
|
// 计算距离
|
|||
|
|
const distance = Math.sqrt(
|
|||
|
|
Math.pow(transform1.position.x - transform2.position.x, 2) +
|
|||
|
|
Math.pow(transform1.position.y - transform2.position.y, 2)
|
|||
|
|
);
|
|||
|
|
|
|||
|
|
// 视线范围内的交互
|
|||
|
|
if (distance <= ai1.config.capabilities.sightRange) {
|
|||
|
|
this.handleVisualContact(ai1, ai2, entity1, entity2, distance);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 听力范围内的交互
|
|||
|
|
if (distance <= ai1.config.capabilities.hearingRange) {
|
|||
|
|
this.handleAudioContact(ai1, ai2, distance);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 创建盟友关系（随机）
|
|||
|
|
if (Math.random() < 0.001 && !ai1.allyIds.includes(entity2.id)) { // 0.1%概率每帧
|
|||
|
|
ai1.addAlly(entity2.id);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 处理视觉接触
|
|||
|
|
*/
|
|||
|
|
private handleVisualContact(ai1: AIComponent, ai2: AIComponent, entity1: Entity, entity2: Entity, distance: number): void {
|
|||
|
|
const currentTime = Date.now();
|
|||
|
|
|
|||
|
|
// 添加到可见实体列表
|
|||
|
|
const existingEntry = ai1.perception.visibleEntities.find(e => e.entityId === entity2.id);
|
|||
|
|
if (existingEntry) {
|
|||
|
|
existingEntry.distance = distance;
|
|||
|
|
existingEntry.lastSeen = currentTime;
|
|||
|
|
existingEntry.componentId = ai2.id; // 使用组件ID避免循环引用
|
|||
|
|
} else {
|
|||
|
|
ai1.perception.visibleEntities.push({
|
|||
|
|
entityId: entity2.id,
|
|||
|
|
position: entity2.getComponent(Transform)!.position.clone(),
|
|||
|
|
distance: distance,
|
|||
|
|
angle: Math.atan2(
|
|||
|
|
entity2.getComponent(Transform)!.position.y - entity1.getComponent(Transform)!.position.y,
|
|||
|
|
entity2.getComponent(Transform)!.position.x - entity1.getComponent(Transform)!.position.x
|
|||
|
|
),
|
|||
|
|
lastSeen: currentTime,
|
|||
|
|
componentId: ai2.id
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 处理音频接触
|
|||
|
|
*/
|
|||
|
|
private handleAudioContact(ai1: AIComponent, ai2: AIComponent, distance: number): void {
|
|||
|
|
const soundVolume = 1.0 - (distance / ai1.config.capabilities.hearingRange);
|
|||
|
|
|
|||
|
|
ai1.perception.audibleSounds.push({
|
|||
|
|
source: ai2.entity.getComponent(Transform)!.position.clone(),
|
|||
|
|
volume: soundVolume,
|
|||
|
|
type: 'movement',
|
|||
|
|
timestamp: Date.now()
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 系统初始化时调用
|
|||
|
|
*/
|
|||
|
|
public initialize(): void {
|
|||
|
|
super.initialize();
|
|||
|
|
console.log('🤖 AI系统已启动');
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取系统统计信息
|
|||
|
|
*/
|
|||
|
|
public getSystemStats(): any {
|
|||
|
|
return {
|
|||
|
|
processedEntities: this.processedEntityCount,
|
|||
|
|
stateTransitions: this.stateTransitionCount,
|
|||
|
|
systemName: 'AISystem'
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
}
|