Simple grid generation and loading

- Grid generation
- Grid loading
- GridIds
- Still having issues with serialization of the grid data structure

Editor/ScenePartitionEditorWindow.cs

@@ -17,6 +17,11 @@ namespace VertexColor.ScenePartition.Editor
 
         private void OnGUI()
         {
+            if (GUILayout.Button("Save"))
+            {
+                ScenePartitionSS.Save();
+            }
+
             DrawSceneDataCache();
         }
 
@@ -55,6 +60,25 @@ namespace VertexColor.ScenePartition.Editor
                                 EditorGUILayout.IntField((int)scenePartition.Value.id);
                             }
                         }
+
+                        if (sceneData.Value.SceneGrid != null)
+                        {
+                            EditorGUILayout.LabelField($"generatedSceneGrid");
+
+                            foreach (KeyValuePair<int, List<uint>> item in sceneData.Value.SceneGrid.Grid)
+                            {
+                                EditorGUILayout.IntField("gridId", item.Key);
+
+                                EditorGUI.indentLevel++;
+
+                                foreach (uint id in item.Value)
+                                {
+                                    EditorGUILayout.IntField((int)id);
+                                }
+
+                                EditorGUI.indentLevel--;
+                            }
+                        }
                     }
                 }
             }

Editor/ScenePartitionSO.cs

@@ -1,5 +1,6 @@
 using System.Collections.Generic;
 using System.IO;
+using System.Reflection;
 using System.Text.RegularExpressions;
 using UnityEditor;
 using UnityEditor.SceneManagement;
@@ -58,7 +59,7 @@ namespace VertexColor.ScenePartition.Editor
         }
 
         /// <summary>
-        /// Load already/previously loaded partitions.
+        /// Discard changes and reload loaded partitions.
         /// </summary>
         public void Reload()
         {
@@ -204,5 +205,49 @@ namespace VertexColor.ScenePartition.Editor
 
             LoadScenePartitions(partitionIds);
         }
+
+        public void GenerateSceneGridData()
+        {
+            if (!Data.HasCreatedPartitions) return;
+
+            LoadAll();
+
+            Scene scene = EditorSceneManager.OpenScene(ScenePartitionUtils.GetScenePath(this), OpenSceneMode.Single);
+
+            GameObject[] rootGameObjects = scene.GetRootGameObjects();
+
+            Data.SceneGrid.Grid.Clear();
+
+            // Maybe later switch to getting the data from disk instead of loading the scene and then unloading it again.
+            foreach (GameObject gameObject in rootGameObjects)
+            {
+                // https://forum.unity.com/threads/how-to-get-the-local-identifier-in-file-for-scene-objects.265686/
+                PropertyInfo inspectorModeInfo = typeof(SerializedObject).GetProperty("inspectorMode", BindingFlags.NonPublic | BindingFlags.Instance);
+                SerializedObject serializedObject = new SerializedObject(gameObject.transform);
+                inspectorModeInfo.SetValue(serializedObject, InspectorMode.Debug, null);
+                SerializedProperty localIdProp = serializedObject.FindProperty("m_LocalIdentfierInFile");
+
+                uint localId = (uint)localIdProp.intValue;
+
+                if (Data.ScenePartitions.ContainsKey(localId))
+                {
+                    Data.SceneGrid.Insert(localId, gameObject.transform.position);
+                }
+                else
+                {
+                    Debug.LogWarning($"Could not find LocalIdentfierInFile for {gameObject.transform} {gameObject.name} {gameObject.transform.GetInstanceID()}");
+                }
+            }
+
+            Unload();
+        }
+
+        public void LoadCell(int gridId)
+        {
+            if (Data.SceneGrid.Grid.TryGetValue(gridId, out List<uint> ids))
+            {
+                LoadPartitions(ids.ToArray());
+            }
+        }
     }
 }

Editor/ScenePartitionSOEditor.cs

@@ -1,3 +1,4 @@
+using System.Collections.Generic;
 using UnityEditor;
 using UnityEngine;
 
@@ -7,6 +8,7 @@ namespace VertexColor.ScenePartition.Editor
     public class ScenePartitionSOEditor : UnityEditor.Editor
     {
         private int id = 0;
+        private int gridId = 0;
 
         public override void OnInspectorGUI()
         {
@@ -64,6 +66,37 @@ namespace VertexColor.ScenePartition.Editor
                 {
                     scenePartitionSO.LoadPartitions(new uint[1] { (uint)id });
                 }
+
+                if (GUILayout.Button("GenerateSceneGrid"))
+                {
+                    scenePartitionSO.GenerateSceneGridData();
+                }
+
+                gridId = EditorGUILayout.IntField("gridId", gridId);
+
+                if (GUILayout.Button("LoadSceneGrid"))
+                {
+                    scenePartitionSO.LoadCell(gridId);
+                }
+
+                if (scenePartitionSO.Data.SceneGrid != null)
+                {
+                    EditorGUILayout.LabelField($"generatedSceneGrid");
+
+                    foreach (KeyValuePair<int, List<uint>> item in scenePartitionSO.Data.SceneGrid.Grid)
+                    {
+                        EditorGUILayout.IntField("gridId", item.Key);
+
+                        EditorGUI.indentLevel++;
+
+                        foreach (uint id in item.Value)
+                        {
+                            EditorGUILayout.IntField((int)id);
+                        }
+
+                        EditorGUI.indentLevel--;
+                    }
+                }
             }
         }
     }

Editor/ScenePartitionSS.cs

@@ -29,7 +29,12 @@ namespace VertexColor.ScenePartition.Editor
 
         private void OnDisable()
         {
-            Save(true);
+            Save();
+        }
+
+        public static void Save()
+        {
+            instance.Save(true);
         }
     }