2009-02-05 Scott Violet <sky@google.com>
[WebKit-https.git] / WebCore / platform / image-encoders / skia / PNGImageEncoder.cpp
1 /*
2  * Copyright (c) 2006-2009, Google Inc. All rights reserved.
3  * 
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions are
6  * met:
7  * 
8  *     * Redistributions of source code must retain the above copyright
9  * notice, this list of conditions and the following disclaimer.
10  *     * Redistributions in binary form must reproduce the above
11  * copyright notice, this list of conditions and the following disclaimer
12  * in the documentation and/or other materials provided with the
13  * distribution.
14  *     * Neither the name of Google Inc. nor the names of its
15  * contributors may be used to endorse or promote products derived from
16  * this software without specific prior written permission.
17  * 
18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29  */
30
31 #include "config.h"
32
33 #include "IntSize.h"
34 #include "OwnArrayPtr.h"
35 #include "PNGImageEncoder.h"
36 #include "Vector.h"
37
38 #include "SkBitmap.h"
39
40 extern "C" {
41 #include "png.h"
42 }
43
44 namespace WebCore {
45
46 // Converts BGRA->RGBA and RGBA->BGRA.
47 static void convertBetweenBGRAandRGBA(const unsigned char* input, int numberOfPixels,
48                                       unsigned char* output)
49 {
50     for (int x = 0; x < numberOfPixels; x++) {
51         const unsigned char* pixelIn = &input[x * 4];
52         unsigned char* pixelOut = &output[x * 4];
53         pixelOut[0] = pixelIn[2];
54         pixelOut[1] = pixelIn[1];
55         pixelOut[2] = pixelIn[0];
56         pixelOut[3] = pixelIn[3];
57     }
58 }
59
60 // Encoder --------------------------------------------------------------------
61 //
62 // This section of the code is based on nsPNGEncoder.cpp in Mozilla
63 // (Copyright 2005 Google Inc.)
64
65 // Passed around as the io_ptr in the png structs so our callbacks know where
66 // to write data.
67 struct PNGEncoderState {
68     PNGEncoderState(Vector<unsigned char>* o) : m_out(o) {}
69     Vector<unsigned char>* m_out;
70 };
71
72 // Called by libpng to flush its internal buffer to ours.
73 void encoderWriteCallback(png_structp png, png_bytep data, png_size_t size)
74 {
75     PNGEncoderState* state = static_cast<PNGEncoderState*>(png_get_io_ptr(png));
76     ASSERT(state->m_out);
77
78     size_t oldSize = state->m_out->size();
79     state->m_out->resize(oldSize + size);
80     memcpy(&(*state->m_out)[oldSize], data, size);
81 }
82
83 // Automatically destroys the given write structs on destruction to make
84 // cleanup and error handling code cleaner.
85 class PNGWriteStructDestroyer {
86 public:
87     PNGWriteStructDestroyer(png_struct** ps, png_info** pi)
88         : m_pngStruct(ps)
89         , m_pngInfo(pi) {
90     }
91
92     ~PNGWriteStructDestroyer() {
93         png_destroy_write_struct(m_pngStruct, m_pngInfo);
94     }
95
96 private:
97     png_struct** m_pngStruct;
98     png_info** m_pngInfo;
99 };
100
101 // static
102 bool PNGImageEncoder::encode(const SkBitmap& image, Vector<unsigned char>* output)
103 {
104     if (image.config() != SkBitmap::kARGB_8888_Config)
105         return false;  // Only support ARGB at 8 bpp now.
106
107     image.lockPixels();
108     bool result = PNGImageEncoder::encode(static_cast<unsigned char*>(
109         image.getPixels()), IntSize(image.width(), image.height()),
110         image.rowBytes(), output);
111     image.unlockPixels();
112     return result;
113 }
114
115 // static
116 bool PNGImageEncoder::encode(const unsigned char* input, const IntSize& size,
117                              int bytesPerRow,
118                              Vector<unsigned char>* output)
119 {
120     int inputColorComponents = 4;
121     int outputColorComponents = 4;
122     int pngOutputColorType = PNG_COLOR_TYPE_RGB_ALPHA;
123     IntSize imageSize(size);
124     imageSize.clampNegativeToZero();
125
126     // Row stride should be at least as long as the length of the data.
127     if (inputColorComponents * imageSize.width() > bytesPerRow) {
128         ASSERT(false);
129         return false;
130     }
131
132     png_struct* pngPtr = png_create_write_struct(PNG_LIBPNG_VER_STRING,
133                                                  png_voidp_NULL,
134                                                  png_error_ptr_NULL,
135                                                  png_error_ptr_NULL);
136     if (!pngPtr)
137         return false;
138
139     png_info* infoPtr = png_create_info_struct(pngPtr);
140     if (!infoPtr) {
141         png_destroy_write_struct(&pngPtr, NULL);
142         return false;
143     }
144     PNGWriteStructDestroyer destroyer(&pngPtr, &infoPtr);
145
146     if (setjmp(png_jmpbuf(pngPtr))) {
147         // The destroyer will ensure that the structures are cleaned up in this
148         // case, even though we may get here as a jump from random parts of the
149         // PNG library called below.
150         return false;
151     }
152
153     // Set our callback for libpng to give us the data.
154     PNGEncoderState state(output);
155     png_set_write_fn(pngPtr, &state, encoderWriteCallback, NULL);
156
157     png_set_IHDR(pngPtr, infoPtr, imageSize.width(), imageSize.height(), 8, pngOutputColorType,
158                  PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_DEFAULT,
159                  PNG_FILTER_TYPE_DEFAULT);
160     png_write_info(pngPtr, infoPtr);
161
162     OwnArrayPtr<unsigned char> rowPixels(new unsigned char[imageSize.width() * outputColorComponents]);
163     for (int y = 0; y < imageSize.height(); y ++) {
164         convertBetweenBGRAandRGBA(&input[y * bytesPerRow], imageSize.width(), rowPixels.get());
165         png_write_row(pngPtr, rowPixels.get());
166     }
167
168     png_write_end(pngPtr, infoPtr);
169     return true;
170 }
171
172 }  // namespace WebCore