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

Unified Diff: components/translate/core/browser/ranker_model_loader.h

Issue 2565873002: [translate] Add translate ranker model loader. (Closed)
Patch Set: for asan testing only - DO NOT COMMIT Created 3 years, 10 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 side-by-side diff with in-line comments
Download patch
Index: components/translate/core/browser/ranker_model_loader.h
diff --git a/components/translate/core/browser/ranker_model_loader.h b/components/translate/core/browser/ranker_model_loader.h
new file mode 100644
index 0000000000000000000000000000000000000000..f20e26573efb185b8ef1f8326e67dac9bb4cc03e
--- /dev/null
+++ b/components/translate/core/browser/ranker_model_loader.h
@@ -0,0 +1,100 @@
+// 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_TRANSLATE_CORE_BROWSER_RANKER_MODEL_LOADER_H_
+#define COMPONENTS_TRANSLATE_CORE_BROWSER_RANKER_MODEL_LOADER_H_
+
+#include <memory>
+#include <string>
+
+#include "base/files/file_path.h"
+#include "base/memory/ref_counted.h"
+#include "base/sequence_checker.h"
+#include "base/threading/thread.h"
+#include "url/gurl.h"
+
+namespace chrome_intelligence {
+class RankerModel;
+} // namespace chrome_intelligence
+
+namespace translate {
+
+// Enumeration denoting the outcome of an attempt to download the model. This
+// must be kept in sync with the TranslateRankerModelStatus enum in
+// histograms.xml
+// TODO(rogerm): rename the enum in histograms.xml to be more generic
groby-ooo-7-16 2017/02/14 20:56:48 Can you do that as part of this CL?
Roger McFarlane (Chromium) 2017/02/21 22:16:51 Done.
+enum class RankerModelStatus : int {
+ OK = 0,
+ DOWNLOAD_THROTTLED = 1,
+ DOWNLOAD_FAILED = 2,
+ PARSE_FAILED = 3,
+ VALIDATION_FAILED = 4,
+ INCOMPATIBLE = 5,
+
+ // Insert new values above this line.
+ MAX
+};
+
+// If enabled, downloads a translate ranker model and uses it to determine
+// whether the user should be given a translation prompt or not.
+class RankerModelLoader {
+ public:
+ // Callback to validate a ranker model on behalf of the model loader client.
+ // For example, the callback might validate that the model is compatible with
+ // the features generated when ranking translation offerings. This callback
+ // may be called on any sequence and must, therefore, be thread-safe.
fdoray 2017/02/15 20:07:40 + // This callback may be invoked after RankerMode
Roger McFarlane (Chromium) 2017/02/21 22:16:50 Done.
+ using ValidateCallback = base::Callback<RankerModelStatus(
hamelphi 2017/02/13 19:18:58 s\ValidateCallback\ValidateModelCallback ? I thin
Roger McFarlane (Chromium) 2017/02/21 22:16:50 Done.
+ const chrome_intelligence::RankerModel& model)>;
+
+ // Called to transfer ownership of a loaded model back to the model loader
+ // client. This will be called on the sequence on which the model loader was
groby-ooo-7-16 2017/02/14 20:56:48 Since we're having a set of callbacks here, does i
fdoray 2017/02/15 20:07:40 Since ValidateCallback can be invoked after the Ra
Roger McFarlane (Chromium) 2017/02/21 22:16:50 Acknowledged.
Roger McFarlane (Chromium) 2017/02/21 22:16:51 fdoray@ suggested to revert to callbacks (instead
+ // constructed.
+ using OnModelAvailableCallback = base::Callback<void(
+ std::unique_ptr<chrome_intelligence::RankerModel> model)>;
+
+ // |valildate_callback| may be called on any sequence; it must, therefore,
groby-ooo-7-16 2017/02/14 20:56:48 s/valildate/validate
Roger McFarlane (Chromium) 2017/02/21 22:16:50 Done.
+ // be thread-safe.
+ //
+ // |on_model_available_callback| will be called on the sequence on which the
+ // ranker model loader is constructed.
+ //
+ // |model_path| denotes the file path
groby-ooo-7-16 2017/02/14 20:56:48 nit: end with period. More importantly: This prob
Roger McFarlane (Chromium) 2017/02/21 22:16:50 Done.
+ //
+ // |uma_prefix| will be used to start all UMA metrics genereted by this
hamelphi 2017/02/13 19:18:58 generated
hamelphi 2017/02/13 19:18:58 Nit. I was confused a bit by "used to start..." Ma
Roger McFarlane (Chromium) 2017/02/21 22:16:50 Done.
Roger McFarlane (Chromium) 2017/02/21 22:16:51 Done.
+ // loader.
+ RankerModelLoader(const ValidateCallback& validate_callback,
+ const OnModelAvailableCallback& on_model_available_callback,
+ const base::FilePath& model_path,
+ const GURL& model_url,
+ const std::string& uma_prefix);
+
+ ~RankerModelLoader();
+
+ // Asynchronously initiates loading the model from the cache file path and URL
+ // previously configured.
groby-ooo-7-16 2017/02/14 20:56:48 Why not pass path and URL here, then?
Roger McFarlane (Chromium) 2017/02/21 22:16:50 The loader is stateful. Passing the path and url h
+ void Start();
+
+ // Call this method periodically to notify the model loader the network might
+ // be available. If a model download is pending, this will trigger (subject to
groby-ooo-7-16 2017/02/14 20:56:48 That's probably not the right name if we call this
Roger McFarlane (Chromium) 2017/02/21 22:16:50 Renamed to NotifyOfRankerActivity
+ // retry and frequency limits) a download attempt.
+ void NotifyOfNetworkAvailability();
+
+ private:
+ // The model loader backend to which the actual loading functionality is
+ // delegated.
+ class Backend;
groby-ooo-7-16 2017/02/14 20:56:48 Why do we have a separate backend?
fdoray 2017/02/15 20:07:40 The Backend performs asynchronous operations and c
Roger McFarlane (Chromium) 2017/02/21 22:16:50 Done.
Roger McFarlane (Chromium) 2017/02/21 22:16:51 The threading and task-scheduling folks recommende
+
+ // The task runner on which backend tasks are performed.
+ const scoped_refptr<base::SequencedTaskRunner> backend_task_runner_;
+
+ // The model loader backend to which the actual loading functionality is
+ // delegated.
fdoray 2017/02/15 20:07:40 + // When the RankerModelLoader is destroyed, owne
Roger McFarlane (Chromium) 2017/02/21 22:16:50 Done.
+ std::unique_ptr<Backend> backend_;
+
+ DISALLOW_COPY_AND_ASSIGN(RankerModelLoader);
+};
+
+} // namespace translate
+
+#endif // COMPONENTS_TRANSLATE_CORE_BROWSER_RANKER_MODEL_LOADER_H_

Powered by Google App Engine
This is Rietveld 408576698