85 lines
2.5 KiB
C#
85 lines
2.5 KiB
C#
|
using System;
|
|||
|
|
using System.Collections.Generic;
|
|||
|
|
using System.Linq;
|
|||
|
|
using System.Text;
|
|||
|
|
using System.Threading.Tasks;
|
|||
|
|
using SWS.CAD.Models;
|
|||
|
|
|
|||
|
|
namespace SWS.CAD.Helper
|
|||
|
|
{
|
|||
|
|
public class TreeHelper
|
|||
|
|
{
|
|||
|
|
#region 获取树节点
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 获取树节点
|
|||
|
|
/// </summary>
|
|||
|
|
public static TreeModel GetTreeModel(TreeModel node, string id)
|
|||
|
|
{
|
|||
|
|
// 如果根节点为空，则返回null
|
|||
|
|
if (node == null) return null;
|
|||
|
|
// 如果找到当前节点，返回当前节点
|
|||
|
|
if (node.ID == id) return node;
|
|||
|
|
//没有子节点就返回null
|
|||
|
|
if (node.ChildNodes == null) return null;
|
|||
|
|
// 否则，递归地在子节点中查找
|
|||
|
|
foreach (var child in node.ChildNodes)
|
|||
|
|
{
|
|||
|
|
var result = GetTreeModel(child, id);
|
|||
|
|
if (result != null)
|
|||
|
|
{
|
|||
|
|
return result; // 找到后立即返回
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
// 如果在当前树中没有找到，返回null
|
|||
|
|
return null;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#endregion
|
|||
|
|
|
|||
|
|
#region 获取树节点
|
|||
|
|
/// <summary>
|
|||
|
|
/// 获取树节点
|
|||
|
|
/// </summary>
|
|||
|
|
public static TreeModel GetTreeModel(List<TreeModel> nodes, string id)
|
|||
|
|
{
|
|||
|
|
TreeModel node = null;
|
|||
|
|
foreach (var dto in GlobalObject.objectTypeTree)
|
|||
|
|
{
|
|||
|
|
//获取节点下的图纸
|
|||
|
|
node = TreeHelper.GetTreeModel(dto, id);
|
|||
|
|
if (node != null)
|
|||
|
|
{
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return node;
|
|||
|
|
}
|
|||
|
|
#endregion
|
|||
|
|
|
|||
|
|
#region 获取树节点所有图纸名
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 获取树节点所有图纸名
|
|||
|
|
/// </summary>
|
|||
|
|
public static void GetTreeText(TreeModel node)
|
|||
|
|
{
|
|||
|
|
// 如果根节点为空，则返回null
|
|||
|
|
if (node == null) return ;
|
|||
|
|
//没有子节点就返回null
|
|||
|
|
if (node.ChildNodes == null) return ;
|
|||
|
|
// 否则，递归地在子节点中查找
|
|||
|
|
foreach (var child in node.ChildNodes)
|
|||
|
|
{
|
|||
|
|
GetTreeText(child);
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
if (node.Text.ToLower().EndsWith(".dwg") && !GlobalObject.AllDwgName.Contains(node.Text))
|
|||
|
|
{ GlobalObject.AllDwgName.Add(node.Text); }
|
|||
|
|
// 如果在当前树中没有找到，返回null
|
|||
|
|
return ;
|
|||
|
|
}
|
|||
|
|
#endregion
|
|||
|
|
}
|
|||
|
|
}
|