109 lines
3.6 KiB
C#
109 lines
3.6 KiB
C#
using System.Linq;
|
|
using Content.Server.NodeContainer.NodeGroups;
|
|
using Content.Server.NodeContainer.Nodes;
|
|
using Content.Server.Power.Components;
|
|
using Content.Server.Power.EntitySystems;
|
|
using JetBrains.Annotations;
|
|
|
|
namespace Content.Server.Power.NodeGroups
|
|
{
|
|
public interface IApcNet : IBasePowerNet
|
|
{
|
|
void AddApc(EntityUid uid, ApcComponent apc);
|
|
|
|
void RemoveApc(EntityUid uid, ApcComponent apc);
|
|
|
|
void AddPowerProvider(ApcPowerProviderComponent provider);
|
|
|
|
void RemovePowerProvider(ApcPowerProviderComponent provider);
|
|
|
|
void QueueNetworkReconnect();
|
|
}
|
|
|
|
[NodeGroup(NodeGroupID.Apc)]
|
|
[UsedImplicitly]
|
|
public sealed partial class ApcNet : BasePowerNet<IApcNet>, IApcNet
|
|
{
|
|
[ViewVariables] public readonly List<ApcComponent> Apcs = new();
|
|
[ViewVariables] public readonly List<ApcPowerProviderComponent> Providers = new();
|
|
|
|
//Debug property
|
|
[ViewVariables] private int TotalReceivers => Providers.Sum(provider => provider.LinkedReceivers.Count);
|
|
|
|
[ViewVariables]
|
|
private IEnumerable<ApcPowerReceiverComponent> AllReceivers =>
|
|
Providers.SelectMany(provider => provider.LinkedReceivers);
|
|
|
|
public override void Initialize(Node sourceNode, IEntityManager entMan)
|
|
{
|
|
base.Initialize(sourceNode, entMan);
|
|
PowerNetSystem.InitApcNet(this);
|
|
}
|
|
|
|
public override void AfterRemake(IEnumerable<IGrouping<INodeGroup?, Node>> newGroups)
|
|
{
|
|
base.AfterRemake(newGroups);
|
|
|
|
PowerNetSystem?.DestroyApcNet(this);
|
|
}
|
|
|
|
public void AddApc(EntityUid uid, ApcComponent apc)
|
|
{
|
|
if (EntMan.TryGetComponent(uid, out PowerNetworkBatteryComponent? netBattery))
|
|
netBattery.NetworkBattery.LinkedNetworkDischarging = default;
|
|
|
|
QueueNetworkReconnect();
|
|
Apcs.Add(apc);
|
|
}
|
|
|
|
public void RemoveApc(EntityUid uid, ApcComponent apc)
|
|
{
|
|
if (EntMan.TryGetComponent(uid, out PowerNetworkBatteryComponent? netBattery))
|
|
netBattery.NetworkBattery.LinkedNetworkDischarging = default;
|
|
|
|
QueueNetworkReconnect();
|
|
Apcs.Remove(apc);
|
|
}
|
|
|
|
public void AddPowerProvider(ApcPowerProviderComponent provider)
|
|
{
|
|
Providers.Add(provider);
|
|
|
|
QueueNetworkReconnect();
|
|
}
|
|
|
|
public void RemovePowerProvider(ApcPowerProviderComponent provider)
|
|
{
|
|
Providers.Remove(provider);
|
|
|
|
QueueNetworkReconnect();
|
|
}
|
|
|
|
public override void QueueNetworkReconnect()
|
|
{
|
|
PowerNetSystem?.QueueReconnectApcNet(this);
|
|
}
|
|
|
|
protected override void SetNetConnectorNet(IBaseNetConnectorComponent<IApcNet> netConnectorComponent)
|
|
{
|
|
netConnectorComponent.Net = this;
|
|
}
|
|
|
|
public override string? GetDebugData()
|
|
{
|
|
// This is just recycling the multi-tool examine.
|
|
|
|
var ps = PowerNetSystem.GetNetworkStatistics(NetworkNode);
|
|
|
|
float storageRatio = ps.InStorageCurrent / Math.Max(ps.InStorageMax, 1.0f);
|
|
float outStorageRatio = ps.OutStorageCurrent / Math.Max(ps.OutStorageMax, 1.0f);
|
|
return @$"Current Supply: {ps.SupplyCurrent:G3}
|
|
From Batteries: {ps.SupplyBatteries:G3}
|
|
Theoretical Supply: {ps.SupplyTheoretical:G3}
|
|
Ideal Consumption: {ps.Consumption:G3}
|
|
Input Storage: {ps.InStorageCurrent:G3} / {ps.InStorageMax:G3} ({storageRatio:P1})
|
|
Output Storage: {ps.OutStorageCurrent:G3} / {ps.OutStorageMax:G3} ({outStorageRatio:P1})";
|
|
}
|
|
}
|
|
}
|