Index: third_party/libwebp/demux/anim_decode.c |
diff --git a/third_party/libwebp/demux/anim_decode.c b/third_party/libwebp/demux/anim_decode.c |
new file mode 100644 |
index 0000000000000000000000000000000000000000..1989eb4ab4c94ca99be8a6b7094b187895cac695 |
--- /dev/null |
+++ b/third_party/libwebp/demux/anim_decode.c |
@@ -0,0 +1,442 @@ |
+// Copyright 2015 Google Inc. All Rights Reserved. |
+// |
+// Use of this source code is governed by a BSD-style license |
+// that can be found in the COPYING file in the root of the source |
+// tree. An additional intellectual property rights grant can be found |
+// in the file PATENTS. All contributing project authors may |
+// be found in the AUTHORS file in the root of the source tree. |
+// ----------------------------------------------------------------------------- |
+// |
+// AnimDecoder implementation. |
+// |
+ |
+#ifdef HAVE_CONFIG_H |
+#include "../webp/config.h" |
+#endif |
+ |
+#include <assert.h> |
+#include <string.h> |
+ |
+#include "../utils/utils.h" |
+#include "../webp/decode.h" |
+#include "../webp/demux.h" |
+ |
+#define NUM_CHANNELS 4 |
+ |
+typedef void (*BlendRowFunc)(uint32_t* const, const uint32_t* const, int); |
+static void BlendPixelRowNonPremult(uint32_t* const src, |
+ const uint32_t* const dst, int num_pixels); |
+static void BlendPixelRowPremult(uint32_t* const src, const uint32_t* const dst, |
+ int num_pixels); |
+ |
+struct WebPAnimDecoder { |
+ WebPDemuxer* demux_; // Demuxer created from given WebP bitstream. |
+ WebPDecoderConfig config_; // Decoder config. |
+ // Note: we use a pointer to a function blending multiple pixels at a time to |
+ // allow possible inlining of per-pixel blending function. |
+ BlendRowFunc blend_func_; // Pointer to the chose blend row function. |
+ WebPAnimInfo info_; // Global info about the animation. |
+ uint8_t* curr_frame_; // Current canvas (not disposed). |
+ uint8_t* prev_frame_disposed_; // Previous canvas (properly disposed). |
+ int prev_frame_timestamp_; // Previous frame timestamp (milliseconds). |
+ WebPIterator prev_iter_; // Iterator object for previous frame. |
+ int prev_frame_was_keyframe_; // True if previous frame was a keyframe. |
+ int next_frame_; // Index of the next frame to be decoded |
+ // (starting from 1). |
+}; |
+ |
+static void DefaultDecoderOptions(WebPAnimDecoderOptions* const dec_options) { |
+ dec_options->color_mode = MODE_RGBA; |
+ dec_options->use_threads = 0; |
+} |
+ |
+int WebPAnimDecoderOptionsInitInternal(WebPAnimDecoderOptions* dec_options, |
+ int abi_version) { |
+ if (dec_options == NULL || |
+ WEBP_ABI_IS_INCOMPATIBLE(abi_version, WEBP_DEMUX_ABI_VERSION)) { |
+ return 0; |
+ } |
+ DefaultDecoderOptions(dec_options); |
+ return 1; |
+} |
+ |
+static int ApplyDecoderOptions(const WebPAnimDecoderOptions* const dec_options, |
+ WebPAnimDecoder* const dec) { |
+ WEBP_CSP_MODE mode; |
+ WebPDecoderConfig* config = &dec->config_; |
+ assert(dec_options != NULL); |
+ |
+ mode = dec_options->color_mode; |
+ if (mode != MODE_RGBA && mode != MODE_BGRA && |
+ mode != MODE_rgbA && mode != MODE_bgrA) { |
+ return 0; |
+ } |
+ dec->blend_func_ = (mode == MODE_RGBA || mode == MODE_BGRA) |
+ ? &BlendPixelRowNonPremult |
+ : &BlendPixelRowPremult; |
+ WebPInitDecoderConfig(config); |
+ config->output.colorspace = mode; |
+ config->output.is_external_memory = 1; |
+ config->options.use_threads = dec_options->use_threads; |
+ // Note: config->output.u.RGBA is set at the time of decoding each frame. |
+ return 1; |
+} |
+ |
+WebPAnimDecoder* WebPAnimDecoderNewInternal( |
+ const WebPData* webp_data, const WebPAnimDecoderOptions* dec_options, |
+ int abi_version) { |
+ WebPAnimDecoderOptions options; |
+ WebPAnimDecoder* dec = NULL; |
+ if (webp_data == NULL || |
+ WEBP_ABI_IS_INCOMPATIBLE(abi_version, WEBP_DEMUX_ABI_VERSION)) { |
+ return NULL; |
+ } |
+ |
+ // Note: calloc() so that the pointer members are initialized to NULL. |
+ dec = (WebPAnimDecoder*)WebPSafeCalloc(1ULL, sizeof(*dec)); |
+ if (dec == NULL) goto Error; |
+ |
+ if (dec_options != NULL) { |
+ options = *dec_options; |
+ } else { |
+ DefaultDecoderOptions(&options); |
+ } |
+ if (!ApplyDecoderOptions(&options, dec)) goto Error; |
+ |
+ dec->demux_ = WebPDemux(webp_data); |
+ if (dec->demux_ == NULL) goto Error; |
+ |
+ dec->info_.canvas_width = WebPDemuxGetI(dec->demux_, WEBP_FF_CANVAS_WIDTH); |
+ dec->info_.canvas_height = WebPDemuxGetI(dec->demux_, WEBP_FF_CANVAS_HEIGHT); |
+ dec->info_.loop_count = WebPDemuxGetI(dec->demux_, WEBP_FF_LOOP_COUNT); |
+ dec->info_.bgcolor = WebPDemuxGetI(dec->demux_, WEBP_FF_BACKGROUND_COLOR); |
+ dec->info_.frame_count = WebPDemuxGetI(dec->demux_, WEBP_FF_FRAME_COUNT); |
+ |
+ { |
+ const int canvas_bytes = |
+ dec->info_.canvas_width * NUM_CHANNELS * dec->info_.canvas_height; |
+ // Note: calloc() because we fill frame with zeroes as well. |
+ dec->curr_frame_ = WebPSafeCalloc(1ULL, canvas_bytes); |
+ if (dec->curr_frame_ == NULL) goto Error; |
+ dec->prev_frame_disposed_ = WebPSafeCalloc(1ULL, canvas_bytes); |
+ if (dec->prev_frame_disposed_ == NULL) goto Error; |
+ } |
+ |
+ WebPAnimDecoderReset(dec); |
+ |
+ return dec; |
+ |
+ Error: |
+ WebPAnimDecoderDelete(dec); |
+ return NULL; |
+} |
+ |
+int WebPAnimDecoderGetInfo(const WebPAnimDecoder* dec, WebPAnimInfo* info) { |
+ if (dec == NULL || info == NULL) return 0; |
+ *info = dec->info_; |
+ return 1; |
+} |
+ |
+// Returns true if the frame covers the full canvas. |
+static int IsFullFrame(int width, int height, int canvas_width, |
+ int canvas_height) { |
+ return (width == canvas_width && height == canvas_height); |
+} |
+ |
+// Clear the canvas to transparent. |
+static void ZeroFillCanvas(uint8_t* buf, uint32_t canvas_width, |
+ uint32_t canvas_height) { |
+ memset(buf, 0, canvas_width * NUM_CHANNELS * canvas_height); |
+} |
+ |
+// Clear given frame rectangle to transparent. |
+static void ZeroFillFrameRect(uint8_t* buf, int buf_stride, int x_offset, |
+ int y_offset, int width, int height) { |
+ int j; |
+ assert(width * NUM_CHANNELS <= buf_stride); |
+ buf += y_offset * buf_stride + x_offset * NUM_CHANNELS; |
+ for (j = 0; j < height; ++j) { |
+ memset(buf, 0, width * NUM_CHANNELS); |
+ buf += buf_stride; |
+ } |
+} |
+ |
+// Copy width * height pixels from 'src' to 'dst'. |
+static void CopyCanvas(const uint8_t* src, uint8_t* dst, |
+ uint32_t width, uint32_t height) { |
+ assert(src != NULL && dst != NULL); |
+ memcpy(dst, src, width * NUM_CHANNELS * height); |
+} |
+ |
+// Returns true if the current frame is a key-frame. |
+static int IsKeyFrame(const WebPIterator* const curr, |
+ const WebPIterator* const prev, |
+ int prev_frame_was_key_frame, |
+ int canvas_width, int canvas_height) { |
+ if (curr->frame_num == 1) { |
+ return 1; |
+ } else if ((!curr->has_alpha || curr->blend_method == WEBP_MUX_NO_BLEND) && |
+ IsFullFrame(curr->width, curr->height, |
+ canvas_width, canvas_height)) { |
+ return 1; |
+ } else { |
+ return (prev->dispose_method == WEBP_MUX_DISPOSE_BACKGROUND) && |
+ (IsFullFrame(prev->width, prev->height, canvas_width, |
+ canvas_height) || |
+ prev_frame_was_key_frame); |
+ } |
+} |
+ |
+ |
+// Blend a single channel of 'src' over 'dst', given their alpha channel values. |
+// 'src' and 'dst' are assumed to be NOT pre-multiplied by alpha. |
+static uint8_t BlendChannelNonPremult(uint32_t src, uint8_t src_a, |
+ uint32_t dst, uint8_t dst_a, |
+ uint32_t scale, int shift) { |
+ const uint8_t src_channel = (src >> shift) & 0xff; |
+ const uint8_t dst_channel = (dst >> shift) & 0xff; |
+ const uint32_t blend_unscaled = src_channel * src_a + dst_channel * dst_a; |
+ assert(blend_unscaled < (1ULL << 32) / scale); |
+ return (blend_unscaled * scale) >> 24; |
+} |
+ |
+// Blend 'src' over 'dst' assuming they are NOT pre-multiplied by alpha. |
+static uint32_t BlendPixelNonPremult(uint32_t src, uint32_t dst) { |
+ const uint8_t src_a = (src >> 24) & 0xff; |
+ |
+ if (src_a == 0) { |
+ return dst; |
+ } else { |
+ const uint8_t dst_a = (dst >> 24) & 0xff; |
+ // This is the approximate integer arithmetic for the actual formula: |
+ // dst_factor_a = (dst_a * (255 - src_a)) / 255. |
+ const uint8_t dst_factor_a = (dst_a * (256 - src_a)) >> 8; |
+ const uint8_t blend_a = src_a + dst_factor_a; |
+ const uint32_t scale = (1UL << 24) / blend_a; |
+ |
+ const uint8_t blend_r = |
+ BlendChannelNonPremult(src, src_a, dst, dst_factor_a, scale, 0); |
+ const uint8_t blend_g = |
+ BlendChannelNonPremult(src, src_a, dst, dst_factor_a, scale, 8); |
+ const uint8_t blend_b = |
+ BlendChannelNonPremult(src, src_a, dst, dst_factor_a, scale, 16); |
+ assert(src_a + dst_factor_a < 256); |
+ |
+ return (blend_r << 0) | |
+ (blend_g << 8) | |
+ (blend_b << 16) | |
+ ((uint32_t)blend_a << 24); |
+ } |
+} |
+ |
+// Blend 'num_pixels' in 'src' over 'dst' assuming they are NOT pre-multiplied |
+// by alpha. |
+static void BlendPixelRowNonPremult(uint32_t* const src, |
+ const uint32_t* const dst, int num_pixels) { |
+ int i; |
+ for (i = 0; i < num_pixels; ++i) { |
+ const uint8_t src_alpha = (src[i] >> 24) & 0xff; |
+ if (src_alpha != 0xff) { |
+ src[i] = BlendPixelNonPremult(src[i], dst[i]); |
+ } |
+ } |
+} |
+ |
+// Individually multiply each channel in 'pix' by 'scale'. |
+static WEBP_INLINE uint32_t ChannelwiseMultiply(uint32_t pix, uint32_t scale) { |
+ uint32_t mask = 0x00FF00FF; |
+ uint32_t rb = ((pix & mask) * scale) >> 8; |
+ uint32_t ag = ((pix >> 8) & mask) * scale; |
+ return (rb & mask) | (ag & ~mask); |
+} |
+ |
+// Blend 'src' over 'dst' assuming they are pre-multiplied by alpha. |
+static uint32_t BlendPixelPremult(uint32_t src, uint32_t dst) { |
+ const uint8_t src_a = (src >> 24) & 0xff; |
+ return src + ChannelwiseMultiply(dst, 256 - src_a); |
+} |
+ |
+// Blend 'num_pixels' in 'src' over 'dst' assuming they are pre-multiplied by |
+// alpha. |
+static void BlendPixelRowPremult(uint32_t* const src, const uint32_t* const dst, |
+ int num_pixels) { |
+ int i; |
+ for (i = 0; i < num_pixels; ++i) { |
+ const uint8_t src_alpha = (src[i] >> 24) & 0xff; |
+ if (src_alpha != 0xff) { |
+ src[i] = BlendPixelPremult(src[i], dst[i]); |
+ } |
+ } |
+} |
+ |
+// Returns two ranges (<left, width> pairs) at row 'canvas_y', that belong to |
+// 'src' but not 'dst'. A point range is empty if the corresponding width is 0. |
+static void FindBlendRangeAtRow(const WebPIterator* const src, |
+ const WebPIterator* const dst, int canvas_y, |
+ int* const left1, int* const width1, |
+ int* const left2, int* const width2) { |
+ const int src_max_x = src->x_offset + src->width; |
+ const int dst_max_x = dst->x_offset + dst->width; |
+ const int dst_max_y = dst->y_offset + dst->height; |
+ assert(canvas_y >= src->y_offset && canvas_y < (src->y_offset + src->height)); |
+ *left1 = -1; |
+ *width1 = 0; |
+ *left2 = -1; |
+ *width2 = 0; |
+ |
+ if (canvas_y < dst->y_offset || canvas_y >= dst_max_y || |
+ src->x_offset >= dst_max_x || src_max_x <= dst->x_offset) { |
+ *left1 = src->x_offset; |
+ *width1 = src->width; |
+ return; |
+ } |
+ |
+ if (src->x_offset < dst->x_offset) { |
+ *left1 = src->x_offset; |
+ *width1 = dst->x_offset - src->x_offset; |
+ } |
+ |
+ if (src_max_x > dst_max_x) { |
+ *left2 = dst_max_x; |
+ *width2 = src_max_x - dst_max_x; |
+ } |
+} |
+ |
+int WebPAnimDecoderGetNext(WebPAnimDecoder* dec, |
+ uint8_t** buf_ptr, int* timestamp_ptr) { |
+ WebPIterator iter; |
+ uint32_t width; |
+ uint32_t height; |
+ int is_key_frame; |
+ int timestamp; |
+ BlendRowFunc blend_row; |
+ |
+ if (dec == NULL || buf_ptr == NULL || timestamp_ptr == NULL) return 0; |
+ if (!WebPAnimDecoderHasMoreFrames(dec)) return 0; |
+ |
+ width = dec->info_.canvas_width; |
+ height = dec->info_.canvas_height; |
+ blend_row = dec->blend_func_; |
+ |
+ // Get compressed frame. |
+ if (!WebPDemuxGetFrame(dec->demux_, dec->next_frame_, &iter)) { |
+ return 0; |
+ } |
+ timestamp = dec->prev_frame_timestamp_ + iter.duration; |
+ |
+ // Initialize. |
+ is_key_frame = IsKeyFrame(&iter, &dec->prev_iter_, |
+ dec->prev_frame_was_keyframe_, width, height); |
+ if (is_key_frame) { |
+ ZeroFillCanvas(dec->curr_frame_, width, height); |
+ } else { |
+ CopyCanvas(dec->prev_frame_disposed_, dec->curr_frame_, width, height); |
+ } |
+ |
+ // Decode. |
+ { |
+ const uint8_t* in = iter.fragment.bytes; |
+ const size_t in_size = iter.fragment.size; |
+ const size_t out_offset = |
+ (iter.y_offset * width + iter.x_offset) * NUM_CHANNELS; |
+ WebPDecoderConfig* const config = &dec->config_; |
+ WebPRGBABuffer* const buf = &config->output.u.RGBA; |
+ buf->stride = NUM_CHANNELS * width; |
+ buf->size = buf->stride * iter.height; |
+ buf->rgba = dec->curr_frame_ + out_offset; |
+ |
+ if (WebPDecode(in, in_size, config) != VP8_STATUS_OK) { |
+ goto Error; |
+ } |
+ } |
+ |
+ // During the decoding of current frame, we may have set some pixels to be |
+ // transparent (i.e. alpha < 255). However, the value of each of these |
+ // pixels should have been determined by blending it against the value of |
+ // that pixel in the previous frame if blending method of is WEBP_MUX_BLEND. |
+ if (iter.frame_num > 1 && iter.blend_method == WEBP_MUX_BLEND && |
+ !is_key_frame) { |
+ if (dec->prev_iter_.dispose_method == WEBP_MUX_DISPOSE_NONE) { |
+ int y; |
+ // Blend transparent pixels with pixels in previous canvas. |
+ for (y = 0; y < iter.height; ++y) { |
+ const size_t offset = |
+ (iter.y_offset + y) * width + iter.x_offset; |
+ blend_row((uint32_t*)dec->curr_frame_ + offset, |
+ (uint32_t*)dec->prev_frame_disposed_ + offset, iter.width); |
+ } |
+ } else { |
+ int y; |
+ assert(dec->prev_iter_.dispose_method == WEBP_MUX_DISPOSE_BACKGROUND); |
+ // We need to blend a transparent pixel with its value just after |
+ // initialization. That is, blend it with: |
+ // * Fully transparent pixel if it belongs to prevRect <-- No-op. |
+ // * The pixel in the previous canvas otherwise <-- Need alpha-blending. |
+ for (y = 0; y < iter.height; ++y) { |
+ const int canvas_y = iter.y_offset + y; |
+ int left1, width1, left2, width2; |
+ FindBlendRangeAtRow(&iter, &dec->prev_iter_, canvas_y, &left1, &width1, |
+ &left2, &width2); |
+ if (width1 > 0) { |
+ const size_t offset1 = canvas_y * width + left1; |
+ blend_row((uint32_t*)dec->curr_frame_ + offset1, |
+ (uint32_t*)dec->prev_frame_disposed_ + offset1, width1); |
+ } |
+ if (width2 > 0) { |
+ const size_t offset2 = canvas_y * width + left2; |
+ blend_row((uint32_t*)dec->curr_frame_ + offset2, |
+ (uint32_t*)dec->prev_frame_disposed_ + offset2, width2); |
+ } |
+ } |
+ } |
+ } |
+ |
+ // Update info of the previous frame and dispose it for the next iteration. |
+ dec->prev_frame_timestamp_ = timestamp; |
+ dec->prev_iter_ = iter; |
+ dec->prev_frame_was_keyframe_ = is_key_frame; |
+ CopyCanvas(dec->curr_frame_, dec->prev_frame_disposed_, width, height); |
+ if (dec->prev_iter_.dispose_method == WEBP_MUX_DISPOSE_BACKGROUND) { |
+ ZeroFillFrameRect(dec->prev_frame_disposed_, width * NUM_CHANNELS, |
+ dec->prev_iter_.x_offset, dec->prev_iter_.y_offset, |
+ dec->prev_iter_.width, dec->prev_iter_.height); |
+ } |
+ ++dec->next_frame_; |
+ |
+ // All OK, fill in the values. |
+ *buf_ptr = dec->curr_frame_; |
+ *timestamp_ptr = timestamp; |
+ return 1; |
+ |
+ Error: |
+ WebPDemuxReleaseIterator(&iter); |
+ return 0; |
+} |
+ |
+int WebPAnimDecoderHasMoreFrames(const WebPAnimDecoder* dec) { |
+ if (dec == NULL) return 0; |
+ return (dec->next_frame_ <= (int)dec->info_.frame_count); |
+} |
+ |
+void WebPAnimDecoderReset(WebPAnimDecoder* dec) { |
+ if (dec != NULL) { |
+ dec->prev_frame_timestamp_ = 0; |
+ memset(&dec->prev_iter_, 0, sizeof(dec->prev_iter_)); |
+ dec->prev_frame_was_keyframe_ = 0; |
+ dec->next_frame_ = 1; |
+ } |
+} |
+ |
+const WebPDemuxer* WebPAnimDecoderGetDemuxer(const WebPAnimDecoder* dec) { |
+ if (dec == NULL) return NULL; |
+ return dec->demux_; |
+} |
+ |
+void WebPAnimDecoderDelete(WebPAnimDecoder* dec) { |
+ if (dec != NULL) { |
+ WebPDemuxDelete(dec->demux_); |
+ WebPSafeFree(dec->curr_frame_); |
+ WebPSafeFree(dec->prev_frame_disposed_); |
+ WebPSafeFree(dec); |
+ } |
+} |