WPF revit二次开发使用树形结构显示revit数据

在功能研发时，需要使用属性结构让revit本身的数据结构更加的清晰，再写代码时采用了mvvm模式对，将数据采集整理再后端完成。利用递归的方式将整个层级进行分类。
本文参照了：https://www.cnblogs.com/yeyunfei/p/5408931.html
https://www.jianshu.com/p/908070cca109
https://www.cnblogs.com/larissa-0464/p/10227483.html
的案例。
xaml代码：

<TreeView Name="TreeView" ItemsSource="{Binding TreeNodes}" HorizontalAlignment="Left" Height="325" Margin="25,60,0,0" VerticalAlignment="Top" Width="220">
            <TreeView.ItemTemplate>
                <HierarchicalDataTemplate  ItemsSource="{Binding Path= Nodes}">
                    <TextBlock Text="{Binding ElementName}"></TextBlock>
                </HierarchicalDataTemplate>
            </TreeView.ItemTemplate>
        </TreeView>

xaml.cs代码：

public MainWindow(List<Command.TreeNode> fds)
      {
          InitializeComponent();
          Datas = fds;
    
          TreeView.DataContext = new WindowData();
    
      }
      
      public class WindowData : INotifyPropertyChanged
      {
          public event PropertyChangedEventHandler PropertyChanged;
          private List<Command.TreeNode> treeNodes = new List<Command.TreeNode>();
    
          public List<Command.TreeNode> TreeNodes
          {
              get { return treeNodes; }
              set
              {
                  treeNodes = value;
                  if (PropertyChanged != null)
                      PropertyChanged(this, new PropertyChangedEventArgs("TreeNodes"));
              }
          }
          public WindowData()
          {
              TreeNodes = GetFamilyData(0, Datas);
          }
    
          private List<Command.TreeNode> GetFamilyData(double parenId, List<Command.TreeNode> datas)
          {
              List<Command.TreeNode> mainDatas = datas.Where(t => t.ParentId == parenId).ToList<Command.TreeNode>();
              List<Command.TreeNode> secondDatas = datas.Where(t => t.ParentId != parenId).ToList<Command.TreeNode>();
              foreach (Command.TreeNode fd in mainDatas)
              {
                  fd.Nodes = GetFamilyData(fd.Id, secondDatas);
              }
    
              return mainDatas;
          }
      }

后端处理数据代码，感觉自动处理代码有点复杂，后期进行优化：

private List<TreeNode> GetTreeNodes(List<FamilyData> fds)
        {
            // The First ID 
            int v = 1;
            List<TreeNode> treeNodes = new List<TreeNode>();
            for (int i = 0; i < fds.Count; i++)
            {
                TreeNode tn = new TreeNode();
                tn.Id = v;
                tn.ParentId = 0;
                tn.ElementName = fds[i].Category.Name;
                v++;
                treeNodes.Add(tn);
                foreach (var dic in fds[i].FamilyAndSymbol)
                {
                    TreeNode tn2 = new TreeNode();
                    tn2.ParentId = tn.Id;
                    tn2.Id = v;
                    v++;
                    tn2.ElementName = dic.Key.Name;
                    treeNodes.Add(tn2);
                    foreach (FamilySymbol sym in dic.Value)
                    {
                        TreeNode tn3 = new TreeNode();
                        tn3.ParentId = tn2.Id;
                        tn3.Id = v;
                        tn3.ElementName = sym.Name;
                        v++;
                        treeNodes.Add(tn3);
                    }
                }
                
            }

            return treeNodes;
        }

数据结构：

public class TreeNode
     {
         public double ParentId { get; set; }
         public string ElementName { get; set; }
         public double Id { get; set; }
         public ElementId ElementId { get; set; }
         public List<TreeNode> Nodes { get; set; }
    
         public TreeNode()
         {
             Nodes = new List<TreeNode>();
         }
     }