Check for Visual or Visual3D in visual ancestry (#2134) (#2276)

Author: greuelpirat

MaterialDesignThemes.Wpf/Extensions.cs

@@ -3,6 +3,7 @@
 using System.Linq;
 using System.Windows;
 using System.Windows.Media;
+using System.Windows.Media.Media3D;
 
 namespace MaterialDesignThemes.Wpf
 {
@@ -50,6 +51,8 @@ public static bool IsAncestorOf(this DependencyObject parent, DependencyObject?
 
         /// <summary>
         /// Returns full visual ancestry, starting at the leaf.
+        /// <para>If element is not of <see cref="Visual"/> or <see cref="Visual3D"/> the
+        /// logical ancestry is used.</para>
         /// </summary>
         /// <param name="leaf"></param>
         /// <returns></returns>
@@ -58,7 +61,9 @@ public static IEnumerable<DependencyObject> GetVisualAncestry(this DependencyObj
             while (leaf is not null)
             {
                 yield return leaf;
-                leaf = VisualTreeHelper.GetParent(leaf);
+                leaf = leaf is Visual || leaf is Visual3D
+                    ? VisualTreeHelper.GetParent(leaf)
+                    : LogicalTreeHelper.GetParent(leaf);
             }
         }