Machine Battery Scaling (SMES/Substation) (#11662)

Co-authored-by: CommieFlowers <rasmus.cedergren@hotmail.com>

Content.Server/Power/Components/UpgradeBatteryComponent.cs

@@ -0,0 +1,28 @@
+using Content.Shared.Construction.Prototypes;
+using Robust.Shared.Serialization.TypeSerializers.Implementations.Custom.Prototype;
+
+namespace Content.Server.Power.Components
+{
+
+    [RegisterComponent]
+    public sealed class UpgradeBatteryComponent : Component
+    {
+        /// <summary>
+        ///     The machine part that affects the power capacity.
+        /// </summary>
+        [DataField("machinePartPowerCapacity", customTypeSerializer: typeof(PrototypeIdSerializer<MachinePartPrototype>))]
+        public string MachinePartPowerCapacity = "Capacitor";
+
+        /// <summary>
+        ///     The machine part rating is raised to this power when calculating power gain
+        /// </summary>
+        [DataField("maxChargeMultiplier")]
+        public float MaxChargeMultiplier = 2f;
+
+        /// <summary>
+        ///     Power gain scaling
+        /// </summary>
+        [DataField("baseMaxCharge")]
+        public float BaseMaxCharge = 8000000;
+    }
+}

Content.Server/Power/EntitySystems/UpgradeBatterySystem.cs

@@ -0,0 +1,27 @@
+using Content.Server.Construction;
+using Content.Server.Power.Components;
+using JetBrains.Annotations;
+
+namespace Content.Server.Power.SMES
+{
+    [UsedImplicitly]
+    public sealed class UpgradeBatterySystem : EntitySystem
+    {
+        public override void Initialize()
+        {
+            base.Initialize();
+
+            SubscribeLocalEvent<UpgradeBatteryComponent, RefreshPartsEvent>(OnRefreshParts);
+        }
+
+        public void OnRefreshParts(EntityUid uid, UpgradeBatteryComponent component, RefreshPartsEvent args)
+        {
+            var capacitorRating = args.PartRatings[component.MachinePartPowerCapacity];
+
+            if (TryComp<BatteryComponent>(uid, out var batteryComp))
+            {
+                batteryComp.MaxCharge = MathF.Pow(component.MaxChargeMultiplier, capacitorRating - 1) * component.BaseMaxCharge;
+            }
+        }
+    }
+}