Optimize variation field trial evaluation.

components/variations/processed_study.cc

 // Copyright 2013 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/variations/processed_study.h"
 
 #include <set>
 #include <string>
 
 #include "base/version.h"
 #include "components/variations/proto/study.pb.h"
 
 namespace variations {
 
 namespace {
 
-// Validates the sanity of |study| and computes the total probability.
+// Validates the sanity of |study| and computes the total probability and
+// whether all assignments are to a single group.
 bool ValidateStudyAndComputeTotalProbability(
     const Study& study,
-    base::FieldTrial::Probability* total_probability) {
+    base::FieldTrial::Probability* total_probability,
+    bool* all_assignments_to_one_group) {
   // At the moment, a missing default_experiment_name makes the study invalid.
   if (study.default_experiment_name().empty()) {
     DVLOG(1) << study.name() << " has no default experiment defined.";
     return false;
   }
   if (study.filter().has_min_version() &&
       !Version::IsValidWildcardString(study.filter().min_version())) {
     DVLOG(1) << study.name() << " has invalid min version: "
              << study.filter().min_version();
     return false;
   }
   if (study.filter().has_max_version() &&
       !Version::IsValidWildcardString(study.filter().max_version())) {
     DVLOG(1) << study.name() << " has invalid max version: "
              << study.filter().max_version();
     return false;
   }
 
   const std::string& default_group_name = study.default_experiment_name();
   base::FieldTrial::Probability divisor = 0;
 
+  bool multiple_assigned_groups = false;
   bool found_default_group = false;
   std::set<std::string> experiment_names;
   for (int i = 0; i < study.experiment_size(); ++i) {
-    if (study.experiment(i).name().empty()) {
+    const Study_Experiment& experiment = study.experiment(i);
+    if (experiment.name().empty()) {
       DVLOG(1) << study.name() << " is missing experiment " << i << " name";
       return false;
     }
-    if (!experiment_names.insert(study.experiment(i).name()).second) {
+    if (!experiment_names.insert(experiment.name()).second) {
       DVLOG(1) << study.name() << " has a repeated experiment name "
                << study.experiment(i).name();
       return false;
     }
 
-    if (!study.experiment(i).has_forcing_flag())
-      divisor += study.experiment(i).probability_weight();
+    if (!experiment.has_forcing_flag() && experiment.probability_weight() > 0) {
+      // If |divisor| is not 0, there was at least one prior non-zero group.
+      if (divisor != 0)
+        multiple_assigned_groups = true;
+      divisor += experiment.probability_weight();
+    }
     if (study.experiment(i).name() == default_group_name)
       found_default_group = true;
   }
 
   if (!found_default_group) {
     DVLOG(1) << study.name() << " is missing default experiment in its "
              << "experiment list";
     // The default group was not found in the list of groups. This study is not
     // valid.
     return false;
   }
 
   *total_probability = divisor;
+  *all_assignments_to_one_group = !multiple_assigned_groups;
   return true;
 }
 
 
 }  // namespace
 
-
 ProcessedStudy::ProcessedStudy()
-    : study_(NULL), total_probability_(0), is_expired_(false) {
+    : study_(NULL),
+      total_probability_(0),
+      all_assignments_to_one_group_(false),
+      is_expired_(false) {
 }
 
 ProcessedStudy::~ProcessedStudy() {
 }
 
 bool ProcessedStudy::Init(const Study* study, bool is_expired) {
   base::FieldTrial::Probability total_probability = 0;
-  if (!ValidateStudyAndComputeTotalProbability(*study, &total_probability))
+  bool all_assignments_to_one_group = false;
+  if (!ValidateStudyAndComputeTotalProbability(*study, &total_probability,
+                                               &all_assignments_to_one_group)) {
     return false;
+  }
 
   study_ = study;
   is_expired_ = is_expired;
   total_probability_ = total_probability;
+  all_assignments_to_one_group_ = all_assignments_to_one_group;
   return true;
 }
 
 int ProcessedStudy::GetExperimentIndexByName(const std::string& name) const {
   for (int i = 0; i < study_->experiment_size(); ++i) {
     if (study_->experiment(i).name() == name)
       return i;
   }
 
   return -1;
 }
 
 // static
 bool ProcessedStudy::ValidateAndAppendStudy(
     const Study* study,
     bool is_expired,
     std::vector<ProcessedStudy>* processed_studies) {
   ProcessedStudy processed_study;
   if (processed_study.Init(study, is_expired)) {
     processed_studies->push_back(processed_study);
     return true;
   }
   return false;
 }
 
 }  // namespace variations