Allow TranslateRanker to override decisions taken by heuristics.

components/translate/core/browser/translate_ranker_impl.h

 // 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_TRANSLATE_RANKER_IMPL_H_
 #define COMPONENTS_TRANSLATE_CORE_BROWSER_TRANSLATE_RANKER_IMPL_H_
 
 #include <memory>
 #include <string>
 
@@ skipping 23 matching lines @@
 
 namespace translate {
 
 class TranslatePrefs;
 
 // Features used to enable ranker query, enforcement and logging. Note that
 // enabling enforcement implies (forces) enabling queries.
 extern const base::Feature kTranslateRankerQuery;
 extern const base::Feature kTranslateRankerEnforcement;
 extern const base::Feature kTranslateRankerLogging;
+extern const base::Feature kTranslateRankerDecisionOverride;
 
 struct TranslateRankerFeatures {
   TranslateRankerFeatures();
 
   TranslateRankerFeatures(int accepted,
                           int denied,
                           int ignored,
                           const std::string& src,
                           const std::string& dst,
                           const std::string& cntry,
@@ skipping 38 matching lines @@
   static base::FilePath GetModelPath(const base::FilePath& data_dir);
 
   // Get the URL from which the download the translate ranker model, by default
   // from Finch.
   static GURL GetModelURL();
 
   // Over-ride the default enabled/disabled state of translate event logging.
   void EnableLogging(bool value);
 
   // TranslateRanker...
-  bool IsLoggingEnabled() override;
-  bool IsQueryEnabled() override;
-  bool IsEnforcementEnabled() override;
-  int GetModelVersion() const override;
-  bool ShouldOfferTranslation(const TranslatePrefs& translate_prefs,
-                              const std::string& src_lang,
-                              const std::string& dst_lang) override;
-  void AddTranslateEvent(const metrics::TranslateEventProto& translate_event,
-                         const GURL& url) override;
+  uint32_t GetModelVersion() const override;
+  bool ShouldOfferTranslation(
+      const TranslatePrefs& translate_prefs,
+      const std::string& src_lang,
+      const std::string& dst_lang,
+      metrics::TranslateEventProto* translate_event) override;
   void FlushTranslateEvents(
       std::vector<metrics::TranslateEventProto>* events) override;
+  void RecordTranslateEvent(
+      int event_type,
+      const GURL& url,
+      metrics::TranslateEventProto* translate_event) override;
+  bool ShouldOverrideDecision(
+      int event_type,
+      const GURL& url,
+      metrics::TranslateEventProto* translate_event) override;
 
   void OnModelAvailable(
       std::unique_ptr<chrome_intelligence::RankerModel> model);
 
   // Calculate the score given to |features| by the |model|.
   double CalculateScore(const TranslateRankerFeatures& features);
 
   // Check if the ModelLoader has been initialized. Used to test ModelLoader
   // logic.
   bool CheckModelLoaderForTesting();
 
  private:
   void SendEventToUKM(const metrics::TranslateEventProto& translate_event,
                       const GURL& url);
 
+  // Caches the translate event.
+  void AddTranslateEvent(const metrics::TranslateEventProto& translate_event,
+                         const GURL& url);
+
   // Used to log URL-keyed metrics. This pointer will outlive |this|.
   ukm::UkmService* ukm_service_;
 
   // Used to sanity check the threading of this ranker.
   base::SequenceChecker sequence_checker_;
 
   // A helper to load the translate ranker model from disk cache or a URL.
   std::unique_ptr<RankerModelLoader> model_loader_;
 
   // The translation ranker model.
   std::unique_ptr<chrome_intelligence::RankerModel> model_;
 
   // Tracks whether or not translate event logging is enabled.
   bool is_logging_enabled_ = true;
 
   // Tracks whether or not translate ranker querying is enabled.
   bool is_query_enabled_ = true;
 
   // Tracks whether or not translate ranker enforcement is enabled. Note that
   // that also enables the code paths for translate ranker querying.
   bool is_enforcement_enabled_ = true;
 
+  // Tracks whether or not translate ranker decision override is enabled. This
+  // will override suppression heuristics and follow ranker's decision. Note
+  // that that also enables the code paths for translate ranker querying and
+  // enforcement.
+  bool is_decision_override_enabled_ = true;
+
   // Saved cache of translate event protos.
   std::vector<metrics::TranslateEventProto> event_cache_;
 
   base::WeakPtrFactory<TranslateRankerImpl> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(TranslateRankerImpl);
 };
 
 }  // namespace translate
 
 std::ostream& operator<<(std::ostream& stream,
                          const translate::TranslateRankerFeatures& features);
 
 #endif  // COMPONENTS_TRANSLATE_CORE_BROWSER_TRANSLATE_RANKER_IMPL_H_