60 lines
1.9 KiB
C#
60 lines
1.9 KiB
C#
using Content.Shared.Mobs.Components;
|
|
using Content.Shared.Xenoarchaeology.Artifact.XAE.Components;
|
|
using Robust.Shared.Random;
|
|
using Robust.Shared.Timing;
|
|
|
|
namespace Content.Shared.Xenoarchaeology.Artifact.XAE;
|
|
|
|
/// <summary>
|
|
/// System that handles mob entities spacial shuffling effect.
|
|
/// </summary>
|
|
public sealed class XAEShuffleSystem : BaseXAESystem<XAEShuffleComponent>
|
|
{
|
|
[Dependency] private readonly EntityLookupSystem _lookup = default!;
|
|
[Dependency] private readonly IRobustRandom _random = default!;
|
|
[Dependency] private readonly SharedTransformSystem _xform = default!;
|
|
[Dependency] private readonly IGameTiming _timing = default!;
|
|
|
|
private EntityQuery<MobStateComponent> _mobState;
|
|
|
|
/// <summary> Pre-allocated and re-used collection.</summary>
|
|
private readonly HashSet<EntityUid> _entities= new();
|
|
|
|
/// <inheritdoc />
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
_mobState = GetEntityQuery<MobStateComponent>();
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
protected override void OnActivated(Entity<XAEShuffleComponent> ent, ref XenoArtifactNodeActivatedEvent args)
|
|
{
|
|
if(!_timing.IsFirstTimePredicted)
|
|
return;
|
|
|
|
List<Entity<TransformComponent>> toShuffle = new();
|
|
_entities.Clear();
|
|
_lookup.GetEntitiesInRange(ent.Owner, ent.Comp.Radius, _entities, LookupFlags.Dynamic | LookupFlags.Sundries);
|
|
foreach (var entity in _entities)
|
|
{
|
|
if (!_mobState.HasComponent(entity))
|
|
continue;
|
|
|
|
var xform = Transform(entity);
|
|
|
|
toShuffle.Add((entity, xform));
|
|
}
|
|
|
|
_random.Shuffle(toShuffle);
|
|
|
|
while (toShuffle.Count > 1)
|
|
{
|
|
var ent1 = _random.PickAndTake(toShuffle);
|
|
var ent2 = _random.PickAndTake(toShuffle);
|
|
_xform.SwapPositions((ent1, ent1), (ent2, ent2));
|
|
}
|
|
}
|
|
}
|