Fluid spread refactor (#11908)

Co-authored-by: metalgearsloth <31366439+metalgearsloth@users.noreply.github.com> Fix undefined
2022-11-15 12:30:59 +01:00
parent 89b959f931
commit 75ea093d78
17 changed files with 719 additions and 336 deletions
--- a/Content.Client/Fluids/PuddleOverlay.cs
+++ b/Content.Client/Fluids/PuddleOverlay.cs
@@ -0,0 +1,117 @@
+using Content.Shared.FixedPoint;
+using Robust.Client.Graphics;
+using Robust.Client.ResourceManagement;
+using Robust.Shared.Enums;
+using Robust.Shared.Map;
+
+namespace Content.Client.Fluids;
+
+public sealed class PuddleOverlay : Overlay
+{
+    [Dependency] private readonly IMapManager _mapManager = default!;
+    [Dependency] private readonly IEyeManager _eyeManager = default!;
+    [Dependency] private readonly IEntityManager _entityManager = default!;
+    [Dependency] private readonly IEntitySystemManager _entitySystemManager = default!;
+    private readonly PuddleDebugOverlaySystem _debugOverlaySystem;
+
+    private readonly Color _heavyPuddle = new(0, 255, 255, 50);
+    private readonly Color _mediumPuddle = new(0, 150, 255, 50);
+    private readonly Color _lightPuddle = new(0, 50, 255, 50);
+
+    private readonly Font _font;
+
+    public override OverlaySpace Space => OverlaySpace.ScreenSpace | OverlaySpace.WorldSpace;
+
+    public PuddleOverlay()
+    {
+        IoCManager.InjectDependencies(this);
+        _debugOverlaySystem = _entitySystemManager.GetEntitySystem<PuddleDebugOverlaySystem>();
+        var cache = IoCManager.Resolve<IResourceCache>();
+        _font = new VectorFont(cache.GetResource<FontResource>("/Fonts/NotoSans/NotoSans-Regular.ttf"), 8);
+    }
+
+    protected override void Draw(in OverlayDrawArgs args)
+    {
+        switch (args.Space)
+        {
+            case OverlaySpace.ScreenSpace:
+                DrawScreen(args);
+                break;
+            case OverlaySpace.WorldSpace:
+                DrawWorld(args);
+                break;
+        }
+    }
+
+    private void DrawWorld(in OverlayDrawArgs args)
+    {
+        var drawHandle = args.WorldHandle;
+        Box2 gridBounds;
+        var xformQuery = _entityManager.GetEntityQuery<TransformComponent>();
+
+        foreach (var gridId in _debugOverlaySystem.TileData.Keys)
+        {
+            if (!_mapManager.TryGetGrid(gridId, out var mapGrid))
+                continue;
+
+            var gridXform = xformQuery.GetComponent(gridId);
+            var (_, _, worldMatrix, invWorldMatrix) = gridXform.GetWorldPositionRotationMatrixWithInv(xformQuery);
+            gridBounds = invWorldMatrix.TransformBox(args.WorldBounds).Enlarged(mapGrid.TileSize * 2);
+            drawHandle.SetTransform(worldMatrix);
+
+            foreach (var debugOverlayData in _debugOverlaySystem.GetData(mapGrid.GridEntityId))
+            {
+                var centre = ((Vector2) debugOverlayData.Pos + 0.5f) * mapGrid.TileSize;
+
+                // is the center of this tile visible
+                if (!gridBounds.Contains(centre))
+                    continue;
+
+                var box = Box2.UnitCentered.Translated(centre);
+                drawHandle.DrawRect(box, Color.Blue, false);
+                drawHandle.DrawRect(box, ColorMap(debugOverlayData.CurrentVolume));
+            }
+        }
+
+        drawHandle.SetTransform(Matrix3.Identity);
+    }
+
+    private void DrawScreen(in OverlayDrawArgs args)
+    {
+        var drawHandle = args.ScreenHandle;
+        var xformQuery = _entityManager.GetEntityQuery<TransformComponent>();
+
+
+        foreach (var gridId in _debugOverlaySystem.TileData.Keys)
+        {
+            if (!_mapManager.TryGetGrid(gridId, out var mapGrid))
+                continue;
+
+            var gridXform = xformQuery.GetComponent(gridId);
+            var (_, _, matrix, invMatrix) = gridXform.GetWorldPositionRotationMatrixWithInv(xformQuery);
+            var gridBounds = invMatrix.TransformBox(args.WorldBounds).Enlarged(mapGrid.TileSize * 2);
+
+            foreach (var debugOverlayData in _debugOverlaySystem.GetData(mapGrid.GridEntityId))
+            {
+                var centre = ((Vector2) debugOverlayData.Pos + 0.5f) * mapGrid.TileSize;
+
+                // // is the center of this tile visible
+                if (!gridBounds.Contains(centre))
+                    continue;
+
+                var screenCenter = _eyeManager.WorldToScreen(matrix.Transform(centre));
+
+                drawHandle.DrawString(_font, screenCenter, debugOverlayData.CurrentVolume.ToString(), Color.White);
+            }
+        }
+    }
+
+    private Color ColorMap(FixedPoint2 intensity)
+    {
+        var fraction = 1 - intensity / FixedPoint2.New(20f);
+        var result  = fraction < 0.5f
+            ? Color.InterpolateBetween(_mediumPuddle, _heavyPuddle, fraction.Float() * 2)
+            : Color.InterpolateBetween(_lightPuddle, _mediumPuddle, (fraction.Float() - 0.5f) * 2);
+        return result;
+    }
+}