mirror of
https://github.com/DCFApixels/DragonECS.git
synced 2025-09-18 01:44:35 +08:00
574 lines
19 KiB
C#
574 lines
19 KiB
C#
using DCFApixels.DragonECS.Internal;
|
||
using System;
|
||
using System.Collections.Generic;
|
||
using System.Runtime.CompilerServices;
|
||
|
||
namespace DCFApixels.DragonECS
|
||
{
|
||
public abstract partial class EcsWorld : IEntityStorage
|
||
{
|
||
public readonly short id;
|
||
private IEcsWorldConfig _config;
|
||
|
||
private bool _isDestroyed = false;
|
||
|
||
private IdDispenser _entityDispenser;
|
||
private int _entitiesCount = 0;
|
||
private int _entitiesCapacity = 0;
|
||
private short[] _gens = Array.Empty<short>(); //старший бит указывает на то жива ли сущность
|
||
private short[] _componentCounts = Array.Empty<short>();
|
||
|
||
private int[] _delEntBuffer = Array.Empty<int>();
|
||
private int _delEntBufferCount = 0;
|
||
private bool _isEnableAutoReleaseDelEntBuffer = true;
|
||
|
||
internal int _entityComponentMaskLength;
|
||
internal int[] _entityComponentMasks = Array.Empty<int>();
|
||
private const int COMPONENT_MATRIX_MASK_BITSIZE = 32;
|
||
|
||
private long _version = 0;
|
||
|
||
private List<WeakReference<EcsGroup>> _groups = new List<WeakReference<EcsGroup>>();
|
||
private Stack<EcsGroup> _groupsPool = new Stack<EcsGroup>(64);
|
||
|
||
private List<IEcsWorldEventListener> _listeners = new List<IEcsWorldEventListener>();
|
||
private List<IEcsEntityEventListener> _entityListeners = new List<IEcsEntityEventListener>();
|
||
|
||
#region Properties
|
||
public IEcsWorldConfig Config
|
||
{
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
get { return _config; }
|
||
}
|
||
public long Version
|
||
{
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
get { return _version; }
|
||
}
|
||
public bool IsDestroyed
|
||
{
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
get { return _isDestroyed; }
|
||
}
|
||
public int Count
|
||
{
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
get { return _entitiesCount; }
|
||
}
|
||
public int Capacity
|
||
{
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
get { return _entitiesCapacity; }
|
||
}
|
||
public int DelEntBufferCount
|
||
{
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
get { return _delEntBufferCount; }
|
||
}
|
||
public bool IsEnableReleaseDelEntBuffer
|
||
{
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
get { return _isEnableAutoReleaseDelEntBuffer; }
|
||
}
|
||
|
||
public EcsSpan Entities
|
||
{
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
get
|
||
{
|
||
if (_isEnableAutoReleaseDelEntBuffer)
|
||
{
|
||
ReleaseDelEntityBufferAll();
|
||
}
|
||
return _entityDispenser.UsedToEcsSpan(id);
|
||
}
|
||
}
|
||
public ReadOnlySpan<IEcsPoolImplementation> AllPools
|
||
{
|
||
// new ReadOnlySpan<IEcsPoolImplementation>(pools, 0, _poolsCount);
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
get { return _pools; }
|
||
}
|
||
#endregion
|
||
|
||
#region Constructors/Destroy
|
||
public EcsWorld(IEcsWorldConfig config = null, short worldID = -1)
|
||
{
|
||
if (config == null)
|
||
{
|
||
config = EcsWorldConfig.Empty;
|
||
}
|
||
_config = config;
|
||
|
||
if (worldID < 0)
|
||
{
|
||
worldID = (short)_worldIdDispenser.UseFree();
|
||
}
|
||
else
|
||
{
|
||
if (worldID != _worldIdDispenser.NullID)
|
||
{
|
||
_worldIdDispenser.Use(worldID);
|
||
}
|
||
if (_worlds[worldID] != null)
|
||
{
|
||
_worldIdDispenser.Release(worldID);
|
||
Throw.UndefinedException();
|
||
}
|
||
}
|
||
id = worldID;
|
||
_worlds[worldID] = this;
|
||
|
||
_poolsMediator = new PoolsMediator(this);
|
||
|
||
int poolsCapacity = ArrayUtility.NormalizeSizeToPowerOfTwo(config.Get_PoolsCapacity());
|
||
_pools = new IEcsPoolImplementation[poolsCapacity];
|
||
_poolComponentCounts = new int[poolsCapacity];
|
||
ArrayUtility.Fill(_pools, _nullPool);
|
||
|
||
int entitiesCapacity = ArrayUtility.NormalizeSizeToPowerOfTwo(config.Get_EntitiesCapacity());
|
||
_entityDispenser = new IdDispenser(entitiesCapacity, 0, OnEntityDispenserResized);
|
||
}
|
||
public void Destroy()
|
||
{
|
||
_entityDispenser = null;
|
||
_gens = null;
|
||
_pools = null;
|
||
_nullPool = null;
|
||
_worlds[id] = null;
|
||
ReleaseData(id);
|
||
_worldIdDispenser.Release(id);
|
||
_isDestroyed = true;
|
||
_poolTypeCode_2_CmpTypeIDs = null;
|
||
_componentTypeCode_2_CmpTypeIDs = null;
|
||
}
|
||
//public void Clear() { }
|
||
#endregion
|
||
|
||
#region Getters
|
||
#if UNITY_2020_3_OR_NEWER
|
||
[UnityEngine.Scripting.Preserve]
|
||
#endif
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public TAspect GetAspect<TAspect>() where TAspect : EcsAspect
|
||
{
|
||
return Get<AspectCache<TAspect>>().instance;
|
||
}
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public TExecutor GetExecutor<TExecutor>() where TExecutor : EcsQueryExecutor, new()
|
||
{
|
||
return Get<ExcecutorCache<TExecutor>>().instance;
|
||
}
|
||
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public ref T Get<T>() where T : struct
|
||
{
|
||
return ref WorldComponentPool<T>.GetForWorld(id);
|
||
}
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public ref T GetUnchecked<T>() where T : struct
|
||
{
|
||
return ref WorldComponentPool<T>.GetForWorldUnchecked(id);
|
||
}
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public static ref T Get<T>(int worldID) where T : struct
|
||
{
|
||
return ref WorldComponentPool<T>.GetForWorld(worldID);
|
||
}
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public static ref T GetUnchecked<T>(int worldID) where T : struct
|
||
{
|
||
return ref WorldComponentPool<T>.GetForWorldUnchecked(worldID);
|
||
}
|
||
#endregion
|
||
|
||
#region Entity
|
||
|
||
#region New/Del
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public int NewEntity()
|
||
{
|
||
int entityID = _entityDispenser.UseFree();
|
||
CreateConcreteEntity(entityID);
|
||
return entityID;
|
||
}
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public void NewEntity(int entityID)
|
||
{
|
||
#if (DEBUG && !DISABLE_DEBUG) || !DISABLE_DRAGONECS_ASSERT_CHEKS
|
||
if (IsUsed(entityID)) { Throw.World_EntityIsAlreadyСontained(entityID); }
|
||
#endif
|
||
_entityDispenser.Use(entityID);
|
||
CreateConcreteEntity(entityID);
|
||
}
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
private void CreateConcreteEntity(int entityID)
|
||
{
|
||
unchecked { _version++; }
|
||
_entitiesCount++;
|
||
if (_entitiesCapacity <= entityID)
|
||
{
|
||
OnEntityDispenserResized(_gens.Length << 1);
|
||
}
|
||
_gens[entityID] &= GEN_MASK;
|
||
_entityListeners.InvokeOnNewEntity(entityID);
|
||
}
|
||
|
||
public entlong NewEntityLong()
|
||
{
|
||
int e = NewEntity();
|
||
return GetEntityLong(e);
|
||
}
|
||
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public void TryDelEntity(int entityID)
|
||
{
|
||
if (IsUsed(entityID))
|
||
{
|
||
DelEntity(entityID);
|
||
}
|
||
}
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public void DelEntity(int entityID)
|
||
{
|
||
#if (DEBUG && !DISABLE_DEBUG) || !DISABLE_DRAGONECS_ASSERT_CHEKS
|
||
if (IsUsed(entityID) == false)
|
||
{
|
||
Throw.World_EntityIsNotContained(entityID);
|
||
}
|
||
#endif
|
||
_delEntBuffer[_delEntBufferCount++] = entityID;
|
||
_gens[entityID] |= DEATH_GEN_BIT;
|
||
_entitiesCount--;
|
||
_entityListeners.InvokeOnDelEntity(entityID);
|
||
}
|
||
#endregion
|
||
|
||
#region Other
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public EcsSpan ToSpan()
|
||
{
|
||
return _entityDispenser.UsedToEcsSpan(id);
|
||
}
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public unsafe entlong GetEntityLong(int entityID)
|
||
{
|
||
long x = (long)id << 48 | (long)_gens[entityID] << 32 | (long)entityID;
|
||
return *(entlong*)&x;
|
||
}
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public bool IsAlive(int entityID, short gen)
|
||
{
|
||
return _gens[entityID] == gen;
|
||
}
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public bool IsUsed(int entityID)
|
||
{
|
||
return _gens[entityID] >= 0;
|
||
}
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public short GetGen(int entityID)
|
||
{
|
||
return _gens[entityID];
|
||
}
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public short GetComponentsCount(int entityID)
|
||
{
|
||
return _componentCounts[entityID];
|
||
}
|
||
|
||
public bool IsMatchesMask(EcsMask mask, int entityID)
|
||
{
|
||
#if (DEBUG && !DISABLE_DEBUG) || !DISABLE_DRAGONECS_ASSERT_CHEKS
|
||
if (mask.worldID != id) { Throw.World_MaskDoesntBelongWorld(); }
|
||
#endif
|
||
for (int i = 0, iMax = mask.incChunckMasks.Length; i < iMax; i++)
|
||
{
|
||
if (!_pools[mask.inc[i]].Has(entityID))
|
||
return false;
|
||
}
|
||
for (int i = 0, iMax = mask.excChunckMasks.Length; i < iMax; i++)
|
||
{
|
||
if (_pools[mask.exc[i]].Has(entityID))
|
||
return false;
|
||
}
|
||
return true;
|
||
}
|
||
|
||
public void DeleteEmptyEntites()
|
||
{
|
||
foreach (var e in Entities)
|
||
{
|
||
if (_componentCounts[e] <= 0)
|
||
{
|
||
DelEntity(e);
|
||
}
|
||
}
|
||
}
|
||
#endregion
|
||
|
||
#region Copy/Clone
|
||
public void CopyEntity(int fromEntityID, int toEntityID)
|
||
{
|
||
foreach (var pool in _pools)
|
||
{
|
||
if (pool.Has(fromEntityID))
|
||
pool.Copy(fromEntityID, toEntityID);
|
||
}
|
||
}
|
||
public void CopyEntity(int fromEntityID, EcsWorld toWorld, int toEntityID)
|
||
{
|
||
foreach (var pool in _pools)
|
||
{
|
||
if (pool.Has(fromEntityID))
|
||
pool.Copy(fromEntityID, toWorld, toEntityID);
|
||
}
|
||
}
|
||
public int CloneEntity(int fromEntityID)
|
||
{
|
||
int newEntity = NewEntity();
|
||
CopyEntity(fromEntityID, newEntity);
|
||
return newEntity;
|
||
}
|
||
public int CloneEntity(int fromEntityID, EcsWorld toWorld)
|
||
{
|
||
int newEntity = NewEntity();
|
||
CopyEntity(fromEntityID, toWorld, newEntity);
|
||
return newEntity;
|
||
}
|
||
public void CloneEntity(int fromEntityID, int toEntityID)
|
||
{
|
||
CopyEntity(fromEntityID, toEntityID);
|
||
foreach (var pool in _pools)
|
||
{
|
||
if (!pool.Has(fromEntityID) && pool.Has(toEntityID))
|
||
pool.Del(toEntityID);
|
||
}
|
||
}
|
||
//public void CloneEntity(int fromEntityID, EcsWorld toWorld, int toEntityID)
|
||
#endregion
|
||
|
||
#endregion
|
||
|
||
#region DelEntBuffer
|
||
public AutoReleaseDelEntBufferLonkUnloker DisableAutoReleaseDelEntBuffer()
|
||
{
|
||
_isEnableAutoReleaseDelEntBuffer = false;
|
||
return new AutoReleaseDelEntBufferLonkUnloker(this);
|
||
}
|
||
public void EnableAutoReleaseDelEntBuffer()
|
||
{
|
||
_isEnableAutoReleaseDelEntBuffer = true;
|
||
}
|
||
public readonly struct AutoReleaseDelEntBufferLonkUnloker : IDisposable
|
||
{
|
||
private readonly EcsWorld _source;
|
||
public AutoReleaseDelEntBufferLonkUnloker(EcsWorld source)
|
||
{
|
||
_source = source;
|
||
}
|
||
public void Dispose()
|
||
{
|
||
_source.EnableAutoReleaseDelEntBuffer();
|
||
}
|
||
}
|
||
public void ReleaseDelEntityBufferAll()
|
||
{
|
||
ReleaseDelEntityBuffer(_delEntBufferCount);
|
||
}
|
||
//private UnsafeArray<int> _delBufferFilter = new UnsafeArray<int>(8);
|
||
public unsafe void ReleaseDelEntityBuffer(int count)
|
||
{
|
||
if (_delEntBufferCount <= 0)
|
||
{
|
||
return;
|
||
}
|
||
unchecked { _version++; }
|
||
count = Math.Clamp(count, 0, _delEntBufferCount);
|
||
_delEntBufferCount -= count;
|
||
ReadOnlySpan<int> buffer = new ReadOnlySpan<int>(_delEntBuffer, _delEntBufferCount, count);
|
||
for (int i = 0; i < _poolsCount; i++)
|
||
{
|
||
_pools[i].OnReleaseDelEntityBuffer(buffer);
|
||
}
|
||
_listeners.InvokeOnReleaseDelEntityBuffer(buffer);
|
||
for (int i = 0; i < buffer.Length; i++)
|
||
{
|
||
int e = buffer[i];
|
||
_entityDispenser.Release(e);
|
||
unchecked { _gens[e]++; }//up gen
|
||
_gens[e] |= DEATH_GEN_BIT;
|
||
}
|
||
Densify();
|
||
}
|
||
private void Densify() //уплотнение свободных айдишников
|
||
{
|
||
_entityDispenser.Sort();
|
||
}
|
||
#endregion
|
||
|
||
#region Upsize
|
||
public void Upsize(int minSize)
|
||
{
|
||
_entityDispenser.Upsize(minSize);
|
||
}
|
||
[MethodImpl(MethodImplOptions.NoInlining)]
|
||
private void OnEntityDispenserResized(int newSize)
|
||
{
|
||
Array.Resize(ref _gens, newSize);
|
||
Array.Resize(ref _componentCounts, newSize);
|
||
Array.Resize(ref _delEntBuffer, newSize);
|
||
_entityComponentMaskLength = _pools.Length / COMPONENT_MATRIX_MASK_BITSIZE + 1;
|
||
Array.Resize(ref _entityComponentMasks, newSize * _entityComponentMaskLength);
|
||
|
||
ArrayUtility.Fill(_gens, DEATH_GEN_BIT, _entitiesCapacity);
|
||
|
||
_entitiesCapacity = newSize;
|
||
|
||
for (int i = 0; i < _groups.Count; i++)
|
||
{
|
||
if (_groups[i].TryGetTarget(out EcsGroup group))
|
||
{
|
||
group.OnWorldResize(newSize);
|
||
}
|
||
else
|
||
{
|
||
int last = _groups.Count - 1;
|
||
_groups[i--] = _groups[last];
|
||
_groups.RemoveAt(last);
|
||
}
|
||
}
|
||
foreach (var item in _pools)
|
||
{
|
||
item.OnWorldResize(newSize);
|
||
}
|
||
|
||
_listeners.InvokeOnWorldResize(newSize);
|
||
}
|
||
#endregion
|
||
|
||
#region Groups Pool
|
||
internal void RegisterGroup(EcsGroup group)
|
||
{
|
||
_groups.Add(new WeakReference<EcsGroup>(group));
|
||
}
|
||
internal EcsGroup GetFreeGroup()
|
||
{
|
||
EcsGroup result = _groupsPool.Count <= 0 ? new EcsGroup(this) : _groupsPool.Pop();
|
||
result._isReleased = false;
|
||
return result;
|
||
}
|
||
internal void ReleaseGroup(EcsGroup group)
|
||
{
|
||
#if (DEBUG && !DISABLE_DEBUG) || !DISABLE_DRAGONECS_ASSERT_CHEKS
|
||
if (group.World != this) Throw.World_GroupDoesNotBelongWorld();
|
||
#endif
|
||
group._isReleased = true;
|
||
group.Clear();
|
||
_groupsPool.Push(group);
|
||
}
|
||
#endregion
|
||
|
||
#region Listeners
|
||
public void AddListener(IEcsWorldEventListener worldEventListener)
|
||
{
|
||
_listeners.Add(worldEventListener);
|
||
}
|
||
public void RemoveListener(IEcsWorldEventListener worldEventListener)
|
||
{
|
||
_listeners.Remove(worldEventListener);
|
||
}
|
||
public void AddListener(IEcsEntityEventListener entityEventListener)
|
||
{
|
||
_entityListeners.Add(entityEventListener);
|
||
}
|
||
public void RemoveListener(IEcsEntityEventListener entityEventListener)
|
||
{
|
||
_entityListeners.Remove(entityEventListener);
|
||
}
|
||
#endregion
|
||
|
||
#region Debug
|
||
public void GetComponents(int entityID, List<object> list)
|
||
{
|
||
list.Clear();
|
||
var itemsCount = GetComponentsCount(entityID);
|
||
for (var i = 0; i < _poolsCount; i++)
|
||
{
|
||
if (_pools[i].Has(entityID))
|
||
{
|
||
itemsCount--;
|
||
list.Add(_pools[i].GetRaw(entityID));
|
||
if (itemsCount <= 0)
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
public void GetComponentTypes(int entityID, HashSet<Type> typeSet)
|
||
{
|
||
typeSet.Clear();
|
||
var itemsCount = GetComponentsCount(entityID);
|
||
for (var i = 0; i < _poolsCount; i++)
|
||
{
|
||
if (_pools[i].Has(entityID))
|
||
{
|
||
itemsCount--;
|
||
typeSet.Add(_pools[i].ComponentType);
|
||
if (itemsCount <= 0)
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
#endregion
|
||
}
|
||
|
||
#region Callbacks Interface
|
||
public interface IEcsWorldEventListener
|
||
{
|
||
void OnWorldResize(int newSize);
|
||
void OnReleaseDelEntityBuffer(ReadOnlySpan<int> buffer);
|
||
void OnWorldDestroy();
|
||
}
|
||
public interface IEcsEntityEventListener
|
||
{
|
||
void OnNewEntity(int entityID);
|
||
void OnDelEntity(int entityID);
|
||
}
|
||
internal static class WorldEventListExtensions
|
||
{
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public static void InvokeOnWorldResize(this List<IEcsWorldEventListener> self, int newSize)
|
||
{
|
||
for (int i = 0, iMax = self.Count; i < iMax; i++) self[i].OnWorldResize(newSize);
|
||
}
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public static void InvokeOnReleaseDelEntityBuffer(this List<IEcsWorldEventListener> self, ReadOnlySpan<int> buffer)
|
||
{
|
||
for (int i = 0, iMax = self.Count; i < iMax; i++) self[i].OnReleaseDelEntityBuffer(buffer);
|
||
}
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public static void InvokeOnWorldDestroy(this List<IEcsWorldEventListener> self)
|
||
{
|
||
for (int i = 0, iMax = self.Count; i < iMax; i++) self[i].OnWorldDestroy();
|
||
}
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public static void InvokeOnNewEntity(this List<IEcsEntityEventListener> self, int entityID)
|
||
{
|
||
for (int i = 0, iMax = self.Count; i < iMax; i++) self[i].OnNewEntity(entityID);
|
||
}
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public static void InvokeOnDelEntity(this List<IEcsEntityEventListener> self, int entityID)
|
||
{
|
||
for (int i = 0, iMax = self.Count; i < iMax; i++) self[i].OnDelEntity(entityID);
|
||
}
|
||
}
|
||
#endregion
|
||
|
||
#region Extensions
|
||
public static class IntExtensions
|
||
{
|
||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
public static entlong ToEntityLong(this int self, EcsWorld world) => world.GetEntityLong(self);
|
||
}
|
||
#endregion
|
||
} |