Dashboard/Dashboard.Drawing/DrawExtension.cs

using System.Collections.Generic;
using System.Collections.Immutable;
using System.Linq;
using System.Numerics;

namespace Dashboard.Drawing
{
    /// <summary>
    /// Interface for all drawing extensions.
    /// </summary>
    public interface IDrawExtension
    {
        /// <summary>
        /// Name of this extension.
        /// </summary>
        public string Name { get; }

        public IReadOnlyList<IDrawExtension> Requires { get; }

        /// <summary>
        /// The list of commands this extension defines, if any.
        /// </summary>
        public IReadOnlyList<IDrawCommand> Commands { get; }
    }

    /// <summary>
    /// A simple draw extension.
    /// </summary>
    public class DrawExtension : IDrawExtension
    {
        private readonly List<IDrawCommand> _drawCommands = new List<IDrawCommand>();

        public string Name { get; }

        public IReadOnlyList<IDrawCommand> Commands { get; }

        public IReadOnlyList<IDrawExtension> Requires { get; }

        public DrawExtension(string name, IEnumerable<IDrawExtension>? requires = null)
        {
            Name = name;
            Commands = _drawCommands.AsReadOnly();
            Requires = (requires ?? Enumerable.Empty<IDrawExtension>()).ToImmutableList();
        }

        protected void AddCommand(IDrawCommand command)
        {
            _drawCommands.Add(command);
        }
    }

    public static class DrawExtensionClass
    {
        /// <summary>
        /// Get the draw controller for the given queue.
        /// </summary>
        /// <param name="extension">The extension instance.</param>
        /// <param name="queue">The draw queue.</param>
        /// <returns>The draw controller for this queue.</returns>
        public static IDrawController GetController(this IDrawExtension extension, DrawQueue queue)
        {
            return queue.GetController(extension);
        }

        public static void Point(this DrawQueue queue, Vector3 position, float size, IBrush brush)
        {
            Vector3 radius = new Vector3(size / 2f);
            Box3d bounds = new Box3d(position - radius, position + radius);

            IDrawController controller = queue.GetController(DbBaseCommands.Instance);
            controller.EnsureBounds(bounds);
            controller.Write(DbBaseCommands.Instance.DrawPoint, new PointCommandArgs(position, size, brush));
        }

        public static void Line(this DrawQueue queue, Vector3 start, Vector3 end, float size, IBrush brush)
        {
            Vector3 radius = new Vector3(size / 2f);
            Vector3 min = Vector3.Min(start, end) - radius;
            Vector3 max = Vector3.Max(start, end) + radius;
            Box3d bounds = new Box3d(min, max);

            IDrawController controller = queue.GetController(DbBaseCommands.Instance);
            controller.EnsureBounds(bounds);
            controller.Write(DbBaseCommands.Instance.DrawLine, new LineCommandArgs(start, end, size, brush));
        }

        public static void Rect(this DrawQueue queue, Vector3 start, Vector3 end, IBrush fillBrush)
        {
            IDrawController controller = queue.GetController(DbBaseCommands.Instance);
            Vector3 min = Vector3.Min(start, end);
            Vector3 max = Vector3.Max(start, end);
            controller.EnsureBounds(new Box3d(min, max));
            controller.Write(DbBaseCommands.Instance.DrawRectF, new RectCommandArgs(start, end, fillBrush));
        }

        public static void Rect(this DrawQueue queue, Vector3 start, Vector3 end, IBrush strikeBrush, float strikeSize,
            StrikeKind kind = StrikeKind.Center)
        {
            IDrawController controller = queue.GetController(DbBaseCommands.Instance);
            Vector3 min = Vector3.Min(start, end);
            Vector3 max = Vector3.Max(start, end);
            controller.EnsureBounds(new Box3d(min, max));
            controller.Write(DbBaseCommands.Instance.DrawRectF, new RectCommandArgs(start, end, strikeBrush, strikeSize, kind));
        }

        public static void Rect(this DrawQueue queue, Vector3 start, Vector3 end, IBrush fillBrush, IBrush strikeBrush,
            float strikeSize, StrikeKind kind = StrikeKind.Center)
        {
            IDrawController controller = queue.GetController(DbBaseCommands.Instance);
            Vector3 min = Vector3.Min(start, end);
            Vector3 max = Vector3.Max(start, end);
            controller.EnsureBounds(new Box3d(min, max));
            controller.Write(DbBaseCommands.Instance.DrawRectF, new RectCommandArgs(start, end, fillBrush, strikeBrush, strikeSize, kind));
        }
    }
}