blob: 7ff15584c4fb3f592eb36077a8726047cf4404af [file] [log] [blame]
epoger@google.comec3ed6a2011-07-28 14:26:00 +00001/*
2 * Copyright 2006 The Android Open Source Project
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
reed@android.com8a1c16f2008-12-17 15:59:43 +00008#include "SkImageDecoder.h"
reed@android.comb08eb2b2009-01-06 20:16:26 +00009#include "SkImageEncoder.h"
reed@android.com8a1c16f2008-12-17 15:59:43 +000010#include "SkColor.h"
11#include "SkColorPriv.h"
12#include "SkDither.h"
13#include "SkMath.h"
halcanary@google.com2a103182013-10-14 12:49:15 +000014#include "SkRTConf.h"
reed@android.com8a1c16f2008-12-17 15:59:43 +000015#include "SkScaledBitmapSampler.h"
16#include "SkStream.h"
17#include "SkTemplates.h"
18#include "SkUtils.h"
epoger@google.com4ce738b2012-11-16 18:44:18 +000019#include "transform_scanline.h"
reed@android.com8a1c16f2008-12-17 15:59:43 +000020extern "C" {
21#include "png.h"
22}
23
commit-bot@chromium.orgc1c56952013-05-02 13:41:34 +000024/* These were dropped in libpng >= 1.4 */
25#ifndef png_infopp_NULL
26#define png_infopp_NULL NULL
27#endif
28
29#ifndef png_bytepp_NULL
30#define png_bytepp_NULL NULL
31#endif
32
33#ifndef int_p_NULL
34#define int_p_NULL NULL
35#endif
36
37#ifndef png_flush_ptr_NULL
38#define png_flush_ptr_NULL NULL
39#endif
40
halcanary@google.comfed30372013-10-04 12:46:45 +000041#if defined(SK_DEBUG)
halcanary@google.com2a103182013-10-14 12:49:15 +000042#define DEFAULT_FOR_SUPPRESS_PNG_IMAGE_DECODER_WARNINGS false
43#else // !defined(SK_DEBUG)
44#define DEFAULT_FOR_SUPPRESS_PNG_IMAGE_DECODER_WARNINGS true
halcanary@google.comfed30372013-10-04 12:46:45 +000045#endif // defined(SK_DEBUG)
halcanary@google.com2a103182013-10-14 12:49:15 +000046SK_CONF_DECLARE(bool, c_suppressPNGImageDecoderWarnings,
47 "images.png.suppressDecoderWarnings",
48 DEFAULT_FOR_SUPPRESS_PNG_IMAGE_DECODER_WARNINGS,
49 "Suppress most PNG warnings when calling image decode "
50 "functions.");
51
halcanary@google.comfed30372013-10-04 12:46:45 +000052
53
commit-bot@chromium.orga936e372013-03-14 14:42:18 +000054class SkPNGImageIndex {
55public:
scroggo@google.comb5571b32013-09-25 21:34:24 +000056 SkPNGImageIndex(SkStreamRewindable* stream, png_structp png_ptr, png_infop info_ptr)
scroggo@google.comc70a3aa2013-07-18 20:03:15 +000057 : fStream(stream)
58 , fPng_ptr(png_ptr)
59 , fInfo_ptr(info_ptr)
reed6c225732014-06-09 19:52:07 -070060 , fColorType(kUnknown_SkColorType) {
scroggo@google.comc70a3aa2013-07-18 20:03:15 +000061 SkASSERT(stream != NULL);
62 stream->ref();
commit-bot@chromium.orga936e372013-03-14 14:42:18 +000063 }
64 ~SkPNGImageIndex() {
scroggo@google.comc70a3aa2013-07-18 20:03:15 +000065 if (NULL != fPng_ptr) {
66 png_destroy_read_struct(&fPng_ptr, &fInfo_ptr, png_infopp_NULL);
commit-bot@chromium.orga936e372013-03-14 14:42:18 +000067 }
68 }
69
scroggo@google.comb5571b32013-09-25 21:34:24 +000070 SkAutoTUnref<SkStreamRewindable> fStream;
71 png_structp fPng_ptr;
72 png_infop fInfo_ptr;
reed6c225732014-06-09 19:52:07 -070073 SkColorType fColorType;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +000074};
75
reed@android.com8a1c16f2008-12-17 15:59:43 +000076class SkPNGImageDecoder : public SkImageDecoder {
77public:
commit-bot@chromium.orga936e372013-03-14 14:42:18 +000078 SkPNGImageDecoder() {
79 fImageIndex = NULL;
80 }
81 virtual Format getFormat() const SK_OVERRIDE {
reed@android.com8a1c16f2008-12-17 15:59:43 +000082 return kPNG_Format;
83 }
scroggo@google.com39edf4c2013-04-25 17:33:51 +000084
commit-bot@chromium.orga936e372013-03-14 14:42:18 +000085 virtual ~SkPNGImageDecoder() {
86 SkDELETE(fImageIndex);
87 }
rmistry@google.comd6176b02012-08-23 18:14:13 +000088
reed@android.com8a1c16f2008-12-17 15:59:43 +000089protected:
commit-bot@chromium.orga936e372013-03-14 14:42:18 +000090#ifdef SK_BUILD_FOR_ANDROID
scroggo@google.comb5571b32013-09-25 21:34:24 +000091 virtual bool onBuildTileIndex(SkStreamRewindable *stream, int *width, int *height) SK_OVERRIDE;
scroggo@google.com7e6fcee2013-05-03 20:14:28 +000092 virtual bool onDecodeSubset(SkBitmap* bitmap, const SkIRect& region) SK_OVERRIDE;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +000093#endif
94 virtual bool onDecode(SkStream* stream, SkBitmap* bm, Mode) SK_OVERRIDE;
95
96private:
97 SkPNGImageIndex* fImageIndex;
98
99 bool onDecodeInit(SkStream* stream, png_structp *png_ptrp, png_infop *info_ptrp);
scroggo@google.com2bbc2c92013-06-14 15:33:20 +0000100 bool decodePalette(png_structp png_ptr, png_infop info_ptr,
101 bool * SK_RESTRICT hasAlphap, bool *reallyHasAlphap,
102 SkColorTable **colorTablep);
reed0689d7b2014-06-14 05:30:20 -0700103 bool getBitmapColorType(png_structp, png_infop, SkColorType*, bool* hasAlpha,
104 SkPMColor* theTranspColor);
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000105
106 typedef SkImageDecoder INHERITED;
reed@android.com8a1c16f2008-12-17 15:59:43 +0000107};
108
109#ifndef png_jmpbuf
110# define png_jmpbuf(png_ptr) ((png_ptr)->jmpbuf)
111#endif
112
113#define PNG_BYTES_TO_CHECK 4
114
115/* Automatically clean up after throwing an exception */
116struct PNGAutoClean {
117 PNGAutoClean(png_structp p, png_infop i): png_ptr(p), info_ptr(i) {}
118 ~PNGAutoClean() {
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000119 png_destroy_read_struct(&png_ptr, &info_ptr, png_infopp_NULL);
reed@android.com8a1c16f2008-12-17 15:59:43 +0000120 }
121private:
122 png_structp png_ptr;
123 png_infop info_ptr;
124};
125
reed@android.com8a1c16f2008-12-17 15:59:43 +0000126static void sk_read_fn(png_structp png_ptr, png_bytep data, png_size_t length) {
commit-bot@chromium.orgc1c56952013-05-02 13:41:34 +0000127 SkStream* sk_stream = (SkStream*) png_get_io_ptr(png_ptr);
reed@android.com8a1c16f2008-12-17 15:59:43 +0000128 size_t bytes = sk_stream->read(data, length);
129 if (bytes != length) {
130 png_error(png_ptr, "Read Error!");
131 }
132}
133
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000134#ifdef SK_BUILD_FOR_ANDROID
135static void sk_seek_fn(png_structp png_ptr, png_uint_32 offset) {
scroggo@google.comb5571b32013-09-25 21:34:24 +0000136 SkStreamRewindable* sk_stream = (SkStreamRewindable*) png_get_io_ptr(png_ptr);
scroggo@google.com4d213ab2013-08-28 13:08:54 +0000137 if (!sk_stream->rewind()) {
138 png_error(png_ptr, "Failed to rewind stream!");
139 }
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000140 (void)sk_stream->skip(offset);
141}
142#endif
143
reed@android.com8a1c16f2008-12-17 15:59:43 +0000144static int sk_read_user_chunk(png_structp png_ptr, png_unknown_chunkp chunk) {
145 SkImageDecoder::Peeker* peeker =
146 (SkImageDecoder::Peeker*)png_get_user_chunk_ptr(png_ptr);
147 // peek() returning true means continue decoding
148 return peeker->peek((const char*)chunk->name, chunk->data, chunk->size) ?
149 1 : -1;
150}
151
152static void sk_error_fn(png_structp png_ptr, png_const_charp msg) {
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000153 SkDEBUGF(("------ png error %s\n", msg));
reed@android.com8a1c16f2008-12-17 15:59:43 +0000154 longjmp(png_jmpbuf(png_ptr), 1);
155}
156
157static void skip_src_rows(png_structp png_ptr, uint8_t storage[], int count) {
158 for (int i = 0; i < count; i++) {
159 uint8_t* tmp = storage;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000160 png_read_rows(png_ptr, &tmp, png_bytepp_NULL, 1);
reed@android.com8a1c16f2008-12-17 15:59:43 +0000161 }
162}
163
164static bool pos_le(int value, int max) {
165 return value > 0 && value <= max;
166}
167
168static bool substituteTranspColor(SkBitmap* bm, SkPMColor match) {
reed0689d7b2014-06-14 05:30:20 -0700169 SkASSERT(bm->colorType() == kN32_SkColorType);
rmistry@google.comd6176b02012-08-23 18:14:13 +0000170
reed@android.com8a1c16f2008-12-17 15:59:43 +0000171 bool reallyHasAlpha = false;
172
173 for (int y = bm->height() - 1; y >= 0; --y) {
174 SkPMColor* p = bm->getAddr32(0, y);
175 for (int x = bm->width() - 1; x >= 0; --x) {
176 if (match == *p) {
177 *p = 0;
178 reallyHasAlpha = true;
179 }
180 p += 1;
181 }
182 }
183 return reallyHasAlpha;
184}
185
reed6c225732014-06-09 19:52:07 -0700186static bool canUpscalePaletteToConfig(SkColorType dstColorType, bool srcHasAlpha) {
187 switch (dstColorType) {
188 case kN32_SkColorType:
189 case kARGB_4444_SkColorType:
reed@android.comb6137c32009-07-29 20:56:52 +0000190 return true;
reed6c225732014-06-09 19:52:07 -0700191 case kRGB_565_SkColorType:
reed@android.comb6137c32009-07-29 20:56:52 +0000192 // only return true if the src is opaque (since 565 is opaque)
193 return !srcHasAlpha;
194 default:
195 return false;
196 }
197}
198
199// call only if color_type is PALETTE. Returns true if the ctable has alpha
200static bool hasTransparencyInPalette(png_structp png_ptr, png_infop info_ptr) {
201 png_bytep trans;
202 int num_trans;
203
204 if (png_get_valid(png_ptr, info_ptr, PNG_INFO_tRNS)) {
205 png_get_tRNS(png_ptr, info_ptr, &trans, &num_trans, NULL);
206 return num_trans > 0;
207 }
208 return false;
reed@android.com11344262009-07-08 20:09:23 +0000209}
210
halcanary@google.comfed30372013-10-04 12:46:45 +0000211void do_nothing_warning_fn(png_structp, png_const_charp) {
212 /* do nothing */
213}
214
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000215bool SkPNGImageDecoder::onDecodeInit(SkStream* sk_stream, png_structp *png_ptrp,
216 png_infop *info_ptrp) {
reed@android.com8a1c16f2008-12-17 15:59:43 +0000217 /* Create and initialize the png_struct with the desired error handler
218 * functions. If you want to use the default stderr and longjump method,
219 * you can supply NULL for the last three parameters. We also supply the
220 * the compiler header file version, so that we know if the application
221 * was compiled with a compatible version of the library. */
halcanary@google.comfed30372013-10-04 12:46:45 +0000222
halcanary@google.comfed30372013-10-04 12:46:45 +0000223 png_error_ptr user_warning_fn =
224 (c_suppressPNGImageDecoderWarnings) ? (&do_nothing_warning_fn) : NULL;
225 /* NULL means to leave as default library behavior. */
halcanary@google.com2a103182013-10-14 12:49:15 +0000226 /* c_suppressPNGImageDecoderWarnings default depends on SK_DEBUG. */
halcanary@google.comfed30372013-10-04 12:46:45 +0000227 /* To suppress warnings with a SK_DEBUG binary, set the
228 * environment variable "skia_images_png_suppressDecoderWarnings"
229 * to "true". Inside a program that links to skia:
230 * SK_CONF_SET("images.png.suppressDecoderWarnings", true); */
halcanary@google.comfed30372013-10-04 12:46:45 +0000231
reed@android.com8a1c16f2008-12-17 15:59:43 +0000232 png_structp png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING,
halcanary@google.comfed30372013-10-04 12:46:45 +0000233 NULL, sk_error_fn, user_warning_fn);
reed@android.com8a1c16f2008-12-17 15:59:43 +0000234 // png_voidp user_error_ptr, user_error_fn, user_warning_fn);
235 if (png_ptr == NULL) {
236 return false;
237 }
halcanary@google.comfed30372013-10-04 12:46:45 +0000238
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000239 *png_ptrp = png_ptr;
reed@android.com8a1c16f2008-12-17 15:59:43 +0000240
241 /* Allocate/initialize the memory for image information. */
242 png_infop info_ptr = png_create_info_struct(png_ptr);
243 if (info_ptr == NULL) {
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000244 png_destroy_read_struct(&png_ptr, png_infopp_NULL, png_infopp_NULL);
reed@android.com8a1c16f2008-12-17 15:59:43 +0000245 return false;
246 }
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000247 *info_ptrp = info_ptr;
reed@android.com8a1c16f2008-12-17 15:59:43 +0000248
249 /* Set error handling if you are using the setjmp/longjmp method (this is
250 * the normal method of doing things with libpng). REQUIRED unless you
251 * set up your own error handlers in the png_create_read_struct() earlier.
252 */
253 if (setjmp(png_jmpbuf(png_ptr))) {
scroggo@google.com5401cd02013-11-12 15:30:06 +0000254 png_destroy_read_struct(&png_ptr, &info_ptr, png_infopp_NULL);
reed@android.com8a1c16f2008-12-17 15:59:43 +0000255 return false;
256 }
257
258 /* If you are using replacement read functions, instead of calling
259 * png_init_io() here you would call:
260 */
261 png_set_read_fn(png_ptr, (void *)sk_stream, sk_read_fn);
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000262#ifdef SK_BUILD_FOR_ANDROID
263 png_set_seek_fn(png_ptr, sk_seek_fn);
264#endif
reed@android.com8a1c16f2008-12-17 15:59:43 +0000265 /* where user_io_ptr is a structure you want available to the callbacks */
266 /* If we have already read some of the signature */
267// png_set_sig_bytes(png_ptr, 0 /* sig_read */ );
268
269 // hookup our peeker so we can see any user-chunks the caller may be interested in
270 png_set_keep_unknown_chunks(png_ptr, PNG_HANDLE_CHUNK_ALWAYS, (png_byte*)"", 0);
271 if (this->getPeeker()) {
272 png_set_read_user_chunk_fn(png_ptr, (png_voidp)this->getPeeker(), sk_read_user_chunk);
273 }
274
275 /* The call to png_read_info() gives us all of the information from the
276 * PNG file before the first IDAT (image data chunk). */
277 png_read_info(png_ptr, info_ptr);
278 png_uint_32 origWidth, origHeight;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000279 int bitDepth, colorType;
280 png_get_IHDR(png_ptr, info_ptr, &origWidth, &origHeight, &bitDepth,
281 &colorType, int_p_NULL, int_p_NULL, int_p_NULL);
reed@android.com8a1c16f2008-12-17 15:59:43 +0000282
283 /* tell libpng to strip 16 bit/color files down to 8 bits/color */
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000284 if (bitDepth == 16) {
reed@android.com8a1c16f2008-12-17 15:59:43 +0000285 png_set_strip_16(png_ptr);
286 }
287 /* Extract multiple pixels with bit depths of 1, 2, and 4 from a single
288 * byte into separate bytes (useful for paletted and grayscale images). */
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000289 if (bitDepth < 8) {
reed@android.com8a1c16f2008-12-17 15:59:43 +0000290 png_set_packing(png_ptr);
291 }
292 /* Expand grayscale images to the full 8 bits from 1, 2, or 4 bits/pixel */
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000293 if (colorType == PNG_COLOR_TYPE_GRAY && bitDepth < 8) {
commit-bot@chromium.orgc1c56952013-05-02 13:41:34 +0000294 png_set_expand_gray_1_2_4_to_8(png_ptr);
reed@android.com8a1c16f2008-12-17 15:59:43 +0000295 }
rmistry@google.comd6176b02012-08-23 18:14:13 +0000296
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000297 return true;
298}
299
300bool SkPNGImageDecoder::onDecode(SkStream* sk_stream, SkBitmap* decodedBitmap,
301 Mode mode) {
302 png_structp png_ptr;
303 png_infop info_ptr;
304
305 if (!onDecodeInit(sk_stream, &png_ptr, &info_ptr)) {
306 return false;
307 }
308
scroggo@google.comfeeca3c2013-11-12 14:38:41 +0000309 PNGAutoClean autoClean(png_ptr, info_ptr);
310
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000311 if (setjmp(png_jmpbuf(png_ptr))) {
312 return false;
313 }
314
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000315 png_uint_32 origWidth, origHeight;
reed6c225732014-06-09 19:52:07 -0700316 int bitDepth, pngColorType, interlaceType;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000317 png_get_IHDR(png_ptr, info_ptr, &origWidth, &origHeight, &bitDepth,
reed6c225732014-06-09 19:52:07 -0700318 &pngColorType, &interlaceType, int_p_NULL, int_p_NULL);
rmistry@google.comd6176b02012-08-23 18:14:13 +0000319
reed6c225732014-06-09 19:52:07 -0700320 SkColorType colorType;
reed@android.com8a1c16f2008-12-17 15:59:43 +0000321 bool hasAlpha = false;
reed@android.com8a1c16f2008-12-17 15:59:43 +0000322 SkPMColor theTranspColor = 0; // 0 tells us not to try to match
rmistry@google.comd6176b02012-08-23 18:14:13 +0000323
reed0689d7b2014-06-14 05:30:20 -0700324 if (!this->getBitmapColorType(png_ptr, info_ptr, &colorType, &hasAlpha, &theTranspColor)) {
reed@android.com8a1c16f2008-12-17 15:59:43 +0000325 return false;
326 }
rmistry@google.comd6176b02012-08-23 18:14:13 +0000327
reed6c225732014-06-09 19:52:07 -0700328 SkAlphaType alphaType = this->getRequireUnpremultipliedColors() ?
329 kUnpremul_SkAlphaType : kPremul_SkAlphaType;
reed@android.com8a1c16f2008-12-17 15:59:43 +0000330 const int sampleSize = this->getSampleSize();
331 SkScaledBitmapSampler sampler(origWidth, origHeight, sampleSize);
reed6c225732014-06-09 19:52:07 -0700332 decodedBitmap->setInfo(SkImageInfo::Make(sampler.scaledWidth(), sampler.scaledHeight(),
333 colorType, alphaType));
djsollen@google.com446cf712014-02-19 21:45:35 +0000334
reed@android.com8a1c16f2008-12-17 15:59:43 +0000335 if (SkImageDecoder::kDecodeBounds_Mode == mode) {
336 return true;
337 }
rmistry@google.comd6176b02012-08-23 18:14:13 +0000338
reed@android.com8a1c16f2008-12-17 15:59:43 +0000339 // from here down we are concerned with colortables and pixels
340
341 // we track if we actually see a non-opaque pixels, since sometimes a PNG sets its colortype
342 // to |= PNG_COLOR_MASK_ALPHA, but all of its pixels are in fact opaque. We care, since we
343 // draw lots faster if we can flag the bitmap has being opaque
344 bool reallyHasAlpha = false;
reed@android.com8a1c16f2008-12-17 15:59:43 +0000345 SkColorTable* colorTable = NULL;
346
reed6c225732014-06-09 19:52:07 -0700347 if (pngColorType == PNG_COLOR_TYPE_PALETTE) {
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000348 decodePalette(png_ptr, info_ptr, &hasAlpha, &reallyHasAlpha, &colorTable);
reed@android.com8a1c16f2008-12-17 15:59:43 +0000349 }
rmistry@google.comd6176b02012-08-23 18:14:13 +0000350
reed@android.com8a1c16f2008-12-17 15:59:43 +0000351 SkAutoUnref aur(colorTable);
352
scroggo@google.combc69ce92013-07-09 15:45:14 +0000353 if (!this->allocPixelRef(decodedBitmap,
reed6c225732014-06-09 19:52:07 -0700354 kIndex_8_SkColorType == colorType ? colorTable : NULL)) {
scroggo@google.combc69ce92013-07-09 15:45:14 +0000355 return false;
reed@android.com8a1c16f2008-12-17 15:59:43 +0000356 }
rmistry@google.comd6176b02012-08-23 18:14:13 +0000357
reed@android.com8a1c16f2008-12-17 15:59:43 +0000358 SkAutoLockPixels alp(*decodedBitmap);
359
reed@android.com8a1c16f2008-12-17 15:59:43 +0000360 /* Turn on interlace handling. REQUIRED if you are not using
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000361 * png_read_image(). To see how to handle interlacing passes,
362 * see the png_read_row() method below:
reed@android.com8a1c16f2008-12-17 15:59:43 +0000363 */
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000364 const int number_passes = (interlaceType != PNG_INTERLACE_NONE) ?
365 png_set_interlace_handling(png_ptr) : 1;
reed@android.com8a1c16f2008-12-17 15:59:43 +0000366
367 /* Optional call to gamma correct and add the background to the palette
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000368 * and update info structure. REQUIRED if you are expecting libpng to
369 * update the palette for you (ie you selected such a transform above).
reed@android.com8a1c16f2008-12-17 15:59:43 +0000370 */
371 png_read_update_info(png_ptr, info_ptr);
372
reed6c225732014-06-09 19:52:07 -0700373 if ((kAlpha_8_SkColorType == colorType || kIndex_8_SkColorType == colorType) &&
374 1 == sampleSize) {
375 if (kAlpha_8_SkColorType == colorType) {
scroggo@google.com5ee18dd2013-10-21 20:47:31 +0000376 // For an A8 bitmap, we assume there is an alpha for speed. It is
377 // possible the bitmap is opaque, but that is an unlikely use case
378 // since it would not be very interesting.
379 reallyHasAlpha = true;
380 // A8 is only allowed if the original was GRAY.
reed6c225732014-06-09 19:52:07 -0700381 SkASSERT(PNG_COLOR_TYPE_GRAY == pngColorType);
scroggo@google.com5ee18dd2013-10-21 20:47:31 +0000382 }
reed@android.com8a1c16f2008-12-17 15:59:43 +0000383 for (int i = 0; i < number_passes; i++) {
384 for (png_uint_32 y = 0; y < origHeight; y++) {
385 uint8_t* bmRow = decodedBitmap->getAddr8(0, y);
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000386 png_read_rows(png_ptr, &bmRow, png_bytepp_NULL, 1);
reed@android.com8a1c16f2008-12-17 15:59:43 +0000387 }
388 }
389 } else {
390 SkScaledBitmapSampler::SrcConfig sc;
391 int srcBytesPerPixel = 4;
rmistry@google.comd6176b02012-08-23 18:14:13 +0000392
reed@android.com11344262009-07-08 20:09:23 +0000393 if (colorTable != NULL) {
reed@android.com8a1c16f2008-12-17 15:59:43 +0000394 sc = SkScaledBitmapSampler::kIndex;
395 srcBytesPerPixel = 1;
reed6c225732014-06-09 19:52:07 -0700396 } else if (kAlpha_8_SkColorType == colorType) {
scroggo@google.comc70a3aa2013-07-18 20:03:15 +0000397 // A8 is only allowed if the original was GRAY.
reed6c225732014-06-09 19:52:07 -0700398 SkASSERT(PNG_COLOR_TYPE_GRAY == pngColorType);
scroggo@google.comc70a3aa2013-07-18 20:03:15 +0000399 sc = SkScaledBitmapSampler::kGray;
400 srcBytesPerPixel = 1;
reed@android.com8a1c16f2008-12-17 15:59:43 +0000401 } else if (hasAlpha) {
402 sc = SkScaledBitmapSampler::kRGBA;
403 } else {
404 sc = SkScaledBitmapSampler::kRGBX;
405 }
reed@android.com11344262009-07-08 20:09:23 +0000406
407 /* We have to pass the colortable explicitly, since we may have one
408 even if our decodedBitmap doesn't, due to the request that we
409 upscale png's palette to a direct model
410 */
411 SkAutoLockColors ctLock(colorTable);
scroggo@google.com8d239242013-10-01 17:27:15 +0000412 if (!sampler.begin(decodedBitmap, sc, *this, ctLock.colors())) {
reed@android.com862e91b2009-04-28 15:27:07 +0000413 return false;
414 }
reed@android.com8a1c16f2008-12-17 15:59:43 +0000415 const int height = decodedBitmap->height();
416
reed@android.com862e91b2009-04-28 15:27:07 +0000417 if (number_passes > 1) {
418 SkAutoMalloc storage(origWidth * origHeight * srcBytesPerPixel);
419 uint8_t* base = (uint8_t*)storage.get();
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000420 size_t rowBytes = origWidth * srcBytesPerPixel;
reed@android.coma8a8b8b2009-05-04 15:00:11 +0000421
reed@android.com862e91b2009-04-28 15:27:07 +0000422 for (int i = 0; i < number_passes; i++) {
423 uint8_t* row = base;
424 for (png_uint_32 y = 0; y < origHeight; y++) {
425 uint8_t* bmRow = row;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000426 png_read_rows(png_ptr, &bmRow, png_bytepp_NULL, 1);
427 row += rowBytes;
reed@android.com862e91b2009-04-28 15:27:07 +0000428 }
reed@android.com8a1c16f2008-12-17 15:59:43 +0000429 }
reed@android.com862e91b2009-04-28 15:27:07 +0000430 // now sample it
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000431 base += sampler.srcY0() * rowBytes;
reed@android.com862e91b2009-04-28 15:27:07 +0000432 for (int y = 0; y < height; y++) {
433 reallyHasAlpha |= sampler.next(base);
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000434 base += sampler.srcDY() * rowBytes;
reed@android.com862e91b2009-04-28 15:27:07 +0000435 }
436 } else {
437 SkAutoMalloc storage(origWidth * srcBytesPerPixel);
reed@android.com8a1c16f2008-12-17 15:59:43 +0000438 uint8_t* srcRow = (uint8_t*)storage.get();
439 skip_src_rows(png_ptr, srcRow, sampler.srcY0());
440
441 for (int y = 0; y < height; y++) {
442 uint8_t* tmp = srcRow;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000443 png_read_rows(png_ptr, &tmp, png_bytepp_NULL, 1);
reed@android.com8a1c16f2008-12-17 15:59:43 +0000444 reallyHasAlpha |= sampler.next(srcRow);
445 if (y < height - 1) {
446 skip_src_rows(png_ptr, srcRow, sampler.srcDY() - 1);
447 }
448 }
reed@android.com862e91b2009-04-28 15:27:07 +0000449
reed@android.com8a1c16f2008-12-17 15:59:43 +0000450 // skip the rest of the rows (if any)
451 png_uint_32 read = (height - 1) * sampler.srcDY() +
452 sampler.srcY0() + 1;
453 SkASSERT(read <= origHeight);
454 skip_src_rows(png_ptr, srcRow, origHeight - read);
455 }
reed@android.com8a1c16f2008-12-17 15:59:43 +0000456 }
457
458 /* read rest of file, and get additional chunks in info_ptr - REQUIRED */
459 png_read_end(png_ptr, info_ptr);
460
461 if (0 != theTranspColor) {
462 reallyHasAlpha |= substituteTranspColor(decodedBitmap, theTranspColor);
463 }
scroggo@google.com5ee18dd2013-10-21 20:47:31 +0000464 if (reallyHasAlpha && this->getRequireUnpremultipliedColors()) {
reed0689d7b2014-06-14 05:30:20 -0700465 switch (decodedBitmap->colorType()) {
466 case kIndex_8_SkColorType:
scroggo@google.com5ee18dd2013-10-21 20:47:31 +0000467 // Fall through.
reed0689d7b2014-06-14 05:30:20 -0700468 case kARGB_4444_SkColorType:
469 // We have chosen not to support unpremul for these colortypes.
scroggo@google.com5ee18dd2013-10-21 20:47:31 +0000470 return false;
471 default: {
reed0689d7b2014-06-14 05:30:20 -0700472 // Fall through to finish the decode. This colortype either
scroggo@google.com5ee18dd2013-10-21 20:47:31 +0000473 // supports unpremul or it is irrelevant because it has no
474 // alpha (or only alpha).
475 // These brackets prevent a warning.
476 }
477 }
scroggo@google.com2bbc2c92013-06-14 15:33:20 +0000478 }
scroggo@google.com5ee18dd2013-10-21 20:47:31 +0000479
djsollen@google.com446cf712014-02-19 21:45:35 +0000480 if (!reallyHasAlpha) {
481 decodedBitmap->setAlphaType(kOpaque_SkAlphaType);
reed@google.com383a6972013-10-21 14:00:07 +0000482 }
reed@android.com8a1c16f2008-12-17 15:59:43 +0000483 return true;
484}
485
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000486
487
reed0689d7b2014-06-14 05:30:20 -0700488bool SkPNGImageDecoder::getBitmapColorType(png_structp png_ptr, png_infop info_ptr,
489 SkColorType* colorTypep,
490 bool* hasAlphap,
491 SkPMColor* SK_RESTRICT theTranspColorp) {
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000492 png_uint_32 origWidth, origHeight;
493 int bitDepth, colorType;
494 png_get_IHDR(png_ptr, info_ptr, &origWidth, &origHeight, &bitDepth,
495 &colorType, int_p_NULL, int_p_NULL, int_p_NULL);
496
497 // check for sBIT chunk data, in case we should disable dithering because
498 // our data is not truely 8bits per component
commit-bot@chromium.orgc1c56952013-05-02 13:41:34 +0000499 png_color_8p sig_bit;
scroggo@google.com8d239242013-10-01 17:27:15 +0000500 if (this->getDitherImage() && png_get_sBIT(png_ptr, info_ptr, &sig_bit)) {
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000501#if 0
commit-bot@chromium.orgc1c56952013-05-02 13:41:34 +0000502 SkDebugf("----- sBIT %d %d %d %d\n", sig_bit->red, sig_bit->green,
503 sig_bit->blue, sig_bit->alpha);
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000504#endif
505 // 0 seems to indicate no information available
commit-bot@chromium.orgc1c56952013-05-02 13:41:34 +0000506 if (pos_le(sig_bit->red, SK_R16_BITS) &&
507 pos_le(sig_bit->green, SK_G16_BITS) &&
508 pos_le(sig_bit->blue, SK_B16_BITS)) {
scroggo@google.com8d239242013-10-01 17:27:15 +0000509 this->setDitherImage(false);
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000510 }
511 }
512
513 if (colorType == PNG_COLOR_TYPE_PALETTE) {
514 bool paletteHasAlpha = hasTransparencyInPalette(png_ptr, info_ptr);
reed6c225732014-06-09 19:52:07 -0700515 *colorTypep = this->getPrefColorType(kIndex_SrcDepth, paletteHasAlpha);
reed0689d7b2014-06-14 05:30:20 -0700516 // now see if we can upscale to their requested colortype
reed6c225732014-06-09 19:52:07 -0700517 if (!canUpscalePaletteToConfig(*colorTypep, paletteHasAlpha)) {
518 *colorTypep = kIndex_8_SkColorType;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000519 }
520 } else {
521 png_color_16p transpColor = NULL;
522 int numTransp = 0;
523
524 png_get_tRNS(png_ptr, info_ptr, NULL, &numTransp, &transpColor);
525
526 bool valid = png_get_valid(png_ptr, info_ptr, PNG_INFO_tRNS);
527
528 if (valid && numTransp == 1 && transpColor != NULL) {
529 /* Compute our transparent color, which we'll match against later.
530 We don't really handle 16bit components properly here, since we
531 do our compare *after* the values have been knocked down to 8bit
532 which means we will find more matches than we should. The real
533 fix seems to be to see the actual 16bit components, do the
534 compare, and then knock it down to 8bits ourselves.
535 */
536 if (colorType & PNG_COLOR_MASK_COLOR) {
537 if (16 == bitDepth) {
538 *theTranspColorp = SkPackARGB32(0xFF, transpColor->red >> 8,
539 transpColor->green >> 8,
540 transpColor->blue >> 8);
541 } else {
halcanary@google.comfed30372013-10-04 12:46:45 +0000542 /* We apply the mask because in a very small
543 number of corrupt PNGs, (transpColor->red > 255)
544 and (bitDepth == 8), for certain versions of libpng. */
545 *theTranspColorp = SkPackARGB32(0xFF,
546 0xFF & (transpColor->red),
547 0xFF & (transpColor->green),
548 0xFF & (transpColor->blue));
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000549 }
550 } else { // gray
551 if (16 == bitDepth) {
552 *theTranspColorp = SkPackARGB32(0xFF, transpColor->gray >> 8,
553 transpColor->gray >> 8,
554 transpColor->gray >> 8);
555 } else {
halcanary@google.comfed30372013-10-04 12:46:45 +0000556 /* We apply the mask because in a very small
557 number of corrupt PNGs, (transpColor->red >
558 255) and (bitDepth == 8), for certain versions
559 of libpng. For safety we assume the same could
560 happen with a grayscale PNG. */
561 *theTranspColorp = SkPackARGB32(0xFF,
562 0xFF & (transpColor->gray),
563 0xFF & (transpColor->gray),
564 0xFF & (transpColor->gray));
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000565 }
566 }
567 }
568
569 if (valid ||
570 PNG_COLOR_TYPE_RGB_ALPHA == colorType ||
571 PNG_COLOR_TYPE_GRAY_ALPHA == colorType) {
572 *hasAlphap = true;
573 }
scroggo@google.comc70a3aa2013-07-18 20:03:15 +0000574
575 SrcDepth srcDepth = k32Bit_SrcDepth;
576 if (PNG_COLOR_TYPE_GRAY == colorType) {
577 srcDepth = k8BitGray_SrcDepth;
scroggo@google.com8e2ef012013-07-18 20:14:45 +0000578 // Remove this assert, which fails on desk_pokemonwiki.skp
579 //SkASSERT(!*hasAlphap);
scroggo@google.comc70a3aa2013-07-18 20:03:15 +0000580 }
581
reed6c225732014-06-09 19:52:07 -0700582 *colorTypep = this->getPrefColorType(srcDepth, *hasAlphap);
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000583 // now match the request against our capabilities
584 if (*hasAlphap) {
reed6c225732014-06-09 19:52:07 -0700585 if (*colorTypep != kARGB_4444_SkColorType) {
586 *colorTypep = kN32_SkColorType;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000587 }
588 } else {
reed6c225732014-06-09 19:52:07 -0700589 if (kAlpha_8_SkColorType == *colorTypep) {
scroggo@google.com354fd972013-10-02 15:50:19 +0000590 if (k8BitGray_SrcDepth != srcDepth) {
591 // Converting a non grayscale image to A8 is not currently supported.
reed6c225732014-06-09 19:52:07 -0700592 *colorTypep = kN32_SkColorType;
scroggo@google.com354fd972013-10-02 15:50:19 +0000593 }
reed6c225732014-06-09 19:52:07 -0700594 } else if (*colorTypep != kRGB_565_SkColorType &&
595 *colorTypep != kARGB_4444_SkColorType) {
596 *colorTypep = kN32_SkColorType;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000597 }
598 }
599 }
600
601 // sanity check for size
602 {
reed@google.com57212f92013-12-30 14:40:38 +0000603 int64_t size = sk_64_mul(origWidth, origHeight);
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000604 // now check that if we are 4-bytes per pixel, we also don't overflow
reed@google.com57212f92013-12-30 14:40:38 +0000605 if (size < 0 || size > (0x7FFFFFFF >> 2)) {
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000606 return false;
607 }
608 }
609
reed5926b862014-06-11 10:33:13 -0700610#ifdef SK_SUPPORT_LEGACY_IMAGEDECODER_CHOOSER
reed6c225732014-06-09 19:52:07 -0700611 if (!this->chooseFromOneChoice(*colorTypep, origWidth, origHeight)) {
scroggo@google.com2bbc2c92013-06-14 15:33:20 +0000612 return false;
613 }
reed5926b862014-06-11 10:33:13 -0700614#endif
scroggo@google.com2bbc2c92013-06-14 15:33:20 +0000615
616 // If the image has alpha and the decoder wants unpremultiplied
reed0689d7b2014-06-14 05:30:20 -0700617 // colors, the only supported colortype is 8888.
scroggo@google.com2bbc2c92013-06-14 15:33:20 +0000618 if (this->getRequireUnpremultipliedColors() && *hasAlphap) {
reed6c225732014-06-09 19:52:07 -0700619 *colorTypep = kN32_SkColorType;
scroggo@google.com2bbc2c92013-06-14 15:33:20 +0000620 }
scroggo@google.comc70a3aa2013-07-18 20:03:15 +0000621
622 if (fImageIndex != NULL) {
reed6c225732014-06-09 19:52:07 -0700623 if (kUnknown_SkColorType == fImageIndex->fColorType) {
scroggo@google.comc70a3aa2013-07-18 20:03:15 +0000624 // This is the first time for this subset decode. From now on,
reed0689d7b2014-06-14 05:30:20 -0700625 // all decodes must be in the same colortype.
reed6c225732014-06-09 19:52:07 -0700626 fImageIndex->fColorType = *colorTypep;
627 } else if (fImageIndex->fColorType != *colorTypep) {
628 // Requesting a different colortype for a subsequent decode is not
scroggo@google.comc70a3aa2013-07-18 20:03:15 +0000629 // supported. Report failure before we make changes to png_ptr.
630 return false;
631 }
632 }
633
reed6c225732014-06-09 19:52:07 -0700634 bool convertGrayToRGB = PNG_COLOR_TYPE_GRAY == colorType && *colorTypep != kAlpha_8_SkColorType;
scroggo@google.comc70a3aa2013-07-18 20:03:15 +0000635
636 // Unless the user is requesting A8, convert a grayscale image into RGB.
637 // GRAY_ALPHA will always be converted to RGB
638 if (convertGrayToRGB || colorType == PNG_COLOR_TYPE_GRAY_ALPHA) {
639 png_set_gray_to_rgb(png_ptr);
640 }
641
642 // Add filler (or alpha) byte (after each RGB triplet) if necessary.
643 if (colorType == PNG_COLOR_TYPE_RGB || convertGrayToRGB) {
644 png_set_filler(png_ptr, 0xff, PNG_FILLER_AFTER);
645 }
646
scroggo@google.com2bbc2c92013-06-14 15:33:20 +0000647 return true;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000648}
649
scroggo@google.com2bbc2c92013-06-14 15:33:20 +0000650typedef uint32_t (*PackColorProc)(U8CPU a, U8CPU r, U8CPU g, U8CPU b);
651
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000652bool SkPNGImageDecoder::decodePalette(png_structp png_ptr, png_infop info_ptr,
653 bool *hasAlphap, bool *reallyHasAlphap,
654 SkColorTable **colorTablep) {
655 int numPalette;
656 png_colorp palette;
657 png_bytep trans;
658 int numTrans;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000659
660 png_get_PLTE(png_ptr, info_ptr, &palette, &numPalette);
661
662 /* BUGGY IMAGE WORKAROUND
663
664 We hit some images (e.g. fruit_.png) who contain bytes that are == colortable_count
665 which is a problem since we use the byte as an index. To work around this we grow
666 the colortable by 1 (if its < 256) and duplicate the last color into that slot.
667 */
668 int colorCount = numPalette + (numPalette < 256);
reed@google.com0a6151d2013-10-10 14:44:56 +0000669 SkPMColor colorStorage[256]; // worst-case storage
670 SkPMColor* colorPtr = colorStorage;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000671
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000672 if (png_get_valid(png_ptr, info_ptr, PNG_INFO_tRNS)) {
673 png_get_tRNS(png_ptr, info_ptr, &trans, &numTrans, NULL);
674 *hasAlphap = (numTrans > 0);
675 } else {
676 numTrans = 0;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000677 }
reed@google.com0a6151d2013-10-10 14:44:56 +0000678
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000679 // check for bad images that might make us crash
680 if (numTrans > numPalette) {
681 numTrans = numPalette;
682 }
683
684 int index = 0;
685 int transLessThanFF = 0;
686
scroggo@google.com2bbc2c92013-06-14 15:33:20 +0000687 // Choose which function to use to create the color table. If the final destination's
reed0689d7b2014-06-14 05:30:20 -0700688 // colortype is unpremultiplied, the color table will store unpremultiplied colors.
scroggo@google.com2bbc2c92013-06-14 15:33:20 +0000689 PackColorProc proc;
690 if (this->getRequireUnpremultipliedColors()) {
691 proc = &SkPackARGB32NoCheck;
692 } else {
693 proc = &SkPreMultiplyARGB;
694 }
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000695 for (; index < numTrans; index++) {
696 transLessThanFF |= (int)*trans - 0xFF;
scroggo@google.com2bbc2c92013-06-14 15:33:20 +0000697 *colorPtr++ = proc(*trans++, palette->red, palette->green, palette->blue);
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000698 palette++;
699 }
reed@google.com0a6151d2013-10-10 14:44:56 +0000700 bool reallyHasAlpha = (transLessThanFF < 0);
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000701
702 for (; index < numPalette; index++) {
703 *colorPtr++ = SkPackARGB32(0xFF, palette->red, palette->green, palette->blue);
704 palette++;
705 }
706
707 // see BUGGY IMAGE WORKAROUND comment above
708 if (numPalette < 256) {
709 *colorPtr = colorPtr[-1];
710 }
reed@google.com0a6151d2013-10-10 14:44:56 +0000711
712 SkAlphaType alphaType = kOpaque_SkAlphaType;
713 if (reallyHasAlpha) {
714 if (this->getRequireUnpremultipliedColors()) {
715 alphaType = kUnpremul_SkAlphaType;
716 } else {
717 alphaType = kPremul_SkAlphaType;
718 }
719 }
720
721 *colorTablep = SkNEW_ARGS(SkColorTable,
722 (colorStorage, colorCount, alphaType));
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000723 *reallyHasAlphap = reallyHasAlpha;
724 return true;
725}
726
727#ifdef SK_BUILD_FOR_ANDROID
728
scroggo@google.comb5571b32013-09-25 21:34:24 +0000729bool SkPNGImageDecoder::onBuildTileIndex(SkStreamRewindable* sk_stream, int *width, int *height) {
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000730 png_structp png_ptr;
731 png_infop info_ptr;
732
733 if (!onDecodeInit(sk_stream, &png_ptr, &info_ptr)) {
734 return false;
735 }
736
737 if (setjmp(png_jmpbuf(png_ptr)) != 0) {
738 png_destroy_read_struct(&png_ptr, &info_ptr, png_infopp_NULL);
739 return false;
740 }
741
742 png_uint_32 origWidth, origHeight;
743 int bitDepth, colorType;
744 png_get_IHDR(png_ptr, info_ptr, &origWidth, &origHeight, &bitDepth,
745 &colorType, int_p_NULL, int_p_NULL, int_p_NULL);
746
747 *width = origWidth;
748 *height = origHeight;
749
750 png_build_index(png_ptr);
751
752 if (fImageIndex) {
753 SkDELETE(fImageIndex);
754 }
scroggo@google.comc70a3aa2013-07-18 20:03:15 +0000755 fImageIndex = SkNEW_ARGS(SkPNGImageIndex, (sk_stream, png_ptr, info_ptr));
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000756
757 return true;
758}
759
scroggo@google.com7e6fcee2013-05-03 20:14:28 +0000760bool SkPNGImageDecoder::onDecodeSubset(SkBitmap* bm, const SkIRect& region) {
761 if (NULL == fImageIndex) {
762 return false;
763 }
764
scroggo@google.comc70a3aa2013-07-18 20:03:15 +0000765 png_structp png_ptr = fImageIndex->fPng_ptr;
766 png_infop info_ptr = fImageIndex->fInfo_ptr;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000767 if (setjmp(png_jmpbuf(png_ptr))) {
768 return false;
769 }
770
771 png_uint_32 origWidth, origHeight;
reed6c225732014-06-09 19:52:07 -0700772 int bitDepth, pngColorType, interlaceType;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000773 png_get_IHDR(png_ptr, info_ptr, &origWidth, &origHeight, &bitDepth,
reed6c225732014-06-09 19:52:07 -0700774 &pngColorType, &interlaceType, int_p_NULL, int_p_NULL);
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000775
776 SkIRect rect = SkIRect::MakeWH(origWidth, origHeight);
777
778 if (!rect.intersect(region)) {
scroggo@google.com7e6fcee2013-05-03 20:14:28 +0000779 // If the requested region is entirely outside the image, just
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000780 // returns false
781 return false;
782 }
783
reed6c225732014-06-09 19:52:07 -0700784 SkColorType colorType;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000785 bool hasAlpha = false;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000786 SkPMColor theTranspColor = 0; // 0 tells us not to try to match
787
reed0689d7b2014-06-14 05:30:20 -0700788 if (!this->getBitmapColorType(png_ptr, info_ptr, &colorType, &hasAlpha, &theTranspColor)) {
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000789 return false;
790 }
791
792 const int sampleSize = this->getSampleSize();
793 SkScaledBitmapSampler sampler(origWidth, rect.height(), sampleSize);
794
795 SkBitmap decodedBitmap;
reed6c225732014-06-09 19:52:07 -0700796 decodedBitmap.setInfo(SkImageInfo::Make(sampler.scaledWidth(), sampler.scaledHeight(),
797 colorType, kPremul_SkAlphaType));
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000798
799 // from here down we are concerned with colortables and pixels
800
801 // we track if we actually see a non-opaque pixels, since sometimes a PNG sets its colortype
802 // to |= PNG_COLOR_MASK_ALPHA, but all of its pixels are in fact opaque. We care, since we
803 // draw lots faster if we can flag the bitmap has being opaque
804 bool reallyHasAlpha = false;
805 SkColorTable* colorTable = NULL;
806
reed6c225732014-06-09 19:52:07 -0700807 if (pngColorType == PNG_COLOR_TYPE_PALETTE) {
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000808 decodePalette(png_ptr, info_ptr, &hasAlpha, &reallyHasAlpha, &colorTable);
809 }
810
811 SkAutoUnref aur(colorTable);
812
813 // Check ahead of time if the swap(dest, src) is possible.
814 // If yes, then we will stick to AllocPixelRef since it's cheaper with the swap happening.
815 // If no, then we will use alloc to allocate pixels to prevent garbage collection.
816 int w = rect.width() / sampleSize;
817 int h = rect.height() / sampleSize;
818 const bool swapOnly = (rect == region) && (w == decodedBitmap.width()) &&
819 (h == decodedBitmap.height()) && bm->isNull();
reed6c225732014-06-09 19:52:07 -0700820 const bool needColorTable = kIndex_8_SkColorType == colorType;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000821 if (swapOnly) {
822 if (!this->allocPixelRef(&decodedBitmap, needColorTable ? colorTable : NULL)) {
823 return false;
824 }
825 } else {
826 if (!decodedBitmap.allocPixels(NULL, needColorTable ? colorTable : NULL)) {
827 return false;
828 }
829 }
830 SkAutoLockPixels alp(decodedBitmap);
831
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000832 /* Turn on interlace handling. REQUIRED if you are not using
833 * png_read_image(). To see how to handle interlacing passes,
834 * see the png_read_row() method below:
835 */
836 const int number_passes = (interlaceType != PNG_INTERLACE_NONE) ?
837 png_set_interlace_handling(png_ptr) : 1;
838
839 /* Optional call to gamma correct and add the background to the palette
840 * and update info structure. REQUIRED if you are expecting libpng to
841 * update the palette for you (ie you selected such a transform above).
842 */
commit-bot@chromium.orgc1c56952013-05-02 13:41:34 +0000843
844 // Direct access to png_ptr fields is deprecated in libpng > 1.2.
845#if defined(PNG_1_0_X) || defined (PNG_1_2_X)
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000846 png_ptr->pass = 0;
commit-bot@chromium.orgc1c56952013-05-02 13:41:34 +0000847#else
scroggo@google.com7e6fcee2013-05-03 20:14:28 +0000848 // FIXME: This sets pass as desired, but also sets iwidth. Is that ok?
849 png_set_interlaced_pass(png_ptr, 0);
commit-bot@chromium.orgc1c56952013-05-02 13:41:34 +0000850#endif
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000851 png_read_update_info(png_ptr, info_ptr);
852
853 int actualTop = rect.fTop;
854
reed6c225732014-06-09 19:52:07 -0700855 if ((kAlpha_8_SkColorType == colorType || kIndex_8_SkColorType == colorType)
scroggo@google.comc70a3aa2013-07-18 20:03:15 +0000856 && 1 == sampleSize) {
reed6c225732014-06-09 19:52:07 -0700857 if (kAlpha_8_SkColorType == colorType) {
scroggo@google.com5ee18dd2013-10-21 20:47:31 +0000858 // For an A8 bitmap, we assume there is an alpha for speed. It is
859 // possible the bitmap is opaque, but that is an unlikely use case
860 // since it would not be very interesting.
861 reallyHasAlpha = true;
862 // A8 is only allowed if the original was GRAY.
reed6c225732014-06-09 19:52:07 -0700863 SkASSERT(PNG_COLOR_TYPE_GRAY == pngColorType);
scroggo@google.com5ee18dd2013-10-21 20:47:31 +0000864 }
scroggo@google.comc70a3aa2013-07-18 20:03:15 +0000865
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000866 for (int i = 0; i < number_passes; i++) {
867 png_configure_decoder(png_ptr, &actualTop, i);
868 for (int j = 0; j < rect.fTop - actualTop; j++) {
869 uint8_t* bmRow = decodedBitmap.getAddr8(0, 0);
870 png_read_rows(png_ptr, &bmRow, png_bytepp_NULL, 1);
871 }
scroggo@google.com7e6fcee2013-05-03 20:14:28 +0000872 png_uint_32 bitmapHeight = (png_uint_32) decodedBitmap.height();
873 for (png_uint_32 y = 0; y < bitmapHeight; y++) {
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000874 uint8_t* bmRow = decodedBitmap.getAddr8(0, y);
875 png_read_rows(png_ptr, &bmRow, png_bytepp_NULL, 1);
876 }
877 }
878 } else {
879 SkScaledBitmapSampler::SrcConfig sc;
880 int srcBytesPerPixel = 4;
881
882 if (colorTable != NULL) {
883 sc = SkScaledBitmapSampler::kIndex;
884 srcBytesPerPixel = 1;
reed6c225732014-06-09 19:52:07 -0700885 } else if (kAlpha_8_SkColorType == colorType) {
scroggo@google.comc70a3aa2013-07-18 20:03:15 +0000886 // A8 is only allowed if the original was GRAY.
reed6c225732014-06-09 19:52:07 -0700887 SkASSERT(PNG_COLOR_TYPE_GRAY == pngColorType);
scroggo@google.comc70a3aa2013-07-18 20:03:15 +0000888 sc = SkScaledBitmapSampler::kGray;
889 srcBytesPerPixel = 1;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000890 } else if (hasAlpha) {
891 sc = SkScaledBitmapSampler::kRGBA;
892 } else {
893 sc = SkScaledBitmapSampler::kRGBX;
894 }
895
896 /* We have to pass the colortable explicitly, since we may have one
897 even if our decodedBitmap doesn't, due to the request that we
898 upscale png's palette to a direct model
899 */
900 SkAutoLockColors ctLock(colorTable);
scroggo@google.com8d239242013-10-01 17:27:15 +0000901 if (!sampler.begin(&decodedBitmap, sc, *this, ctLock.colors())) {
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000902 return false;
903 }
904 const int height = decodedBitmap.height();
905
906 if (number_passes > 1) {
907 SkAutoMalloc storage(origWidth * origHeight * srcBytesPerPixel);
908 uint8_t* base = (uint8_t*)storage.get();
909 size_t rb = origWidth * srcBytesPerPixel;
910
911 for (int i = 0; i < number_passes; i++) {
912 png_configure_decoder(png_ptr, &actualTop, i);
913 for (int j = 0; j < rect.fTop - actualTop; j++) {
914 uint8_t* bmRow = (uint8_t*)decodedBitmap.getPixels();
915 png_read_rows(png_ptr, &bmRow, png_bytepp_NULL, 1);
916 }
917 uint8_t* row = base;
918 for (int32_t y = 0; y < rect.height(); y++) {
919 uint8_t* bmRow = row;
920 png_read_rows(png_ptr, &bmRow, png_bytepp_NULL, 1);
921 row += rb;
922 }
923 }
924 // now sample it
925 base += sampler.srcY0() * rb;
926 for (int y = 0; y < height; y++) {
927 reallyHasAlpha |= sampler.next(base);
928 base += sampler.srcDY() * rb;
929 }
930 } else {
931 SkAutoMalloc storage(origWidth * srcBytesPerPixel);
932 uint8_t* srcRow = (uint8_t*)storage.get();
933
934 png_configure_decoder(png_ptr, &actualTop, 0);
935 skip_src_rows(png_ptr, srcRow, sampler.srcY0());
936
937 for (int i = 0; i < rect.fTop - actualTop; i++) {
938 uint8_t* bmRow = (uint8_t*)decodedBitmap.getPixels();
939 png_read_rows(png_ptr, &bmRow, png_bytepp_NULL, 1);
940 }
941 for (int y = 0; y < height; y++) {
942 uint8_t* tmp = srcRow;
943 png_read_rows(png_ptr, &tmp, png_bytepp_NULL, 1);
944 reallyHasAlpha |= sampler.next(srcRow);
945 if (y < height - 1) {
946 skip_src_rows(png_ptr, srcRow, sampler.srcDY() - 1);
947 }
948 }
949 }
950 }
951
952 if (0 != theTranspColor) {
953 reallyHasAlpha |= substituteTranspColor(&decodedBitmap, theTranspColor);
954 }
scroggo@google.com5ee18dd2013-10-21 20:47:31 +0000955 if (reallyHasAlpha && this->getRequireUnpremultipliedColors()) {
reed6c225732014-06-09 19:52:07 -0700956 switch (decodedBitmap.colorType()) {
957 case kIndex_8_SkColorType:
scroggo@google.com5ee18dd2013-10-21 20:47:31 +0000958 // Fall through.
reed6c225732014-06-09 19:52:07 -0700959 case kARGB_4444_SkColorType:
960 // We have chosen not to support unpremul for these colortypess.
scroggo@google.com5ee18dd2013-10-21 20:47:31 +0000961 return false;
962 default: {
963 // Fall through to finish the decode. This config either
964 // supports unpremul or it is irrelevant because it has no
965 // alpha (or only alpha).
966 // These brackets prevent a warning.
967 }
968 }
commit-bot@chromium.org546f70c2013-10-03 17:13:38 +0000969 }
reed@google.com383a6972013-10-21 14:00:07 +0000970 SkAlphaType alphaType = kOpaque_SkAlphaType;
971 if (reallyHasAlpha) {
972 if (this->getRequireUnpremultipliedColors()) {
973 alphaType = kUnpremul_SkAlphaType;
974 } else {
975 alphaType = kPremul_SkAlphaType;
976 }
977 }
978 decodedBitmap.setAlphaType(alphaType);
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000979
980 if (swapOnly) {
981 bm->swap(decodedBitmap);
scroggo@google.com7e6fcee2013-05-03 20:14:28 +0000982 return true;
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000983 }
scroggo@google.com7e6fcee2013-05-03 20:14:28 +0000984 return this->cropBitmap(bm, &decodedBitmap, sampleSize, region.x(), region.y(),
985 region.width(), region.height(), 0, rect.y());
commit-bot@chromium.orga936e372013-03-14 14:42:18 +0000986}
987#endif
988
reed@android.com8a1c16f2008-12-17 15:59:43 +0000989///////////////////////////////////////////////////////////////////////////////
990
reed@android.com8a1c16f2008-12-17 15:59:43 +0000991#include "SkColorPriv.h"
992#include "SkUnPreMultiply.h"
993
994static void sk_write_fn(png_structp png_ptr, png_bytep data, png_size_t len) {
commit-bot@chromium.orgc1c56952013-05-02 13:41:34 +0000995 SkWStream* sk_stream = (SkWStream*)png_get_io_ptr(png_ptr);
reed@android.com8a1c16f2008-12-17 15:59:43 +0000996 if (!sk_stream->write(data, len)) {
997 png_error(png_ptr, "sk_write_fn Error!");
998 }
999}
1000
reed0689d7b2014-06-14 05:30:20 -07001001static transform_scanline_proc choose_proc(SkColorType ct, bool hasAlpha) {
reed@android.com8a1c16f2008-12-17 15:59:43 +00001002 // we don't care about search on alpha if we're kIndex8, since only the
1003 // colortable packing cares about that distinction, not the pixels
reed0689d7b2014-06-14 05:30:20 -07001004 if (kIndex_8_SkColorType == ct) {
reed@android.com8a1c16f2008-12-17 15:59:43 +00001005 hasAlpha = false; // we store false in the table entries for kIndex8
1006 }
rmistry@google.comd6176b02012-08-23 18:14:13 +00001007
reed@android.com8a1c16f2008-12-17 15:59:43 +00001008 static const struct {
reed0689d7b2014-06-14 05:30:20 -07001009 SkColorType fColorType;
reed@android.com8a1c16f2008-12-17 15:59:43 +00001010 bool fHasAlpha;
1011 transform_scanline_proc fProc;
1012 } gMap[] = {
reed0689d7b2014-06-14 05:30:20 -07001013 { kRGB_565_SkColorType, false, transform_scanline_565 },
1014 { kN32_SkColorType, false, transform_scanline_888 },
1015 { kN32_SkColorType, true, transform_scanline_8888 },
1016 { kARGB_4444_SkColorType, false, transform_scanline_444 },
1017 { kARGB_4444_SkColorType, true, transform_scanline_4444 },
1018 { kIndex_8_SkColorType, false, transform_scanline_memcpy },
reed@android.com8a1c16f2008-12-17 15:59:43 +00001019 };
1020
1021 for (int i = SK_ARRAY_COUNT(gMap) - 1; i >= 0; --i) {
reed0689d7b2014-06-14 05:30:20 -07001022 if (gMap[i].fColorType == ct && gMap[i].fHasAlpha == hasAlpha) {
reed@android.com8a1c16f2008-12-17 15:59:43 +00001023 return gMap[i].fProc;
1024 }
1025 }
1026 sk_throw();
1027 return NULL;
1028}
1029
1030// return the minimum legal bitdepth (by png standards) for this many colortable
1031// entries. SkBitmap always stores in 8bits per pixel, but for colorcount <= 16,
1032// we can use fewer bits per in png
1033static int computeBitDepth(int colorCount) {
1034#if 0
1035 int bits = SkNextLog2(colorCount);
1036 SkASSERT(bits >= 1 && bits <= 8);
1037 // now we need bits itself to be a power of 2 (e.g. 1, 2, 4, 8)
1038 return SkNextPow2(bits);
1039#else
1040 // for the moment, we don't know how to pack bitdepth < 8
1041 return 8;
1042#endif
1043}
1044
1045/* Pack palette[] with the corresponding colors, and if hasAlpha is true, also
1046 pack trans[] and return the number of trans[] entries written. If hasAlpha
1047 is false, the return value will always be 0.
rmistry@google.comd6176b02012-08-23 18:14:13 +00001048
reed@android.com8a1c16f2008-12-17 15:59:43 +00001049 Note: this routine takes care of unpremultiplying the RGB values when we
1050 have alpha in the colortable, since png doesn't support premul colors
1051*/
reed@android.com6f252972009-01-14 16:46:16 +00001052static inline int pack_palette(SkColorTable* ctable,
reed@android.comb50a60c2009-01-14 17:51:08 +00001053 png_color* SK_RESTRICT palette,
1054 png_byte* SK_RESTRICT trans, bool hasAlpha) {
reed@android.com8a1c16f2008-12-17 15:59:43 +00001055 SkAutoLockColors alc(ctable);
1056 const SkPMColor* SK_RESTRICT colors = alc.colors();
1057 const int ctCount = ctable->count();
1058 int i, num_trans = 0;
1059
1060 if (hasAlpha) {
1061 /* first see if we have some number of fully opaque at the end of the
1062 ctable. PNG allows num_trans < num_palette, but all of the trans
1063 entries must come first in the palette. If I was smarter, I'd
1064 reorder the indices and ctable so that all non-opaque colors came
1065 first in the palette. But, since that would slow down the encode,
1066 I'm leaving the indices and ctable order as is, and just looking
1067 at the tail of the ctable for opaqueness.
1068 */
1069 num_trans = ctCount;
1070 for (i = ctCount - 1; i >= 0; --i) {
1071 if (SkGetPackedA32(colors[i]) != 0xFF) {
1072 break;
1073 }
1074 num_trans -= 1;
1075 }
rmistry@google.comd6176b02012-08-23 18:14:13 +00001076
reed@android.com8a1c16f2008-12-17 15:59:43 +00001077 const SkUnPreMultiply::Scale* SK_RESTRICT table =
1078 SkUnPreMultiply::GetScaleTable();
1079
1080 for (i = 0; i < num_trans; i++) {
1081 const SkPMColor c = *colors++;
1082 const unsigned a = SkGetPackedA32(c);
1083 const SkUnPreMultiply::Scale s = table[a];
1084 trans[i] = a;
1085 palette[i].red = SkUnPreMultiply::ApplyScale(s, SkGetPackedR32(c));
1086 palette[i].green = SkUnPreMultiply::ApplyScale(s,SkGetPackedG32(c));
1087 palette[i].blue = SkUnPreMultiply::ApplyScale(s, SkGetPackedB32(c));
rmistry@google.comd6176b02012-08-23 18:14:13 +00001088 }
reed@android.com8a1c16f2008-12-17 15:59:43 +00001089 // now fall out of this if-block to use common code for the trailing
1090 // opaque entries
1091 }
rmistry@google.comd6176b02012-08-23 18:14:13 +00001092
reed@android.com8a1c16f2008-12-17 15:59:43 +00001093 // these (remaining) entries are opaque
1094 for (i = num_trans; i < ctCount; i++) {
1095 SkPMColor c = *colors++;
1096 palette[i].red = SkGetPackedR32(c);
1097 palette[i].green = SkGetPackedG32(c);
1098 palette[i].blue = SkGetPackedB32(c);
1099 }
1100 return num_trans;
1101}
1102
1103class SkPNGImageEncoder : public SkImageEncoder {
1104protected:
commit-bot@chromium.orga936e372013-03-14 14:42:18 +00001105 virtual bool onEncode(SkWStream* stream, const SkBitmap& bm, int quality) SK_OVERRIDE;
tomhudson@google.com5c210c72011-07-28 21:06:40 +00001106private:
1107 bool doEncode(SkWStream* stream, const SkBitmap& bm,
1108 const bool& hasAlpha, int colorType,
reed0689d7b2014-06-14 05:30:20 -07001109 int bitDepth, SkColorType ct,
tomhudson@google.com5c210c72011-07-28 21:06:40 +00001110 png_color_8& sig_bit);
commit-bot@chromium.orga936e372013-03-14 14:42:18 +00001111
1112 typedef SkImageEncoder INHERITED;
reed@android.com8a1c16f2008-12-17 15:59:43 +00001113};
1114
reed0689d7b2014-06-14 05:30:20 -07001115bool SkPNGImageEncoder::onEncode(SkWStream* stream, const SkBitmap& bitmap, int /*quality*/) {
1116 SkColorType ct = bitmap.colorType();
reed@android.com8a1c16f2008-12-17 15:59:43 +00001117
1118 const bool hasAlpha = !bitmap.isOpaque();
1119 int colorType = PNG_COLOR_MASK_COLOR;
1120 int bitDepth = 8; // default for color
1121 png_color_8 sig_bit;
1122
reed0689d7b2014-06-14 05:30:20 -07001123 switch (ct) {
1124 case kIndex_8_SkColorType:
reed@android.com8a1c16f2008-12-17 15:59:43 +00001125 colorType |= PNG_COLOR_MASK_PALETTE;
1126 // fall through to the ARGB_8888 case
reed0689d7b2014-06-14 05:30:20 -07001127 case kN32_SkColorType:
reed@android.com8a1c16f2008-12-17 15:59:43 +00001128 sig_bit.red = 8;
1129 sig_bit.green = 8;
1130 sig_bit.blue = 8;
1131 sig_bit.alpha = 8;
1132 break;
reed0689d7b2014-06-14 05:30:20 -07001133 case kARGB_4444_SkColorType:
reed@android.com8a1c16f2008-12-17 15:59:43 +00001134 sig_bit.red = 4;
1135 sig_bit.green = 4;
1136 sig_bit.blue = 4;
1137 sig_bit.alpha = 4;
1138 break;
reed0689d7b2014-06-14 05:30:20 -07001139 case kRGB_565_SkColorType:
reed@android.com8a1c16f2008-12-17 15:59:43 +00001140 sig_bit.red = 5;
1141 sig_bit.green = 6;
1142 sig_bit.blue = 5;
1143 sig_bit.alpha = 0;
1144 break;
1145 default:
1146 return false;
1147 }
rmistry@google.comd6176b02012-08-23 18:14:13 +00001148
reed@android.com8a1c16f2008-12-17 15:59:43 +00001149 if (hasAlpha) {
1150 // don't specify alpha if we're a palette, even if our ctable has alpha
1151 if (!(colorType & PNG_COLOR_MASK_PALETTE)) {
1152 colorType |= PNG_COLOR_MASK_ALPHA;
1153 }
1154 } else {
1155 sig_bit.alpha = 0;
1156 }
rmistry@google.comd6176b02012-08-23 18:14:13 +00001157
reed@android.com8a1c16f2008-12-17 15:59:43 +00001158 SkAutoLockPixels alp(bitmap);
1159 // readyToDraw checks for pixels (and colortable if that is required)
1160 if (!bitmap.readyToDraw()) {
1161 return false;
1162 }
1163
1164 // we must do this after we have locked the pixels
1165 SkColorTable* ctable = bitmap.getColorTable();
1166 if (NULL != ctable) {
1167 if (ctable->count() == 0) {
1168 return false;
1169 }
1170 // check if we can store in fewer than 8 bits
1171 bitDepth = computeBitDepth(ctable->count());
1172 }
1173
reed0689d7b2014-06-14 05:30:20 -07001174 return doEncode(stream, bitmap, hasAlpha, colorType, bitDepth, ct, sig_bit);
tomhudson@google.com5c210c72011-07-28 21:06:40 +00001175}
1176
1177bool SkPNGImageEncoder::doEncode(SkWStream* stream, const SkBitmap& bitmap,
1178 const bool& hasAlpha, int colorType,
reed0689d7b2014-06-14 05:30:20 -07001179 int bitDepth, SkColorType ct,
tomhudson@google.com5c210c72011-07-28 21:06:40 +00001180 png_color_8& sig_bit) {
1181
reed@android.com8a1c16f2008-12-17 15:59:43 +00001182 png_structp png_ptr;
1183 png_infop info_ptr;
1184
1185 png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, sk_error_fn,
1186 NULL);
1187 if (NULL == png_ptr) {
1188 return false;
1189 }
1190
1191 info_ptr = png_create_info_struct(png_ptr);
1192 if (NULL == info_ptr) {
commit-bot@chromium.orga936e372013-03-14 14:42:18 +00001193 png_destroy_write_struct(&png_ptr, png_infopp_NULL);
reed@android.com8a1c16f2008-12-17 15:59:43 +00001194 return false;
1195 }
1196
1197 /* Set error handling. REQUIRED if you aren't supplying your own
1198 * error handling functions in the png_create_write_struct() call.
1199 */
1200 if (setjmp(png_jmpbuf(png_ptr))) {
1201 png_destroy_write_struct(&png_ptr, &info_ptr);
1202 return false;
1203 }
1204
commit-bot@chromium.orga936e372013-03-14 14:42:18 +00001205 png_set_write_fn(png_ptr, (void*)stream, sk_write_fn, png_flush_ptr_NULL);
reed@android.com8a1c16f2008-12-17 15:59:43 +00001206
1207 /* Set the image information here. Width and height are up to 2^31,
1208 * bit_depth is one of 1, 2, 4, 8, or 16, but valid values also depend on
1209 * the color_type selected. color_type is one of PNG_COLOR_TYPE_GRAY,
1210 * PNG_COLOR_TYPE_GRAY_ALPHA, PNG_COLOR_TYPE_PALETTE, PNG_COLOR_TYPE_RGB,
1211 * or PNG_COLOR_TYPE_RGB_ALPHA. interlace is either PNG_INTERLACE_NONE or
1212 * PNG_INTERLACE_ADAM7, and the compression_type and filter_type MUST
1213 * currently be PNG_COMPRESSION_TYPE_BASE and PNG_FILTER_TYPE_BASE. REQUIRED
1214 */
1215
1216 png_set_IHDR(png_ptr, info_ptr, bitmap.width(), bitmap.height(),
1217 bitDepth, colorType,
1218 PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_BASE,
1219 PNG_FILTER_TYPE_BASE);
1220
reed@android.com61898772009-07-07 19:38:01 +00001221 // set our colortable/trans arrays if needed
1222 png_color paletteColors[256];
1223 png_byte trans[256];
reed0689d7b2014-06-14 05:30:20 -07001224 if (kIndex_8_SkColorType == ct) {
reed@android.com61898772009-07-07 19:38:01 +00001225 SkColorTable* ct = bitmap.getColorTable();
1226 int numTrans = pack_palette(ct, paletteColors, trans, hasAlpha);
1227 png_set_PLTE(png_ptr, info_ptr, paletteColors, ct->count());
1228 if (numTrans > 0) {
1229 png_set_tRNS(png_ptr, info_ptr, trans, numTrans, NULL);
1230 }
1231 }
reed@android.com8a1c16f2008-12-17 15:59:43 +00001232
1233 png_set_sBIT(png_ptr, info_ptr, &sig_bit);
1234 png_write_info(png_ptr, info_ptr);
1235
1236 const char* srcImage = (const char*)bitmap.getPixels();
1237 SkAutoSMalloc<1024> rowStorage(bitmap.width() << 2);
1238 char* storage = (char*)rowStorage.get();
reed0689d7b2014-06-14 05:30:20 -07001239 transform_scanline_proc proc = choose_proc(ct, hasAlpha);
reed@android.com8a1c16f2008-12-17 15:59:43 +00001240
1241 for (int y = 0; y < bitmap.height(); y++) {
1242 png_bytep row_ptr = (png_bytep)storage;
1243 proc(srcImage, bitmap.width(), storage);
1244 png_write_rows(png_ptr, &row_ptr, 1);
1245 srcImage += bitmap.rowBytes();
1246 }
1247
1248 png_write_end(png_ptr, info_ptr);
1249
1250 /* clean up after the write, and free any memory allocated */
1251 png_destroy_write_struct(&png_ptr, &info_ptr);
1252 return true;
1253}
1254
reed@android.com00bf85a2009-01-22 13:04:56 +00001255///////////////////////////////////////////////////////////////////////////////
robertphillips@google.comec51cb82012-03-23 18:13:47 +00001256DEFINE_DECODER_CREATOR(PNGImageDecoder);
1257DEFINE_ENCODER_CREATOR(PNGImageEncoder);
1258///////////////////////////////////////////////////////////////////////////////
reed@android.com00bf85a2009-01-22 13:04:56 +00001259
scroggo@google.comb5571b32013-09-25 21:34:24 +00001260static bool is_png(SkStreamRewindable* stream) {
reed@android.com00bf85a2009-01-22 13:04:56 +00001261 char buf[PNG_BYTES_TO_CHECK];
1262 if (stream->read(buf, PNG_BYTES_TO_CHECK) == PNG_BYTES_TO_CHECK &&
1263 !png_sig_cmp((png_bytep) buf, (png_size_t)0, PNG_BYTES_TO_CHECK)) {
scroggo@google.com39edf4c2013-04-25 17:33:51 +00001264 return true;
1265 }
1266 return false;
1267}
1268
scroggo@google.comb5571b32013-09-25 21:34:24 +00001269SkImageDecoder* sk_libpng_dfactory(SkStreamRewindable* stream) {
scroggo@google.com39edf4c2013-04-25 17:33:51 +00001270 if (is_png(stream)) {
reed@android.com00bf85a2009-01-22 13:04:56 +00001271 return SkNEW(SkPNGImageDecoder);
1272 }
1273 return NULL;
reed@android.com8a1c16f2008-12-17 15:59:43 +00001274}
1275
scroggo@google.comb5571b32013-09-25 21:34:24 +00001276static SkImageDecoder::Format get_format_png(SkStreamRewindable* stream) {
scroggo@google.com39edf4c2013-04-25 17:33:51 +00001277 if (is_png(stream)) {
1278 return SkImageDecoder::kPNG_Format;
1279 }
1280 return SkImageDecoder::kUnknown_Format;
1281}
1282
reed@android.comdfee5792010-04-15 14:24:50 +00001283SkImageEncoder* sk_libpng_efactory(SkImageEncoder::Type t) {
reed@android.com00bf85a2009-01-22 13:04:56 +00001284 return (SkImageEncoder::kPNG_Type == t) ? SkNEW(SkPNGImageEncoder) : NULL;
1285}
1286
mtklein@google.combd6343b2013-09-04 17:20:18 +00001287static SkImageDecoder_DecodeReg gDReg(sk_libpng_dfactory);
1288static SkImageDecoder_FormatReg gFormatReg(get_format_png);
1289static SkImageEncoder_EncodeReg gEReg(sk_libpng_efactory);