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

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

Powered by Google App Engine
This is Rietveld 408576698