1 /*
2 * Copyright 2013 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 #include "SkBitmap.h"
10 #include "SkColorPriv.h"
11 #include "SkStream.h"
12 #include "SkTemplates.h"
13
14 class SkARGBImageEncoder : public SkImageEncoder {
15 protected:
16 virtual bool onEncode(SkWStream* stream, const SkBitmap& bm, int quality) SK_OVERRIDE;
17
18 private:
19 typedef SkImageEncoder INHERITED;
20 };
21
22 typedef void (*ScanlineImporter)(const uint8_t* in, uint8_t* argb, int width,
23 const SkPMColor* SK_RESTRICT ctable);
24
ARGB_8888_To_ARGB(const uint8_t * in,uint8_t * argb,int width,const SkPMColor *)25 static void ARGB_8888_To_ARGB(const uint8_t* in, uint8_t* argb, int width, const SkPMColor*) {
26 const uint32_t* SK_RESTRICT src = (const uint32_t*)in;
27 for (int i = 0; i < width; ++i) {
28 const uint32_t c = *src++;
29 argb[0] = SkGetPackedA32(c);
30 argb[1] = SkGetPackedR32(c);
31 argb[2] = SkGetPackedG32(c);
32 argb[3] = SkGetPackedB32(c);
33 argb += 4;
34 }
35 }
36
RGB_565_To_ARGB(const uint8_t * in,uint8_t * argb,int width,const SkPMColor *)37 static void RGB_565_To_ARGB(const uint8_t* in, uint8_t* argb, int width, const SkPMColor*) {
38 const uint16_t* SK_RESTRICT src = (const uint16_t*)in;
39 for (int i = 0; i < width; ++i) {
40 const uint16_t c = *src++;
41 argb[0] = 0xFF;
42 argb[1] = SkPacked16ToR32(c);
43 argb[2] = SkPacked16ToG32(c);
44 argb[3] = SkPacked16ToB32(c);
45 argb += 4;
46 }
47 }
48
ARGB_4444_To_ARGB(const uint8_t * in,uint8_t * argb,int width,const SkPMColor *)49 static void ARGB_4444_To_ARGB(const uint8_t* in, uint8_t* argb, int width, const SkPMColor*) {
50 const SkPMColor16* SK_RESTRICT src = (const SkPMColor16*)in;
51 for (int i = 0; i < width; ++i) {
52 const SkPMColor16 c = *src++;
53 argb[0] = SkPacked4444ToA32(c);
54 argb[1] = SkPacked4444ToR32(c);
55 argb[2] = SkPacked4444ToG32(c);
56 argb[3] = SkPacked4444ToB32(c);
57 argb += 4;
58 }
59 }
60
Index8_To_ARGB(const uint8_t * in,uint8_t * argb,int width,const SkPMColor * SK_RESTRICT ctable)61 static void Index8_To_ARGB(const uint8_t* in, uint8_t* argb, int width,
62 const SkPMColor* SK_RESTRICT ctable) {
63 const uint8_t* SK_RESTRICT src = (const uint8_t*)in;
64 for (int i = 0; i < width; ++i) {
65 const uint32_t c = ctable[*src++];
66 argb[0] = SkGetPackedA32(c);
67 argb[1] = SkGetPackedR32(c);
68 argb[2] = SkGetPackedG32(c);
69 argb[3] = SkGetPackedB32(c);
70 argb += 4;
71 }
72 }
73
ChooseImporter(const SkBitmap::Config & config)74 static ScanlineImporter ChooseImporter(const SkBitmap::Config& config) {
75 switch (config) {
76 case SkBitmap::kARGB_8888_Config:
77 return ARGB_8888_To_ARGB;
78 case SkBitmap::kRGB_565_Config:
79 return RGB_565_To_ARGB;
80 case SkBitmap::kARGB_4444_Config:
81 return ARGB_4444_To_ARGB;
82 case SkBitmap::kIndex8_Config:
83 return Index8_To_ARGB;
84 default:
85 return NULL;
86 }
87 }
88
onEncode(SkWStream * stream,const SkBitmap & bitmap,int)89 bool SkARGBImageEncoder::onEncode(SkWStream* stream, const SkBitmap& bitmap, int) {
90 const SkBitmap::Config config = bitmap.config();
91 const ScanlineImporter scanline_import = ChooseImporter(config);
92 if (NULL == scanline_import) {
93 return false;
94 }
95
96 SkAutoLockPixels alp(bitmap);
97 const uint8_t* src = (uint8_t*)bitmap.getPixels();
98 if (NULL == bitmap.getPixels()) {
99 return false;
100 }
101
102 SkAutoLockColors ctLocker;
103 const SkPMColor* colors = ctLocker.lockColors(bitmap);
104
105 const int argbStride = bitmap.width() * 4;
106 SkAutoTDeleteArray<uint8_t> ada(new uint8_t[argbStride]);
107 uint8_t* argb = ada.get();
108 for (int y = 0; y < bitmap.height(); ++y) {
109 scanline_import(src + y * bitmap.rowBytes(), argb, bitmap.width(), colors);
110 stream->write(argb, argbStride);
111 }
112
113 return true;
114 }
115
116
117 ///////////////////////////////////////////////////////////////////////////////
118 DEFINE_ENCODER_CREATOR(ARGBImageEncoder);
119 ///////////////////////////////////////////////////////////////////////////////
120