emitter visuals update (#13382)

2023-01-08 22:42:31 -05:00
parent 692079745f
commit 2933f030d9
7 changed files with 210 additions and 186 deletions
--- a/Content.Client/Singularity/Systems/EmitterSystem.cs
+++ b/Content.Client/Singularity/Systems/EmitterSystem.cs
@@ -0,0 +1,59 @@
+using Content.Client.Storage.Visualizers;
+using Content.Shared.Singularity.Components;
+using Content.Shared.Singularity.EntitySystems;
+using Content.Shared.Storage;
+using Robust.Client.GameObjects;
+
+namespace Content.Client.Singularity.Systems;
+
+public sealed class EmitterSystem : SharedEmitterSystem
+{
+    [Dependency] private readonly AppearanceSystem _appearance = default!;
+
+    /// <inheritdoc/>
+    public override void Initialize()
+    {
+        SubscribeLocalEvent<EmitterComponent, AppearanceChangeEvent>(OnAppearanceChange);
+    }
+
+    private void OnAppearanceChange(EntityUid uid, EmitterComponent component, ref AppearanceChangeEvent args)
+    {
+        if (args.Sprite == null)
+            return;
+
+        if (args.Sprite.LayerMapTryGet(StorageVisualLayers.Lock, out var lockLayer))
+        {
+            if (!_appearance.TryGetData(uid, StorageVisuals.Locked, out bool locked, args.Component))
+                locked = false;
+
+            args.Sprite.LayerSetVisible(lockLayer, locked);
+        }
+
+        if (!_appearance.TryGetData(uid, EmitterVisuals.VisualState, out EmitterVisualState state, args.Component))
+            state = EmitterVisualState.Off;
+
+        if (!args.Sprite.LayerMapTryGet(EmitterVisualLayers.Lights, out var layer))
+            return;
+
+        switch (state)
+        {
+            case EmitterVisualState.On:
+                if (component.OnState == null)
+                    break;
+                args.Sprite.LayerSetVisible(layer, true);
+                args.Sprite.LayerSetState(layer, component.OnState);
+                break;
+            case EmitterVisualState.Underpowered:
+                if (component.UnderpoweredState == null)
+                    break;
+                args.Sprite.LayerSetVisible(layer, true);
+                args.Sprite.LayerSetState(layer, component.UnderpoweredState);
+                break;
+            case EmitterVisualState.Off:
+                args.Sprite.LayerSetVisible(layer, false);
+                break;
+            default:
+                throw new ArgumentOutOfRangeException();
+        }
+    }
+}