43 lines
1.6 KiB
C#
43 lines
1.6 KiB
C#
using Content.Server.Storage.Components;
|
|
using Content.Shared.Rounding;
|
|
using Content.Shared.Storage.Components;
|
|
using Robust.Shared.Containers;
|
|
|
|
namespace Content.Server.Storage.EntitySystems;
|
|
|
|
public sealed class StorageFillVisualizerSystem : EntitySystem
|
|
{
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
SubscribeLocalEvent<StorageFillVisualizerComponent, ComponentInit>(OnInit);
|
|
SubscribeLocalEvent<StorageFillVisualizerComponent, EntInsertedIntoContainerMessage>(OnInserted);
|
|
SubscribeLocalEvent<StorageFillVisualizerComponent, EntRemovedFromContainerMessage>(OnRemoved);
|
|
}
|
|
|
|
private void OnInit(EntityUid uid, StorageFillVisualizerComponent component, ComponentInit args)
|
|
{
|
|
UpdateAppearance(uid, component: component);
|
|
}
|
|
|
|
private void OnInserted(EntityUid uid, StorageFillVisualizerComponent component, EntInsertedIntoContainerMessage args)
|
|
{
|
|
UpdateAppearance(uid, component: component);
|
|
}
|
|
|
|
private void OnRemoved(EntityUid uid, StorageFillVisualizerComponent component, EntRemovedFromContainerMessage args)
|
|
{
|
|
UpdateAppearance(uid, component: component);
|
|
}
|
|
|
|
private void UpdateAppearance(EntityUid uid, ServerStorageComponent? storage = null, AppearanceComponent? appearance = null,
|
|
StorageFillVisualizerComponent? component = null)
|
|
{
|
|
if (!Resolve(uid, ref storage, ref appearance, ref component, false))
|
|
return;
|
|
|
|
var level = ContentHelpers.RoundToEqualLevels(storage.StorageUsed, storage.StorageCapacityMax, component.MaxFillLevels);
|
|
appearance.SetData(StorageFillVisuals.FillLevel, level);
|
|
}
|
|
}
|