64 lines
2.0 KiB
C#
64 lines
2.0 KiB
C#
using Content.Shared.Sprite;
|
|
using Robust.Client.GameObjects;
|
|
using Robust.Shared.GameStates;
|
|
using Robust.Shared.Reflection;
|
|
|
|
namespace Content.Client.Sprite;
|
|
|
|
public sealed class RandomSpriteSystem : SharedRandomSpriteSystem
|
|
{
|
|
[Dependency] private readonly IReflectionManager _reflection = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
SubscribeLocalEvent<RandomSpriteComponent, ComponentHandleState>(OnHandleState);
|
|
}
|
|
|
|
private void OnHandleState(EntityUid uid, RandomSpriteComponent component, ref ComponentHandleState args)
|
|
{
|
|
if (args.Current is not RandomSpriteColorComponentState state)
|
|
return;
|
|
|
|
if (state.Selected.Equals(component.Selected))
|
|
return;
|
|
|
|
component.Selected.Clear();
|
|
component.Selected.EnsureCapacity(state.Selected.Count);
|
|
|
|
foreach (var layer in state.Selected)
|
|
{
|
|
component.Selected.Add(layer.Key, layer.Value);
|
|
}
|
|
|
|
UpdateAppearance(uid, component);
|
|
}
|
|
|
|
private void UpdateAppearance(EntityUid uid, RandomSpriteComponent component, SpriteComponent? sprite = null)
|
|
{
|
|
if (!Resolve(uid, ref sprite, false))
|
|
return;
|
|
|
|
foreach (var layer in component.Selected)
|
|
{
|
|
int index;
|
|
if (_reflection.TryParseEnumReference(layer.Key, out var @enum))
|
|
{
|
|
if (!sprite.LayerMapTryGet(@enum, out index, logError: true))
|
|
return;
|
|
}
|
|
else if (!sprite.LayerMapTryGet(layer.Key, out index))
|
|
{
|
|
if (layer.Key is not string strKey || !int.TryParse(strKey, out index))
|
|
{
|
|
Logger.Error($"Invalid key `{layer.Key}` for entity with random sprite {ToPrettyString(uid)}");
|
|
return;
|
|
}
|
|
}
|
|
|
|
sprite.LayerSetState(index, layer.Value.State);
|
|
sprite.LayerSetColor(index, layer.Value.Color ?? Color.White);
|
|
}
|
|
}
|
|
}
|