switch (selection)
{
case ObjectsTabSelection.Stations:
- entities.AddRange(_entityManager.EntitySysManager.GetEntitySystem<StationSystem>().Stations);
+ entities.AddRange(_entityManager.EntitySysManager.GetEntitySystem<StationSystem>().GetStationNames());
break;
case ObjectsTabSelection.Grids:
{
namespace Content.Client.Station;
-/// <summary>
-/// This handles letting the client know stations are a thing. Only really used by an admin menu.
-/// </summary>
-public sealed partial class StationSystem : SharedStationSystem
-{
- private readonly List<(string Name, NetEntity Entity)> _stations = new();
-
- /// <summary>
- /// All stations that currently exist.
- /// </summary>
- /// <remarks>
- /// I'd have this just invoke an entity query, but we're on the client and the client barely knows about stations.
- /// </remarks>
- // TODO: Stations have a global PVS override now, this can probably be changed into a query.
- public IReadOnlyList<(string Name, NetEntity Entity)> Stations => _stations;
-
- /// <inheritdoc/>
- public override void Initialize()
- {
- base.Initialize();
-
- SubscribeNetworkEvent<StationsUpdatedEvent>(StationsUpdated);
- }
-
- private void StationsUpdated(StationsUpdatedEvent ev)
- {
- _stations.Clear();
- // TODO this needs to be done in component states and with the Ensure() methods
- _stations.AddRange(ev.Stations);
- }
-}
+/// <inheritdoc/>
+public sealed partial class StationSystem : SharedStationSystem;
using Content.Server.GameTicking;
using Content.Server.Shuttles.Components;
using Content.Server.Shuttles.Systems;
-using Content.Server.Station.Components;
using Content.Shared.CCVar;
using Content.Shared.Shuttles.Components;
+using Content.Shared.Station.Components;
using Robust.Shared.GameObjects;
using Robust.Shared.Map.Components;
using System.Linq;
using System.Numerics;
using Content.Server.Administration.Components;
-using Content.Server.Atmos;
-using Content.Server.Atmos.Components;
using Content.Server.Cargo.Components;
using Content.Server.Doors.Systems;
using Content.Server.Hands.Systems;
using Content.Server.Power.Components;
using Content.Server.Power.EntitySystems;
using Content.Server.Stack;
-using Content.Server.Station.Components;
using Content.Server.Station.Systems;
using Content.Server.Weapons.Ranged.Systems;
using Content.Shared.Access;
using Content.Shared.Inventory;
using Content.Shared.PDA;
using Content.Shared.Stacks;
+using Content.Shared.Station.Components;
using Content.Shared.Verbs;
using Content.Shared.Weapons.Ranged.Components;
using Robust.Server.Physics;
using Content.Server.Anomaly.Components;
using Content.Server.Power.EntitySystems;
-using Content.Server.Station.Components;
using Content.Shared.Anomaly;
using Content.Shared.CCVar;
using Content.Shared.Materials;
var xform = Transform(uid);
if (_station.GetStationInMap(xform.MapID) is not { } station ||
- !TryComp<StationDataComponent>(station, out var data) ||
- _station.GetLargestGrid(data) is not { } grid)
+ _station.GetLargestGrid(station) is not { } grid)
{
if (xform.GridUid == null)
return;
using System.Linq;
-using Content.Server.Station.Components;
using Content.Shared.Cargo;
using Content.Shared.Cargo.Components;
using Content.Shared.Cargo.Prototypes;
+using Content.Shared.Station.Components;
using Robust.Shared.Prototypes;
namespace Content.Server.Cargo.Components;
using System.Diagnostics.CodeAnalysis;
using System.Linq;
using Content.Server.Cargo.Components;
-using Content.Server.Station.Components;
using Content.Shared.Cargo;
using Content.Shared.Cargo.BUI;
using Content.Shared.Cargo.Components;
using Content.Shared.Interaction;
using Content.Shared.Labels.Components;
using Content.Shared.Paper;
+using Content.Shared.Station.Components;
using JetBrains.Annotations;
-using Robust.Shared.Audio;
using Robust.Shared.Map;
using Robust.Shared.Prototypes;
using Robust.Shared.Timing;
using Content.Server.Cargo.Components;
using Content.Server.Power.Components;
using Content.Server.Power.EntitySystems;
-using Content.Server.Station.Components;
using Content.Shared.Cargo;
using Content.Shared.Cargo.Components;
using Content.Shared.DeviceLinking;
using Content.Shared.Power;
+using Content.Shared.Station.Components;
using Robust.Shared.Audio;
using Robust.Shared.Random;
using Robust.Shared.Utility;
using Content.Server.GameTicking;
using Content.Server.Speech.EntitySystems;
using Content.Server.Speech.Prototypes;
-using Content.Server.Station.Components;
using Content.Server.Station.Systems;
using Content.Shared.ActionBlocker;
using Content.Shared.Administration;
using Content.Shared.Players;
using Content.Shared.Players.RateLimiting;
using Content.Shared.Radio;
+using Content.Shared.Station.Components;
using Content.Shared.Whitelist;
using Robust.Server.Player;
using Robust.Shared.Audio;
-using Content.Server.Station.Components;
+using Content.Shared.Station.Components;
using Robust.Shared.Console;
namespace Content.Server.Commands;
using System.Linq;
using Content.Server.Administration;
using Content.Server.EUI;
-using Content.Server.Station.Components;
using Content.Server.Station.Systems;
using Content.Server.StationRecords;
using Content.Server.StationRecords.Systems;
using Content.Shared.CrewManifest;
using Content.Shared.GameTicking;
using Content.Shared.Roles;
+using Content.Shared.Station.Components;
using Content.Shared.StationRecords;
using Robust.Shared.Configuration;
using Robust.Shared.Console;
using Robust.Shared.Player;
using Robust.Shared.Prototypes;
-using Robust.Shared.Utility;
namespace Content.Server.CrewManifest;
using Content.Server.Antag;
-using Content.Server.Dragon;
using Content.Server.GameTicking.Rules.Components;
using Content.Server.Mind;
using Content.Server.Roles;
-using Content.Server.Station.Components;
using Content.Server.Station.Systems;
-using Content.Shared.CharacterInfo;
using Content.Shared.Localizations;
using Robust.Server.GameObjects;
var dragonXform = Transform(dragon);
- var station = _station.GetStationInMap(dragonXform.MapID);
EntityUid? stationGrid = null;
- if (TryComp<StationDataComponent>(station, out var stationData))
- stationGrid = _station.GetLargestGrid(stationData);
+ if (_station.GetStationInMap(dragonXform.MapID) is { } station)
+ stationGrid = _station.GetLargestGrid(station);
if (stationGrid is not null)
{
using System.Diagnostics.CodeAnalysis;
using System.Linq;
-using Content.Server.GameTicking.Rules.Components;
using Content.Server.Station.Components;
using Content.Shared.GameTicking.Components;
using Content.Shared.Random.Helpers;
-using Robust.Server.GameObjects;
+using Content.Shared.Station.Components;
using Robust.Shared.Collections;
using Robust.Shared.Map;
using Robust.Shared.Map.Components;
-using Robust.Shared.Random;
namespace Content.Server.GameTicking.Rules;
using Robust.Shared.Random;
using Robust.Shared.Utility;
using System.Linq;
+using Content.Shared.Station.Components;
using Content.Shared.Store.Components;
using Robust.Shared.Prototypes;
-using System.Linq;
-using Content.Server.GameTicking.Rules.Components;
+using Content.Server.GameTicking.Rules.Components;
using Content.Server.Shuttles.Systems;
using Content.Server.Station.Components;
using Content.Server.Station.Events;
using Content.Shared.GameTicking.Components;
+using Content.Shared.Station.Components;
using Content.Shared.Storage;
-using Robust.Shared.Prototypes;
using Robust.Shared.Random;
namespace Content.Server.GameTicking.Rules;
Transform(newEnt).LocalRotation = rot;
}
- Log.Debug($"Entity replacement took {stopwatch.Elapsed} with {Stations.GetTileCount(args.Station)} tiles");
+ Log.Debug($"Entity replacement took {stopwatch.Elapsed} with {Stations.GetTileCount(args.Station.AsNullable())} tiles");
}
private void QueueReplace(Entity<TransformComponent> ent, List<EntitySpawnEntry> replacements)
{
protected override void ApplyVariation(Entity<EntitySpawnVariationPassComponent> ent, ref StationVariationPassEvent args)
{
- var totalTiles = Stations.GetTileCount(args.Station);
+ var totalTiles = Stations.GetTileCount(args.Station.AsNullable());
var dirtyMod = Random.NextGaussian(ent.Comp.TilesPerEntityAverage, ent.Comp.TilesPerEntityStdDev);
var trashTiles = Math.Max((int) (totalTiles * (1 / dirtyMod)), 0);
protected override void ApplyVariation(Entity<PuddleMessVariationPassComponent> ent, ref StationVariationPassEvent args)
{
- var totalTiles = Stations.GetTileCount(args.Station);
+ var totalTiles = Stations.GetTileCount(args.Station.AsNullable());
if (!_proto.TryIndex(ent.Comp.RandomPuddleSolutionFill, out var proto))
return;
using Content.Server.Popups;
using Content.Server.Roles;
using Content.Server.RoundEnd;
-using Content.Server.Station.Components;
using Content.Server.Station.Systems;
using Content.Server.Zombies;
using Content.Shared.GameTicking.Components;
{
foreach (var station in _station.GetStationsSet())
{
- if (TryComp<StationDataComponent>(station, out var data) && _station.GetLargestGrid(data) is { } grid)
+ if (_station.GetLargestGrid(station) is { } grid)
stationGrids.Add(grid);
}
}
-using Content.Server.Station;
using JetBrains.Annotations;
using Robust.Shared.Prototypes;
using Robust.Shared.Utility;
using System.Diagnostics;
-using System.Numerics;
+using Content.Shared.Station;
namespace Content.Server.Maps;
using Content.Server.Administration;
-using Content.Server.Station.Components;
using Content.Shared.Administration;
+using Content.Shared.Station.Components;
using Robust.Shared.Console;
namespace Content.Server.Nuke.Commands;
using Content.Server.Chat.Systems;
using Content.Server.Fax;
using Content.Shared.Fax.Components;
-using Content.Server.Station.Components;
using Content.Server.Station.Systems;
using Content.Shared.Paper;
+using Content.Shared.Station.Components;
using Robust.Shared.Random;
using Robust.Shared.Utility;
using Content.Server.Screens.Components;
using Content.Server.Shuttles.Components;
using Content.Server.Shuttles.Systems;
-using Content.Server.Station.Components;
using Content.Server.Station.Systems;
using Content.Shared.Database;
using Content.Shared.DeviceNetwork;
using Robust.Shared.Prototypes;
using Robust.Shared.Timing;
using Content.Shared.DeviceNetwork.Components;
+using Content.Shared.Station.Components;
using Timer = Robust.Shared.Timing.Timer;
namespace Content.Server.RoundEnd
/// </summary>
public EntityUid? GetStation()
{
- AllEntityQuery<StationEmergencyShuttleComponent, StationDataComponent>().MoveNext(out _, out _, out var data);
+ AllEntityQuery<StationEmergencyShuttleComponent, StationDataComponent>().MoveNext(out var uid, out _, out var data);
if (data == null)
return null;
- var targetGrid = _stationSystem.GetLargestGrid(data);
+ var targetGrid = _stationSystem.GetLargestGrid((uid, data));
return targetGrid == null ? null : Transform(targetGrid.Value).MapUid;
}
using Content.Server.Salvage.Expeditions;
using Content.Server.Shuttles.Components;
using Content.Server.Shuttles.Events;
-using Content.Server.Station.Components;
using Content.Shared.Chat;
using Content.Shared.Humanoid;
using Content.Shared.Mobs.Components;
using Content.Shared.Salvage.Expeditions;
using Content.Shared.Shuttles.Components;
using Content.Shared.Localizations;
+using Content.Shared.Station.Components;
using Robust.Shared.Map.Components;
using Robust.Shared.Player;
using Content.Server.Shuttles.Events;
using Content.Server.Spawners.Components;
using Content.Server.Spawners.EntitySystems;
-using Content.Server.Station.Components;
using Content.Server.Station.Events;
using Content.Server.Station.Systems;
using Content.Shared.Administration;
if (component.FirstRun)
{
- var station = _station.GetLargestGrid(Comp<StationDataComponent>(component.Station));
+ var station = _station.GetLargestGrid(component.Station);
sourceMap = station == null ? null : Transform(station.Value)?.MapUid;
arrivalsDelay += RoundStartFTLDuration;
component.FirstRun = false;
{
while (query.MoveNext(out var uid, out var comp, out var shuttle, out var xform))
{
- if (comp.NextTransfer > curTime || !TryComp<StationDataComponent>(comp.Station, out var data))
+ if (comp.NextTransfer > curTime)
continue;
var tripTime = _shuttles.DefaultTravelTime + _shuttles.DefaultStartupTime;
// Go to station
else
{
- var targetGrid = _station.GetLargestGrid(data);
+ var targetGrid = _station.GetLargestGrid(comp.Station);
if (targetGrid != null)
_shuttles.FTLToDock(uid, shuttle, targetGrid.Value);
using Content.Server.Screens.Components;
using Content.Server.Shuttles.Components;
using Content.Server.Shuttles.Events;
-using Content.Server.Station.Components;
using Content.Server.Station.Events;
using Content.Server.Station.Systems;
using Content.Shared.Access.Systems;
return;
}
- var targetGrid = _station.GetLargestGrid(Comp<StationDataComponent>(station.Value));
+ var targetGrid = _station.GetLargestGrid(station.Value);
if (targetGrid == null)
return;
return null;
}
- var targetGrid = _station.GetLargestGrid(Comp<StationDataComponent>(stationUid));
+ var targetGrid = _station.GetLargestGrid(stationUid);
// UHH GOOD LUCK
if (targetGrid == null)
using System.Numerics;
using Content.Server.Shuttles.Components;
-using Content.Server.Station.Components;
using Content.Server.Station.Events;
using Content.Shared.CCVar;
-using Content.Shared.Salvage;
using Content.Shared.Shuttles.Components;
using Content.Shared.Station.Components;
using Robust.Shared.Collections;
if (!_cfg.GetCVar(CCVars.GridFill))
return;
- if (!TryComp(uid, out StationDataComponent? dataComp))
- return;
-
- var targetGrid = _station.GetLargestGrid(dataComp);
+ var targetGrid = _station.GetLargestGrid(uid);
if (targetGrid == null)
return;
if (!_cfg.GetCVar(CCVars.GridFill))
return;
- if (!TryComp<StationDataComponent>(uid, out var data))
- {
- return;
- }
-
- var targetGrid = _station.GetLargestGrid(data);
+ var targetGrid = _station.GetLargestGrid(uid);
if (targetGrid == null)
return;
using System.Numerics;
using Content.Server.Administration.Logs;
using Content.Server.Singularity.Events;
-using Content.Server.Station.Components;
using Content.Shared.Database;
-using Content.Shared.Ghost;
using Content.Shared.Mind.Components;
using Content.Shared.Singularity.Components;
using Content.Shared.Singularity.EntitySystems;
+using Content.Shared.Station.Components;
using Content.Shared.Tag;
using Robust.Shared.Containers;
-using Robust.Shared.GameObjects;
using Robust.Shared.Map;
using Robust.Shared.Map.Components;
using Robust.Shared.Physics.Components;
using System.Linq;
using Content.Server.Administration;
using Content.Server.Cargo.Systems;
-using Content.Server.Station.Components;
using Content.Server.Station.Systems;
using Content.Shared.Administration;
+using Content.Shared.Station;
+using Content.Shared.Station.Components;
using Robust.Shared.Toolshed;
using Robust.Shared.Toolshed.Errors;
-using Robust.Shared.Toolshed.Syntax;
using Robust.Shared.Utility;
namespace Content.Server.Station.Commands;
public EntityUid? LargestGrid([PipedArgument] EntityUid input)
{
_station ??= GetSys<StationSystem>();
- return _station.GetLargestGrid(Comp<StationDataComponent>(input));
+ return _station.GetLargestGrid(input);
}
[CommandImplementation("largestgrid")]
-using Content.Server.Station.Components;
+using Content.Shared.Station.Components;
namespace Content.Server.Station.Events;
private void OnStationPostInit(Entity<StationBiomeComponent> map, ref StationPostInitEvent args)
{
- if (!TryComp(map, out StationDataComponent? dataComp))
+ var station = _station.GetLargestGrid(map.Owner);
+ if (station == null)
return;
- var station = _station.GetLargestGrid(dataComp);
- if (station == null) return;
-
var mapId = Transform(station.Value).MapID;
var mapUid = _map.GetMapOrInvalid(mapId);
using Content.Server.Station.Components;
+using Content.Shared.Station.Components;
namespace Content.Server.Station.Systems;
using Content.Server.GameTicking;
using Content.Server.Station.Components;
using Content.Server.Station.Events;
-using Content.Shared.CCVar;
using Content.Shared.Station;
using Content.Shared.Station.Components;
using JetBrains.Annotations;
-using Robust.Server.GameObjects;
using Robust.Server.GameStates;
using Robust.Server.Player;
using Robust.Shared.Collections;
-using Robust.Shared.Configuration;
using Robust.Shared.Enums;
using Robust.Shared.Map;
using Robust.Shared.Map.Components;
using Robust.Shared.Player;
-using Robust.Shared.Random;
using Robust.Shared.Utility;
namespace Content.Server.Station.Systems;
[Dependency] private readonly ChatSystem _chatSystem = default!;
[Dependency] private readonly SharedTransformSystem _transform = default!;
[Dependency] private readonly MetaDataSystem _metaData = default!;
- [Dependency] private readonly MapSystem _map = default!;
[Dependency] private readonly PvsOverrideSystem _pvsOverride = default!;
private ISawmill _sawmill = default!;
private EntityQuery<MapGridComponent> _gridQuery;
private EntityQuery<TransformComponent> _xformQuery;
- private ValueList<MapId> _mapIds = new();
- private ValueList<(Box2Rotated Bounds, MapId MapId)> _gridBounds = new();
+ private ValueList<MapId> _mapIds;
+ private ValueList<(Box2Rotated Bounds, MapId MapId)> _gridBounds;
/// <inheritdoc/>
public override void Initialize()
return;
stationData.Grids.Remove(uid);
+ Dirty(uid, component);
}
public override void Shutdown()
#endregion Event handlers
- /// <summary>
- /// Gets the largest member grid from a station.
- /// </summary>
- public EntityUid? GetLargestGrid(StationDataComponent component)
- {
- EntityUid? largestGrid = null;
- Box2 largestBounds = new Box2();
-
- foreach (var gridUid in component.Grids)
- {
- if (!TryComp<MapGridComponent>(gridUid, out var grid) ||
- grid.LocalAABB.Size.LengthSquared() < largestBounds.Size.LengthSquared())
- continue;
-
- largestBounds = grid.LocalAABB;
- largestGrid = gridUid;
- }
-
- return largestGrid;
- }
-
- /// <summary>
- /// Returns the total number of tiles contained in the station's grids.
- /// </summary>
- public int GetTileCount(StationDataComponent component)
- {
- var count = 0;
- foreach (var gridUid in component.Grids)
- {
- if (!TryComp<MapGridComponent>(gridUid, out var grid))
- continue;
-
- count += _map.GetAllTiles(gridUid, grid).Count();
- }
-
- return count;
- }
-
/// <summary>
/// Tries to retrieve a filter for everything in the station the source is on.
/// </summary>
var stationMember = EnsureComp<StationMemberComponent>(mapGrid);
stationMember.Station = station;
stationData.Grids.Add(mapGrid);
+ Dirty(station, stationData);
RaiseLocalEvent(station, new StationGridAddedEvent(mapGrid, station, false), true);
RemComp<StationMemberComponent>(mapGrid);
stationData.Grids.Remove(mapGrid);
+ Dirty(station, stationData);
RaiseLocalEvent(station, new StationGridRemovedEvent(mapGrid, station), true);
_sawmill.Info($"Removing grid {mapGrid} from station {Name(station)} ({station})");
QueueDel(station);
}
-
- public EntityUid? GetOwningStation(EntityUid? entity, TransformComponent? xform = null)
- {
- if (entity == null)
- return null;
-
- return GetOwningStation(entity.Value, xform);
- }
-
- /// <summary>
- /// Gets the station that "owns" the given entity (essentially, the station the grid it's on is attached to)
- /// </summary>
- /// <param name="entity">Entity to find the owner of.</param>
- /// <param name="xform">Resolve pattern, transform of the entity.</param>
- /// <returns>The owning station, if any.</returns>
- /// <remarks>
- /// This does not remember what station an entity started on, it simply checks where it is currently located.
- /// </remarks>
- public EntityUid? GetOwningStation(EntityUid entity, TransformComponent? xform = null)
- {
- if (!Resolve(entity, ref xform))
- throw new ArgumentException("Tried to use an abstract entity!", nameof(entity));
-
- if (TryComp<StationDataComponent>(entity, out _))
- {
- // We are the station, just return ourselves.
- return entity;
- }
-
- if (TryComp<MapGridComponent>(entity, out _))
- {
- // We are the station, just check ourselves.
- return CompOrNull<StationMemberComponent>(entity)?.Station;
- }
-
- if (xform.GridUid == EntityUid.Invalid)
- {
- Log.Debug("Unable to get owning station - GridUid invalid.");
- return null;
- }
-
- return CompOrNull<StationMemberComponent>(xform.GridUid)?.Station;
- }
-
- public List<EntityUid> GetStations()
- {
- var stations = new List<EntityUid>();
- var query = EntityQueryEnumerator<StationDataComponent>();
- while (query.MoveNext(out var uid, out _))
- {
- stations.Add(uid);
- }
-
- return stations;
- }
-
- public HashSet<EntityUid> GetStationsSet()
- {
- var stations = new HashSet<EntityUid>();
- var query = EntityQueryEnumerator<StationDataComponent>();
- while (query.MoveNext(out var uid, out _))
- {
- stations.Add(uid);
- }
-
- return stations;
- }
-
- public List<(string Name, NetEntity Entity)> GetStationNames()
- {
- var stations = GetStationsSet();
- var stats = new List<(string Name, NetEntity Station)>();
-
- foreach (var weh in stations)
- {
- stats.Add((MetaData(weh).EntityName, GetNetEntity(weh)));
- }
-
- return stats;
- }
-
- /// <summary>
- /// Returns the first station that has a grid in a certain map.
- /// If the map has no stations, null is returned instead.
- /// </summary>
- /// <remarks>
- /// If there are multiple stations on a map it is probably arbitrary which one is returned.
- /// </remarks>
- public EntityUid? GetStationInMap(MapId map)
- {
- var query = EntityQueryEnumerator<StationDataComponent>();
- while (query.MoveNext(out var uid, out var data))
- {
- foreach (var gridUid in data.Grids)
- {
- if (Transform(gridUid).MapID == map)
- {
- return uid;
- }
- }
- }
-
- return null;
- }
}
/// <summary>
using Content.Server.Anomaly;
-using Content.Server.Station.Components;
using Content.Server.StationEvents.Components;
-using Content.Shared.GameTicking.Components;
+using Content.Shared.GameTicking.Components;
+using Content.Shared.Station.Components;
namespace Content.Server.StationEvents.Events;
if (!TryComp<StationDataComponent>(chosenStation, out var stationData))
return;
- var grid = StationSystem.GetLargestGrid(stationData);
+ var grid = StationSystem.GetLargestGrid((chosenStation.Value, stationData));
if (grid is null)
return;
using Content.Server.Cargo.Components;
using Content.Server.Cargo.Systems;
using Content.Server.GameTicking;
-using Content.Server.Station.Components;
using Content.Server.StationEvents.Components;
using Content.Shared.GameTicking.Components;
+using Content.Shared.Station.Components;
using Robust.Shared.Prototypes;
namespace Content.Server.StationEvents.Events;
using System.Numerics;
using Content.Server.Chat.Systems;
using Content.Server.GameTicking.Rules;
-using Content.Server.Station.Components;
using Content.Server.Station.Systems;
using Content.Server.StationEvents.Components;
using Content.Shared.GameTicking.Components;
return;
var station = RobustRandom.Pick(_station.GetStations());
- if (_station.GetLargestGrid(Comp<StationDataComponent>(station)) is not { } grid)
+ if (_station.GetLargestGrid(station) is not { } grid)
return;
var mapId = Transform(grid).MapID;
using Content.Server.Antag;
-using Content.Server.GameTicking.Rules.Components;
-using Content.Server.Station.Components;
using Content.Server.StationEvents.Components;
using Content.Shared.GameTicking.Components;
using Robust.Shared.Map;
return;
}
- var stationData = Comp<StationDataComponent>(station.Value);
-
// find a station grid
- var gridUid = StationSystem.GetLargestGrid(stationData);
+ var gridUid = StationSystem.GetLargestGrid(station.Value);
if (gridUid == null || !TryComp<MapGridComponent>(gridUid, out var grid))
{
Sawmill.Warning("Chosen station has no grids, cannot pick location for {ToPrettyString(uid):rule}");
-using Content.Server.Shuttles.Systems;
-using Content.Server.Station.Systems;
-using Robust.Shared.Serialization.TypeSerializers.Implementations;
-using Robust.Shared.Utility;
+using Robust.Shared.GameStates;
-namespace Content.Server.Station.Components;
+namespace Content.Shared.Station.Components;
/// <summary>
/// Stores core information about a station, namely its config and associated grids.
/// All station entities will have this component.
/// </summary>
-[RegisterComponent, Access(typeof(StationSystem))]
+[RegisterComponent, NetworkedComponent, AutoGenerateComponentState, Access(typeof(SharedStationSystem))]
public sealed partial class StationDataComponent : Component
{
/// <summary>
/// The game map prototype, if any, associated with this station.
/// </summary>
- [DataField("stationConfig")]
- public StationConfig? StationConfig = null;
+ [DataField]
+ public StationConfig? StationConfig;
/// <summary>
/// List of all grids this station is part of.
/// </summary>
- [DataField("grids")]
+ [DataField, AutoNetworkedField]
public HashSet<EntityUid> Grids = new();
}
--- /dev/null
+using Content.Shared.Station.Components;
+using JetBrains.Annotations;
+using Robust.Shared.Map;
+
+namespace Content.Shared.Station;
+
+public abstract partial class SharedStationSystem
+{
+ private void InitializeTracker()
+ {
+ SubscribeLocalEvent<StationTrackerComponent, MapInitEvent>(OnTrackerMapInit);
+ SubscribeLocalEvent<StationTrackerComponent, ComponentRemove>(OnTrackerRemove);
+ SubscribeLocalEvent<StationTrackerComponent, GridUidChangedEvent>(OnTrackerGridChanged);
+ SubscribeLocalEvent<StationTrackerComponent, MetaFlagRemoveAttemptEvent>(OnMetaFlagRemoveAttempt);
+ }
+
+ private void OnTrackerMapInit(Entity<StationTrackerComponent> ent, ref MapInitEvent args)
+ {
+ _meta.AddFlag(ent, MetaDataFlags.ExtraTransformEvents);
+ UpdateStationTracker(ent.AsNullable());
+ }
+
+ private void OnTrackerRemove(Entity<StationTrackerComponent> ent, ref ComponentRemove args)
+ {
+ _meta.RemoveFlag(ent, MetaDataFlags.ExtraTransformEvents);
+ }
+
+ private void OnTrackerGridChanged(Entity<StationTrackerComponent> ent, ref GridUidChangedEvent args)
+ {
+ UpdateStationTracker((ent, ent.Comp, args.Transform));
+ }
+
+ private void OnMetaFlagRemoveAttempt(Entity<StationTrackerComponent> ent, ref MetaFlagRemoveAttemptEvent args)
+ {
+ if ((args.ToRemove & MetaDataFlags.ExtraTransformEvents) != 0 &&
+ ent.Comp.LifeStage <= ComponentLifeStage.Running)
+ {
+ args.ToRemove &= ~MetaDataFlags.ExtraTransformEvents;
+ }
+ }
+
+ /// <summary>
+ /// Updates the station tracker component based on entity's current location.
+ /// </summary>
+ [PublicAPI]
+ public void UpdateStationTracker(Entity<StationTrackerComponent?, TransformComponent?> ent)
+ {
+ if (!Resolve(ent, ref ent.Comp1))
+ return;
+
+ var xform = ent.Comp2;
+
+ if (!_xformQuery.Resolve(ent, ref xform))
+ return;
+
+ // Entity is in nullspace or not on a grid
+ if (xform.MapID == MapId.Nullspace || xform.GridUid == null)
+ {
+ SetStation(ent, null);
+ return;
+ }
+
+ // Check if the grid is part of a station
+ if (!_stationMemberQuery.TryGetComponent(xform.GridUid.Value, out var stationMember))
+ {
+ SetStation(ent, null);
+ return;
+ }
+
+ SetStation(ent, stationMember.Station);
+ }
+
+ /// <summary>
+ /// Sets the station for a StationTrackerComponent.
+ /// </summary>
+ [PublicAPI]
+ public void SetStation(Entity<StationTrackerComponent?> ent, EntityUid? station)
+ {
+ if (!Resolve(ent, ref ent.Comp))
+ return;
+
+ if (ent.Comp.Station == station)
+ return;
+
+ ent.Comp.Station = station;
+ Dirty(ent);
+ }
+}
+using System.Linq;
using Content.Shared.Station.Components;
using JetBrains.Annotations;
using Robust.Shared.Map;
+using Robust.Shared.Map.Components;
namespace Content.Shared.Station;
public abstract partial class SharedStationSystem : EntitySystem
{
+ [Dependency] private readonly SharedMapSystem _map = default!;
[Dependency] private readonly MetaDataSystem _meta = default!;
private EntityQuery<TransformComponent> _xformQuery;
{
base.Initialize();
+ InitializeTracker();
+
_xformQuery = GetEntityQuery<TransformComponent>();
_stationMemberQuery = GetEntityQuery<StationMemberComponent>();
-
- SubscribeLocalEvent<StationTrackerComponent, MapInitEvent>(OnTrackerMapInit);
- SubscribeLocalEvent<StationTrackerComponent, ComponentRemove>(OnTrackerRemove);
- SubscribeLocalEvent<StationTrackerComponent, GridUidChangedEvent>(OnTrackerGridChanged);
- SubscribeLocalEvent<StationTrackerComponent, MetaFlagRemoveAttemptEvent>(OnMetaFlagRemoveAttempt);
}
- private void OnTrackerMapInit(Entity<StationTrackerComponent> ent, ref MapInitEvent args)
+ /// <summary>
+ /// Gets the largest member grid from a station.
+ /// </summary>
+ public EntityUid? GetLargestGrid(Entity<StationDataComponent?> ent)
{
- _meta.AddFlag(ent, MetaDataFlags.ExtraTransformEvents);
- UpdateStationTracker(ent.AsNullable());
- }
+ if (!Resolve(ent, ref ent.Comp))
+ return null;
- private void OnTrackerRemove(Entity<StationTrackerComponent> ent, ref ComponentRemove args)
- {
- _meta.RemoveFlag(ent, MetaDataFlags.ExtraTransformEvents);
- }
+ EntityUid? largestGrid = null;
+ Box2 largestBounds = new Box2();
- private void OnTrackerGridChanged(Entity<StationTrackerComponent> ent, ref GridUidChangedEvent args)
- {
- UpdateStationTracker((ent, ent.Comp, args.Transform));
+ foreach (var gridUid in ent.Comp.Grids)
+ {
+ if (!TryComp<MapGridComponent>(gridUid, out var grid) ||
+ grid.LocalAABB.Size.LengthSquared() < largestBounds.Size.LengthSquared())
+ continue;
+
+ largestBounds = grid.LocalAABB;
+ largestGrid = gridUid;
+ }
+
+ return largestGrid;
}
- private void OnMetaFlagRemoveAttempt(Entity<StationTrackerComponent> ent, ref MetaFlagRemoveAttemptEvent args)
+ /// <summary>
+ /// Returns the total number of tiles contained in the station's grids.
+ /// </summary>
+ public int GetTileCount(Entity<StationDataComponent?> ent)
{
- if ((args.ToRemove & MetaDataFlags.ExtraTransformEvents) != 0 &&
- ent.Comp.LifeStage <= ComponentLifeStage.Running)
+ if (!Resolve(ent, ref ent.Comp))
+ return 0;
+
+ var count = 0;
+ foreach (var gridUid in ent.Comp.Grids)
{
- args.ToRemove &= ~MetaDataFlags.ExtraTransformEvents;
+ if (!TryComp<MapGridComponent>(gridUid, out var grid))
+ continue;
+
+ count += _map.GetAllTiles(gridUid, grid).Count();
}
+
+ return count;
+ }
+
+ [PublicAPI]
+ public EntityUid? GetOwningStation(EntityUid? entity, TransformComponent? xform = null)
+ {
+ if (entity == null)
+ return null;
+
+ return GetOwningStation(entity.Value, xform);
}
/// <summary>
- /// Updates the station tracker component based on entity's current location.
+ /// Gets the station that "owns" the given entity (essentially, the station the grid it's on is attached to)
/// </summary>
- [PublicAPI]
- public void UpdateStationTracker(Entity<StationTrackerComponent?, TransformComponent?> ent)
+ /// <param name="entity">Entity to find the owner of.</param>
+ /// <param name="xform">Resolve pattern, transform of the entity.</param>
+ /// <returns>The owning station, if any.</returns>
+ /// <remarks>
+ /// This does not remember what station an entity started on, it simply checks where it is currently located.
+ /// </remarks>
+ public EntityUid? GetOwningStation(EntityUid entity, TransformComponent? xform = null)
{
- if (!Resolve(ent, ref ent.Comp1))
- return;
+ if (!Resolve(entity, ref xform))
+ throw new ArgumentException("Tried to use an abstract entity!", nameof(entity));
- var xform = ent.Comp2;
+ if (TryComp<StationTrackerComponent>(entity, out var stationTracker))
+ {
+ // We have a specific station we are tracking and are tethered to.
+ return stationTracker.Station;
+ }
- if (!_xformQuery.Resolve(ent, ref xform))
- return;
+ if (HasComp<StationDataComponent>(entity))
+ {
+ // We are the station, just return ourselves.
+ return entity;
+ }
- // Entity is in nullspace or not on a grid
- if (xform.MapID == MapId.Nullspace || xform.GridUid == null)
+ if (HasComp<MapGridComponent>(entity))
{
- SetStation(ent, null);
- return;
+ // We are the station, just check ourselves.
+ return CompOrNull<StationMemberComponent>(entity)?.Station;
}
- // Check if the grid is part of a station
- if (!_stationMemberQuery.TryGetComponent(xform.GridUid.Value, out var stationMember))
+ if (xform.GridUid == EntityUid.Invalid)
{
- SetStation(ent, null);
- return;
+ Log.Debug("Unable to get owning station - GridUid invalid.");
+ return null;
}
- SetStation(ent, stationMember.Station);
+ return CompOrNull<StationMemberComponent>(xform.GridUid)?.Station;
}
- /// <summary>
- /// Sets the station for a StationTrackerComponent.
- /// </summary>
- [PublicAPI]
- public void SetStation(Entity<StationTrackerComponent?> ent, EntityUid? station)
+ public List<EntityUid> GetStations()
{
- if (!Resolve(ent, ref ent.Comp))
- return;
+ var stations = new List<EntityUid>();
+ var query = EntityQueryEnumerator<StationDataComponent>();
+ while (query.MoveNext(out var uid, out _))
+ {
+ stations.Add(uid);
+ }
- if (ent.Comp.Station == station)
- return;
+ return stations;
+ }
- ent.Comp.Station = station;
- Dirty(ent);
+ public HashSet<EntityUid> GetStationsSet()
+ {
+ var stations = new HashSet<EntityUid>();
+ var query = EntityQueryEnumerator<StationDataComponent>();
+ while (query.MoveNext(out var uid, out _))
+ {
+ stations.Add(uid);
+ }
+
+ return stations;
+ }
+
+ public List<(string Name, NetEntity Entity)> GetStationNames()
+ {
+ var stations = GetStationsSet();
+ var stats = new List<(string Name, NetEntity Station)>();
+
+ foreach (var weh in stations)
+ {
+ stats.Add((MetaData(weh).EntityName, GetNetEntity(weh)));
+ }
+
+ return stats;
}
/// <summary>
- /// Gets the station an entity is currently on, if any.
+ /// Returns the first station that has a grid in a certain map.
+ /// If the map has no stations, null is returned instead.
/// </summary>
- [PublicAPI]
- public EntityUid? GetCurrentStation(Entity<StationTrackerComponent?> ent)
+ /// <remarks>
+ /// If there are multiple stations on a map it is probably arbitrary which one is returned.
+ /// </remarks>
+ public EntityUid? GetStationInMap(MapId map)
{
- if (!Resolve(ent, ref ent.Comp, logMissing: false))
- return null;
+ var query = EntityQueryEnumerator<StationDataComponent>();
+ while (query.MoveNext(out var uid, out var data))
+ {
+ foreach (var gridUid in data.Grids)
+ {
+ if (Transform(gridUid).MapID == map)
+ {
+ return uid;
+ }
+ }
+ }
- return ent.Comp.Station;
+ return null;
}
}
-using Content.Server.Maps.NameGenerators;
-using JetBrains.Annotations;
-using Robust.Shared.Prototypes;
+using Robust.Shared.Prototypes;
-namespace Content.Server.Station;
+namespace Content.Shared.Station;
/// <summary>
/// A config for a station. Specifies name and component modifications.
/// </summary>
-[DataDefinition, PublicAPI]
+[DataDefinition]
public sealed partial class StationConfig
{
[DataField("stationProto", required: true)]
- public string StationPrototype = default!;
+ public EntProtoId StationPrototype;
[DataField("components", required: true)]
public ComponentRegistry StationComponentOverrides = default!;
projects / space-station-14.git / commitdiff