Files
tbd-station-14/Content.Client/Markers/MarkerSystem.cs
2022-05-12 21:11:50 +10:00

49 lines
1.2 KiB
C#

using Robust.Client.GameObjects;
using Robust.Shared.GameObjects;
namespace Content.Client.Markers
{
public sealed class MarkerSystem : EntitySystem
{
private bool _markersVisible;
public bool MarkersVisible
{
get => _markersVisible;
set
{
_markersVisible = value;
UpdateMarkers();
}
}
public override void Initialize()
{
base.Initialize();
SubscribeLocalEvent<MarkerComponent, ComponentStartup>(OnStartup);
}
private void OnStartup(EntityUid uid, MarkerComponent marker, ComponentStartup args)
{
UpdateVisibility(marker);
}
private void UpdateVisibility(MarkerComponent marker)
{
if (EntityManager.TryGetComponent(marker.Owner, out SpriteComponent? sprite))
{
sprite.Visible = MarkersVisible;
}
}
private void UpdateMarkers()
{
foreach (var markerComponent in EntityManager.EntityQuery<MarkerComponent>(true))
{
UpdateVisibility(markerComponent);
}
}
}
}