#nullable enable using System; using System.Collections; using System.Collections.Generic; using System.Linq; using System.Runtime.CompilerServices; using Content.Server.Atmos; using Content.Server.GameObjects.Components.Atmos.Piping; using Content.Server.GameObjects.Components.NodeContainer.NodeGroups; using Content.Shared.Atmos; using Content.Shared.Maps; using Robust.Server.Interfaces.GameObjects; using Robust.Shared.GameObjects; using Robust.Shared.GameObjects.Components.Map; using Robust.Shared.GameObjects.Components.Transform; using Robust.Shared.Interfaces.Map; using Robust.Shared.Interfaces.Timing; using Robust.Shared.IoC; using Robust.Shared.Map; using Robust.Shared.Maths; using Robust.Shared.Serialization; using Robust.Shared.Timing; using Robust.Shared.ViewVariables; namespace Content.Server.GameObjects.Components.Atmos { /// /// This is our SSAir equivalent. /// [ComponentReference(typeof(IGridAtmosphereComponent))] [RegisterComponent, Serializable] public class GridAtmosphereComponent : Component, IGridAtmosphereComponent { [Robust.Shared.IoC.Dependency] private IMapManager _mapManager = default!; [Robust.Shared.IoC.Dependency] private ITileDefinitionManager _tileDefinitionManager = default!; [Robust.Shared.IoC.Dependency] private IServerEntityManager _serverEntityManager = default!; /// /// Check current execution time every n instances processed. /// private const int LagCheckIterations = 30; /// /// Max milliseconds allowed for atmos updates. /// private const float LagCheckMaxMilliseconds = 5f; /// /// How much time before atmos updates are ran. /// private const float AtmosTime = 1/26f; public override string Name => "GridAtmosphere"; private bool _paused = false; private float _timer = 0f; private Stopwatch _stopwatch = new Stopwatch(); [ViewVariables] public int UpdateCounter { get; private set; } = 0; [ViewVariables] private double _tileEqualizeLastProcess; [ViewVariables] private readonly HashSet _excitedGroups = new HashSet(1000); [ViewVariables] private int ExcitedGroupCount => _excitedGroups.Count; [ViewVariables] private double _excitedGroupLastProcess; [ViewVariables] private readonly Dictionary _tiles = new Dictionary(1000); [ViewVariables] private readonly HashSet _activeTiles = new HashSet(1000); [ViewVariables] private int ActiveTilesCount => _activeTiles.Count; [ViewVariables] private double _activeTilesLastProcess; [ViewVariables] private readonly HashSet _hotspotTiles = new HashSet(1000); [ViewVariables] private int HotspotTilesCount => _hotspotTiles.Count; [ViewVariables] private double _hotspotsLastProcess; [ViewVariables] private readonly HashSet _superconductivityTiles = new HashSet(1000); [ViewVariables] private int SuperconductivityTilesCount => _superconductivityTiles.Count; [ViewVariables] private double _superconductivityLastProcess; [ViewVariables] private readonly HashSet _invalidatedCoords = new HashSet(1000); [ViewVariables] private int InvalidatedCoordsCount => _invalidatedCoords.Count; [ViewVariables] private HashSet _highPressureDelta = new HashSet(1000); [ViewVariables] private int HighPressureDeltaCount => _highPressureDelta.Count; [ViewVariables] private double _highPressureDeltaLastProcess; [ViewVariables] private readonly HashSet _pipeNets = new HashSet(); [ViewVariables] private double _pipeNetLastProcess; [ViewVariables] private readonly HashSet _pipeNetDevices = new HashSet(); [ViewVariables] private double _pipeNetDevicesLastProcess; [ViewVariables] private Queue _currentRunTiles = new Queue(); [ViewVariables] private Queue _currentRunExcitedGroups = new Queue(); [ViewVariables] private Queue _currentRunPipeNet = new Queue(); [ViewVariables] private Queue _currentRunPipeNetDevice = new Queue(); [ViewVariables] private ProcessState _state = ProcessState.TileEqualize; private enum ProcessState { TileEqualize, ActiveTiles, ExcitedGroups, HighPressureDelta, Hotspots, Superconductivity, PipeNet, PipeNetDevices, } /// public void PryTile(MapIndices indices) { if (!Owner.TryGetComponent(out IMapGridComponent? mapGridComponent)) return; if (IsSpace(indices) || IsAirBlocked(indices)) return; var mapGrid = mapGridComponent.Grid; indices.PryTile(mapGrid.Index, _mapManager, _tileDefinitionManager, _serverEntityManager); } public override void Initialize() { base.Initialize(); RepopulateTiles(); } public override void OnAdd() { base.OnAdd(); RepopulateTiles(); } public void RepopulateTiles() { if (!Owner.TryGetComponent(out IMapGridComponent? mapGrid)) return; foreach (var tile in mapGrid.Grid.GetAllTiles()) { if(!_tiles.ContainsKey(tile.GridIndices)) _tiles.Add(tile.GridIndices, new TileAtmosphere(this, tile.GridIndex, tile.GridIndices, new GasMixture(GetVolumeForCells(1)){Temperature = Atmospherics.T20C})); } foreach (var (_, tile) in _tiles.ToArray()) { tile.UpdateAdjacent(); tile.UpdateVisuals(); } } /// public void Invalidate(MapIndices indices) { _invalidatedCoords.Add(indices); } private void Revalidate() { if (!Owner.TryGetComponent(out IMapGridComponent? mapGrid)) return; foreach (var indices in _invalidatedCoords.ToArray()) { var tile = GetTile(indices); AddActiveTile(tile); if (tile == null) { tile = new TileAtmosphere(this, mapGrid.Grid.Index, indices, new GasMixture(GetVolumeForCells(1)){Temperature = Atmospherics.T20C}); _tiles[indices] = tile; } if (IsSpace(indices)) { tile.Air = new GasMixture(GetVolumeForCells(1)); tile.Air.MarkImmutable(); _tiles[indices] = tile; } else if (IsAirBlocked(indices)) { tile.Air = null; } else { var obs = GetObstructingComponent(indices); if (obs != null) { if (tile.Air == null && obs.FixVacuum) { FixVacuum(tile.GridIndices); } } tile.Air ??= new GasMixture(GetVolumeForCells(1)){Temperature = Atmospherics.T20C}; } tile.UpdateAdjacent(); tile.UpdateVisuals(); for (var i = 0; i < Atmospherics.Directions; i++) { var direction = (AtmosDirection) (1 << i); var otherIndices = indices.Offset(direction.ToDirection()); var otherTile = GetTile(otherIndices); AddActiveTile(otherTile); otherTile?.UpdateAdjacent(direction.GetOpposite()); } } _invalidatedCoords.Clear(); } /// public void FixVacuum(MapIndices indices) { if (!Owner.TryGetComponent(out IMapGridComponent? mapGrid)) return; var tile = GetTile(indices); if (tile?.GridIndex != mapGrid.Grid.Index) return; var adjacent = GetAdjacentTiles(indices); tile.Air = new GasMixture(GetVolumeForCells(1)){Temperature = Atmospherics.T20C}; _tiles[indices] = tile; var ratio = 1f / adjacent.Count; foreach (var (direction, adj) in adjacent) { var mix = adj.Air.RemoveRatio(ratio); tile.Air.Merge(mix); adj.Air.Merge(mix); } } /// [MethodImpl(MethodImplOptions.AggressiveInlining)] public void AddActiveTile(TileAtmosphere? tile) { if (!Owner.TryGetComponent(out IMapGridComponent? mapGrid)) return; if (tile?.GridIndex != mapGrid.Grid.Index) return; tile.Excited = true; _activeTiles.Add(tile); } /// [MethodImpl(MethodImplOptions.AggressiveInlining)] public void RemoveActiveTile(TileAtmosphere? tile) { if (tile == null) return; _activeTiles.Remove(tile); tile.Excited = false; tile.ExcitedGroup?.Dispose(); } /// [MethodImpl(MethodImplOptions.AggressiveInlining)] public void AddHotspotTile(TileAtmosphere? tile) { if (!Owner.TryGetComponent(out IMapGridComponent? mapGrid)) return; if (tile?.GridIndex != mapGrid.Grid.Index || tile?.Air == null) return; _hotspotTiles.Add(tile); } /// [MethodImpl(MethodImplOptions.AggressiveInlining)] public void RemoveHotspotTile(TileAtmosphere? tile) { if (tile == null) return; _hotspotTiles.Remove(tile); } public void AddSuperconductivityTile(TileAtmosphere? tile) { if (!Owner.TryGetComponent(out IMapGridComponent? mapGrid)) return; if (tile?.GridIndex != mapGrid.Grid.Index) return; _superconductivityTiles.Add(tile); } public void RemoveSuperconductivityTile(TileAtmosphere? tile) { if (tile == null) return; _superconductivityTiles.Remove(tile); } /// [MethodImpl(MethodImplOptions.AggressiveInlining)] public void AddHighPressureDelta(TileAtmosphere? tile) { if (!Owner.TryGetComponent(out IMapGridComponent? mapGrid)) return; if (tile?.GridIndex != mapGrid.Grid.Index) return; _highPressureDelta.Add(tile); } /// [MethodImpl(MethodImplOptions.AggressiveInlining)] public bool HasHighPressureDelta(TileAtmosphere tile) { return _highPressureDelta.Contains(tile); } /// [MethodImpl(MethodImplOptions.AggressiveInlining)] public void AddExcitedGroup(ExcitedGroup excitedGroup) { _excitedGroups.Add(excitedGroup); } /// [MethodImpl(MethodImplOptions.AggressiveInlining)] public void RemoveExcitedGroup(ExcitedGroup excitedGroup) { _excitedGroups.Remove(excitedGroup); } public void AddPipeNet(IPipeNet pipeNet) { _pipeNets.Add(pipeNet); } public void RemovePipeNet(IPipeNet pipeNet) { _pipeNets.Remove(pipeNet); } public void AddPipeNetDevice(PipeNetDeviceComponent pipeNetDevice) { _pipeNetDevices.Add(pipeNetDevice); } public void RemovePipeNetDevice(PipeNetDeviceComponent pipeNetDevice) { _pipeNetDevices.Remove(pipeNetDevice); } /// public TileAtmosphere? GetTile(GridCoordinates coordinates, bool createSpace = true) { return GetTile(coordinates.ToMapIndices(_mapManager), createSpace); } /// public TileAtmosphere? GetTile(MapIndices indices, bool createSpace = true) { if (!Owner.TryGetComponent(out IMapGridComponent? mapGrid)) return null; if (_tiles.TryGetValue(indices, out var tile)) return tile; // We don't have that tile! if (IsSpace(indices) && createSpace) { return new TileAtmosphere(this, mapGrid.Grid.Index, indices, new GasMixture(GetVolumeForCells(1)){Temperature = Atmospherics.TCMB}, true); } return null; } /// public bool IsAirBlocked(MapIndices indices) { var ac = GetObstructingComponent(indices); return ac != null && ac.AirBlocked; } /// public bool IsSpace(MapIndices indices) { // TODO ATMOS use ContentTileDefinition to define in YAML whether or not a tile is considered space if (!Owner.TryGetComponent(out IMapGridComponent? mapGrid)) return default; return mapGrid.Grid.GetTileRef(indices).Tile.IsEmpty; } public Dictionary GetAdjacentTiles(MapIndices indices, bool includeAirBlocked = false) { var sides = new Dictionary(); for (var i = 0; i < Atmospherics.Directions; i++) { var direction = (AtmosDirection) (1 << i); var side = indices.Offset(direction.ToDirection()); var tile = GetTile(side); if (tile != null && (tile.Air != null || includeAirBlocked)) sides[direction] = tile; } return sides; } public long EqualizationQueueCycleControl { get; set; } /// public float GetVolumeForCells(int cellCount) { if (!Owner.TryGetComponent(out IMapGridComponent? mapGrid)) return default; return mapGrid.Grid.TileSize * cellCount * Atmospherics.CellVolume; } /// public void Update(float frameTime) { _timer += frameTime; if (_invalidatedCoords.Count != 0) Revalidate(); if (_timer < AtmosTime) return; // We subtract it so it takes lost time into account. _timer -= AtmosTime; switch (_state) { case ProcessState.TileEqualize: if (!ProcessTileEqualize(_paused)) { _paused = true; return; } _paused = false; _state = ProcessState.ActiveTiles; return; case ProcessState.ActiveTiles: if (!ProcessActiveTiles(_paused)) { _paused = true; return; } _paused = false; _state = ProcessState.ExcitedGroups; return; case ProcessState.ExcitedGroups: if (!ProcessExcitedGroups(_paused)) { _paused = true; return; } _paused = false; _state = ProcessState.HighPressureDelta; return; case ProcessState.HighPressureDelta: if (!ProcessHighPressureDelta(_paused)) { _paused = true; return; } _paused = false; _state = ProcessState.Hotspots; break; case ProcessState.Hotspots: if (!ProcessHotspots(_paused)) { _paused = true; return; } _paused = false; _state = ProcessState.Superconductivity; break; case ProcessState.Superconductivity: if (!ProcessSuperconductivity(_paused)) { _paused = true; return; } _paused = false; _state = ProcessState.PipeNet; break; case ProcessState.PipeNet: if (!ProcessPipeNets(_paused)) { _paused = true; return; } _paused = false; _state = ProcessState.PipeNetDevices; break; case ProcessState.PipeNetDevices: if (!ProcessPipeNetDevices(_paused)) { _paused = true; return; } _paused = false; _state = ProcessState.TileEqualize; break; } UpdateCounter++; } public bool ProcessTileEqualize(bool resumed = false) { _stopwatch.Restart(); if(!resumed) _currentRunTiles = new Queue(_activeTiles); var number = 0; while (_currentRunTiles.Count > 0) { var tile = _currentRunTiles.Dequeue(); tile.EqualizePressureInZone(UpdateCounter); if (number++ < LagCheckIterations) continue; number = 0; // Process the rest next time. if (_stopwatch.Elapsed.TotalMilliseconds >= LagCheckMaxMilliseconds) { _tileEqualizeLastProcess = _stopwatch.Elapsed.TotalMilliseconds; return false; } } _tileEqualizeLastProcess = _stopwatch.Elapsed.TotalMilliseconds; return true; } public bool ProcessActiveTiles(bool resumed = false) { _stopwatch.Restart(); if(!resumed) _currentRunTiles = new Queue(_activeTiles); var number = 0; while (_currentRunTiles.Count > 0) { var tile = _currentRunTiles.Dequeue(); tile.ProcessCell(UpdateCounter); if (number++ < LagCheckIterations) continue; number = 0; // Process the rest next time. if (_stopwatch.Elapsed.TotalMilliseconds >= LagCheckMaxMilliseconds) { _activeTilesLastProcess = _stopwatch.Elapsed.TotalMilliseconds; return false; } } _activeTilesLastProcess = _stopwatch.Elapsed.TotalMilliseconds; return true; } public bool ProcessExcitedGroups(bool resumed = false) { _stopwatch.Restart(); if(!resumed) _currentRunExcitedGroups = new Queue(_excitedGroups); var number = 0; while (_currentRunExcitedGroups.Count > 0) { var excitedGroup = _currentRunExcitedGroups.Dequeue(); excitedGroup.BreakdownCooldown++; excitedGroup.DismantleCooldown++; if(excitedGroup.BreakdownCooldown > Atmospherics.ExcitedGroupBreakdownCycles) excitedGroup.SelfBreakdown(); else if(excitedGroup.DismantleCooldown > Atmospherics.ExcitedGroupsDismantleCycles) excitedGroup.Dismantle(); if (number++ < LagCheckIterations) continue; number = 0; // Process the rest next time. if (_stopwatch.Elapsed.TotalMilliseconds >= LagCheckMaxMilliseconds) { _excitedGroupLastProcess = _stopwatch.Elapsed.TotalMilliseconds; return false; } } _excitedGroupLastProcess = _stopwatch.Elapsed.TotalMilliseconds; return true; } public bool ProcessHighPressureDelta(bool resumed = false) { _stopwatch.Restart(); if(!resumed) _currentRunTiles = new Queue(_highPressureDelta); var number = 0; while (_currentRunTiles.Count > 0) { var tile = _currentRunTiles.Dequeue(); tile.HighPressureMovements(); tile.PressureDifference = 0f; tile.PressureSpecificTarget = null; _highPressureDelta.Remove(tile); if (number++ < LagCheckIterations) continue; number = 0; // Process the rest next time. if (_stopwatch.Elapsed.TotalMilliseconds >= LagCheckMaxMilliseconds) { _highPressureDeltaLastProcess = _stopwatch.Elapsed.TotalMilliseconds; return false; } } _highPressureDeltaLastProcess = _stopwatch.Elapsed.TotalMilliseconds; return true; } private bool ProcessHotspots(bool resumed = false) { _stopwatch.Restart(); if(!resumed) _currentRunTiles = new Queue(_hotspotTiles); var number = 0; while (_currentRunTiles.Count > 0) { var hotspot = _currentRunTiles.Dequeue(); hotspot.ProcessHotspot(); if (number++ < LagCheckIterations) continue; number = 0; // Process the rest next time. if (_stopwatch.Elapsed.TotalMilliseconds >= LagCheckMaxMilliseconds) { _hotspotsLastProcess = _stopwatch.Elapsed.TotalMilliseconds; return false; } } _hotspotsLastProcess = _stopwatch.Elapsed.TotalMilliseconds; return true; } private bool ProcessSuperconductivity(bool resumed = false) { _stopwatch.Restart(); if(!resumed) _currentRunTiles = new Queue(_superconductivityTiles); var number = 0; while (_currentRunTiles.Count > 0) { var superconductivity = _currentRunTiles.Dequeue(); superconductivity.Superconduct(); if (number++ < LagCheckIterations) continue; number = 0; // Process the rest next time. if (_stopwatch.Elapsed.TotalMilliseconds >= LagCheckMaxMilliseconds) { _superconductivityLastProcess = _stopwatch.Elapsed.TotalMilliseconds; return false; } } _superconductivityLastProcess = _stopwatch.Elapsed.TotalMilliseconds; return true; } private bool ProcessPipeNets(bool resumed = false) { _stopwatch.Restart(); if(!resumed) _currentRunPipeNet = new Queue(_pipeNets); var number = 0; while (_currentRunPipeNet.Count > 0) { var pipenet = _currentRunPipeNet.Dequeue(); pipenet.Update(); if (number++ < LagCheckIterations) continue; number = 0; // Process the rest next time. if (_stopwatch.Elapsed.TotalMilliseconds >= LagCheckMaxMilliseconds) { _pipeNetLastProcess = _stopwatch.Elapsed.TotalMilliseconds; return false; } } _pipeNetLastProcess = _stopwatch.Elapsed.TotalMilliseconds; return true; } private bool ProcessPipeNetDevices(bool resumed = false) { _stopwatch.Restart(); if(!resumed) _currentRunPipeNetDevice = new Queue(_pipeNetDevices); var number = 0; while (_currentRunPipeNetDevice.Count > 0) { var device = _currentRunPipeNetDevice.Dequeue(); device.Update(); if (number++ < LagCheckIterations) continue; number = 0; // Process the rest next time. if (_stopwatch.Elapsed.TotalMilliseconds >= LagCheckMaxMilliseconds) { _pipeNetDevicesLastProcess = _stopwatch.Elapsed.TotalMilliseconds; return false; } } _pipeNetDevicesLastProcess = _stopwatch.Elapsed.TotalMilliseconds; return true; } private AirtightComponent? GetObstructingComponent(MapIndices indices) { if (!Owner.TryGetComponent(out IMapGridComponent? mapGrid)) return default; foreach (var v in mapGrid.Grid.GetSnapGridCell(indices, SnapGridOffset.Center)) { if (v.Owner.TryGetComponent(out var ac)) return ac; } return null; } public void Dispose() { } public override void ExposeData(ObjectSerializer serializer) { base.ExposeData(serializer); if (serializer.Reading && Owner.TryGetComponent(out IMapGridComponent? mapGrid)) { var gridId = mapGrid.Grid.Index; if (!serializer.TryReadDataField("uniqueMixes", out List? uniqueMixes) || !serializer.TryReadDataField("tiles", out Dictionary? tiles)) return; _tiles.Clear(); foreach (var (indices, mix) in tiles!) { _tiles.Add(indices, new TileAtmosphere(this, gridId, indices, (GasMixture)uniqueMixes![mix].Clone())); Invalidate(indices); } } else if (serializer.Writing) { var uniqueMixes = new List(); var uniqueMixHash = new Dictionary(); var tiles = new Dictionary(); foreach (var (indices, tile) in _tiles) { if (tile.Air == null) continue; if (uniqueMixHash.TryGetValue(tile.Air, out var index)) { tiles[indices] = index; continue; } uniqueMixes.Add(tile.Air); var newIndex = uniqueMixes.Count - 1; uniqueMixHash[tile.Air] = newIndex; tiles[indices] = newIndex; } serializer.DataField(ref uniqueMixes, "uniqueMixes", new List()); serializer.DataField(ref tiles, "tiles", new Dictionary()); } } public IEnumerator GetEnumerator() { return _tiles.Values.GetEnumerator(); } IEnumerator IEnumerable.GetEnumerator() { return GetEnumerator(); } /// public void BurnTile(MapIndices gridIndices) { // TODO ATMOS } } }