tbd-station-14/Content.Server/Procedural/DungeonJob/DungeonJob.CorridorClutter.cs

using System.Threading.Tasks;
using Content.Shared.Procedural;
using Content.Shared.Procedural.PostGeneration;
using Content.Shared.Storage;
using Robust.Shared.Physics.Components;
using Robust.Shared.Random;

namespace Content.Server.Procedural.DungeonJob;

public sealed partial class DungeonJob
{
    /// <summary>
    /// <see cref="CorridorClutterDunGen"/>
    /// </summary>
    private async Task PostGen(CorridorClutterDunGen gen, Dungeon dungeon, HashSet<Vector2i> reservedTiles, Random random)
    {
        var physicsQuery = _entManager.GetEntityQuery<PhysicsComponent>();
        var count = (int) Math.Ceiling(dungeon.CorridorTiles.Count * gen.Chance);
        var contents = _prototype.Index(gen.Contents);

        while (count > 0)
        {
            var tile = random.Pick(dungeon.CorridorTiles);

            var enumerator = _maps.GetAnchoredEntitiesEnumerator(_gridUid, _grid, tile);
            var blocked = false;

            while (enumerator.MoveNext(out var ent))
            {
                if (!physicsQuery.TryGetComponent(ent, out var physics) ||
                    !physics.CanCollide ||
                    !physics.Hard)
                {
                    continue;
                }

                blocked = true;
                break;
            }

            if (blocked)
                continue;

            count--;

            if (reservedTiles.Contains(tile))
                continue;

            var protos = _entTable.GetSpawns(contents, random);
            var coords = _maps.ToCenterCoordinates(_gridUid, tile, _grid);
            var uids = _entManager.SpawnEntitiesAttachedTo(coords, protos);

            foreach (var uid in uids)
            {
                AddLoadedEntity(tile, uid);
            }

            await SuspendIfOutOfTime();

            if (!ValidateResume())
                return;
        }
    }
}