Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(961)

Side by Side Diff: chrome/browser/predictors/autocomplete_action_predictor.h

Issue 9610006: Refactoring, moving and renaming the NetworkActionPredictor. (Closed) Base URL: http://git.chromium.org/chromium/src.git@master
Patch Set: Breaking the CL to be only the moved files. Created 8 years, 9 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #ifndef CHROME_BROWSER_AUTOCOMPLETE_NETWORK_ACTION_PREDICTOR_H_ 5 #ifndef CHROME_BROWSER_PREDICTORS_AUTOCOMPLETE_ACTION_PREDICTOR_H_
6 #define CHROME_BROWSER_AUTOCOMPLETE_NETWORK_ACTION_PREDICTOR_H_ 6 #define CHROME_BROWSER_PREDICTORS_AUTOCOMPLETE_ACTION_PREDICTOR_H_
7 #pragma once 7 #pragma once
8 8
9 #include <map> 9 #include <map>
10 10
11 #include "base/gtest_prod_util.h" 11 #include "base/gtest_prod_util.h"
12 #include "base/memory/scoped_ptr.h" 12 #include "base/memory/scoped_ptr.h"
13 #include "base/memory/weak_ptr.h" 13 #include "base/memory/weak_ptr.h"
14 #include "base/string16.h" 14 #include "base/string16.h"
15 #include "chrome/browser/autocomplete/network_action_predictor_database.h" 15 #include "chrome/browser/predictors/autocomplete_action_predictor_database.h"
16 #include "chrome/browser/profiles/profile_keyed_service.h" 16 #include "chrome/browser/profiles/profile_keyed_service.h"
17 #include "content/public/browser/notification_observer.h" 17 #include "content/public/browser/notification_observer.h"
18 #include "content/public/browser/notification_registrar.h" 18 #include "content/public/browser/notification_registrar.h"
19 #include "googleurl/src/gurl.h" 19 #include "googleurl/src/gurl.h"
20 20
21 struct AutocompleteLog; 21 struct AutocompleteLog;
22 struct AutocompleteMatch; 22 struct AutocompleteMatch;
23 class AutocompleteResult; 23 class AutocompleteResult;
24 class HistoryService; 24 class HistoryService;
25 class Profile; 25 class Profile;
26 26
27 namespace history { 27 namespace history {
28 class URLDatabase; 28 class URLDatabase;
29 } 29 }
30 30
31 // This class is responsible for determining the correct predictive network 31 // This class is responsible for determining the correct predictive network
32 // action to take given for a given AutocompleteMatch and entered text. it uses 32 // action to take given for a given AutocompleteMatch and entered text. it uses
33 // a NetworkActionPredictorDatabase accessed asynchronously on the DB thread to 33 // a AutocompleteActionPredictorDatabase accessed asynchronously on the DB
cbentzel 2012/03/29 00:34:28 Nit: an Autocomplete
Shishir 2012/03/29 00:43:55 Done.
34 // permanently store the data used to make predictions, and keeps local caches 34 // thread to permanently store the data used to make predictions, and keeps
35 // of that data to be able to make predictions synchronously on the UI thread 35 // local caches of that data to be able to make predictions synchronously on the
36 // where it lives. It can be accessed as a weak pointer so that it can safely 36 // UI thread where it lives. It can be accessed as a weak pointer so that it can
37 // use PostTaskAndReply without fear of crashes if it is destroyed before the 37 // safely use PostTaskAndReply without fear of crashes if it is destroyed before
38 // reply triggers. This is necessary during initialization. 38 // the reply triggers. This is necessary during initialization.
39 class NetworkActionPredictor 39 class AutocompleteActionPredictor
40 : public ProfileKeyedService, 40 : public ProfileKeyedService,
41 public content::NotificationObserver, 41 public content::NotificationObserver,
42 public base::SupportsWeakPtr<NetworkActionPredictor> { 42 public base::SupportsWeakPtr<AutocompleteActionPredictor> {
43 public: 43 public:
44 enum Action { 44 enum Action {
45 ACTION_PRERENDER = 0, 45 ACTION_PRERENDER = 0,
46 ACTION_PRECONNECT, 46 ACTION_PRECONNECT,
47 ACTION_NONE, 47 ACTION_NONE,
48 LAST_PREDICT_ACTION = ACTION_NONE 48 LAST_PREDICT_ACTION = ACTION_NONE
49 }; 49 };
50 50
51 explicit NetworkActionPredictor(Profile* profile); 51 explicit AutocompleteActionPredictor(Profile* profile);
52 virtual ~NetworkActionPredictor(); 52 virtual ~AutocompleteActionPredictor();
53 53
54 static void set_hit_weight(double weight) { hit_weight_ = weight; } 54 static void set_hit_weight(double weight) { hit_weight_ = weight; }
55 static double get_hit_weight() { return hit_weight_; } 55 static double get_hit_weight() { return hit_weight_; }
56 56
57 // Registers an AutocompleteResult for a given |user_text|. This will be used 57 // Registers an AutocompleteResult for a given |user_text|. This will be used
58 // when the user navigates from the Omnibox to determine early opportunities 58 // when the user navigates from the Omnibox to determine early opportunities
59 // to predict their actions. 59 // to predict their actions.
60 void RegisterTransitionalMatches(const string16& user_text, 60 void RegisterTransitionalMatches(const string16& user_text,
61 const AutocompleteResult& result); 61 const AutocompleteResult& result);
62 62
63 // Clears any transitional matches that have been registered. Called when, for 63 // Clears any transitional matches that have been registered. Called when, for
64 // example, the AutocompleteEditModel is reverted. 64 // example, the AutocompleteEditModel is reverted.
65 void ClearTransitionalMatches(); 65 void ClearTransitionalMatches();
66 66
67 // Return the recommended action given |user_text|, the text the user has 67 // Return the recommended action given |user_text|, the text the user has
68 // entered in the Omnibox, and |match|, the suggestion from Autocomplete. 68 // entered in the Omnibox, and |match|, the suggestion from Autocomplete.
69 // This method uses information from the ShortcutsBackend including how much 69 // This method uses information from the ShortcutsBackend including how much
70 // of the matching entry the user typed, and how long it's been since the user 70 // of the matching entry the user typed, and how long it's been since the user
71 // visited the matching URL, to calculate a score between 0 and 1. This score 71 // visited the matching URL, to calculate a score between 0 and 1. This score
72 // is then mapped to an Action. 72 // is then mapped to an Action.
73 Action RecommendAction(const string16& user_text, 73 Action RecommendAction(const string16& user_text,
74 const AutocompleteMatch& match) const; 74 const AutocompleteMatch& match) const;
75 75
76 // Return true if the suggestion type warrants a TCP/IP preconnection. 76 // Return true if the suggestion type warrants a TCP/IP preconnection.
77 // i.e., it is now quite likely that the user will select the related domain. 77 // i.e., it is now quite likely that the user will select the related domain.
78 static bool IsPreconnectable(const AutocompleteMatch& match); 78 static bool IsPreconnectable(const AutocompleteMatch& match);
79 79
80 private: 80 private:
81 friend class NetworkActionPredictorTest; 81 friend class AutocompleteActionPredictorTest;
82 friend class NetworkActionPredictorDOMHandler; 82 friend class AutocompleteActionPredictorDOMHandler;
83 83
84 struct TransitionalMatch { 84 struct TransitionalMatch {
85 TransitionalMatch(); 85 TransitionalMatch();
86 ~TransitionalMatch(); 86 ~TransitionalMatch();
87 87
88 string16 user_text; 88 string16 user_text;
89 std::vector<GURL> urls; 89 std::vector<GURL> urls;
90 90
91 bool operator==(const string16& other_user_text) const { 91 bool operator==(const string16& other_user_text) const {
92 return user_text == other_user_text; 92 return user_text == other_user_text;
(...skipping 13 matching lines...) Expand all
106 return (user_text == rhs.user_text) && (url == rhs.url); 106 return (user_text == rhs.user_text) && (url == rhs.url);
107 } 107 }
108 }; 108 };
109 109
110 struct DBCacheValue { 110 struct DBCacheValue {
111 int number_of_hits; 111 int number_of_hits;
112 int number_of_misses; 112 int number_of_misses;
113 }; 113 };
114 114
115 typedef std::map<DBCacheKey, DBCacheValue> DBCacheMap; 115 typedef std::map<DBCacheKey, DBCacheValue> DBCacheMap;
116 typedef std::map<DBCacheKey, NetworkActionPredictorDatabase::Row::Id> 116 typedef std::map<DBCacheKey, AutocompleteActionPredictorDatabase::Row::Id>
117 DBIdCacheMap; 117 DBIdCacheMap;
118 118
119 static const int kMaximumDaysToKeepEntry; 119 static const int kMaximumDaysToKeepEntry;
120 120
121 // Multiplying factor applied to the |number_of_hits| for a database entry 121 // Multiplying factor applied to the |number_of_hits| for a database entry
122 // when calculating the confidence. It is currently set by a field trial so is 122 // when calculating the confidence. It is currently set by a field trial so is
123 // static. Once the field trial ends, this will be a constant value. 123 // static. Once the field trial ends, this will be a constant value.
124 static double hit_weight_; 124 static double hit_weight_;
125 125
126 // ProfileKeyedService 126 // ProfileKeyedService
127 virtual void Shutdown() OVERRIDE; 127 virtual void Shutdown() OVERRIDE;
128 128
129 // NotificationObserver 129 // NotificationObserver
130 virtual void Observe(int type, 130 virtual void Observe(int type,
131 const content::NotificationSource& source, 131 const content::NotificationSource& source,
132 const content::NotificationDetails& details) OVERRIDE; 132 const content::NotificationDetails& details) OVERRIDE;
133 133
134 // Called when NOTIFICATION_OMNIBOX_OPENED_URL is observed. 134 // Called when NOTIFICATION_OMNIBOX_OPENED_URL is observed.
135 void OnOmniboxOpenedUrl(const AutocompleteLog& log); 135 void OnOmniboxOpenedUrl(const AutocompleteLog& log);
136 136
137 // Deletes any old or invalid entries from the local caches. |url_db| and 137 // Deletes any old or invalid entries from the local caches. |url_db| and
138 // |id_list| must not be NULL. Every row id deleted will be added to id_list. 138 // |id_list| must not be NULL. Every row id deleted will be added to id_list.
139 void DeleteOldIdsFromCaches( 139 void DeleteOldIdsFromCaches(
140 history::URLDatabase* url_db, 140 history::URLDatabase* url_db,
141 std::vector<NetworkActionPredictorDatabase::Row::Id>* id_list); 141 std::vector<AutocompleteActionPredictorDatabase::Row::Id>* id_list);
142 142
143 // Called to delete any old or invalid entries from the database. Called after 143 // Called to delete any old or invalid entries from the database. Called after
144 // the local caches are created once the history service is available. 144 // the local caches are created once the history service is available.
145 void DeleteOldEntries(history::URLDatabase* url_db); 145 void DeleteOldEntries(history::URLDatabase* url_db);
146 146
147 // Called to populate the local caches. This also calls DeleteOldEntries 147 // Called to populate the local caches. This also calls DeleteOldEntries
148 // if the history service is available, or registers for the notification of 148 // if the history service is available, or registers for the notification of
149 // it becoming available. 149 // it becoming available.
150 void CreateCaches( 150 void CreateCaches(
151 std::vector<NetworkActionPredictorDatabase::Row>* row_buffer); 151 std::vector<AutocompleteActionPredictorDatabase::Row>* row_buffer);
152 152
153 // Attempts to call DeleteOldEntries if the in-memory database has been loaded 153 // Attempts to call DeleteOldEntries if the in-memory database has been loaded
154 // by |service|. Returns success as a boolean. 154 // by |service|. Returns success as a boolean.
155 bool TryDeleteOldEntries(HistoryService* service); 155 bool TryDeleteOldEntries(HistoryService* service);
156 156
157 // Uses local caches to calculate an exact percentage prediction that the user 157 // Uses local caches to calculate an exact percentage prediction that the user
158 // will take a particular match given what they have typed. |is_in_db| is set 158 // will take a particular match given what they have typed. |is_in_db| is set
159 // to differentiate trivial zero results resulting from a match not being 159 // to differentiate trivial zero results resulting from a match not being
160 // found from actual zero results where the calculation returns 0.0. 160 // found from actual zero results where the calculation returns 0.0.
161 double CalculateConfidence(const string16& user_text, 161 double CalculateConfidence(const string16& user_text,
162 const AutocompleteMatch& match, 162 const AutocompleteMatch& match,
163 bool* is_in_db) const; 163 bool* is_in_db) const;
164 164
165 // Calculates the confidence for an entry in the DBCacheMap. 165 // Calculates the confidence for an entry in the DBCacheMap.
166 double CalculateConfidenceForDbEntry(DBCacheMap::const_iterator iter) const; 166 double CalculateConfidenceForDbEntry(DBCacheMap::const_iterator iter) const;
167 167
168 // Adds a row to the database and caches. 168 // Adds a row to the database and caches.
169 void AddRow(const DBCacheKey& key, 169 void AddRow(const DBCacheKey& key,
170 const NetworkActionPredictorDatabase::Row& row); 170 const AutocompleteActionPredictorDatabase::Row& row);
171 171
172 // Updates a row in the database and the caches. 172 // Updates a row in the database and the caches.
173 void UpdateRow(DBCacheMap::iterator it, 173 void UpdateRow(DBCacheMap::iterator it,
174 const NetworkActionPredictorDatabase::Row& row); 174 const AutocompleteActionPredictorDatabase::Row& row);
175 175
176 // Removes all rows from the database and caches. 176 // Removes all rows from the database and caches.
177 void DeleteAllRows(); 177 void DeleteAllRows();
178 178
179 // Removes rows from the database and caches that contain a URL in |urls|. 179 // Removes rows from the database and caches that contain a URL in |urls|.
180 void DeleteRowsWithURLs(const std::set<GURL>& urls); 180 void DeleteRowsWithURLs(const std::set<GURL>& urls);
181 181
182 // Used to batch operations on the database. 182 // Used to batch operations on the database.
183 void BeginTransaction(); 183 void BeginTransaction();
184 void CommitTransaction(); 184 void CommitTransaction();
185 185
186 Profile* profile_; 186 Profile* profile_;
187 scoped_refptr<NetworkActionPredictorDatabase> db_; 187 scoped_refptr<AutocompleteActionPredictorDatabase> db_;
188 content::NotificationRegistrar notification_registrar_; 188 content::NotificationRegistrar notification_registrar_;
189 189
190 // This is cleared after every Omnibox navigation. 190 // This is cleared after every Omnibox navigation.
191 std::vector<TransitionalMatch> transitional_matches_; 191 std::vector<TransitionalMatch> transitional_matches_;
192 192
193 // This allows us to predict the effect of confidence threshold changes on 193 // This allows us to predict the effect of confidence threshold changes on
194 // accuracy. 194 // accuracy.
195 mutable std::vector<std::pair<GURL, double> > tracked_urls_; 195 mutable std::vector<std::pair<GURL, double> > tracked_urls_;
196 196
197 DBCacheMap db_cache_; 197 DBCacheMap db_cache_;
198 DBIdCacheMap db_id_cache_; 198 DBIdCacheMap db_id_cache_;
199 199
200 bool initialized_; 200 bool initialized_;
201 201
202 DISALLOW_COPY_AND_ASSIGN(NetworkActionPredictor); 202 DISALLOW_COPY_AND_ASSIGN(AutocompleteActionPredictor);
203 }; 203 };
204 204
205 #endif // CHROME_BROWSER_AUTOCOMPLETE_NETWORK_ACTION_PREDICTOR_H_ 205 #endif // CHROME_BROWSER_PREDICTORS_AUTOCOMPLETE_ACTION_PREDICTOR_H_
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698