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