93 lines
3.1 KiB
C#
93 lines
3.1 KiB
C#
using System.Threading.Tasks;
|
|
using Content.Server.Parallax;
|
|
using Content.Shared.Maps;
|
|
using Content.Shared.Parallax.Biomes;
|
|
using Content.Shared.Procedural;
|
|
using Content.Shared.Procedural.DungeonLayers;
|
|
using Content.Shared.Procedural.PostGeneration;
|
|
using Robust.Shared.Map;
|
|
using Robust.Shared.Utility;
|
|
|
|
namespace Content.Server.Procedural.DungeonJob;
|
|
|
|
public sealed partial class DungeonJob
|
|
{
|
|
/// <summary>
|
|
/// <see cref="BiomeDunGen"/>
|
|
/// </summary>
|
|
private async Task PostGen(CP14BiomeDunGen dunGen,
|
|
List<Dungeon> dungeons,
|
|
HashSet<Vector2i> reservedTiles,
|
|
Random random)
|
|
{
|
|
if (!_prototype.TryIndex(dunGen.BiomeTemplate, out var indexedBiome))
|
|
return;
|
|
|
|
HashSet<Vector2i> processedTiles = new();
|
|
var biomeSystem = _entManager.System<BiomeSystem>();
|
|
|
|
var seed = random.Next();
|
|
var xformQuery = _entManager.GetEntityQuery<TransformComponent>();
|
|
|
|
foreach (var dun in dungeons)
|
|
{
|
|
foreach (var node in dun.AllTiles)
|
|
{
|
|
if (processedTiles.Contains(node))
|
|
continue;
|
|
|
|
processedTiles.Add(node);
|
|
|
|
var tileRef = _maps.GetTileRef(_gridUid, _grid, node);
|
|
|
|
if (reservedTiles.Contains(node))
|
|
continue;
|
|
|
|
if (dunGen.TileMask is not null)
|
|
{
|
|
if (!dunGen.TileMask.Contains(((ContentTileDefinition)_tileDefManager[tileRef.Tile.TypeId]).ID))
|
|
continue;
|
|
}
|
|
|
|
if (dunGen.TileMask is not null)
|
|
{
|
|
if (!dunGen.TileMask.Contains(((ContentTileDefinition) _tileDefManager[tileRef.Tile.TypeId]).ID))
|
|
continue;
|
|
}
|
|
|
|
// Need to set per-tile to override data.
|
|
if (biomeSystem.TryGetTile(node, indexedBiome.Layers, seed, (_gridUid, _grid), out var tile))
|
|
{
|
|
_maps.SetTile(_gridUid, _grid, node, tile.Value);
|
|
}
|
|
|
|
if (biomeSystem.TryGetDecals(node, indexedBiome.Layers, seed, (_gridUid, _grid), out var decals))
|
|
{
|
|
foreach (var decal in decals)
|
|
{
|
|
_decals.TryAddDecal(decal.ID, new EntityCoordinates(_gridUid, decal.Position), out _);
|
|
}
|
|
}
|
|
|
|
if (biomeSystem.TryGetEntity(node, indexedBiome.Layers, tile ?? tileRef.Tile, seed, (_gridUid, _grid), out var entityProto))
|
|
{
|
|
var ent = _entManager.SpawnEntity(entityProto, new EntityCoordinates(_gridUid, node + _grid.TileSizeHalfVector));
|
|
var xform = xformQuery.Get(ent);
|
|
|
|
if (!xform.Comp.Anchored)
|
|
{
|
|
_transform.AnchorEntity(ent, xform);
|
|
}
|
|
|
|
// TODO: Engine bug with SpawnAtPosition
|
|
DebugTools.Assert(xform.Comp.Anchored);
|
|
}
|
|
|
|
await SuspendDungeon();
|
|
if (!ValidateResume())
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
}
|