116 lines
4.5 KiB
C#
116 lines
4.5 KiB
C#
using Content.Shared.Chemistry;
|
|
using Content.Shared.Chemistry.Reagent;
|
|
using Content.Shared.Rounding;
|
|
using Robust.Client.GameObjects;
|
|
using Robust.Shared.Prototypes;
|
|
|
|
namespace Content.Client.Chemistry.Visualizers;
|
|
|
|
public sealed class SolutionContainerVisualsSystem : VisualizerSystem<SolutionContainerVisualsComponent>
|
|
{
|
|
[Dependency] private readonly IPrototypeManager _prototype = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
SubscribeLocalEvent<SolutionContainerVisualsComponent, MapInitEvent>(OnMapInit);
|
|
}
|
|
|
|
private void OnMapInit(EntityUid uid, SolutionContainerVisualsComponent component, MapInitEvent args)
|
|
{
|
|
var meta = MetaData(uid);
|
|
component.InitialName = meta.EntityName;
|
|
component.InitialDescription = meta.EntityDescription;
|
|
}
|
|
|
|
protected override void OnAppearanceChange(EntityUid uid, SolutionContainerVisualsComponent component, ref AppearanceChangeEvent args)
|
|
{
|
|
// Check if the solution that was updated is the one set as represented
|
|
if (!string.IsNullOrEmpty(component.SolutionName))
|
|
{
|
|
if (AppearanceSystem.TryGetData<string>(uid, SolutionContainerVisuals.SolutionName, out var name,
|
|
args.Component) && name != component.SolutionName)
|
|
{
|
|
return;
|
|
}
|
|
}
|
|
|
|
if (!AppearanceSystem.TryGetData<float>(uid, SolutionContainerVisuals.FillFraction, out var fraction, args.Component))
|
|
return;
|
|
|
|
if (args.Sprite == null)
|
|
return;
|
|
|
|
if (!args.Sprite.LayerMapTryGet(component.FillLayer, out var fillLayer))
|
|
return;
|
|
|
|
// Currently some solution methods such as overflowing will try to update appearance with a
|
|
// volume greater than the max volume. We'll clamp it so players don't see
|
|
// a giant error sign and error for debug.
|
|
if (fraction > 1f)
|
|
{
|
|
Logger.Error("Attempted to set solution container visuals volume ratio on " + ToPrettyString(uid) + " to a value greater than 1. Volume should never be greater than max volume!");
|
|
fraction = 1f;
|
|
}
|
|
if (component.Metamorphic)
|
|
{
|
|
if (args.Sprite.LayerMapTryGet(component.BaseLayer, out var baseLayer))
|
|
{
|
|
var hasOverlay = args.Sprite.LayerMapTryGet(component.OverlayLayer, out var overlayLayer);
|
|
|
|
if (AppearanceSystem.TryGetData<string>(uid, SolutionContainerVisuals.BaseOverride,
|
|
out var baseOverride,
|
|
args.Component))
|
|
{
|
|
_prototype.TryIndex<ReagentPrototype>(baseOverride, out var reagentProto);
|
|
|
|
if (reagentProto?.MetamorphicSprite is { } sprite)
|
|
{
|
|
args.Sprite.LayerSetSprite(baseLayer, sprite);
|
|
args.Sprite.LayerSetVisible(fillLayer, false);
|
|
if (hasOverlay)
|
|
args.Sprite.LayerSetVisible(overlayLayer, false);
|
|
return;
|
|
}
|
|
else
|
|
{
|
|
if (hasOverlay)
|
|
args.Sprite.LayerSetVisible(overlayLayer, true);
|
|
if (component.MetamorphicDefaultSprite != null)
|
|
args.Sprite.LayerSetSprite(baseLayer, component.MetamorphicDefaultSprite);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
int closestFillSprite = ContentHelpers.RoundToLevels(fraction, 1, component.MaxFillLevels + 1);
|
|
|
|
if (closestFillSprite > 0)
|
|
{
|
|
if (component.FillBaseName == null)
|
|
return;
|
|
|
|
args.Sprite.LayerSetVisible(fillLayer, true);
|
|
|
|
var stateName = component.FillBaseName + closestFillSprite;
|
|
args.Sprite.LayerSetState(fillLayer, stateName);
|
|
|
|
if (component.ChangeColor && AppearanceSystem.TryGetData<Color>(uid, SolutionContainerVisuals.Color, out var color, args.Component))
|
|
args.Sprite.LayerSetColor(fillLayer, color);
|
|
}
|
|
else
|
|
{
|
|
if (component.EmptySpriteName == null)
|
|
args.Sprite.LayerSetVisible(fillLayer, false);
|
|
else
|
|
{
|
|
args.Sprite.LayerSetState(fillLayer, component.EmptySpriteName);
|
|
if (component.ChangeColor)
|
|
args.Sprite.LayerSetColor(fillLayer, component.EmptySpriteColor);
|
|
}
|
|
}
|
|
|
|
|
|
}
|
|
}
|