DesktopAura: Track windows "owned" via the DesktopWindowTreeHost

ui/views/widget/native_widget_aura.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "ui/views/widget/native_widget_aura.h"
 
 #include "base/bind.h"
 #include "base/location.h"
 #include "base/memory/ptr_util.h"
 #include "base/single_thread_task_runner.h"
@@ skipping 770 matching lines @@
 }
 
 void NativeWidgetAura::RepostNativeEvent(gfx::NativeEvent native_event) {
   OnEvent(native_event);
 }
 
 std::string NativeWidgetAura::GetName() const {
   return window_ ? window_->GetName() : std::string();
 }
 
+Widget::Widgets NativeWidgetAura::GetAllOwnedTopLevelWidgets() const {
+  // Only the root window is top level.
+  return Widget::Widgets();
+}
+
 ////////////////////////////////////////////////////////////////////////////////
 // NativeWidgetAura, aura::WindowDelegate implementation:
 
 gfx::Size NativeWidgetAura::GetMinimumSize() const {
   return delegate_->GetMinimumSize();
 }
 
 gfx::Size NativeWidgetAura::GetMaximumSize() const {
   // A window should not have a maximum size and also be maximizable.
   DCHECK(delegate_->GetMaximumSize().IsEmpty() ||
@@ skipping 333 matching lines @@
     window = window->parent();
   }
   return top_level_native_widget;
 }
 
 // static
 void NativeWidgetPrivate::GetAllChildWidgets(gfx::NativeView native_view,
                                              Widget::Widgets* children) {
   {
     // Code expects widget for |native_view| to be added to |children|.
-    NativeWidgetPrivate* native_widget = static_cast<NativeWidgetPrivate*>(
-        GetNativeWidgetForNativeView(native_view));
+    NativeWidgetPrivate* native_widget =
+        GetNativeWidgetForNativeView(native_view);
     if (native_widget && native_widget->GetWidget())
       children->insert(native_widget->GetWidget());
   }
 
   const aura::Window::Windows& child_windows = native_view->children();
   for (aura::Window::Windows::const_iterator i = child_windows.begin();
        i != child_windows.end(); ++i) {
     GetAllChildWidgets((*i), children);
   }
 }
 
 // static
 void NativeWidgetPrivate::GetAllOwnedWidgets(gfx::NativeView native_view,
-                                             Widget::Widgets* owned) {
+                                             Widget::Widgets* owned,
+                                             bool include_toplevel) {
   // Add all owned widgets.
   for (aura::Window* transient_child : wm::GetTransientChildren(native_view)) {
-    NativeWidgetPrivate* native_widget = static_cast<NativeWidgetPrivate*>(
-        GetNativeWidgetForNativeView(transient_child));
+    NativeWidgetPrivate* native_widget =
+        GetNativeWidgetForNativeView(transient_child);
     if (native_widget && native_widget->GetWidget())
       owned->insert(native_widget->GetWidget());
-    GetAllOwnedWidgets(transient_child, owned);
+    GetAllOwnedWidgets(transient_child, owned, include_toplevel);
   }
 
   // Add all child windows.
   for (aura::Window* child : native_view->children())
     GetAllChildWidgets(child, owned);
+
+  if (!include_toplevel)
+    return;
+
+  // Discover potential top-level children owned via the WindowTreeHost.
+  NativeWidgetPrivate* native_widget =
+      GetNativeWidgetForNativeView(native_view);
+  if (!native_widget)
+    return;
+
+  for (Widget* child : native_widget->GetAllOwnedTopLevelWidgets()) {
+    owned->insert(child);
+    GetAllOwnedWidgets(child->GetNativeView(), owned, true);
+  }
 }
 
 // static
 void NativeWidgetPrivate::ReparentNativeView(gfx::NativeView native_view,
                                              gfx::NativeView new_parent) {
   DCHECK(native_view != new_parent);
 
   gfx::NativeView previous_parent = native_view->parent();
   if (previous_parent == new_parent)
     return;
@@ skipping 57 matching lines @@
     gfx::NativeView native_view) {
   aura::client::CaptureClient* capture_client =
       aura::client::GetCaptureClient(native_view->GetRootWindow());
   if (!capture_client)
     return nullptr;
   return capture_client->GetGlobalCaptureWindow();
 }
 
 }  // namespace internal
 }  // namespace views