Merge pull request #91 from pancake-llc/main

Add: tree map and row separator for hierarchy

Alchemy/Assets/Alchemy/Editor/AlchemySettings.cs

@@ -68,6 +68,32 @@ internal static SettingsProvider CreateSettingsProvider()
  EditorGUILayout.PropertyField(serializedObject.FindProperty("hierarchyObjectMode"));
  EditorGUILayout.PropertyField(serializedObject.FindProperty("showHierarchyToggles"), new GUIContent("Show Toggles"));
  EditorGUILayout.PropertyField(serializedObject.FindProperty("showComponentIcons"));
+ var showTreeMap = serializedObject.FindProperty("showTreeMap");
+ EditorGUILayout.PropertyField(showTreeMap);
+ if (showTreeMap.boolValue)
+ {
+ EditorGUI.indentLevel++;
+ EditorGUILayout.PropertyField(serializedObject.FindProperty("treeMapColor"), new GUIContent("Color"));
+ EditorGUI.indentLevel--;
+ }
+
+ var showSeparator = serializedObject.FindProperty("showSeparator");
+ EditorGUILayout.PropertyField(showSeparator, new GUIContent("Show Row Separator"));
+ if (showSeparator.boolValue)
+ {
+ EditorGUI.indentLevel++;
+ EditorGUILayout.PropertyField(serializedObject.FindProperty("separatorColor"), new GUIContent("Color"));
+ EditorGUI.indentLevel--;
+ var showRowShading = serializedObject.FindProperty("showRowShading");
+ EditorGUILayout.PropertyField(showRowShading);
+ if (showRowShading.boolValue)
+ {
+ EditorGUI.indentLevel++;
+ EditorGUILayout.PropertyField(serializedObject.FindProperty("evenRowColor"));
+ EditorGUILayout.PropertyField(serializedObject.FindProperty("oddRowColor"));
+ EditorGUI.indentLevel--;
+ }
+ }
 
  if (changeCheck.changed)
  {
@@ -84,9 +110,23 @@ internal static SettingsProvider CreateSettingsProvider()
  [SerializeField] HierarchyObjectMode hierarchyObjectMode = HierarchyObjectMode.RemoveInBuild;
  [SerializeField] bool showHierarchyToggles;
  [SerializeField] bool showComponentIcons;
+ [SerializeField] bool showTreeMap;
+ [SerializeField] Color treeMapColor = new(0.53f, 0.53f, 0.53f, 0.45f);
+ [SerializeField] bool showSeparator;
+ [SerializeField] bool showRowShading;
+ [SerializeField] Color separatorColor = new(0.19f, 0.19f, 0.19f, 0f);
+ [SerializeField] Color evenRowColor = new(0f, 0f, 0f, 0.07f);
+ [SerializeField] Color oddRowColor = Color.clear;
 
  public HierarchyObjectMode HierarchyObjectMode => hierarchyObjectMode;
  public bool ShowHierarchyToggles => showHierarchyToggles;
  public bool ShowComponentIcons => showComponentIcons;
+ public bool ShowTreeMap => showTreeMap;
+ public Color TreeMapColor => treeMapColor;
+ public bool ShowSeparator => showSeparator;
+ public bool ShowRowShading => showRowShading;
+ public Color SeparatorColor => separatorColor;
+ public Color EvenRowColor => evenRowColor;
+ public Color OddRowColor => oddRowColor;
  }
 }

Alchemy/Assets/Alchemy/Editor/Hierarchy/HierarchyRowSeparatorDrawer.cs

@@ -0,0 +1,24 @@
+using UnityEditor;
+using UnityEngine;
+
+namespace Alchemy.Editor
+{
+ public class HierarchyRowSeparatorDrawer : HierarchyDrawer
+ {
+ public override void OnGUI(int instanceID, Rect selectionRect)
+ {
+ var settings = AlchemySettings.GetOrCreateSettings();
+ if (!settings.ShowSeparator) return;
+ var rect = new Rect {y = selectionRect.y, width = selectionRect.width + selectionRect.x, height = 1, x = 0};
+
+ EditorGUI.DrawRect(rect, settings.SeparatorColor);
+
+ if (!settings.ShowRowShading) return;
+ selectionRect.width += selectionRect.x;
+ selectionRect.x = 0;
+ selectionRect.height -= 1;
+ selectionRect.y += 1;
+ EditorGUI.DrawRect(selectionRect, Mathf.FloorToInt((selectionRect.y - 4) / 16 % 2) == 0 ? settings.EvenRowColor : settings.OddRowColor);
+ }
+ }
+}

