DesktopAura: Track windows "owned" via the DesktopWindowTreeHost

ui/views/widget/desktop_aura/desktop_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/desktop_aura/desktop_native_widget_aura.h"
 
 #include "base/bind.h"
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/trace_event/trace_event.h"
@@ skipping 944 matching lines @@
 }
 
 void DesktopNativeWidgetAura::RepostNativeEvent(gfx::NativeEvent native_event) {
   OnEvent(native_event);
 }
 
 std::string DesktopNativeWidgetAura::GetName() const {
   return name_;
 }
 
+Widget::Widgets DesktopNativeWidgetAura::GetAllOwnedTopLevelWidgets() const {
+  Widget::Widgets owned;
+  for (DesktopWindowTreeHost* owned_host :
+       desktop_window_tree_host_->GetOwnedTopLevelHosts()) {
+    DesktopNativeWidgetAura* child =
+        ForWindow(owned_host->AsWindowTreeHost()->window());
+    DCHECK(child);
+    owned.insert(child->GetWidget());
+  }
+  return owned;
+}
+
 ////////////////////////////////////////////////////////////////////////////////
 // DesktopNativeWidgetAura, aura::WindowDelegate implementation:
 
 gfx::Size DesktopNativeWidgetAura::GetMinimumSize() const {
   return native_widget_delegate_->GetMinimumSize();
 }
 
 gfx::Size DesktopNativeWidgetAura::GetMaximumSize() const {
   return native_widget_delegate_->GetMaximumSize();
 }
@@ skipping 233 matching lines @@
   if (cursor_reference_count_ == 0) {
     // We are the last DesktopNativeWidgetAura instance, and we are responsible
     // for cleaning up |cursor_manager_|.
     delete cursor_manager_;
     native_cursor_manager_ = NULL;
     cursor_manager_ = NULL;
   }
 }
 
 }  // namespace views