Files
tbd-station-14/Content.Server/Power/Nodes/CableTerminalPortNode.cs

31 lines
1.0 KiB
C#

using System.Collections.Generic;
using Content.Server.NodeContainer.Nodes;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Map;
using Robust.Shared.Serialization.Manager.Attributes;
namespace Content.Server.Power.Nodes
{
[DataDefinition]
public class CableTerminalPortNode : Node
{
public override IEnumerable<Node> GetReachableNodes()
{
if (Owner.Transform.GridID == GridId.Invalid)
yield break; // No funny nodes in spess.
var entMan = IoCManager.Resolve<IEntityManager>();
var grid = IoCManager.Resolve<IMapManager>().GetGrid(Owner.Transform.GridID);
var gridIndex = grid.TileIndicesFor(Owner.Transform.Coordinates);
var nodes = NodeHelpers.GetCardinalNeighborNodes(entMan, grid, gridIndex, includeSameTile: false);
foreach (var (_, node) in nodes)
{
if (node is CableTerminalNode)
yield return node;
}
}
}
}