代码

基于https://github.com/smkplus/CustomToolbar项目中思路，实现简易的扩展工具栏功能

  1. using System.Reflection;
  2. using UnityEditor;
  3. using UnityEngine;
  4. using System;
  5. #if UNITY_2019_1_OR_NEWER
  6. using UnityEngine.UIElements;
  7. #else
  8. using UnityEngine.Experimental.UIElements;
  9. #endif
  11. /// <summary>
  12. /// 扩展Unity的按钮栏
  13. /// </summary>
  14. [InitializeOnLoad]
  15. public static class CustomUnityToolbar
  16. {
  17.     public static Type m_toolbarType = typeof(Editor).Assembly.GetType("UnityEditor.Toolbar");
  18.     public static ScriptableObject m_currentToolbar;
  20.     static CustomUnityToolbar()
  21.     {
  22.         EditorApplication.delayCall += OnUpdate;
  23.     }
  25.     public static void OnUpdate()
  26.     {
  27.         // Relying on the fact that toolbar is ScriptableObject and gets deleted when layout changes
  28.         if(m_currentToolbar == null)
  29.         {
  30.             // Find toolbar
  31.             var toolbars = Resources.FindObjectsOfTypeAll(m_toolbarType);
  32.             m_currentToolbar = toolbars.Length > 0 ? (ScriptableObject)toolbars[0] : null;
  34.             if(m_currentToolbar != null)
  35.             {
  36.                 var root = m_currentToolbar.GetType().GetField("m_Root", BindingFlags.NonPublic | BindingFlags.Instance);
  37.                 var rawRoot = root.GetValue(m_currentToolbar);
  38.                 var mRoot = rawRoot as VisualElement;
  39.                 RegisterCallback("ToolbarZoneLeftAlign", GUILeft);
  40.                 RegisterCallback("ToolbarZoneRightAlign", GUIRight);
  42.                 void RegisterCallback(string root, Action cb)
  43.                 {
  44.                     var toolbarZone = mRoot.Q(root);
  46.                     var parent = new VisualElement()
  47.                     {
  48.                         style = {
  49.                             flexGrow = 1,
  50.                             flexDirection = FlexDirection.Row,
  51.                         }
  52.                     };
  53.                     var container = new IMGUIContainer();
  54.                     container.onGUIHandler += () =>
  55.                     {
  56.                         cb?.Invoke();
  57.                     };
  58.                     parent.Add(container);
  59.                     toolbarZone.Add(parent);
  60.                 }
  61.             }
  62.         }
  63.     }
  66.     /// <summary>
  67.     /// 绘制左侧的元素
  68.     /// </summary>
  69.     private static void GUILeft()
  70.     {
  71.         GUILayout.BeginHorizontal();
  72.         if(GUILayout.Button("可扩展"))
  73.         { }
  74.         GUILayout.EndHorizontal();
  75.     }
  77.     /// <summary>
  78.     /// 绘制右侧的元素
  79.     /// </summary>
  80.     private static void GUIRight()
  81.     {
  82.         GUILayout.BeginHorizontal();
  83.         if(GUILayout.Button("可扩展"))
  84.         { }
  85.         GUILayout.EndHorizontal();
  86.     }
  87. }

效果

扩展Unity的工具(按钮)栏 - 图1

参考