61 lines
2.3 KiB
C#
61 lines
2.3 KiB
C#
using Content.Server.NodeContainer.Nodes;
|
|
using Content.Server.Power.Components;
|
|
using Content.Server.Power.EntitySystems;
|
|
using Content.Server.Power.Pow3r;
|
|
using Robust.Shared.Utility;
|
|
|
|
namespace Content.Server.Power.NodeGroups;
|
|
|
|
public abstract class BasePowerNet<TNetType> : BaseNetConnectorNodeGroup<TNetType>, IBasePowerNet
|
|
where TNetType : IBasePowerNet
|
|
{
|
|
[ViewVariables] public readonly List<PowerConsumerComponent> Consumers = new();
|
|
[ViewVariables] public readonly List<PowerSupplierComponent> Suppliers = new();
|
|
public PowerNetSystem PowerNetSystem = default!;
|
|
|
|
[ViewVariables]
|
|
public PowerState.Network NetworkNode { get; } = new();
|
|
|
|
public override void Initialize(Node sourceNode, IEntityManager entMan)
|
|
{
|
|
base.Initialize(sourceNode, entMan);
|
|
PowerNetSystem = entMan.EntitySysManager.GetEntitySystem<PowerNetSystem>();
|
|
}
|
|
|
|
public void AddConsumer(PowerConsumerComponent consumer)
|
|
{
|
|
DebugTools.Assert(consumer.NetworkLoad.LinkedNetwork == default);
|
|
consumer.NetworkLoad.LinkedNetwork = default;
|
|
Consumers.Add(consumer);
|
|
QueueNetworkReconnect();
|
|
}
|
|
|
|
public void RemoveConsumer(PowerConsumerComponent consumer)
|
|
{
|
|
// Linked network can be default if it was re-connected twice in one tick.
|
|
DebugTools.Assert(consumer.NetworkLoad.LinkedNetwork == default || consumer.NetworkLoad.LinkedNetwork == NetworkNode.Id);
|
|
consumer.NetworkLoad.LinkedNetwork = default;
|
|
Consumers.Remove(consumer);
|
|
QueueNetworkReconnect();
|
|
}
|
|
|
|
public void AddSupplier(PowerSupplierComponent supplier)
|
|
{
|
|
DebugTools.Assert(supplier.NetworkSupply.LinkedNetwork == default);
|
|
supplier.NetworkSupply.LinkedNetwork = default;
|
|
Suppliers.Add(supplier);
|
|
QueueNetworkReconnect();
|
|
}
|
|
|
|
public void RemoveSupplier(PowerSupplierComponent supplier)
|
|
{
|
|
// Linked network can be default if it was re-connected twice in one tick.
|
|
DebugTools.Assert(supplier.NetworkSupply.LinkedNetwork == default || supplier.NetworkSupply.LinkedNetwork == NetworkNode.Id);
|
|
supplier.NetworkSupply.LinkedNetwork = default;
|
|
Suppliers.Remove(supplier);
|
|
QueueNetworkReconnect();
|
|
}
|
|
|
|
public abstract void QueueNetworkReconnect();
|
|
}
|