40 lines
1.3 KiB
C#
40 lines
1.3 KiB
C#
using Content.Shared.Weapons.Marker;
|
|
using Robust.Client.GameObjects;
|
|
using Robust.Shared.Timing;
|
|
|
|
namespace Content.Client.Weapons.Marker;
|
|
|
|
public sealed class DamageMarkerSystem : SharedDamageMarkerSystem
|
|
{
|
|
[Dependency] private readonly IGameTiming _timing = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
SubscribeLocalEvent<DamageMarkerComponent, ComponentStartup>(OnMarkerStartup);
|
|
SubscribeLocalEvent<DamageMarkerComponent, ComponentShutdown>(OnMarkerShutdown);
|
|
}
|
|
|
|
private void OnMarkerStartup(EntityUid uid, DamageMarkerComponent component, ComponentStartup args)
|
|
{
|
|
if (!_timing.ApplyingState || component.Effect == null || !TryComp<SpriteComponent>(uid, out var sprite))
|
|
return;
|
|
|
|
var layer = sprite.LayerMapReserveBlank(DamageMarkerKey.Key);
|
|
sprite.LayerSetState(layer, component.Effect.RsiState, component.Effect.RsiPath);
|
|
}
|
|
|
|
private void OnMarkerShutdown(EntityUid uid, DamageMarkerComponent component, ComponentShutdown args)
|
|
{
|
|
if (!_timing.ApplyingState || !TryComp<SpriteComponent>(uid, out var sprite) || !sprite.LayerMapTryGet(DamageMarkerKey.Key, out var weh))
|
|
return;
|
|
|
|
sprite.RemoveLayer(weh);
|
|
}
|
|
|
|
private enum DamageMarkerKey : byte
|
|
{
|
|
Key
|
|
}
|
|
}
|