Files
tbd-station-14/Content.Client/MachineLinking/SignalSwitchVisualizer.cs
2021-12-05 18:09:01 +01:00

45 lines
1.2 KiB
C#

using Content.Shared.MachineLinking;
using JetBrains.Annotations;
using Robust.Client.GameObjects;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Serialization.Manager.Attributes;
namespace Content.Client.MachineLinking
{
[UsedImplicitly]
public class SignalSwitchVisualizer : AppearanceVisualizer
{
[DataField("layer")]
private int Layer { get; }
public override void InitializeEntity(EntityUid entity)
{
base.InitializeEntity(entity);
if (IoCManager.Resolve<IEntityManager>().TryGetComponent(entity, out SpriteComponent? sprite))
{
sprite.LayerMapReserveBlank(Layer);
}
}
public override void OnChangeData(AppearanceComponent component)
{
base.OnChangeData(component);
var entities = IoCManager.Resolve<IEntityManager>();
if (!entities.TryGetComponent(component.Owner, out SpriteComponent? sprite))
{
return;
}
if (!component.TryGetData(SignalSwitchVisuals.On, out bool on))
{
return;
}
sprite.LayerSetState(0, on ? "on" : "off");
}
}
}