50 lines
1.8 KiB
C#
50 lines
1.8 KiB
C#
using System;
|
|
using System.Collections;
|
|
using System.Collections.Generic;
|
|
using UnityEngine;
|
|
using UnityEngine.UIElements;
|
|
using UnityEditor;
|
|
using UnityEditor.UIElements;
|
|
|
|
namespace Hierarchy2
|
|
{
|
|
[CustomEditor(typeof(HierarchyFolder))]
|
|
internal class HierarchyFolderEditor : Editor
|
|
{
|
|
private void OnEnable()
|
|
{
|
|
}
|
|
|
|
public override VisualElement CreateInspectorGUI()
|
|
{
|
|
var script = target as HierarchyFolder;
|
|
|
|
var root = new VisualElement();
|
|
|
|
IMGUIContainer imguiContainer = new IMGUIContainer(() =>
|
|
{
|
|
script.flattenMode = (HierarchyFolder.FlattenMode) EditorGUILayout.EnumPopup("Flatten Mode", script.flattenMode);
|
|
if (script.flattenMode != HierarchyFolder.FlattenMode.None)
|
|
{
|
|
script.flattenSpace = (HierarchyFolder.FlattenSpace) EditorGUILayout.EnumPopup("Flatten Space", script.flattenSpace);
|
|
script.destroyAfterFlatten = EditorGUILayout.Toggle("Destroy After Flatten", script.destroyAfterFlatten);
|
|
}
|
|
});
|
|
root.Add(imguiContainer);
|
|
|
|
return root;
|
|
}
|
|
|
|
[MenuItem("Tools/Hierarchy 2/Hierarchy Folder", priority = 0)]
|
|
static void CreateInstance(UnityEditor.MenuCommand command)
|
|
{
|
|
GameObject gameObject = new GameObject("Folder", new Type[1] {typeof(HierarchyFolder)});
|
|
|
|
Undo.RegisterCreatedObjectUndo(gameObject, "Create Hierarchy Folder");
|
|
if (command.context)
|
|
Undo.SetTransformParent(gameObject.transform, ((GameObject) command.context).transform, "Create Hierarchy Folder");
|
|
|
|
Selection.activeTransform = gameObject.transform;
|
|
}
|
|
}
|
|
} |