Avalonia/src/Avalonia.Base/Rendering/SceneGraph/RectangleNode.cs

using System;
using System.Collections.Generic;
using Avalonia.Media;
using Avalonia.Media.Immutable;
using Avalonia.Platform;
using Avalonia.VisualTree;

namespace Avalonia.Rendering.SceneGraph
{
    /// <summary>
    /// A node in the scene graph which represents a rectangle draw.
    /// </summary>
    internal class RectangleNode : BrushDrawOperation
    {
        /// <summary>
        /// Initializes a new instance of the <see cref="RectangleNode"/> class.
        /// </summary>
        /// <param name="transform">The transform.</param>
        /// <param name="brush">The fill brush.</param>
        /// <param name="pen">The stroke pen.</param>
        /// <param name="rect">The rectangle to draw.</param>
        /// <param name="boxShadows">The box shadow parameters</param>
        public RectangleNode(
            Matrix transform,
            IImmutableBrush? brush,
            IPen? pen,
            RoundedRect rect,
            BoxShadows boxShadows)
            : base(boxShadows.TransformBounds(rect.Rect).Inflate((pen?.Thickness ?? 0) / 2), transform, brush)
        {
            Pen = pen?.ToImmutable();
            Rect = rect;
            BoxShadows = boxShadows;
        }

        /// <summary>
        /// Gets the stroke pen.
        /// </summary>
        public ImmutablePen? Pen { get; }

        /// <summary>
        /// Gets the rectangle to draw.
        /// </summary>
        public RoundedRect Rect { get; }
        
        /// <summary>
        /// The parameters for the box-shadow effect
        /// </summary>
        public BoxShadows BoxShadows { get; }

        /// <summary>
        /// Determines if this draw operation equals another.
        /// </summary>
        /// <param name="transform">The transform of the other draw operation.</param>
        /// <param name="brush">The fill of the other draw operation.</param>
        /// <param name="pen">The stroke of the other draw operation.</param>
        /// <param name="rect">The rectangle of the other draw operation.</param>
        /// <param name="boxShadows">The box shadow parameters of the other draw operation</param>
        /// <returns>True if the draw operations are the same, otherwise false.</returns>
        /// <remarks>
        /// The properties of the other draw operation are passed in as arguments to prevent
        /// allocation of a not-yet-constructed draw operation object.
        /// </remarks>
        public bool Equals(Matrix transform, IBrush? brush, IPen? pen, RoundedRect rect, BoxShadows boxShadows)
        {
            return transform == Transform &&
                   Equals(brush, Brush) &&
                   Equals(Pen, pen) &&
                   BoxShadows.Equals(boxShadows) &&
                   rect.Equals(Rect);
        }

        /// <inheritdoc/>
        public override void Render(IDrawingContextImpl context) => context.DrawRectangle(Brush, Pen, Rect, BoxShadows);

        /// <inheritdoc/>
        public override bool HitTestTransformed(Point p)
        {
            if (Brush != null)
            {
                var rect = Rect.Rect.Inflate((Pen?.Thickness / 2) ?? 0);
                return rect.ContainsExclusive(p);
            }
            else
            {
                var borderRect = Rect.Rect.Inflate((Pen?.Thickness / 2) ?? 0);
                var emptyRect = Rect.Rect.Deflate((Pen?.Thickness / 2) ?? 0);
                return borderRect.ContainsExclusive(p) && !emptyRect.ContainsExclusive(p);
            }
        }
    }
}