Shrlalgo.RvKits/WPFluent/Controls/Primitives/IconElementEx.cs

using WPFluent.Converters;

using System;
using System.ComponentModel;
using System.Windows;
using System.Windows.Controls;
using System.Windows.Data;
using System.Windows.Documents;
using System.Windows.Media;

using Grid = System.Windows.Controls.Grid;

namespace WPFluent.Controls.Primitives;

[TypeConverter(typeof(IconElementConverter))]
public abstract class IconElementEx : IconElement
{
    private protected IconElementEx()
    {
    }

    protected override void OnForegroundChanged(DependencyPropertyChangedEventArgs args)
    {
        var baseValueSource = DependencyPropertyHelper.GetValueSource(this, args.Property).BaseValueSource;
        _isForegroundDefaultOrInherited = baseValueSource <= BaseValueSource.Inherited;
        UpdateShouldInheritForegroundFromVisualParent();
    }

    private static readonly DependencyProperty VisualParentForegroundProperty =
        DependencyProperty.Register(
            nameof(VisualParentForeground),
            typeof(Brush),
            typeof(IconElementEx),
            new PropertyMetadata(null, OnVisualParentForegroundPropertyChanged));

    protected Brush VisualParentForeground
    {
        get => (Brush)GetValue(VisualParentForegroundProperty);
        set => SetValue(VisualParentForegroundProperty, value);
    }

    private static void OnVisualParentForegroundPropertyChanged(DependencyObject sender, DependencyPropertyChangedEventArgs args)
    {
        ((IconElementEx)sender).OnVisualParentForegroundPropertyChanged(args);
    }

    private protected virtual void OnVisualParentForegroundPropertyChanged(DependencyPropertyChangedEventArgs args)
    {
    }

    protected bool ShouldInheritForegroundFromVisualParent
    {
        get => _shouldInheritForegroundFromVisualParent;
        private set
        {
            if (_shouldInheritForegroundFromVisualParent != value)
            {
                _shouldInheritForegroundFromVisualParent = value;

                if (_shouldInheritForegroundFromVisualParent)
                {
                    SetBinding(VisualParentForegroundProperty,
                        new Binding
                        {
                            Path = new PropertyPath(TextElement.ForegroundProperty),
                            Source = VisualParent
                        });
                }
                else
                {
                    ClearValue(VisualParentForegroundProperty);
                }

                OnShouldInheritForegroundFromVisualParentChanged();
            }
        }
    }

    private protected virtual void OnShouldInheritForegroundFromVisualParentChanged()
    {
    }

    private void UpdateShouldInheritForegroundFromVisualParent()
    {
        ShouldInheritForegroundFromVisualParent =
            _isForegroundDefaultOrInherited &&
            Parent != null &&
            VisualParent != null &&
            Parent != VisualParent;
    }

    protected UIElementCollection Children
    {
        get
        {
            EnsureLayoutRoot();
            return _layoutRoot.Children;
        }
    }

    protected override int VisualChildrenCount => 1;

    protected override Visual GetVisualChild(int index)
    {
        if (index == 0)
        {
            EnsureLayoutRoot();
            return _layoutRoot;
        }
        else
        {
            throw new ArgumentOutOfRangeException(nameof(index));
        }
    }

    protected override Size MeasureOverride(Size availableSize)
    {
        EnsureLayoutRoot();
        _layoutRoot.Measure(availableSize);
        return _layoutRoot.DesiredSize;
    }

    protected override Size ArrangeOverride(Size finalSize)
    {
        EnsureLayoutRoot();
        _layoutRoot.Arrange(new Rect(new Point(), finalSize));
        return finalSize;
    }

    protected override void OnVisualParentChanged(DependencyObject oldParent)
    {
        base.OnVisualParentChanged(oldParent);
        UpdateShouldInheritForegroundFromVisualParent();
    }

    private void EnsureLayoutRoot()
    {
        if (_layoutRoot != null)
            return;

        _layoutRoot = new Grid
        {
            Background = Brushes.Transparent,
            SnapsToDevicePixels = true,
        };
        InitializeChildren();

        AddVisualChild(_layoutRoot);
    }

    private Grid _layoutRoot = null!;
    private bool _isForegroundDefaultOrInherited = true;
    private bool _shouldInheritForegroundFromVisualParent;
}