108 lines
3.6 KiB
C#
108 lines
3.6 KiB
C#
#nullable enable
|
|
using System.Collections.Generic;
|
|
using Content.Client.Atmos;
|
|
using Content.Shared.GameObjects.EntitySystems.Atmos;
|
|
using Content.Shared.Atmos;
|
|
using Content.Shared.GameTicking;
|
|
using JetBrains.Annotations;
|
|
using Robust.Client.Interfaces.Graphics.Overlays;
|
|
using Robust.Shared.Interfaces.Map;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.Map;
|
|
using Robust.Shared.Maths;
|
|
|
|
namespace Content.Client.GameObjects.EntitySystems
|
|
{
|
|
[UsedImplicitly]
|
|
internal sealed class AtmosDebugOverlaySystem : SharedAtmosDebugOverlaySystem, IResettingEntitySystem
|
|
{
|
|
[Dependency] private readonly IMapManager _mapManager = default!;
|
|
|
|
private readonly Dictionary<GridId, AtmosDebugOverlayMessage> _tileData =
|
|
new();
|
|
|
|
// Configuration set by debug commands and used by AtmosDebugOverlay {
|
|
/// <summary>Value source for display</summary>
|
|
public AtmosDebugOverlayMode CfgMode;
|
|
/// <summary>This is subtracted from value (applied before CfgScale)</summary>
|
|
public float CfgBase = 0;
|
|
/// <summary>The value is divided by this (applied after CfgBase)</summary>
|
|
public float CfgScale = Atmospherics.MolesCellStandard * 2;
|
|
/// <summary>Gas ID used by GasMoles mode</summary>
|
|
public int CfgSpecificGas = 0;
|
|
/// <summary>Uses black-to-white interpolation (as opposed to red-green-blue) for colourblind users</summary>
|
|
public bool CfgCBM = false;
|
|
// }
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeNetworkEvent<AtmosDebugOverlayMessage>(HandleAtmosDebugOverlayMessage);
|
|
SubscribeNetworkEvent<AtmosDebugOverlayDisableMessage>(HandleAtmosDebugOverlayDisableMessage);
|
|
|
|
_mapManager.OnGridRemoved += OnGridRemoved;
|
|
|
|
var overlayManager = IoCManager.Resolve<IOverlayManager>();
|
|
if(!overlayManager.HasOverlay(nameof(AtmosDebugOverlay)))
|
|
overlayManager.AddOverlay(new AtmosDebugOverlay());
|
|
}
|
|
|
|
private void HandleAtmosDebugOverlayMessage(AtmosDebugOverlayMessage message)
|
|
{
|
|
_tileData[message.GridId] = message;
|
|
}
|
|
|
|
private void HandleAtmosDebugOverlayDisableMessage(AtmosDebugOverlayDisableMessage ev)
|
|
{
|
|
_tileData.Clear();
|
|
}
|
|
|
|
public override void Shutdown()
|
|
{
|
|
base.Shutdown();
|
|
_mapManager.OnGridRemoved -= OnGridRemoved;
|
|
var overlayManager = IoCManager.Resolve<IOverlayManager>();
|
|
if(!overlayManager.HasOverlay(nameof(GasTileOverlay)))
|
|
overlayManager.RemoveOverlay(nameof(GasTileOverlay));
|
|
}
|
|
|
|
public void Reset()
|
|
{
|
|
_tileData.Clear();
|
|
}
|
|
|
|
private void OnGridRemoved(GridId gridId)
|
|
{
|
|
if (_tileData.ContainsKey(gridId))
|
|
{
|
|
_tileData.Remove(gridId);
|
|
}
|
|
}
|
|
|
|
public bool HasData(GridId gridId)
|
|
{
|
|
return _tileData.ContainsKey(gridId);
|
|
}
|
|
|
|
public AtmosDebugOverlayData? GetData(GridId gridIndex, Vector2i indices)
|
|
{
|
|
if (!_tileData.TryGetValue(gridIndex, out var srcMsg))
|
|
return null;
|
|
|
|
var relative = indices - srcMsg.BaseIdx;
|
|
if (relative.X < 0 || relative.Y < 0 || relative.X >= LocalViewRange || relative.Y >= LocalViewRange)
|
|
return null;
|
|
|
|
return srcMsg.OverlayData[relative.X + (relative.Y * LocalViewRange)];
|
|
}
|
|
}
|
|
|
|
internal enum AtmosDebugOverlayMode : byte
|
|
{
|
|
TotalMoles,
|
|
GasMoles,
|
|
Temperature
|
|
}
|
|
}
|