Pow3r: stage 1 (#4208)

Co-authored-by: 20kdc <asdd2808@gmail.com>

Content.Server/Power/SMES/SmesComponent.cs

@@ -0,0 +1,91 @@
+#nullable enable
+using System;
+using Content.Server.Power.Components;
+using Content.Shared.Power;
+using Content.Shared.Rounding;
+using Content.Shared.SMES;
+using Robust.Server.GameObjects;
+using Robust.Shared.GameObjects;
+using Robust.Shared.IoC;
+using Robust.Shared.Timing;
+
+namespace Content.Server.Power.SMES
+{
+    /// <summary>
+    ///     Handles the "user-facing" side of the actual SMES object.
+    ///     This is operations that are specific to the SMES, like UI and visuals.
+    ///     Code interfacing with the powernet is handled in <see cref="BatteryStorageComponent"/> and <see cref="BatteryDischargerComponent"/>.
+    /// </summary>
+    [RegisterComponent]
+    public class SmesComponent : Component
+    {
+        [Dependency] private readonly IGameTiming _gameTiming = default!;
+
+        public override string Name => "Smes";
+
+        private int _lastChargeLevel;
+
+        private TimeSpan _lastChargeLevelChange;
+
+        private ChargeState _lastChargeState;
+
+        private TimeSpan _lastChargeStateChange;
+
+        private const int VisualsChangeDelay = 1;
+
+        protected override void Initialize()
+        {
+            base.Initialize();
+
+            Owner.EnsureComponentWarn<AppearanceComponent>();
+        }
+
+        public void OnUpdate()
+        {
+            var newLevel = GetNewChargeLevel();
+            if (newLevel != _lastChargeLevel && _lastChargeLevelChange + TimeSpan.FromSeconds(VisualsChangeDelay) < _gameTiming.CurTime)
+            {
+                _lastChargeLevel = newLevel;
+                _lastChargeLevelChange = _gameTiming.CurTime;
+
+                if (Owner.TryGetComponent(out AppearanceComponent? appearance))
+                {
+                    appearance.SetData(SmesVisuals.LastChargeLevel, newLevel);
+                }
+            }
+
+            var newChargeState = GetNewChargeState();
+            if (newChargeState != _lastChargeState && _lastChargeStateChange + TimeSpan.FromSeconds(VisualsChangeDelay) < _gameTiming.CurTime)
+            {
+                _lastChargeState = newChargeState;
+                _lastChargeStateChange = _gameTiming.CurTime;
+
+                if (Owner.TryGetComponent(out AppearanceComponent? appearance))
+                {
+                    appearance.SetData(SmesVisuals.LastChargeState, newChargeState);
+                }
+            }
+        }
+
+        private int GetNewChargeLevel()
+        {
+            if (!Owner.TryGetComponent(out BatteryComponent? battery))
+            {
+                return 0;
+            }
+
+            return ContentHelpers.RoundToLevels(battery.CurrentCharge, battery.MaxCharge, 6);
+        }
+
+        private ChargeState GetNewChargeState()
+        {
+            var battery = Owner.GetComponent<PowerNetworkBatteryComponent>();
+            return (battery.CurrentSupply - battery.CurrentReceiving) switch
+            {
+                > 0 => ChargeState.Discharging,
+                < 0 => ChargeState.Charging,
+                _ => ChargeState.Still
+            };
+        }
+    }
+}