Remove use of window list in Viewport.cpp

src/openrct2-ui/WindowManager.cpp

@@ -502,6 +502,18 @@ public:
     {
         window_all_wheel_input();
     }
+
+    rct_window* GetOwner(const rct_viewport* viewport) override
+    {
+        for (auto w = g_window_list; w < gWindowNextSlot; w++)
+        {
+            if (w->viewport == viewport)
+            {
+                return w;
+            }
+        }
+        return nullptr;
+    }
 };
 
 IWindowManager * OpenRCT2::Ui::CreateWindowManager()

src/openrct2/interface/Viewport.cpp

@@ -24,11 +24,15 @@
 #include "../world/Climate.h"
 #include "../world/Map.h"
 #include "../world/Sprite.h"
+#include "../ui/UiContext.h"
+#include "../ui/WindowManager.h"
 #include "Colour.h"
 #include "Viewport.h"
 #include "Window.h"
 #include "Window_internal.h"
 
+using namespace OpenRCT2;
+
 //#define DEBUG_SHOW_DIRTY_BOX
 uint8 gShowGridLinesRefCount;
 uint8 gShowLandRightsRefCount;
@@ -1538,17 +1542,17 @@ void viewport_invalidate(rct_viewport *viewport, sint32 left, sint32 top, sint32
     // if unknown viewport visibility, use the containing window to discover the status
     if (viewport->visibility == VC_UNKNOWN)
     {
-        for (rct_window *w = g_window_list; w < gWindowNextSlot; w++)
-        {
-            if (w->classification != WC_MAIN_WINDOW && w->viewport != nullptr && w->viewport == viewport)
+        auto windowManager = GetContext()->GetUiContext()->GetWindowManager();
+        auto owner = windowManager->GetOwner(viewport);
+        if (owner != nullptr && owner->classification != WC_MAIN_WINDOW)
         {
             // note, window_is_visible will update viewport->visibility, so this should have a low hit count
-                if (!window_is_visible(w)) {
+            if (!window_is_visible(owner))
+            {
                 return;
             }
         }
     }
-    }
 
     if (viewport->visibility == VC_COVERED) return;
 

src/openrct2/ui/DummyWindowManager.cpp

@@ -27,6 +27,7 @@ namespace OpenRCT2::Ui
         std::string GetKeyboardShortcutString(sint32 /*shortcut*/) override { return std::string(); }
         void SetMainView(sint32 x, sint32 y, sint32 zoom, sint32 rotation) override { }
         void UpdateMouseWheel() override { }
+        rct_window* GetOwner(const rct_viewport* viewport) override { return nullptr; }
     };
 
     IWindowManager * CreateDummyWindowManager()

src/openrct2/ui/WindowManager.h

@@ -37,6 +37,7 @@ namespace OpenRCT2::Ui
         virtual std::string GetKeyboardShortcutString(sint32 shortcut) abstract;
         virtual void SetMainView(sint32 x, sint32 y, sint32 zoom, sint32 rotation) abstract;
         virtual void UpdateMouseWheel() abstract;
+        virtual rct_window* GetOwner(const rct_viewport* viewport) abstract;
     };
 
     IWindowManager * CreateDummyWindowManager();