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

Fix planet command being ran on existing maps (#21775)

Browse Source
This commit is contained in:
metalgearsloth
2023-12-11 19:51:02 +11:00
committed by GitHub
parent 5cf38a6eb3
commit 342b08418e
6 changed files with 148 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
Content.Server/Gateway/Systems/GatewayGeneratorSystem.cs
View File

@@ -225,7 +225,7 @@ public sealed class GatewayGeneratorSystem : EntitySystem
var layer = lootLayers[layerIdx]; var layer = lootLayers[layerIdx];
lootLayers.RemoveSwap(layerIdx); lootLayers.RemoveSwap(layerIdx);
_biome.AddMarkerLayer(biomeComp, layer.Id); _biome.AddMarkerLayer(ent.Owner, biomeComp, layer.Id);
} }
// - Mobs // - Mobs
@@ -237,7 +237,7 @@ public sealed class GatewayGeneratorSystem : EntitySystem
var layer = mobLayers[layerIdx]; var layer = mobLayers[layerIdx];
mobLayers.RemoveSwap(layerIdx); mobLayers.RemoveSwap(layerIdx);
_biome.AddMarkerLayer(biomeComp, layer.Id); _biome.AddMarkerLayer(ent.Owner, biomeComp, layer.Id);
} }
} }
} }

10
Content.Server/Parallax/BiomeSystem.Commands.cs
View File

@@ -29,14 +29,15 @@ public sealed partial class BiomeSystem
int.TryParse(args[0], out var mapInt); int.TryParse(args[0], out var mapInt);
var mapId = new MapId(mapInt); var mapId = new MapId(mapInt);
var mapUid = _mapManager.GetMapEntityId(mapId);
if (_mapManager.MapExists(mapId) || if (_mapManager.MapExists(mapId) ||
!TryComp<BiomeComponent>(_mapManager.GetMapEntityId(mapId), out var biome)) !TryComp<BiomeComponent>(mapUid, out var biome))
{ {
return; return;
} }
ClearTemplate(biome); ClearTemplate(mapUid, biome);
} }
private CompletionResult BiomeClearCallbackHelper(IConsoleShell shell, string[] args) private CompletionResult BiomeClearCallbackHelper(IConsoleShell shell, string[] args)
@@ -63,8 +64,9 @@ public sealed partial class BiomeSystem
} }
var mapId = new MapId(mapInt); var mapId = new MapId(mapInt);
var mapUid = _mapManager.GetMapEntityId(mapId);
if (!_mapManager.MapExists(mapId) || !TryComp<BiomeComponent>(_mapManager.GetMapEntityId(mapId), out var biome)) if (!_mapManager.MapExists(mapId) || !TryComp<BiomeComponent>(mapUid, out var biome))
{ {
return; return;
} }
@@ -81,7 +83,7 @@ public sealed partial class BiomeSystem
int.TryParse(args[3], out offset); int.TryParse(args[3], out offset);
} }
AddTemplate(biome, args[2], template, offset); AddTemplate(mapUid, biome, args[2], template, offset);
} }
private CompletionResult AddLayerCallbackHelp(IConsoleShell shell, string[] args) private CompletionResult AddLayerCallbackHelp(IConsoleShell shell, string[] args)

155
Content.Server/Parallax/BiomeSystem.cs
View File

