1/* 2 * Copyright 2012 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 "SkImageEncoder.h" 9 10#include "SkBitmap.h" 11#include "SkCanvas.h" 12#include "SkStream.h" 13#include "Test.h" 14 15static SkColorType gColorTypes[] = { 16 kRGB_565_SkColorType, 17 kN32_SkColorType, 18}; 19 20DEF_TEST(ARGBImageEncoder, reporter) { 21 // Bytes we expect to get: 22 const int kWidth = 3; 23 const int kHeight = 5; 24 const unsigned char comparisonBuffer[] = { 25 // kHeight rows, each with kWidth pixels, premultiplied ARGB for each pixel 26 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,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 31 }; 32 33 SkAutoTDelete<SkImageEncoder> enc(CreateARGBImageEncoder()); 34 for (size_t ctIndex = 0; ctIndex < SK_ARRAY_COUNT(gColorTypes); ++ctIndex) { 35 // A bitmap that should generate the above bytes: 36 SkBitmap bitmap; 37 { 38 bitmap.allocPixels(SkImageInfo::Make(kWidth, kHeight, gColorTypes[ctIndex], 39 kOpaque_SkAlphaType)); 40 bitmap.eraseColor(SK_ColorBLUE); 41 // Change rows [0,1] from blue to [red,green]. 42 SkCanvas canvas(bitmap); 43 SkPaint paint; 44 paint.setColor(SK_ColorRED); 45 canvas.drawIRect(SkIRect::MakeLTRB(0, 0, kWidth, 1), paint); 46 paint.setColor(SK_ColorGREEN); 47 canvas.drawIRect(SkIRect::MakeLTRB(0, 1, kWidth, 2), paint); 48 } 49 50 // Transform the bitmap. 51 int bufferSize = bitmap.width() * bitmap.height() * 4; 52 SkAutoMalloc pixelBufferManager(bufferSize); 53 char *pixelBuffer = static_cast<char *>(pixelBufferManager.get()); 54 SkMemoryWStream out(pixelBuffer, bufferSize); 55 REPORTER_ASSERT(reporter, enc->encodeStream(&out, bitmap, SkImageEncoder::kDefaultQuality)); 56 57 // Confirm we got the expected results. 58 REPORTER_ASSERT(reporter, bufferSize == sizeof(comparisonBuffer)); 59 REPORTER_ASSERT(reporter, memcmp(pixelBuffer, comparisonBuffer, bufferSize) == 0); 60 } 61} 62