Make launching apps from shelf more intuitive

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"
 #include "chrome/browser/ui/browser_list.h"
 #include "chrome/browser/ui/browser_window.h"
 #include "chrome/browser/ui/tab_contents/tab_contents_iterator.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "content/public/browser/navigation_controller.h"
+#include "ui/display/display.h"
+#include "ui/display/screen.h"
+#include "ui/display/types/display_constants.h"
 
 #if defined(OS_CHROMEOS)
 #include "chrome/browser/ui/ash/multi_user/multi_user_util.h"
 #include "chrome/browser/ui/ash/multi_user/multi_user_window_manager.h"
 #include "components/signin/core/account_id/account_id.h"
 #endif
 
 using content::WebContents;
 
 namespace {
 
 
 // Type used to indicate to match anything.
 const int kMatchAny                     = 0;
 
 // See BrowserMatches for details.
 const int kMatchOriginalProfile         = 1 << 0;
 const int kMatchCanSupportWindowFeature = 1 << 1;
 const int kMatchTabbed                  = 1 << 2;
+const int kMatchDisplayId               = 1 << 3;
 
 // Returns true if the specified |browser| matches the specified arguments.
 // |match_types| is a bitmask dictating what parameters to match:
 // . If it contains kMatchOriginalProfile then the original profile of the
 //   browser must match |profile->GetOriginalProfile()|. This is used to match
 //   incognito windows.
 // . If it contains kMatchCanSupportWindowFeature
 //   |CanSupportWindowFeature(window_feature)| must return true.
 // . If it contains kMatchTabbed, the browser must be a tabbed browser.
 bool BrowserMatches(Browser* browser,
                     Profile* profile,
                     Browser::WindowFeature window_feature,
-                    uint32_t match_types) {
+                    uint32_t match_types,
+                    int64_t display_id) {
   if ((match_types & kMatchCanSupportWindowFeature) &&
       !browser->CanSupportWindowFeature(window_feature)) {
     return false;
   }
 
 #if defined(OS_CHROMEOS)
   // Get the profile on which the window is currently shown.
   // MultiUserWindowManager might be NULL under test scenario.
   chrome::MultiUserWindowManager* const window_manager =
       chrome::MultiUserWindowManager::GetInstance();
@@ skipping 20 matching lines @@
 #endif
   } else {
     if (browser->profile() != profile)
       return false;
 #if defined(OS_CHROMEOS)
     if (shown_profile && shown_profile != profile)
       return false;
 #endif
   }
 
-  if (match_types & kMatchTabbed)
-    return browser->is_type_tabbed();
+  if ((match_types & kMatchTabbed) && !browser->is_type_tabbed())
+    return false;
+
+  if (match_types & kMatchDisplayId) {
+    return display::Screen::GetScreen()
+               ->GetDisplayNearestWindow(browser->window()->GetNativeWindow())
+               .id() == display_id;
+  }
 
   return true;
 }
 
 // Returns the first browser in the specified iterator that returns true from
 // |BrowserMatches|, or null if no browsers match the arguments. See
 // |BrowserMatches| for details on the arguments.
 template <class T>
 Browser* FindBrowserMatching(const T& begin,
                              const T& end,
                              Profile* profile,
                              Browser::WindowFeature window_feature,
-                             uint32_t match_types) {
+                             uint32_t match_types,
+                             int64_t display_id) {
   for (T i = begin; i != end; ++i) {
-    if (BrowserMatches(*i, profile, window_feature, match_types))
+    if (BrowserMatches(*i, profile, window_feature, match_types, display_id))
       return *i;
   }
   return NULL;
 }
 
 Browser* FindBrowserWithTabbedOrAnyType(Profile* profile,
                                         bool match_tabbed,
-                                        bool match_original_profiles) {
+                                        bool match_original_profiles,
+                                        int64_t display_id) {
   BrowserList* browser_list_impl = BrowserList::GetInstance();
   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);
