using System.Collections.Generic; using System.IO; using JetBrains.Annotations; using UnityEditor; using UnityEditor.SceneManagement; using UnityEngine; using UnityEngine.SceneManagement; namespace Module.NavigationTool.Editor.Toolbar { [UsedImplicitly] internal sealed class ToolScenePicker : AbstractToolbarDrawer { public override bool Visible => ToolbarSettings.IsSceneEnabled; public override bool Enabled => !EditorApplication.isPlaying && !EditorApplication.isPlayingOrWillChangePlaymode; public override EToolbarPlacement Placement => EToolbarPlacement.Right; public override int Priority => (int)EToolbarPriority.Medium; private static bool IS_DIRTY = true; private static int SELECTED_INDEX = -1; private static string[] OPTIONS = new string[0]; private static string[] PATHS = new string[0]; private static void Initialize() { if (!IS_DIRTY) return; var listNames = new List(); var listPaths = new List(); InitializeBuildSettingsScenes(listNames, listPaths); InitializeRemainingScenes(listNames, listPaths); OPTIONS = listNames.ToArray(); PATHS = listPaths.ToArray(); Scene activeScene = SceneManager.GetActiveScene(); SELECTED_INDEX = listPaths.IndexOf(activeScene.path); IS_DIRTY = false; } private static void InitializeBuildSettingsScenes(List listNames, List listPaths) { EditorBuildSettingsScene[] scenes = EditorBuildSettings.scenes; for (var i = 0; i < scenes.Length; i++) { if (string.IsNullOrEmpty(scenes[i].path)) continue; listNames.Add($"{Path.GetFileNameWithoutExtension(scenes[i].path)}"); listPaths.Add(scenes[i].path); } } private static void InitializeRemainingScenes(List listNames, List listPaths) { string[] guids = AssetDatabase.FindAssets("t:scene"); int startIndex = listNames.Count; for (var i = 0; i < guids.Length; i++) { string path = AssetDatabase.GUIDToAssetPath(guids[i]); if (listPaths.Contains(path)) continue; var scene = AssetDatabase.LoadAssetAtPath(path); if (scene == null) continue; listNames.Add(path.Substring(7, path.Length - 13).Replace('/', '\\')); listPaths.Add(path); } if (startIndex >= listNames.Count) return; listNames.Insert(startIndex, string.Empty); listPaths.Insert(startIndex, string.Empty); } public override void Update() { Initialize(); } protected override void Draw(Rect rect) { Initialize(); int temp = EditorGUI.Popup(rect, SELECTED_INDEX, OPTIONS, styles.popup); if (temp > -1 && temp != SELECTED_INDEX && OPTIONS.Length != 0) { string path = PATHS[temp]; if (string.IsNullOrEmpty(path)) return; SELECTED_INDEX = temp; EditorSceneManager.OpenScene(PATHS[SELECTED_INDEX], OpenSceneMode.Single); } } public override float CalculateWidth() { return 100.0f; } public static void SetAsDirty() { IS_DIRTY = true; } } }