28 lines
857 B
C#
28 lines
857 B
C#
using Content.Shared.Weapons.Ranged.Components;
|
|
|
|
namespace Content.Client.Weapons.Ranged.Systems;
|
|
|
|
public sealed partial class GunSystem
|
|
{
|
|
protected override void InitializeBattery()
|
|
{
|
|
base.InitializeBattery();
|
|
|
|
SubscribeLocalEvent<BatteryAmmoProviderComponent, UpdateAmmoCounterEvent>(OnAmmoCountUpdate);
|
|
SubscribeLocalEvent<BatteryAmmoProviderComponent, AmmoCounterControlEvent>(OnControl);
|
|
}
|
|
|
|
private void OnAmmoCountUpdate(Entity<BatteryAmmoProviderComponent> ent, ref UpdateAmmoCounterEvent args)
|
|
{
|
|
if (args.Control is not BoxesStatusControl boxes)
|
|
return;
|
|
|
|
boxes.Update(ent.Comp.Shots, ent.Comp.Capacity);
|
|
}
|
|
|
|
private void OnControl(Entity<BatteryAmmoProviderComponent> ent, ref AmmoCounterControlEvent args)
|
|
{
|
|
args.Control = new BoxesStatusControl();
|
|
}
|
|
}
|