/
NodeFilter.cs
77 lines (66 loc) · 2.73 KB
/
NodeFilter.cs
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
using System.Collections.Generic;
using System.Drawing;
using System.Linq;
using System.Windows.Forms;
using SwqlStudio.Metadata;
using SwqlStudio.Properties;
namespace SwqlStudio.Filtering
{
internal class NodeFilter
{
private readonly List<INodeFilterStrategy> _filters = new List<INodeFilterStrategy>();
private const int _limitOfExpandedNodes = 5000;
private int _visibleNodesCount;
public NodeFilter(TreeView treeData, string filter)
{
InitializeFilters(treeData, filter);
}
private void InitializeFilters(TreeView treeData, string filter)
{
if (!Settings.Default.ShowObsolete)
{
_filters.Add(new ObsoleteNodeFilter());
}
if (filter != null)
{
_filters.Add(new SearchNodeFilter(filter));
}
TreeNodeUtils.IterateNodes(treeData, node => _filters.ForEach(strategy => strategy.Initialize(node)));
TreeNodeUtils.IterateNodes(treeData, node =>
{
if (FilterAction(node)) _visibleNodesCount++;
});
}
private VisibilityStatus NodeVisibility(TreeNode node) => _filters.Aggregate(VisibilityStatus.None, (current, filter) => current | filter.GetVisibility(node));
internal bool FilterAction(TreeNode node) => !NodeVisibility(node).HasFlag(VisibilityStatus.NotVisible);
internal void UpdateAction(TreeNode data, TreeNode display)
{
var nodeVisibility = NodeVisibility(data);
if (nodeVisibility.HasFlag(VisibilityStatus.ChildVisible))
// this one is not directly visible, gray it out
display.ForeColor = Color.LightBlue;
else if (nodeVisibility.HasFlag(VisibilityStatus.ParentVisible))
// this one is not directly visible, gray it out
display.ForeColor = Color.DarkGray;
// this one is visible directly, ensure it is visible (parents are expanded).
// it's children are visible as well, but may be not expanded
else if (nodeVisibility.HasFlag(VisibilityStatus.Visible))
{
//expand parents of visible node
if (_visibleNodesCount < _limitOfExpandedNodes)
{
display.EnsureVisible();
}
//If too many objects are visible, treeView performance drops dramatically.
//In case of many object only entity level is expanded
else
{
if (data.Tag is Entity)
{
display.EnsureVisible();
}
}
}
}
}
}