54 lines
1.6 KiB
C#
54 lines
1.6 KiB
C#
using System.Linq;
|
|
using Content.Shared.Decals;
|
|
using Content.Shared.Sprite;
|
|
using Robust.Shared.GameStates;
|
|
using Robust.Shared.Prototypes;
|
|
using Robust.Shared.Random;
|
|
|
|
namespace Content.Server.Sprite;
|
|
|
|
public sealed class RandomSpriteSystem: SharedRandomSpriteSystem
|
|
{
|
|
[Dependency] private readonly IPrototypeManager _prototype = default!;
|
|
[Dependency] private readonly IRobustRandom _random = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
SubscribeLocalEvent<RandomSpriteComponent, ComponentGetState>(OnGetState);
|
|
SubscribeLocalEvent<RandomSpriteComponent, MapInitEvent>(OnMapInit);
|
|
}
|
|
|
|
private void OnMapInit(EntityUid uid, RandomSpriteComponent component, MapInitEvent args)
|
|
{
|
|
if (component.Selected.Count > 0)
|
|
return;
|
|
|
|
if (component.Available.Count == 0)
|
|
return;
|
|
|
|
var group = _random.Pick(component.Available);
|
|
component.Selected.EnsureCapacity(group.Count);
|
|
|
|
foreach (var layer in group)
|
|
{
|
|
Color? color = null;
|
|
|
|
if (!string.IsNullOrEmpty(layer.Value.Color))
|
|
color = _random.Pick(_prototype.Index<ColorPalettePrototype>(layer.Value.Color).Colors.Values);
|
|
|
|
component.Selected.Add(layer.Key, (layer.Value.State, color));
|
|
}
|
|
|
|
Dirty(component);
|
|
}
|
|
|
|
private void OnGetState(EntityUid uid, RandomSpriteComponent component, ref ComponentGetState args)
|
|
{
|
|
args.State = new RandomSpriteColorComponentState()
|
|
{
|
|
Selected = component.Selected,
|
|
};
|
|
}
|
|
}
|