#include "SkUnPreMultiply.h"
#include "SkUtils.h"
-// A WebP decoder only, on top of (subset of) libwebp
+// A WebP encoder only, on top of (subset of) libwebp
// For more information on WebP image format, and libwebp library, see:
// http://code.google.com/speed/webp/
// http://www.webmproject.org/code/#libwebp_webp_image_decoder_library
// If moving libwebp out of skia source tree, path for webp headers must be
// updated accordingly. Here, we enforce using local copy in webp sub-directory.
#include "webp/encode.h"
+#include "webp/mux.h"
}
static transform_scanline_proc choose_proc(const SkImageInfo& info,
WebPPicture pic;
WebPPictureInit(&pic);
+ SkAutoTCallVProc<WebPPicture, WebPPictureFree> autoPic(&pic);
pic.width = pixmap.width();
pic.height = pixmap.height();
pic.writer = stream_writer;
- pic.custom_ptr = (void*)stream;
+
+ // If there is no need to embed an ICC profile, we write directly to the input stream.
+ // Otherwise, we will first encode to |tmp| and use a mux to add the ICC chunk. libwebp
+ // forces us to have an encoded image before we can add a profile.
+ sk_sp<SkData> icc = pixmap.colorSpace() ? icc_from_color_space(*pixmap.colorSpace()) : nullptr;
+ SkDynamicMemoryWStream tmp;
+ pic.custom_ptr = icc ? (void*)&tmp : (void*)stream;
const uint8_t* src = (uint8_t*)pixmap.addr();
const int rgbStride = pic.width * bpp;
proc((char*) &rgb[y * rgbStride], (const char*) &src[y * rowBytes], pic.width, bpp, colors);
}
- bool ok;
- if (bpp == 3) {
- ok = SkToBool(WebPPictureImportRGB(&pic, &rgb[0], rgbStride));
- } else {
+ auto importProc = WebPPictureImportRGB;
+ if (3 != bpp) {
if (pixmap.isOpaque()) {
- ok = SkToBool(WebPPictureImportRGBX(&pic, &rgb[0], rgbStride));
+ importProc = WebPPictureImportRGBX;
} else {
- ok = SkToBool(WebPPictureImportRGBA(&pic, &rgb[0], rgbStride));
+ importProc = WebPPictureImportRGBA;
}
}
- ok = ok && WebPEncode(&webp_config, &pic);
- WebPPictureFree(&pic);
+ if (!importProc(&pic, &rgb[0], rgbStride)) {
+ return false;
+ }
+
+ if (!WebPEncode(&webp_config, &pic)) {
+ return false;
+ }
+
+ if (icc) {
+ sk_sp<SkData> encodedData = tmp.detachAsData();
+ WebPData encoded = { encodedData->bytes(), encodedData->size() };
+ WebPData iccChunk = { icc->bytes(), icc->size() };
+
+ SkAutoTCallVProc<WebPMux, WebPMuxDelete> mux(WebPMuxNew());
+ if (WEBP_MUX_OK != WebPMuxSetImage(mux, &encoded, 0)) {
+ return false;
+ }
+
+ if (WEBP_MUX_OK != WebPMuxSetChunk(mux, "ICCP", &iccChunk, 0)) {
+ return false;
+ }
+
+ WebPData assembled;
+ if (WEBP_MUX_OK != WebPMuxAssemble(mux, &assembled)) {
+ return false;
+ }
+
+ stream->write(assembled.bytes, assembled.size);
+ WebPDataClear(&assembled);
+ }
- return ok;
+ return true;
}
bool SkEncodeImageAsWEBP(SkWStream* stream, const SkPixmap& src, int quality) {
DEF_TEST(Codec_EncodeICC, r) {
test_encode_icc(r, SkEncodedImageFormat::kPNG, SkTransferFunctionBehavior::kRespect);
test_encode_icc(r, SkEncodedImageFormat::kJPEG, SkTransferFunctionBehavior::kRespect);
+ test_encode_icc(r, SkEncodedImageFormat::kWEBP, SkTransferFunctionBehavior::kRespect);
test_encode_icc(r, SkEncodedImageFormat::kPNG, SkTransferFunctionBehavior::kIgnore);
test_encode_icc(r, SkEncodedImageFormat::kJPEG, SkTransferFunctionBehavior::kIgnore);
+ test_encode_icc(r, SkEncodedImageFormat::kWEBP, SkTransferFunctionBehavior::kIgnore);
}
"../externals/libwebp/src/enc/tree_enc.c",
"../externals/libwebp/src/enc/vp8l_enc.c",
"../externals/libwebp/src/enc/webp_enc.c",
+ "../externals/libwebp/src/mux/anim_encode.c",
+ "../externals/libwebp/src/mux/muxedit.c",
+ "../externals/libwebp/src/mux/muxinternal.c",
+ "../externals/libwebp/src/mux/muxread.c",
"../externals/libwebp/src/utils/bit_reader_utils.c",
"../externals/libwebp/src/utils/bit_writer_utils.c",
"../externals/libwebp/src/utils/color_cache_utils.c",