52 lines
1.4 KiB
C#
52 lines
1.4 KiB
C#
using Content.Shared.GameObjects.Components.Fluids;
|
|
using Robust.Client.GameObjects;
|
|
using Robust.Client.Interfaces.GameObjects.Components;
|
|
using Robust.Shared.Utility;
|
|
using YamlDotNet.RepresentationModel;
|
|
|
|
namespace Content.Client.GameObjects.Components.Fluids
|
|
{
|
|
public class SprayVisualizer : AppearanceVisualizer
|
|
{
|
|
private string _safetyOnState;
|
|
private string _safetyOffState;
|
|
|
|
public override void LoadData(YamlMappingNode node)
|
|
{
|
|
base.LoadData(node);
|
|
|
|
if (node.TryGetNode("safety_on_state", out var safetyOn))
|
|
{
|
|
_safetyOnState = safetyOn.AsString();
|
|
}
|
|
|
|
if (node.TryGetNode("safety_off_state", out var safetyOff))
|
|
{
|
|
_safetyOffState = safetyOff.AsString();
|
|
}
|
|
}
|
|
|
|
public override void OnChangeData(AppearanceComponent component)
|
|
{
|
|
base.OnChangeData(component);
|
|
|
|
if (component.TryGetData<bool>(SprayVisuals.Safety, out var safety))
|
|
{
|
|
SetSafety(component, safety);
|
|
}
|
|
}
|
|
|
|
private void SetSafety(AppearanceComponent component, bool safety)
|
|
{
|
|
var sprite = component.Owner.GetComponent<ISpriteComponent>();
|
|
|
|
sprite.LayerSetState(SprayVisualLayers.Base, safety ? _safetyOnState : _safetyOffState);
|
|
}
|
|
}
|
|
|
|
public enum SprayVisualLayers : byte
|
|
{
|
|
Base
|
|
}
|
|
}
|