Index: third_party/libwebp/dec/alpha.c |
diff --git a/third_party/libwebp/dec/alpha.c b/third_party/libwebp/dec/alpha.c |
index b5e68919ca699b237b70588e3aa034f0150fa3d8..93729a035f1897290098656a3c42e17277a45ac7 100644 |
--- a/third_party/libwebp/dec/alpha.c |
+++ b/third_party/libwebp/dec/alpha.c |
@@ -12,104 +12,150 @@ |
// Author: Skal (pascal.massimino@gmail.com) |
#include <stdlib.h> |
+#include "./alphai.h" |
#include "./vp8i.h" |
#include "./vp8li.h" |
-#include "../utils/filters.h" |
#include "../utils/quant_levels_dec.h" |
#include "../webp/format_constants.h" |
-#if defined(__cplusplus) || defined(c_plusplus) |
-extern "C" { |
-#endif |
+//------------------------------------------------------------------------------ |
+// ALPHDecoder object. |
+ |
+ALPHDecoder* ALPHNew(void) { |
+ ALPHDecoder* const dec = (ALPHDecoder*)calloc(1, sizeof(*dec)); |
+ return dec; |
+} |
+ |
+void ALPHDelete(ALPHDecoder* const dec) { |
+ if (dec != NULL) { |
+ VP8LDelete(dec->vp8l_dec_); |
+ dec->vp8l_dec_ = NULL; |
+ free(dec); |
+ } |
+} |
//------------------------------------------------------------------------------ |
-// Decodes the compressed data 'data' of size 'data_size' into the 'output'. |
-// The 'output' buffer should be pre-allocated and must be of the same |
-// dimension 'height'x'width', as that of the image. |
-// |
-// Returns 1 on successfully decoding the compressed alpha and |
-// 0 if either: |
-// error in bit-stream header (invalid compression mode or filter), or |
-// error returned by appropriate compression method. |
- |
-static int DecodeAlpha(const uint8_t* data, size_t data_size, |
- int width, int height, uint8_t* output) { |
- WEBP_FILTER_TYPE filter; |
- int pre_processing; |
- int rsrv; |
+// Decoding. |
+ |
+// Initialize alpha decoding by parsing the alpha header and decoding the image |
+// header for alpha data stored using lossless compression. |
+// 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) { |
int ok = 0; |
- int method; |
const uint8_t* const alpha_data = data + ALPHA_HEADER_LEN; |
const size_t alpha_data_size = data_size - ALPHA_HEADER_LEN; |
+ int rsrv; |
assert(width > 0 && height > 0); |
assert(data != NULL && output != NULL); |
+ dec->width_ = width; |
+ dec->height_ = height; |
+ |
if (data_size <= ALPHA_HEADER_LEN) { |
return 0; |
} |
- method = (data[0] >> 0) & 0x03; |
- filter = (data[0] >> 2) & 0x03; |
- pre_processing = (data[0] >> 4) & 0x03; |
+ dec->method_ = (data[0] >> 0) & 0x03; |
+ dec->filter_ = (data[0] >> 2) & 0x03; |
+ dec->pre_processing_ = (data[0] >> 4) & 0x03; |
rsrv = (data[0] >> 6) & 0x03; |
- if (method < ALPHA_NO_COMPRESSION || |
- method > ALPHA_LOSSLESS_COMPRESSION || |
- filter >= WEBP_FILTER_LAST || |
- pre_processing > ALPHA_PREPROCESSED_LEVELS || |
+ if (dec->method_ < ALPHA_NO_COMPRESSION || |
+ dec->method_ > ALPHA_LOSSLESS_COMPRESSION || |
+ dec->filter_ >= WEBP_FILTER_LAST || |
+ dec->pre_processing_ > ALPHA_PREPROCESSED_LEVELS || |
rsrv != 0) { |
return 0; |
} |
- if (method == ALPHA_NO_COMPRESSION) { |
- const size_t alpha_decoded_size = height * width; |
+ if (dec->method_ == ALPHA_NO_COMPRESSION) { |
+ const size_t alpha_decoded_size = dec->width_ * dec->height_; |
ok = (alpha_data_size >= alpha_decoded_size); |
- if (ok) memcpy(output, alpha_data, alpha_decoded_size); |
} else { |
- ok = VP8LDecodeAlphaImageStream(width, height, alpha_data, alpha_data_size, |
- output); |
+ assert(dec->method_ == ALPHA_LOSSLESS_COMPRESSION); |
+ ok = VP8LDecodeAlphaHeader(dec, alpha_data, alpha_data_size, output); |
} |
+ return ok; |
+} |
- if (ok) { |
- WebPUnfilterFunc unfilter_func = WebPUnfilters[filter]; |
- if (unfilter_func != NULL) { |
- // TODO(vikas): Implement on-the-fly decoding & filter mechanism to decode |
- // and apply filter per image-row. |
- unfilter_func(width, height, width, output); |
+// Decodes, unfilters and dequantizes *at least* 'num_rows' rows of alpha |
+// starting from row number 'row'. It assumes that rows up to (row - 1) have |
+// already been decoded. |
+// Returns false in case of bitstream error. |
+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_; |
+ 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); |
+ } else { // alph_dec->method_ == ALPHA_LOSSLESS_COMPRESSION |
+ assert(alph_dec->vp8l_dec_ != NULL); |
+ if (!VP8LDecodeAlphaImageStream(alph_dec, row + num_rows)) { |
+ return 0; |
} |
- if (pre_processing == ALPHA_PREPROCESSED_LEVELS) { |
- ok = DequantizeLevels(output, width, height); |
+ } |
+ |
+ if (unfilter_func != NULL) { |
+ unfilter_func(width, height, width, row, num_rows, output); |
+ } |
+ |
+ if (alph_dec->pre_processing_ == ALPHA_PREPROCESSED_LEVELS) { |
+ if (!DequantizeLevels(output, width, height, row, num_rows)) { |
+ return 0; |
} |
} |
- return ok; |
+ if (row + num_rows == dec->pic_hdr_.height_) { |
+ dec->is_alpha_decoded_ = 1; |
+ } |
+ return 1; |
} |
//------------------------------------------------------------------------------ |
+// Main entry point. |
const uint8_t* VP8DecompressAlphaRows(VP8Decoder* const dec, |
int row, int num_rows) { |
const int width = dec->pic_hdr_.width_; |
const int height = dec->pic_hdr_.height_; |
- if (row < 0 || num_rows < 0 || row + num_rows > height) { |
+ if (row < 0 || num_rows <= 0 || row + num_rows > height) { |
return NULL; // sanity check. |
} |
if (row == 0) { |
- // Decode everything during the first call. |
- assert(!dec->is_alpha_decoded_); |
- if (!DecodeAlpha(dec->alpha_data_, (size_t)dec->alpha_data_size_, |
- width, height, dec->alpha_plane_)) { |
- return NULL; // Error. |
+ // 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; |
} |
- dec->is_alpha_decoded_ = 1; |
+ } |
+ |
+ if (!dec->is_alpha_decoded_) { |
+ int ok = 0; |
+ assert(dec->alph_dec_ != NULL); |
+ ok = ALPHDecode(dec, row, num_rows); |
+ if (!ok || dec->is_alpha_decoded_) { |
+ ALPHDelete(dec->alph_dec_); |
+ dec->alph_dec_ = NULL; |
+ } |
+ if (!ok) return NULL; // Error. |
} |
// Return a pointer to the current decoded row. |
return dec->alpha_plane_ + row * width; |
} |
-#if defined(__cplusplus) || defined(c_plusplus) |
-} // extern "C" |
-#endif |