Files
tbd-station-14/Content.Server/Power/Nodes/CableDeviceNode.cs
Pieter-Jan Briers 103bc19508 Pow3r: stage 1 (#4208)
Co-authored-by: 20kdc <asdd2808@gmail.com>
2021-07-04 18:11:52 +02:00

30 lines
957 B
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
{
/// <summary>
/// Type of node that connects to a <see cref="CableNode"/> below it.
/// </summary>
[DataDefinition]
public class CableDeviceNode : Node
{
public override IEnumerable<Node> GetReachableNodes()
{
var compMgr = IoCManager.Resolve<IComponentManager>();
var grid = IoCManager.Resolve<IMapManager>().GetGrid(Owner.Transform.GridID);
var gridIndex = grid.TileIndicesFor(Owner.Transform.Coordinates);
foreach (var node in NodeHelpers.GetNodesInTile(compMgr, grid, gridIndex))
{
if (node is CableNode)
yield return node;
}
}
}
}