61 lines
1.7 KiB
C#
61 lines
1.7 KiB
C#
using System.Collections.Generic;
|
|
using SepCore.Entity;
|
|
|
|
namespace SepCore.EnemyManager
|
|
{
|
|
public class EnemyRegistry
|
|
{
|
|
private readonly Dictionary<int, EntityBase> _enemyById;
|
|
// PruneInvalidEntries 由 OnUpdate 每帧调用，复用缓冲避免每帧分配。
|
|
private readonly List<int> _pruneBuffer;
|
|
|
|
public int Count => _enemyById.Count;
|
|
public IReadOnlyCollection<EntityBase> Enemies => _enemyById.Values;
|
|
|
|
public EnemyRegistry()
|
|
{
|
|
_enemyById = new Dictionary<int, EntityBase>();
|
|
_pruneBuffer = new List<int>();
|
|
}
|
|
|
|
public void Register(EnemyBase enemy)
|
|
{
|
|
_enemyById[enemy.Id] = enemy;
|
|
}
|
|
|
|
public void Remove(int entityId)
|
|
{
|
|
// 移除是幂等的。ClearEnemies 会同步清空 registry，而 HideEntity 的完成回调
|
|
// 是异步的，晚几帧才到达，此时该 id 早已不在集合中，属正常情况，静默忽略即可。
|
|
_enemyById.Remove(entityId);
|
|
}
|
|
|
|
public bool TryGet(int entityId, out EntityBase enemy)
|
|
{
|
|
return _enemyById.TryGetValue(entityId, out enemy);
|
|
}
|
|
|
|
public void PruneInvalidEntries()
|
|
{
|
|
_pruneBuffer.Clear();
|
|
foreach (var kvp in _enemyById)
|
|
{
|
|
if (kvp.Value == null || !kvp.Value.Available)
|
|
{
|
|
_pruneBuffer.Add(kvp.Key);
|
|
}
|
|
}
|
|
|
|
foreach (int id in _pruneBuffer)
|
|
{
|
|
_enemyById.Remove(id);
|
|
}
|
|
}
|
|
|
|
public void Clear()
|
|
{
|
|
_enemyById.Clear();
|
|
}
|
|
}
|
|
}
|