potato battery visualizer exception fix (#12862)

This commit is contained in:
Nemanja
2022-12-04 18:50:47 -05:00
committed by GitHub
parent dcffadb526
commit 64fbb7908a

View File

@@ -15,17 +15,21 @@ public sealed class PowerCellSystem : SharedPowerCellSystem
private void OnPowerCellVisualsChange(EntityUid uid, PowerCellVisualsComponent component, ref AppearanceChangeEvent args)
{
if (args.Sprite == null) return;
if (args.Sprite == null)
return;
if (!args.Sprite.TryGetLayer((int) PowerCellVisualLayers.Unshaded, out var unshadedLayer))
return;
if (args.Component.TryGetData(PowerCellVisuals.ChargeLevel, out byte level))
{
if (level == 0)
{
args.Sprite.LayerSetVisible(PowerCellVisualLayers.Unshaded, false);
unshadedLayer.Visible = false;
return;
}
args.Sprite.LayerSetVisible(PowerCellVisualLayers.Unshaded, true);
unshadedLayer.Visible = true;
args.Sprite.LayerSetState(PowerCellVisualLayers.Unshaded, $"o{level}");
}
}