Extracting more than one wallpaper prominent color

ui/gfx/color_analysis_unittest.cc

Index: ui/gfx/color_analysis_unittest.cc
diff --git a/ui/gfx/color_analysis_unittest.cc b/ui/gfx/color_analysis_unittest.cc
index 93fc2db63bc4bccc3ae67a12db9c93aa26ae8a02..488700e3f03da5abc4d8e5b338a217ddf5028f7c 100644
--- a/ui/gfx/color_analysis_unittest.cc
+++ b/ui/gfx/color_analysis_unittest.cc
@@ -497,16 +497,15 @@ TEST_F(ColorAnalysisTest, ComputePrincipalComponentImage) {
   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};
+  std::vector<ColorProfile> color_profiles;
+  for (auto s : saturations) {
+    for (auto l : lumas)
+      color_profiles.emplace_back(l, s);
+  }
 
   // A totally dark gray image, which yields no prominent color as it's too
   // close to black.
@@ -515,12 +514,10 @@ TEST_F(ColorAnalysisTest, ComputeProminentColor) {
   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);
@@ -528,22 +525,16 @@ TEST_F(ColorAnalysisTest, ComputeProminentColor) {
   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.
@@ -551,11 +542,8 @@ TEST_F(ColorAnalysisTest, ComputeProminentColor) {
   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