KOP_PIbd-33_Volkov_N.A._Tik.../VolkovLabs/WinFormsLibraryVolkov/CustomTreeCell.cs

118 lines
3.5 KiB
C#

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Drawing;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Forms;
namespace WinFormsLibraryVolkov
{
public partial class CustomTreeCell : UserControl
{
public CustomTreeCell()
{
InitializeComponent();
}
public void Clear()
{
treeView.Nodes.Clear();
}
//Публичное свойство для установки и получения индекса выбранной ветки(set, get).
protected DataTreeNodeConfig Levels { get; set; }
public void LoadConfig(DataTreeNodeConfig levels)
{
if (levels != null)
{
Levels = levels;
}
}
public T GetSelectedObject<T>() where T : class, new()
{
if (treeView.SelectedNode == null || Levels == null || treeView.SelectedNode.Nodes.Count > 0)
{
return null;
}
Stack<string> stack = new Stack<string>();
for (TreeNode treeNode = treeView.SelectedNode; treeNode != null; treeNode = treeNode.Parent)
{
stack.Push(treeNode.Text);
}
if (stack.Count != Levels.NodeNames.Count)
{
return null;
}
T val = new T();
foreach (string nodeName in Levels.NodeNames)
{
PropertyInfo property = val.GetType().GetProperty(nodeName);
string value = stack.Pop();
property?.SetValue(val, Convert.ChangeType(value, property?.PropertyType));
}
return val;
}
public void AddCells<T>(int columnIndex, List<T> elements)
{
if (elements == null || elements.Count == 0)
{
return;
}
foreach (T element in elements)
{
AddCell(columnIndex, element);
}
}
// Обновленный метод для добавления одного элемента в дерево
public void AddCell<T>(int columnIndex, T element)
{
if (Levels == null || element == null || columnIndex < 0 || columnIndex >= Levels.NodeNames.Count)
{
return;
}
TreeNodeCollection treeNodeCollection = treeView.Nodes;
int num = 0;
foreach (string nodeName in Levels.NodeNames)
{
// Получение значения свойства или поля
PropertyInfo property = element.GetType().GetProperty(nodeName);
string text = property?.GetValue(element, null)?.ToString() ?? nodeName;
TreeNode treeNode = null;
foreach (TreeNode item in treeNodeCollection)
{
if (item.Text == text)
{
treeNode = item;
break;
}
}
// Если узел не найден, добавляем новый
treeNodeCollection = (treeNode == null) ? treeNodeCollection.Add(text).Nodes : treeNode.Nodes;
if (num >= columnIndex)
{
break;
}
num++;
}
}
}
}