105 lines
3.8 KiB
C#
105 lines
3.8 KiB
C#
using Content.Client.SubFloor;
|
|
using Content.Shared.Atmos;
|
|
using Content.Shared.Atmos.Components;
|
|
using Content.Shared.Atmos.Piping;
|
|
using Content.Shared.SubFloor;
|
|
using JetBrains.Annotations;
|
|
using Robust.Client.GameObjects;
|
|
using Robust.Client.ResourceManagement;
|
|
|
|
namespace Content.Client.Atmos.EntitySystems;
|
|
|
|
[UsedImplicitly]
|
|
public sealed class AtmosPipeAppearanceSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly IResourceCache _resCache = default!;
|
|
[Dependency] private readonly SubFloorHideSystem _subfloorSys = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<PipeAppearanceComponent, ComponentInit>(OnInit);
|
|
SubscribeLocalEvent<PipeAppearanceComponent, AppearanceChangeEvent>(OnAppearanceChanged, after: new[] { typeof(SubFloorHideSystem) });
|
|
}
|
|
|
|
private void OnInit(EntityUid uid, PipeAppearanceComponent component, ComponentInit args)
|
|
{
|
|
if (!TryComp(uid, out SpriteComponent? sprite))
|
|
return;
|
|
|
|
if (!_resCache.TryGetResource(SharedSpriteComponent.TextureRoot / component.RsiPath, out RSIResource? rsi))
|
|
{
|
|
Logger.Error($"{nameof(AtmosPipeAppearanceSystem)} could not load to load RSI {component.RsiPath}.");
|
|
return;
|
|
}
|
|
|
|
foreach (PipeConnectionLayer layerKey in Enum.GetValues(typeof(PipeConnectionLayer)))
|
|
{
|
|
sprite.LayerMapReserveBlank(layerKey);
|
|
var layer = sprite.LayerMapGet(layerKey);
|
|
sprite.LayerSetRSI(layer, rsi.RSI);
|
|
var layerState = component.State;
|
|
sprite.LayerSetState(layer, layerState);
|
|
sprite.LayerSetDirOffset(layer, ToOffset(layerKey));
|
|
}
|
|
}
|
|
|
|
private void OnAppearanceChanged(EntityUid uid, PipeAppearanceComponent component, ref AppearanceChangeEvent args)
|
|
{
|
|
if (args.Sprite == null)
|
|
return;
|
|
|
|
if (!args.Sprite.Visible)
|
|
{
|
|
// This entity is probably below a floor and is not even visible to the user -> don't bother updating sprite data.
|
|
// Note that if the subfloor visuals change, then another AppearanceChangeEvent will get triggered.
|
|
return;
|
|
}
|
|
|
|
if (!args.Component.TryGetData(PipeColorVisuals.Color, out Color color))
|
|
color = Color.White;
|
|
|
|
if (!args.Component.TryGetData(PipeVisuals.VisualState, out PipeDirection worldConnectedDirections))
|
|
return;
|
|
|
|
// transform connected directions to local-coordinates
|
|
var connectedDirections = worldConnectedDirections.RotatePipeDirection(-Transform(uid).LocalRotation);
|
|
|
|
foreach (PipeConnectionLayer layerKey in Enum.GetValues(typeof(PipeConnectionLayer)))
|
|
{
|
|
if (!args.Sprite.LayerMapTryGet(layerKey, out var key))
|
|
continue;
|
|
|
|
var layer = args.Sprite[key];
|
|
var dir = (PipeDirection) layerKey;
|
|
var visible = connectedDirections.HasDirection(dir);
|
|
|
|
layer.Visible &= visible;
|
|
|
|
if (!visible) continue;
|
|
|
|
layer.Color = color;
|
|
}
|
|
}
|
|
|
|
private SpriteComponent.DirectionOffset ToOffset(PipeConnectionLayer layer)
|
|
{
|
|
return layer switch
|
|
{
|
|
PipeConnectionLayer.NorthConnection => SpriteComponent.DirectionOffset.Flip,
|
|
PipeConnectionLayer.EastConnection => SpriteComponent.DirectionOffset.CounterClockwise,
|
|
PipeConnectionLayer.WestConnection => SpriteComponent.DirectionOffset.Clockwise,
|
|
_ => SpriteComponent.DirectionOffset.None,
|
|
};
|
|
}
|
|
|
|
private enum PipeConnectionLayer : byte
|
|
{
|
|
NorthConnection = PipeDirection.North,
|
|
SouthConnection = PipeDirection.South,
|
|
EastConnection = PipeDirection.East,
|
|
WestConnection = PipeDirection.West,
|
|
}
|
|
}
|