Make the material ejection control completely generic (#23308)

* Finish decoupling material ejection from lathes

* commented
This commit is contained in:
Nemanja
2024-01-01 02:16:25 -05:00
committed by GitHub
parent 10c09c7655
commit 2b01899d63
12 changed files with 282 additions and 166 deletions

View File

@@ -0,0 +1,92 @@
using System.Linq;
using Content.Shared.Materials;
using Robust.Client.AutoGenerated;
using Robust.Client.UserInterface.Controls;
using Robust.Client.UserInterface.XAML;
using Robust.Shared.Timing;
namespace Content.Client.Materials.UI;
/// <summary>
/// This widget is one row in the lathe eject menu.
/// </summary>
[GenerateTypedNameReferences]
public sealed partial class MaterialStorageControl : BoxContainer
{
[Dependency] private readonly IEntityManager _entityManager = default!;
private EntityUid? _owner;
private Dictionary<string, int> _currentMaterials = new();
public MaterialStorageControl()
{
RobustXamlLoader.Load(this);
IoCManager.InjectDependencies(this);
}
public void SetOwner(EntityUid owner)
{
_owner = owner;
}
protected override void FrameUpdate(FrameEventArgs args)
{
base.FrameUpdate(args);
if (_owner == null)
return;
if (!_entityManager.TryGetComponent<MaterialStorageComponent>(_owner, out var materialStorage))
{
Dispose();
return;
}
var canEject = materialStorage.CanEjectStoredMaterials;
var mats = materialStorage.Storage.Select(pair => (pair.Key.Id, pair.Value)).ToDictionary();
if (_currentMaterials.Equals(mats))
return;
var missing = new List<string>();
var extra = new List<string>();
foreach (var (mat, amount) in mats)
{
if (!_currentMaterials.ContainsKey(mat) ||
_currentMaterials[mat] == 0 && _currentMaterials[mat] != amount)
missing.Add(mat);
}
foreach (var (mat, amount) in _currentMaterials)
{
if (!mats.ContainsKey(mat) || amount == 0)
extra.Add(mat);
}
var children = new List<MaterialDisplay>();
children.AddRange(Children.OfType<MaterialDisplay>());
foreach (var display in children)
{
var mat = display.Material;
if (extra.Contains(mat))
{
RemoveChild(display);
continue;
}
if (!mats.TryGetValue(mat, out var newAmount))
continue;
display.UpdateVolume(newAmount);
}
foreach (var mat in missing)
{
var volume = mats[mat];
AddChild(new MaterialDisplay(_owner.Value, mat, volume, canEject));
}
_currentMaterials = mats;
NoMatsLabel.Visible = ChildCount == 1;
}
}