| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "webkit/tools/test_shell/image_decoder_unittest.h" | 5 #include "webkit/tools/test_shell/image_decoder_unittest.h" |
| 6 | 6 |
| 7 #include "base/file_path.h" | 7 #include "base/file_path.h" |
| 8 #include "base/file_util.h" | 8 #include "base/file_util.h" |
| 9 #include "base/md5.h" | 9 #include "base/md5.h" |
| 10 #include "base/memory/scoped_ptr.h" | 10 #include "base/memory/scoped_ptr.h" |
| (...skipping 16 matching lines...) Expand all Loading... |
| 27 ImageDecoderTestFileSelection file_selection, | 27 ImageDecoderTestFileSelection file_selection, |
| 28 const int64 threshold) { | 28 const int64 threshold) { |
| 29 if (file_selection == TEST_ALL) | 29 if (file_selection == TEST_ALL) |
| 30 return false; | 30 return false; |
| 31 | 31 |
| 32 int64 image_size = 0; | 32 int64 image_size = 0; |
| 33 file_util::GetFileSize(path, &image_size); | 33 file_util::GetFileSize(path, &image_size); |
| 34 return (file_selection == TEST_SMALLER) == (image_size > threshold); | 34 return (file_selection == TEST_SMALLER) == (image_size > threshold); |
| 35 } | 35 } |
| 36 | 36 |
| 37 } // anonymous namespace | 37 } // namespace |
| 38 | 38 |
| 39 void ReadFileToVector(const FilePath& path, std::vector<char>* contents) { | 39 void ReadFileToVector(const FilePath& path, std::vector<char>* contents) { |
| 40 std::string raw_image_data; | 40 std::string raw_image_data; |
| 41 file_util::ReadFileToString(path, &raw_image_data); | 41 file_util::ReadFileToString(path, &raw_image_data); |
| 42 contents->resize(raw_image_data.size()); | 42 contents->resize(raw_image_data.size()); |
| 43 memcpy(&contents->at(0), raw_image_data.data(), raw_image_data.size()); | 43 memcpy(&contents->at(0), raw_image_data.data(), raw_image_data.size()); |
| 44 } | 44 } |
| 45 | 45 |
| 46 FilePath GetMD5SumPath(const FilePath& path) { | 46 FilePath GetMD5SumPath(const FilePath& path) { |
| 47 static const FilePath::StringType kDecodedDataExtension( | 47 static const FilePath::StringType kDecodedDataExtension( |
| 48 FILE_PATH_LITERAL(".md5sum")); | 48 FILE_PATH_LITERAL(".md5sum")); |
| 49 return FilePath(path.value() + kDecodedDataExtension); | 49 return FilePath(path.value() + kDecodedDataExtension); |
| 50 } | 50 } |
| 51 | 51 |
| 52 #if defined(CALCULATE_MD5_SUMS) | 52 #if defined(CALCULATE_MD5_SUMS) |
| 53 void SaveMD5Sum(const std::wstring& path, const WebKit::WebImage& web_image) { | 53 void SaveMD5Sum(const std::wstring& path, const WebKit::WebImage& web_image) { |
| 54 // Calculate MD5 sum. | 54 // Calculate MD5 sum. |
| 55 MD5Digest digest; | 55 base::MD5Digest digest; |
| 56 web_image.getSkBitmap().lockPixels(); | 56 web_image.getSkBitmap().lockPixels(); |
| 57 MD5Sum(web_image.getSkBitmap().getPixels(), | 57 base::MD5Sum(web_image.getSkBitmap().getPixels(), |
| 58 web_image.getSkBitmap().width() * web_image.getSkBitmap().height() * | 58 web_image.getSkBitmap().width() * web_image.getSkBitmap().height() * |
| 59 sizeof(uint32_t), | 59 sizeof(uint32_t), |
| 60 &digest); | 60 &digest); |
| 61 | 61 |
| 62 // Write sum to disk. | 62 // Write sum to disk. |
| 63 int bytes_written = file_util::WriteFile(path, | 63 int bytes_written = file_util::WriteFile(path, |
| 64 reinterpret_cast<const char*>(&digest), sizeof digest); | 64 reinterpret_cast<const char*>(&digest), sizeof digest); |
| 65 ASSERT_EQ(sizeof digest, bytes_written); | 65 ASSERT_EQ(sizeof digest, bytes_written); |
| 66 web_image.getSkBitmap().unlockPixels(); | 66 web_image.getSkBitmap().unlockPixels(); |
| 67 } | 67 } |
| 68 #endif | 68 #endif |
| 69 | 69 |
| 70 #if !defined(CALCULATE_MD5_SUMS) | 70 #if !defined(CALCULATE_MD5_SUMS) |
| 71 void VerifyImage(const WebKit::WebImageDecoder& decoder, | 71 void VerifyImage(const WebKit::WebImageDecoder& decoder, |
| 72 const FilePath& path, | 72 const FilePath& path, |
| 73 const FilePath& md5_sum_path, | 73 const FilePath& md5_sum_path, |
| 74 size_t frame_index) { | 74 size_t frame_index) { |
| 75 // Make sure decoding can complete successfully. | 75 // Make sure decoding can complete successfully. |
| 76 EXPECT_TRUE(decoder.isSizeAvailable()) << path.value(); | 76 EXPECT_TRUE(decoder.isSizeAvailable()) << path.value(); |
| 77 EXPECT_GE(decoder.frameCount(), frame_index) << path.value(); | 77 EXPECT_GE(decoder.frameCount(), frame_index) << path.value(); |
| 78 EXPECT_TRUE(decoder.isFrameCompleteAtIndex(frame_index)) << path.value(); | 78 EXPECT_TRUE(decoder.isFrameCompleteAtIndex(frame_index)) << path.value(); |
| 79 EXPECT_FALSE(decoder.isFailed()); | 79 EXPECT_FALSE(decoder.isFailed()); |
| 80 | 80 |
| 81 // Calculate MD5 sum. | 81 // Calculate MD5 sum. |
| 82 MD5Digest actual_digest; | 82 base::MD5Digest actual_digest; |
| 83 WebKit::WebImage web_image = decoder.getFrameAtIndex(frame_index); | 83 WebKit::WebImage web_image = decoder.getFrameAtIndex(frame_index); |
| 84 web_image.getSkBitmap().lockPixels(); | 84 web_image.getSkBitmap().lockPixels(); |
| 85 MD5Sum(web_image.getSkBitmap().getPixels(), | 85 base::MD5Sum(web_image.getSkBitmap().getPixels(), |
| 86 web_image.getSkBitmap().width() * web_image.getSkBitmap().height() * | 86 web_image.getSkBitmap().width() * web_image.getSkBitmap().height() * |
| 87 sizeof(uint32_t), | 87 sizeof(uint32_t), |
| 88 &actual_digest); | 88 &actual_digest); |
| 89 | 89 |
| 90 // Read the MD5 sum off disk. | 90 // Read the MD5 sum off disk. |
| 91 std::string file_bytes; | 91 std::string file_bytes; |
| 92 file_util::ReadFileToString(md5_sum_path, &file_bytes); | 92 file_util::ReadFileToString(md5_sum_path, &file_bytes); |
| 93 MD5Digest expected_digest; | 93 base::MD5Digest expected_digest; |
| 94 ASSERT_EQ(sizeof expected_digest, file_bytes.size()) << path.value(); | 94 ASSERT_EQ(sizeof expected_digest, file_bytes.size()) << path.value(); |
| 95 memcpy(&expected_digest, file_bytes.data(), sizeof expected_digest); | 95 memcpy(&expected_digest, file_bytes.data(), sizeof expected_digest); |
| 96 | 96 |
| 97 // Verify that the sums are the same. | 97 // Verify that the sums are the same. |
| 98 EXPECT_EQ(0, memcmp(&expected_digest, &actual_digest, sizeof(MD5Digest))) | 98 EXPECT_EQ(0, |
| 99 memcmp(&expected_digest, &actual_digest, sizeof(base::MD5Digest))) |
| 99 << path.value(); | 100 << path.value(); |
| 100 web_image.getSkBitmap().unlockPixels(); | 101 web_image.getSkBitmap().unlockPixels(); |
| 101 } | 102 } |
| 102 #endif | 103 #endif |
| 103 | 104 |
| 104 void ImageDecoderTest::SetUp() { | 105 void ImageDecoderTest::SetUp() { |
| 105 FilePath data_dir; | 106 FilePath data_dir; |
| 106 ASSERT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT, &data_dir)); | 107 ASSERT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT, &data_dir)); |
| 107 data_dir_ = data_dir.AppendASCII("webkit"). | 108 data_dir_ = data_dir.AppendASCII("webkit"). |
| 108 AppendASCII("data"). | 109 AppendASCII("data"). |
| (...skipping 97 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 206 // Since WebImage does not expose get data by frame, get the size | 207 // Since WebImage does not expose get data by frame, get the size |
| 207 // through decoder and pass it to fromData so that the closest | 208 // through decoder and pass it to fromData so that the closest |
| 208 // image dats to the size is returned. | 209 // image dats to the size is returned. |
| 209 WebKit::WebSize size(decoder->getImage(desired_frame_index).size()); | 210 WebKit::WebSize size(decoder->getImage(desired_frame_index).size()); |
| 210 const WebKit::WebImage& image = WebKit::WebImage::fromData(data, size); | 211 const WebKit::WebImage& image = WebKit::WebImage::fromData(data, size); |
| 211 SaveMD5Sum(md5_sum_path.value(), image); | 212 SaveMD5Sum(md5_sum_path.value(), image); |
| 212 #else | 213 #else |
| 213 VerifyImage(*decoder, image_path, md5_sum_path, desired_frame_index); | 214 VerifyImage(*decoder, image_path, md5_sum_path, desired_frame_index); |
| 214 #endif | 215 #endif |
| 215 } | 216 } |
| OLD | NEW |