Repaint editor GUIs on shortcut

rogerbarton · Mar 21, 2020 · 1d497da · 1d497da
1 parent 0b77423
commit 1d497da
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/scripts/Editor/DllManipulatorEditor.cs b/scripts/Editor/DllManipulatorEditor.cs
@@ -328,6 +328,8 @@ static string GetFirstAssemblyToList(string[] allAssemblies)
         public static void LoadAllShortcut()
         {
             DllManipulator.LoadAll();
+            RepaintAll();
+            DllManipulatorWindowEditor.RepaintAll();
         }
 
         #if UNITY_2019_1_OR_NEWER
@@ -338,6 +340,16 @@ public static void LoadAllShortcut()
         public static void UnloadAll()
         {
             DllManipulator.UnloadAll();
+            RepaintAll();
+            DllManipulatorWindowEditor.RepaintAll();
+        }
+
+        public static void RepaintAll()
+        {
+            var editors = Resources.FindObjectsOfTypeAll<DllManipulatorEditor>();
+            if(editors == null) return;
+            foreach (var editor in editors)
+                editor.Repaint();
         }
     }
 }
diff --git a/scripts/Editor/DllManipulatorWindowEditor.cs b/scripts/Editor/DllManipulatorWindowEditor.cs
@@ -6,13 +6,20 @@ namespace UnityNativeTool.Internal
 {
     public class DllManipulatorWindowEditor : EditorWindow
     {
+        private static EditorWindow window;
+
         [MenuItem("Window/Dll manipulator")]
         static void Init()
         {
-            var window = GetWindow<DllManipulatorWindowEditor>();
+            window = GetWindow<DllManipulatorWindowEditor>();
             window.Show();
         }
 
+        public static void RepaintAll()
+        {
+            if(window) window.Repaint();
+        }
+
         void OnGUI()
         {
             var dllManipulator = FindObjectOfType<DllManipulatorScript>();