| // Copyright 2017 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 COMPONENTS_WALLPAPER_WALLPAPER_COLOR_CALCULATOR_H_ |
| #define COMPONENTS_WALLPAPER_WALLPAPER_COLOR_CALCULATOR_H_ |
| |
| #include "base/macros.h" |
| #include "base/memory/ref_counted.h" |
| #include "base/memory/weak_ptr.h" |
| #include "base/observer_list.h" |
| #include "base/time/time.h" |
| #include "components/wallpaper/wallpaper_export.h" |
| #include "third_party/skia/include/core/SkColor.h" |
| #include "ui/gfx/image/image_skia.h" |
| |
| namespace base { |
| class TaskRunner; |
| } |
| |
| namespace color_utils { |
| struct ColorProfile; |
| } |
| |
| namespace wallpaper { |
| class WallpaperColorCalculatorObserver; |
| |
| // Calculates colors based on a wallpaper image. |
| class WALLPAPER_EXPORT WallpaperColorCalculator { |
| public: |
| // |image|, |color_profiles| are the input parameters to the color calculation |
| // that is executed on the |task_runner|. |
| WallpaperColorCalculator( |
| const gfx::ImageSkia& image, |
| const std::vector<color_utils::ColorProfile>& color_profiles, |
| scoped_refptr<base::TaskRunner> task_runner); |
| ~WallpaperColorCalculator(); |
| |
| void AddObserver(WallpaperColorCalculatorObserver* observer); |
| |
| void RemoveObserver(WallpaperColorCalculatorObserver* observer); |
| |
| // Initiates the calculation and returns false if the calculation fails to be |
| // initiated. Observers may be notified synchronously or asynchronously. |
| // Callers should be aware that this will make |image_| read-only. |
| bool StartCalculation() WARN_UNUSED_RESULT; |
| |
| std::vector<SkColor> prominent_colors() const { return prominent_colors_; } |
| |
| void set_prominent_colors_for_test( |
| const std::vector<SkColor>& prominent_colors) { |
| prominent_colors_ = prominent_colors; |
| } |
| |
| // Explicitly sets the |task_runner_| for testing. |
| void SetTaskRunnerForTest(scoped_refptr<base::TaskRunner> task_runner); |
| |
| private: |
| // Handles asynchronous calculation results. |async_start_time| is used to |
| // record duration metrics. |
| void OnAsyncCalculationComplete(base::TimeTicks async_start_time, |
| const std::vector<SkColor>& prominent_colors); |
| |
| // Notifies observers that a color calulation has completed. Called on the |
| // same thread that constructed |this|. |
| void NotifyCalculationComplete(const std::vector<SkColor>& prominent_colors); |
| |
| // The result of the color calculation. |
| std::vector<SkColor> prominent_colors_; |
| |
| // The image to calculate colors from. |
| gfx::ImageSkia image_; |
| |
| // The color profiles used to calculate colors. |
| std::vector<color_utils::ColorProfile> color_profiles_; |
| |
| // The task runner to run the calculation on. |
| scoped_refptr<base::TaskRunner> task_runner_; |
| |
| base::ObserverList<WallpaperColorCalculatorObserver> observers_; |
| |
| base::WeakPtrFactory<WallpaperColorCalculator> weak_ptr_factory_; |
| |
| DISALLOW_COPY_AND_ASSIGN(WallpaperColorCalculator); |
| }; |
| |
| } // namespace wallpaper |
| |
| #endif // COMPONENTS_WALLPAPER_WALLPAPER_COLOR_CALCULATOR_H_ |