105 lines
4.0 KiB
C#
105 lines
4.0 KiB
C#
using System.Collections.Generic;
|
||
using UnityEngine;
|
||
using BaseGames.Core.Events;
|
||
|
||
namespace BaseGames.Enemies
|
||
{
|
||
/// <summary>
|
||
/// 敌人 BehaviorTree 配额管理器（架构 07_EnemyModule §13）。
|
||
/// 每 REBALANCE_INTERVAL 帧，按到玩家距离排序已注册敌人，
|
||
/// 仅对最近的 _maxActiveBehaviorTrees 个敌人启用 BT。
|
||
/// 挂载在场景管理器 GameObject 上（每场景一个实例）。
|
||
/// </summary>
|
||
public class EnemyQuotaManager : MonoBehaviour
|
||
{
|
||
[SerializeField, Min(1)] private int _maxActiveBehaviorTrees = 12;
|
||
[Tooltip("PlayerController.Start() 广播此频道，替代 FindWithTag")]
|
||
[SerializeField] private TransformEventChannelSO _onPlayerSpawned;
|
||
|
||
private const int REBALANCE_INTERVAL = 10;
|
||
|
||
private int _frameCount;
|
||
// _registeredSet 用于 O(1) 重复检测，_registered List 用于 Sort
|
||
private readonly HashSet<EnemyBase> _registeredSet = new();
|
||
private readonly List<EnemyBase> _registered = new();
|
||
private readonly Dictionary<EnemyBase, int> _indexMap = new();
|
||
// 缓存玩家 Transform
|
||
private Transform _playerTransform;
|
||
private readonly CompositeDisposable _subs = new();
|
||
|
||
private void OnEnable()
|
||
{
|
||
_onPlayerSpawned?.Subscribe(OnPlayerSpawned).AddTo(_subs);
|
||
}
|
||
|
||
private void OnDisable() => _subs.Clear();
|
||
|
||
private void OnPlayerSpawned(Transform playerTransform)
|
||
=> _playerTransform = playerTransform;
|
||
|
||
// ── 注册 / 注销 ───────────────────────────────────────────────────
|
||
public void Register(EnemyBase enemy)
|
||
{
|
||
if (enemy != null && _registeredSet.Add(enemy))
|
||
{
|
||
_indexMap[enemy] = _registered.Count;
|
||
_registered.Add(enemy);
|
||
}
|
||
}
|
||
|
||
public void Unregister(EnemyBase enemy)
|
||
{
|
||
if (enemy == null || !_registeredSet.Remove(enemy)) return;
|
||
int idx = _indexMap[enemy];
|
||
int last = _registered.Count - 1;
|
||
if (idx != last)
|
||
{
|
||
var moved = _registered[last];
|
||
_registered[idx] = moved;
|
||
_indexMap[moved] = idx;
|
||
}
|
||
_registered.RemoveAt(last);
|
||
_indexMap.Remove(enemy);
|
||
}
|
||
|
||
// ── Unity 生命周期 ────────────────────────────────────────────────
|
||
private void Update()
|
||
{
|
||
if (++_frameCount % REBALANCE_INTERVAL == 0)
|
||
Rebalance();
|
||
}
|
||
|
||
// ── 内部 ──────────────────────────────────────────────────────────
|
||
private void Rebalance()
|
||
{
|
||
if (_registered.Count == 0) return;
|
||
|
||
var playerPos = _playerTransform != null ? _playerTransform.position : Vector3.zero;
|
||
|
||
// 按距离平方升序排序（避免开方，性能更好）
|
||
_registered.Sort((a, b) =>
|
||
{
|
||
if (a == null) return 1;
|
||
if (b == null) return -1;
|
||
float sqA = (a.transform.position - playerPos).sqrMagnitude;
|
||
float sqB = (b.transform.position - playerPos).sqrMagnitude;
|
||
return sqA.CompareTo(sqB);
|
||
});
|
||
|
||
#if GRAPH_DESIGNER
|
||
for (int i = _registered.Count - 1; i >= 0; i--)
|
||
{
|
||
var enemy = _registered[i];
|
||
if (enemy == null) { _registered.RemoveAt(i); continue; }
|
||
|
||
var bt = enemy.BehaviorTree;
|
||
bool active = i < _maxActiveBehaviorTrees;
|
||
|
||
if (bt != null && bt.enabled != active)
|
||
bt.enabled = active;
|
||
}
|
||
#endif
|
||
}
|
||
}
|
||
}
|