UnityGame/Library/PackageCache/com.unity.inputsystem/InputSystem/Editor/Internal/TreeViewHelpers.cs

52 lines
1.4 KiB
C#
Raw Normal View History

2024-10-27 10:53:47 +03:00
#if UNITY_EDITOR
using System;
using UnityEditor.IMGUI.Controls;
namespace UnityEngine.InputSystem.Editor
{
/// <summary>
/// Extension methods for working with tree views.
/// </summary>
/// <seealso cref="TreeView"/>
internal static class TreeViewHelpers
{
public static TItem TryFindItemInHierarchy<TItem>(this TreeViewItem item)
where TItem : TreeViewItem
{
while (item != null)
{
if (item is TItem result)
return result;
item = item.parent;
}
return null;
}
public static bool IsParentOf(this TreeViewItem parent, TreeViewItem child)
{
if (parent == null)
throw new ArgumentNullException(nameof(parent));
if (child == null)
throw new ArgumentNullException(nameof(child));
do
{
child = child.parent;
}
while (child != null && child != parent);
return child != null;
}
public static void ExpandChildren(this TreeView treeView, TreeViewItem item)
{
if (!item.hasChildren)
return;
foreach (var child in item.children)
treeView.SetExpanded(child.id, true);
}
}
}
#endif // UNITY_EDITOR