1 /*
2  * Copyright 2011 Google Inc.
3  *
4  * Use of this source code is governed by a BSD-style license that can be
5  * found in the LICENSE file.
6  */
7 
8 #include "SkBitmap.h"
9 #include "SkRect.h"
10 #include "SkTemplates.h"
11 #include "Test.h"
12 
boolStr(bool value)13 static const char* boolStr(bool value) {
14     return value ? "true" : "false";
15 }
16 
17 // these are in the same order as the SkColorType enum
18 static const char* gColorTypeName[] = {
19     "None", "A8", "565", "4444", "RGBA", "BGRA", "Index8"
20 };
21 
report_opaqueness(skiatest::Reporter * reporter,const SkBitmap & src,const SkBitmap & dst)22 static void report_opaqueness(skiatest::Reporter* reporter, const SkBitmap& src,
23                               const SkBitmap& dst) {
24     ERRORF(reporter, "src %s opaque:%d, dst %s opaque:%d",
25            gColorTypeName[src.colorType()], src.isOpaque(),
26            gColorTypeName[dst.colorType()], dst.isOpaque());
27 }
28 
canHaveAlpha(SkColorType ct)29 static bool canHaveAlpha(SkColorType ct) {
30     return kRGB_565_SkColorType != ct;
31 }
32 
33 // copyTo() should preserve isOpaque when it makes sense
test_isOpaque(skiatest::Reporter * reporter,const SkBitmap & srcOpaque,const SkBitmap & srcPremul,SkColorType dstColorType)34 static void test_isOpaque(skiatest::Reporter* reporter,
35                           const SkBitmap& srcOpaque, const SkBitmap& srcPremul,
36                           SkColorType dstColorType) {
37     SkBitmap dst;
38 
39     if (canHaveAlpha(srcPremul.colorType()) && canHaveAlpha(dstColorType)) {
40         REPORTER_ASSERT(reporter, srcPremul.copyTo(&dst, dstColorType));
41         REPORTER_ASSERT(reporter, dst.colorType() == dstColorType);
42         if (srcPremul.isOpaque() != dst.isOpaque()) {
43             report_opaqueness(reporter, srcPremul, dst);
44         }
45     }
46 
47     REPORTER_ASSERT(reporter, srcOpaque.copyTo(&dst, dstColorType));
48     REPORTER_ASSERT(reporter, dst.colorType() == dstColorType);
49     if (srcOpaque.isOpaque() != dst.isOpaque()) {
50         report_opaqueness(reporter, srcOpaque, dst);
51     }
52 }
53 
init_src(const SkBitmap & bitmap)54 static void init_src(const SkBitmap& bitmap) {
55     SkAutoLockPixels lock(bitmap);
56     if (bitmap.getPixels()) {
57         if (bitmap.getColorTable()) {
58             sk_bzero(bitmap.getPixels(), bitmap.getSize());
59         } else {
60             bitmap.eraseColor(SK_ColorWHITE);
61         }
62     }
63 }
64 
init_ctable()65 static SkColorTable* init_ctable() {
66     static const SkColor colors[] = {
67         SK_ColorBLACK, SK_ColorRED, SK_ColorGREEN, SK_ColorBLUE, SK_ColorWHITE
68     };
69     return new SkColorTable(colors, SK_ARRAY_COUNT(colors));
70 }
71 
72 struct Pair {
73     SkColorType fColorType;
74     const char* fValid;
75 };
76 
77 // Utility functions for copyPixelsTo()/copyPixelsFrom() tests.
78 // getPixel()
79 // setPixel()
80 // getSkConfigName()
81 // struct Coordinates
82 // reportCopyVerification()
83 // writeCoordPixels()
84 
85 // Utility function to read the value of a given pixel in bm. All
86 // values converted to uint32_t for simplification of comparisons.
getPixel(int x,int y,const SkBitmap & bm)87 static uint32_t getPixel(int x, int y, const SkBitmap& bm) {
88     uint32_t val = 0;
89     uint16_t val16;
90     uint8_t val8;
91     SkAutoLockPixels lock(bm);
92     const void* rawAddr = bm.getAddr(x,y);
93 
94     switch (bm.bytesPerPixel()) {
95         case 4:
96             memcpy(&val, rawAddr, sizeof(uint32_t));
97             break;
98         case 2:
99             memcpy(&val16, rawAddr, sizeof(uint16_t));
100             val = val16;
101             break;
102         case 1:
103             memcpy(&val8, rawAddr, sizeof(uint8_t));
104             val = val8;
105             break;
106         default:
107             break;
108     }
109     return val;
110 }
111 
112 // Utility function to set value of any pixel in bm.
113 // bm.getConfig() specifies what format 'val' must be
114 // converted to, but at present uint32_t can handle all formats.
setPixel(int x,int y,uint32_t val,SkBitmap & bm)115 static void setPixel(int x, int y, uint32_t val, SkBitmap& bm) {
116     uint16_t val16;
117     uint8_t val8;
118     SkAutoLockPixels lock(bm);
119     void* rawAddr = bm.getAddr(x,y);
120 
121     switch (bm.bytesPerPixel()) {
122         case 4:
123             memcpy(rawAddr, &val, sizeof(uint32_t));
124             break;
125         case 2:
126             val16 = val & 0xFFFF;
127             memcpy(rawAddr, &val16, sizeof(uint16_t));
128             break;
129         case 1:
130             val8 = val & 0xFF;
131             memcpy(rawAddr, &val8, sizeof(uint8_t));
132             break;
133         default:
134             // Ignore.
135             break;
136     }
137 }
138 
139 // Helper struct to contain pixel locations, while avoiding need for STL.
140 struct Coordinates {
141 
142     const int length;
143     SkIPoint* const data;
144 
CoordinatesCoordinates145     explicit Coordinates(int _length): length(_length)
146                                      , data(new SkIPoint[length]) { }
147 
~CoordinatesCoordinates148     ~Coordinates(){
149         delete [] data;
150     }
151 
operator []Coordinates152     SkIPoint* operator[](int i) const {
153         // Use with care, no bounds checking.
154         return data + i;
155     }
156 };
157 
158 // A function to verify that two bitmaps contain the same pixel values
159 // at all coordinates indicated by coords. Simplifies verification of
160 // copied bitmaps.
reportCopyVerification(const SkBitmap & bm1,const SkBitmap & bm2,Coordinates & coords,const char * msg,skiatest::Reporter * reporter)161 static void reportCopyVerification(const SkBitmap& bm1, const SkBitmap& bm2,
162                             Coordinates& coords,
163                             const char* msg,
164                             skiatest::Reporter* reporter){
165     // Confirm all pixels in the list match.
166     for (int i = 0; i < coords.length; ++i) {
167         uint32_t p1 = getPixel(coords[i]->fX, coords[i]->fY, bm1);
168         uint32_t p2 = getPixel(coords[i]->fX, coords[i]->fY, bm2);
169 //        SkDebugf("[%d] (%d %d) p1=%x p2=%x\n", i, coords[i]->fX, coords[i]->fY, p1, p2);
170         if (p1 != p2) {
171             ERRORF(reporter, "%s [colortype = %s]", msg, gColorTypeName[bm1.colorType()]);
172             break;
173         }
174     }
175 }
176 
177 // Writes unique pixel values at locations specified by coords.
writeCoordPixels(SkBitmap & bm,const Coordinates & coords)178 static void writeCoordPixels(SkBitmap& bm, const Coordinates& coords) {
179     for (int i = 0; i < coords.length; ++i)
180         setPixel(coords[i]->fX, coords[i]->fY, i, bm);
181 }
182 
183 static const Pair gPairs[] = {
184     { kUnknown_SkColorType,     "000000"  },
185     { kAlpha_8_SkColorType,     "010101"  },
186     { kIndex_8_SkColorType,     "011111"  },
187     { kRGB_565_SkColorType,     "010101"  },
188     { kARGB_4444_SkColorType,   "010111"  },
189     { kN32_SkColorType,         "010111"  },
190 };
191 
192 static const int W = 20;
193 static const int H = 33;
194 
setup_src_bitmaps(SkBitmap * srcOpaque,SkBitmap * srcPremul,SkColorType ct)195 static void setup_src_bitmaps(SkBitmap* srcOpaque, SkBitmap* srcPremul,
196                               SkColorType ct) {
197     SkColorTable* ctable = nullptr;
198     if (kIndex_8_SkColorType == ct) {
199         ctable = init_ctable();
200     }
201 
202     srcOpaque->allocPixels(SkImageInfo::Make(W, H, ct, kOpaque_SkAlphaType),
203                            nullptr, ctable);
204     srcPremul->allocPixels(SkImageInfo::Make(W, H, ct, kPremul_SkAlphaType),
205                            nullptr, ctable);
206     SkSafeUnref(ctable);
207     init_src(*srcOpaque);
208     init_src(*srcPremul);
209 }
210 
DEF_TEST(BitmapCopy_extractSubset,reporter)211 DEF_TEST(BitmapCopy_extractSubset, reporter) {
212     for (size_t i = 0; i < SK_ARRAY_COUNT(gPairs); i++) {
213         SkBitmap srcOpaque, srcPremul;
214         setup_src_bitmaps(&srcOpaque, &srcPremul, gPairs[i].fColorType);
215 
216         SkBitmap bitmap(srcOpaque);
217         SkBitmap subset;
218         SkIRect r;
219         // Extract a subset which has the same width as the original. This
220         // catches a bug where we cloned the genID incorrectly.
221         r.set(0, 1, W, 3);
222         bitmap.setIsVolatile(true);
223         // Relies on old behavior of extractSubset failing if colortype is unknown
224         if (kUnknown_SkColorType != bitmap.colorType() && bitmap.extractSubset(&subset, r)) {
225             REPORTER_ASSERT(reporter, subset.width() == W);
226             REPORTER_ASSERT(reporter, subset.height() == 2);
227             REPORTER_ASSERT(reporter, subset.alphaType() == bitmap.alphaType());
228             REPORTER_ASSERT(reporter, subset.isVolatile() == true);
229 
230             // Test copying an extracted subset.
231             for (size_t j = 0; j < SK_ARRAY_COUNT(gPairs); j++) {
232                 SkBitmap copy;
233                 bool success = subset.copyTo(&copy, gPairs[j].fColorType);
234                 if (!success) {
235                     // Skip checking that success matches fValid, which is redundant
236                     // with the code below.
237                     REPORTER_ASSERT(reporter, gPairs[i].fColorType != gPairs[j].fColorType);
238                     continue;
239                 }
240 
241                 // When performing a copy of an extracted subset, the gen id should
242                 // change.
243                 REPORTER_ASSERT(reporter, copy.getGenerationID() != subset.getGenerationID());
244 
245                 REPORTER_ASSERT(reporter, copy.width() == W);
246                 REPORTER_ASSERT(reporter, copy.height() == 2);
247 
248                 if (gPairs[i].fColorType == gPairs[j].fColorType) {
249                     SkAutoLockPixels alp0(subset);
250                     SkAutoLockPixels alp1(copy);
251                     // they should both have, or both not-have, a colortable
252                     bool hasCT = subset.getColorTable() != nullptr;
253                     REPORTER_ASSERT(reporter, (copy.getColorTable() != nullptr) == hasCT);
254                 }
255             }
256         }
257 
258         bitmap = srcPremul;
259         bitmap.setIsVolatile(false);
260         if (bitmap.extractSubset(&subset, r)) {
261             REPORTER_ASSERT(reporter, subset.alphaType() == bitmap.alphaType());
262             REPORTER_ASSERT(reporter, subset.isVolatile() == false);
263         }
264     }
265 }
266 
DEF_TEST(BitmapCopy,reporter)267 DEF_TEST(BitmapCopy, reporter) {
268     static const bool isExtracted[] = {
269         false, true
270     };
271 
272     for (size_t i = 0; i < SK_ARRAY_COUNT(gPairs); i++) {
273         SkBitmap srcOpaque, srcPremul;
274         setup_src_bitmaps(&srcOpaque, &srcPremul, gPairs[i].fColorType);
275 
276         for (size_t j = 0; j < SK_ARRAY_COUNT(gPairs); j++) {
277             SkBitmap dst;
278 
279             bool success = srcPremul.copyTo(&dst, gPairs[j].fColorType);
280             bool expected = gPairs[i].fValid[j] != '0';
281             if (success != expected) {
282                 ERRORF(reporter, "SkBitmap::copyTo from %s to %s. expected %s "
283                        "returned %s", gColorTypeName[i], gColorTypeName[j],
284                        boolStr(expected), boolStr(success));
285             }
286 
287             bool canSucceed = srcPremul.canCopyTo(gPairs[j].fColorType);
288             if (success != canSucceed) {
289                 ERRORF(reporter, "SkBitmap::copyTo from %s to %s. returned %s "
290                        "canCopyTo %s", gColorTypeName[i], gColorTypeName[j],
291                        boolStr(success), boolStr(canSucceed));
292             }
293 
294             if (success) {
295                 REPORTER_ASSERT(reporter, srcPremul.width() == dst.width());
296                 REPORTER_ASSERT(reporter, srcPremul.height() == dst.height());
297                 REPORTER_ASSERT(reporter, dst.colorType() == gPairs[j].fColorType);
298                 test_isOpaque(reporter, srcOpaque, srcPremul, dst.colorType());
299                 if (srcPremul.colorType() == dst.colorType()) {
300                     SkAutoLockPixels srcLock(srcPremul);
301                     SkAutoLockPixels dstLock(dst);
302                     REPORTER_ASSERT(reporter, srcPremul.readyToDraw());
303                     REPORTER_ASSERT(reporter, dst.readyToDraw());
304                     const char* srcP = (const char*)srcPremul.getAddr(0, 0);
305                     const char* dstP = (const char*)dst.getAddr(0, 0);
306                     REPORTER_ASSERT(reporter, srcP != dstP);
307                     REPORTER_ASSERT(reporter, !memcmp(srcP, dstP,
308                                                       srcPremul.getSize()));
309                     REPORTER_ASSERT(reporter, srcPremul.getGenerationID() == dst.getGenerationID());
310                 } else {
311                     REPORTER_ASSERT(reporter, srcPremul.getGenerationID() != dst.getGenerationID());
312                 }
313             } else {
314                 // dst should be unchanged from its initial state
315                 REPORTER_ASSERT(reporter, dst.colorType() == kUnknown_SkColorType);
316                 REPORTER_ASSERT(reporter, dst.width() == 0);
317                 REPORTER_ASSERT(reporter, dst.height() == 0);
318             }
319         } // for (size_t j = ...
320 
321         // Tests for getSafeSize(), getSafeSize64(), copyPixelsTo(),
322         // copyPixelsFrom().
323         //
324         for (size_t copyCase = 0; copyCase < SK_ARRAY_COUNT(isExtracted);
325              ++copyCase) {
326             // Test copying to/from external buffer.
327             // Note: the tests below have hard-coded values ---
328             //       Please take care if modifying.
329 
330             // Tests for getSafeSize64().
331             // Test with a very large configuration without pixel buffer
332             // attached.
333             SkBitmap tstSafeSize;
334             tstSafeSize.setInfo(SkImageInfo::Make(100000000U, 100000000U,
335                                                   gPairs[i].fColorType, kPremul_SkAlphaType));
336             int64_t safeSize = tstSafeSize.computeSafeSize64();
337             if (safeSize < 0) {
338                 ERRORF(reporter, "getSafeSize64() negative: %s",
339                        gColorTypeName[tstSafeSize.colorType()]);
340             }
341             bool sizeFail = false;
342             // Compare against hand-computed values.
343             switch (gPairs[i].fColorType) {
344                 case kUnknown_SkColorType:
345                     break;
346 
347                 case kAlpha_8_SkColorType:
348                 case kIndex_8_SkColorType:
349                     if (safeSize != 0x2386F26FC10000LL) {
350                         sizeFail = true;
351                     }
352                     break;
353 
354                 case kRGB_565_SkColorType:
355                 case kARGB_4444_SkColorType:
356                     if (safeSize != 0x470DE4DF820000LL) {
357                         sizeFail = true;
358                     }
359                     break;
360 
361                 case kN32_SkColorType:
362                     if (safeSize != 0x8E1BC9BF040000LL) {
363                         sizeFail = true;
364                     }
365                     break;
366 
367                 default:
368                     break;
369             }
370             if (sizeFail) {
371                 ERRORF(reporter, "computeSafeSize64() wrong size: %s",
372                        gColorTypeName[tstSafeSize.colorType()]);
373             }
374 
375             int subW = 2;
376             int subH = 2;
377 
378             // Create bitmap to act as source for copies and subsets.
379             SkBitmap src, subset;
380             SkColorTable* ct = nullptr;
381             if (kIndex_8_SkColorType == src.colorType()) {
382                 ct = init_ctable();
383             }
384 
385             int localSubW;
386             if (isExtracted[copyCase]) { // A larger image to extract from.
387                 localSubW = 2 * subW + 1;
388             } else { // Tests expect a 2x2 bitmap, so make smaller.
389                 localSubW = subW;
390             }
391             // could fail if we pass kIndex_8 for the colortype
392             if (src.tryAllocPixels(SkImageInfo::Make(localSubW, subH, gPairs[i].fColorType,
393                                                      kPremul_SkAlphaType))) {
394                 // failure is fine, as we will notice later on
395             }
396             SkSafeUnref(ct);
397 
398             // Either copy src or extract into 'subset', which is used
399             // for subsequent calls to copyPixelsTo/From.
400             bool srcReady = false;
401             // Test relies on older behavior that extractSubset will fail on
402             // kUnknown_SkColorType
403             if (kUnknown_SkColorType != src.colorType() &&
404                 isExtracted[copyCase]) {
405                 // The extractedSubset() test case allows us to test copy-
406                 // ing when src and dst mave possibly different strides.
407                 SkIRect r;
408                 r.set(1, 0, 1 + subW, subH); // 2x2 extracted bitmap
409 
410                 srcReady = src.extractSubset(&subset, r);
411             } else {
412                 srcReady = src.copyTo(&subset);
413             }
414 
415             // Not all configurations will generate a valid 'subset'.
416             if (srcReady) {
417 
418                 // Allocate our target buffer 'buf' for all copies.
419                 // To simplify verifying correctness of copies attach
420                 // buf to a SkBitmap, but copies are done using the
421                 // raw buffer pointer.
422                 const size_t bufSize = subH *
423                     SkColorTypeMinRowBytes(src.colorType(), subW) * 2;
424                 SkAutoTMalloc<uint8_t> autoBuf (bufSize);
425                 uint8_t* buf = autoBuf.get();
426 
427                 SkBitmap bufBm; // Attach buf to this bitmap.
428                 bool successExpected;
429 
430                 // Set up values for each pixel being copied.
431                 Coordinates coords(subW * subH);
432                 for (int x = 0; x < subW; ++x)
433                     for (int y = 0; y < subH; ++y)
434                     {
435                         int index = y * subW + x;
436                         SkASSERT(index < coords.length);
437                         coords[index]->fX = x;
438                         coords[index]->fY = y;
439                     }
440 
441                 writeCoordPixels(subset, coords);
442 
443                 // Test #1 ////////////////////////////////////////////
444 
445                 const SkImageInfo info = SkImageInfo::Make(subW, subH,
446                                                            gPairs[i].fColorType,
447                                                            kPremul_SkAlphaType);
448                 // Before/after comparisons easier if we attach buf
449                 // to an appropriately configured SkBitmap.
450                 memset(buf, 0xFF, bufSize);
451                 // Config with stride greater than src but that fits in buf.
452                 bufBm.installPixels(info, buf, info.minRowBytes() * 2);
453                 successExpected = false;
454                 // Then attempt to copy with a stride that is too large
455                 // to fit in the buffer.
456                 REPORTER_ASSERT(reporter,
457                     subset.copyPixelsTo(buf, bufSize, bufBm.rowBytes() * 3)
458                     == successExpected);
459 
460                 if (successExpected)
461                     reportCopyVerification(subset, bufBm, coords,
462                         "copyPixelsTo(buf, bufSize, 1.5*maxRowBytes)",
463                         reporter);
464 
465                 // Test #2 ////////////////////////////////////////////
466                 // This test should always succeed, but in the case
467                 // of extracted bitmaps only because we handle the
468                 // issue of getSafeSize(). Without getSafeSize()
469                 // buffer overrun/read would occur.
470                 memset(buf, 0xFF, bufSize);
471                 bufBm.installPixels(info, buf, subset.rowBytes());
472                 successExpected = subset.getSafeSize() <= bufSize;
473                 REPORTER_ASSERT(reporter,
474                     subset.copyPixelsTo(buf, bufSize) ==
475                         successExpected);
476                 if (successExpected)
477                     reportCopyVerification(subset, bufBm, coords,
478                     "copyPixelsTo(buf, bufSize)", reporter);
479 
480                 // Test #3 ////////////////////////////////////////////
481                 // Copy with different stride between src and dst.
482                 memset(buf, 0xFF, bufSize);
483                 bufBm.installPixels(info, buf, subset.rowBytes()+1);
484                 successExpected = true; // Should always work.
485                 REPORTER_ASSERT(reporter,
486                         subset.copyPixelsTo(buf, bufSize,
487                             subset.rowBytes()+1) == successExpected);
488                 if (successExpected)
489                     reportCopyVerification(subset, bufBm, coords,
490                     "copyPixelsTo(buf, bufSize, rowBytes+1)", reporter);
491 
492                 // Test #4 ////////////////////////////////////////////
493                 // Test copy with stride too small.
494                 memset(buf, 0xFF, bufSize);
495                 bufBm.installPixels(info, buf, info.minRowBytes());
496                 successExpected = false;
497                 // Request copy with stride too small.
498                 REPORTER_ASSERT(reporter,
499                     subset.copyPixelsTo(buf, bufSize, bufBm.rowBytes()-1)
500                         == successExpected);
501                 if (successExpected)
502                     reportCopyVerification(subset, bufBm, coords,
503                     "copyPixelsTo(buf, bufSize, rowBytes()-1)", reporter);
504 
505 #if 0   // copyPixelsFrom is gone
506                 // Test #5 ////////////////////////////////////////////
507                 // Tests the case where the source stride is too small
508                 // for the source configuration.
509                 memset(buf, 0xFF, bufSize);
510                 bufBm.installPixels(info, buf, info.minRowBytes());
511                 writeCoordPixels(bufBm, coords);
512                 REPORTER_ASSERT(reporter,
513                     subset.copyPixelsFrom(buf, bufSize, 1) == false);
514 
515                 // Test #6 ///////////////////////////////////////////
516                 // Tests basic copy from an external buffer to the bitmap.
517                 // If the bitmap is "extracted", this also tests the case
518                 // where the source stride is different from the dest.
519                 // stride.
520                 // We've made the buffer large enough to always succeed.
521                 bufBm.installPixels(info, buf, info.minRowBytes());
522                 writeCoordPixels(bufBm, coords);
523                 REPORTER_ASSERT(reporter,
524                     subset.copyPixelsFrom(buf, bufSize, bufBm.rowBytes()) ==
525                         true);
526                 reportCopyVerification(bufBm, subset, coords,
527                     "copyPixelsFrom(buf, bufSize)",
528                     reporter);
529 
530                 // Test #7 ////////////////////////////////////////////
531                 // Tests the case where the source buffer is too small
532                 // for the transfer.
533                 REPORTER_ASSERT(reporter,
534                     subset.copyPixelsFrom(buf, 1, subset.rowBytes()) ==
535                         false);
536 
537 #endif
538             }
539         } // for (size_t copyCase ...
540     }
541 }
542 
543 #include "SkColorPriv.h"
544 #include "SkUtils.h"
545 
546 /**
547  *  Construct 4x4 pixels where we can look at a color and determine where it should be in the grid.
548  *  alpha = 0xFF, blue = 0x80, red = x, green = y
549  */
fill_4x4_pixels(SkPMColor colors[16])550 static void fill_4x4_pixels(SkPMColor colors[16]) {
551     for (int y = 0; y < 4; ++y) {
552         for (int x = 0; x < 4; ++x) {
553             colors[y*4+x] = SkPackARGB32(0xFF, x, y, 0x80);
554         }
555     }
556 }
557 
check_4x4_pixel(SkPMColor color,unsigned x,unsigned y)558 static bool check_4x4_pixel(SkPMColor color, unsigned x, unsigned y) {
559     SkASSERT(x < 4 && y < 4);
560     return  0xFF == SkGetPackedA32(color) &&
561             x    == SkGetPackedR32(color) &&
562             y    == SkGetPackedG32(color) &&
563             0x80 == SkGetPackedB32(color);
564 }
565 
566 /**
567  *  Fill with all zeros, which will never match any value from fill_4x4_pixels
568  */
clear_4x4_pixels(SkPMColor colors[16])569 static void clear_4x4_pixels(SkPMColor colors[16]) {
570     sk_memset32(colors, 0, 16);
571 }
572 
573 // Much of readPixels is exercised by copyTo testing, since readPixels is the backend for that
574 // method. Here we explicitly test subset copies.
575 //
DEF_TEST(BitmapReadPixels,reporter)576 DEF_TEST(BitmapReadPixels, reporter) {
577     const int W = 4;
578     const int H = 4;
579     const size_t rowBytes = W * sizeof(SkPMColor);
580     const SkImageInfo srcInfo = SkImageInfo::MakeN32Premul(W, H);
581     SkPMColor srcPixels[16];
582     fill_4x4_pixels(srcPixels);
583     SkBitmap srcBM;
584     srcBM.installPixels(srcInfo, srcPixels, rowBytes);
585 
586     SkImageInfo dstInfo = SkImageInfo::MakeN32Premul(W, H);
587     SkPMColor dstPixels[16];
588 
589     const struct {
590         bool     fExpectedSuccess;
591         SkIPoint fRequestedSrcLoc;
592         SkISize  fRequestedDstSize;
593         // If fExpectedSuccess, check these, otherwise ignore
594         SkIPoint fExpectedDstLoc;
595         SkIRect  fExpectedSrcR;
596     } gRec[] = {
597         { true,  { 0, 0 }, { 4, 4 }, { 0, 0 }, { 0, 0, 4, 4 } },
598         { true,  { 1, 1 }, { 2, 2 }, { 0, 0 }, { 1, 1, 3, 3 } },
599         { true,  { 2, 2 }, { 4, 4 }, { 0, 0 }, { 2, 2, 4, 4 } },
600         { true,  {-1,-1 }, { 2, 2 }, { 1, 1 }, { 0, 0, 1, 1 } },
601         { false, {-1,-1 }, { 1, 1 }, { 0, 0 }, { 0, 0, 0, 0 } },
602     };
603 
604     for (size_t i = 0; i < SK_ARRAY_COUNT(gRec); ++i) {
605         clear_4x4_pixels(dstPixels);
606 
607         dstInfo = dstInfo.makeWH(gRec[i].fRequestedDstSize.width(),
608                                  gRec[i].fRequestedDstSize.height());
609         bool success = srcBM.readPixels(dstInfo, dstPixels, rowBytes,
610                                         gRec[i].fRequestedSrcLoc.x(), gRec[i].fRequestedSrcLoc.y());
611 
612         REPORTER_ASSERT(reporter, gRec[i].fExpectedSuccess == success);
613         if (success) {
614             const SkIRect srcR = gRec[i].fExpectedSrcR;
615             const int dstX = gRec[i].fExpectedDstLoc.x();
616             const int dstY = gRec[i].fExpectedDstLoc.y();
617             // Walk the dst pixels, and check if we got what we expected
618             for (int y = 0; y < H; ++y) {
619                 for (int x = 0; x < W; ++x) {
620                     SkPMColor dstC = dstPixels[y*4+x];
621                     // get into src coordinates
622                     int sx = x - dstX + srcR.x();
623                     int sy = y - dstY + srcR.y();
624                     if (srcR.contains(sx, sy)) {
625                         REPORTER_ASSERT(reporter, check_4x4_pixel(dstC, sx, sy));
626                     } else {
627                         REPORTER_ASSERT(reporter, 0 == dstC);
628                     }
629                 }
630             }
631         }
632     }
633 }
634 
635 #if SK_SUPPORT_GPU
636 
637 #include "GrContext.h"
638 #include "SkGr.h"
639 #include "SkColorPriv.h"
640 /** Tests calling copyTo on a texture backed bitmap. Tests that all BGRA_8888/RGBA_8888 combinations
641     of src and dst work. This test should be removed when SkGrPixelRef is removed. */
DEF_GPUTEST_FOR_RENDERING_CONTEXTS(BitmapCopy_Texture,reporter,ctx)642 DEF_GPUTEST_FOR_RENDERING_CONTEXTS(BitmapCopy_Texture, reporter, ctx) {
643     static const SkPMColor kData[] = {
644         0xFF112233, 0xAF224499,
645         0xEF004466, 0x80773311
646     };
647 
648     uint32_t swizData[SK_ARRAY_COUNT(kData)];
649     for (size_t i = 0; i < SK_ARRAY_COUNT(kData); ++i) {
650         swizData[i] = SkSwizzle_RB(kData[i]);
651     }
652 
653     static const GrPixelConfig kSrcConfigs[] = {
654         kRGBA_8888_GrPixelConfig,
655         kBGRA_8888_GrPixelConfig,
656     };
657 
658     for (size_t srcC = 0; srcC < SK_ARRAY_COUNT(kSrcConfigs); ++srcC) {
659         for (int rt = 0; rt < 2; ++rt) {
660             GrSurfaceDesc desc;
661             desc.fConfig = kSrcConfigs[srcC];
662             desc.fFlags = rt ? kRenderTarget_GrSurfaceFlag : kNone_GrSurfaceFlags;
663             desc.fWidth = 2;
664             desc.fHeight = 2;
665             desc.fOrigin = kTopLeft_GrSurfaceOrigin;
666 
667             const void* srcData = (kSkia8888_GrPixelConfig == desc.fConfig) ? kData : swizData;
668 
669             SkAutoTUnref<GrTexture> texture(
670                 ctx->textureProvider()->createTexture(desc, SkBudgeted::kNo, srcData, 0));
671 
672             if (!texture) {
673                 continue;
674             }
675 
676             SkBitmap srcBmp;
677             GrWrapTextureInBitmap(texture, 2, 2, false, &srcBmp);
678             if (srcBmp.isNull()) {
679                 ERRORF(reporter, "Could not wrap texture in bitmap.");
680                 continue;
681             }
682             static const SkColorType kDstCTs[] = { kRGBA_8888_SkColorType, kBGRA_8888_SkColorType };
683             for (size_t dCT = 0; dCT < SK_ARRAY_COUNT(kDstCTs); ++dCT) {
684                 SkBitmap dstBmp;
685                 if (!srcBmp.copyTo(&dstBmp, kDstCTs[dCT])) {
686                     ERRORF(reporter, "CopyTo failed.");
687                 }
688                 if (dstBmp.colorType() != kDstCTs[dCT]) {
689                     ERRORF(reporter, "SkBitmap::CopyTo did not respect passed in color type.");
690                 }
691                 SkAutoLockPixels alp(dstBmp);
692                 uint8_t* dstBmpPixels = static_cast<uint8_t*>(dstBmp.getPixels());
693                 const uint32_t* refData;
694 #if defined(SK_PMCOLOR_IS_RGBA)
695                 refData = (kRGBA_8888_SkColorType == dstBmp.colorType()) ? kData : swizData;
696 #elif defined(SK_PMCOLOR_IS_BGRA)
697                 refData = (kBGRA_8888_SkColorType == dstBmp.colorType()) ? kData : swizData;
698 #else
699     #error "PM Color must be BGRA or RGBA to use GPU backend."
700 #endif
701                 bool foundError = false;
702                 for (int y = 0; y < 2 && !foundError; ++y) {
703                     uint32_t* dstBmpRow = reinterpret_cast<uint32_t*>(dstBmpPixels);
704                     for (int x = 0; x < 2 && !foundError; ++x) {
705                         if (refData[2 * y + x] != dstBmpRow[x]) {
706                             ERRORF(reporter, "Expected pixel 0x%08x, found 0x%08x.",
707                                    refData[2 * y + x], dstBmpRow[x]);
708                             foundError = true;
709                         }
710                     }
711                     dstBmpPixels += dstBmp.rowBytes();
712                 }
713             }
714         }
715     }
716 }
717 
718 #endif
719