38 lines
1.4 KiB
C#
38 lines
1.4 KiB
C#
using Content.Server.Construction;
|
|
using Content.Server.Power.Components;
|
|
using JetBrains.Annotations;
|
|
|
|
namespace Content.Server.Power.EntitySystems
|
|
{
|
|
[UsedImplicitly]
|
|
public sealed class UpgradeBatterySystem : EntitySystem
|
|
{
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<UpgradeBatteryComponent, RefreshPartsEvent>(OnRefreshParts);
|
|
SubscribeLocalEvent<UpgradeBatteryComponent, UpgradeExamineEvent>(OnUpgradeExamine);
|
|
}
|
|
|
|
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;
|
|
}
|
|
}
|
|
|
|
private void OnUpgradeExamine(EntityUid uid, UpgradeBatteryComponent component, UpgradeExamineEvent args)
|
|
{
|
|
// UpgradeBatteryComponent.MaxChargeMultiplier is not the actual multiplier, so we have to do this.
|
|
if (TryComp<BatteryComponent>(uid, out var batteryComp))
|
|
{
|
|
args.AddPercentageUpgrade("upgrade-max-charge", batteryComp.MaxCharge / component.BaseMaxCharge);
|
|
}
|
|
}
|
|
}
|
|
}
|