SanAndreasUnity/Assets/Scripts/Editor/NavMeshGenerator.cs

using System.Collections.Generic;
using UnityEditor;
using UnityEngine;
using UnityEngine.AI;
using SanAndreasUnity.Behaviours.World;
using UnityEditor.SceneManagement;
using System.IO;
using System.Collections;

namespace SanAndreasUnity.Editor
{
    public class NavMeshGenerator : EditorWindow
    {
        private static int s_selectedAgentId = 0;
        private static NavMeshData s_navMeshData;
        private static IEnumerator s_coroutine;

        public NavMeshGenerator()
        {
            this.titleContent = new GUIContent("Nav mesh generator");

            EditorApplication.update -= EditorUpdate;
            EditorApplication.update += EditorUpdate;
        }

        void EditorUpdate()
        {
            if (s_coroutine != null)
            {
                try
                {
                    if (!s_coroutine.MoveNext())
                    {
                        s_coroutine = null;
                        Cleanup();
                    }
                }
                catch
                {
                    s_coroutine = null;
                    Cleanup();
                }
            }
        }

        void Cleanup()
        {
            EditorUtility.ClearProgressBar();

            if (s_navMeshData != null)
            {
                NavMeshBuilder.Cancel(s_navMeshData);
                Destroy(s_navMeshData);
                s_navMeshData = null;
            }
        }

        [MenuItem(EditorCore.MenuName + "/" + "Generate nav mesh")]
        static void Init()
        {
            var window = GetWindow<NavMeshGenerator>();
            window.Show();
        }

        void OnGUI()
        {
            EditorGUILayout.HelpBox(
                "Generate nav mesh from loaded world.\nOnly high LOD objects are included.",
                MessageType.Info,
                true);

            GUILayout.Space(20);

            EditorGUILayout.PrefixLabel("Agent ID:");
            s_selectedAgentId = EditorGUILayout.IntField(s_selectedAgentId);

            GUILayout.Space(10);

            if (!string.IsNullOrEmpty(NavMesh.GetSettingsNameFromID(s_selectedAgentId)))
            {
                var navMeshBuildSettings = NavMesh.GetSettingsByID(s_selectedAgentId);
                EditorUtils.DrawFieldsAndPropertiesInInspector(navMeshBuildSettings, 0);
            }

            GUILayout.Space(20);

            if (GUILayout.Button("Generate"))
                Generate();
            
        }

        void Generate()
        {
            if (s_coroutine != null)
                return;

            Cleanup();

            s_coroutine = DoGenerate();
        }

        IEnumerator DoGenerate()
        {
            if (string.IsNullOrEmpty(NavMesh.GetSettingsNameFromID(s_selectedAgentId)))
            {
                EditorUtility.DisplayDialog("", "Invalid agent id", "Ok");
                yield break;
            }

            var cell = Cell.Instance;
            if (null == cell)
            {
                EditorUtility.DisplayDialog("", $"{nameof(Cell)} script not found in scene. Make sure you loaded the correct scene.", "Ok");
                yield break;
            }

            UnityEngine.SceneManagement.Scene activeScene = EditorSceneManager.GetActiveScene();
            if (!activeScene.IsValid())
            {
                EditorUtility.DisplayDialog("", "Active scene is not valid", "Ok");
                yield break;
            }

            NavMeshBuildSettings navMeshBuildSettings = NavMesh.GetSettingsByID(s_selectedAgentId);

            var navMeshData = s_navMeshData = new NavMeshData(s_selectedAgentId);

            EditorUtility.DisplayProgressBar("Generating nav mesh", "Collecting objects...", 0f);

            var navMeshBuildSources = new List<NavMeshBuildSource>(1024 * 20);
            int numObjects = 0;

            foreach (StaticGeometry staticGeometry in cell.StaticGeometries.Values)
            {
                if (null != staticGeometry.LodParent)
                    continue;

                numObjects++;

                navMeshBuildSources.AddRange(Cell.GetNavMeshBuildSources(staticGeometry.transform));
            }

            navMeshBuildSources.AddRange(cell.GetWaterNavMeshBuildSources());

            EditorUtility.ClearProgressBar();

            if (!EditorUtility.DisplayDialog("", $"Found total of {numObjects} objects with {navMeshBuildSources.Count} sources. Proceed ?", "Ok", "Cancel"))
                yield break;

            var asyncOperation = NavMeshBuilder.UpdateNavMeshDataAsync(
                navMeshData,
                navMeshBuildSettings,
                navMeshBuildSources,
                new Bounds(cell.transform.position, Vector3.one * cell.WorldSize));

            while (!(asyncOperation.isDone || asyncOperation.progress == 1f))
            {
                yield return null;

                if (EditorUtility.DisplayCancelableProgressBar("", "Updating nav mesh...", asyncOperation.progress))
                    yield break;
            }

            EditorUtility.ClearProgressBar();

            if (asyncOperation.progress != 1f)
            {
                EditorUtility.DisplayDialog("", $"Updating nav mesh did not finish completely, progress is {asyncOperation.progress}", "Ok");
                yield break;
            }

            // save nav mesh

            EditorUtility.DisplayProgressBar("Generating nav mesh", "Saving nav mesh...", 1f);

            string parentDirectory = Path.GetDirectoryName(activeScene.path);
            string directory = parentDirectory + "/" + activeScene.name;
            if (!AssetDatabase.IsValidFolder(directory))
                AssetDatabase.CreateFolder(parentDirectory, activeScene.name);
            AssetDatabase.CreateAsset(s_navMeshData, directory + "/NavMesh.asset");

            s_navMeshData = null; // this is now an asset, we don't want it to be destroyed

            EditorUtility.ClearProgressBar();
            EditorUtility.DisplayDialog("", "Done", "Ok");
        }
    }
}