@@ -6,6 +6,7 @@ using Content.Server.Atmos.EntitySystems;
using Content.Server.Decals; using Content.Server.Decals;
using Content.Server.Ghost.Roles.Components; using Content.Server.Ghost.Roles.Components;
using Content.Server.Shuttles.Events; using Content.Server.Shuttles.Events;
using Content.Server.Shuttles.Systems;
using Content.Shared.Atmos; using Content.Shared.Atmos;
using Content.Shared.Decals; using Content.Shared.Decals;
using Content.Shared.Gravity; using Content.Shared.Gravity;
@@ -21,6 +22,9 @@ using Robust.Shared.Console;
using Robust.Shared.Map; using Robust.Shared.Map;
using Robust.Shared.Map.Components; using Robust.Shared.Map.Components;
using Robust.Shared.Noise; using Robust.Shared.Noise;
using Robust.Shared.Physics;
using Robust.Shared.Physics.Components;
using Robust.Shared.Physics.Systems;
using Robust.Shared.Player; using Robust.Shared.Player;
using Robust.Shared.Prototypes; using Robust.Shared.Prototypes;
using Robust.Shared.Random; using Robust.Shared.Random;
@@ -41,8 +45,12 @@ public sealed partial class BiomeSystem : SharedBiomeSystem
[Dependency] private readonly AtmosphereSystem _atmos = default!; [Dependency] private readonly AtmosphereSystem _atmos = default!;
[Dependency] private readonly DecalSystem _decals = default!; [Dependency] private readonly DecalSystem _decals = default!;
[Dependency] private readonly SharedMapSystem _mapSystem = default!; [Dependency] private readonly SharedMapSystem _mapSystem = default!;
[Dependency] private readonly SharedPhysicsSystem _physics = default!;
[Dependency] private readonly SharedTransformSystem _transform = default!; [Dependency] private readonly SharedTransformSystem _transform = default!;
[Dependency] private readonly ShuttleSystem _shuttles = default!;
private EntityQuery<BiomeComponent> _biomeQuery;
private EntityQuery<FixturesComponent> _fixturesQuery;
private EntityQuery<TransformComponent> _xformQuery; private EntityQuery<TransformComponent> _xformQuery;
private readonly HashSet<EntityUid> _handledEntities = new(); private readonly HashSet<EntityUid> _handledEntities = new();
@@ -69,6 +77,8 @@ public sealed partial class BiomeSystem : SharedBiomeSystem
{ {
base.Initialize(); base.Initialize();
Log.Level = LogLevel.Debug; Log.Level = LogLevel.Debug;
_biomeQuery = GetEntityQuery<BiomeComponent>();
_fixturesQuery = GetEntityQuery<FixturesComponent>();
_xformQuery = GetEntityQuery<TransformComponent>(); _xformQuery = GetEntityQuery<TransformComponent>();
SubscribeLocalEvent<BiomeComponent, MapInitEvent>(OnBiomeMapInit); SubscribeLocalEvent<BiomeComponent, MapInitEvent>(OnBiomeMapInit);
SubscribeLocalEvent<FTLStartedEvent>(OnFTLStarted); SubscribeLocalEvent<FTLStartedEvent>(OnFTLStarted);
@@ -92,12 +102,12 @@ public sealed partial class BiomeSystem : SharedBiomeSystem
var query = AllEntityQuery<BiomeComponent>(); var query = AllEntityQuery<BiomeComponent>();
while (query.MoveNext(out var biome)) while (query.MoveNext(out var uid, out var biome))
{ {
if (biome.Template == null || !reloads.Modified.TryGetValue(biome.Template, out var proto)) if (biome.Template == null || !reloads.Modified.TryGetValue(biome.Template, out var proto))
continue; continue;
SetTemplate(biome, (BiomeTemplatePrototype) proto); SetTemplate(uid, biome, (BiomeTemplatePrototype) proto);
} }
} }
@@ -110,29 +120,64 @@ public sealed partial class BiomeSystem : SharedBiomeSystem
private void OnBiomeMapInit(EntityUid uid, BiomeComponent component, MapInitEvent args) private void OnBiomeMapInit(EntityUid uid, BiomeComponent component, MapInitEvent args)
{ {
if (component.Seed != -1) if (component.Seed == -1)
return; {
SetSeed(uid, component, _random.Next());
SetSeed(component, _random.Next());
} }
public void SetSeed(BiomeComponent component, int seed) var xform = Transform(uid);
var mapId = xform.MapID;
if (mapId != MapId.Nullspace && TryComp(uid, out MapGridComponent? mapGrid))
{
var setTiles = new List<(Vector2i Index, Tile tile)>();
foreach (var grid in _mapManager.GetAllMapGrids(mapId))
{
var gridUid = grid.Owner;
if (!_fixturesQuery.TryGetComponent(gridUid, out var fixtures))
continue;
// Don't want shuttles flying around now do we.
_shuttles.Disable(gridUid);
var pTransform = _physics.GetPhysicsTransform(gridUid);
foreach (var fixture in fixtures.Fixtures.Values)
{
for (var i = 0; i < fixture.Shape.ChildCount; i++)
{
var aabb = fixture.Shape.ComputeAABB(pTransform, i);
setTiles.Clear();
ReserveTiles(uid, aabb, setTiles);
}
}
}
}
}
public void SetSeed(EntityUid uid, BiomeComponent component, int seed, bool dirty = true)
{ {
component.Seed = seed; component.Seed = seed;
Dirty(component);
if (dirty)
Dirty(uid, component);
} }
public void ClearTemplate(BiomeComponent component) public void ClearTemplate(EntityUid uid, BiomeComponent component, bool dirty = true)
{ {
component.Layers.Clear(); component.Layers.Clear();
component.Template = null; component.Template = null;
Dirty(component);
if (dirty)
Dirty(uid, component);
} }
/// <summary> /// <summary>
/// Sets the <see cref="BiomeComponent.Template"/> and refreshes layers. /// Sets the <see cref="BiomeComponent.Template"/> and refreshes layers.
/// </summary> /// </summary>
public void SetTemplate(BiomeComponent component, BiomeTemplatePrototype template) public void SetTemplate(EntityUid uid, BiomeComponent component, BiomeTemplatePrototype template, bool dirty = true)
{ {
component.Layers.Clear(); component.Layers.Clear();
component.Template = template.ID; component.Template = template.ID;
@@ -142,13 +187,14 @@ public sealed partial class BiomeSystem : SharedBiomeSystem
component.Layers.Add(layer); component.Layers.Add(layer);
} }
Dirty(component); if (dirty)
Dirty(uid, component);
} }
/// <summary> /// <summary>
/// Adds the specified layer at the specified marker if it exists. /// Adds the specified layer at the specified marker if it exists.
/// </summary> /// </summary>
public void AddLayer(BiomeComponent component, string id, IBiomeLayer addedLayer, int seedOffset = 0) public void AddLayer(EntityUid uid, BiomeComponent component, string id, IBiomeLayer addedLayer, int seedOffset = 0)
{ {
for (var i = 0; i < component.Layers.Count; i++) for (var i = 0; i < component.Layers.Count; i++)
{ {
@@ -162,25 +208,19 @@ public sealed partial class BiomeSystem : SharedBiomeSystem
break; break;
} }
Dirty(component); Dirty(uid, component);
} }
public void AddMarkerLayer(BiomeComponent component, string marker) public void AddMarkerLayer(EntityUid uid, BiomeComponent component, string marker)
{ {
if (!ProtoManager.HasIndex<BiomeMarkerLayerPrototype>(marker))
{
// TODO: Log when we get a sawmill
return;
}
component.MarkerLayers.Add(marker); component.MarkerLayers.Add(marker);
Dirty(component); Dirty(uid, component);
} }
/// <summary> /// <summary>
/// Adds the specified template at the specified marker if it exists, withour overriding every layer. /// Adds the specified template at the specified marker if it exists, withour overriding every layer.
/// </summary> /// </summary>
public void AddTemplate(BiomeComponent component, string id, BiomeTemplatePrototype template, int seedOffset = 0) public void AddTemplate(EntityUid uid, BiomeComponent component, string id, BiomeTemplatePrototype template, int seedOffset = 0)
{ {
for (var i = 0; i < component.Layers.Count; i++) for (var i = 0; i < component.Layers.Count; i++)
{ {
@@ -199,7 +239,7 @@ public sealed partial class BiomeSystem : SharedBiomeSystem
break; break;
} }
Dirty(component); Dirty(uid, component);
} }
private void OnFTLStarted(ref FTLStartedEvent ev) private void OnFTLStarted(ref FTLStartedEvent ev)
@@ -258,7 +298,7 @@ public sealed partial class BiomeSystem : SharedBiomeSystem
foreach (var layer in markers) foreach (var layer in markers)
{ {
var proto = ProtoManager.Index<BiomeMarkerLayerPrototype>(layer); var proto = ProtoManager.Index(layer);
var enumerator = new ChunkIndicesEnumerator(area, proto.Size); var enumerator = new ChunkIndicesEnumerator(area, proto.Size);
while (enumerator.MoveNext(out var chunk)) while (enumerator.MoveNext(out var chunk))
@@ -273,8 +313,6 @@ public sealed partial class BiomeSystem : SharedBiomeSystem
public override void Update(float frameTime) public override void Update(float frameTime)
{ {
base.Update(frameTime); base.Update(frameTime);
var biomeQuery = GetEntityQuery<BiomeComponent>();
var xformQuery = GetEntityQuery<TransformComponent>();
var biomes = AllEntityQuery<BiomeComponent>(); var biomes = AllEntityQuery<BiomeComponent>();
while (biomes.MoveNext(out var biome)) while (biomes.MoveNext(out var biome))
@@ -286,17 +324,16 @@ public sealed partial class BiomeSystem : SharedBiomeSystem
// Get chunks in range // Get chunks in range
foreach (var pSession in Filter.GetAllPlayers(_playerManager)) foreach (var pSession in Filter.GetAllPlayers(_playerManager))
{ {
if (_xformQuery.TryGetComponent(pSession.AttachedEntity, out var xform) &&
if (xformQuery.TryGetComponent(pSession.AttachedEntity, out var xform) &&
_handledEntities.Add(pSession.AttachedEntity.Value) && _handledEntities.Add(pSession.AttachedEntity.Value) &&
biomeQuery.TryGetComponent(xform.MapUid, out var biome)) _biomeQuery.TryGetComponent(xform.MapUid, out var biome))
{ {
var worldPos = _transform.GetWorldPosition(xform, xformQuery); var worldPos = _transform.GetWorldPosition(xform);
AddChunksInRange(biome, worldPos); AddChunksInRange(biome, worldPos);
foreach (var layer in biome.MarkerLayers) foreach (var layer in biome.MarkerLayers)
{ {
var layerProto = ProtoManager.Index<BiomeMarkerLayerPrototype>(layer); var layerProto = ProtoManager.Index(layer);
AddMarkerChunksInRange(biome, worldPos, layerProto); AddMarkerChunksInRange(biome, worldPos, layerProto);
} }
} }
@@ -304,18 +341,18 @@ public sealed partial class BiomeSystem : SharedBiomeSystem
foreach (var viewer in pSession.ViewSubscriptions) foreach (var viewer in pSession.ViewSubscriptions)
{ {
if (!_handledEntities.Add(viewer) || if (!_handledEntities.Add(viewer) ||
!xformQuery.TryGetComponent(viewer, out xform) || !_xformQuery.TryGetComponent(viewer, out xform) ||
!biomeQuery.TryGetComponent(xform.MapUid, out biome)) !_biomeQuery.TryGetComponent(xform.MapUid, out biome))
{ {
continue; continue;
} }
var worldPos = _transform.GetWorldPosition(xform, xformQuery); var worldPos = _transform.GetWorldPosition(xform);
AddChunksInRange(biome, worldPos); AddChunksInRange(biome, worldPos);
foreach (var layer in biome.MarkerLayers) foreach (var layer in biome.MarkerLayers)
{ {
var layerProto = ProtoManager.Index<BiomeMarkerLayerPrototype>(layer); var layerProto = ProtoManager.Index(layer);
AddMarkerChunksInRange(biome, worldPos, layerProto); AddMarkerChunksInRange(biome, worldPos, layerProto);
} }
} }
@@ -326,7 +363,7 @@ public sealed partial class BiomeSystem : SharedBiomeSystem
while (loadBiomes.MoveNext(out var gridUid, out var biome, out var grid)) while (loadBiomes.MoveNext(out var gridUid, out var biome, out var grid))
{ {
// Load new chunks // Load new chunks
LoadChunks(biome, gridUid, grid, biome.Seed, xformQuery); LoadChunks(biome, gridUid, grid, biome.Seed, _xformQuery);
// Unload old chunks // Unload old chunks
UnloadChunks(biome, gridUid, grid, biome.Seed); UnloadChunks(biome, gridUid, grid, biome.Seed);
} }
@@ -867,10 +904,12 @@ public sealed partial class BiomeSystem : SharedBiomeSystem
if (!Resolve(mapUid, ref metadata)) if (!Resolve(mapUid, ref metadata))
return; return;
var biome = EnsureComp<BiomeComponent>(mapUid); EnsureComp<MapGridComponent>(mapUid);
var biome = (BiomeComponent) EntityManager.ComponentFactory.GetComponent(typeof(BiomeComponent));
seed ??= _random.Next(); seed ??= _random.Next();
SetSeed(biome, seed.Value); SetSeed(mapUid, biome, seed.Value, false);
SetTemplate(biome, biomeTemplate); SetTemplate(mapUid, biome, biomeTemplate, false);
AddComp(mapUid, biome, true);
Dirty(mapUid, biome, metadata); Dirty(mapUid, biome, metadata);
var gravity = EnsureComp<GravityComponent>(mapUid); var gravity = EnsureComp<GravityComponent>(mapUid);
@@ -902,7 +941,41 @@ public sealed partial class BiomeSystem : SharedBiomeSystem
}; };
_atmos.SetMapAtmosphere(mapUid, false, mixture, atmos); _atmos.SetMapAtmosphere(mapUid, false, mixture, atmos);
}
EnsureComp<MapGridComponent>(mapUid); /// <summary>
/// Sets the specified tiles as relevant and marks them as modified.
/// </summary>
public void ReserveTiles(EntityUid mapUid, Box2 bounds, List<(Vector2i Index, Tile Tile)> tiles, BiomeComponent? biome = null, MapGridComponent? mapGrid = null)
{
if (!Resolve(mapUid, ref biome, ref mapGrid, false))
return;
foreach (var tileSet in _mapSystem.GetLocalTilesIntersecting(mapUid, mapGrid, bounds, false))
{
Vector2i chunkOrigin;
HashSet<Vector2i> modified;
// Existing, ignore
if (_mapSystem.TryGetTileRef(mapUid, mapGrid, tileSet.GridIndices, out var existingRef) && !existingRef.Tile.IsEmpty)
{
chunkOrigin = SharedMapSystem.GetChunkIndices(tileSet.GridIndices, ChunkSize) * ChunkSize;
modified = biome.ModifiedTiles.GetOrNew(chunkOrigin);
modified.Add(tileSet.GridIndices);
continue;
}
if (!TryGetBiomeTile(tileSet.GridIndices, biome.Layers, biome.Seed, mapGrid, out var tile))
{
continue;
}
chunkOrigin = SharedMapSystem.GetChunkIndices(tileSet.GridIndices, ChunkSize) * ChunkSize;
modified = biome.ModifiedTiles.GetOrNew(chunkOrigin);
modified.Add(tileSet.GridIndices);
tiles.Add((tileSet.GridIndices, tile.Value));
}
_mapSystem.SetTiles(mapUid, mapGrid, tiles);
} }
} }

