| Index: third_party/libwebp/dec/alpha.c
|
| diff --git a/third_party/libwebp/dec/alpha.c b/third_party/libwebp/dec/alpha.c
|
| index 52216fc4d608f4fd94724219739d2354be86af38..028eb3d50ba4a32a2aeb35a96aaee4821fc0d53d 100644
|
| --- a/third_party/libwebp/dec/alpha.c
|
| +++ b/third_party/libwebp/dec/alpha.c
|
| @@ -23,12 +23,14 @@
|
| //------------------------------------------------------------------------------
|
| // ALPHDecoder object.
|
|
|
| -ALPHDecoder* ALPHNew(void) {
|
| +// Allocates a new alpha decoder instance.
|
| +static ALPHDecoder* ALPHNew(void) {
|
| ALPHDecoder* const dec = (ALPHDecoder*)WebPSafeCalloc(1ULL, sizeof(*dec));
|
| return dec;
|
| }
|
|
|
| -void ALPHDelete(ALPHDecoder* const dec) {
|
| +// Clears and deallocates an alpha decoder instance.
|
| +static void ALPHDelete(ALPHDecoder* const dec) {
|
| if (dec != NULL) {
|
| VP8LDelete(dec->vp8l_dec_);
|
| dec->vp8l_dec_ = NULL;
|
| @@ -44,17 +46,21 @@ void ALPHDelete(ALPHDecoder* const dec) {
|
| // Returns false in case of error in alpha header (data too short, invalid
|
| // compression method or filter, error in lossless header data etc).
|
| static int ALPHInit(ALPHDecoder* const dec, const uint8_t* data,
|
| - size_t data_size, int width, int height, uint8_t* output) {
|
| + size_t data_size, const VP8Io* const src_io,
|
| + uint8_t* output) {
|
| int ok = 0;
|
| const uint8_t* const alpha_data = data + ALPHA_HEADER_LEN;
|
| const size_t alpha_data_size = data_size - ALPHA_HEADER_LEN;
|
| int rsrv;
|
| + VP8Io* const io = &dec->io_;
|
|
|
| - assert(width > 0 && height > 0);
|
| - assert(data != NULL && output != NULL);
|
| + assert(data != NULL && output != NULL && src_io != NULL);
|
|
|
| - dec->width_ = width;
|
| - dec->height_ = height;
|
| + VP8FiltersInit();
|
| + dec->output_ = output;
|
| + dec->width_ = src_io->width;
|
| + dec->height_ = src_io->height;
|
| + assert(dec->width_ > 0 && dec->height_ > 0);
|
|
|
| if (data_size <= ALPHA_HEADER_LEN) {
|
| return 0;
|
| @@ -72,14 +78,28 @@ static int ALPHInit(ALPHDecoder* const dec, const uint8_t* data,
|
| return 0;
|
| }
|
|
|
| + // Copy the necessary parameters from src_io to io
|
| + VP8InitIo(io);
|
| + WebPInitCustomIo(NULL, io);
|
| + io->opaque = dec;
|
| + io->width = src_io->width;
|
| + io->height = src_io->height;
|
| +
|
| + io->use_cropping = src_io->use_cropping;
|
| + io->crop_left = src_io->crop_left;
|
| + io->crop_right = src_io->crop_right;
|
| + io->crop_top = src_io->crop_top;
|
| + io->crop_bottom = src_io->crop_bottom;
|
| + // No need to copy the scaling parameters.
|
| +
|
| if (dec->method_ == ALPHA_NO_COMPRESSION) {
|
| const size_t alpha_decoded_size = dec->width_ * dec->height_;
|
| ok = (alpha_data_size >= alpha_decoded_size);
|
| } else {
|
| assert(dec->method_ == ALPHA_LOSSLESS_COMPRESSION);
|
| - ok = VP8LDecodeAlphaHeader(dec, alpha_data, alpha_data_size, output);
|
| + ok = VP8LDecodeAlphaHeader(dec, alpha_data, alpha_data_size);
|
| }
|
| - VP8FiltersInit();
|
| +
|
| return ok;
|
| }
|
|
|
| @@ -90,15 +110,30 @@ static int ALPHInit(ALPHDecoder* const dec, const uint8_t* data,
|
| static int ALPHDecode(VP8Decoder* const dec, int row, int num_rows) {
|
| ALPHDecoder* const alph_dec = dec->alph_dec_;
|
| const int width = alph_dec->width_;
|
| - const int height = alph_dec->height_;
|
| - WebPUnfilterFunc unfilter_func = WebPUnfilters[alph_dec->filter_];
|
| - uint8_t* const output = dec->alpha_plane_;
|
| + const int height = alph_dec->io_.crop_bottom;
|
| if (alph_dec->method_ == ALPHA_NO_COMPRESSION) {
|
| - const size_t offset = row * width;
|
| - const size_t num_pixels = num_rows * width;
|
| - assert(dec->alpha_data_size_ >= ALPHA_HEADER_LEN + offset + num_pixels);
|
| - memcpy(dec->alpha_plane_ + offset,
|
| - dec->alpha_data_ + ALPHA_HEADER_LEN + offset, num_pixels);
|
| + int y;
|
| + const uint8_t* prev_line = dec->alpha_prev_line_;
|
| + const uint8_t* deltas = dec->alpha_data_ + ALPHA_HEADER_LEN + row * width;
|
| + uint8_t* dst = dec->alpha_plane_ + row * width;
|
| + assert(deltas <= &dec->alpha_data_[dec->alpha_data_size_]);
|
| + if (alph_dec->filter_ != WEBP_FILTER_NONE) {
|
| + assert(WebPUnfilters[alph_dec->filter_] != NULL);
|
| + for (y = 0; y < num_rows; ++y) {
|
| + WebPUnfilters[alph_dec->filter_](prev_line, deltas, dst, width);
|
| + prev_line = dst;
|
| + dst += width;
|
| + deltas += width;
|
| + }
|
| + } else {
|
| + for (y = 0; y < num_rows; ++y) {
|
| + memcpy(dst, deltas, width * sizeof(*dst));
|
| + prev_line = dst;
|
| + dst += width;
|
| + deltas += width;
|
| + }
|
| + }
|
| + dec->alpha_prev_line_ = prev_line;
|
| } else { // alph_dec->method_ == ALPHA_LOSSLESS_COMPRESSION
|
| assert(alph_dec->vp8l_dec_ != NULL);
|
| if (!VP8LDecodeAlphaImageStream(alph_dec, row + num_rows)) {
|
| @@ -106,62 +141,92 @@ static int ALPHDecode(VP8Decoder* const dec, int row, int num_rows) {
|
| }
|
| }
|
|
|
| - if (unfilter_func != NULL) {
|
| - unfilter_func(width, height, width, row, num_rows, output);
|
| + if (row + num_rows >= height) {
|
| + dec->is_alpha_decoded_ = 1;
|
| }
|
| + return 1;
|
| +}
|
|
|
| - if (row + num_rows == dec->pic_hdr_.height_) {
|
| - dec->is_alpha_decoded_ = 1;
|
| +static int AllocateAlphaPlane(VP8Decoder* const dec, const VP8Io* const io) {
|
| + const int stride = io->width;
|
| + const int height = io->crop_bottom;
|
| + const uint64_t alpha_size = (uint64_t)stride * height;
|
| + assert(dec->alpha_plane_mem_ == NULL);
|
| + dec->alpha_plane_mem_ =
|
| + (uint8_t*)WebPSafeMalloc(alpha_size, sizeof(*dec->alpha_plane_));
|
| + if (dec->alpha_plane_mem_ == NULL) {
|
| + return 0;
|
| }
|
| + dec->alpha_plane_ = dec->alpha_plane_mem_;
|
| + dec->alpha_prev_line_ = NULL;
|
| return 1;
|
| }
|
|
|
| +void WebPDeallocateAlphaMemory(VP8Decoder* const dec) {
|
| + assert(dec != NULL);
|
| + WebPSafeFree(dec->alpha_plane_mem_);
|
| + dec->alpha_plane_mem_ = NULL;
|
| + dec->alpha_plane_ = NULL;
|
| + ALPHDelete(dec->alph_dec_);
|
| + dec->alph_dec_ = NULL;
|
| +}
|
| +
|
| //------------------------------------------------------------------------------
|
| // Main entry point.
|
|
|
| const uint8_t* VP8DecompressAlphaRows(VP8Decoder* const dec,
|
| + const VP8Io* const io,
|
| int row, int num_rows) {
|
| - const int width = dec->pic_hdr_.width_;
|
| - const int height = dec->pic_hdr_.height_;
|
| + const int width = io->width;
|
| + const int height = io->crop_bottom;
|
| +
|
| + assert(dec != NULL && io != NULL);
|
|
|
| if (row < 0 || num_rows <= 0 || row + num_rows > height) {
|
| return NULL; // sanity check.
|
| }
|
|
|
| - if (row == 0) {
|
| - // Initialize decoding.
|
| - assert(dec->alpha_plane_ != NULL);
|
| - dec->alph_dec_ = ALPHNew();
|
| - if (dec->alph_dec_ == NULL) return NULL;
|
| - if (!ALPHInit(dec->alph_dec_, dec->alpha_data_, dec->alpha_data_size_,
|
| - width, height, dec->alpha_plane_)) {
|
| - ALPHDelete(dec->alph_dec_);
|
| - dec->alph_dec_ = NULL;
|
| - return NULL;
|
| - }
|
| - // if we allowed use of alpha dithering, check whether it's needed at all
|
| - if (dec->alph_dec_->pre_processing_ != ALPHA_PREPROCESSED_LEVELS) {
|
| - dec->alpha_dithering_ = 0; // disable dithering
|
| - } else {
|
| - num_rows = height; // decode everything in one pass
|
| + if (!dec->is_alpha_decoded_) {
|
| + if (dec->alph_dec_ == NULL) { // Initialize decoder.
|
| + dec->alph_dec_ = ALPHNew();
|
| + if (dec->alph_dec_ == NULL) return NULL;
|
| + if (!AllocateAlphaPlane(dec, io)) goto Error;
|
| + if (!ALPHInit(dec->alph_dec_, dec->alpha_data_, dec->alpha_data_size_,
|
| + io, dec->alpha_plane_)) {
|
| + goto Error;
|
| + }
|
| + // if we allowed use of alpha dithering, check whether it's needed at all
|
| + if (dec->alph_dec_->pre_processing_ != ALPHA_PREPROCESSED_LEVELS) {
|
| + dec->alpha_dithering_ = 0; // disable dithering
|
| + } else {
|
| + num_rows = height - row; // decode everything in one pass
|
| + }
|
| }
|
| - }
|
|
|
| - if (!dec->is_alpha_decoded_) {
|
| - int ok = 0;
|
| assert(dec->alph_dec_ != NULL);
|
| - ok = ALPHDecode(dec, row, num_rows);
|
| - if (ok && dec->alpha_dithering_ > 0) {
|
| - ok = WebPDequantizeLevels(dec->alpha_plane_, width, height,
|
| - dec->alpha_dithering_);
|
| - }
|
| - if (!ok || dec->is_alpha_decoded_) {
|
| + assert(row + num_rows <= height);
|
| + if (!ALPHDecode(dec, row, num_rows)) goto Error;
|
| +
|
| + if (dec->is_alpha_decoded_) { // finished?
|
| ALPHDelete(dec->alph_dec_);
|
| dec->alph_dec_ = NULL;
|
| + if (dec->alpha_dithering_ > 0) {
|
| + uint8_t* const alpha = dec->alpha_plane_ + io->crop_top * width
|
| + + io->crop_left;
|
| + if (!WebPDequantizeLevels(alpha,
|
| + io->crop_right - io->crop_left,
|
| + io->crop_bottom - io->crop_top,
|
| + width, dec->alpha_dithering_)) {
|
| + goto Error;
|
| + }
|
| + }
|
| }
|
| - if (!ok) return NULL; // Error.
|
| }
|
|
|
| // Return a pointer to the current decoded row.
|
| return dec->alpha_plane_ + row * width;
|
| +
|
| + Error:
|
| + WebPDeallocateAlphaMemory(dec);
|
| + return NULL;
|
| }
|
|
|