Updated ScenePartitionViewEditor

- Grid is now drawn based on the cell size of the current active open scene
- Added function to get ScenePartitionSO of current active scene

Editor/EditorGUIUtils.cs

@@ -0,0 +1,20 @@
+using UnityEditor;
+using UnityEngine;
+
+namespace VertexColor.ScenePartition.Editor
+{
+    public static class EditorGUIUtils
+    {
+        public static readonly Color horizontalLineColor = Color.white;
+
+        public static void HorizontalLine(Color color)
+        {
+            Color prev = GUI.color;
+            GUI.color = color;
+            EditorGUILayout.LabelField("", GUI.skin.horizontalSlider);
+            GUI.color = prev;
+        }
+
+        public static void HorizontalLine() => HorizontalLine(horizontalLineColor);
+    }
+}

Editor/EditorGUIUtils.cs.meta

@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: befa2e31a4a6ae04696dcd5b72554ea4
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 

Editor/ScenePartitionEditorWindow.cs

@@ -19,11 +19,23 @@ public static void ShowExample()
 
         private void OnGUI()
         {
-            if (GUILayout.Button("Save"))
+            EditorGUILayout.LabelField("Cache", EditorStyles.boldLabel);
+
+            using (new EditorGUILayout.HorizontalScope())
+            {
+                if (GUILayout.Button("Save", EditorStyles.miniButtonLeft))
                 {
                     ScenePartitionSS.Save();
                 }
 
+                if (GUILayout.Button("Clear", EditorStyles.miniButtonRight))
+                {
+                    ScenePartitionSS.Save();
+                }
+            }
+
+            EditorGUIUtils.HorizontalLine();
+
             DrawSceneDataCache();
         }
 

Editor/ScenePartitionSOEditor.cs

@@ -83,6 +83,8 @@ public override void OnInspectorGUI()
                 {
                     EditorGUILayout.LabelField($"generatedSceneGrid");
 
+                    scenePartitionSO.Data.SceneGrid.cellSize = EditorGUILayout.IntSlider("cellSize", scenePartitionSO.Data.SceneGrid.cellSize, 10, 1000);
+
                     foreach (KeyValuePair<int, GridList> item in scenePartitionSO.Data.SceneGrid.Grid)
                     {
                         EditorGUILayout.LongField("gridId", item.Key);
@@ -97,12 +99,6 @@ public override void OnInspectorGUI()
                         EditorGUI.indentLevel--;
                     }
                 }
-
-                if (GUILayout.Button("ClearCache"))
-                {
-                    scenePartitionSO.ClearCache();
-                    ScenePartitionSS.instance.SceneDataCache.Clear();
-                }
             }
         }
     }

Editor/ScenePartitionSceneViewEditor.cs

