84 lines
2.7 KiB
C#
84 lines
2.7 KiB
C#
using Content.Shared.Emp;
|
|
using Content.Shared.Power.Components;
|
|
using JetBrains.Annotations;
|
|
|
|
namespace Content.Shared.Power.EntitySystems;
|
|
|
|
public abstract class SharedBatterySystem : EntitySystem
|
|
{
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<BatteryComponent, EmpPulseEvent>(OnEmpPulse);
|
|
}
|
|
|
|
private void OnEmpPulse(Entity<BatteryComponent> ent, ref EmpPulseEvent args)
|
|
{
|
|
args.Affected = true;
|
|
UseCharge(ent.AsNullable(), args.EnergyConsumption);
|
|
// Apply a cooldown to the entity's self recharge if needed to avoid it immediately self recharging after an EMP.
|
|
TrySetChargeCooldown(ent.Owner);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Changes the battery's charge by the given amount
|
|
/// and resets the self-recharge cooldown if it exists.
|
|
/// A positive value will add charge, a negative value will remove charge.
|
|
/// </summary>
|
|
/// <returns>The actually changed amount.</returns>
|
|
[PublicAPI]
|
|
public virtual float ChangeCharge(Entity<BatteryComponent?> ent, float amount)
|
|
{
|
|
return 0f;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Removes the given amount of charge from the battery
|
|
/// and resets the self-recharge cooldown if it exists.
|
|
/// </summary>
|
|
/// <returns>The actually changed amount.</returns>
|
|
[PublicAPI]
|
|
public virtual float UseCharge(Entity<BatteryComponent?> ent, float amount)
|
|
{
|
|
return 0f;
|
|
}
|
|
|
|
/// <summary>
|
|
/// If sufficient charge is available on the battery, use it. Otherwise, don't.
|
|
/// Resets the self-recharge cooldown if it exists.
|
|
/// Always returns false on the client.
|
|
/// </summary>
|
|
/// <returns>If the full amount was able to be removed.</returns>
|
|
[PublicAPI]
|
|
public virtual bool TryUseCharge(Entity<BatteryComponent?> ent, float amount)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Sets the battery's charge.
|
|
/// </summary>
|
|
[PublicAPI]
|
|
public virtual void SetCharge(Entity<BatteryComponent?> ent, float value) { }
|
|
|
|
/// <summary>
|
|
/// Sets the battery's maximum charge.
|
|
/// </summary>
|
|
[PublicAPI]
|
|
public virtual void SetMaxCharge(Entity<BatteryComponent?> ent, float value) { }
|
|
|
|
/// <summary>
|
|
/// Checks if the entity has a self recharge and puts it on cooldown if applicable.
|
|
/// Uses the cooldown time given in the component.
|
|
/// </summary>
|
|
[PublicAPI]
|
|
public virtual void TrySetChargeCooldown(Entity<BatterySelfRechargerComponent?> ent) { }
|
|
|
|
/// <summary>
|
|
/// Puts the entity's self recharge on cooldown for the specified time.
|
|
/// </summary>
|
|
[PublicAPI]
|
|
public virtual void SetChargeCooldown(Entity<BatterySelfRechargerComponent?> ent, TimeSpan cooldown) { }
|
|
}
|