| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright 2012 Google Inc. | 2 * Copyright 2012 Google Inc. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license that can be | 4 * Use of this source code is governed by a BSD-style license that can be |
| 5 * found in the LICENSE file. | 5 * found in the LICENSE file. |
| 6 */ | 6 */ |
| 7 | 7 |
| 8 #include "SkImageEncoder.h" | 8 #include "SkImageEncoder.h" |
| 9 | 9 |
| 10 #include "SkBitmap.h" | 10 #include "SkBitmap.h" |
| 11 #include "SkCanvas.h" | 11 #include "SkCanvas.h" |
| 12 #include "SkImageEncoder_argb.h" |
| 12 #include "SkStream.h" | 13 #include "SkStream.h" |
| 13 #include "Test.h" | 14 #include "Test.h" |
| 14 | 15 |
| 15 static SkColorType gColorTypes[] = { | 16 static SkColorType gColorTypes[] = { |
| 16 kRGB_565_SkColorType, | 17 kRGB_565_SkColorType, |
| 17 kN32_SkColorType, | 18 kN32_SkColorType, |
| 18 }; | 19 }; |
| 19 | 20 |
| 20 DEF_TEST(ARGBImageEncoder, reporter) { | 21 DEF_TEST(ARGBImageEncoder, reporter) { |
| 21 // Bytes we expect to get: | 22 // Bytes we expect to get: |
| 22 const int kWidth = 3; | 23 const int kWidth = 3; |
| 23 const int kHeight = 5; | 24 const int kHeight = 5; |
| 24 const unsigned char comparisonBuffer[] = { | 25 const unsigned char comparisonBuffer[] = { |
| 25 // kHeight rows, each with kWidth pixels, premultiplied ARGB for each pi
xel | 26 // kHeight rows, each with kWidth pixels, premultiplied ARGB for each pi
xel |
| 26 0xff,0xff,0x00,0x00, 0xff,0xff,0x00,0x00, 0xff,0xff,0x00,0x00, // red | 27 0xff,0xff,0x00,0x00, 0xff,0xff,0x00,0x00, 0xff,0xff,0x00,0x00, // red |
| 27 0xff,0x00,0xff,0x00, 0xff,0x00,0xff,0x00, 0xff,0x00,0xff,0x00, // green | 28 0xff,0x00,0xff,0x00, 0xff,0x00,0xff,0x00, 0xff,0x00,0xff,0x00, // green |
| 28 0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, // blue | 29 0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, // blue |
| 29 0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, // blue | 30 0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, // blue |
| 30 0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, // blue | 31 0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, 0xff,0x00,0x00,0xff, // blue |
| 31 }; | 32 }; |
| 32 | 33 |
| 33 SkAutoTDelete<SkImageEncoder> enc(CreateARGBImageEncoder()); | 34 SkAutoTDelete<SkImageEncoder> enc(SkNEW(SkImageEncoder_argb)); |
| 34 for (size_t ctIndex = 0; ctIndex < SK_ARRAY_COUNT(gColorTypes); ++ctIndex) { | 35 for (size_t ctIndex = 0; ctIndex < SK_ARRAY_COUNT(gColorTypes); ++ctIndex) { |
| 35 // A bitmap that should generate the above bytes: | 36 // A bitmap that should generate the above bytes: |
| 36 SkBitmap bitmap; | 37 SkBitmap bitmap; |
| 37 { | 38 { |
| 38 bitmap.allocPixels(SkImageInfo::Make(kWidth, kHeight, gColorTypes[ct
Index], | 39 bitmap.allocPixels(SkImageInfo::Make(kWidth, kHeight, gColorTypes[ct
Index], |
| 39 kOpaque_SkAlphaType)); | 40 kOpaque_SkAlphaType)); |
| 40 bitmap.eraseColor(SK_ColorBLUE); | 41 bitmap.eraseColor(SK_ColorBLUE); |
| 41 // Change rows [0,1] from blue to [red,green]. | 42 // Change rows [0,1] from blue to [red,green]. |
| 42 SkCanvas canvas(bitmap); | 43 SkCanvas canvas(bitmap); |
| 43 SkPaint paint; | 44 SkPaint paint; |
| 44 paint.setColor(SK_ColorRED); | 45 paint.setColor(SK_ColorRED); |
| 45 canvas.drawIRect(SkIRect::MakeLTRB(0, 0, kWidth, 1), paint); | 46 canvas.drawIRect(SkIRect::MakeLTRB(0, 0, kWidth, 1), paint); |
| 46 paint.setColor(SK_ColorGREEN); | 47 paint.setColor(SK_ColorGREEN); |
| 47 canvas.drawIRect(SkIRect::MakeLTRB(0, 1, kWidth, 2), paint); | 48 canvas.drawIRect(SkIRect::MakeLTRB(0, 1, kWidth, 2), paint); |
| 48 } | 49 } |
| 49 | 50 |
| 50 // Transform the bitmap. | 51 // Transform the bitmap. |
| 51 int bufferSize = bitmap.width() * bitmap.height() * 4; | 52 int bufferSize = bitmap.width() * bitmap.height() * 4; |
| 52 SkAutoMalloc pixelBufferManager(bufferSize); | 53 SkAutoMalloc pixelBufferManager(bufferSize); |
| 53 char *pixelBuffer = static_cast<char *>(pixelBufferManager.get()); | 54 char *pixelBuffer = static_cast<char *>(pixelBufferManager.get()); |
| 54 SkMemoryWStream out(pixelBuffer, bufferSize); | 55 SkMemoryWStream out(pixelBuffer, bufferSize); |
| 55 REPORTER_ASSERT(reporter, enc->encodeStream(&out, bitmap, SkImageEncoder
::kDefaultQuality)); | 56 REPORTER_ASSERT(reporter, enc->encodeStream(&out, bitmap, SkImageEncoder
::kDefaultQuality)); |
| 56 | 57 |
| 57 // Confirm we got the expected results. | 58 // Confirm we got the expected results. |
| 58 REPORTER_ASSERT(reporter, bufferSize == sizeof(comparisonBuffer)); | 59 REPORTER_ASSERT(reporter, bufferSize == sizeof(comparisonBuffer)); |
| 59 REPORTER_ASSERT(reporter, memcmp(pixelBuffer, comparisonBuffer, bufferSi
ze) == 0); | 60 REPORTER_ASSERT(reporter, memcmp(pixelBuffer, comparisonBuffer, bufferSi
ze) == 0); |
| 60 } | 61 } |
| 61 } | 62 } |
| OLD | NEW |