Alchemy/Assets/Alchemy/Editor/Hierarchy/HierarchyTreeMapDrawer.cs

@@ -0,0 +1,137 @@
+using System.Collections.Generic;
+using Alchemy.Hierarchy;
+using UnityEditor;
+using UnityEngine;
+
+namespace Alchemy.Editor
+{
+ public class HierarchyTreeMapDrawer : HierarchyDrawer
+ {
+ private static readonly Dictionary<string, Texture2D> TextureCached = new();
+
+ public static Texture2D TreeMapCurrent
+ {
+ get
+ {
+ TextureCached.TryGetValue(nameof(TreeMapCurrent), out var tex);
+
+ if (tex != null) return tex;
+ tex = AssetHelper.FindAssetWithPath<Texture2D>("tree_map_current.png", "Editor/Hierarchy/Textures");
+ TextureCached[nameof(TreeMapCurrent)] = tex;
+ return tex;
+ }
+ }
+
+ public static Texture2D TreeMapLast
+ {
+ get
+ {
+ TextureCached.TryGetValue(nameof(TreeMapLast), out var tex);
+
+ if (tex != null) return tex;
+ tex = AssetHelper.FindAssetWithPath<Texture2D>("tree_map_last.png", "Editor/Hierarchy/Textures");
+ TextureCached[nameof(TreeMapLast)] = tex;
+ return tex;
+ }
+ }
+
+ public static Texture2D TreeMapLevel
+ {
+ get
+ {
+ TextureCached.TryGetValue(nameof(TreeMapLevel), out var tex);
+
+ if (tex != null) return tex;
+ tex = AssetHelper.FindAssetWithPath<Texture2D>("tree_map_level.png", "Editor/Hierarchy/Textures");
+ TextureCached[nameof(TreeMapLevel)] = tex;
+ return tex;
+ }
+ }
+
+ public static Texture2D TreeMapLine
+ {
+ get
+ {
+ TextureCached.TryGetValue(nameof(TreeMapLine), out var tex);
+
+ if (tex != null) return tex;
+ tex = AssetHelper.FindAssetWithPath<Texture2D>("tree_map_line.png", "Editor/Hierarchy/Textures");
+ TextureCached[nameof(TreeMapLine)] = tex;
+ return tex;
+ }
+ }
+
+ public override void OnGUI(int instanceID, Rect selectionRect)
+ {
+ var gameObject = EditorUtility.InstanceIDToObject(instanceID) as GameObject;
+ if (gameObject == null) return;
+
+ var settings = AlchemySettings.GetOrCreateSettings();
+ if (settings.ShowTreeMap)
+ {
+ selectionRect.width = 14;
+ selectionRect.height = 16;
+
+ int childCount = gameObject.transform.childCount;
+ int level = Mathf.RoundToInt(selectionRect.x / 14f);
+ var t = gameObject.transform;
+ Transform parent = null;
+
+ for (int i = 0, j = level - 1; j >= 0; i++, j--)
+ {
+ selectionRect.x = 14 * j;
+ if (i == 0)
+ {
+ if (childCount == 0)
+ {
+ GUI.color = settings.TreeMapColor;
+ GUI.DrawTexture(selectionRect, TreeMapLine);
+ }
+
+ t = gameObject.transform;
+ }
+ else if (i == 1)
+ {
+ GUI.color = settings.TreeMapColor;
+ if (parent == null)
+ {
+ if (t.GetSiblingIndex() == gameObject.scene.rootCount - 1)
+ {
+ GUI.DrawTexture(selectionRect, TreeMapLast);
+ }
+ else
+ {
+ GUI.DrawTexture(selectionRect, TreeMapCurrent);
+ }
+ }
+ else if (t.GetSiblingIndex() == parent.childCount - 1)
+ {
+ GUI.DrawTexture(selectionRect, TreeMapLast);
+ }
+ else
+ {
+ GUI.DrawTexture(selectionRect, TreeMapCurrent);
+ }
+
+ t = parent;
+ }
+ else
+ {
+ if (parent == null)
+ {
+ if (t.GetSiblingIndex() != gameObject.scene.rootCount - 1) GUI.DrawTexture(selectionRect, TreeMapLevel);
+ }
+ else if (t.GetSiblingIndex() != parent.childCount - 1) GUI.DrawTexture(selectionRect, TreeMapLevel);
+
+ t = parent;
+ }
+
+ if (t != null) parent = t.parent;
+ else break;
+ }
+
+ GUI.color = Color.white;
+ }
+ }
+ }
+}