Extracting more than one wallpaper prominent color

ui/gfx/color_analysis_unittest.cc

 // 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.
 
 #include "ui/gfx/color_analysis.h"
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include <vector>
@@ skipping 479 matching lines @@
   uint8_t max_gl = 0;
   Calculate8bitBitmapMinMax(result, &min_gl, &max_gl);
 
   EXPECT_EQ(0, min_gl);
   EXPECT_EQ(255, max_gl);
   EXPECT_EQ(min_gl, SkColorGetA(result.getColor(0, 0)));
   EXPECT_EQ(max_gl, SkColorGetA(result.getColor(299, 199)));
   EXPECT_EQ(93U, SkColorGetA(result.getColor(150, 0)));
 }
 
-TEST_F(ColorAnalysisTest, ComputeProminentColor) {
-  struct {
-    LumaRange luma;
-    SaturationRange saturation;
-  } color_profiles[] = {{LumaRange::DARK, SaturationRange::VIBRANT},
-                        {LumaRange::NORMAL, SaturationRange::VIBRANT},
-                        {LumaRange::LIGHT, SaturationRange::VIBRANT},
-                        {LumaRange::DARK, SaturationRange::MUTED},
-                        {LumaRange::NORMAL, SaturationRange::MUTED},
-                        {LumaRange::LIGHT, SaturationRange::MUTED}};
+TEST_F(ColorAnalysisTest, ComputeProminentColors) {
+  LumaRange lumas[] = {LumaRange::DARK, LumaRange::NORMAL, LumaRange::LIGHT};
+  SaturationRange saturations[] = {SaturationRange::VIBRANT,
+                                   SaturationRange::MUTED};
+  ColorProfiles color_profiles;
+  for (auto s : saturations) {
+    for (auto l : lumas)
+      color_profiles.push_back(ColorProfile(l, s));
+  }
 
   // A totally dark gray image, which yields no prominent color as it's too
   // close to black.
   gfx::Canvas canvas(gfx::Size(300, 200), 1.0f, true);
   canvas.FillRect(gfx::Rect(0, 0, 300, 200), SkColorSetRGB(10, 10, 10));
   SkBitmap bitmap = canvas.GetBitmap();
 
   // All expectations start at SK_ColorTRANSPARENT (i.e. 0).
-  SkColor expectations[arraysize(color_profiles)] = {};
-  for (size_t i = 0; i < arraysize(color_profiles); ++i) {
-    EXPECT_EQ(expectations[i],
-              CalculateProminentColorOfBitmap(bitmap, color_profiles[i].luma,
-                                              color_profiles[i].saturation));
-  }
+  std::vector<SkColor> expectations(color_profiles.size(), 0);
+  std::vector<SkColor> computations =
+      CalculateProminentColorsOfBitmap(bitmap, color_profiles);
+  EXPECT_EQ(expectations, computations);
 
   // Add a green that could hit a couple values.
   const SkColor kVibrantGreen = SkColorSetRGB(25, 200, 25);
   canvas.FillRect(gfx::Rect(0, 1, 300, 1), kVibrantGreen);
   bitmap = canvas.GetBitmap();
   expectations[0] = kVibrantGreen;
   expectations[1] = kVibrantGreen;
-  for (size_t i = 0; i < arraysize(color_profiles); ++i) {
-    EXPECT_EQ(expectations[i],
-              CalculateProminentColorOfBitmap(bitmap, color_profiles[i].luma,
-                                              color_profiles[i].saturation));
-  }
+  computations = CalculateProminentColorsOfBitmap(bitmap, color_profiles);
+  EXPECT_EQ(expectations, computations);
 
   // Add a stripe of a dark, muted green (saturation .33, luma .29).
   const SkColor kDarkGreen = SkColorSetRGB(50, 100, 50);
   canvas.FillRect(gfx::Rect(0, 2, 300, 1), kDarkGreen);
   bitmap = canvas.GetBitmap();
   expectations[3] = kDarkGreen;
-  for (size_t i = 0; i < arraysize(color_profiles); ++i) {
-    EXPECT_EQ(expectations[i],
-              CalculateProminentColorOfBitmap(bitmap, color_profiles[i].luma,
-                                              color_profiles[i].saturation));
-  }
+  computations = CalculateProminentColorsOfBitmap(bitmap, color_profiles);
+  EXPECT_EQ(expectations, computations);
 
   // Now draw a little bit of pure green. That should be closer to the goal for
   // normal vibrant, but is out of range for other color profiles.
   const SkColor kPureGreen = SkColorSetRGB(0, 255, 0);
   canvas.FillRect(gfx::Rect(0, 3, 300, 1), kPureGreen);
   bitmap = canvas.GetBitmap();
   expectations[1] = kPureGreen;
-  for (size_t i = 0; i < arraysize(color_profiles); ++i) {
-    EXPECT_EQ(expectations[i],
-              CalculateProminentColorOfBitmap(bitmap, color_profiles[i].luma,
-                                              color_profiles[i].saturation));
-  }
+  computations = CalculateProminentColorsOfBitmap(bitmap, color_profiles);
+  EXPECT_EQ(expectations, computations);
 }
 
 }  // namespace color_utils