DesktopAura: Track windows "owned" via the DesktopWindowTreeHost

ui/views/widget/desktop_aura/desktop_window_tree_host_win.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_window_tree_host_win.h"
 
 #include "base/memory/ptr_util.h"
 #include "third_party/skia/include/core/SkPath.h"
 #include "third_party/skia/include/core/SkRegion.h"
 #include "ui/aura/client/aura_constants.h"
@@ skipping 463 matching lines @@
 }
 
 bool DesktopWindowTreeHostWin::ShouldUpdateWindowTransparency() const {
   return true;
 }
 
 bool DesktopWindowTreeHostWin::ShouldUseDesktopNativeCursorManager() const {
   return true;
 }
 
+std::vector<DesktopWindowTreeHost*>
+DesktopWindowTreeHostWin::GetOwnedTopLevelHosts() const {
+  std::vector<DesktopWindowTreeHost*> hosts;
+  std::vector<HWND> child_windows = message_handler_->GetOwnedWindows();
+  for (HWND hwnd : child_windows) {
+    if (aura::WindowTreeHost* host = GetForAcceleratedWidget(hwnd)) {
+      if (DesktopWindowTreeHostWin* target =
+              host->window()->GetProperty(kDesktopWindowTreeHostKey)) {
+        hosts.push_back(target);
+      }
+    }
+  }
+  return hosts;
+}
+
 ////////////////////////////////////////////////////////////////////////////////
 // DesktopWindowTreeHostWin, WindowTreeHost implementation:
 
 ui::EventSource* DesktopWindowTreeHostWin::GetEventSource() {
   return this;
 }
 
 gfx::AcceleratedWidget DesktopWindowTreeHostWin::GetAcceleratedWidget() {
   return message_handler_->hwnd();
 }
@@ skipping 497 matching lines @@
 
 // static
 DesktopWindowTreeHost* DesktopWindowTreeHost::Create(
     internal::NativeWidgetDelegate* native_widget_delegate,
     DesktopNativeWidgetAura* desktop_native_widget_aura) {
   return new DesktopWindowTreeHostWin(native_widget_delegate,
                                       desktop_native_widget_aura);
 }
 
 }  // namespace views