michael@0: /*
michael@0:  * Copyright 2013 Google Inc.
michael@0:  *
michael@0:  * Use of this source code is governed by a BSD-style license that can be
michael@0:  * found in the LICENSE file.
michael@0:  */
michael@0: 
michael@0: #include "SkImageEncoder.h"
michael@0: #include "SkBitmap.h"
michael@0: #include "SkColorPriv.h"
michael@0: #include "SkStream.h"
michael@0: #include "SkTemplates.h"
michael@0: 
michael@0: class SkARGBImageEncoder : public SkImageEncoder {
michael@0: protected:
michael@0:     virtual bool onEncode(SkWStream* stream, const SkBitmap& bm, int quality) SK_OVERRIDE;
michael@0: 
michael@0: private:
michael@0:     typedef SkImageEncoder INHERITED;
michael@0: };
michael@0: 
michael@0: typedef void (*ScanlineImporter)(const uint8_t* in, uint8_t* argb, int width,
michael@0:                                  const SkPMColor* SK_RESTRICT ctable);
michael@0: 
michael@0: static void ARGB_8888_To_ARGB(const uint8_t* in, uint8_t* argb, int width, const SkPMColor*) {
michael@0:   const uint32_t* SK_RESTRICT src = (const uint32_t*)in;
michael@0:   for (int i = 0; i < width; ++i) {
michael@0:       const uint32_t c = *src++;
michael@0:       argb[0] = SkGetPackedA32(c);
michael@0:       argb[1] = SkGetPackedR32(c);
michael@0:       argb[2] = SkGetPackedG32(c);
michael@0:       argb[3] = SkGetPackedB32(c);
michael@0:       argb += 4;
michael@0:   }
michael@0: }
michael@0: 
michael@0: static void RGB_565_To_ARGB(const uint8_t* in, uint8_t* argb, int width, const SkPMColor*) {
michael@0:   const uint16_t* SK_RESTRICT src = (const uint16_t*)in;
michael@0:   for (int i = 0; i < width; ++i) {
michael@0:       const uint16_t c = *src++;
michael@0:       argb[0] = 0xFF;
michael@0:       argb[1] = SkPacked16ToR32(c);
michael@0:       argb[2] = SkPacked16ToG32(c);
michael@0:       argb[3] = SkPacked16ToB32(c);
michael@0:       argb += 4;
michael@0:   }
michael@0: }
michael@0: 
michael@0: static void ARGB_4444_To_ARGB(const uint8_t* in, uint8_t* argb, int width, const SkPMColor*) {
michael@0:   const SkPMColor16* SK_RESTRICT src = (const SkPMColor16*)in;
michael@0:   for (int i = 0; i < width; ++i) {
michael@0:       const SkPMColor16 c = *src++;
michael@0:       argb[0] = SkPacked4444ToA32(c);
michael@0:       argb[1] = SkPacked4444ToR32(c);
michael@0:       argb[2] = SkPacked4444ToG32(c);
michael@0:       argb[3] = SkPacked4444ToB32(c);
michael@0:       argb += 4;
michael@0:   }
michael@0: }
michael@0: 
michael@0: static void Index8_To_ARGB(const uint8_t* in, uint8_t* argb, int width,
michael@0:                            const SkPMColor* SK_RESTRICT ctable) {
michael@0:   const uint8_t* SK_RESTRICT src = (const uint8_t*)in;
michael@0:   for (int i = 0; i < width; ++i) {
michael@0:       const uint32_t c = ctable[*src++];
michael@0:       argb[0] = SkGetPackedA32(c);
michael@0:       argb[1] = SkGetPackedR32(c);
michael@0:       argb[2] = SkGetPackedG32(c);
michael@0:       argb[3] = SkGetPackedB32(c);
michael@0:       argb += 4;
michael@0:   }
michael@0: }
michael@0: 
michael@0: static ScanlineImporter ChooseImporter(const SkBitmap::Config& config) {
michael@0:     switch (config) {
michael@0:         case SkBitmap::kARGB_8888_Config:
michael@0:             return ARGB_8888_To_ARGB;
michael@0:         case SkBitmap::kRGB_565_Config:
michael@0:             return RGB_565_To_ARGB;
michael@0:         case SkBitmap::kARGB_4444_Config:
michael@0:             return ARGB_4444_To_ARGB;
michael@0:         case SkBitmap::kIndex8_Config:
michael@0:             return Index8_To_ARGB;
michael@0:         default:
michael@0:             return NULL;
michael@0:     }
michael@0: }
michael@0: 
michael@0: bool SkARGBImageEncoder::onEncode(SkWStream* stream, const SkBitmap& bitmap, int) {
michael@0:     const SkBitmap::Config config = bitmap.config();
michael@0:     const ScanlineImporter scanline_import = ChooseImporter(config);
michael@0:     if (NULL == scanline_import) {
michael@0:         return false;
michael@0:     }
michael@0: 
michael@0:     SkAutoLockPixels alp(bitmap);
michael@0:     const uint8_t* src = (uint8_t*)bitmap.getPixels();
michael@0:     if (NULL == bitmap.getPixels()) {
michael@0:         return false;
michael@0:     }
michael@0: 
michael@0:     SkAutoLockColors ctLocker;
michael@0:     const SkPMColor* colors = ctLocker.lockColors(bitmap);
michael@0: 
michael@0:     const int argbStride = bitmap.width() * 4;
michael@0:     SkAutoTDeleteArray<uint8_t> ada(new uint8_t[argbStride]);
michael@0:     uint8_t* argb = ada.get();
michael@0:     for (int y = 0; y < bitmap.height(); ++y) {
michael@0:         scanline_import(src + y * bitmap.rowBytes(), argb, bitmap.width(), colors);
michael@0:         stream->write(argb, argbStride);
michael@0:     }
michael@0: 
michael@0:     return true;
michael@0: }
michael@0: 
michael@0: 
michael@0: ///////////////////////////////////////////////////////////////////////////////
michael@0: DEFINE_ENCODER_CREATOR(ARGBImageEncoder);
michael@0: ///////////////////////////////////////////////////////////////////////////////