Optimize shortcuts provider

components/omnibox/browser/match_compare.h

Index: components/omnibox/browser/match_compare.h
diff --git a/components/omnibox/browser/match_compare.h b/components/omnibox/browser/match_compare.h
new file mode 100644
index 0000000000000000000000000000000000000000..66f3beb0ba824cbe12667665000a29e2c26f6bed
--- /dev/null
+++ b/components/omnibox/browser/match_compare.h
@@ -0,0 +1,70 @@
+// Copyright 2016 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.
+
+#ifndef COMPONENTS_OMNIBOX_BROWSER_MATCH_COMPARE_H_
+#define COMPONENTS_OMNIBOX_BROWSER_MATCH_COMPARE_H_
+
+#include "components/omnibox/browser/omnibox_field_trial.h"
+
+using metrics::OmniboxEventProto;
+
+// This class implements a special version of AutocompleteMatch::MoreRelevant
+// that allows matches of particular types to be demoted in AutocompleteResult.
+template <class Match> class CompareWithDemoteByType {
+ public:
+  CompareWithDemoteByType(
+      OmniboxEventProto::PageClassification current_page_classification) {
+    OmniboxFieldTrial::GetDemotionsByType(current_page_classification,
+                                          &demotions_);
+  }
+
+  // Returns the relevance score of |match| demoted appropriately by
+  // |demotions_by_type_|.
+  int GetDemotedRelevance(const Match& match) {

[Peter Kasting, 2016/04/12 00:55:08]

Nit: Can be const

[Alexander Yashkin, 2016/04/12 09:09:20]

Done.

+      OmniboxFieldTrial::DemotionMultipliers::const_iterator demotion_it =
+      demotions_.find(match.type);
+    return (demotion_it == demotions_.end()) ?
+        match.relevance : (match.relevance * demotion_it->second);
+  }
+
+  // Comparison function.
+  bool operator()(const Match& elem1, const Match& elem2) {
+    // Compute demoted relevance scores for each match.
+    const int demoted_relevance1 = GetDemotedRelevance(elem1);
+    const int demoted_relevance2 = GetDemotedRelevance(elem2);
+    // For equal-relevance matches, we sort alphabetically, so that providers
+    // who return multiple elements at the same priority get a "stable" sort
+    // across multiple updates.
+    return (demoted_relevance1 == demoted_relevance2) ?
+        (elem1.contents < elem2.contents) :
+        (demoted_relevance1 > demoted_relevance2);
+  }
+
+ private:
+  OmniboxFieldTrial::DemotionMultipliers demotions_;
+};
+
+template<class Match> class DestinationSort {
+ public:
+  DestinationSort(
+      OmniboxEventProto::PageClassification current_page_classification) :
+      demote_by_type_(current_page_classification) {}
+  bool operator()(const Match& elem1,
+                  const Match& elem2) {
+    // Sort identical destination_urls together.
+    // Place the most relevant matches first, so that when we call
+    // std::unique(), these are the ones that get preserved.
+    if (Match::DestinationsEqual(elem1, elem2) ||
+        (elem1.stripped_destination_url.is_empty() &&
+         elem2.stripped_destination_url.is_empty())) {
+      return demote_by_type_(elem1, elem2);
+    }
+    return elem1.stripped_destination_url < elem2.stripped_destination_url;
+  }
+
+ private:
+  CompareWithDemoteByType<Match> demote_by_type_;
+};
+
+#endif  // COMPONENTS_OMNIBOX_BROWSER_MATCH_COMPARE_H_