module-inspector/Editor/Drawers/DrawerSceneDropdown.cs

74 lines
2.4 KiB
C#

using System;
using System.Collections.Generic;
using Module.Inspector.Editor.Utilities;
using UnityEditor;
using UnityEngine;
namespace Module.Inspector.Editor
{
[CustomPropertyDrawer(typeof(SceneDropdown))]
internal sealed class DrawerSceneDropdown : DrawerPropertyDrawer
{
private static GUIContent[] NAMES;
private static string[] PATHS;
public override bool Draw(Rect position, DrawerPropertyAttribute attribute, SerializedProperty property, GUIContent label, EditorPropertyUtility.Result result)
{
if (property.propertyType != SerializedPropertyType.String)
return false;
FetchScenes();
EditorGUI.BeginChangeCheck();
EditorGUI.BeginProperty(position, label, property);
{
int index = Array.IndexOf(PATHS, property.stringValue);
int newIndex = EditorGUI.Popup(position, label, index, NAMES);
if (newIndex != -1 && index != newIndex)
property.stringValue = PATHS[newIndex];
}
EditorGUI.EndProperty();
bool changed = EditorGUI.EndChangeCheck();
if (changed)
property.serializedObject.ApplyModifiedProperties();
return true;
}
public override string GetErrorMessage(SerializedProperty property)
{
return "Only supports strings";
}
private static string ToSceneName(string path)
{
return !string.IsNullOrEmpty(path) ? path.Substring(7, path.Length - 13).Replace('/', '\\') : string.Empty;
}
private static void FetchScenes()
{
if (NAMES != null)
return;
EditorBuildSettingsScene[] scenes = EditorBuildSettings.scenes;
var listNames = new List<GUIContent>(scenes.Length);
var listPaths = new List<string>(scenes.Length);
for (var i = 0; i < scenes.Length; i++)
{
string path = scenes[i].path;
if (string.IsNullOrEmpty(path))
continue;
listNames.Add(new GUIContent(ToSceneName(path)));
listPaths.Add(path);
}
NAMES = listNames.ToArray();
PATHS = listPaths.ToArray();
}
}
}