fix blank window bug #40

hecomi · Oct 16, 2021 · 268882a · 268882a
1 parent 5e5b334
commit 268882a
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 6 deletions.
diff --git a/Assets/uWindowCapture/Examples/Window List/UwcWindowList.cs b/Assets/uWindowCapture/Examples/Window List/UwcWindowList.cs
@@ -35,13 +35,16 @@ void OnWindowAdded(UwcWindow window)
 
         window.RequestCaptureIcon();
         window.RequestCapture(CapturePriority.Low);
+
+        listItem.OnClick();
     }
 
     void OnWindowRemoved(UwcWindow window)
     {
         UwcWindowListItem listItem;
         items_.TryGetValue(window.id, out listItem);
         if (listItem) {
+            listItem.RemoveWindow();
             Destroy(listItem.gameObject);
         }
     }

diff --git a/Assets/uWindowCapture/Examples/Window List/UwcWindowListItem.cs b/Assets/uWindowCapture/Examples/Window List/UwcWindowListItem.cs
@@ -58,16 +58,27 @@ void Update()
 
     public void OnClick()
     {
-        var manager = list.windowTextureManager;
         if (windowTexture == null) {
-            windowTexture = manager.AddWindowTexture(window);
-            image_.color = selected;
+            AddWindow();
         } else {
-            manager.RemoveWindowTexture(window);
-            windowTexture = null;
-            image_.color = notSelected;
+            RemoveWindow();
         }
     }
+
+    void AddWindow()
+    {
+        var manager = list.windowTextureManager;
+        windowTexture = manager.AddWindowTexture(window);
+        image_.color = selected;
+    }
+
+    public void RemoveWindow()
+    {
+        var manager = list.windowTextureManager;
+        manager.RemoveWindowTexture(window);
+        windowTexture = null;
+        image_.color = notSelected;
+    }
 }
 
 }