Add a new distillability model for the new set of page features

components/dom_distiller/core/distillable_page_detector.cc

 // Copyright 2015 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.
 
 #include "components/dom_distiller/core/distillable_page_detector.h"
 
 #include "base/logging.h"
 #include "grit/components_resources.h"
 #include "ui/base/resource/resource_bundle.h"
 
 namespace dom_distiller {
 
 const DistillablePageDetector* DistillablePageDetector::GetDefault() {
   static DistillablePageDetector* detector = nullptr;
   if (!detector) {
     std::string serialized_proto =
         ResourceBundle::GetSharedInstance()
             .GetRawDataResource(IDR_DISTILLABLE_PAGE_SERIALIZED_MODEL)
             .as_string();
     scoped_ptr<AdaBoostProto> proto(new AdaBoostProto);
     CHECK(proto->ParseFromString(serialized_proto));
     detector = new DistillablePageDetector(proto.Pass());
   }
   return detector;
 }
 
+const DistillablePageDetector* DistillablePageDetector::GetNewModel() {
+  static DistillablePageDetector* detector = nullptr;
+  if (!detector) {
+    std::string serialized_proto =
+        ResourceBundle::GetSharedInstance()
+            .GetRawDataResource(IDR_DISTILLABLE_PAGE_SERIALIZED_MODEL_NEW)
+            .as_string();
+    scoped_ptr<AdaBoostProto> proto(new AdaBoostProto);
+    CHECK(proto->ParseFromString(serialized_proto));
+    detector = new DistillablePageDetector(proto.Pass());
+  }
+  return detector;
+}
+
 DistillablePageDetector::DistillablePageDetector(
     scoped_ptr<AdaBoostProto> proto)
     : proto_(proto.Pass()), threshold_(0.0) {
   CHECK(proto_->num_stumps() == proto_->stump_size());
   for (int i = 0; i < proto_->num_stumps(); ++i) {
     const StumpProto& stump = proto_->stump(i);
     CHECK(stump.feature_number() >= 0);
     CHECK(stump.feature_number() < proto_->num_features());
     threshold_ += stump.weight() / 2.0;
   }
@@ skipping 20 matching lines @@
     }
   }
   return score;
 }
 
 double DistillablePageDetector::GetThreshold() const {
   return threshold_;
 }
 
 }  // namespace dom_distiller