1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 // Because the unit tests for gfx::Image are spread across multiple 6 // implementation files, this header contains the reusable components. 7 8 #include "ui/gfx/image/image_unittest_util.h" 9 10 #include <cmath> 11 12 #include "base/memory/scoped_ptr.h" 13 #include "testing/gtest/include/gtest/gtest.h" 14 #include "third_party/skia/include/core/SkBitmap.h" 15 #include "ui/gfx/codec/png_codec.h" 16 #include "ui/gfx/image/image_skia.h" 17 18 #if defined(OS_IOS) 19 #include "base/mac/foundation_util.h" 20 #include "base/mac/scoped_cftyperef.h" 21 #include "skia/ext/skia_utils_ios.h" 22 #elif defined(OS_MACOSX) 23 #include "base/mac/mac_util.h" 24 #include "skia/ext/skia_utils_mac.h" 25 #endif 26 27 namespace gfx { 28 namespace test { 29 30 namespace { 31 32 bool ColorComponentsClose(SkColor component1, SkColor component2) { 33 int c1 = static_cast<int>(component1); 34 int c2 = static_cast<int>(component2); 35 return std::abs(c1 - c2) <= 40; 36 } 37 38 bool ColorsClose(SkColor color1, SkColor color2) { 39 // Be tolerant of floating point rounding and lossy color space conversions. 40 return ColorComponentsClose(SkColorGetR(color1), SkColorGetR(color2)) && 41 ColorComponentsClose(SkColorGetG(color1), SkColorGetG(color2)) && 42 ColorComponentsClose(SkColorGetB(color1), SkColorGetB(color2)) && 43 ColorComponentsClose(SkColorGetA(color1), SkColorGetA(color2)); 44 } 45 46 } // namespace 47 48 std::vector<float> Get1xAnd2xScales() { 49 std::vector<float> scales; 50 scales.push_back(1.0f); 51 scales.push_back(2.0f); 52 return scales; 53 } 54 55 const SkBitmap CreateBitmap(int width, int height) { 56 SkBitmap bitmap; 57 bitmap.setConfig(SkBitmap::kARGB_8888_Config, width, height); 58 bitmap.allocPixels(); 59 bitmap.eraseARGB(255, 0, 255, 0); 60 return bitmap; 61 } 62 63 gfx::ImageSkia CreateImageSkia(int width, int height) { 64 return gfx::ImageSkia::CreateFrom1xBitmap(CreateBitmap(width, height)); 65 } 66 67 scoped_refptr<base::RefCountedMemory> CreatePNGBytes(int edge_size) { 68 SkBitmap bitmap = CreateBitmap(edge_size, edge_size); 69 scoped_refptr<base::RefCountedBytes> bytes(new base::RefCountedBytes()); 70 PNGCodec::EncodeBGRASkBitmap(bitmap, false, &bytes->data()); 71 return bytes; 72 } 73 74 gfx::Image CreateImage() { 75 return CreateImage(100, 50); 76 } 77 78 gfx::Image CreateImage(int width, int height) { 79 return gfx::Image::CreateFrom1xBitmap(CreateBitmap(width, height)); 80 } 81 82 bool IsEqual(const gfx::Image& img1, const gfx::Image& img2) { 83 img1.AsImageSkia().EnsureRepsForSupportedScales(); 84 img2.AsImageSkia().EnsureRepsForSupportedScales(); 85 std::vector<gfx::ImageSkiaRep> img1_reps = img1.AsImageSkia().image_reps(); 86 gfx::ImageSkia image_skia2 = img2.AsImageSkia(); 87 if (image_skia2.image_reps().size() != img1_reps.size()) 88 return false; 89 90 for (size_t i = 0; i < img1_reps.size(); ++i) { 91 float scale = img1_reps[i].scale(); 92 const gfx::ImageSkiaRep& image_rep2 = image_skia2.GetRepresentation(scale); 93 if (image_rep2.scale() != scale || 94 !IsEqual(img1_reps[i].sk_bitmap(), image_rep2.sk_bitmap())) { 95 return false; 96 } 97 } 98 return true; 99 } 100 101 bool IsEqual(const SkBitmap& bmp1, const SkBitmap& bmp2) { 102 if (bmp1.isNull() && bmp2.isNull()) 103 return true; 104 105 if (bmp1.width() != bmp2.width() || 106 bmp1.height() != bmp2.height() || 107 bmp1.config() != SkBitmap::kARGB_8888_Config || 108 bmp2.config() != SkBitmap::kARGB_8888_Config) { 109 return false; 110 } 111 112 SkAutoLockPixels lock1(bmp1); 113 SkAutoLockPixels lock2(bmp2); 114 if (!bmp1.getPixels() || !bmp2.getPixels()) 115 return false; 116 117 for (int y = 0; y < bmp1.height(); ++y) { 118 for (int x = 0; x < bmp1.width(); ++x) { 119 if (!ColorsClose(bmp1.getColor(x,y), bmp2.getColor(x,y))) 120 return false; 121 } 122 } 123 124 return true; 125 } 126 127 bool IsEqual(const scoped_refptr<base::RefCountedMemory>& bytes, 128 const SkBitmap& bitmap) { 129 SkBitmap decoded; 130 if (!bytes.get() || 131 !PNGCodec::Decode(bytes->front(), bytes->size(), &decoded)) { 132 return bitmap.isNull(); 133 } 134 135 return IsEqual(bitmap, decoded); 136 } 137 138 void CheckImageIndicatesPNGDecodeFailure(const gfx::Image& image) { 139 SkBitmap bitmap = image.AsBitmap(); 140 EXPECT_FALSE(bitmap.isNull()); 141 EXPECT_LE(16, bitmap.width()); 142 EXPECT_LE(16, bitmap.height()); 143 SkAutoLockPixels auto_lock(bitmap); 144 CheckColors(bitmap.getColor(10, 10), SK_ColorRED); 145 } 146 147 bool ImageSkiaStructureMatches( 148 const gfx::ImageSkia& image_skia, 149 int width, 150 int height, 151 const std::vector<float>& scales) { 152 if (image_skia.isNull() || 153 image_skia.width() != width || 154 image_skia.height() != height || 155 image_skia.image_reps().size() != scales.size()) { 156 return false; 157 } 158 159 for (size_t i = 0; i < scales.size(); ++i) { 160 gfx::ImageSkiaRep image_rep = 161 image_skia.GetRepresentation(scales[i]); 162 if (image_rep.is_null() || image_rep.scale() != scales[i]) 163 return false; 164 165 if (image_rep.pixel_width() != static_cast<int>(width * scales[i]) || 166 image_rep.pixel_height() != static_cast<int>(height * scales[i])) { 167 return false; 168 } 169 } 170 return true; 171 } 172 173 bool IsEmpty(const gfx::Image& image) { 174 const SkBitmap& bmp = *image.ToSkBitmap(); 175 return bmp.isNull() || 176 (bmp.width() == 0 && bmp.height() == 0); 177 } 178 179 PlatformImage CreatePlatformImage() { 180 const SkBitmap bitmap(CreateBitmap(25, 25)); 181 #if defined(OS_IOS) 182 float scale = ImageSkia::GetMaxSupportedScale(); 183 184 base::ScopedCFTypeRef<CGColorSpaceRef> color_space( 185 CGColorSpaceCreateDeviceRGB()); 186 UIImage* image = 187 gfx::SkBitmapToUIImageWithColorSpace(bitmap, scale, color_space); 188 base::mac::NSObjectRetain(image); 189 return image; 190 #elif defined(OS_MACOSX) 191 NSImage* image = gfx::SkBitmapToNSImage(bitmap); 192 base::mac::NSObjectRetain(image); 193 return image; 194 #else 195 return gfx::ImageSkia::CreateFrom1xBitmap(bitmap); 196 #endif 197 } 198 199 gfx::Image::RepresentationType GetPlatformRepresentationType() { 200 #if defined(OS_IOS) 201 return gfx::Image::kImageRepCocoaTouch; 202 #elif defined(OS_MACOSX) 203 return gfx::Image::kImageRepCocoa; 204 #else 205 return gfx::Image::kImageRepSkia; 206 #endif 207 } 208 209 PlatformImage ToPlatformType(const gfx::Image& image) { 210 #if defined(OS_IOS) 211 return image.ToUIImage(); 212 #elif defined(OS_MACOSX) 213 return image.ToNSImage(); 214 #else 215 return image.AsImageSkia(); 216 #endif 217 } 218 219 PlatformImage CopyPlatformType(const gfx::Image& image) { 220 #if defined(OS_IOS) 221 return image.CopyUIImage(); 222 #elif defined(OS_MACOSX) 223 return image.CopyNSImage(); 224 #else 225 return image.AsImageSkia(); 226 #endif 227 } 228 229 #if defined(OS_MACOSX) 230 // Defined in image_unittest_util_mac.mm. 231 #else 232 SkColor GetPlatformImageColor(PlatformImage image, int x, int y) { 233 SkBitmap bitmap = *image.bitmap(); 234 SkAutoLockPixels auto_lock(bitmap); 235 return bitmap.getColor(x, y); 236 } 237 #endif 238 239 void CheckColors(SkColor color1, SkColor color2) { 240 EXPECT_TRUE(ColorsClose(color1, color2)); 241 } 242 243 void CheckIsTransparent(SkColor color) { 244 EXPECT_LT(SkColorGetA(color) / 255.0, 0.05); 245 } 246 247 bool IsPlatformImageValid(PlatformImage image) { 248 #if defined(OS_MACOSX) 249 return image != NULL; 250 #else 251 return !image.isNull(); 252 #endif 253 } 254 255 bool PlatformImagesEqual(PlatformImage image1, PlatformImage image2) { 256 #if defined(OS_MACOSX) 257 return image1 == image2; 258 #else 259 return image1.BackedBySameObjectAs(image2); 260 #endif 261 } 262 263 } // namespace test 264 } // namespace gfx 265
