| /* |
| * Copyright (C) 2013 Google Inc. All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions are |
| * met: |
| * |
| * * Redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer. |
| * * Redistributions in binary form must reproduce the above |
| * copyright notice, this list of conditions and the following disclaimer |
| * in the documentation and/or other materials provided with the |
| * distribution. |
| * * Neither the name of Google Inc. nor the names of its |
| * contributors may be used to endorse or promote products derived from |
| * this software without specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| #include "platform/image-decoders/jpeg/JPEGImageDecoder.h" |
| |
| #include "platform/SharedBuffer.h" |
| #include "platform/image-decoders/ImageAnimation.h" |
| #include "platform/image-decoders/ImageDecoderTestHelpers.h" |
| #include "public/platform/WebData.h" |
| #include "public/platform/WebSize.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| #include "wtf/PtrUtil.h" |
| #include "wtf/typed_arrays/ArrayBuffer.h" |
| #include <memory> |
| |
| namespace blink { |
| |
| static const size_t LargeEnoughSize = 1000 * 1000; |
| |
| namespace { |
| |
| std::unique_ptr<ImageDecoder> createDecoder(size_t maxDecodedBytes) { |
| return wrapUnique(new JPEGImageDecoder( |
| ImageDecoder::AlphaNotPremultiplied, |
| ImageDecoder::GammaAndColorProfileApplied, maxDecodedBytes)); |
| } |
| |
| std::unique_ptr<ImageDecoder> createDecoder() { |
| return createDecoder(ImageDecoder::noDecodedImageByteLimit); |
| } |
| |
| } // anonymous namespace |
| |
| void downsample(size_t maxDecodedBytes, |
| unsigned* outputWidth, |
| unsigned* outputHeight, |
| const char* imageFilePath) { |
| RefPtr<SharedBuffer> data = readFile(imageFilePath); |
| ASSERT_TRUE(data); |
| |
| std::unique_ptr<ImageDecoder> decoder = createDecoder(maxDecodedBytes); |
| decoder->setData(data.get(), true); |
| |
| ImageFrame* frame = decoder->frameBufferAtIndex(0); |
| ASSERT_TRUE(frame); |
| *outputWidth = frame->bitmap().width(); |
| *outputHeight = frame->bitmap().height(); |
| EXPECT_EQ(IntSize(*outputWidth, *outputHeight), decoder->decodedSize()); |
| } |
| |
| void readYUV(size_t maxDecodedBytes, |
| unsigned* outputYWidth, |
| unsigned* outputYHeight, |
| unsigned* outputUVWidth, |
| unsigned* outputUVHeight, |
| const char* imageFilePath) { |
| RefPtr<SharedBuffer> data = readFile(imageFilePath); |
| ASSERT_TRUE(data); |
| |
| std::unique_ptr<ImageDecoder> decoder = createDecoder(maxDecodedBytes); |
| decoder->setData(data.get(), true); |
| |
| // Setting a dummy ImagePlanes object signals to the decoder that we want to |
| // do YUV decoding. |
| std::unique_ptr<ImagePlanes> dummyImagePlanes = wrapUnique(new ImagePlanes()); |
| decoder->setImagePlanes(std::move(dummyImagePlanes)); |
| |
| bool sizeIsAvailable = decoder->isSizeAvailable(); |
| ASSERT_TRUE(sizeIsAvailable); |
| |
| IntSize size = decoder->decodedSize(); |
| IntSize ySize = decoder->decodedYUVSize(0); |
| IntSize uSize = decoder->decodedYUVSize(1); |
| IntSize vSize = decoder->decodedYUVSize(2); |
| |
| ASSERT_TRUE(size.width() == ySize.width()); |
| ASSERT_TRUE(size.height() == ySize.height()); |
| ASSERT_TRUE(uSize.width() == vSize.width()); |
| ASSERT_TRUE(uSize.height() == vSize.height()); |
| |
| *outputYWidth = ySize.width(); |
| *outputYHeight = ySize.height(); |
| *outputUVWidth = uSize.width(); |
| *outputUVHeight = uSize.height(); |
| |
| size_t rowBytes[3]; |
| rowBytes[0] = decoder->decodedYUVWidthBytes(0); |
| rowBytes[1] = decoder->decodedYUVWidthBytes(1); |
| rowBytes[2] = decoder->decodedYUVWidthBytes(2); |
| |
| RefPtr<ArrayBuffer> buffer(ArrayBuffer::create( |
| rowBytes[0] * ySize.height() + rowBytes[1] * uSize.height() + |
| rowBytes[2] * vSize.height(), |
| 1)); |
| void* planes[3]; |
| planes[0] = buffer->data(); |
| planes[1] = ((char*)planes[0]) + rowBytes[0] * ySize.height(); |
| planes[2] = ((char*)planes[1]) + rowBytes[1] * uSize.height(); |
| |
| std::unique_ptr<ImagePlanes> imagePlanes = |
| wrapUnique(new ImagePlanes(planes, rowBytes)); |
| decoder->setImagePlanes(std::move(imagePlanes)); |
| |
| ASSERT_TRUE(decoder->decodeToYUV()); |
| } |
| |
| // Tests failure on a too big image. |
| TEST(JPEGImageDecoderTest, tooBig) { |
| std::unique_ptr<ImageDecoder> decoder = createDecoder(100); |
| EXPECT_FALSE(decoder->setSize(10000, 10000)); |
| EXPECT_TRUE(decoder->failed()); |
| } |
| |
| // Tests that the JPEG decoder can downsample image whose width and height are |
| // multiples of 8, to ensure we compute the correct decodedSize and pass correct |
| // parameters to libjpeg to output the image with the expected size. |
| TEST(JPEGImageDecoderTest, downsampleImageSizeMultipleOf8) { |
| const char* jpegFile = |
| "/LayoutTests/fast/images/resources/lenna.jpg"; // 256x256 |
| unsigned outputWidth, outputHeight; |
| |
| // 1/8 downsample. |
| downsample(40 * 40 * 4, &outputWidth, &outputHeight, jpegFile); |
| EXPECT_EQ(32u, outputWidth); |
| EXPECT_EQ(32u, outputHeight); |
| |
| // 2/8 downsample. |
| downsample(70 * 70 * 4, &outputWidth, &outputHeight, jpegFile); |
| EXPECT_EQ(64u, outputWidth); |
| EXPECT_EQ(64u, outputHeight); |
| |
| // 3/8 downsample. |
| downsample(100 * 100 * 4, &outputWidth, &outputHeight, jpegFile); |
| EXPECT_EQ(96u, outputWidth); |
| EXPECT_EQ(96u, outputHeight); |
| |
| // 4/8 downsample. |
| downsample(130 * 130 * 4, &outputWidth, &outputHeight, jpegFile); |
| EXPECT_EQ(128u, outputWidth); |
| EXPECT_EQ(128u, outputHeight); |
| |
| // 5/8 downsample. |
| downsample(170 * 170 * 4, &outputWidth, &outputHeight, jpegFile); |
| EXPECT_EQ(160u, outputWidth); |
| EXPECT_EQ(160u, outputHeight); |
| |
| // 6/8 downsample. |
| downsample(200 * 200 * 4, &outputWidth, &outputHeight, jpegFile); |
| EXPECT_EQ(192u, outputWidth); |
| EXPECT_EQ(192u, outputHeight); |
| |
| // 7/8 downsample. |
| downsample(230 * 230 * 4, &outputWidth, &outputHeight, jpegFile); |
| EXPECT_EQ(224u, outputWidth); |
| EXPECT_EQ(224u, outputHeight); |
| } |
| |
| // Tests that JPEG decoder can downsample image whose width and height are not |
| // multiple of 8. Ensures that we round using the same algorithm as libjpeg. |
| TEST(JPEGImageDecoderTest, downsampleImageSizeNotMultipleOf8) { |
| const char* jpegFile = |
| "/LayoutTests/fast/images/resources/icc-v2-gbr.jpg"; // 275x207 |
| unsigned outputWidth, outputHeight; |
| |
| // 1/8 downsample. |
| downsample(40 * 40 * 4, &outputWidth, &outputHeight, jpegFile); |
| EXPECT_EQ(35u, outputWidth); |
| EXPECT_EQ(26u, outputHeight); |
| |
| // 2/8 downsample. |
| downsample(70 * 70 * 4, &outputWidth, &outputHeight, jpegFile); |
| EXPECT_EQ(69u, outputWidth); |
| EXPECT_EQ(52u, outputHeight); |
| |
| // 3/8 downsample. |
| downsample(100 * 100 * 4, &outputWidth, &outputHeight, jpegFile); |
| EXPECT_EQ(104u, outputWidth); |
| EXPECT_EQ(78u, outputHeight); |
| |
| // 4/8 downsample. |
| downsample(130 * 130 * 4, &outputWidth, &outputHeight, jpegFile); |
| EXPECT_EQ(138u, outputWidth); |
| EXPECT_EQ(104u, outputHeight); |
| |
| // 5/8 downsample. |
| downsample(170 * 170 * 4, &outputWidth, &outputHeight, jpegFile); |
| EXPECT_EQ(172u, outputWidth); |
| EXPECT_EQ(130u, outputHeight); |
| |
| // 6/8 downsample. |
| downsample(200 * 200 * 4, &outputWidth, &outputHeight, jpegFile); |
| EXPECT_EQ(207u, outputWidth); |
| EXPECT_EQ(156u, outputHeight); |
| |
| // 7/8 downsample. |
| downsample(230 * 230 * 4, &outputWidth, &outputHeight, jpegFile); |
| EXPECT_EQ(241u, outputWidth); |
| EXPECT_EQ(182u, outputHeight); |
| } |
| |
| // Tests that upsampling is not allowed. |
| TEST(JPEGImageDecoderTest, upsample) { |
| const char* jpegFile = |
| "/LayoutTests/fast/images/resources/lenna.jpg"; // 256x256 |
| unsigned outputWidth, outputHeight; |
| downsample(LargeEnoughSize, &outputWidth, &outputHeight, jpegFile); |
| EXPECT_EQ(256u, outputWidth); |
| EXPECT_EQ(256u, outputHeight); |
| } |
| |
| TEST(JPEGImageDecoderTest, yuv) { |
| const char* jpegFile = |
| "/LayoutTests/fast/images/resources/lenna.jpg"; // 256x256, YUV 4:2:0 |
| unsigned outputYWidth, outputYHeight, outputUVWidth, outputUVHeight; |
| readYUV(LargeEnoughSize, &outputYWidth, &outputYHeight, &outputUVWidth, |
| &outputUVHeight, jpegFile); |
| EXPECT_EQ(256u, outputYWidth); |
| EXPECT_EQ(256u, outputYHeight); |
| EXPECT_EQ(128u, outputUVWidth); |
| EXPECT_EQ(128u, outputUVHeight); |
| |
| const char* jpegFileImageSizeNotMultipleOf8 = |
| "/LayoutTests/fast/images/resources/cropped_mandrill.jpg"; // 439x154 |
| readYUV(LargeEnoughSize, &outputYWidth, &outputYHeight, &outputUVWidth, |
| &outputUVHeight, jpegFileImageSizeNotMultipleOf8); |
| EXPECT_EQ(439u, outputYWidth); |
| EXPECT_EQ(154u, outputYHeight); |
| EXPECT_EQ(220u, outputUVWidth); |
| EXPECT_EQ(77u, outputUVHeight); |
| |
| // Make sure we revert to RGBA decoding when we're about to downscale, |
| // which can occur on memory-constrained android devices. |
| RefPtr<SharedBuffer> data = readFile(jpegFile); |
| ASSERT_TRUE(data); |
| |
| std::unique_ptr<ImageDecoder> decoder = createDecoder(230 * 230 * 4); |
| decoder->setData(data.get(), true); |
| |
| std::unique_ptr<ImagePlanes> imagePlanes = wrapUnique(new ImagePlanes()); |
| decoder->setImagePlanes(std::move(imagePlanes)); |
| ASSERT_TRUE(decoder->isSizeAvailable()); |
| ASSERT_FALSE(decoder->canDecodeToYUV()); |
| } |
| |
| TEST(JPEGImageDecoderTest, |
| byteByByteBaselineJPEGWithColorProfileAndRestartMarkers) { |
| testByteByByteDecode(&createDecoder, |
| "/LayoutTests/fast/images/resources/" |
| "small-square-with-colorspin-profile.jpg", |
| 1u, cAnimationNone); |
| } |
| |
| TEST(JPEGImageDecoderTest, byteByByteProgressiveJPEG) { |
| testByteByByteDecode(&createDecoder, |
| "/LayoutTests/fast/images/resources/bug106024.jpg", 1u, |
| cAnimationNone); |
| } |
| |
| TEST(JPEGImageDecoderTest, byteByByteRGBJPEGWithAdobeMarkers) { |
| testByteByByteDecode( |
| &createDecoder, |
| "/LayoutTests/fast/images/resources/rgb-jpeg-with-adobe-marker-only.jpg", |
| 1u, cAnimationNone); |
| } |
| |
| // This test verifies that calling SharedBuffer::mergeSegmentsIntoBuffer() does |
| // not break JPEG decoding at a critical point: in between a call to decode the |
| // size (when JPEGImageDecoder stops while it may still have input data to |
| // read) and a call to do a full decode. |
| TEST(JPEGImageDecoderTest, mergeBuffer) { |
| const char* jpegFile = "/LayoutTests/fast/images/resources/lenna.jpg"; |
| testMergeBuffer(&createDecoder, jpegFile); |
| } |
| |
| } // namespace blink |