勇气小铺

Hierarchy Folder

发表于 2023-05-08
思路

新建一个Tag Folder 用于标识显示为Folder的Empty GameObject
针对Tag为Folder的GameObject进行特殊绘制处理
用到的UnityEditor方法

Full Code

using UnityEngine;
using UnityEditor;

namespace kkmaple.EditorTool
{
    [InitializeOnLoad]
    public partial class CustomHierarchyFolder
    {
        #region 静态构造
        static CustomHierarchyFolder()
        {
            EditorApplication.hierarchyWindowItemOnGUI += HierarchyWindowItemOnGUI;
        }
        #endregion
        #region 变量
        public static GameObject selectObj;
        #endregion
        #region 菜单
        [MenuItem("GameObject/Folder/Create Folder", false, -1)]
        static void CreateEmptyFolder()
        {
            GameObject obj = new GameObject();
            Undo.RegisterCreatedObjectUndo(obj, obj.name);
            if (Selection.activeTransform != null)
            {
                Object[] selectedObjects = Selection.objects;
                GameObject selectedGameObject = selectedObjects[0] as GameObject;
                obj.transform.SetParent(selectedGameObject.transform);
            }
            obj.name = "New Folder";
            EditorUtility.SetDirty(obj);
            Selection.activeGameObject = obj;
            AddTag(obj, "Folder");

        }
        [MenuItem("GameObject/Folder/Create Folder", true, -1)]
        private static bool ValidateCreateEmpty()
        {
            return Selection.objects.Length <= 1;
        }
        #endregion
        #region 工具方法
        /// <summary>
        /// 向工程中添加tag
        /// </summary>
        /// <param name="tag">tag名</param>
        public static void AddTagToProject(string tag)
        {
            Object[] asset = AssetDatabase.LoadAllAssetsAtPath("ProjectSettings/TagManager.asset");
            if ((asset != null) && (asset.Length > 0))
            {
                SerializedObject so = new SerializedObject(asset[0]);
                SerializedProperty tags = so.FindProperty("tags");
                for (int i = 0; i < tags.arraySize; ++i)
                {
                    if (tags.GetArrayElementAtIndex(i).stringValue == tag)
                    {
                        //tag已经存在啥也不做
                        return;
                    }
                }
                tags.InsertArrayElementAtIndex(0);
                tags.GetArrayElementAtIndex(0).stringValue = tag;
                so.ApplyModifiedProperties();
                so.Update();
            }
        }
        /// <summary>
        /// 给指定对象添加tag
        /// </summary>
        /// <param name="obj"></param>
        /// <param name="tag"></param>
        public static void AddTag(GameObject obj, string tag)
        {
            AddTagToProject(tag);
            obj.tag = tag;
        }
        /// <summary>
        /// 重置GameObject位置
        /// </summary>
        /// <param name="obj"></param>
        internal static void TransfromReset(GameObject obj)
        {
            obj.transform.localPosition = Vector3.zero;
            obj.transform.localRotation = Quaternion.identity;
            obj.transform.localScale = Vector3.one;
        }
        /// <summary>
        /// 勾选按钮
        /// </summary>
        /// <param name="obj"></param>
        /// <param name="selectionRect"></param>
        internal static void EnableDisableToggle(GameObject obj, Rect selectionRect)
        {
            Rect r = selectionRect;
            r.xMax = r.width;
            r.x = selectionRect.xMax;
            var tmpColor = GUI.color;
            //设置透明
            GUI.color = new Color(0, 0, 0, 0f);
            if (GUI.Button(r, ""))
            {
                obj.SetActive(!obj.activeInHierarchy);
            }
            //还原颜色
            GUI.color = tmpColor;
            r.x = selectionRect.xMax;
            GUI.Label(r, obj.activeInHierarchy ? EditorGUIUtility.IconContent("d_toggle_on_focus") : EditorGUIUtility.IconContent("d_toggle_bg"));
        }
        /// <summary>
        /// 检查工程中是否存在tag
        /// </summary>
        /// <param name="tag">tag名</param>
        public static bool IsTagExsist(string tag)
        {
            Object[] asset = AssetDatabase.LoadAllAssetsAtPath("ProjectSettings/TagManager.asset");
            if ((asset != null) && (asset.Length > 0))
            {
                SerializedObject so = new SerializedObject(asset[0]);
                SerializedProperty tags = so.FindProperty("tags");
                bool exsist = false;
                for (int i = 0; i < tags.arraySize; ++i)
                {
                    if (tags.GetArrayElementAtIndex(i).stringValue == tag)
                    {
                        exsist = true;
                        break;
                    }
                    else
                    {
                        exsist = false;
                    }
                }
                if (exsist) return true;
                else { return false; }
            }
            else
            {
                return false;
            }
        }
        /// <summary>
        /// 处理Folder显示
        /// </summary>
        /// <param name="instanceID">Object id</param>
        /// <param name="selectionRect"></param>
        internal static void HierarchyWindowItemOnGUI(int instanceID, Rect selectionRect)
        {
            GameObject obj = EditorUtility.InstanceIDToObject(instanceID) as GameObject;

            if (obj == null)
            {
                selectObj = null;
                return;
            }
            if (obj.transform == Selection.activeTransform)
            {
                selectObj = obj;
            }

            Rect r = selectionRect;

            if (!IsTagExsist("Folder")) return;

            //处理Folder显示
            if (obj.CompareTag("Folder"))
            {
                EnableDisableToggle(obj, selectionRect);

                //重置位置
                TransfromReset(obj);

                //隐藏编辑选项
                obj.transform.hideFlags = HideFlags.NotEditable | HideFlags.HideInInspector;
                obj.hideFlags = HideFlags.HideInInspector;

                //在Hierarchy中不可pick
                var sv = SceneVisibilityManager.instance;
                sv.DisablePicking(obj, false);

                GUI.color = new Color(1, 1, 1, 1);
                r.xMin = selectionRect.x + 1;
                r.xMax = r.xMin + 15f;

                bool isPro = EditorGUIUtility.isProSkin;
                if (obj.transform == Selection.activeTransform)
                {
                    if (isPro) EditorGUI.DrawRect(r, new Color(0.17f, 0.36f, 0.52f, 1f));
                    else { EditorGUI.DrawRect(r, new Color(0.23f, 0.45f, 0.69f, 1f)); }
                }
                else
                {
                    if (isPro) EditorGUI.DrawRect(r, new Color(0.22f, 0.22f, 0.22f, 1f));
                    else { EditorGUI.DrawRect(r, new Color(0.8f, 0.8f, 0.8f, 1f)); }
                }

                r = selectionRect;
                r.x = selectionRect.xMin + 25;

                if (obj.activeInHierarchy)
                {
                    ChangeFolderIconActive(obj, selectionRect);
                }
                else
                {
                    ChangeFolderIconDactive(obj, selectionRect);
                }
            }
        }
        /// <summary>
        /// Folder Icon Active
        /// </summary>
        static void ChangeFolderIconActive(GameObject obj, Rect selectionRect)
        {
            int childcout = obj.transform.childCount;
            bool isExtended = true;
            string iconeName = string.Empty;
            bool hasChild = childcout > 0;
            if (hasChild)
            {
                if (EditorGUIUtility.isProSkin)
                {
                    if (isExtended)
                    {
                        iconeName = "FolderOpened On Icon";
                    }
                    else
                    {
                        iconeName = "Folder On Icon";
                    }
                }
                else
                {
                    if (isExtended)
                    {
                        iconeName = "FolderOpened Icon";
                    }
                    else
                    {
                        iconeName = "Folder Icon";
                    }
                }
            }
            else
            {
                if (EditorGUIUtility.isProSkin)
                {
                    iconeName = "FolderEmpty On Icon";
                }
                else
                {
                    iconeName = "FolderEmpty Icon";
                }
            }
            EditorGUI.LabelField(selectionRect, EditorGUIUtility.IconContent(iconeName));
        }

        /// <summary>
        /// Folder Icon Dactive
        /// </summary>
        static void ChangeFolderIconDactive(GameObject obj, Rect selectionRect)
        {
            int childcout = obj.transform.childCount;
            bool isExtended = true;
            string iconeName = string.Empty;
            bool hasChild = childcout > 0;
            if (hasChild)
            {
                if (EditorGUIUtility.isProSkin)
                {
                    if (isExtended)
                    {
                        iconeName = "FolderOpened Icon";
                    }
                    else
                    {
                        iconeName = "Folder Icon";
                    }
                }
                else
                {
                    if (isExtended)
                    {
                        iconeName = "FolderOpened On Icon";
                    }
                    else
                    {
                        iconeName = "Folder On Icon";
                    }
                }
            }
            else
            {
                if (EditorGUIUtility.isProSkin)
                {
                    iconeName = "FolderEmpty Icon";
                }
                else
                {
                    iconeName = "FolderEmpty On Icon";
                }
            }
            EditorGUI.LabelField(selectionRect, EditorGUIUtility.IconContent(iconeName));
        }
    }
    #endregion
}