隐藏树对象递归的节点

Question

我正在寻找一种在树中隐藏节点的方法。 我正在过滤具有节点类型的节点，这是我的原始结构：

root
 +FolderA
    -File1
    -File2
    -File3
 +FolderB
    -File4
    -File5
    +FolderB-1
            -File6

这就是我想要做的：

+root
 -File1
 -File2
 -File3
 -File4
 -File5
 -File6

我创建了我的类 XNode，其中包含一个类型和一个子项列表。 （这是伪语言的缩写）

Class XNode
  MyType type;
  string Name;
  List<XNode> childrens; 
End Class

我创建了一个工厂类来查询我的数据库并创建原始树。

我创建了一个 Xaml treeView UserControl 来绑定我创建的根 XNode。

无法通过 Xaml 隐藏节点，此操作必须使用绑定对象（我创建的根 Xnode）完成。

现在我的问题是： 有递归算法可以取消“文件夹类型”节点，获取其子节点并将它们添加到父节点？

我试试：

  public XNode RemoveFoldersElements(ref XNode rootNode)
    {

        if (rootNode != null)
        {
            if (rootNode.Children.Count > 0)
            {
                for (int i = 0; i < rootNode.Children.Count; i++)
                {
                    XNode children = rootNode.Children.ElementAt(i);
                    if (children.WType == NodeType.Type_FOLDER)
                    {
                        XNode tempNode = RemoveFoldersElements(ref children);
                        if (tempNode != null)
                            rootNode.Children.Add(tempNode);
                        rootNode.Children.RemoveAt(i);
                    }
                    else
                    {
                        RemoveFoldersElements(ref children);
                    }

                }
            }

        }
        return null;
    }

但不是成功，因为并非所有 Type_FOLDER 节点都被擦除（它跳转一级）！ 有什么想法吗？

Answer 1

好的，想法是这样，在过滤的同时展平树：

public static IEnumerable<TSource> Flatten<TSource>(this IEnumerable<TSource> source, Func<TSource, IEnumerable<TSource>> childCollectionSelector, Func<TSource, bool> predicate)
{
    foreach (var item in source)
    {
        if(predicate(item))
            yield return item;
        foreach (var subitem in childCollectionSelector(item).Flatten(childCollectionSelector, predicate).Where(predicate))
        {
            yield return subitem;
        }
    }
}

你可以这样使用它：

var rootList = new []{rootElement}; // Flattening works on the collection
var files = rootList.Flatten(x => x.children, x => x.type == MyType.FILE);

如果您没有任何循环引用并且所有集合都已初始化，它将起作用。如果你确实有循环依赖，你会向StackOverflowException打个招呼...