Files
crystall-punk-14/Content.Server/Power/Nodes/CableTerminalPortNode.cs

34 lines
1.2 KiB
C#
Raw Normal View History

2022-02-07 01:10:33 +13:00
using Content.Server.NodeContainer;
using Content.Server.NodeContainer.Nodes;
using Content.Shared.NodeContainer;
using Robust.Shared.Map;
using Robust.Shared.Map.Components;
namespace Content.Server.Power.Nodes
{
[DataDefinition]
public sealed partial class CableTerminalPortNode : Node
{
2022-02-07 01:10:33 +13:00
public override IEnumerable<Node> GetReachableNodes(TransformComponent xform,
EntityQuery<NodeContainerComponent> nodeQuery,
EntityQuery<TransformComponent> xformQuery,
MapGridComponent? grid,
2022-02-07 01:10:33 +13:00
IEntityManager entMan)
{
2022-02-07 01:10:33 +13:00
if (!xform.Anchored || grid == null)
yield break;
2022-02-07 01:10:33 +13:00
var gridIndex = grid.TileIndicesFor(xform.Coordinates);
2022-02-07 01:10:33 +13:00
var nodes = NodeHelpers.GetCardinalNeighborNodes(nodeQuery, grid, gridIndex, includeSameTile: false);
foreach (var (dir, node) in nodes)
{
if (node is CableTerminalNode
&& dir != Direction.Invalid
&& xformQuery.GetComponent(node.Owner).LocalRotation.GetCardinalDir().GetOpposite() == dir)
yield return node;
}
}
}
}