MoonsideGames
/
MoonTools.ECS
1
0
Fork 0
Code Issues Pull Requests Projects Releases Activity

register filter components with storage

pull/1/head
cosmonaut 2022-03-07 12:05:06 -08:00
parent 1d7db58d43
commit 227aae9a1a
2 changed files with 19 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
src/ComponentDepot.cs
View File

@ -10,6 +10,14 @@ internal class ComponentDepot
private Dictionary<int, HashSet<Type>> entityComponentMap = new Dictionary<int, HashSet<Type>>();
internal void Register<TComponent>() where TComponent : struct
{
if (!storages.ContainsKey(typeof(TComponent)))
{
storages.Add(typeof(TComponent), new ComponentStorage<TComponent>());
}
}
private ComponentStorage Lookup(Type type)
{
return storages[type];
@ -18,11 +26,7 @@ internal class ComponentDepot
private ComponentStorage<TComponent> Lookup<TComponent>() where TComponent : struct
{
// TODO: is it possible to optimize this?
if (!storages.ContainsKey(typeof(TComponent)))
{
storages.Add(typeof(TComponent), new ComponentStorage<TComponent>());
}
Register<TComponent>();
return storages[typeof(TComponent)] as ComponentStorage<TComponent>;
}
@ -64,10 +68,7 @@ internal class ComponentDepot
{
foreach (var filterSignature in filterSignatures)
{
if (CheckFilter(filterSignature, entityID))
{
filterSignatureToEntityIDs[filterSignature].Add(entityID);
}
CheckFilter(filterSignature, entityID);
}
}
}
@ -96,10 +97,7 @@ internal class ComponentDepot
{
foreach (var filterSignature in filterSignatures)
{
if (!CheckFilter(filterSignature, entityID))
{
filterSignatureToEntityIDs[filterSignature].Remove(entityID);
}
CheckFilter(filterSignature, entityID);
}
}
}
@ -175,13 +173,14 @@ internal class ComponentDepot
}
}
private bool CheckFilter(FilterSignature filterSignature, int entityID)
private void CheckFilter(FilterSignature filterSignature, int entityID)
{
foreach (var type in filterSignature.Included)
{
if (!Has(type, entityID))
{
return false;
filterSignatureToEntityIDs[filterSignature].Remove(entityID);
return;
}
}
@ -189,10 +188,11 @@ internal class ComponentDepot
{
if (Has(type, entityID))
{
return false;
filterSignatureToEntityIDs[filterSignature].Remove(entityID);
return;
}
}
return true;
filterSignatureToEntityIDs[filterSignature].Add(entityID);
}
}

2
src/FilterBuilder.cs
View File

@ -22,12 +22,14 @@ public struct FilterBuilder
public FilterBuilder Include<TComponent>() where TComponent : struct
{
ComponentDepot.Register<TComponent>();
Included.Add(typeof(TComponent));
return new FilterBuilder(ComponentDepot, Included, Excluded);
}
public FilterBuilder Exclude<TComponent>() where TComponent : struct
{
ComponentDepot.Register<TComponent>();
Excluded.Add(typeof(TComponent));
return new FilterBuilder(ComponentDepot, Included, Excluded);
}