18c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com/* 28c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com * Copyright 2013 Google Inc. 38c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com * 48c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com * Use of this source code is governed by a BSD-style license that can be 58c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com * found in the LICENSE file. 68c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com */ 78c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com 88c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com#include "SkImageEncoder.h" 98c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com#include "SkBitmap.h" 108c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com#include "SkColorPriv.h" 118c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com#include "SkStream.h" 128c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com#include "SkTemplates.h" 138c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com 148c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.comclass SkARGBImageEncoder : public SkImageEncoder { 158c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.comprotected: 168c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com virtual bool onEncode(SkWStream* stream, const SkBitmap& bm, int quality) SK_OVERRIDE; 178c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com 188c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.comprivate: 198c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com typedef SkImageEncoder INHERITED; 208c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com}; 218c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com 228c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.comtypedef void (*ScanlineImporter)(const uint8_t* in, uint8_t* argb, int width, 238c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com const SkPMColor* SK_RESTRICT ctable); 248c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com 258c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.comstatic void ARGB_8888_To_ARGB(const uint8_t* in, uint8_t* argb, int width, const SkPMColor*) { 268c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com const uint32_t* SK_RESTRICT src = (const uint32_t*)in; 278c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com for (int i = 0; i < width; ++i) { 288c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com const uint32_t c = *src++; 298c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb[0] = SkGetPackedA32(c); 308c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb[1] = SkGetPackedR32(c); 318c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb[2] = SkGetPackedG32(c); 328c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb[3] = SkGetPackedB32(c); 338c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb += 4; 348c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com } 358c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com} 368c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com 378c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.comstatic void RGB_565_To_ARGB(const uint8_t* in, uint8_t* argb, int width, const SkPMColor*) { 388c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com const uint16_t* SK_RESTRICT src = (const uint16_t*)in; 398c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com for (int i = 0; i < width; ++i) { 408c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com const uint16_t c = *src++; 418c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb[0] = 0xFF; 428c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb[1] = SkPacked16ToR32(c); 438c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb[2] = SkPacked16ToG32(c); 448c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb[3] = SkPacked16ToB32(c); 458c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb += 4; 468c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com } 478c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com} 488c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com 498c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.comstatic void ARGB_4444_To_ARGB(const uint8_t* in, uint8_t* argb, int width, const SkPMColor*) { 508c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com const SkPMColor16* SK_RESTRICT src = (const SkPMColor16*)in; 518c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com for (int i = 0; i < width; ++i) { 528c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com const SkPMColor16 c = *src++; 538c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb[0] = SkPacked4444ToA32(c); 548c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb[1] = SkPacked4444ToR32(c); 558c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb[2] = SkPacked4444ToG32(c); 568c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb[3] = SkPacked4444ToB32(c); 578c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb += 4; 588c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com } 598c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com} 608c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com 618c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.comstatic void Index8_To_ARGB(const uint8_t* in, uint8_t* argb, int width, 628c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com const SkPMColor* SK_RESTRICT ctable) { 638c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com const uint8_t* SK_RESTRICT src = (const uint8_t*)in; 648c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com for (int i = 0; i < width; ++i) { 658c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com const uint32_t c = ctable[*src++]; 668c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb[0] = SkGetPackedA32(c); 678c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb[1] = SkGetPackedR32(c); 688c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb[2] = SkGetPackedG32(c); 698c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb[3] = SkGetPackedB32(c); 708c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com argb += 4; 718c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com } 728c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com} 738c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com 740689d7b12e7c427a077b003d3d8ae759d86f798freedstatic ScanlineImporter ChooseImporter(SkColorType ct) { 750689d7b12e7c427a077b003d3d8ae759d86f798freed switch (ct) { 760689d7b12e7c427a077b003d3d8ae759d86f798freed case kN32_SkColorType: 778c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com return ARGB_8888_To_ARGB; 780689d7b12e7c427a077b003d3d8ae759d86f798freed case kRGB_565_SkColorType: 798c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com return RGB_565_To_ARGB; 800689d7b12e7c427a077b003d3d8ae759d86f798freed case kARGB_4444_SkColorType: 818c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com return ARGB_4444_To_ARGB; 820689d7b12e7c427a077b003d3d8ae759d86f798freed case kIndex_8_SkColorType: 838c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com return Index8_To_ARGB; 848c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com default: 858c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com return NULL; 868c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com } 878c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com} 888c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com 898c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.combool SkARGBImageEncoder::onEncode(SkWStream* stream, const SkBitmap& bitmap, int) { 900689d7b12e7c427a077b003d3d8ae759d86f798freed const ScanlineImporter scanline_import = ChooseImporter(bitmap.colorType()); 918c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com if (NULL == scanline_import) { 928c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com return false; 938c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com } 948c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com 958c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com SkAutoLockPixels alp(bitmap); 968c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com const uint8_t* src = (uint8_t*)bitmap.getPixels(); 978c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com if (NULL == bitmap.getPixels()) { 988c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com return false; 998c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com } 1008c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com 1018c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com SkAutoLockColors ctLocker; 1028c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com const SkPMColor* colors = ctLocker.lockColors(bitmap); 1038c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com 1048c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com const int argbStride = bitmap.width() * 4; 1058c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com SkAutoTDeleteArray<uint8_t> ada(new uint8_t[argbStride]); 1068c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com uint8_t* argb = ada.get(); 1078c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com for (int y = 0; y < bitmap.height(); ++y) { 1088c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com scanline_import(src + y * bitmap.rowBytes(), argb, bitmap.width(), colors); 1098c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com stream->write(argb, argbStride); 1108c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com } 1118c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com 1128c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com return true; 1138c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com} 1148c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com 1158c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com 1168c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com/////////////////////////////////////////////////////////////////////////////// 1178c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.comDEFINE_ENCODER_CREATOR(ARGBImageEncoder); 1188c6a4f24d331503b3eb9a5c918d5876772b9a5eebungeman@google.com/////////////////////////////////////////////////////////////////////////////// 119