@@ -6,7 +6,7 @@ namespace VertexColor.ScenePartition.Editor
     [InitializeOnLoad]
     public class ScenePartitionSceneViewEditor : UnityEditor.Editor
     {
-        private const int cellSize = 10;
+        private static ScenePartitionSO scenePartitionSO = null;
 
         static ScenePartitionSceneViewEditor()
         {
@@ -16,10 +16,32 @@ static ScenePartitionSceneViewEditor()
         private static void HandleDuringSceneGui(SceneView sceneView)
         {
             if (Event.current.modifiers != EventModifiers.Control) return;
+            if (!ScenePartitionUtils.TryGetScenePartitionSOForActiveScene(out scenePartitionSO)) return;
 
+            int cellSize = scenePartitionSO.Data.SceneGrid.cellSize;
             Vector3 gridPosition = CalculateGridPosition(Event.current.mousePosition);
-            int gridId = SceneGrid.CalculateGridPosition(gridPosition, 10);
+            int gridId = SceneGrid.CalculateGridPosition(gridPosition, cellSize);
 
+            {   // Draw Cells.
+                foreach (int id in scenePartitionSO.Data.SceneGrid.Grid.Keys)
+                {
+                    (int x, int y) = SceneGrid.IntToIntPair(id);
+
+                    Vector3 cellOrign = new Vector3(x * cellSize, 0, y * cellSize);
+                    Vector3[] lines = new Vector3[] {
+                        new Vector3(0, 0, 0) + cellOrign, new Vector3(0, 0, cellSize) + cellOrign,                  // Left Bottom -> Left Top
+                        new Vector3(cellSize, 0, 0) + cellOrign, new Vector3(cellSize, 0, cellSize) + cellOrign,    // Right Bottom -> Right Top
+                        new Vector3(0, 0, 0) + cellOrign, new Vector3(cellSize, 0, 0) + cellOrign,                  // Left Bottom -> Right Bottom
+                        new Vector3(0, 0, cellSize) + cellOrign, new Vector3(cellSize, 0, cellSize) + cellOrign,    // Left Top -> Right Top
+                    };
+
+                    Handles.DrawLines(lines);
+                }
+            }
+
+            if (!scenePartitionSO.Data.SceneGrid.Grid.ContainsKey(gridId)) return;
+
+            {   // Draw Selection.
                 int x = Mathf.FloorToInt(gridPosition.x / cellSize);
                 int z = Mathf.FloorToInt(gridPosition.z / cellSize);
 
@@ -31,9 +53,13 @@ private static void HandleDuringSceneGui(SceneView sceneView)
                     new Vector3(0, 0, cellSize) + cellOrign, new Vector3(cellSize, 0, cellSize) + cellOrign,    // Left Top -> Right Top
                 };
 
-            Handles.DrawDottedLines(lines, 10.0f);
+                Color c = Handles.color;
-            Handles.Label(cellOrign + new Vector3(cellSize * 0.5f, 0, cellSize * 0.5f), new GUIContent($"{gridId}"));
+                Handles.color = Color.yellow;
+                Handles.DrawLines(lines);
+                Handles.color = c;
+            }
 
+            // Context menu to load.
             if (Event.current.button == 1 && Event.current.type == EventType.MouseDown)
             {
                 GenericMenu menu = new GenericMenu();
@@ -62,10 +88,11 @@ private static Vector3 CalculateGridPosition(Vector2 mousePosition)
 
         private static void Load(object gridId)
         {
+            if (scenePartitionSO == null) return;
             if (gridId == null) return;
-            if (gridId is not int gridPos) return;
+            if (gridId is not int cellId) return;
-
 
+            scenePartitionSO.LoadCell(cellId);
         }
     }
 }

Editor/ScenePartitionUtils.cs

@@ -57,5 +57,35 @@ private static void FindDeeplyLinkedObjectsRecursive(SortedList<ulong, ScenePart
                 }
             }
         }
+
+        public static bool TryGetScenePartitionSOForActiveScene(out ScenePartitionSO scenePartitionSO)
+        {
+            scenePartitionSO = null;
+
+            UnityEngine.SceneManagement.Scene activeScene = UnityEngine.SceneManagement.SceneManager.GetActiveScene();
+
+            if (activeScene == null) return false;
+
+            string filter = $"t:{nameof(ScenePartitionSO)}";
+            string[] assets = AssetDatabase.FindAssets(filter);
+
+            if (assets == null || assets.Length <= 0) return false;
+
+            for (int i = 0; i < assets.Length; i++)
+            {
+                string path = AssetDatabase.GUIDToAssetPath(assets[i]);
+                ScenePartitionSO scenePartition = AssetDatabase.LoadAssetAtPath<ScenePartitionSO>(path);
+
+                if (scenePartition == null) continue;
+
+                if (scenePartition.SceneName == activeScene.name)
+                {
+                    scenePartitionSO = scenePartition;
+                    return true;
+                }
+            }
+
+            return false;
+        }
     }
 }

Runtime/SceneGrid.cs

@@ -6,7 +6,7 @@ namespace VertexColor.ScenePartition
     public class SceneGrid
     {
         [SerializeField]
-        private int cellSize = 10;
+        public int cellSize = 10;
 
         [SerializeField]
         private SceneGridDictionary grid = new SceneGridDictionary();