Desktop chrome tab capture-chooseDesktopMedia()

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/shell.h"
 #include "ash/shell_window_ids.h"
 #include "base/hash.h"
 #include "base/logging.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/threading/sequenced_worker_pool.h"
 #include "chrome/browser/media/desktop_media_list_observer.h"
+#include "chrome/browser/profiles/profile_manager.h"
+#include "chrome/browser/ui/browser_finder.h"
+#include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/grit/generated_resources.h"
+#include "components/favicon/content/content_favicon_driver.h"
 #include "content/public/browser/browser_thread.h"
+#include "content/public/browser/render_frame_host.h"
+#include "content/public/browser/render_process_host.h"
 #include "media/base/video_util.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/compositor/dip_util.h"
 #include "ui/gfx/image/image.h"
 #include "ui/snapshot/snapshot.h"
 
 using content::BrowserThread;
 using content::DesktopMediaID;
 
 namespace {
@@ skipping 134 matching lines @@
         content::DesktopMediaID::TYPE_WINDOW, *it);
     if (id.aura_id == view_dialog_id_.aura_id)
       continue;
     SourceDescription window_source(id, (*it)->title());
     sources->push_back(window_source);
 
     CaptureThumbnail(window_source.id, *it);
   }
 }
 
+void DesktopMediaListAsh::EnumerateTabs(
+    std::vector<DesktopMediaListAsh::SourceDescription>* sources) {
+  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
+
+  Profile* profile = ProfileManager::GetLastUsedProfileAllowedByPolicy();
+  std::vector<Browser*> browsers = FindAllTabbedBrowsersWithProfile(
+      profile, chrome::HOST_DESKTOP_TYPE_NATIVE);
+
+  std::map<base::TimeTicks, SourceDescription> tab_map;
+
+  for (auto browser : browsers) {
+    TabStripModel* tab_strip_model = browser->tab_strip_model();
+    DCHECK(tab_strip_model);
+
+    const int kNumTabs = tab_strip_model->count();

[miu, 2015/12/08 01:54:38]

naming: This variable is not assigned from a literal value.  Please call it
|number_of_tabs| instead.

[GeorgeZ, 2015/12/09 19:36:37]

Changed for loop to 
   for (int i = 0; i < tab_strip_model->count(); i++) {
to remove this variable

+    for (int i = 0; i < kNumTabs; i++) {
+      // Create DesktopMediaID.
+      content::WebContents* contents = tab_strip_model->GetWebContentsAt(i);
+      gfx::NativeWindow tab_window = contents->GetTopLevelNativeWindow();
+      DesktopMediaID media_id = DesktopMediaID::RegisterAuraWindow(
+          DesktopMediaID::TYPE_TAB, tab_window);
+
+      content::RenderFrameHost* const main_frame = contents->GetMainFrame();
+      DCHECK(main_frame);
+
+      media_id.render_process_id = main_frame->GetProcess()->GetID();
+      media_id.main_render_frame_id = main_frame->GetRoutingID();
+
+      // Get tab title.
+      base::string16 title =
+          base::UTF8ToUTF16("Chrome tab_") + contents->GetTitle();

[qiangchen, 2015/12/07 22:45:56]

Just wonder do we need to throw "Chrome tab_" to resource for translation?

[miu, 2015/12/08 01:54:38]

1. This string needs to be in generated_resources.grd so it can be translated
into multiple languages.

2. "Chrome tab_" is kind of weird.  How about "Tab -- "?  Or, maybe a UX
designer can help you with this?

[GeorgeZ, 2015/12/09 19:36:37]

Done.

[GeorgeZ, 2015/12/09 19:36:37]

For 1, I moved string into resources.grd. For 2, I used to used "tab_", however
some reviewers think "chrome tab_" is  better. Any way, there is going to have
UI refactor later and prefix string most likely will disappear.

+
+      // Get tab's last active time stamp.
+      base::TimeTicks t = contents->GetLastActiveTime();
+      tab_map.insert(std::make_pair(t, SourceDescription(media_id, title)));
+
+      // Get thumbnail for tab.

[miu, 2015/12/08 01:54:38]

These aren't tab thumbnails.  They're favicons.  Please use consistent
terminology in all of the variable/function names and code comments throughout
this CL.

[GeorgeZ, 2015/12/09 19:36:37]

The whole block is for thumbnail creation. I will change comment to "Create
thumbnail based on favicon for tab." to remove ambiguous.


I Created a background thumbnail and put the scaled/Enlarged favicon in its
Center. One reason to do so is that if I use Favicon directly for
tab_map[t].thumbnail. Mac and windows system will behave diffently.

+      favicon::FaviconDriver* favicon_driver =
+          favicon::ContentFaviconDriver::FromWebContents(contents);
+      if (favicon_driver) {
+        ++pending_window_capture_requests_;
+        gfx::Image tab_icon = favicon_driver->GetFavicon();
+        tab_map[t].thumbnail =
+            CreateImageEncloseFavicon(thumbnail_size_, tab_icon);
+      }
+    }
+  }
+
+  // Add timely sorted tab sources into vector. Most recent one first.
+  for (auto it = tab_map.rbegin(); it != tab_map.rend(); ++it) {
+    SourceDescription& source = it->second;
+    sources->push_back(source);
+    if (!it->second.thumbnail.isNull())
+      OnThumbnailCaptured(source.id, gfx::Image(source.thumbnail));
+  }
+}
+
 void DesktopMediaListAsh::EnumerateSources(
     std::vector<DesktopMediaListAsh::SourceDescription>* sources) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
 
   aura::Window::Windows root_windows = ash::Shell::GetAllRootWindows();
 
   for (size_t i = 0; i < root_windows.size(); ++i) {
     if (source_types_ & SCREENS) {
       SourceDescription screen_source(
           content::DesktopMediaID::RegisterAuraWindow(
@@ skipping 20 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 (source_types_ & TABS)
+      EnumerateTabs(sources);
   }
 }
 
 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_;
@@ skipping 23 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_);
   }
 }