Source/core/platform/image-decoders/webp/WEBPImageDecoderTest.cpp - platform/external/chromium_org/third_party/WebKit - Git at Google

 /*
  * 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 "config.h"

 #include "core/platform/image-decoders/webp/WEBPImageDecoder.h"

 #include "RuntimeEnabledFeatures.h"
 #include "core/platform/SharedBuffer.h"
 #include "public/platform/Platform.h"
 #include "public/platform/WebData.h"
 #include "public/platform/WebSize.h"
 #include "public/platform/WebUnitTestSupport.h"
 #include "wtf/OwnPtr.h"
 #include "wtf/PassOwnPtr.h"
 #include "wtf/StringHasher.h"
 #include "wtf/Vector.h"
 #include "wtf/dtoa/utils.h"
 #include <gtest/gtest.h>

 using namespace WebCore;
 using namespace WebKit;

 #if !OS(ANDROID)

 namespace {

 PassRefPtr<SharedBuffer> readFile(const char* fileName)
 {
     String filePath = Platform::current()->unitTestSupport()->webKitRootDir();
     filePath.append(fileName);

     return Platform::current()->unitTestSupport()->readFromFile(filePath);
 }

 PassOwnPtr<WEBPImageDecoder> createDecoder()
 {
     return adoptPtr(new WEBPImageDecoder(ImageSource::AlphaNotPremultiplied, ImageSource::GammaAndColorProfileApplied));
 }

 unsigned hashSkBitmap(const SkBitmap& bitmap)
 {
     return StringHasher::hashMemory(bitmap.getPixels(), bitmap.getSize());
 }

 void createDecodingBaseline(SharedBuffer* data, Vector<unsigned>* baselineHashes)
 {
     OwnPtr<WEBPImageDecoder> decoder = createDecoder();
     decoder->setData(data, true);
     size_t frameCount = decoder->frameCount();
     for (size_t i = 0; i < frameCount; ++i) {
         ImageFrame* frame = decoder->frameBufferAtIndex(i);
         baselineHashes->append(hashSkBitmap(frame->getSkBitmap()));
     }
 }

 void testRandomFrameDecode(const char* webpFile)
 {
     SCOPED_TRACE(webpFile);

     RefPtr<SharedBuffer> fullData = readFile(webpFile);
     ASSERT_TRUE(fullData.get());
     Vector<unsigned> baselineHashes;
     createDecodingBaseline(fullData.get(), &baselineHashes);
     size_t frameCount = baselineHashes.size();

     // Random decoding should get the same results as sequential decoding.
     OwnPtr<WEBPImageDecoder> decoder = createDecoder();
     decoder->setData(fullData.get(), true);
     const size_t skippingStep = 5;
     for (size_t i = 0; i < skippingStep; ++i) {
         for (size_t j = i; j < frameCount; j += skippingStep) {
             SCOPED_TRACE(testing::Message() << "Random i:" << i << " j:" << j);
             ImageFrame* frame = decoder->frameBufferAtIndex(j);
             EXPECT_EQ(baselineHashes[j], hashSkBitmap(frame->getSkBitmap()));
         }
     }

     // Decoding in reverse order.
     decoder = createDecoder();
     decoder->setData(fullData.get(), true);
     for (size_t i = frameCount; i; --i) {
         SCOPED_TRACE(testing::Message() << "Reverse i:" << i);
         ImageFrame* frame = decoder->frameBufferAtIndex(i - 1);
         EXPECT_EQ(baselineHashes[i - 1], hashSkBitmap(frame->getSkBitmap()));
     }
 }

 void testRandomDecodeAfterClearFrameBufferCache(const char* webpFile)
 {
     SCOPED_TRACE(webpFile);

     RefPtr<SharedBuffer> data = readFile(webpFile);
     ASSERT_TRUE(data.get());
     Vector<unsigned> baselineHashes;
     createDecodingBaseline(data.get(), &baselineHashes);
     size_t frameCount = baselineHashes.size();

     OwnPtr<WEBPImageDecoder> decoder = createDecoder();
     decoder->setData(data.get(), true);
     for (size_t clearExceptFrame = 0; clearExceptFrame < frameCount; ++clearExceptFrame) {
         decoder->clearCacheExceptFrame(clearExceptFrame);
         const size_t skippingStep = 5;
         for (size_t i = 0; i < skippingStep; ++i) {
             for (size_t j = 0; j < frameCount; j += skippingStep) {
                 SCOPED_TRACE(testing::Message() << "Random i:" << i << " j:" << j);
                 ImageFrame* frame = decoder->frameBufferAtIndex(j);
                 EXPECT_EQ(baselineHashes[j], hashSkBitmap(frame->getSkBitmap()));
             }
         }
     }
 }

 void testDecodeAfterReallocatingData(const char* webpFile)
 {
     OwnPtr<WEBPImageDecoder> decoder = createDecoder();
     RefPtr<SharedBuffer> data = readFile(webpFile);
     ASSERT_TRUE(data.get());

     // Parse from 'data'.
     decoder->setData(data.get(), true);
     size_t frameCount = decoder->frameCount();

     // ... and then decode frames from 'reallocatedData'.
     RefPtr<SharedBuffer> reallocatedData = data.get()->copy();
     ASSERT_TRUE(reallocatedData.get());
     data.clear();
     decoder->setData(reallocatedData.get(), true);

     for (size_t i = 0; i < frameCount; ++i) {
         const ImageFrame* const frame = decoder->frameBufferAtIndex(i);
         EXPECT_EQ(ImageFrame::FrameComplete, frame->status());
     }
 }

 } // namespace

 class AnimatedWebPTests : public ::testing::Test {
 protected:
     virtual void SetUp()
     {
         // Enable animated WebP for all the tests.
         WebCore::RuntimeEnabledFeatures::setAnimatedWebPEnabled(true);
     }
 };

 TEST_F(AnimatedWebPTests, verifyAnimationParametersTransparentImage)
 {
     OwnPtr<WEBPImageDecoder> decoder = createDecoder();
     EXPECT_EQ(cAnimationLoopOnce, decoder->repetitionCount());

     RefPtr<SharedBuffer> data = readFile("/LayoutTests/fast/images/resources/webp-animated.webp");
     ASSERT_TRUE(data.get());
     decoder->setData(data.get(), true);

     const int canvasWidth = 11;
     const int canvasHeight = 29;
     const struct AnimParam {
         int xOffset, yOffset, width, height;
         ImageFrame::FrameDisposalMethod dispose;
         unsigned duration;
         bool hasAlpha;
     } frameParameters[] = {
         { 0, 0, 11, 29, ImageFrame::DisposeKeep, 1000u, true },
         { 2, 10, 7, 17, ImageFrame::DisposeKeep, 500u, true },
         { 2, 2, 7, 16, ImageFrame::DisposeKeep, 1000u, true },
     };

     for (size_t i = 0; i < ARRAY_SIZE(frameParameters); ++i) {
         const ImageFrame* const frame = decoder->frameBufferAtIndex(i);
         EXPECT_EQ(ImageFrame::FrameComplete, frame->status());
         EXPECT_EQ(canvasWidth, frame->getSkBitmap().width());
         EXPECT_EQ(canvasHeight, frame->getSkBitmap().height());
         EXPECT_EQ(frameParameters[i].xOffset, frame->originalFrameRect().x());
         EXPECT_EQ(frameParameters[i].yOffset, frame->originalFrameRect().y());
         EXPECT_EQ(frameParameters[i].width, frame->originalFrameRect().width());
         EXPECT_EQ(frameParameters[i].height, frame->originalFrameRect().height());
         EXPECT_EQ(frameParameters[i].dispose, frame->disposalMethod());
         EXPECT_EQ(frameParameters[i].duration, frame->duration());
         EXPECT_EQ(frameParameters[i].hasAlpha, frame->hasAlpha());
     }

     EXPECT_EQ(ARRAY_SIZE(frameParameters), decoder->frameCount());
     EXPECT_EQ(cAnimationLoopInfinite, decoder->repetitionCount());
 }

 TEST_F(AnimatedWebPTests, verifyAnimationParametersOpaqueFramesTransparentBackground)
 {
     OwnPtr<WEBPImageDecoder> decoder = createDecoder();
     EXPECT_EQ(cAnimationLoopOnce, decoder->repetitionCount());

     RefPtr<SharedBuffer> data = readFile("/LayoutTests/fast/images/resources/webp-animated-opaque.webp");
     ASSERT_TRUE(data.get());
     decoder->setData(data.get(), true);

     const int canvasWidth = 94;
     const int canvasHeight = 87;
     const struct AnimParam {
         int xOffset, yOffset, width, height;
         ImageFrame::FrameDisposalMethod dispose;
         unsigned duration;
         bool hasAlpha;
     } frameParameters[] = {
         { 4, 10, 33, 32, ImageFrame::DisposeOverwriteBgcolor, 1000u, true },
         { 34, 30, 33, 32, ImageFrame::DisposeOverwriteBgcolor, 1000u, true },
         { 62, 50, 32, 32, ImageFrame::DisposeOverwriteBgcolor, 1000u, true },
         { 10, 54, 32, 33, ImageFrame::DisposeOverwriteBgcolor, 1000u, true },
     };

     for (size_t i = 0; i < ARRAY_SIZE(frameParameters); ++i) {
         const ImageFrame* const frame = decoder->frameBufferAtIndex(i);
         EXPECT_EQ(ImageFrame::FrameComplete, frame->status());
         EXPECT_EQ(canvasWidth, frame->getSkBitmap().width());
         EXPECT_EQ(canvasHeight, frame->getSkBitmap().height());
         EXPECT_EQ(frameParameters[i].xOffset, frame->originalFrameRect().x());
         EXPECT_EQ(frameParameters[i].yOffset, frame->originalFrameRect().y());
         EXPECT_EQ(frameParameters[i].width, frame->originalFrameRect().width());
         EXPECT_EQ(frameParameters[i].height, frame->originalFrameRect().height());
         EXPECT_EQ(frameParameters[i].dispose, frame->disposalMethod());
         EXPECT_EQ(frameParameters[i].duration, frame->duration());
         EXPECT_EQ(frameParameters[i].hasAlpha, frame->hasAlpha());
     }

     EXPECT_EQ(ARRAY_SIZE(frameParameters), decoder->frameCount());
     EXPECT_EQ(cAnimationLoopInfinite, decoder->repetitionCount());
 }

 TEST_F(AnimatedWebPTests, parseAndDecodeByteByByte)
 {
     const struct TestImage {
         const char* filename;
         unsigned frameCount;
         int repetitionCount;
     } testImages[] = {
         { "/LayoutTests/fast/images/resources/webp-animated.webp", 3u, cAnimationLoopInfinite },
         { "/LayoutTests/fast/images/resources/webp-animated-icc-xmp.webp", 13u, 32000 },
     };

     for (size_t i = 0; i < ARRAY_SIZE(testImages); ++i) {
         OwnPtr<WEBPImageDecoder> decoder = createDecoder();
         RefPtr<SharedBuffer> data = readFile(testImages[i].filename);
         ASSERT_TRUE(data.get());

         size_t frameCount = 0;
         size_t framesDecoded = 0;

         // Pass data to decoder byte by byte.
         for (size_t length = 1; length <= data->size(); ++length) {
             RefPtr<SharedBuffer> tempData = SharedBuffer::create(data->data(), length);
             decoder->setData(tempData.get(), length == data->size());

             EXPECT_LE(frameCount, decoder->frameCount());
             frameCount = decoder->frameCount();

             ImageFrame* frame = decoder->frameBufferAtIndex(frameCount - 1);
             if (frame && frame->status() == ImageFrame::FrameComplete && framesDecoded < frameCount)
                 ++framesDecoded;
         }

         EXPECT_EQ(testImages[i].frameCount, decoder->frameCount());
         EXPECT_EQ(testImages[i].frameCount, framesDecoded);
         EXPECT_EQ(testImages[i].repetitionCount, decoder->repetitionCount());
     }
 }

 TEST_F(AnimatedWebPTests, invalidImage)
 {
     OwnPtr<WEBPImageDecoder> decoder = createDecoder();

     RefPtr<SharedBuffer> data = readFile("/LayoutTests/fast/images/resources/invalid-animated-webp.webp");
     ASSERT_TRUE(data.get());
     decoder->setData(data.get(), true);

     EXPECT_EQ(0u, decoder->frameCount());
     ImageFrame* frame = decoder->frameBufferAtIndex(0);
     EXPECT_FALSE(frame);
     EXPECT_EQ(cAnimationLoopOnce, decoder->repetitionCount());
 }

 TEST_F(AnimatedWebPTests, progressiveDecode)
 {
     RefPtr<SharedBuffer> fullData = readFile("/LayoutTests/fast/images/resources/webp-animated.webp");
     ASSERT_TRUE(fullData.get());
     const size_t fullLength = fullData->size();

     OwnPtr<WEBPImageDecoder>  decoder;
     ImageFrame* frame;

     Vector<unsigned> truncatedHashes;
     Vector<unsigned> progressiveHashes;

     // Compute hashes when the file is truncated.
     const size_t increment = 1;
     for (size_t i = 1; i <= fullLength; i += increment) {
         decoder = createDecoder();
         RefPtr<SharedBuffer> data = SharedBuffer::create(fullData->data(), i);
         decoder->setData(data.get(), i == fullLength);
         frame = decoder->frameBufferAtIndex(0);
         if (!frame) {
             truncatedHashes.append(0);
             continue;
         }
         truncatedHashes.append(hashSkBitmap(frame->getSkBitmap()));
     }

     // Compute hashes when the file is progressively decoded.
     decoder = createDecoder();
     for (size_t i = 1; i <= fullLength; i += increment) {
         RefPtr<SharedBuffer> data = SharedBuffer::create(fullData->data(), i);
         decoder->setData(data.get(), i == fullLength);
         frame = decoder->frameBufferAtIndex(0);
         if (!frame) {
             progressiveHashes.append(0);
             continue;
         }
         progressiveHashes.append(hashSkBitmap(frame->getSkBitmap()));
     }

     bool match = true;
     for (size_t i = 0; i < truncatedHashes.size(); ++i) {
         if (truncatedHashes[i] != progressiveHashes[i]) {
             match = false;
             break;
         }
     }
     EXPECT_TRUE(match);
 }

 TEST_F(AnimatedWebPTests, frameIsCompleteAndDuration)
 {
     OwnPtr<WEBPImageDecoder> decoder = createDecoder();

     RefPtr<SharedBuffer> data = readFile("/LayoutTests/fast/images/resources/webp-animated.webp");
     ASSERT_TRUE(data.get());

     ASSERT_GE(data->size(), 10u);
     RefPtr<SharedBuffer> tempData = SharedBuffer::create(data->data(), data->size() - 10);
     decoder->setData(tempData.get(), false);

     EXPECT_EQ(2u, decoder->frameCount());
     EXPECT_FALSE(decoder->failed());
     EXPECT_TRUE(decoder->frameIsCompleteAtIndex(0));
     EXPECT_EQ(1000, decoder->frameDurationAtIndex(0));
     EXPECT_TRUE(decoder->frameIsCompleteAtIndex(1));
     EXPECT_EQ(500, decoder->frameDurationAtIndex(1));

     decoder->setData(data.get(), true);
     EXPECT_EQ(3u, decoder->frameCount());
     EXPECT_TRUE(decoder->frameIsCompleteAtIndex(0));
     EXPECT_EQ(1000, decoder->frameDurationAtIndex(0));
     EXPECT_TRUE(decoder->frameIsCompleteAtIndex(1));
     EXPECT_EQ(500, decoder->frameDurationAtIndex(1));
     EXPECT_TRUE(decoder->frameIsCompleteAtIndex(2));
     EXPECT_EQ(1000.0, decoder->frameDurationAtIndex(2));
 }

 TEST_F(AnimatedWebPTests, updateRequiredPreviousFrameAfterFirstDecode)
 {
     OwnPtr<WEBPImageDecoder> decoder = createDecoder();

     RefPtr<SharedBuffer> fullData = readFile("/LayoutTests/fast/images/resources/webp-animated.webp");
     ASSERT_TRUE(fullData.get());

     // Give it data that is enough to parse but not decode in order to check the status
     // of requiredPreviousFrameIndex before decoding.
     size_t partialSize = 1;
     do {
         RefPtr<SharedBuffer> data = SharedBuffer::create(fullData->data(), partialSize);
         decoder->setData(data.get(), false);
         ++partialSize;
     } while (!decoder->frameCount() || decoder->frameBufferAtIndex(0)->status() == ImageFrame::FrameEmpty);

     EXPECT_EQ(notFound, decoder->frameBufferAtIndex(0)->requiredPreviousFrameIndex());
     unsigned frameCount = decoder->frameCount();
     for (size_t i = 1; i < frameCount; ++i)
         EXPECT_EQ(i - 1, decoder->frameBufferAtIndex(i)->requiredPreviousFrameIndex());

     decoder->setData(fullData.get(), true);
     for (size_t i = 0; i < frameCount; ++i)
         EXPECT_EQ(notFound, decoder->frameBufferAtIndex(i)->requiredPreviousFrameIndex());
 }

 TEST_F(AnimatedWebPTests, randomFrameDecode)
 {
     testRandomFrameDecode("/LayoutTests/fast/images/resources/webp-animated.webp");
     testRandomFrameDecode("/LayoutTests/fast/images/resources/webp-animated-opaque.webp");
     testRandomFrameDecode("/LayoutTests/fast/images/resources/webp-animated-large.webp");
     testRandomFrameDecode("/LayoutTests/fast/images/resources/webp-animated-icc-xmp.webp");
 }

 TEST_F(AnimatedWebPTests, randomDecodeAfterClearFrameBufferCache)
 {
     testRandomDecodeAfterClearFrameBufferCache("/LayoutTests/fast/images/resources/webp-animated.webp");
     testRandomDecodeAfterClearFrameBufferCache("/LayoutTests/fast/images/resources/webp-animated-opaque.webp");
     testRandomDecodeAfterClearFrameBufferCache("/LayoutTests/fast/images/resources/webp-animated-large.webp");
     testRandomDecodeAfterClearFrameBufferCache("/LayoutTests/fast/images/resources/webp-animated-icc-xmp.webp");
 }

 TEST_F(AnimatedWebPTests, resumePartialDecodeAfterClearFrameBufferCache)
 {
     RefPtr<SharedBuffer> fullData = readFile("/LayoutTests/fast/images/resources/webp-animated-large.webp");
     ASSERT_TRUE(fullData.get());
     Vector<unsigned> baselineHashes;
     createDecodingBaseline(fullData.get(), &baselineHashes);
     size_t frameCount = baselineHashes.size();

     OwnPtr<WEBPImageDecoder> decoder = createDecoder();

     // Let frame 0 be partially decoded.
     size_t partialSize = 1;
     do {
         RefPtr<SharedBuffer> data = SharedBuffer::create(fullData->data(), partialSize);
         decoder->setData(data.get(), false);
         ++partialSize;
     } while (!decoder->frameCount() || decoder->frameBufferAtIndex(0)->status() == ImageFrame::FrameEmpty);

     // Skip to the last frame and clear.
     decoder->setData(fullData.get(), true);
     EXPECT_EQ(frameCount, decoder->frameCount());
     ImageFrame* lastFrame = decoder->frameBufferAtIndex(frameCount - 1);
     EXPECT_EQ(baselineHashes[frameCount - 1], hashSkBitmap(lastFrame->getSkBitmap()));
     decoder->clearCacheExceptFrame(notFound);

     // Resume decoding of the first frame.
     ImageFrame* firstFrame = decoder->frameBufferAtIndex(0);
     EXPECT_EQ(ImageFrame::FrameComplete, firstFrame->status());
     EXPECT_EQ(baselineHashes[0], hashSkBitmap(firstFrame->getSkBitmap()));
 }

 TEST_F(AnimatedWebPTests, decodeAfterReallocatingData)
 {
     testDecodeAfterReallocatingData("/LayoutTests/fast/images/resources/webp-animated.webp");
     testDecodeAfterReallocatingData("/LayoutTests/fast/images/resources/webp-animated-icc-xmp.webp");
 }

 #endif
	/*
	* 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 "config.h"

	#include "core/platform/image-decoders/webp/WEBPImageDecoder.h"

	#include "RuntimeEnabledFeatures.h"
	#include "core/platform/SharedBuffer.h"
	#include "public/platform/Platform.h"
	#include "public/platform/WebData.h"
	#include "public/platform/WebSize.h"
	#include "public/platform/WebUnitTestSupport.h"
	#include "wtf/OwnPtr.h"
	#include "wtf/PassOwnPtr.h"
	#include "wtf/StringHasher.h"
	#include "wtf/Vector.h"
	#include "wtf/dtoa/utils.h"
	#include <gtest/gtest.h>

	using namespace WebCore;
	using namespace WebKit;

	#if !OS(ANDROID)

	namespace {

	PassRefPtr<SharedBuffer> readFile(const char* fileName)
	{
	String filePath = Platform::current()->unitTestSupport()->webKitRootDir();
	filePath.append(fileName);

	return Platform::current()->unitTestSupport()->readFromFile(filePath);
	}

	PassOwnPtr<WEBPImageDecoder> createDecoder()
	{
	return adoptPtr(new WEBPImageDecoder(ImageSource::AlphaNotPremultiplied, ImageSource::GammaAndColorProfileApplied));
	}

	unsigned hashSkBitmap(const SkBitmap& bitmap)
	{
	return StringHasher::hashMemory(bitmap.getPixels(), bitmap.getSize());
	}

	void createDecodingBaseline(SharedBuffer* data, Vector<unsigned>* baselineHashes)
	{
	OwnPtr<WEBPImageDecoder> decoder = createDecoder();
	decoder->setData(data, true);
	size_t frameCount = decoder->frameCount();
	for (size_t i = 0; i < frameCount; ++i) {
	ImageFrame* frame = decoder->frameBufferAtIndex(i);
	baselineHashes->append(hashSkBitmap(frame->getSkBitmap()));
	}
	}

	void testRandomFrameDecode(const char* webpFile)
	{
	SCOPED_TRACE(webpFile);

	RefPtr<SharedBuffer> fullData = readFile(webpFile);
	ASSERT_TRUE(fullData.get());
	Vector<unsigned> baselineHashes;
	createDecodingBaseline(fullData.get(), &baselineHashes);
	size_t frameCount = baselineHashes.size();

	// Random decoding should get the same results as sequential decoding.
	OwnPtr<WEBPImageDecoder> decoder = createDecoder();
	decoder->setData(fullData.get(), true);
	const size_t skippingStep = 5;
	for (size_t i = 0; i < skippingStep; ++i) {
	for (size_t j = i; j < frameCount; j += skippingStep) {
	SCOPED_TRACE(testing::Message() << "Random i:" << i << " j:" << j);
	ImageFrame* frame = decoder->frameBufferAtIndex(j);
	EXPECT_EQ(baselineHashes[j], hashSkBitmap(frame->getSkBitmap()));
	}
	}

	// Decoding in reverse order.
	decoder = createDecoder();
	decoder->setData(fullData.get(), true);
	for (size_t i = frameCount; i; --i) {
	SCOPED_TRACE(testing::Message() << "Reverse i:" << i);
	ImageFrame* frame = decoder->frameBufferAtIndex(i - 1);
	EXPECT_EQ(baselineHashes[i - 1], hashSkBitmap(frame->getSkBitmap()));
	}
	}

	void testRandomDecodeAfterClearFrameBufferCache(const char* webpFile)
	{
	SCOPED_TRACE(webpFile);

	RefPtr<SharedBuffer> data = readFile(webpFile);
	ASSERT_TRUE(data.get());
	Vector<unsigned> baselineHashes;
	createDecodingBaseline(data.get(), &baselineHashes);
	size_t frameCount = baselineHashes.size();

	OwnPtr<WEBPImageDecoder> decoder = createDecoder();
	decoder->setData(data.get(), true);
	for (size_t clearExceptFrame = 0; clearExceptFrame < frameCount; ++clearExceptFrame) {
	decoder->clearCacheExceptFrame(clearExceptFrame);
	const size_t skippingStep = 5;
	for (size_t i = 0; i < skippingStep; ++i) {
	for (size_t j = 0; j < frameCount; j += skippingStep) {
	SCOPED_TRACE(testing::Message() << "Random i:" << i << " j:" << j);
	ImageFrame* frame = decoder->frameBufferAtIndex(j);
	EXPECT_EQ(baselineHashes[j], hashSkBitmap(frame->getSkBitmap()));
	}
	}
	}
	}

	void testDecodeAfterReallocatingData(const char* webpFile)
	{
	OwnPtr<WEBPImageDecoder> decoder = createDecoder();
	RefPtr<SharedBuffer> data = readFile(webpFile);
	ASSERT_TRUE(data.get());

	// Parse from 'data'.
	decoder->setData(data.get(), true);
	size_t frameCount = decoder->frameCount();

	// ... and then decode frames from 'reallocatedData'.
	RefPtr<SharedBuffer> reallocatedData = data.get()->copy();
	ASSERT_TRUE(reallocatedData.get());
	data.clear();
	decoder->setData(reallocatedData.get(), true);

	for (size_t i = 0; i < frameCount; ++i) {
	const ImageFrame* const frame = decoder->frameBufferAtIndex(i);
	EXPECT_EQ(ImageFrame::FrameComplete, frame->status());
	}
	}

	} // namespace

	class AnimatedWebPTests : public ::testing::Test {
	protected:
	virtual void SetUp()
	{
	// Enable animated WebP for all the tests.
	WebCore::RuntimeEnabledFeatures::setAnimatedWebPEnabled(true);
	}
	};

	TEST_F(AnimatedWebPTests, verifyAnimationParametersTransparentImage)
	{
	OwnPtr<WEBPImageDecoder> decoder = createDecoder();
	EXPECT_EQ(cAnimationLoopOnce, decoder->repetitionCount());

	RefPtr<SharedBuffer> data = readFile("/LayoutTests/fast/images/resources/webp-animated.webp");
	ASSERT_TRUE(data.get());
	decoder->setData(data.get(), true);

	const int canvasWidth = 11;
	const int canvasHeight = 29;
	const struct AnimParam {
	int xOffset, yOffset, width, height;
	ImageFrame::FrameDisposalMethod dispose;
	unsigned duration;
	bool hasAlpha;
	} frameParameters[] = {
	{ 0, 0, 11, 29, ImageFrame::DisposeKeep, 1000u, true },
	{ 2, 10, 7, 17, ImageFrame::DisposeKeep, 500u, true },
	{ 2, 2, 7, 16, ImageFrame::DisposeKeep, 1000u, true },
	};

	for (size_t i = 0; i < ARRAY_SIZE(frameParameters); ++i) {
	const ImageFrame* const frame = decoder->frameBufferAtIndex(i);
	EXPECT_EQ(ImageFrame::FrameComplete, frame->status());
	EXPECT_EQ(canvasWidth, frame->getSkBitmap().width());
	EXPECT_EQ(canvasHeight, frame->getSkBitmap().height());
	EXPECT_EQ(frameParameters[i].xOffset, frame->originalFrameRect().x());
	EXPECT_EQ(frameParameters[i].yOffset, frame->originalFrameRect().y());
	EXPECT_EQ(frameParameters[i].width, frame->originalFrameRect().width());
	EXPECT_EQ(frameParameters[i].height, frame->originalFrameRect().height());
	EXPECT_EQ(frameParameters[i].dispose, frame->disposalMethod());
	EXPECT_EQ(frameParameters[i].duration, frame->duration());
	EXPECT_EQ(frameParameters[i].hasAlpha, frame->hasAlpha());
	}

	EXPECT_EQ(ARRAY_SIZE(frameParameters), decoder->frameCount());
	EXPECT_EQ(cAnimationLoopInfinite, decoder->repetitionCount());
	}

	TEST_F(AnimatedWebPTests, verifyAnimationParametersOpaqueFramesTransparentBackground)
	{
	OwnPtr<WEBPImageDecoder> decoder = createDecoder();
	EXPECT_EQ(cAnimationLoopOnce, decoder->repetitionCount());

	RefPtr<SharedBuffer> data = readFile("/LayoutTests/fast/images/resources/webp-animated-opaque.webp");
	ASSERT_TRUE(data.get());
	decoder->setData(data.get(), true);

	const int canvasWidth = 94;
	const int canvasHeight = 87;
	const struct AnimParam {
	int xOffset, yOffset, width, height;
	ImageFrame::FrameDisposalMethod dispose;
	unsigned duration;
	bool hasAlpha;
	} frameParameters[] = {
	{ 4, 10, 33, 32, ImageFrame::DisposeOverwriteBgcolor, 1000u, true },
	{ 34, 30, 33, 32, ImageFrame::DisposeOverwriteBgcolor, 1000u, true },
	{ 62, 50, 32, 32, ImageFrame::DisposeOverwriteBgcolor, 1000u, true },
	{ 10, 54, 32, 33, ImageFrame::DisposeOverwriteBgcolor, 1000u, true },
	};

	for (size_t i = 0; i < ARRAY_SIZE(frameParameters); ++i) {
	const ImageFrame* const frame = decoder->frameBufferAtIndex(i);
	EXPECT_EQ(ImageFrame::FrameComplete, frame->status());
	EXPECT_EQ(canvasWidth, frame->getSkBitmap().width());
	EXPECT_EQ(canvasHeight, frame->getSkBitmap().height());
	EXPECT_EQ(frameParameters[i].xOffset, frame->originalFrameRect().x());
	EXPECT_EQ(frameParameters[i].yOffset, frame->originalFrameRect().y());
	EXPECT_EQ(frameParameters[i].width, frame->originalFrameRect().width());
	EXPECT_EQ(frameParameters[i].height, frame->originalFrameRect().height());
	EXPECT_EQ(frameParameters[i].dispose, frame->disposalMethod());
	EXPECT_EQ(frameParameters[i].duration, frame->duration());
	EXPECT_EQ(frameParameters[i].hasAlpha, frame->hasAlpha());
	}

	EXPECT_EQ(ARRAY_SIZE(frameParameters), decoder->frameCount());
	EXPECT_EQ(cAnimationLoopInfinite, decoder->repetitionCount());
	}

	TEST_F(AnimatedWebPTests, parseAndDecodeByteByByte)
	{
	const struct TestImage {
	const char* filename;
	unsigned frameCount;
	int repetitionCount;
	} testImages[] = {
	{ "/LayoutTests/fast/images/resources/webp-animated.webp", 3u, cAnimationLoopInfinite },
	{ "/LayoutTests/fast/images/resources/webp-animated-icc-xmp.webp", 13u, 32000 },
	};

	for (size_t i = 0; i < ARRAY_SIZE(testImages); ++i) {
	OwnPtr<WEBPImageDecoder> decoder = createDecoder();
	RefPtr<SharedBuffer> data = readFile(testImages[i].filename);
	ASSERT_TRUE(data.get());

	size_t frameCount = 0;
	size_t framesDecoded = 0;

	// Pass data to decoder byte by byte.
	for (size_t length = 1; length <= data->size(); ++length) {
	RefPtr<SharedBuffer> tempData = SharedBuffer::create(data->data(), length);
	decoder->setData(tempData.get(), length == data->size());

	EXPECT_LE(frameCount, decoder->frameCount());
	frameCount = decoder->frameCount();

	ImageFrame* frame = decoder->frameBufferAtIndex(frameCount - 1);
	if (frame && frame->status() == ImageFrame::FrameComplete && framesDecoded < frameCount)
	++framesDecoded;
	}

	EXPECT_EQ(testImages[i].frameCount, decoder->frameCount());
	EXPECT_EQ(testImages[i].frameCount, framesDecoded);
	EXPECT_EQ(testImages[i].repetitionCount, decoder->repetitionCount());
	}
	}

	TEST_F(AnimatedWebPTests, invalidImage)
	{
	OwnPtr<WEBPImageDecoder> decoder = createDecoder();

	RefPtr<SharedBuffer> data = readFile("/LayoutTests/fast/images/resources/invalid-animated-webp.webp");
	ASSERT_TRUE(data.get());
	decoder->setData(data.get(), true);

	EXPECT_EQ(0u, decoder->frameCount());
	ImageFrame* frame = decoder->frameBufferAtIndex(0);
	EXPECT_FALSE(frame);
	EXPECT_EQ(cAnimationLoopOnce, decoder->repetitionCount());
	}

	TEST_F(AnimatedWebPTests, progressiveDecode)
	{
	RefPtr<SharedBuffer> fullData = readFile("/LayoutTests/fast/images/resources/webp-animated.webp");
	ASSERT_TRUE(fullData.get());
	const size_t fullLength = fullData->size();

	OwnPtr<WEBPImageDecoder> decoder;
	ImageFrame* frame;

	Vector<unsigned> truncatedHashes;
	Vector<unsigned> progressiveHashes;

	// Compute hashes when the file is truncated.
	const size_t increment = 1;
	for (size_t i = 1; i <= fullLength; i += increment) {
	decoder = createDecoder();
	RefPtr<SharedBuffer> data = SharedBuffer::create(fullData->data(), i);
	decoder->setData(data.get(), i == fullLength);
	frame = decoder->frameBufferAtIndex(0);
	if (!frame) {
	truncatedHashes.append(0);
	continue;
	}
	truncatedHashes.append(hashSkBitmap(frame->getSkBitmap()));
	}

	// Compute hashes when the file is progressively decoded.
	decoder = createDecoder();
	for (size_t i = 1; i <= fullLength; i += increment) {
	RefPtr<SharedBuffer> data = SharedBuffer::create(fullData->data(), i);
	decoder->setData(data.get(), i == fullLength);
	frame = decoder->frameBufferAtIndex(0);
	if (!frame) {
	progressiveHashes.append(0);
	continue;
	}
	progressiveHashes.append(hashSkBitmap(frame->getSkBitmap()));
	}

	bool match = true;
	for (size_t i = 0; i < truncatedHashes.size(); ++i) {
	if (truncatedHashes[i] != progressiveHashes[i]) {
	match = false;
	break;
	}
	}
	EXPECT_TRUE(match);
	}

	TEST_F(AnimatedWebPTests, frameIsCompleteAndDuration)
	{
	OwnPtr<WEBPImageDecoder> decoder = createDecoder();

	RefPtr<SharedBuffer> data = readFile("/LayoutTests/fast/images/resources/webp-animated.webp");
	ASSERT_TRUE(data.get());

	ASSERT_GE(data->size(), 10u);
	RefPtr<SharedBuffer> tempData = SharedBuffer::create(data->data(), data->size() - 10);
	decoder->setData(tempData.get(), false);

	EXPECT_EQ(2u, decoder->frameCount());
	EXPECT_FALSE(decoder->failed());
	EXPECT_TRUE(decoder->frameIsCompleteAtIndex(0));
	EXPECT_EQ(1000, decoder->frameDurationAtIndex(0));
	EXPECT_TRUE(decoder->frameIsCompleteAtIndex(1));
	EXPECT_EQ(500, decoder->frameDurationAtIndex(1));

	decoder->setData(data.get(), true);
	EXPECT_EQ(3u, decoder->frameCount());
	EXPECT_TRUE(decoder->frameIsCompleteAtIndex(0));
	EXPECT_EQ(1000, decoder->frameDurationAtIndex(0));
	EXPECT_TRUE(decoder->frameIsCompleteAtIndex(1));
	EXPECT_EQ(500, decoder->frameDurationAtIndex(1));
	EXPECT_TRUE(decoder->frameIsCompleteAtIndex(2));
	EXPECT_EQ(1000.0, decoder->frameDurationAtIndex(2));
	}

	TEST_F(AnimatedWebPTests, updateRequiredPreviousFrameAfterFirstDecode)
	{
	OwnPtr<WEBPImageDecoder> decoder = createDecoder();

	RefPtr<SharedBuffer> fullData = readFile("/LayoutTests/fast/images/resources/webp-animated.webp");
	ASSERT_TRUE(fullData.get());

	// Give it data that is enough to parse but not decode in order to check the status
	// of requiredPreviousFrameIndex before decoding.
	size_t partialSize = 1;
	do {
	RefPtr<SharedBuffer> data = SharedBuffer::create(fullData->data(), partialSize);
	decoder->setData(data.get(), false);
	++partialSize;
	} while (!decoder->frameCount() \|\| decoder->frameBufferAtIndex(0)->status() == ImageFrame::FrameEmpty);

	EXPECT_EQ(notFound, decoder->frameBufferAtIndex(0)->requiredPreviousFrameIndex());
	unsigned frameCount = decoder->frameCount();
	for (size_t i = 1; i < frameCount; ++i)
	EXPECT_EQ(i - 1, decoder->frameBufferAtIndex(i)->requiredPreviousFrameIndex());

	decoder->setData(fullData.get(), true);
	for (size_t i = 0; i < frameCount; ++i)
	EXPECT_EQ(notFound, decoder->frameBufferAtIndex(i)->requiredPreviousFrameIndex());
	}

	TEST_F(AnimatedWebPTests, randomFrameDecode)
	{
	testRandomFrameDecode("/LayoutTests/fast/images/resources/webp-animated.webp");
	testRandomFrameDecode("/LayoutTests/fast/images/resources/webp-animated-opaque.webp");
	testRandomFrameDecode("/LayoutTests/fast/images/resources/webp-animated-large.webp");
	testRandomFrameDecode("/LayoutTests/fast/images/resources/webp-animated-icc-xmp.webp");
	}

	TEST_F(AnimatedWebPTests, randomDecodeAfterClearFrameBufferCache)
	{
	testRandomDecodeAfterClearFrameBufferCache("/LayoutTests/fast/images/resources/webp-animated.webp");
	testRandomDecodeAfterClearFrameBufferCache("/LayoutTests/fast/images/resources/webp-animated-opaque.webp");
	testRandomDecodeAfterClearFrameBufferCache("/LayoutTests/fast/images/resources/webp-animated-large.webp");
	testRandomDecodeAfterClearFrameBufferCache("/LayoutTests/fast/images/resources/webp-animated-icc-xmp.webp");
	}

	TEST_F(AnimatedWebPTests, resumePartialDecodeAfterClearFrameBufferCache)
	{
	RefPtr<SharedBuffer> fullData = readFile("/LayoutTests/fast/images/resources/webp-animated-large.webp");
	ASSERT_TRUE(fullData.get());
	Vector<unsigned> baselineHashes;
	createDecodingBaseline(fullData.get(), &baselineHashes);
	size_t frameCount = baselineHashes.size();

	OwnPtr<WEBPImageDecoder> decoder = createDecoder();

	// Let frame 0 be partially decoded.
	size_t partialSize = 1;
	do {
	RefPtr<SharedBuffer> data = SharedBuffer::create(fullData->data(), partialSize);
	decoder->setData(data.get(), false);
	++partialSize;
	} while (!decoder->frameCount() \|\| decoder->frameBufferAtIndex(0)->status() == ImageFrame::FrameEmpty);

	// Skip to the last frame and clear.
	decoder->setData(fullData.get(), true);
	EXPECT_EQ(frameCount, decoder->frameCount());
	ImageFrame* lastFrame = decoder->frameBufferAtIndex(frameCount - 1);
	EXPECT_EQ(baselineHashes[frameCount - 1], hashSkBitmap(lastFrame->getSkBitmap()));
	decoder->clearCacheExceptFrame(notFound);

	// Resume decoding of the first frame.
	ImageFrame* firstFrame = decoder->frameBufferAtIndex(0);
	EXPECT_EQ(ImageFrame::FrameComplete, firstFrame->status());
	EXPECT_EQ(baselineHashes[0], hashSkBitmap(firstFrame->getSkBitmap()));
	}

	TEST_F(AnimatedWebPTests, decodeAfterReallocatingData)
	{
	testDecodeAfterReallocatingData("/LayoutTests/fast/images/resources/webp-animated.webp");
	testDecodeAfterReallocatingData("/LayoutTests/fast/images/resources/webp-animated-icc-xmp.webp");
	}

	#endif