12
Content.Server/Salvage/SpawnSalvageMissionJob.cs
View File

@@ -106,14 +106,14 @@ public sealed class SpawnSalvageMissionJob : Job<bool>
{ {
var biome = _entManager.AddComponent<BiomeComponent>(mapUid); var biome = _entManager.AddComponent<BiomeComponent>(mapUid);
var biomeSystem = _entManager.System<BiomeSystem>(); var biomeSystem = _entManager.System<BiomeSystem>();
biomeSystem.SetTemplate(biome, _prototypeManager.Index<BiomeTemplatePrototype>(missionBiome.BiomePrototype)); biomeSystem.SetTemplate(mapUid, biome, _prototypeManager.Index<BiomeTemplatePrototype>(missionBiome.BiomePrototype));
biomeSystem.SetSeed(biome, mission.Seed); biomeSystem.SetSeed(mapUid, biome, mission.Seed);
_entManager.Dirty(biome); _entManager.Dirty(mapUid, biome);
// Gravity // Gravity
var gravity = _entManager.EnsureComponent<GravityComponent>(mapUid); var gravity = _entManager.EnsureComponent<GravityComponent>(mapUid);
gravity.Enabled = true; gravity.Enabled = true;
_entManager.Dirty(gravity, metadata); _entManager.Dirty(mapUid, gravity, metadata);
// Atmos // Atmos
var air = _prototypeManager.Index<SalvageAirMod>(mission.Air); var air = _prototypeManager.Index<SalvageAirMod>(mission.Air);
@@ -310,7 +310,7 @@ public sealed class SpawnSalvageMissionJob : Job<bool>
{ {
if (_entManager.TryGetComponent<BiomeComponent>(gridUid, out var biome)) if (_entManager.TryGetComponent<BiomeComponent>(gridUid, out var biome))
{ {
_biome.AddMarkerLayer(biome, biomeLoot.Prototype); _biome.AddMarkerLayer(gridUid, biome, biomeLoot.Prototype);
} }
} }
break; break;
@@ -318,7 +318,7 @@ public sealed class SpawnSalvageMissionJob : Job<bool>
{ {
if (_entManager.TryGetComponent<BiomeComponent>(gridUid, out var biome)) if (_entManager.TryGetComponent<BiomeComponent>(gridUid, out var biome))
{ {
_biome.AddTemplate(biome, "Loot", _prototypeManager.Index<BiomeTemplatePrototype>(biomeLoot.Prototype), i); _biome.AddTemplate(gridUid, biome, "Loot", _prototypeManager.Index<BiomeTemplatePrototype>(biomeLoot.Prototype), i);
} }
} }
break; break;

11
Content.Server/Shuttles/Systems/ShuttleSystem.FasterThanLight.cs
View File

@@ -274,7 +274,7 @@ public sealed partial class ShuttleSystem
if (TryComp(uid, out body)) if (TryComp(uid, out body))
{ {
if (shuttle != null) if (shuttle != null)
Enable(uid, body, shuttle); Enable(uid, component: body, shuttle: shuttle);
_physics.SetLinearVelocity(uid, new Vector2(0f, 20f), body: body); _physics.SetLinearVelocity(uid, new Vector2(0f, 20f), body: body);
_physics.SetAngularVelocity(uid, 0f, body: body); _physics.SetAngularVelocity(uid, 0f, body: body);
_physics.SetLinearDamping(body, 0f); _physics.SetLinearDamping(body, 0f);
@@ -365,11 +365,11 @@ public sealed partial class ShuttleSystem
// to event ordering and awake body shenanigans (at least for now). // to event ordering and awake body shenanigans (at least for now).
if (HasComp<MapGridComponent>(xform.MapUid)) if (HasComp<MapGridComponent>(xform.MapUid))
{ {
Disable(uid, body); Disable(uid, component: body);
} }
else if (shuttle != null) else if (shuttle != null)
{ {
Enable(uid, body, shuttle); Enable(uid, component: body, shuttle: shuttle);
} }
} }
@@ -701,6 +701,7 @@ public sealed partial class ShuttleSystem
var transform = _physics.GetPhysicsTransform(uid, xform, _xformQuery); var transform = _physics.GetPhysicsTransform(uid, xform, _xformQuery);
var aabbs = new List<Box2>(manager.Fixtures.Count); var aabbs = new List<Box2>(manager.Fixtures.Count);
var immune = new HashSet<EntityUid>(); var immune = new HashSet<EntityUid>();
var tileSet = new List<(Vector2i, Tile)>();
foreach (var fixture in manager.Fixtures.Values) foreach (var fixture in manager.Fixtures.Values)
{ {
@@ -712,6 +713,10 @@ public sealed partial class ShuttleSystem
aabb = aabb.Enlarged(0.2f); aabb = aabb.Enlarged(0.2f);
aabbs.Add(aabb); aabbs.Add(aabb);
// Handle clearing biome stuff as relevant.
tileSet.Clear();
_biomes.ReserveTiles(xform.MapUid.Value, aabb, tileSet);
foreach (var ent in _lookup.GetEntitiesIntersecting(xform.MapUid.Value, aabb, LookupFlags.Uncontained)) foreach (var ent in _lookup.GetEntitiesIntersecting(xform.MapUid.Value, aabb, LookupFlags.Uncontained))
{ {
if (ent == uid || immune.Contains(ent)) if (ent == uid || immune.Contains(ent))

26
Content.Server/Shuttles/Systems/ShuttleSystem.cs
View File

@@ -1,5 +1,6 @@
using Content.Server.Body.Systems; using Content.Server.Body.Systems;
using Content.Server.Doors.Systems; using Content.Server.Doors.Systems;
using Content.Server.Parallax;
using Content.Server.Shuttles.Components; using Content.Server.Shuttles.Components;
using Content.Server.Station.Systems; using Content.Server.Station.Systems;
using Content.Server.Stunnable; using Content.Server.Stunnable;
@@ -28,6 +29,7 @@ public sealed partial class ShuttleSystem : SharedShuttleSystem
[Dependency] private readonly IMapManager _mapManager = default!; [Dependency] private readonly IMapManager _mapManager = default!;
[Dependency] private readonly IRobustRandom _random = default!; [Dependency] private readonly IRobustRandom _random = default!;
[Dependency] private readonly ITileDefinitionManager _tileDefManager = default!; [Dependency] private readonly ITileDefinitionManager _tileDefManager = default!;
[Dependency] private readonly BiomeSystem _biomes = default!;
[Dependency] private readonly BodySystem _bobby = default!; [Dependency] private readonly BodySystem _bobby = default!;
[Dependency] private readonly DockingSystem _dockSystem = default!; [Dependency] private readonly DockingSystem _dockSystem = default!;
[Dependency] private readonly DoorSystem _doors = default!; [Dependency] private readonly DoorSystem _doors = default!;
@@ -37,6 +39,7 @@ public sealed partial class ShuttleSystem : SharedShuttleSystem
[Dependency] private readonly MapLoaderSystem _loader = default!; [Dependency] private readonly MapLoaderSystem _loader = default!;
[Dependency] private readonly MetaDataSystem _metadata = default!; [Dependency] private readonly MetaDataSystem _metadata = default!;
[Dependency] private readonly SharedAudioSystem _audio = default!; [Dependency] private readonly SharedAudioSystem _audio = default!;
[Dependency] private readonly SharedMapSystem _maps = default!;
[Dependency] private readonly SharedPhysicsSystem _physics = default!; [Dependency] private readonly SharedPhysicsSystem _physics = default!;
[Dependency] private readonly SharedTransformSystem _transform = default!; [Dependency] private readonly SharedTransformSystem _transform = default!;
[Dependency] private readonly ShuttleConsoleSystem _console = default!; [Dependency] private readonly ShuttleConsoleSystem _console = default!;
@@ -115,7 +118,7 @@ public sealed partial class ShuttleSystem : SharedShuttleSystem
if (component.Enabled) if (component.Enabled)
{ {
Enable(uid, physicsComponent, component); Enable(uid, component: physicsComponent, shuttle: component);
} }
} }
@@ -128,17 +131,18 @@ public sealed partial class ShuttleSystem : SharedShuttleSystem
if (component.Enabled) if (component.Enabled)
{ {
Enable(uid, physicsComponent, component); Enable(uid, component: physicsComponent, shuttle: component);
} }
else else
{ {
Disable(uid, physicsComponent); Disable(uid, component: physicsComponent);
} }
} }
private void Enable(EntityUid uid, PhysicsComponent component, ShuttleComponent shuttle) public void Enable(EntityUid uid, FixturesComponent? manager = null, PhysicsComponent? component = null, ShuttleComponent? shuttle = null)
{ {
FixturesComponent? manager = null; if (!Resolve(uid, ref manager, ref component, ref shuttle, false))
return;
_physics.SetBodyType(uid, BodyType.Dynamic, manager: manager, body: component); _physics.SetBodyType(uid, BodyType.Dynamic, manager: manager, body: component);
_physics.SetBodyStatus(component, BodyStatus.InAir); _physics.SetBodyStatus(component, BodyStatus.InAir);
@@ -147,9 +151,10 @@ public sealed partial class ShuttleSystem : SharedShuttleSystem
_physics.SetAngularDamping(component, shuttle.AngularDamping); _physics.SetAngularDamping(component, shuttle.AngularDamping);
} }
private void Disable(EntityUid uid, PhysicsComponent component) public void Disable(EntityUid uid, FixturesComponent? manager = null, PhysicsComponent? component = null)
{ {
FixturesComponent? manager = null; if (!Resolve(uid, ref manager, ref component, false))
return;
_physics.SetBodyType(uid, BodyType.Static, manager: manager, body: component); _physics.SetBodyType(uid, BodyType.Static, manager: manager, body: component);
_physics.SetBodyStatus(component, BodyStatus.OnGround); _physics.SetBodyStatus(component, BodyStatus.OnGround);
@@ -162,11 +167,6 @@ public sealed partial class ShuttleSystem : SharedShuttleSystem
if (EntityManager.GetComponent<MetaDataComponent>(uid).EntityLifeStage >= EntityLifeStage.Terminating) if (EntityManager.GetComponent<MetaDataComponent>(uid).EntityLifeStage >= EntityLifeStage.Terminating)
return; return;
if (!EntityManager.TryGetComponent(uid, out PhysicsComponent? physicsComponent)) Disable(uid);
{
return;
}
Disable(uid, physicsComponent);
} }
} }