63 lines
1.9 KiB
C#
63 lines
1.9 KiB
C#
using System.Diagnostics.CodeAnalysis;
|
|
using Content.Shared.Construction.NodeEntities;
|
|
using Content.Shared.Construction.Serialization;
|
|
using Robust.Shared.Prototypes;
|
|
using Robust.Shared.Serialization.TypeSerializers.Implementations.Custom.Prototype;
|
|
|
|
namespace Content.Shared.Construction
|
|
{
|
|
[Serializable]
|
|
[DataDefinition]
|
|
public sealed partial class ConstructionGraphNode
|
|
{
|
|
[DataField("actions", serverOnly: true)]
|
|
private IGraphAction[] _actions = Array.Empty<IGraphAction>();
|
|
|
|
[DataField("edges")]
|
|
private ConstructionGraphEdge[] _edges = Array.Empty<ConstructionGraphEdge>();
|
|
|
|
[DataField("node", required: true)]
|
|
public string Name { get; private set; } = default!;
|
|
|
|
[ViewVariables]
|
|
public IReadOnlyList<ConstructionGraphEdge> Edges => _edges;
|
|
|
|
[ViewVariables]
|
|
public IReadOnlyList<IGraphAction> Actions => _actions;
|
|
|
|
[DataField("transform")]
|
|
public IGraphTransform[] TransformLogic = Array.Empty<IGraphTransform>();
|
|
|
|
[DataField("entity", customTypeSerializer: typeof(GraphNodeEntitySerializer), serverOnly:true)]
|
|
public IGraphNodeEntity Entity { get; private set; } = new NullNodeEntity();
|
|
|
|
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;
|
|
}
|
|
}
|
|
}
|