110 lines
2.4 KiB
C#
110 lines
2.4 KiB
C#
using System.Collections.Generic;
|
||
using System.IO;
|
||
using Godot;
|
||
using Learn.Models;
|
||
|
||
public partial class LearnTree : Tree
|
||
{
|
||
[Export]
|
||
private FileDirSelector _dirSelector;
|
||
|
||
public override void _Ready()
|
||
{
|
||
|
||
|
||
ItemSelected += OnItemSelected;
|
||
|
||
|
||
}
|
||
|
||
private void OnItemSelected()
|
||
{
|
||
TreeItem selected = GetSelected();
|
||
if (selected != null)
|
||
{
|
||
GD.Print($"选中了: {selected.GetText(0)}");
|
||
}
|
||
}
|
||
|
||
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);
|
||
}
|
||
}
|