94 lines
3.7 KiB
C#
94 lines
3.7 KiB
C#
using Content.Shared.IdentityManagement;
|
|
using Robust.Client.GameObjects;
|
|
using System.Linq;
|
|
using Robust.Client.UserInterface.Controllers;
|
|
|
|
namespace Content.Client.ContextMenu.UI
|
|
{
|
|
public sealed partial class EntityMenuUIController
|
|
{
|
|
public const int GroupingTypesCount = 2;
|
|
private int GroupingContextMenuType { get; set; }
|
|
public void OnGroupingChanged(int obj)
|
|
{
|
|
_context.Close();
|
|
GroupingContextMenuType = obj;
|
|
}
|
|
|
|
private List<List<EntityUid>> GroupEntities(IEnumerable<EntityUid> entities, int depth = 0)
|
|
{
|
|
if (GroupingContextMenuType == 0)
|
|
{
|
|
var newEntities = entities.GroupBy(e => Identity.Name(e, _entityManager) + (_entityManager.GetComponent<MetaDataComponent>(e).EntityPrototype?.ID ?? string.Empty)).ToList();
|
|
return newEntities.Select(grp => grp.ToList()).ToList();
|
|
}
|
|
else
|
|
{
|
|
var newEntities = entities.GroupBy(e => e, new PrototypeAndStatesContextMenuComparer(depth, _entityManager)).ToList();
|
|
return newEntities.Select(grp => grp.ToList()).ToList();
|
|
}
|
|
}
|
|
|
|
private sealed class PrototypeAndStatesContextMenuComparer : IEqualityComparer<EntityUid>
|
|
{
|
|
private static readonly List<Func<EntityUid, EntityUid, IEntityManager, bool>> EqualsList = new()
|
|
{
|
|
(a, b, entMan) => entMan.GetComponent<MetaDataComponent>(a).EntityPrototype!.ID == entMan.GetComponent<MetaDataComponent>(b).EntityPrototype!.ID,
|
|
(a, b, entMan) =>
|
|
{
|
|
entMan.TryGetComponent<SpriteComponent?>(a, out var spriteA);
|
|
entMan.TryGetComponent<SpriteComponent?>(b, out var spriteB);
|
|
|
|
if (spriteA == null || spriteB == null)
|
|
return spriteA == spriteB;
|
|
|
|
var xStates = spriteA.AllLayers.Where(e => e.Visible).Select(s => s.RsiState.Name);
|
|
var yStates = spriteB.AllLayers.Where(e => e.Visible).Select(s => s.RsiState.Name);
|
|
|
|
return xStates.OrderBy(t => t).SequenceEqual(yStates.OrderBy(t => t));
|
|
},
|
|
};
|
|
private static readonly List<Func<EntityUid, IEntityManager, int>> GetHashCodeList = new()
|
|
{
|
|
(e, entMan) => EqualityComparer<string>.Default.GetHashCode(entMan.GetComponent<MetaDataComponent>(e).EntityPrototype!.ID),
|
|
(e, entMan) =>
|
|
{
|
|
var hash = 0;
|
|
foreach (var element in entMan.GetComponent<SpriteComponent>(e).AllLayers.Where(obj => obj.Visible).Select(s => s.RsiState.Name))
|
|
{
|
|
hash ^= EqualityComparer<string>.Default.GetHashCode(element!);
|
|
}
|
|
return hash;
|
|
},
|
|
};
|
|
|
|
private static int Count => EqualsList.Count - 1;
|
|
|
|
private readonly int _depth;
|
|
private readonly IEntityManager _entMan;
|
|
public PrototypeAndStatesContextMenuComparer(int step = 0, IEntityManager? entMan = null)
|
|
{
|
|
IoCManager.Resolve(ref entMan);
|
|
|
|
_depth = step > Count ? Count : step;
|
|
_entMan = entMan;
|
|
}
|
|
|
|
public bool Equals(EntityUid x, EntityUid y)
|
|
{
|
|
if (x == default)
|
|
{
|
|
return y == default;
|
|
}
|
|
|
|
return y != default && EqualsList[_depth](x, y, _entMan);
|
|
}
|
|
|
|
public int GetHashCode(EntityUid e)
|
|
{
|
|
return GetHashCodeList[_depth](e, _entMan);
|
|
}
|
|
}
|
|
}
|
|
}
|