Cinka/OldThink
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Reduce atmos allocs a bunch (#7228)

Browse Source
This commit is contained in:
metalgearsloth
2022-03-27 17:49:26 +11:00
committed by GitHub
parent 0ab1a2d4c9
commit b506fdcf65
2 changed files with 54 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
Content.Server/Atmos/EntitySystems/AtmosphereSystem.Grid.cs
View File

@@ -195,13 +195,25 @@ namespace Content.Server.Atmos.EntitySystems
/// <returns></returns>
public IEnumerable<AirtightComponent> GetObstructingComponents(IMapGrid mapGrid, Vector2i tile)
{
foreach (var uid in mapGrid.GetAnchoredEntities(tile))
var airQuery = GetEntityQuery<AirtightComponent>();
var enumerator = mapGrid.GetAnchoredEntitiesEnumerator(tile);
while (enumerator.MoveNext(out var uid))
{
if (TryComp<AirtightComponent>(uid, out var ac))
yield return ac;
if (!airQuery.TryGetComponent(uid.Value, out var airtight)) continue;
yield return airtight;
}
}
public AtmosObstructionEnumerator GetObstructingComponentsEnumerator(IMapGrid mapGrid, Vector2i tile)
{
var ancEnumerator = mapGrid.GetAnchoredEntitiesEnumerator(tile);
var airQuery = GetEntityQuery<AirtightComponent>();
var enumerator = new AtmosObstructionEnumerator(ancEnumerator, airQuery);
return enumerator;
}
private AtmosDirection GetBlockedDirections(IMapGrid mapGrid, Vector2i indices)
{
var value = AtmosDirection.Invalid;
@@ -717,7 +729,9 @@ namespace Content.Server.Atmos.EntitySystems
{
var directions = AtmosDirection.Invalid;
foreach (var obstructingComponent in GetObstructingComponents(mapGrid, tile))
var enumerator = GetObstructingComponentsEnumerator(mapGrid, tile);
while (enumerator.MoveNext(out var obstructingComponent))
{
if (!obstructingComponent.AirBlocked)
continue;