98 lines
2.9 KiB
C#
98 lines
2.9 KiB
C#
using Content.Shared.Atmos.EntitySystems;
|
|
using Robust.Shared.Serialization;
|
|
using Robust.Shared.Timing;
|
|
using Robust.Shared.Utility;
|
|
using System.Diagnostics.CodeAnalysis;
|
|
using static Content.Shared.Atmos.EntitySystems.SharedGasTileOverlaySystem;
|
|
|
|
namespace Content.Shared.Atmos
|
|
{
|
|
[Serializable, NetSerializable]
|
|
[Access(typeof(SharedGasTileOverlaySystem))]
|
|
public sealed class GasOverlayChunk
|
|
{
|
|
/// <summary>
|
|
/// The index of this chunk
|
|
/// </summary>
|
|
public readonly Vector2i Index;
|
|
public readonly Vector2i Origin;
|
|
|
|
public GasOverlayData?[][] TileData = new GasOverlayData?[ChunkSize][];
|
|
|
|
[NonSerialized]
|
|
public GameTick LastUpdate;
|
|
|
|
public GasOverlayChunk(Vector2i index)
|
|
{
|
|
Index = index;
|
|
Origin = Index * ChunkSize;
|
|
|
|
// For whatever reason, net serialize does not like multi_D arrays. So Jagged it is.
|
|
for (var i = 0; i < ChunkSize; i++)
|
|
{
|
|
TileData[i] = new GasOverlayData?[ChunkSize];
|
|
}
|
|
}
|
|
|
|
public GasOverlayData? GetData(Vector2i gridIndices)
|
|
{
|
|
DebugTools.Assert(InBounds(gridIndices));
|
|
return TileData[gridIndices.X - Origin.X][gridIndices.Y - Origin.Y];
|
|
}
|
|
|
|
public GasOverlayData? SetData(Vector2i gridIndices, GasOverlayData? data)
|
|
{
|
|
DebugTools.Assert(InBounds(gridIndices));
|
|
return TileData[gridIndices.X - Origin.X][gridIndices.Y - Origin.Y] = data;
|
|
}
|
|
|
|
private bool InBounds(Vector2i gridIndices)
|
|
{
|
|
return gridIndices.X >= Origin.X &&
|
|
gridIndices.Y >= Origin.Y &&
|
|
gridIndices.X < Origin.X + ChunkSize &&
|
|
gridIndices.Y < Origin.Y + ChunkSize;
|
|
}
|
|
}
|
|
|
|
public struct GasChunkEnumerator
|
|
{
|
|
private GasOverlayChunk _chunk;
|
|
public int X = 0;
|
|
public int Y = -1;
|
|
private GasOverlayData?[] _column;
|
|
|
|
|
|
public GasChunkEnumerator(GasOverlayChunk chunk)
|
|
{
|
|
_chunk = chunk;
|
|
_column = _chunk.TileData[0];
|
|
}
|
|
|
|
public bool MoveNext([NotNullWhen(true)] out GasOverlayData? gas)
|
|
{
|
|
while (X < ChunkSize)
|
|
{
|
|
// We want to increment Y before returning, but we also want it to match the current Y coordinate for
|
|
// the returned gas, so using a slightly different logic for the Y loop.
|
|
while (Y < ChunkSize - 1)
|
|
{
|
|
Y++;
|
|
gas = _column[Y];
|
|
|
|
if (gas != null)
|
|
return true;
|
|
}
|
|
|
|
X++;
|
|
if (X < ChunkSize)
|
|
_column = _chunk.TileData[X];
|
|
Y = -1;
|
|
}
|
|
|
|
gas = null;
|
|
return false;
|
|
}
|
|
}
|
|
}
|