48 lines
1.5 KiB
C#
48 lines
1.5 KiB
C#
using Content.Shared.Weapons.Ranged.Components;
|
|
using Robust.Shared.Map;
|
|
|
|
namespace Content.Client.Weapons.Ranged.Systems;
|
|
|
|
public sealed partial class GunSystem
|
|
{
|
|
protected override void InitializeBallistic()
|
|
{
|
|
base.InitializeBallistic();
|
|
SubscribeLocalEvent<BallisticAmmoProviderComponent, UpdateAmmoCounterEvent>(OnBallisticAmmoCount);
|
|
}
|
|
|
|
private void OnBallisticAmmoCount(EntityUid uid, BallisticAmmoProviderComponent component, UpdateAmmoCounterEvent args)
|
|
{
|
|
if (args.Control is DefaultStatusControl control)
|
|
{
|
|
control.Update(GetBallisticShots(component), component.Capacity);
|
|
}
|
|
}
|
|
|
|
protected override void Cycle(BallisticAmmoProviderComponent component, MapCoordinates coordinates)
|
|
{
|
|
if (!Timing.IsFirstTimePredicted) return;
|
|
|
|
EntityUid? ent = null;
|
|
|
|
// TODO: Combine with TakeAmmo
|
|
if (component.Entities.Count > 0)
|
|
{
|
|
var existing = component.Entities[^1];
|
|
component.Entities.RemoveAt(component.Entities.Count - 1);
|
|
|
|
component.Container.Remove(existing);
|
|
EnsureComp<AmmoComponent>(existing);
|
|
}
|
|
else if (component.UnspawnedCount > 0)
|
|
{
|
|
component.UnspawnedCount--;
|
|
ent = Spawn(component.FillProto, coordinates);
|
|
EnsureComp<AmmoComponent>(ent.Value);
|
|
}
|
|
|
|
if (ent != null && ent.Value.IsClientSide())
|
|
Del(ent.Value);
|
|
}
|
|
}
|