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

Add a test for the weightless status (#2437)

Browse Source 
* Add wip test for weightless status

* Expand and fix weightless test
This commit is contained in:
DrSmugleaf
2020-10-30 02:38:21 +01:00
committed by GitHub
parent 86829b5191
commit 57729063b7
5 changed files with 120 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
Content.Server/GameObjects/EntitySystems/WeightlessSystem.cs
View File

@@ -7,6 +7,8 @@ using JetBrains.Annotations;
using Robust.Shared.GameObjects.Components.Map;
using Robust.Shared.GameObjects.EntitySystemMessages;
using Robust.Shared.GameObjects.Systems;
using Robust.Shared.Interfaces.Map;
using Robust.Shared.IoC;
using Robust.Shared.Map;
using Robust.Shared.Utility;
@@ -15,6 +17,8 @@ namespace Content.Server.GameObjects.EntitySystems
[UsedImplicitly]
public class WeightlessSystem : EntitySystem, IResettingEntitySystem
{
[Dependency] private readonly IMapManager _mapManager = default!;
private readonly Dictionary<GridId, List<ServerStatusEffectsComponent>> _statuses = new Dictionary<GridId, List<ServerStatusEffectsComponent>>();
public override void Initialize()
@@ -32,10 +36,22 @@ namespace Content.Server.GameObjects.EntitySystems
public void AddStatus(ServerStatusEffectsComponent status)
{
var grid = status.Owner.Transform.GridID;
var statuses = _statuses.GetOrNew(grid);
var gridId = status.Owner.Transform.GridID;
var statuses = _statuses.GetOrNew(gridId);
statuses.Add(status);
if (_mapManager.TryGetGrid(status.Owner.Transform.GridID, out var grid))
{
if (grid.HasGravity)
{
RemoveWeightless(status);
}
else
{
AddWeightless(status);
}
}
}
public void RemoveStatus(ServerStatusEffectsComponent status)