105 lines
3.5 KiB
C#
105 lines
3.5 KiB
C#
using System.Collections.Generic;
|
|
using Content.Client.Atmos.Overlays;
|
|
using Content.Shared.Atmos;
|
|
using Content.Shared.Atmos.EntitySystems;
|
|
using Content.Shared.GameTicking;
|
|
using JetBrains.Annotations;
|
|
using Robust.Client.Graphics;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.Map;
|
|
using Robust.Shared.Maths;
|
|
|
|
namespace Content.Client.Atmos.EntitySystems
|
|
{
|
|
[UsedImplicitly]
|
|
internal sealed class AtmosDebugOverlaySystem : SharedAtmosDebugOverlaySystem
|
|
{
|
|
|
|
private readonly Dictionary<EntityUid, 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<RoundRestartCleanupEvent>(Reset);
|
|
SubscribeNetworkEvent<AtmosDebugOverlayMessage>(HandleAtmosDebugOverlayMessage);
|
|
SubscribeNetworkEvent<AtmosDebugOverlayDisableMessage>(HandleAtmosDebugOverlayDisableMessage);
|
|
|
|
SubscribeLocalEvent<GridRemovalEvent>(OnGridRemoved);
|
|
|
|
var overlayManager = IoCManager.Resolve<IOverlayManager>();
|
|
if(!overlayManager.HasOverlay<AtmosDebugOverlay>())
|
|
overlayManager.AddOverlay(new AtmosDebugOverlay(this));
|
|
}
|
|
|
|
private void OnGridRemoved(GridRemovalEvent ev)
|
|
{
|
|
if (_tileData.ContainsKey(ev.EntityUid))
|
|
{
|
|
_tileData.Remove(ev.EntityUid);
|
|
}
|
|
}
|
|
|
|
private void HandleAtmosDebugOverlayMessage(AtmosDebugOverlayMessage message)
|
|
{
|
|
_tileData[message.GridId] = message;
|
|
}
|
|
|
|
private void HandleAtmosDebugOverlayDisableMessage(AtmosDebugOverlayDisableMessage ev)
|
|
{
|
|
_tileData.Clear();
|
|
}
|
|
|
|
public override void Shutdown()
|
|
{
|
|
base.Shutdown();
|
|
var overlayManager = IoCManager.Resolve<IOverlayManager>();
|
|
if (overlayManager.HasOverlay<AtmosDebugOverlay>())
|
|
overlayManager.RemoveOverlay<AtmosDebugOverlay>();
|
|
}
|
|
|
|
public void Reset(RoundRestartCleanupEvent ev)
|
|
{
|
|
_tileData.Clear();
|
|
}
|
|
|
|
public bool HasData(EntityUid gridId)
|
|
{
|
|
return _tileData.ContainsKey(gridId);
|
|
}
|
|
|
|
public AtmosDebugOverlayData? GetData(EntityUid 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
|
|
}
|
|
}
|