31 lines
1.1 KiB
C#
31 lines
1.1 KiB
C#
|
using Content.Shared.Weapons.Ranged.Components;
|
||
|
|
|
||
|
|
namespace Content.Client.Weapons.Ranged.Systems;
|
||
|
|
|
||
|
|
public sealed partial class GunSystem
|
||
|
|
{
|
||
|
|
protected override void InitializeBattery()
|
||
|
|
{
|
||
|
|
base.InitializeBattery();
|
||
|
|
// Hitscan
|
||
|
|
SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, AmmoCounterControlEvent>(OnControl);
|
||
|
|
SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, UpdateAmmoCounterEvent>(OnAmmoCountUpdate);
|
||
|
|
|
||
|
|
// Projectile
|
||
|
|
SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, AmmoCounterControlEvent>(OnControl);
|
||
|
|
SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, UpdateAmmoCounterEvent>(OnAmmoCountUpdate);
|
||
|
|
}
|
||
|
|
|
||
|
|
private void OnAmmoCountUpdate(EntityUid uid, BatteryAmmoProviderComponent component, UpdateAmmoCounterEvent args)
|
||
|
|
{
|
||
|
|
if (args.Control is not BoxesStatusControl boxes) return;
|
||
|
|
|
||
|
|
boxes.Update(component.Shots, component.Capacity);
|
||
|
|
}
|
||
|
|
|
||
|
|
private void OnControl(EntityUid uid, BatteryAmmoProviderComponent component, AmmoCounterControlEvent args)
|
||
|
|
{
|
||
|
|
args.Control = new BoxesStatusControl();
|
||
|
|
}
|
||
|
|
}
|