StarHorizon_Public/Content.Client/_NF/Research/UI/RoundedStyleBoxFlat.cs

using System.Numerics;
using Robust.Client.Graphics;

namespace Content.Client._NF.Research.UI;

/// <summary>
/// A StyleBoxFlat with rounded corners support
/// </summary>
public sealed class RoundedStyleBoxFlat : StyleBox
{
    public Color BackgroundColor { get; set; }
    public Color BorderColor { get; set; }

    /// <summary>
    /// Thickness of the border, in virtual pixels.
    /// </summary>
    public Thickness BorderThickness { get; set; }

    /// <summary>
    /// Radius of the rounded corners, in virtual pixels.
    /// </summary>
    public float CornerRadius { get; set; } = 0f;

    protected override void DoDraw(DrawingHandleScreen handle, UIBox2 box, float uiScale)
    {
        var scaledRadius = CornerRadius * uiScale;
        var thickness = BorderThickness.Scale(uiScale);

        if (scaledRadius <= 0)
        {
            // Draw as regular rectangle if no corner radius
            DrawRegularRect(handle, box, thickness);
        }
        else
        {
            // Draw rounded rectangle
            DrawRoundedRect(handle, box, thickness, scaledRadius);
        }
    }

    private void DrawRegularRect(DrawingHandleScreen handle, UIBox2 box, Thickness thickness)
    {
        var (btl, btt, btr, btb) = thickness;

        // Draw borders
        if (btl > 0)
            handle.DrawRect(new UIBox2(box.Left, box.Top, box.Left + btl, box.Bottom), BorderColor);

        if (btt > 0)
            handle.DrawRect(new UIBox2(box.Left, box.Top, box.Right, box.Top + btt), BorderColor);

        if (btr > 0)
            handle.DrawRect(new UIBox2(box.Right - btr, box.Top, box.Right, box.Bottom), BorderColor);

        if (btb > 0)
            handle.DrawRect(new UIBox2(box.Left, box.Bottom - btb, box.Right, box.Bottom), BorderColor);

        // Draw background
        handle.DrawRect(thickness.Deflate(box), BackgroundColor);
    }

    private void DrawRoundedRect(DrawingHandleScreen handle, UIBox2 box, Thickness thickness, float radius)
    {
        // Clamp radius to not exceed half the box dimensions
        var maxRadius = Math.Min(box.Width, box.Height) / 2f;
        radius = Math.Min(radius, maxRadius);

        // Draw background first
        DrawRoundedBackground(handle, box, radius);

        // Draw border if needed
        if (thickness.Left > 0 || thickness.Top > 0 || thickness.Right > 0 || thickness.Bottom > 0)
        {
            DrawRoundedBorder(handle, box, thickness, radius);
        }
    }

    private List<Vector2> CreateRoundedRectVertices(UIBox2 box, float radius)
    {
        var vertices = new List<Vector2>();
        const int segments = 16;

        // Top-left
        var tl = new Vector2(box.Left + radius, box.Top + radius);
        for (var i = 0; i <= segments; i++)
        {
            var angle = MathF.PI + (MathF.PI / 2) * (i / (float)segments);
            vertices.Add(tl + new Vector2(MathF.Cos(angle), MathF.Sin(angle)) * radius);
        }

        // Top-right
        var tr = new Vector2(box.Right - radius, box.Top + radius);
        for (var i = 0; i <= segments; i++)
        {
            var angle = 3 * MathF.PI / 2 + (MathF.PI / 2) * (i / (float)segments);
            vertices.Add(tr + new Vector2(MathF.Cos(angle), MathF.Sin(angle)) * radius);
        }

        // Bottom-right
        var br = new Vector2(box.Right - radius, box.Bottom - radius);
        for (var i = 0; i <= segments; i++)
        {
            var angle = 0 + (MathF.PI / 2) * (i / (float)segments);
            vertices.Add(br + new Vector2(MathF.Cos(angle), MathF.Sin(angle)) * radius);
        }

        // Bottom-left
        var bl = new Vector2(box.Left + radius, box.Bottom - radius);
        for (var i = 0; i <= segments; i++)
        {
            var angle = MathF.PI / 2 + (MathF.PI / 2) * (i / (float)segments);
            vertices.Add(bl + new Vector2(MathF.Cos(angle), MathF.Sin(angle)) * radius);
        }

        return vertices;
    }

    private void DrawRoundedBackground(DrawingHandleScreen handle, UIBox2 box, float radius)
    {
        var vertices = CreateRoundedRectVertices(box, radius);
        handle.DrawPrimitives(DrawPrimitiveTopology.TriangleFan, vertices.ToArray(), BackgroundColor);
    }

    private void DrawRoundedBorder(DrawingHandleScreen handle, UIBox2 box, Thickness thickness, float radius)
    {
        var outer = CreateRoundedRectVertices(box, radius);
        var innerBox = new UIBox2(box.Left + thickness.Left, box.Top + thickness.Top, box.Right - thickness.Right, box.Bottom - thickness.Bottom);
        var innerRadius = Math.Max(0, radius - Math.Max(thickness.Left, thickness.Top));
        var inner = CreateRoundedRectVertices(innerBox, innerRadius);

        var vertices = new List<Vector2>();
        for (var i = 0; i < outer.Count; i++)
        {
            vertices.Add(outer[i]);
            vertices.Add(inner[i]);
        }
        vertices.Add(outer[0]);
        vertices.Add(inner[0]);

        handle.DrawPrimitives(DrawPrimitiveTopology.TriangleStrip, vertices.ToArray(), BorderColor);
    }

    public RoundedStyleBoxFlat()
    {
    }

    public RoundedStyleBoxFlat(Color backgroundColor, float cornerRadius = 0f)
    {
        BackgroundColor = backgroundColor;
        CornerRadius = cornerRadius;
    }

    public RoundedStyleBoxFlat(RoundedStyleBoxFlat other) : base(other)
    {
        BackgroundColor = other.BackgroundColor;
        BorderColor = other.BorderColor;
        BorderThickness = other.BorderThickness;
        CornerRadius = other.CornerRadius;
    }

    protected override float GetDefaultContentMargin(Margin margin)
    {
        return margin switch
        {
            Margin.Top => BorderThickness.Top,
            Margin.Bottom => BorderThickness.Bottom,
            Margin.Right => BorderThickness.Right,
            Margin.Left => BorderThickness.Left,
            _ => throw new ArgumentOutOfRangeException(nameof(margin), margin, null)
        };
    }
}