| Index: src/codec/SkJpegCodec.cpp
|
| diff --git a/src/codec/SkJpegCodec.cpp b/src/codec/SkJpegCodec.cpp
|
| index f1b55df06eb98a2ef4002495d226edf839441752..b5780c55817e670ae545d1c7f16bddc9e77368ea 100644
|
| --- a/src/codec/SkJpegCodec.cpp
|
| +++ b/src/codec/SkJpegCodec.cpp
|
| @@ -594,21 +594,21 @@
|
| (4 == hSampY && 2 == vSampY);
|
| }
|
|
|
| -bool SkJpegCodec::onQueryYUV8(SkYUVSizeInfo* sizeInfo, SkYUVColorSpace* colorSpace) const {
|
| +bool SkJpegCodec::onQueryYUV8(YUVSizeInfo* sizeInfo, SkYUVColorSpace* colorSpace) const {
|
| jpeg_decompress_struct* dinfo = fDecoderMgr->dinfo();
|
| if (!is_yuv_supported(dinfo)) {
|
| return false;
|
| }
|
|
|
| - sizeInfo->fSizes[SkYUVSizeInfo::kY].set(dinfo->comp_info[0].downsampled_width,
|
| - dinfo->comp_info[0].downsampled_height);
|
| - sizeInfo->fSizes[SkYUVSizeInfo::kU].set(dinfo->comp_info[1].downsampled_width,
|
| - dinfo->comp_info[1].downsampled_height);
|
| - sizeInfo->fSizes[SkYUVSizeInfo::kV].set(dinfo->comp_info[2].downsampled_width,
|
| - dinfo->comp_info[2].downsampled_height);
|
| - sizeInfo->fWidthBytes[SkYUVSizeInfo::kY] = dinfo->comp_info[0].width_in_blocks * DCTSIZE;
|
| - sizeInfo->fWidthBytes[SkYUVSizeInfo::kU] = dinfo->comp_info[1].width_in_blocks * DCTSIZE;
|
| - sizeInfo->fWidthBytes[SkYUVSizeInfo::kV] = dinfo->comp_info[2].width_in_blocks * DCTSIZE;
|
| + sizeInfo->fYSize.set(dinfo->comp_info[0].downsampled_width,
|
| + dinfo->comp_info[0].downsampled_height);
|
| + sizeInfo->fUSize.set(dinfo->comp_info[1].downsampled_width,
|
| + dinfo->comp_info[1].downsampled_height);
|
| + sizeInfo->fVSize.set(dinfo->comp_info[2].downsampled_width,
|
| + dinfo->comp_info[2].downsampled_height);
|
| + sizeInfo->fYWidthBytes = dinfo->comp_info[0].width_in_blocks * DCTSIZE;
|
| + sizeInfo->fUWidthBytes = dinfo->comp_info[1].width_in_blocks * DCTSIZE;
|
| + sizeInfo->fVWidthBytes = dinfo->comp_info[2].width_in_blocks * DCTSIZE;
|
|
|
| if (colorSpace) {
|
| *colorSpace = kJPEG_SkYUVColorSpace;
|
| @@ -617,18 +617,17 @@
|
| return true;
|
| }
|
|
|
| -SkCodec::Result SkJpegCodec::onGetYUV8Planes(const SkYUVSizeInfo& sizeInfo, void* planes[3]) {
|
| - SkYUVSizeInfo defaultInfo;
|
| +SkCodec::Result SkJpegCodec::onGetYUV8Planes(const YUVSizeInfo& sizeInfo, void* pixels[3]) {
|
| + YUVSizeInfo defaultInfo;
|
|
|
| // This will check is_yuv_supported(), so we don't need to here.
|
| bool supportsYUV = this->onQueryYUV8(&defaultInfo, nullptr);
|
| - if (!supportsYUV ||
|
| - sizeInfo.fSizes[SkYUVSizeInfo::kY] != defaultInfo.fSizes[SkYUVSizeInfo::kY] ||
|
| - sizeInfo.fSizes[SkYUVSizeInfo::kU] != defaultInfo.fSizes[SkYUVSizeInfo::kU] ||
|
| - sizeInfo.fSizes[SkYUVSizeInfo::kV] != defaultInfo.fSizes[SkYUVSizeInfo::kV] ||
|
| - sizeInfo.fWidthBytes[SkYUVSizeInfo::kY] < defaultInfo.fWidthBytes[SkYUVSizeInfo::kY] ||
|
| - sizeInfo.fWidthBytes[SkYUVSizeInfo::kU] < defaultInfo.fWidthBytes[SkYUVSizeInfo::kU] ||
|
| - sizeInfo.fWidthBytes[SkYUVSizeInfo::kV] < defaultInfo.fWidthBytes[SkYUVSizeInfo::kV]) {
|
| + if (!supportsYUV || sizeInfo.fYSize != defaultInfo.fYSize ||
|
| + sizeInfo.fUSize != defaultInfo.fUSize ||
|
| + sizeInfo.fVSize != defaultInfo.fVSize ||
|
| + sizeInfo.fYWidthBytes < defaultInfo.fYWidthBytes ||
|
| + sizeInfo.fUWidthBytes < defaultInfo.fUWidthBytes ||
|
| + sizeInfo.fVWidthBytes < defaultInfo.fVWidthBytes) {
|
| return fDecoderMgr->returnFailure("onGetYUV8Planes", kInvalidInput);
|
| }
|
|
|
| @@ -652,9 +651,9 @@
|
|
|
| // Currently, we require that the Y plane dimensions match the image dimensions
|
| // and that the U and V planes are the same dimensions.
|
| - SkASSERT(sizeInfo.fSizes[SkYUVSizeInfo::kU] == sizeInfo.fSizes[SkYUVSizeInfo::kV]);
|
| - SkASSERT((uint32_t) sizeInfo.fSizes[SkYUVSizeInfo::kY].width() == dinfo->output_width &&
|
| - (uint32_t) sizeInfo.fSizes[SkYUVSizeInfo::kY].height() == dinfo->output_height);
|
| + SkASSERT(sizeInfo.fUSize == sizeInfo.fVSize);
|
| + SkASSERT((uint32_t) sizeInfo.fYSize.width() == dinfo->output_width &&
|
| + (uint32_t) sizeInfo.fYSize.height() == dinfo->output_height);
|
|
|
| // Build a JSAMPIMAGE to handle output from libjpeg-turbo. A JSAMPIMAGE has
|
| // a 2-D array of pixels for each of the components (Y, U, V) in the image.
|
| @@ -671,20 +670,17 @@
|
| // Initialize rowptrs.
|
| int numYRowsPerBlock = DCTSIZE * dinfo->comp_info[0].v_samp_factor;
|
| for (int i = 0; i < numYRowsPerBlock; i++) {
|
| - rowptrs[i] = SkTAddOffset<JSAMPLE>(planes[SkYUVSizeInfo::kY],
|
| - i * sizeInfo.fWidthBytes[SkYUVSizeInfo::kY]);
|
| + rowptrs[i] = SkTAddOffset<JSAMPLE>(pixels[0], i * sizeInfo.fYWidthBytes);
|
| }
|
| for (int i = 0; i < DCTSIZE; i++) {
|
| - rowptrs[i + 2 * DCTSIZE] = SkTAddOffset<JSAMPLE>(planes[SkYUVSizeInfo::kU],
|
| - i * sizeInfo.fWidthBytes[SkYUVSizeInfo::kU]);
|
| - rowptrs[i + 3 * DCTSIZE] = SkTAddOffset<JSAMPLE>(planes[SkYUVSizeInfo::kV],
|
| - i * sizeInfo.fWidthBytes[SkYUVSizeInfo::kV]);
|
| + rowptrs[i + 2 * DCTSIZE] = SkTAddOffset<JSAMPLE>(pixels[1], i * sizeInfo.fUWidthBytes);
|
| + rowptrs[i + 3 * DCTSIZE] = SkTAddOffset<JSAMPLE>(pixels[2], i * sizeInfo.fVWidthBytes);
|
| }
|
|
|
| // After each loop iteration, we will increment pointers to Y, U, and V.
|
| - size_t blockIncrementY = numYRowsPerBlock * sizeInfo.fWidthBytes[SkYUVSizeInfo::kY];
|
| - size_t blockIncrementU = DCTSIZE * sizeInfo.fWidthBytes[SkYUVSizeInfo::kU];
|
| - size_t blockIncrementV = DCTSIZE * sizeInfo.fWidthBytes[SkYUVSizeInfo::kV];
|
| + size_t blockIncrementY = numYRowsPerBlock * sizeInfo.fYWidthBytes;
|
| + size_t blockIncrementU = DCTSIZE * sizeInfo.fUWidthBytes;
|
| + size_t blockIncrementV = DCTSIZE * sizeInfo.fVWidthBytes;
|
|
|
| uint32_t numRowsPerBlock = numYRowsPerBlock;
|
|
|
| @@ -717,7 +713,7 @@
|
| // this requirement using a dummy row buffer.
|
| // FIXME: Should SkCodec have an extra memory buffer that can be shared among
|
| // all of the implementations that use temporary/garbage memory?
|
| - SkAutoTMalloc<JSAMPLE> dummyRow(sizeInfo.fWidthBytes[SkYUVSizeInfo::kY]);
|
| + SkAutoTMalloc<JSAMPLE> dummyRow(sizeInfo.fYWidthBytes);
|
| for (int i = remainingRows; i < numYRowsPerBlock; i++) {
|
| rowptrs[i] = dummyRow.get();
|
| }
|
|
|