Added auto reload on UnityEditor's play state change

hadashiA · Nov 10, 2023 · 78d6e78 · 78d6e78
1 parent d1f0f35
commit 78d6e78
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/VContainer/Assets/VContainer/Editor/Diagnostics/VContainerDiagnosticsWindow.cs b/VContainer/Assets/VContainer/Editor/Diagnostics/VContainerDiagnosticsWindow.cs
@@ -66,6 +66,12 @@ public void Reload(IObjectResolver resolver)
             Repaint();
         }
 
+        void OnPlayModeStateChange(PlayModeStateChange state)
+        {
+            treeView.ReloadAndSort();
+            Repaint();
+        }
+
         void OnEnable()
         {
             window = this; // set singleton.
@@ -76,11 +82,13 @@ void OnEnable()
             searchField = new SearchField();
 
             DiagnositcsContext.OnContainerBuilt += Reload;
+            EditorApplication.playModeStateChanged += OnPlayModeStateChange;
         }
 
         void OnDisable()
         {
             DiagnositcsContext.OnContainerBuilt -= Reload;
+            EditorApplication.playModeStateChanged -= OnPlayModeStateChange;
         }
 
         void OnGUI()
@@ -205,4 +213,4 @@ void RenderStackTracePanel()
             }
         }
    }
-}
+}