Index: src/images/SkImageDecoder_libwebp.cpp |
diff --git a/src/images/SkImageDecoder_libwebp.cpp b/src/images/SkImageDecoder_libwebp.cpp |
index 2db08cee83dd5fd3ceb054a2e515ae12af8cd2cf..116608a253e2d1db65bc5b2d3e5258bf7c285435 100644 |
--- a/src/images/SkImageDecoder_libwebp.cpp |
+++ b/src/images/SkImageDecoder_libwebp.cpp |
@@ -14,10 +14,9 @@ |
* limitations under the License. |
*/ |
-#include "SkImageDecoder.h" |
+#include "SkBitmap.h" |
#include "SkImageEncoder.h" |
#include "SkColorPriv.h" |
-#include "SkScaledBitmapSampler.h" |
#include "SkStream.h" |
#include "SkTemplates.h" |
#include "SkUtils.h" |
@@ -32,299 +31,9 @@ |
extern "C" { |
// 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/decode.h" |
#include "webp/encode.h" |
} |
-// this enables timing code to report milliseconds for a decode |
-//#define TIME_DECODE |
- |
-////////////////////////////////////////////////////////////////////////// |
-////////////////////////////////////////////////////////////////////////// |
- |
-// Define VP8 I/O on top of Skia stream |
- |
-////////////////////////////////////////////////////////////////////////// |
-////////////////////////////////////////////////////////////////////////// |
- |
-static const size_t WEBP_VP8_HEADER_SIZE = 64; |
-static const size_t WEBP_IDECODE_BUFFER_SZ = (1 << 16); |
- |
-// Parse headers of RIFF container, and check for valid Webp (VP8) content. |
-static bool webp_parse_header(SkStream* stream, int* width, int* height, int* alpha) { |
- unsigned char buffer[WEBP_VP8_HEADER_SIZE]; |
- size_t bytesToRead = WEBP_VP8_HEADER_SIZE; |
- size_t totalBytesRead = 0; |
- do { |
- unsigned char* dst = buffer + totalBytesRead; |
- const size_t bytesRead = stream->read(dst, bytesToRead); |
- if (0 == bytesRead) { |
- SkASSERT(stream->isAtEnd()); |
- break; |
- } |
- bytesToRead -= bytesRead; |
- totalBytesRead += bytesRead; |
- SkASSERT(bytesToRead + totalBytesRead == WEBP_VP8_HEADER_SIZE); |
- } while (!stream->isAtEnd() && bytesToRead > 0); |
- |
- WebPBitstreamFeatures features; |
- VP8StatusCode status = WebPGetFeatures(buffer, totalBytesRead, &features); |
- if (VP8_STATUS_OK != status) { |
- return false; // Invalid WebP file. |
- } |
- *width = features.width; |
- *height = features.height; |
- *alpha = features.has_alpha; |
- |
- // sanity check for image size that's about to be decoded. |
- { |
- int64_t size = sk_64_mul(*width, *height); |
- if (!sk_64_isS32(size)) { |
- return false; |
- } |
- // now check that if we are 4-bytes per pixel, we also don't overflow |
- if (sk_64_asS32(size) > (0x7FFFFFFF >> 2)) { |
- return false; |
- } |
- } |
- return true; |
-} |
- |
-class SkWEBPImageDecoder: public SkImageDecoder { |
-public: |
- SkWEBPImageDecoder() { |
- fOrigWidth = 0; |
- fOrigHeight = 0; |
- fHasAlpha = 0; |
- } |
- |
- Format getFormat() const override { |
- return kWEBP_Format; |
- } |
- |
-protected: |
- Result onDecode(SkStream* stream, SkBitmap* bm, Mode) override; |
- |
-private: |
- /** |
- * Called when determining the output config to request to webp. |
- * If the image does not have alpha, there is no need to premultiply. |
- * If the caller wants unpremultiplied colors, that is respected. |
- */ |
- bool shouldPremultiply() const { |
- return SkToBool(fHasAlpha) && !this->getRequireUnpremultipliedColors(); |
- } |
- |
- bool setDecodeConfig(SkBitmap* decodedBitmap, int width, int height); |
- |
- SkAutoTDelete<SkStream> fInputStream; |
- int fOrigWidth; |
- int fOrigHeight; |
- int fHasAlpha; |
- |
- typedef SkImageDecoder INHERITED; |
-}; |
- |
-////////////////////////////////////////////////////////////////////////// |
- |
-#ifdef TIME_DECODE |
- |
-#include "SkTime.h" |
- |
-class AutoTimeMillis { |
-public: |
- AutoTimeMillis(const char label[]) : |
- fLabel(label) { |
- if (nullptr == fLabel) { |
- fLabel = ""; |
- } |
- fNow = SkTime::GetMSecs(); |
- } |
- ~AutoTimeMillis() { |
- SkDebugf("---- Time (ms): %s %d\n", fLabel, SkTime::GetMSecs() - fNow); |
- } |
-private: |
- const char* fLabel; |
- SkMSec fNow; |
-}; |
- |
-#endif |
- |
-/////////////////////////////////////////////////////////////////////////////// |
- |
-// This guy exists just to aid in debugging, as it allows debuggers to just |
-// set a break-point in one place to see all error exists. |
-static void print_webp_error(const SkBitmap& bm, const char msg[]) { |
- SkDEBUGF(("libwebp error %s [%d %d]", msg, bm.width(), bm.height())); |
-} |
- |
-static SkImageDecoder::Result return_failure(const SkBitmap& bm, const char msg[]) { |
- print_webp_error(bm, msg); |
- return SkImageDecoder::kFailure; // must always return kFailure |
-} |
- |
-/////////////////////////////////////////////////////////////////////////////// |
- |
-static WEBP_CSP_MODE webp_decode_mode(const SkBitmap* decodedBitmap, bool premultiply) { |
- WEBP_CSP_MODE mode = MODE_LAST; |
- const SkColorType ct = decodedBitmap->colorType(); |
- |
- if (ct == kN32_SkColorType) { |
- #if SK_PMCOLOR_BYTE_ORDER(B,G,R,A) |
- mode = premultiply ? MODE_bgrA : MODE_BGRA; |
- #elif SK_PMCOLOR_BYTE_ORDER(R,G,B,A) |
- mode = premultiply ? MODE_rgbA : MODE_RGBA; |
- #else |
- #error "Skia uses BGRA or RGBA byte order" |
- #endif |
- } else if (ct == kARGB_4444_SkColorType) { |
- mode = premultiply ? MODE_rgbA_4444 : MODE_RGBA_4444; |
- } else if (ct == kRGB_565_SkColorType) { |
- mode = MODE_RGB_565; |
- } |
- SkASSERT(MODE_LAST != mode); |
- return mode; |
-} |
- |
-// Incremental WebP image decoding. Reads input buffer of 64K size iteratively |
-// and decodes this block to appropriate color-space as per config object. |
-static bool webp_idecode(SkStream* stream, WebPDecoderConfig* config) { |
- WebPIDecoder* idec = WebPIDecode(nullptr, 0, config); |
- if (nullptr == idec) { |
- WebPFreeDecBuffer(&config->output); |
- return false; |
- } |
- |
- if (!stream->rewind()) { |
- SkDebugf("Failed to rewind webp stream!"); |
- return false; |
- } |
- const size_t readBufferSize = stream->hasLength() ? |
- SkTMin(stream->getLength(), WEBP_IDECODE_BUFFER_SZ) : WEBP_IDECODE_BUFFER_SZ; |
- SkAutoTMalloc<unsigned char> srcStorage(readBufferSize); |
- unsigned char* input = srcStorage.get(); |
- if (nullptr == input) { |
- WebPIDelete(idec); |
- WebPFreeDecBuffer(&config->output); |
- return false; |
- } |
- |
- bool success = true; |
- VP8StatusCode status = VP8_STATUS_SUSPENDED; |
- do { |
- const size_t bytesRead = stream->read(input, readBufferSize); |
- if (0 == bytesRead) { |
- success = false; |
- break; |
- } |
- |
- status = WebPIAppend(idec, input, bytesRead); |
- if (VP8_STATUS_OK != status && VP8_STATUS_SUSPENDED != status) { |
- success = false; |
- break; |
- } |
- } while (VP8_STATUS_OK != status); |
- srcStorage.reset(); |
- WebPIDelete(idec); |
- WebPFreeDecBuffer(&config->output); |
- |
- return success; |
-} |
- |
-static bool webp_get_config_resize(WebPDecoderConfig* config, |
- SkBitmap* decodedBitmap, |
- int width, int height, bool premultiply) { |
- WEBP_CSP_MODE mode = webp_decode_mode(decodedBitmap, premultiply); |
- if (MODE_LAST == mode) { |
- return false; |
- } |
- |
- if (0 == WebPInitDecoderConfig(config)) { |
- return false; |
- } |
- |
- config->output.colorspace = mode; |
- config->output.u.RGBA.rgba = (uint8_t*)decodedBitmap->getPixels(); |
- config->output.u.RGBA.stride = (int) decodedBitmap->rowBytes(); |
- config->output.u.RGBA.size = decodedBitmap->getSize(); |
- config->output.is_external_memory = 1; |
- |
- if (width != decodedBitmap->width() || height != decodedBitmap->height()) { |
- config->options.use_scaling = 1; |
- config->options.scaled_width = decodedBitmap->width(); |
- config->options.scaled_height = decodedBitmap->height(); |
- } |
- |
- return true; |
-} |
- |
-bool SkWEBPImageDecoder::setDecodeConfig(SkBitmap* decodedBitmap, int width, int height) { |
- SkColorType colorType = this->getPrefColorType(k32Bit_SrcDepth, SkToBool(fHasAlpha)); |
- |
- // YUV converter supports output in RGB565, RGBA4444 and RGBA8888 formats. |
- if (fHasAlpha) { |
- if (colorType != kARGB_4444_SkColorType) { |
- colorType = kN32_SkColorType; |
- } |
- } else { |
- if (colorType != kRGB_565_SkColorType && colorType != kARGB_4444_SkColorType) { |
- colorType = kN32_SkColorType; |
- } |
- } |
- |
- SkAlphaType alphaType = kOpaque_SkAlphaType; |
- if (SkToBool(fHasAlpha)) { |
- if (this->getRequireUnpremultipliedColors()) { |
- alphaType = kUnpremul_SkAlphaType; |
- } else { |
- alphaType = kPremul_SkAlphaType; |
- } |
- } |
- return decodedBitmap->setInfo(SkImageInfo::Make(width, height, colorType, alphaType)); |
-} |
- |
-SkImageDecoder::Result SkWEBPImageDecoder::onDecode(SkStream* stream, SkBitmap* decodedBitmap, |
- Mode mode) { |
-#ifdef TIME_DECODE |
- AutoTimeMillis atm("WEBP Decode"); |
-#endif |
- |
- int origWidth, origHeight, hasAlpha; |
- if (!webp_parse_header(stream, &origWidth, &origHeight, &hasAlpha)) { |
- return kFailure; |
- } |
- this->fHasAlpha = hasAlpha; |
- |
- const int sampleSize = this->getSampleSize(); |
- SkScaledBitmapSampler sampler(origWidth, origHeight, sampleSize); |
- if (!setDecodeConfig(decodedBitmap, sampler.scaledWidth(), |
- sampler.scaledHeight())) { |
- return kFailure; |
- } |
- |
- // If only bounds are requested, done |
- if (SkImageDecoder::kDecodeBounds_Mode == mode) { |
- return kSuccess; |
- } |
- |
- if (!this->allocPixelRef(decodedBitmap, nullptr)) { |
- return return_failure(*decodedBitmap, "allocPixelRef"); |
- } |
- |
- SkAutoLockPixels alp(*decodedBitmap); |
- |
- WebPDecoderConfig config; |
- if (!webp_get_config_resize(&config, decodedBitmap, origWidth, origHeight, |
- this->shouldPremultiply())) { |
- return kFailure; |
- } |
- |
- // Decode the WebP image data stream using WebP incremental decoding. |
- return webp_idecode(stream, &config) ? kSuccess : kFailure; |
-} |
- |
-/////////////////////////////////////////////////////////////////////////////// |
- |
#include "SkUnPreMultiply.h" |
typedef void (*ScanlineImporter)(const uint8_t* in, uint8_t* out, int width, |
@@ -528,32 +237,11 @@ bool SkWEBPImageEncoder::onEncode(SkWStream* stream, const SkBitmap& bm, |
/////////////////////////////////////////////////////////////////////////////// |
-DEFINE_DECODER_CREATOR(WEBPImageDecoder); |
DEFINE_ENCODER_CREATOR(WEBPImageEncoder); |
/////////////////////////////////////////////////////////////////////////////// |
-static SkImageDecoder* sk_libwebp_dfactory(SkStreamRewindable* stream) { |
- int width, height, hasAlpha; |
- if (!webp_parse_header(stream, &width, &height, &hasAlpha)) { |
- return nullptr; |
- } |
- |
- // Magic matches, call decoder |
- return new SkWEBPImageDecoder; |
-} |
- |
-static SkImageDecoder::Format get_format_webp(SkStreamRewindable* stream) { |
- int width, height, hasAlpha; |
- if (webp_parse_header(stream, &width, &height, &hasAlpha)) { |
- return SkImageDecoder::kWEBP_Format; |
- } |
- return SkImageDecoder::kUnknown_Format; |
-} |
- |
static SkImageEncoder* sk_libwebp_efactory(SkImageEncoder::Type t) { |
return (SkImageEncoder::kWEBP_Type == t) ? new SkWEBPImageEncoder : nullptr; |
} |
-static SkImageDecoder_DecodeReg gDReg(sk_libwebp_dfactory); |
-static SkImageDecoder_FormatReg gFormatReg(get_format_webp); |
static SkImageEncoder_EncodeReg gEReg(sk_libwebp_efactory); |