超级NB Gemini大师手作快速选择器（shift+右键唤起）
Signed-off-by: TRAfoer <lhf190@outlook.com>
This commit is contained in:
231
Assets/Editor/QuickSelector.CS
Normal file
231
Assets/Editor/QuickSelector.CS
Normal file
@@ -0,0 +1,231 @@
|
||||
using UnityEngine;
|
||||
using UnityEditor;
|
||||
using UnityEngine.UI;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
|
||||
[InitializeOnLoad]
|
||||
public class QuickSelectorHud
|
||||
{
|
||||
public static GameObject HoveredObject;
|
||||
|
||||
static QuickSelectorHud()
|
||||
{
|
||||
SceneView.duringSceneGui += OnSceneGUI;
|
||||
}
|
||||
|
||||
private static void OnSceneGUI(SceneView sceneView)
|
||||
{
|
||||
if (HoveredObject != null)
|
||||
{
|
||||
DrawHighlight(HoveredObject);
|
||||
sceneView.Repaint();
|
||||
}
|
||||
|
||||
Event e = Event.current;
|
||||
if (e.type == EventType.MouseDown && e.button == 1 && e.shift)
|
||||
{
|
||||
var entries = CollectData(sceneView, e.mousePosition);
|
||||
QuickSelectorPopup popup = new QuickSelectorPopup(entries);
|
||||
PopupWindow.Show(new Rect(e.mousePosition.x, e.mousePosition.y, 0, 0), popup);
|
||||
e.Use();
|
||||
}
|
||||
}
|
||||
|
||||
private static void DrawHighlight(GameObject go)
|
||||
{
|
||||
Handles.color = new Color(0f, 0.7f, 1f, 1f); // 亮蓝色
|
||||
RectTransform rt = go.GetComponent<RectTransform>();
|
||||
if (rt != null)
|
||||
{
|
||||
Vector3[] corners = new Vector3[4];
|
||||
rt.GetWorldCorners(corners);
|
||||
Handles.DrawPolyLine(corners[0], corners[1], corners[2], corners[3], corners[0]);
|
||||
Handles.color = new Color(0f, 0.7f, 1f, 0.1f);
|
||||
Handles.DrawAAConvexPolygon(corners);
|
||||
}
|
||||
else
|
||||
{
|
||||
Renderer r = go.GetComponent<Renderer>();
|
||||
if (r != null) Handles.DrawWireCube(r.bounds.center, r.bounds.size);
|
||||
}
|
||||
}
|
||||
|
||||
private static List<SelectionEntry> CollectData(SceneView sceneView, Vector2 mousePos)
|
||||
{
|
||||
bool enableUI = EditorPrefs.GetBool("QS_UI", true);
|
||||
bool enable3D = EditorPrefs.GetBool("QS_3D", true);
|
||||
Vector2 guiPos = mousePos;
|
||||
guiPos.y = sceneView.camera.pixelHeight - guiPos.y;
|
||||
Ray ray = HandleUtility.GUIPointToWorldRay(mousePos);
|
||||
|
||||
HashSet<GameObject> processed = new HashSet<GameObject>();
|
||||
List<SelectionEntry> list = new List<SelectionEntry>();
|
||||
|
||||
GameObject smart = HandleUtility.PickGameObject(mousePos, false);
|
||||
if (smart != null) AddEntry(list, processed, smart, "PICK");
|
||||
|
||||
if (enableUI)
|
||||
{
|
||||
var rects = GameObject.FindObjectsByType<RectTransform>(FindObjectsSortMode.None);
|
||||
foreach (var rect in rects)
|
||||
{
|
||||
if (rect.gameObject.activeInHierarchy && RectTransformUtility.RectangleContainsScreenPoint(rect, guiPos, sceneView.camera))
|
||||
AddEntry(list, processed, rect.gameObject, "UI");
|
||||
}
|
||||
}
|
||||
|
||||
if (enable3D)
|
||||
{
|
||||
RaycastHit[] hits = Physics.RaycastAll(ray, float.MaxValue);
|
||||
foreach (var hit in hits) AddEntry(list, processed, hit.collider.gameObject, "3D");
|
||||
foreach (var r in GameObject.FindObjectsByType<Renderer>(FindObjectsSortMode.None))
|
||||
if (!processed.Contains(r.gameObject) && r.bounds.IntersectRay(ray, out _))
|
||||
AddEntry(list, processed, r.gameObject, "Mesh");
|
||||
}
|
||||
return list;
|
||||
}
|
||||
|
||||
private static void AddEntry(List<SelectionEntry> list, HashSet<GameObject> set, GameObject go, string src)
|
||||
{
|
||||
if (go == null || !set.Add(go)) return;
|
||||
string maj = (go.GetComponent<RectTransform>() != null) ? "UI" : (go.GetComponent<Collider>() != null ? "3D" : "Mesh");
|
||||
string min = (maj == "UI") ? GetUIType(go) : (go.GetComponent<Renderer>()?.GetType().Name ?? "Object");
|
||||
list.Add(new SelectionEntry { go = go, root = GetRoot(go.transform), depth = GetDepth(go.transform), sibling = go.transform.GetSiblingIndex(), major = maj, minor = min });
|
||||
}
|
||||
|
||||
private static string GetUIType(GameObject go)
|
||||
{
|
||||
if (go.GetComponent("TextMeshProUGUI")) return "TMPro";
|
||||
if (go.GetComponent<Text>()) return "Text";
|
||||
if (go.GetComponent<Image>()) return "Image";
|
||||
return "Rect";
|
||||
}
|
||||
private static GameObject GetRoot(Transform t) { while (t.parent != null) t = t.parent; return t.gameObject; }
|
||||
private static int GetDepth(Transform t) { int d = 0; while (t.parent != null) { d++; t = t.parent; } return d; }
|
||||
}
|
||||
|
||||
public class SelectionEntry { public GameObject go, root; public int depth, sibling; public string major, minor; }
|
||||
|
||||
public class QuickSelectorPopup : PopupWindowContent
|
||||
{
|
||||
private List<SelectionEntry> _entries;
|
||||
private List<DisplayItem> _displayItems;
|
||||
private Vector2 _scroll;
|
||||
private GUIStyle _hoverStyle;
|
||||
private GUIStyle _richLabelStyle; // 修正点：手动创建支持富文本的 Style
|
||||
|
||||
public QuickSelectorPopup(List<SelectionEntry> entries)
|
||||
{
|
||||
_entries = entries;
|
||||
RefreshList();
|
||||
}
|
||||
|
||||
public override Vector2 GetWindowSize() => new Vector2(280, Mathf.Min((_displayItems.Count * 22) + 30, 450));
|
||||
|
||||
public override void OnGUI(Rect rect)
|
||||
{
|
||||
// 样式初始化
|
||||
if (_hoverStyle == null)
|
||||
{
|
||||
_hoverStyle = new GUIStyle(EditorStyles.label);
|
||||
_hoverStyle.normal.background = MakeTex(2, 2, new Color(0.2f, 0.5f, 1f, 0.4f));
|
||||
}
|
||||
if (_richLabelStyle == null)
|
||||
{
|
||||
_richLabelStyle = new GUIStyle(EditorStyles.label);
|
||||
_richLabelStyle.richText = true; // 关键修正：开启富文本支持
|
||||
}
|
||||
|
||||
_scroll = EditorGUILayout.BeginScrollView(_scroll);
|
||||
Event e = Event.current;
|
||||
|
||||
foreach (var item in _displayItems)
|
||||
{
|
||||
if (item.isSep) { EditorGUILayout.LabelField("", GUI.skin.horizontalSlider); continue; }
|
||||
|
||||
Rect r = EditorGUILayout.GetControlRect(false, 20);
|
||||
bool isHover = r.Contains(e.mousePosition);
|
||||
|
||||
if (isHover)
|
||||
{
|
||||
GUI.Box(r, "", _hoverStyle);
|
||||
if (QuickSelectorHud.HoveredObject != item.ent.go)
|
||||
{
|
||||
QuickSelectorHud.HoveredObject = item.ent.go;
|
||||
EditorGUIUtility.PingObject(item.ent.go);
|
||||
SceneView.RepaintAll();
|
||||
}
|
||||
if (e.type == EventType.MouseDown && e.button == 0)
|
||||
{
|
||||
Selection.activeGameObject = item.ent.go;
|
||||
this.editorWindow.Close();
|
||||
}
|
||||
}
|
||||
|
||||
// 使用修正后的富文本样式绘制
|
||||
GUI.Label(r, item.label, _richLabelStyle);
|
||||
}
|
||||
EditorGUILayout.EndScrollView();
|
||||
|
||||
DrawToolbar();
|
||||
|
||||
if (e.type == EventType.MouseMove && !rect.Contains(e.mousePosition))
|
||||
{
|
||||
QuickSelectorHud.HoveredObject = null;
|
||||
}
|
||||
}
|
||||
|
||||
private void DrawToolbar()
|
||||
{
|
||||
GUILayout.BeginHorizontal(EditorStyles.toolbar);
|
||||
if (GUILayout.Toggle(EditorPrefs.GetBool("QS_UI", true), "UI", EditorStyles.toolbarButton) != EditorPrefs.GetBool("QS_UI", true))
|
||||
{
|
||||
EditorPrefs.SetBool("QS_UI", !EditorPrefs.GetBool("QS_UI")); RefreshList();
|
||||
}
|
||||
if (GUILayout.Toggle(EditorPrefs.GetBool("QS_3D", true), "3D", EditorStyles.toolbarButton) != EditorPrefs.GetBool("QS_3D", true))
|
||||
{
|
||||
EditorPrefs.SetBool("QS_3D", !EditorPrefs.GetBool("QS_3D")); RefreshList();
|
||||
}
|
||||
GUILayout.FlexibleSpace();
|
||||
if (GUILayout.Toggle(EditorPrefs.GetBool("QS_REV", false), "Reverse", EditorStyles.toolbarButton) != EditorPrefs.GetBool("QS_REV", false))
|
||||
{
|
||||
EditorPrefs.SetBool("QS_REV", !EditorPrefs.GetBool("QS_REV")); RefreshList();
|
||||
}
|
||||
GUILayout.EndHorizontal();
|
||||
}
|
||||
|
||||
public override void OnClose() { QuickSelectorHud.HoveredObject = null; SceneView.RepaintAll(); }
|
||||
|
||||
private void RefreshList()
|
||||
{
|
||||
_displayItems = new List<DisplayItem>();
|
||||
bool rev = EditorPrefs.GetBool("QS_REV", false);
|
||||
var groups = _entries.GroupBy(x => x.root).OrderByDescending(g => g.Any(e => e.major == "UI")).ThenBy(g => g.Key.name);
|
||||
bool first = true;
|
||||
foreach (var g in groups)
|
||||
{
|
||||
if (!first) _displayItems.Add(new DisplayItem { isSep = true });
|
||||
first = false;
|
||||
var sorted = rev ? g.OrderByDescending(x => x.depth).ThenByDescending(x => x.sibling) : g.OrderBy(x => x.depth).ThenBy(x => x.sibling);
|
||||
int minD = g.Min(x => x.depth);
|
||||
int maxD = g.Max(x => x.depth);
|
||||
foreach (var ent in sorted)
|
||||
{
|
||||
int ind = rev ? (maxD - ent.depth) : (ent.depth - minD);
|
||||
string color = (ent.major == "UI" ? "#00E6FF" : "#AAAAAA"); // 调整 UI 标签为更亮的青色
|
||||
string label = $"{new string(' ', ind * 4)}<color={color}>[{ent.major}|{ent.minor}]</color> {ent.go.name}";
|
||||
_displayItems.Add(new DisplayItem { ent = ent, label = label });
|
||||
}
|
||||
}
|
||||
if (editorWindow != null) editorWindow.Repaint();
|
||||
}
|
||||
|
||||
private Texture2D MakeTex(int w, int h, Color col)
|
||||
{
|
||||
Color[] pix = new Color[w * h]; for (int i = 0; i < pix.Length; i++) pix[i] = col;
|
||||
Texture2D t = new Texture2D(w, h); t.SetPixels(pix); t.Apply(); return t;
|
||||
}
|
||||
|
||||
private class DisplayItem { public bool isSep; public SelectionEntry ent; public string label; }
|
||||
}
|
||||
Reference in New Issue
Block a user