71 lines
2.5 KiB
C#
71 lines
2.5 KiB
C#
using Content.Client.AME.Components;
|
|
using Robust.Client.GameObjects;
|
|
using static Content.Shared.AME.SharedAMEShieldComponent;
|
|
|
|
namespace Content.Client.AME;
|
|
|
|
public sealed class AMEShieldingVisualizerSystem : VisualizerSystem<AMEShieldingVisualsComponent>
|
|
{
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<AMEShieldingVisualsComponent, ComponentInit>(OnComponentInit);
|
|
}
|
|
|
|
private void OnComponentInit(EntityUid uid, AMEShieldingVisualsComponent component, ComponentInit args)
|
|
{
|
|
if (TryComp<SpriteComponent>(uid, out var sprite))
|
|
{
|
|
sprite.LayerMapSet(AMEShieldingVisualsLayer.Core, sprite.AddLayerState("core"));
|
|
sprite.LayerSetVisible(AMEShieldingVisualsLayer.Core, false);
|
|
sprite.LayerMapSet(AMEShieldingVisualsLayer.CoreState, sprite.AddLayerState("core_weak"));
|
|
sprite.LayerSetVisible(AMEShieldingVisualsLayer.CoreState, false);
|
|
}
|
|
}
|
|
|
|
protected override void OnAppearanceChange(EntityUid uid, AMEShieldingVisualsComponent component, ref AppearanceChangeEvent args)
|
|
{
|
|
if (args.Sprite == null)
|
|
return;
|
|
|
|
if (AppearanceSystem.TryGetData<string>(uid, AMEShieldVisuals.Core, out var core, args.Component))
|
|
{
|
|
if (core == "isCore")
|
|
{
|
|
args.Sprite.LayerSetState(AMEShieldingVisualsLayer.Core, "core");
|
|
args.Sprite.LayerSetVisible(AMEShieldingVisualsLayer.Core, true);
|
|
|
|
}
|
|
else
|
|
{
|
|
args.Sprite.LayerSetVisible(AMEShieldingVisualsLayer.Core, false);
|
|
}
|
|
}
|
|
|
|
if (AppearanceSystem.TryGetData<string>(uid, AMEShieldVisuals.CoreState, out var coreState, args.Component))
|
|
{
|
|
switch (coreState)
|
|
{
|
|
case "weak":
|
|
args.Sprite.LayerSetState(AMEShieldingVisualsLayer.CoreState, "core_weak");
|
|
args.Sprite.LayerSetVisible(AMEShieldingVisualsLayer.CoreState, true);
|
|
break;
|
|
case "strong":
|
|
args.Sprite.LayerSetState(AMEShieldingVisualsLayer.CoreState, "core_strong");
|
|
args.Sprite.LayerSetVisible(AMEShieldingVisualsLayer.CoreState, true);
|
|
break;
|
|
case "off":
|
|
args.Sprite.LayerSetVisible(AMEShieldingVisualsLayer.CoreState, false);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public enum AMEShieldingVisualsLayer : byte
|
|
{
|
|
Core,
|
|
CoreState,
|
|
}
|