+  if (display_id != display::kInvalidDisplayId)
+    match_types |= kMatchDisplayId;
+  Browser* browser =
+      FindBrowserMatching(browser_list_impl->begin_last_active(),
+                          browser_list_impl->end_last_active(), profile,
+                          Browser::FEATURE_NONE, match_types, display_id);
   // 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);
+  return browser ? browser
+                 : FindBrowserMatching(
+                       browser_list_impl->begin(), browser_list_impl->end(),
+                       profile, Browser::FEATURE_NONE, match_types, display_id);
 }
 
 size_t GetBrowserCountImpl(Profile* profile,
-                           uint32_t match_types) {
+                           uint32_t match_types,
+                           int64_t display_id) {
   BrowserList* browser_list_impl = BrowserList::GetInstance();
   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))
+      if (BrowserMatches(*i, profile, Browser::FEATURE_NONE, match_types,
+                         display_id))
         count++;
     }
   }
   return count;
 }
 
 }  // namespace
 
 namespace chrome {
 
 Browser* FindTabbedBrowser(Profile* profile,
                            bool match_original_profiles) {
-  return FindBrowserWithTabbedOrAnyType(profile, true, match_original_profiles);
+  return FindBrowserWithTabbedOrAnyType(profile, true, match_original_profiles,
+                                        display::kInvalidDisplayId);
+}
+
+Browser* FindTabbedBrowserOnDisplay(Profile* profile,
+                                    bool match_original_profiles,
+                                    int64_t display_id) {
+  return FindBrowserWithTabbedOrAnyType(profile, true, match_original_profiles,
+                                        display_id);
 }
 
 Browser* FindAnyBrowser(Profile* profile,
                         bool match_original_profiles) {
-  return FindBrowserWithTabbedOrAnyType(profile,
-                                        false,
-                                        match_original_profiles);
+  return FindBrowserWithTabbedOrAnyType(profile, false, match_original_profiles,
+                                        display::kInvalidDisplayId);

[Peter Kasting, 2017/05/26 01:25:03]

Nit: We could reduce the number of callers restating kInvalidDisplayId values
they don't care about by using more default args above.

[weidongg, 2017/05/26 02:49:47]

Done.

 }
 
 Browser* FindBrowserWithProfile(Profile* profile) {
-  return FindBrowserWithTabbedOrAnyType(profile, false, false);
+  return FindBrowserWithTabbedOrAnyType(profile, false, false,
+                                        display::kInvalidDisplayId);
 }
 
 Browser* FindBrowserWithID(SessionID::id_type desired_id) {
   for (auto* browser : *BrowserList::GetInstance()) {
     if (browser->session_id().id() == desired_id)
       return browser;
   }
   return NULL;
 }
 
@@ skipping 14 matching lines @@
       return it.browser();
   }
   return NULL;
 }
 
 Browser* FindLastActiveWithProfile(Profile* profile) {
   BrowserList* list = BrowserList::GetInstance();
   // We are only interested in last active browsers, so we don't fall back to
   // all browsers like FindBrowserWith* do.
   return FindBrowserMatching(list->begin_last_active(), list->end_last_active(),
-                             profile, Browser::FEATURE_NONE, kMatchAny);
+                             profile, Browser::FEATURE_NONE, kMatchAny,
+                             display::kInvalidDisplayId);
 }
 
 Browser* FindLastActive() {
   BrowserList* browser_list_impl = BrowserList::GetInstance();
   if (browser_list_impl)
     return browser_list_impl->GetLastActive();
   return NULL;
 }
 
 size_t GetTotalBrowserCount() {
   return BrowserList::GetInstance()->size();
 }
 
 size_t GetBrowserCount(Profile* profile) {
-  return GetBrowserCountImpl(profile, kMatchAny);
+  return GetBrowserCountImpl(profile, kMatchAny, display::kInvalidDisplayId);
 }
 
 size_t GetTabbedBrowserCount(Profile* profile) {
-  return GetBrowserCountImpl(profile, kMatchTabbed);
+  return GetBrowserCountImpl(profile, kMatchTabbed, display::kInvalidDisplayId);
 }
 
 }  // namespace chrome