Cinka/OldThink
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Fancier powergrid check (#6901)

Browse Source
This commit is contained in:
Moony
2022-02-27 01:01:39 -06:00
committed by GitHub
parent 3cc5efc86d
commit 7d9c7e46ab
1 changed files with 49 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
Content.Server/StationEvents/Events/PowerGridCheck.cs
View File

@@ -4,6 +4,7 @@ using JetBrains.Annotations;
using Robust.Shared.Audio;
using Robust.Shared.Player;
using Robust.Shared.Random;
using Robust.Shared.Utility;
using Timer = Robust.Shared.Timing.Timer;
namespace Content.Server.StationEvents.Events
@@ -11,6 +12,9 @@ namespace Content.Server.StationEvents.Events
[UsedImplicitly]
public sealed class PowerGridCheck : StationEvent
{
[Dependency] private readonly IEntityManager _entityManager = default!;
[Dependency] private readonly IRobustRandom _random = default!;
public override string Name => "PowerGridCheck";
public override float Weight => WeightNormal;
public override int? MaxOccurrences => 3;
@@ -25,6 +29,13 @@ namespace Content.Server.StationEvents.Events
private CancellationTokenSource? _announceCancelToken;
private readonly List<EntityUid> _powered = new();
private readonly List<EntityUid> _unpowered = new();
private const float SecondsUntilOff = 30.0f;
private int _numberPerSecond = 0;
private float UpdateRate => 1.0f / _numberPerSecond;
private float _frameTimeAccumulator = 0.0f;
public override void Announce()
{
@@ -34,26 +45,53 @@ namespace Content.Server.StationEvents.Events
public override void Startup()
{
var entityManager = IoCManager.Resolve<IEntityManager>();
foreach (var component in entityManager.EntityQuery<ApcPowerReceiverComponent>(true))
foreach (var component in _entityManager.EntityQuery<ApcPowerReceiverComponent>(true))
{
component.PowerDisabled = true;
_powered.Add(component.Owner);
if (!component.PowerDisabled)
_powered.Add(component.Owner);
}
_random.Shuffle(_powered);
_numberPerSecond = Math.Max(1, (int)(_powered.Count / SecondsUntilOff)); // Number of APCs to turn off every second. At least one.
base.Startup();
}
public override void Update(float frameTime)
{
base.Update(frameTime);
_frameTimeAccumulator += frameTime;
var updates = 0;
if (_frameTimeAccumulator > UpdateRate)
{
updates = (int) (_frameTimeAccumulator / UpdateRate);
_frameTimeAccumulator -= UpdateRate * updates;
}
for (var i = 0; i < updates; i++)
{
if (_powered.Count == 0)
break;
var selected = _powered.Pop();
if (_entityManager.Deleted(selected)) continue;
if (_entityManager.TryGetComponent<ApcPowerReceiverComponent>(selected, out var powerReceiverComponent))
{
powerReceiverComponent.PowerDisabled = true;
}
_unpowered.Add(selected);
}
}
public override void Shutdown()
{
var entMan = IoCManager.Resolve<IEntityManager>();
foreach (var entity in _powered)
foreach (var entity in _unpowered)
{
if (entMan.Deleted(entity)) continue;
if (_entityManager.Deleted(entity)) continue;
if (entMan.TryGetComponent(entity, out ApcPowerReceiverComponent? powerReceiverComponent))
if (_entityManager.TryGetComponent(entity, out ApcPowerReceiverComponent? powerReceiverComponent))
{
powerReceiverComponent.PowerDisabled = false;
}
@@ -66,7 +104,7 @@ namespace Content.Server.StationEvents.Events
{
SoundSystem.Play(Filter.Broadcast(), "/Audio/Announcements/power_on.ogg", AudioParams);
}, _announceCancelToken.Token);
_powered.Clear();
_unpowered.Clear();
base.Shutdown();
}