Files
tbd-station-14/Content.Shared/Construction/ConstructionGraphNode.cs
2022-02-16 18:23:23 +11:00

62 lines
1.6 KiB
C#

using System;
using System.Collections.Generic;
using System.Diagnostics.CodeAnalysis;
using Robust.Shared.Serialization.Manager.Attributes;
using Robust.Shared.ViewVariables;
namespace Content.Shared.Construction
{
[Serializable]
[DataDefinition]
public sealed class ConstructionGraphNode
{
[DataField("actions", serverOnly: true)]
private IGraphAction[] _actions = Array.Empty<IGraphAction>();
[DataField("edges")]
private ConstructionGraphEdge[] _edges = Array.Empty<ConstructionGraphEdge>();
[ViewVariables]
[DataField("node", required: true)]
public string Name { get; private set; } = default!;
[ViewVariables]
public IReadOnlyList<ConstructionGraphEdge> Edges => _edges;
[ViewVariables]
public IReadOnlyList<IGraphAction> Actions => _actions;
[ViewVariables]
[DataField("entity")]
public string? Entity { get; private set; }
public ConstructionGraphEdge? GetEdge(string target)
{
foreach (var edge in _edges)
{
if (edge.Target == target)
return edge;
}
return null;
}
public int? GetEdgeIndex(string target)
{
for (var i = 0; i < _edges.Length; i++)
{
var edge = _edges[i];
if (edge.Target == target)
return i;
}
return null;
}
public bool TryGetEdge(string target, [NotNullWhen(true)] out ConstructionGraphEdge? edge)
{
return (edge = GetEdge(target)) != null;
}
}
}