Extracting more than one wallpaper prominent color

ash/wallpaper/wallpaper_controller.h

 // Copyright (c) 2012 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 ASH_WALLPAPER_WALLPAPER_CONTROLLER_H_
 #define ASH_WALLPAPER_WALLPAPER_CONTROLLER_H_
 
 #include <memory>
 
 #include "ash/ash_export.h"
@@ skipping 21 matching lines @@
 class WallpaperResizer;
 }
 
 namespace ash {
 
 class WallpaperControllerObserver;
 
 // Controls the desktop background wallpaper:
 //   - Sets a wallpaper image and layout;
 //   - Handles display change (add/remove display, configuration change etc);
-//   - Calculates prominent color for shelf;
+//   - Calculates prominent colors.
 //   - Move wallpaper to locked container(s) when session state is not ACTIVE to
 //     hide the user desktop and move it to unlocked container when session
 //     state is ACTIVE;
 class ASH_EXPORT WallpaperController
     : public NON_EXPORTED_BASE(mojom::WallpaperController),
       public WmDisplayObserver,
       public ShellObserver,
       public wallpaper::WallpaperResizerObserver,
       public wallpaper::WallpaperColorCalculatorObserver,
       public SessionObserver,
       public NON_EXPORTED_BASE(ui::CompositorLockClient) {
  public:
   enum WallpaperMode { WALLPAPER_NONE, WALLPAPER_IMAGE };
 
-  // The value assigned to |prominent_color_| if extraction fails or the feature
-  // is disabled (e.g. command line, lock/login screens).
-  static constexpr SkColor kInvalidColor = SK_ColorTRANSPARENT;
+  // This enum is used to get the corresponding prominent color from the results
+  // of wallpaper color calculator.
+  enum ColorProfileIndex {
+    COLOR_PROFILE_INDEX_DARK_VIBRANT = 0,
+    COLOR_PROFILE_INDEX_NORMAL_VIBRANT,
+    COLOR_PROFILE_INDEX_LIGHT_VIBRANT,
+    COLOR_PROFILE_INDEX_DARK_MUTED,
+    COLOR_PROFILE_INDEX_NORMAL_MUTED,
+    COLOR_PROFILE_INDEX_LIGHT_MUTED,
+  };
+
+  // The value assigned if extraction fails or the feature is disabled (e.g.
+  // command line, lock/login screens).
+  static const SkColor kInvalidColor;
 
   WallpaperController();
   ~WallpaperController() override;
 
   // Binds the mojom::WallpaperController interface request to this object.
   void BindRequest(mojom::WallpaperControllerRequest request);
 
   // Add/Remove observers.
   void AddObserver(WallpaperControllerObserver* observer);
   void RemoveObserver(WallpaperControllerObserver* observer);
 
-  SkColor prominent_color() const { return prominent_color_; }
+  // Returns the prominent color based on the given |color_profile|.
+  SkColor prominent_color(ColorProfileIndex color_profile_index) const {
+    return prominent_colors_[color_profile_index];
+  }
+
+  // Returns the color profile index used for shelf.
+  WallpaperController::ColorProfileIndex GetShelfColorProfileIndex() const;
 
   // Provides current image on the wallpaper, or empty gfx::ImageSkia if there
   // is no image, e.g. wallpaper is none.
   gfx::ImageSkia GetWallpaper() const;
   uint32_t GetWallpaperOriginalImageId() const;
 
   wallpaper::WallpaperLayout GetWallpaperLayout() const;
 
   // Sets the wallpaper and alerts observers of changes.
   void SetWallpaperImage(const gfx::ImageSkia& image,
@@ skipping 63 matching lines @@
   // Returns true if a wallpaper moved.
   bool ReparentWallpaper(int container);
 
   // Returns the wallpaper container id for unlocked and locked states.
   int GetWallpaperContainerId(bool locked);
 
   // Reload the wallpaper. |clear_cache| specifies whether to clear the
   // wallpaper cahce or not.
   void UpdateWallpaper(bool clear_cache);
 
-  // Sets |prominent_color_| and notifies the observers if there is a change.
-  void SetProminentColor(SkColor color);
+  // Sets |prominent_colors_| and notifies the observers if there is a change.
+  void SetProminentColors(const std::vector<SkColor>& prominent_colors);
 
-  // Calculates a prominent color based on the wallpaper image and notifies
+  // Returns color profile index based on |color_profile|.
+  WallpaperController::ColorProfileIndex GetColorProfileIndex(
+      color_utils::ColorProfile color_profile) const;
+
+  // Calculates prominent colors based on the wallpaper image and notifies
   // |observers_| of the value, either synchronously or asynchronously. In some
   // cases the wallpaper image will not actually be processed (e.g. user isn't
   // logged in, feature isn't enabled).
   // If an existing calculation is in progress it is destroyed.
   void CalculateWallpaperColors();
 
   // Returns false when the color extraction algorithm shouldn't be run based on
   // system state (e.g. wallpaper image, SessionState, etc.).
   bool ShouldCalculateColors() const;
 
@@ skipping 21 matching lines @@
   // Bindings for the WallpaperController interface.
   mojo::BindingSet<mojom::WallpaperController> bindings_;
 
   base::ObserverList<WallpaperControllerObserver> observers_;
 
   std::unique_ptr<wallpaper::WallpaperResizer> current_wallpaper_;
 
   // Asynchronous task to extract colors from the wallpaper.
   std::unique_ptr<wallpaper::WallpaperColorCalculator> color_calculator_;
 
-  // The prominent color extracted from the current wallpaper.
+  // The prominent colors extracted from the current wallpaper.
   // kInvalidColor is used by default or if extracting colors fails.
-  SkColor prominent_color_;
+  std::vector<SkColor> prominent_colors_;
+
+  // Caches the color profiles that need to do wallpaper color extracting.
+  const color_utils::ColorProfiles color_profiles_;
 
   gfx::Size current_max_display_size_;
 
   base::OneShotTimer timer_;
 
   int wallpaper_reload_delay_;
 
   scoped_refptr<base::SequencedTaskRunner> sequenced_task_runner_;
 
   ScopedSessionObserver scoped_session_observer_;
 
   std::unique_ptr<ui::CompositorLock> compositor_lock_;
 
   DISALLOW_COPY_AND_ASSIGN(WallpaperController);
 };
 
 }  // namespace ash
 
 #endif  // ASH_WALLPAPER_WALLPAPER_CONTROLLER_H_