Desktop chrome tab capture-chooseDesktopMedia()

chrome/browser/ui/browser_finder.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 "chrome/browser/ui/browser_finder.h"
 
 #include <stdint.h>
 
 #include "build/build_config.h"
 #include "chrome/browser/profiles/profile.h"
@@ skipping 90 matching lines @@
                              Profile* profile,
                              Browser::WindowFeature window_feature,
                              uint32_t match_types) {
   for (T i = begin; i != end; ++i) {
     if (BrowserMatches(*i, profile, window_feature, match_types))
       return *i;
   }
   return NULL;
 }
 
+// Returns all browsers in the specified iterator that returns true from
+// |BrowserMatches|. See |BrowserMatches| for details on the arguments.
+template <class T>
+std::vector<Browser*> FindAllBrowsersMatching(
+    const T& begin,
+    const T& end,
+    Profile* profile,
+    Browser::WindowFeature window_feature,
+    uint32_t match_types) {
+  std::vector<Browser*> matches;
+  for (T i = begin; i != end; ++i) {
+    if (BrowserMatches(*i, profile, window_feature, match_types))
+      matches.push_back(*i);
+  }
+  return matches;
+}
+
 Browser* FindBrowserWithTabbedOrAnyType(Profile* profile,
                                         chrome::HostDesktopType desktop_type,
                                         bool match_tabbed,
                                         bool match_original_profiles) {
   BrowserList* browser_list_impl = BrowserList::GetInstance(desktop_type);
   if (!browser_list_impl)
     return NULL;
   uint32_t match_types = kMatchAny;
   if (match_tabbed)
     match_types |= kMatchTabbed;
   if (match_original_profiles)
     match_types |= kMatchOriginalProfile;
   Browser* browser = FindBrowserMatching(browser_list_impl->begin_last_active(),
                                          browser_list_impl->end_last_active(),
                                          profile,
                                          Browser::FEATURE_NONE,
                                          match_types);
   // Fall back to a forward scan of all Browsers if no active one was found.
   return browser ? browser : FindBrowserMatching(browser_list_impl->begin(),
                                                  browser_list_impl->end(),
                                                  profile,
                                                  Browser::FEATURE_NONE,
                                                  match_types);
 }
 
+std::vector<Browser*> FindAllBrowsersWithTabbedOrAnyType(
+    Profile* profile,
+    chrome::HostDesktopType desktop_type,
+    bool match_tabbed,
+    bool match_original_profiles) {
+  std::vector<Browser*> matches;
+  BrowserList* browser_list_impl = BrowserList::GetInstance(desktop_type);
+  if (!browser_list_impl)
+    return matches;
+  uint32_t match_types = kMatchAny;
+  if (match_tabbed)
+    match_types |= kMatchTabbed;
+  if (match_original_profiles)
+    match_types |= kMatchOriginalProfile;
+  matches =
+      FindAllBrowsersMatching(browser_list_impl->begin_last_active(),
+                              browser_list_impl->end_last_active(), profile,
+                              Browser::FEATURE_NONE, match_types);
+  if (!matches.empty())
+    return matches;
+
+  // Fall back to a forward scan of all Browsers if no active one was found.
+  matches = FindAllBrowsersMatching(browser_list_impl->begin(),
+                                    browser_list_impl->end(), profile,
+                                    Browser::FEATURE_NONE, match_types);
+
+  return matches;
+}
+
 size_t GetBrowserCountImpl(Profile* profile,
                            chrome::HostDesktopType desktop_type,
                            uint32_t match_types) {
   BrowserList* browser_list_impl = BrowserList::GetInstance(desktop_type);
   size_t count = 0;
   if (browser_list_impl) {
     for (BrowserList::const_iterator i = browser_list_impl->begin();
          i != browser_list_impl->end(); ++i) {
       if (BrowserMatches(*i, profile, Browser::FEATURE_NONE, match_types))
         count++;
@@ skipping 22 matching lines @@
                                         type,
                                         false,
                                         match_original_profiles);
 }
 
 Browser* FindBrowserWithProfile(Profile* profile,
                                 HostDesktopType desktop_type) {
   return FindBrowserWithTabbedOrAnyType(profile, desktop_type, false, false);
 }
 
+std::vector<Browser*> FindAllTabbedBrowsersWithProfile(
+    Profile* profile,
+    HostDesktopType desktop_type) {
+  return FindAllBrowsersWithTabbedOrAnyType(profile, desktop_type, true, false);
+}
+
 Browser* FindBrowserWithID(SessionID::id_type desired_id) {
   for (BrowserIterator it; !it.done(); it.Next()) {
     if (it->session_id().id() == desired_id)
       return *it;
   }
   return NULL;
 }
 
 Browser* FindBrowserWithWindow(gfx::NativeWindow window) {
   if (!window)
@@ skipping 50 matching lines @@
 
 size_t GetBrowserCount(Profile* profile, HostDesktopType type) {
   return GetBrowserCountImpl(profile, type, kMatchAny);
 }
 
 size_t GetTabbedBrowserCount(Profile* profile, HostDesktopType type) {
   return GetBrowserCountImpl(profile, type, kMatchTabbed);
 }
 
 }  // namespace chrome