civ14/Content.Client/Fluids/UI/AbsorbentItemStatus.xaml.cs

using System.Linq;
using Content.Client.UserInterface.Controls;
using Content.Shared.Fluids;
using Robust.Client.AutoGenerated;
using Robust.Client.UserInterface;
using Robust.Client.UserInterface.XAML;
using Robust.Shared.Timing;
using Robust.Shared.Utility;

namespace Content.Client.Fluids.UI
{
    [GenerateTypedNameReferences]
    public sealed partial class AbsorbentItemStatus : SplitBar
    {
        private readonly IEntityManager _entManager;
        private readonly EntityUid _uid;
        private Dictionary<Color, float> _progress = new();

        public AbsorbentItemStatus(EntityUid uid, IEntityManager entManager)
        {
            RobustXamlLoader.Load(this);
            _uid = uid;
            _entManager = entManager;
        }

        protected override void FrameUpdate(FrameEventArgs args)
        {
            base.FrameUpdate(args);
            if (!_entManager.TryGetComponent<AbsorbentComponent>(_uid, out var absorbent))
                return;

            var oldProgress = _progress.ShallowClone();
            _progress.Clear();

            foreach (var item in absorbent.Progress)
            {
                _progress[item.Key] = item.Value;
            }

            if (oldProgress.OrderBy(x => x.Key.ToArgb()).SequenceEqual(_progress))
                return;

            Bar.Clear();

            foreach (var (key, value) in absorbent.Progress)
            {
                Bar.AddEntry(value, key);
            }
        }
    }
}