blob: 0859affc225ddfd3244379a0da3666d21cd02a99 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Controls;
public static class TreeViewExtensions
{
public static void SetExpansion(this TreeView treeView, bool isExpanded) => SetExpansion((ItemsControl)treeView, isExpanded);
private static void SetExpansion(ItemsControl parent, bool isExpanded)
{
if (parent == null) return;
if (parent is TreeViewItem tvi)
tvi.IsExpanded = isExpanded;
if (parent.HasItems)
foreach (var item in parent.Items.Cast<object>()
.Select(i => GetTreeViewItem(parent, i, isExpanded)))
SetExpansion(item, isExpanded);
}
private static TreeViewItem GetTreeViewItem(ItemsControl parent, object item, bool isExpanded)
{
if (item is TreeViewItem tvi)
return tvi;
var result = ContainerFromItem(parent, item);
if (result == null && isExpanded)
{
parent.UpdateLayout();
result = ContainerFromItem(parent, item);
}
return result;
}
static TreeViewItem ContainerFromItem(ItemsControl parent, object item) => (TreeViewItem)parent.ItemContainerGenerator.ContainerFromItem(item);
}
|
