提交

LearnTree.cs

@@ -1,4 +1,7 @@
+using System.Collections.Generic;
+using System.IO;
 using Godot;
+using Learn.Models;
 
 public partial class LearnTree : Tree
 {
@@ -7,7 +10,11 @@ public partial class LearnTree : Tree
 	
 	public override void _Ready()
 	{
+		
+		
 		ItemSelected += OnItemSelected;
+		
+		
 	}
 
 	private void OnItemSelected()
@@ -19,10 +26,84 @@ public partial class LearnTree : Tree
 		}
 	}
 
+	private void Setup()
+	{
+		// 1. 设置列数
+		Columns = 2;
+        
+		// 设置标题 (需要在属性面板开启 Column Title Visible，或者代码开启)
+		SetColumnTitle(0, "角色 (Role)");
+		SetColumnTitle(1, "状态 (Status)");
+		SetColumnTitlesVisible(true);
+
+		// 2. 创建根节点 (Root)
+		// CreateItem() 不传参数即为根节点
+		TreeItem root = CreateItem();
+		root.SetText(0, "所有单位");
+
+		// 3. 创建子节点
+		// 传入父节点作为参数
+		TreeItem warrior = CreateItem(root);
+		warrior.SetText(0, "战士");
+		warrior.SetText(1, "HP: 100");
+
+		// 4. 创建层级更深的节点
+		TreeItem sword = CreateItem(warrior);
+		sword.SetText(0, "铁剑");
+		sword.SetText(1, "攻击力: 12");
+	}
+
+	private readonly Dictionary<TreeItem, Item> _mapper = new();
+
+	private TreeItem CreateNode(TreeItem father, Item item)
+	{
+		var node = CreateItem(father);
+        node.SetText(0, item.Info["Name"]);
+        return node;
+	}
+	
+	private Item InitItem(string path, bool isDir)
+	{
+		var name = Path.GetFileName(path);
+		var item = new Item();
+		item.Info["Name"] = name;
+		item.Info["IsDir"] = isDir ? "True" : "False";
+		return item;
+	}
+	
+	private void BuildTree(TreeItem root, string path)
+	{
+		var nodes = new Stack<(TreeItem node, string path)>();
+		nodes.Push((root, path));
+		while (nodes.Count > 0)
+		{
+			var node = nodes.Pop();
+
+			foreach (var filePath in Directory.GetFiles(node.path))
+			{
+				var item = InitItem(filePath, false);
+				var chNode = CreateNode(node.node, item);
+				_mapper[chNode] = item;
+			}
+			
+			foreach (var subDirPath in Directory.GetDirectories(node.path))
+			{
+				var item = InitItem(subDirPath, true);
+				var chNode = CreateNode(node.node, item);
+				_mapper[chNode] = item;
+				nodes.Push((chNode, subDirPath));
+			}
+		}
+	}
+	
 	private async void Open()
 	{
 		var path = await _dirSelector.SelectFolderAsync() ?? "None";
 		if (string.IsNullOrEmpty(path)) return;
+		Clear();
+		_mapper.Clear();
 		
+		TreeItem root = CreateItem();
+		BuildTree(root, path);
 	}
 }

Models/Item.cs

@@ -0,0 +1,8 @@
+using Godot.Collections;
+
+namespace Learn.Models;
+
+public class Item
+{
+    public Dictionary<string, string> Info { get; } = new ();
+}

Models/Item.cs.uid

@@ -0,0 +1 @@
+uid://44p6d5b3xerf