Extracting more than one wallpaper prominent color

ash/shelf/shelf_background_animator.cc

 // 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.
 
 #include "ash/shelf/shelf_background_animator.h"
 
 #include <algorithm>
 
 #include "ash/animation/animation_change_type.h"
+#include "ash/ash_switches.h"
 #include "ash/shelf/shelf.h"
 #include "ash/shelf/shelf_background_animator_observer.h"
 #include "ash/shelf/shelf_constants.h"
 #include "ash/wallpaper/wallpaper_controller.h"
+#include "base/command_line.h"
 #include "ui/gfx/animation/slide_animation.h"
 #include "ui/gfx/color_utils.h"
 
+using ColorProfile = color_utils::ColorProfile;
+using LumaRange = color_utils::LumaRange;
+using SaturationRange = color_utils::SaturationRange;
+
 namespace ash {
 
+namespace {
+
+// Returns color profile used for shelf based on the kAshShelfColorScheme
+// command line arg.
+ColorProfile GetShelfColorProfile() {
+  const std::string switch_value =
+      base::CommandLine::ForCurrentProcess()->GetSwitchValueASCII(
+          switches::kAshShelfColorScheme);
+
+  ColorProfile color_profile(LumaRange::DARK, SaturationRange::MUTED);
+
+  if (switch_value.find("light") != std::string::npos)
+    color_profile.luma = LumaRange::LIGHT;
+  else if (switch_value.find("normal") != std::string::npos)
+    color_profile.luma = LumaRange::NORMAL;
+  else if (switch_value.find("dark") != std::string::npos)
+    color_profile.luma = LumaRange::DARK;
+
+  if (switch_value.find("vibrant") != std::string::npos)
+    color_profile.saturation = SaturationRange::VIBRANT;
+  else if (switch_value.find("muted") != std::string::npos)
+    color_profile.saturation = SaturationRange::MUTED;
+
+  return color_profile;
+}
+
+}  // namespace
+
 ShelfBackgroundAnimator::AnimationValues::AnimationValues() {}
 
 ShelfBackgroundAnimator::AnimationValues::~AnimationValues() {}
 
 void ShelfBackgroundAnimator::AnimationValues::UpdateCurrentValues(double t) {
   current_color_ =
       gfx::Tween::ColorValueBetween(t, initial_color_, target_color_);
 }
 
 void ShelfBackgroundAnimator::AnimationValues::SetTargetValues(
@@ skipping 177 matching lines @@
     case SHELF_BACKGROUND_OVERLAP:
       target_shelf_color_alpha = kShelfTranslucentAlpha;
       target_item_color_alpha = 0;
       break;
     case SHELF_BACKGROUND_MAXIMIZED:
       target_shelf_color_alpha = kMaxAlpha;
       target_item_color_alpha = 0;
       break;
   }
 
+  WallpaperController::ColorProfileIndex index =
+      wallpaper_controller_->GetColorProfileIndex(GetShelfColorProfile());
   SkColor target_color = wallpaper_controller_
-                             ? wallpaper_controller_->prominent_color()
+                             ? wallpaper_controller_->prominent_colors()[index]
                              : kShelfDefaultBaseColor;
   if (target_color == WallpaperController::kInvalidColor) {
     target_color = kShelfDefaultBaseColor;
   } else {
     int darkening_alpha = 0;
 
     switch (background_type) {
       case SHELF_BACKGROUND_DEFAULT:
       case SHELF_BACKGROUND_OVERLAP:
         darkening_alpha = kShelfTranslucentColorDarkenAlpha;
@@ skipping 14 matching lines @@
 
 void ShelfBackgroundAnimator::SetAnimationValues(double t) {
   DCHECK_GE(t, 0.0);
   DCHECK_LE(t, 1.0);
   shelf_background_values_.UpdateCurrentValues(t);
   item_background_values_.UpdateCurrentValues(t);
   NotifyObservers();
 }
 
 }  // namespace ash