Add support for a virtual display on ChromeOS

chrome/browser/media/desktop_media_list_ash.cc

 // Copyright 2013 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 "chrome/browser/media/desktop_media_list_ash.h"
 
 #include <set>
 
+#include "ash/display/display_controller.h"
 #include "ash/shell.h"
 #include "ash/shell_window_ids.h"
 #include "base/hash.h"
 #include "base/logging.h"
+#include "base/memory/singleton.h"
 #include "base/strings/utf_string_conversions.h"
+#include "base/sys_info.h"
 #include "base/threading/sequenced_worker_pool.h"
 #include "chrome/browser/media/desktop_media_list_observer.h"
 #include "chrome/grit/generated_resources.h"
 #include "content/public/browser/browser_thread.h"
 #include "media/base/video_util.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/compositor/dip_util.h"
+#include "ui/display/chromeos/display_configurator.h"
+#include "ui/gfx/display_observer.h"
 #include "ui/gfx/image/image.h"
 #include "ui/snapshot/snapshot.h"
 
 using content::BrowserThread;
 using content::DesktopMediaID;
 
 namespace {
 
 // Update the list twice per second.
 const int kDefaultUpdatePeriod = 500;
 
+#if defined(OS_CHROMEOS)
+// Watches for the creation of a virtual display and updates
+// DesktopMediaID::AuraWindowRegistry.
+class VirtualScreenMonitor : public gfx::DisplayObserver {
+ public:
+  VirtualScreenMonitor();
+  ~VirtualScreenMonitor();
+
+  // gfx::DisplayObserver overrides:
+  void OnDisplayAdded(const gfx::Display& display) override;
+  void OnDisplayRemoved(const gfx::Display& display) override {}
+  void OnDisplayMetricsChanged(const gfx::Display& display,
+                               uint32_t metrics) override {}
+
+  // static
+  static VirtualScreenMonitor* GetInstance();
+
+ private:
+  friend struct DefaultSingletonTraits<VirtualScreenMonitor>;
+
+  DISALLOW_COPY_AND_ASSIGN(VirtualScreenMonitor);
+};
+
+VirtualScreenMonitor::VirtualScreenMonitor() {
+  gfx::Screen::GetNativeScreen()->AddObserver(this);
+}
+
+VirtualScreenMonitor::~VirtualScreenMonitor() {
+}
+
+// Subsequent observers (e.g. from the VirtualDisplayCapturer class) will see
+// the updated virtual screen window as the observers are called in the order
+// they are added.
+void VirtualScreenMonitor::OnDisplayAdded(const gfx::Display& display) {
+  DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
+  if (display.id() ==
+      ui::DisplayConfigurator::GetInstance()->GetVirtualDisplayId()) {
+    aura::Window* window = ash::Shell::GetInstance()
+                               ->display_controller()
+                               ->GetRootWindowForDisplayId(display.id());
+
+    content::DesktopMediaID::RegisterVirtualScreenAuraWindow(window);
+  }
+}
+
+// static
+VirtualScreenMonitor* VirtualScreenMonitor::GetInstance() {
+  return Singleton<VirtualScreenMonitor>::get();
+}
+
+#endif  // defined(OS_CHROMEOS)
+
 }  // namespace
 
 DesktopMediaListAsh::SourceDescription::SourceDescription(
     DesktopMediaID id,
     const base::string16& name)
     : id(id),
       name(name) {
 }
 
 DesktopMediaListAsh::DesktopMediaListAsh(int source_types)
@@ skipping 160 matching lines @@
 
     if (source_types_ & WINDOWS) {
       EnumerateWindowsForRoot(
           sources, root_windows[i], ash::kShellWindowId_DefaultContainer);
       EnumerateWindowsForRoot(
           sources, root_windows[i], ash::kShellWindowId_AlwaysOnTopContainer);
       EnumerateWindowsForRoot(
           sources, root_windows[i], ash::kShellWindowId_DockedContainer);
     }
   }
+#if defined(OS_CHROMEOS)
+  // DisplayConfigurator only works when running on ChromeOS.
+  if (base::SysInfo::IsRunningOnChromeOS()) {
+    if (ui::DisplayConfigurator::GetInstance()->GetVirtualDisplayId() ==
+        gfx::Display::kInvalidDisplayID) {
+      // Instantiate the VirtualScreenMonitor object which will observe for
+      // the virtual display being created and store the window so that it can
+      // be looked up from the content::DesktopMediaID.
+      (void)VirtualScreenMonitor::GetInstance();
+      SourceDescription virtual_screen_source(
+          content::DesktopMediaID(
+              content::DesktopMediaID::TYPE_AURA_VIRTUAL_SCREEN, 0),
+          l10n_util::GetStringUTF16(
+              IDS_DESTOP_MEDIA_PICKER_VIRTUAL_SCREEN_NAME));
+      sources->push_back(virtual_screen_source);
+      // TODO(robert.bradford): crbug.com/425060 There is no thumbnail for the
+      // virtual screen right now.
+    }
+  }
+#endif  // defined(OS_CHROMEOS)
 }
 
 void DesktopMediaListAsh::CaptureThumbnail(content::DesktopMediaID id,
                                            aura::Window* window) {
   gfx::Rect window_rect(window->bounds().width(), window->bounds().height());
   gfx::Rect scaled_rect = media::ComputeLetterboxRegion(
       gfx::Rect(thumbnail_size_), window_rect.size());
 
   ++pending_window_capture_requests_;
   ui::GrabWindowSnapshotAndScaleAsync(
@@ skipping 22 matching lines @@
   if (!pending_window_capture_requests_) {
     // Once we've finished capturing all windows post a task for the next list
     // update.
     BrowserThread::PostDelayedTask(
         BrowserThread::UI, FROM_HERE,
         base::Bind(&DesktopMediaListAsh::Refresh,
                    weak_factory_.GetWeakPtr()),
         update_period_);
   }
 }