1
0
mirror of synced 2024-12-03 19:47:29 +01:00
Switch-Toolbox/Switch_Toolbox_Library/Forms/Custom/TreeViewCustom.cs

248 lines
8.7 KiB
C#

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Forms;
using System.Drawing;
using System.Runtime.InteropServices;
using Switch_Toolbox.Library.Forms;
namespace Switch_Toolbox.Library
{
public abstract class TreeNodeCustom : TreeNode
{
public virtual void OnClick(TreeView treeview) { }
public virtual void OnMouseLeftClick(TreeView treeview) { }
public virtual void OnMouseRightClick(TreeView treeview) { }
public virtual void OnDoubleMouseClick(TreeView treeview) { }
public TreeNodeCustom()
{
}
}
public class TreeNodeFile : TreeNodeCustom
{
public bool CanDelete
{
set
{
if (value == true)
{
if (ContextMenuStrip == null)
ContextMenuStrip = new STContextMenuStrip();
ContextMenuStrip.Items.Add(new STToolStipMenuItem("Delete", null, Delete, Keys.Control | Keys.Delete));
}
}
}
public TreeNodeFile()
{
}
public TreeNodeFile(string text)
{
Text = text;
}
private void Delete(object sender, EventArgs args)
{
var editor = LibraryGUI.Instance.GetObjectEditor();
if (editor != null)
{
editor.RemoveFile(this);
editor.ResetControls();
}
}
public virtual void OnAfterAdded() //After added to treeview
{
}
}
public class TreeViewCustom : TreeView
{
private readonly Dictionary<int, TreeNode> _treeNodes = new Dictionary<int, TreeNode>();
public TreeViewCustom()
{
ReloadImages();
BackColor = FormThemes.BaseTheme.FormBackColor;
ForeColor = FormThemes.BaseTheme.FormForeColor;
}
public static ImageList imgList = new ImageList();
public void AddImage(Image image)
{
imgList.Images.Add(image);
}
public int ImageWidth = 21;
public int ImageHeight = 21;
public void ReloadImages()
{
imgList = new ImageList();
imgList.ColorDepth = ColorDepth.Depth32Bit;
imgList.ImageSize = new Size(ImageWidth, ImageHeight);
imgList.Images.Add("folder", Properties.Resources.Folder);
imgList.Images.Add("resource", Properties.Resources.Folder);
imgList.Images.Add("Texture", Properties.Resources.Texture);
imgList.Images.Add("fileBlank", Properties.Resources.FileBlank);
imgList.Images.Add("bfres", Properties.Resources.Bfres);
imgList.Images.Add("byaml", Properties.Resources.Byaml);
imgList.Images.Add("aamp", Properties.Resources.Aamp);
imgList.Images.Add("bntx", Properties.Resources.Bntx);
imgList.Images.Add("bfsha", Properties.Resources.Bfsha);
imgList.Images.Add("bnsh", Properties.Resources.Bnsh);
imgList.Images.Add("mesh", Properties.Resources.mesh);
imgList.Images.Add("skeletonAnimation", Properties.Resources.skeletonAnimation);
imgList.Images.Add("bone", Properties.Resources.Bone);
imgList.Images.Add("bfwav", Properties.Resources.Music1);
imgList.Images.Add("bfstp", Properties.Resources.Music2);
imgList.Images.Add("material", Properties.Resources.materialSphere);
imgList.Images.Add("model", Properties.Resources.model);
imgList.Images.Add("folder", Properties.Resources.skeleton);
imgList.Images.Add("TextureMaterialMap", Properties.Resources.TextureMaterialMap);
imgList.Images.Add("MaterialTranslucent", Properties.Resources.materialSphereTranslucent);
imgList.Images.Add("MaterialTransparent", Properties.Resources.materialSphereTransparent);
//Data types
imgList.Images.Add("bool", Properties.Resources.IconBool);
imgList.Images.Add("buffer", Properties.Resources.IconBuffer);
imgList.Images.Add("curve", Properties.Resources.IconCurve);
imgList.Images.Add("float", Properties.Resources.IconFloat);
imgList.Images.Add("int", Properties.Resources.IconInt);
imgList.Images.Add("list", Properties.Resources.IconList);
imgList.Images.Add("object", Properties.Resources.IconObject);
imgList.Images.Add("vec2", Properties.Resources.IconVec2);
imgList.Images.Add("vec3", Properties.Resources.IconVec3);
imgList.Images.Add("vec4", Properties.Resources.IconVec4);
Bitmap bmp = new Bitmap(32, 32);
Graphics g = Graphics.FromImage(bmp);
g.Clear(Color.Transparent);
imgList.Images.Add("empty", bmp);
this.ImageList = imgList;
}
/// <summary>
/// Load the TreeView with items.
/// </summary>
/// <typeparam name="T">Item type</typeparam>
/// <param name="items">Collection of items</param>
/// <param name="getId">Function to parse Id value from item object</param>
/// <param name="getParentId">Function to parse parentId value from item object</param>
/// <param name="getDisplayName">Function to parse display name
/// value from item object. This is used as node text.</param>
public void LoadItems<T>(IEnumerable<T> items, Func<T, int> getId,
Func<T, int?> getParentId, Func<T, string> getDisplayName)
{
// Clear view and internal dictionary
Nodes.Clear();
_treeNodes.Clear();
// Load internal dictionary with nodes
foreach (var item in items)
{
var id = getId(item);
var displayName = getDisplayName(item);
var node = new TreeNode
{
Name = id.ToString(),
Text = displayName,
Tag = item
};
_treeNodes.Add(getId(item), node);
}
// Create hierarchy and load into view
foreach (var id in _treeNodes.Keys)
{
var node = GetNode(id);
var obj = (T)node.Tag;
var parentId = getParentId(obj);
if (parentId.HasValue)
{
var parentNode = GetNode(parentId.Value);
parentNode.Nodes.Add(node);
}
else
{
Nodes.Add(node);
}
}
}
/// <summary>
/// Retrieve TreeNode by Id.
/// Useful when you want to select a specific node.
/// </summary>
/// <param name="id">Item id</param>
public TreeNode GetNode(int id)
{
return _treeNodes[id];
}
}
public static class TreeViewExtensions
{
public static IEnumerable<TreeNode> Collect(TreeNodeCollection nodes)
{
foreach (TreeNode node in nodes)
{
yield return node;
foreach (var child in Collect(node.Nodes))
yield return child;
}
}
private const int TVIF_STATE = 0x8;
private const int TVIS_STATEIMAGEMASK = 0xF000;
private const int TV_FIRST = 0x1100;
private const int TVM_SETITEM = TV_FIRST + 63;
[StructLayout(LayoutKind.Sequential, Pack = 8, CharSet = CharSet.Auto)]
private struct TVITEM
{
public int mask;
public IntPtr hItem;
public int state;
public int stateMask;
[MarshalAs(UnmanagedType.LPTStr)]
public string lpszText;
public int cchTextMax;
public int iImage;
public int iSelectedImage;
public int cChildren;
public IntPtr lParam;
}
[DllImport("user32.dll", CharSet = CharSet.Auto)]
private static extern IntPtr SendMessage(IntPtr hWnd, int Msg, IntPtr wParam,
ref TVITEM lParam);
/// <summary>
/// Hides the checkbox for the specified node on a TreeView control.
/// </summary>
public static void HideCheckBox(this TreeNode node)
{
TVITEM tvi = new TVITEM();
tvi.hItem = node.Handle;
tvi.mask = TVIF_STATE;
tvi.stateMask = TVIS_STATEIMAGEMASK;
tvi.state = 0;
SendMessage(node.TreeView.Handle, TVM_SETITEM, IntPtr.Zero, ref tvi);
}
}
}