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"
 #include "ash/display/window_tree_host_manager.h"
 #include "ash/public/interfaces/wallpaper.mojom.h"
 #include "ash/session/session_observer.h"
 #include "ash/shell_observer.h"
 #include "base/memory/ref_counted.h"
 #include "base/observer_list.h"
 #include "base/timer/timer.h"
 #include "components/wallpaper/wallpaper_color_calculator_observer.h"
 #include "components/wallpaper/wallpaper_layout.h"
 #include "components/wallpaper/wallpaper_resizer_observer.h"
 #include "mojo/public/cpp/bindings/binding_set.h"
 #include "ui/compositor/compositor_lock.h"
-#include "ui/gfx/color_analysis.h"
-#include "ui/gfx/image/image_skia.h"
 
 namespace base {
 class SequencedTaskRunner;
 }
 
+namespace color_utils {
+struct ColorProfile;
+}
+
 namespace wallpaper {
 class WallpaperColorCalculator;
 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 WindowTreeHostManager::Observer,
       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;
+  // 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 |color_profile|.
+  SkColor GetProminentColor(color_utils::ColorProfile color_profile) 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
+  // 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 std::vector<color_utils::ColorProfile> 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_