91 lines
3.5 KiB
C#
91 lines
3.5 KiB
C#
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Windows;
|
|
using System.Windows.Media;
|
|
|
|
namespace DMS.WPF.Helper
|
|
{
|
|
public static class VisualTreeFinder
|
|
{
|
|
/// <summary>
|
|
/// (按名称查找) 在可视化树中查找指定名称的特定类型 T 的子元素。
|
|
/// </summary>
|
|
/// <typeparam name="T">要查找的元素的类型，必须是 FrameworkElement。</typeparam>
|
|
/// <param name="parent">要开始查找的父元素。</param>
|
|
/// <param name="name">要查找的元素的 Name 属性。</param>
|
|
/// <returns>找到的第一个匹配的元素，如果未找到则返回 null。</returns>
|
|
public static T FindVisualChildByName<T>(DependencyObject parent, string name) where T : FrameworkElement
|
|
{
|
|
if (parent == null || string.IsNullOrEmpty(name))
|
|
return null;
|
|
|
|
int childrenCount = VisualTreeHelper.GetChildrenCount(parent);
|
|
for (int i = 0; i < childrenCount; i++)
|
|
{
|
|
var child = VisualTreeHelper.GetChild(parent, i);
|
|
|
|
// 检查当前子元素是否是我们要找的目标
|
|
if (child is T frameworkElement && frameworkElement.Name == name)
|
|
{
|
|
return frameworkElement;
|
|
}
|
|
|
|
// 如果不是，则递归进入子元素的子级中查找
|
|
var result = FindVisualChildByName<T>(child, name);
|
|
if (result != null)
|
|
{
|
|
// 如果在子级中找到了，立即返回结果，停止继续搜索
|
|
return result;
|
|
}
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
/// <summary>
|
|
/// (非泛型) 查找指定元素的所有可视化子元素。
|
|
/// </summary>
|
|
/// <param name="parent">要开始查找的父元素。</param>
|
|
/// <returns>一个包含所有找到的子元素的 IEnumerable<DependencyObject> 集合。</returns>
|
|
public static IEnumerable<DependencyObject> FindAllVisualChildren(DependencyObject parent)
|
|
{
|
|
if (parent == null)
|
|
yield break;
|
|
|
|
int childrenCount = VisualTreeHelper.GetChildrenCount(parent);
|
|
for (int i = 0; i < childrenCount; i++)
|
|
{
|
|
DependencyObject child = VisualTreeHelper.GetChild(parent, i);
|
|
|
|
// 先返回直接子元素
|
|
yield return child;
|
|
|
|
// 然后递归查找并返回其所有子元素
|
|
foreach (DependencyObject nestedChild in FindAllVisualChildren(child))
|
|
{
|
|
yield return nestedChild;
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// (泛型) 查找指定元素的可视化子元素中所有符合条件的特定类型 T 的元素。
|
|
/// </summary>
|
|
/// <typeparam name="T">要查找的子元素的类型，必须是 DependencyObject。</typeparam>
|
|
/// <param name="parent">要开始查找的父元素。</param>
|
|
/// <returns>一个包含所有找到的子元素的 IEnumerable<T> 集合。</returns>
|
|
public static IEnumerable<T> FindVisualChildren<T>(DependencyObject parent) where T : DependencyObject
|
|
{
|
|
// 复用 FindAllVisualChildren 的逻辑，避免代码重复
|
|
foreach (var child in FindAllVisualChildren(parent))
|
|
{
|
|
if (child is T typedChild)
|
|
{
|
|
yield return typedChild;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|