RobustToolbox/Robust.Client/GameObjects/Components/Renderable/SpriteBoundsOverlay.cs

using System.Numerics;
using Robust.Client.ComponentTrees;
using Robust.Client.Graphics;
using Robust.Shared.Console;
using Robust.Shared.Enums;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Maths;
using Robust.Shared.Utility;

namespace Robust.Client.GameObjects
{
    public sealed class ShowSpriteBBCommand : LocalizedEntityCommands
    {
        [Dependency] private readonly SpriteBoundsSystem _system = default!;

        public override string Command => "showspritebb";

        public override void Execute(IConsoleShell shell, string argStr, string[] args)
        {
            _system.Enabled ^= true;
        }
    }

    public sealed class SpriteBoundsSystem : EntitySystem
    {
        [Dependency] private readonly IOverlayManager _overlayManager = default!;

        private SpriteBoundsOverlay? _overlay;

        public bool Enabled
        {
            get => _enabled;
            set
            {
                if (_enabled == value) return;

                _enabled = value;

                if (_enabled)
                {
                    DebugTools.AssertNull(_overlay);
                    _overlay = new SpriteBoundsOverlay(EntityManager);
                    _overlayManager.AddOverlay(_overlay);
                }
                else
                {
                    if (_overlay == null) return;
                    _overlayManager.RemoveOverlay(_overlay);
                    _overlay = null;
                }
            }
        }

        private bool _enabled;
    }

    public sealed class SpriteBoundsOverlay(IEntityManager entMan) : Overlay
    {
        public override OverlaySpace Space => OverlaySpace.WorldSpace;

        private readonly SharedTransformSystem _xformSystem = entMan.System<SharedTransformSystem>();
        private readonly SpriteSystem _spriteSystem = entMan.System<SpriteSystem>();
        private readonly SpriteTreeSystem _renderTree = entMan.System<SpriteTreeSystem>();

        protected internal override void Draw(in OverlayDrawArgs args)
        {
            var handle = args.WorldHandle;
            var currentMap = args.MapId;
            var viewport = args.WorldBounds;

            foreach (var entry in _renderTree.QueryAabb(currentMap, viewport))
            {
                var (sprite, xform) = entry;
                var (worldPos, worldRot) = _xformSystem.GetWorldPositionRotation(xform);
                var bounds = _spriteSystem.CalculateBounds((entry.Uid, sprite), worldPos, worldRot, args.Viewport.Eye?.Rotation ?? default);

                // Get scaled down bounds used to indicate the "south" of a sprite.
                var localBound = bounds.Box;
                var smallLocal = localBound.Scale(0.2f).Translated(-new Vector2(0f, localBound.Extents.Y));
                var southIndicator = new Box2Rotated(smallLocal, bounds.Rotation, bounds.Origin);

                handle.DrawRect(bounds, Color.Red.WithAlpha(0.2f));
                handle.DrawRect(southIndicator, Color.Blue.WithAlpha(0.5f));
            }
        }
    }
}