tommy/tbd-station-14
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Move all logic from SubFloorHideComponent to SubFloorHideSystem. (#3824)

Browse Source
This commit is contained in:
Vera Aguilera Puerto
2021-04-09 16:46:45 +02:00
committed by GitHub
parent a03e9686ea
commit 0827a95bee
2 changed files with 45 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
Content.Shared/GameObjects/EntitySystems/SubFloorHideSystem.cs
View File

@@ -1,6 +1,7 @@
#nullable enable
using Content.Shared.GameObjects.Components;
using Content.Shared.Maps;
using JetBrains.Annotations;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Map;
@@ -12,6 +13,7 @@ namespace Content.Shared.GameObjects.EntitySystems
/// <summary>
/// Entity system backing <see cref="SubFloorHideComponent"/>.
/// </summary>
[UsedImplicitly]
public class SubFloorHideSystem : EntitySystem
{
[Dependency] private readonly IMapManager _mapManager = default!;
@@ -52,18 +54,44 @@ namespace Content.Shared.GameObjects.EntitySystems
_mapManager.GridChanged += MapManagerOnGridChanged;
_mapManager.TileChanged += MapManagerOnTileChanged;
SubscribeLocalEvent<SubFloorHideDirtyEvent>(HandleDirtyEvent);
// TODO: Make this sane when EntityStarted becomes a directed event.
EntityManager.EntityStarted += OnEntityStarted;
SubscribeLocalEvent<SubFloorHideComponent, EntityTerminatingEvent>(OnSubFloorTerminating);
SubscribeLocalEvent<SubFloorHideComponent, SnapGridPositionChangedEvent>(OnSnapGridPositionChanged);
}
private void HandleDirtyEvent(SubFloorHideDirtyEvent ev)
public override void Shutdown()
{
if (!_mapManager.TryGetGrid(ev.Sender.Transform.GridID, out var grid))
{
return;
}
base.Shutdown();
var indices = grid.WorldToTile(ev.Sender.Transform.WorldPosition);
UpdateTile(grid, indices);
_mapManager.GridChanged -= MapManagerOnGridChanged;
_mapManager.TileChanged -= MapManagerOnTileChanged;
EntityManager.EntityStarted -= OnEntityStarted;
UnsubscribeLocalEvent<SubFloorHideComponent, EntityTerminatingEvent>(OnSubFloorTerminating);
UnsubscribeLocalEvent<SubFloorHideComponent, SnapGridPositionChangedEvent>(OnSnapGridPositionChanged);
}
private void OnEntityStarted(object? sender, EntityUid uid)
{
if (ComponentManager.HasComponent<SubFloorHideComponent>(uid))
{
UpdateEntity(uid);
}
}
private void OnSubFloorTerminating(EntityUid uid, SubFloorHideComponent component, EntityTerminatingEvent args)
{
UpdateEntity(uid);
}
private void OnSnapGridPositionChanged(EntityUid uid, SubFloorHideComponent component, SnapGridPositionChangedEvent ev)
{
// We do this directly instead of calling UpdateEntity.
if(_mapManager.TryGetGrid(ev.NewGrid, out var grid))
UpdateTile(grid, ev.Position);
}
private void MapManagerOnTileChanged(object? sender, TileChangedEventArgs e)
@@ -79,6 +107,14 @@ namespace Content.Shared.GameObjects.EntitySystems
}
}
private void UpdateEntity(EntityUid uid)
{
if (!ComponentManager.TryGetComponent(uid, out ITransformComponent? transform) ||
!_mapManager.TryGetGrid(transform.GridID, out var grid)) return;
UpdateTile(grid, grid.WorldToTile(transform.WorldPosition));
}
private void UpdateTile(IMapGrid grid, Vector2i position)
{
var tile = grid.GetTileRef(position);