SceneLauncherEditor.cs

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License. See LICENSE in the project root for license information.

using UnityEngine;
using UnityEditor;
using UnityEngine.SceneManagement;

namespace HoloToolkit.Unity
{
    [CustomEditor(typeof(SceneLauncher))]
    public class SceneLauncherEditor : Editor
    {
        private SerializedProperty sceneMappingProperty;
        private SerializedProperty buttonSpawnLocationProperty;
        private SerializedProperty buttonPrefabProperty;
        private SerializedProperty buttonRowMaxProperty;

        private void OnEnable()
        {
            sceneMappingProperty = serializedObject.FindProperty("sceneMapping");
            buttonSpawnLocationProperty = serializedObject.FindProperty("ButtonSpawnLocation");
            buttonPrefabProperty = serializedObject.FindProperty("SceneButtonPrefab");
            buttonRowMaxProperty = serializedObject.FindProperty("MaxRows");

            CheckBuildScenes(sceneMappingProperty);
        }

        public override void OnInspectorGUI()
        {
            serializedObject.Update();
            EditorGUILayout.PropertyField(buttonSpawnLocationProperty);
            EditorGUILayout.PropertyField(buttonPrefabProperty);
            EditorGUILayout.IntSlider(buttonRowMaxProperty, 1, 10);
            EditorGUILayout.HelpBox("To add scenes to the Scene Mapper by adding scenes in the build window.", MessageType.Info);
            CheckBuildScenes(sceneMappingProperty);
            ShowSceneList(sceneMappingProperty);
            serializedObject.ApplyModifiedProperties();
        }

        private void ShowSceneList(SerializedProperty sceneList)
        {
            // property name with expansion widget
            EditorGUILayout.PropertyField(sceneList);

            if (EditorBuildSettings.scenes.Length == 0)
            {
                sceneList.ClearArray();

                int index = sceneList.arraySize;

                sceneList.InsertArrayElementAtIndex(index);
                sceneList.GetArrayElementAtIndex(index).FindPropertyRelative("ScenePath").stringValue = SceneManager.GetActiveScene().path;
                sceneList.GetArrayElementAtIndex(index).FindPropertyRelative("IsButtonEnabled").boolValue = true;

                EditorBuildSettings.scenes = new[] { new EditorBuildSettingsScene(SceneManager.GetActiveScene().path, true) };
            }

            if (sceneList.isExpanded)
            {
                EditorGUI.indentLevel++;

                // Scene rows
                for (int i = 0; i < sceneList.arraySize; i++)
                {
                    EditorGUILayout.BeginHorizontal();
                    // Disable the toggle if the scene is not enabled in the build settings.
                    GUI.enabled = EditorBuildSettings.scenes[i].enabled;

                    EditorGUILayout.PropertyField(sceneList.GetArrayElementAtIndex(i).FindPropertyRelative("IsButtonEnabled"));

                    GUI.enabled = false;

                    string path = sceneList.GetArrayElementAtIndex(i).FindPropertyRelative("ScenePath").stringValue;
                    var sceneAsset = AssetDatabase.LoadAssetAtPath(path, typeof(SceneAsset)) as SceneAsset;
                    EditorGUILayout.ObjectField(sceneAsset, typeof(SceneAsset), false);

                    GUI.enabled = true;
                    EditorGUILayout.EndHorizontal();
                }

                EditorGUI.indentLevel--;
            }
        }

        private void CheckBuildScenes(SerializedProperty list)
        {
            if (EditorBuildSettings.scenes.Length == 0)
            {
                return;
            }

            bool reBuildList = list.arraySize != EditorBuildSettings.scenes.Length;

            if (!reBuildList)
            {
                // Check if the build settings list is the same as ours.
                for (int i = 0; i < list.arraySize; i++)
                {
                    if (list.GetArrayElementAtIndex(i).FindPropertyRelative("ScenePath").stringValue != EditorBuildSettings.scenes[i].path)
                    {
                        reBuildList = true;
                    }
                }
            }

            if (reBuildList)
            {
                // if it's not then we need to store a copy of our mappings.
                var oldBuildSceneMapping = new EditorBuildSettingsScene[list.arraySize];
                for (int i = 0; i < list.arraySize; i++)
                {
                    oldBuildSceneMapping[i] = new EditorBuildSettingsScene
                    {
                        path = list.GetArrayElementAtIndex(i).FindPropertyRelative("ScenePath").stringValue,
                        enabled = list.GetArrayElementAtIndex(i).FindPropertyRelative("IsButtonEnabled").boolValue
                    };
                }

                // Then re assign the mapping to the right scene in the build settings window.
                list.ClearArray();

                for (int i = 0; i < EditorBuildSettings.scenes.Length; i++)
                {
                    list.InsertArrayElementAtIndex(i);
                    list.GetArrayElementAtIndex(i).FindPropertyRelative("ScenePath").stringValue = EditorBuildSettings.scenes[i].path;
                    list.GetArrayElementAtIndex(i).FindPropertyRelative("IsButtonEnabled").boolValue = false;

                    for (var j = 0; j < oldBuildSceneMapping.Length; j++)
                    {
                        if (oldBuildSceneMapping[j].path == EditorBuildSettings.scenes[i].path)
                        {
                            list.GetArrayElementAtIndex(i).FindPropertyRelative("IsButtonEnabled").boolValue = oldBuildSceneMapping[j].enabled;
                        }
                    }
                }
            }
        }
    }
}