Index: media/base/yuv_convert_unittest.cc |
diff --git a/media/base/yuv_convert_unittest.cc b/media/base/yuv_convert_unittest.cc |
deleted file mode 100644 |
index ec872c89b684bf0cb2be0ff62e8a031eacf1b4f0..0000000000000000000000000000000000000000 |
--- a/media/base/yuv_convert_unittest.cc |
+++ /dev/null |
@@ -1,876 +0,0 @@ |
-// Copyright (c) 2012 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
- |
-#include "media/base/yuv_convert.h" |
- |
-#include <stddef.h> |
-#include <stdint.h> |
- |
-#include <memory> |
- |
-#include "base/base_paths.h" |
-#include "base/cpu.h" |
-#include "base/files/file_util.h" |
-#include "base/logging.h" |
-#include "base/path_service.h" |
-#include "build/build_config.h" |
-#include "media/base/djb2.h" |
-#include "media/base/simd/convert_rgb_to_yuv.h" |
-#include "media/base/simd/convert_yuv_to_rgb.h" |
-#include "media/base/simd/filter_yuv.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
-#include "ui/gfx/geometry/rect.h" |
- |
-// Size of raw image. |
-static const int kSourceWidth = 640; |
-static const int kSourceHeight = 360; |
-static const int kSourceYSize = kSourceWidth * kSourceHeight; |
-static const int kSourceUOffset = kSourceYSize; |
-static const int kSourceVOffset = kSourceYSize * 5 / 4; |
-static const int kScaledWidth = 1024; |
-static const int kScaledHeight = 768; |
-static const int kDownScaledWidth = 512; |
-static const int kDownScaledHeight = 320; |
-static const int kBpp = 4; |
- |
-// Surface sizes for various test files. |
-static const int kYUV12Size = kSourceYSize * 12 / 8; |
-static const int kYUV16Size = kSourceYSize * 16 / 8; |
-static const int kRGBSize = kSourceYSize * kBpp; |
-static const int kRGBSizeScaled = kScaledWidth * kScaledHeight * kBpp; |
-static const int kRGB24Size = kSourceYSize * 3; |
-static const int kRGBSizeConverted = kSourceYSize * kBpp; |
- |
-#if !defined(ARCH_CPU_ARM_FAMILY) && !defined(ARCH_CPU_MIPS_FAMILY) && \ |
- !defined(OS_ANDROID) |
-static const int kSourceAOffset = kSourceYSize * 12 / 8; |
-static const int kYUVA12Size = kSourceYSize * 20 / 8; |
-#endif |
- |
-// Helper for reading test data into a std::unique_ptr<uint8_t[]>. |
-static void ReadData(const base::FilePath::CharType* filename, |
- int expected_size, |
- std::unique_ptr<uint8_t[]>* data) { |
- data->reset(new uint8_t[expected_size]); |
- |
- base::FilePath path; |
- CHECK(PathService::Get(base::DIR_SOURCE_ROOT, &path)); |
- path = path.Append(FILE_PATH_LITERAL("media")) |
- .Append(FILE_PATH_LITERAL("test")) |
- .Append(FILE_PATH_LITERAL("data")) |
- .Append(filename); |
- |
- // Verify file size is correct. |
- int64_t actual_size = 0; |
- base::GetFileSize(path, &actual_size); |
- CHECK_EQ(actual_size, expected_size); |
- |
- // Verify bytes read are correct. |
- int bytes_read = base::ReadFile( |
- path, reinterpret_cast<char*>(data->get()), expected_size); |
- CHECK_EQ(bytes_read, expected_size); |
-} |
- |
-static void ReadYV12Data(std::unique_ptr<uint8_t[]>* data) { |
- ReadData(FILE_PATH_LITERAL("bali_640x360_P420.yuv"), kYUV12Size, data); |
-} |
- |
-static void ReadYV16Data(std::unique_ptr<uint8_t[]>* data) { |
- ReadData(FILE_PATH_LITERAL("bali_640x360_P422.yuv"), kYUV16Size, data); |
-} |
- |
-#if !defined(ARCH_CPU_ARM_FAMILY) && !defined(ARCH_CPU_MIPS_FAMILY) && \ |
- !defined(OS_ANDROID) |
-static void ReadYV12AData(std::unique_ptr<uint8_t[]>* data) { |
- ReadData(FILE_PATH_LITERAL("bali_640x360_P420_alpha.yuv"), kYUVA12Size, data); |
-} |
-#endif |
- |
-static void ReadRGB24Data(std::unique_ptr<uint8_t[]>* data) { |
- ReadData(FILE_PATH_LITERAL("bali_640x360_RGB24.rgb"), kRGB24Size, data); |
-} |
- |
-#if defined(OS_ANDROID) |
-// Helper for swapping red and blue channels of RGBA or BGRA. |
-static void SwapRedAndBlueChannels(unsigned char* pixels, size_t buffer_size) { |
- for (size_t i = 0; i < buffer_size; i += 4) { |
- std::swap(pixels[i], pixels[i + 2]); |
- } |
-} |
-#endif |
- |
-namespace media { |
- |
-TEST(YUVConvertTest, YV12) { |
- // Allocate all surfaces. |
- std::unique_ptr<uint8_t[]> yuv_bytes; |
- std::unique_ptr<uint8_t[]> rgb_bytes(new uint8_t[kRGBSize]); |
- std::unique_ptr<uint8_t[]> rgb_converted_bytes( |
- new uint8_t[kRGBSizeConverted]); |
- |
- // Read YUV reference data from file. |
- ReadYV12Data(&yuv_bytes); |
- |
- // Convert a frame of YUV to 32 bit ARGB. |
- media::ConvertYUVToRGB32(yuv_bytes.get(), |
- yuv_bytes.get() + kSourceUOffset, |
- yuv_bytes.get() + kSourceVOffset, |
- rgb_converted_bytes.get(), // RGB output |
- kSourceWidth, kSourceHeight, // Dimensions |
- kSourceWidth, // YStride |
- kSourceWidth / 2, // UVStride |
- kSourceWidth * kBpp, // RGBStride |
- media::YV12); |
- |
-#if defined(OS_ANDROID) |
- SwapRedAndBlueChannels(rgb_converted_bytes.get(), kRGBSizeConverted); |
-#endif |
- |
- uint32_t rgb_hash = |
- DJB2Hash(rgb_converted_bytes.get(), kRGBSizeConverted, kDJB2HashSeed); |
- EXPECT_EQ(2413171226u, rgb_hash); |
-} |
- |
-TEST(YUVConvertTest, YV16) { |
- // Allocate all surfaces. |
- std::unique_ptr<uint8_t[]> yuv_bytes; |
- std::unique_ptr<uint8_t[]> rgb_bytes(new uint8_t[kRGBSize]); |
- std::unique_ptr<uint8_t[]> rgb_converted_bytes( |
- new uint8_t[kRGBSizeConverted]); |
- |
- // Read YUV reference data from file. |
- ReadYV16Data(&yuv_bytes); |
- |
- // Convert a frame of YUV to 32 bit ARGB. |
- media::ConvertYUVToRGB32(yuv_bytes.get(), // Y |
- yuv_bytes.get() + kSourceUOffset, // U |
- yuv_bytes.get() + kSourceYSize * 3 / 2, // V |
- rgb_converted_bytes.get(), // RGB output |
- kSourceWidth, kSourceHeight, // Dimensions |
- kSourceWidth, // YStride |
- kSourceWidth / 2, // UVStride |
- kSourceWidth * kBpp, // RGBStride |
- media::YV16); |
- |
-#if defined(OS_ANDROID) |
- SwapRedAndBlueChannels(rgb_converted_bytes.get(), kRGBSizeConverted); |
-#endif |
- |
- uint32_t rgb_hash = |
- DJB2Hash(rgb_converted_bytes.get(), kRGBSizeConverted, kDJB2HashSeed); |
- EXPECT_EQ(4222342047u, rgb_hash); |
-} |
- |
-struct YUVScaleTestData { |
- YUVScaleTestData(media::YUVType y, media::ScaleFilter s, uint32_t r) |
- : yuv_type(y), scale_filter(s), rgb_hash(r) {} |
- |
- media::YUVType yuv_type; |
- media::ScaleFilter scale_filter; |
- uint32_t rgb_hash; |
-}; |
- |
-class YUVScaleTest : public ::testing::TestWithParam<YUVScaleTestData> { |
- public: |
- YUVScaleTest() { |
- switch (GetParam().yuv_type) { |
- case media::YV12: |
- case media::YV12J: |
- case media::YV12HD: |
- ReadYV12Data(&yuv_bytes_); |
- break; |
- case media::YV16: |
- ReadYV16Data(&yuv_bytes_); |
- break; |
- } |
- |
- rgb_bytes_.reset(new uint8_t[kRGBSizeScaled]); |
- } |
- |
- // Helpers for getting the proper Y, U and V plane offsets. |
- uint8_t* y_plane() { return yuv_bytes_.get(); } |
- uint8_t* u_plane() { return yuv_bytes_.get() + kSourceYSize; } |
- uint8_t* v_plane() { |
- switch (GetParam().yuv_type) { |
- case media::YV12: |
- case media::YV12J: |
- case media::YV12HD: |
- return yuv_bytes_.get() + kSourceVOffset; |
- case media::YV16: |
- return yuv_bytes_.get() + kSourceYSize * 3 / 2; |
- } |
- return NULL; |
- } |
- |
- std::unique_ptr<uint8_t[]> yuv_bytes_; |
- std::unique_ptr<uint8_t[]> rgb_bytes_; |
-}; |
- |
-TEST_P(YUVScaleTest, NoScale) { |
- media::ScaleYUVToRGB32(y_plane(), // Y |
- u_plane(), // U |
- v_plane(), // V |
- rgb_bytes_.get(), // RGB output |
- kSourceWidth, kSourceHeight, // Dimensions |
- kSourceWidth, kSourceHeight, // Dimensions |
- kSourceWidth, // YStride |
- kSourceWidth / 2, // UvStride |
- kSourceWidth * kBpp, // RgbStride |
- GetParam().yuv_type, |
- media::ROTATE_0, |
- GetParam().scale_filter); |
- |
- uint32_t yuv_hash = DJB2Hash(rgb_bytes_.get(), kRGBSize, kDJB2HashSeed); |
- |
- media::ConvertYUVToRGB32(y_plane(), // Y |
- u_plane(), // U |
- v_plane(), // V |
- rgb_bytes_.get(), // RGB output |
- kSourceWidth, kSourceHeight, // Dimensions |
- kSourceWidth, // YStride |
- kSourceWidth / 2, // UVStride |
- kSourceWidth * kBpp, // RGBStride |
- GetParam().yuv_type); |
- |
- uint32_t rgb_hash = DJB2Hash(rgb_bytes_.get(), kRGBSize, kDJB2HashSeed); |
- |
- EXPECT_EQ(yuv_hash, rgb_hash); |
-} |
- |
-TEST_P(YUVScaleTest, Normal) { |
- media::ScaleYUVToRGB32(y_plane(), // Y |
- u_plane(), // U |
- v_plane(), // V |
- rgb_bytes_.get(), // RGB output |
- kSourceWidth, kSourceHeight, // Dimensions |
- kScaledWidth, kScaledHeight, // Dimensions |
- kSourceWidth, // YStride |
- kSourceWidth / 2, // UvStride |
- kScaledWidth * kBpp, // RgbStride |
- GetParam().yuv_type, |
- media::ROTATE_0, |
- GetParam().scale_filter); |
- |
-#if defined(OS_ANDROID) |
- SwapRedAndBlueChannels(rgb_bytes_.get(), kRGBSizeScaled); |
-#endif |
- |
- uint32_t rgb_hash = DJB2Hash(rgb_bytes_.get(), kRGBSizeScaled, kDJB2HashSeed); |
- EXPECT_EQ(GetParam().rgb_hash, rgb_hash); |
-} |
- |
-TEST_P(YUVScaleTest, ZeroSourceSize) { |
- media::ScaleYUVToRGB32(y_plane(), // Y |
- u_plane(), // U |
- v_plane(), // V |
- rgb_bytes_.get(), // RGB output |
- 0, 0, // Dimensions |
- kScaledWidth, kScaledHeight, // Dimensions |
- kSourceWidth, // YStride |
- kSourceWidth / 2, // UvStride |
- kScaledWidth * kBpp, // RgbStride |
- GetParam().yuv_type, |
- media::ROTATE_0, |
- GetParam().scale_filter); |
- |
- // Testing for out-of-bound read/writes with AddressSanitizer. |
-} |
- |
-TEST_P(YUVScaleTest, ZeroDestinationSize) { |
- media::ScaleYUVToRGB32(y_plane(), // Y |
- u_plane(), // U |
- v_plane(), // V |
- rgb_bytes_.get(), // RGB output |
- kSourceWidth, kSourceHeight, // Dimensions |
- 0, 0, // Dimensions |
- kSourceWidth, // YStride |
- kSourceWidth / 2, // UvStride |
- kScaledWidth * kBpp, // RgbStride |
- GetParam().yuv_type, |
- media::ROTATE_0, |
- GetParam().scale_filter); |
- |
- // Testing for out-of-bound read/writes with AddressSanitizer. |
-} |
- |
-TEST_P(YUVScaleTest, OddWidthAndHeightNotCrash) { |
- media::ScaleYUVToRGB32(y_plane(), // Y |
- u_plane(), // U |
- v_plane(), // V |
- rgb_bytes_.get(), // RGB output |
- kSourceWidth, kSourceHeight, // Dimensions |
- 3, 3, // Dimensions |
- kSourceWidth, // YStride |
- kSourceWidth / 2, // UvStride |
- kScaledWidth * kBpp, // RgbStride |
- GetParam().yuv_type, |
- media::ROTATE_0, |
- GetParam().scale_filter); |
-} |
- |
-INSTANTIATE_TEST_CASE_P( |
- YUVScaleFormats, YUVScaleTest, |
- ::testing::Values( |
- YUVScaleTestData(media::YV12, media::FILTER_NONE, 4136904952u), |
- YUVScaleTestData(media::YV16, media::FILTER_NONE, 1501777547u), |
- YUVScaleTestData(media::YV12, media::FILTER_BILINEAR, 3164274689u), |
- YUVScaleTestData(media::YV16, media::FILTER_BILINEAR, 3095878046u))); |
- |
-// This tests a known worst case YUV value, and for overflow. |
-TEST(YUVConvertTest, Clamp) { |
- // Allocate all surfaces. |
- std::unique_ptr<uint8_t[]> yuv_bytes(new uint8_t[1]); |
- std::unique_ptr<uint8_t[]> rgb_bytes(new uint8_t[1]); |
- std::unique_ptr<uint8_t[]> rgb_converted_bytes(new uint8_t[1]); |
- |
- // Values that failed previously in bug report. |
- unsigned char y = 255u; |
- unsigned char u = 255u; |
- unsigned char v = 19u; |
- |
- // Prefill extra large destination buffer to test for overflow. |
- unsigned char rgb[8] = { 0, 1, 2, 3, 4, 5, 6, 7 }; |
- unsigned char expected[8] = { 255, 255, 104, 255, 4, 5, 6, 7 }; |
- // Convert a frame of YUV to 32 bit ARGB. |
- media::ConvertYUVToRGB32(&y, // Y |
- &u, // U |
- &v, // V |
- &rgb[0], // RGB output |
- 1, 1, // Dimensions |
- 0, // YStride |
- 0, // UVStride |
- 0, // RGBStride |
- media::YV12); |
- |
-#if defined(OS_ANDROID) |
- SwapRedAndBlueChannels(rgb, kBpp); |
-#endif |
- |
- int expected_test = memcmp(rgb, expected, sizeof(expected)); |
- EXPECT_EQ(0, expected_test); |
-} |
- |
-TEST(YUVConvertTest, RGB24ToYUV) { |
- // Allocate all surfaces. |
- std::unique_ptr<uint8_t[]> rgb_bytes; |
- std::unique_ptr<uint8_t[]> yuv_converted_bytes(new uint8_t[kYUV12Size]); |
- |
- // Read RGB24 reference data from file. |
- ReadRGB24Data(&rgb_bytes); |
- |
- // Convert to I420. |
- media::ConvertRGB24ToYUV(rgb_bytes.get(), |
- yuv_converted_bytes.get(), |
- yuv_converted_bytes.get() + kSourceUOffset, |
- yuv_converted_bytes.get() + kSourceVOffset, |
- kSourceWidth, kSourceHeight, // Dimensions |
- kSourceWidth * 3, // RGBStride |
- kSourceWidth, // YStride |
- kSourceWidth / 2); // UVStride |
- |
- uint32_t rgb_hash = |
- DJB2Hash(yuv_converted_bytes.get(), kYUV12Size, kDJB2HashSeed); |
- EXPECT_EQ(320824432u, rgb_hash); |
-} |
- |
-TEST(YUVConvertTest, RGB32ToYUV) { |
- // Allocate all surfaces. |
- std::unique_ptr<uint8_t[]> yuv_bytes(new uint8_t[kYUV12Size]); |
- std::unique_ptr<uint8_t[]> rgb_bytes(new uint8_t[kRGBSize]); |
- std::unique_ptr<uint8_t[]> yuv_converted_bytes(new uint8_t[kYUV12Size]); |
- std::unique_ptr<uint8_t[]> rgb_converted_bytes(new uint8_t[kRGBSize]); |
- |
- // Read YUV reference data from file. |
- base::FilePath yuv_url; |
- EXPECT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT, &yuv_url)); |
- yuv_url = yuv_url.Append(FILE_PATH_LITERAL("media")) |
- .Append(FILE_PATH_LITERAL("test")) |
- .Append(FILE_PATH_LITERAL("data")) |
- .Append(FILE_PATH_LITERAL("bali_640x360_P420.yuv")); |
- EXPECT_EQ(static_cast<int>(kYUV12Size), |
- base::ReadFile(yuv_url, |
- reinterpret_cast<char*>(yuv_bytes.get()), |
- static_cast<int>(kYUV12Size))); |
- |
- // Convert a frame of YUV to 32 bit ARGB. |
- media::ConvertYUVToRGB32(yuv_bytes.get(), |
- yuv_bytes.get() + kSourceUOffset, |
- yuv_bytes.get() + kSourceVOffset, |
- rgb_bytes.get(), // RGB output |
- kSourceWidth, kSourceHeight, // Dimensions |
- kSourceWidth, // YStride |
- kSourceWidth / 2, // UVStride |
- kSourceWidth * kBpp, // RGBStride |
- media::YV12); |
- |
- // Convert RGB32 to YV12. |
- media::ConvertRGB32ToYUV(rgb_bytes.get(), |
- yuv_converted_bytes.get(), |
- yuv_converted_bytes.get() + kSourceUOffset, |
- yuv_converted_bytes.get() + kSourceVOffset, |
- kSourceWidth, kSourceHeight, // Dimensions |
- kSourceWidth * 4, // RGBStride |
- kSourceWidth, // YStride |
- kSourceWidth / 2); // UVStride |
- |
- // Convert YV12 back to RGB32. |
- media::ConvertYUVToRGB32(yuv_converted_bytes.get(), |
- yuv_converted_bytes.get() + kSourceUOffset, |
- yuv_converted_bytes.get() + kSourceVOffset, |
- rgb_converted_bytes.get(), // RGB output |
- kSourceWidth, kSourceHeight, // Dimensions |
- kSourceWidth, // YStride |
- kSourceWidth / 2, // UVStride |
- kSourceWidth * kBpp, // RGBStride |
- media::YV12); |
- |
- int error = 0; |
- for (int i = 0; i < kRGBSize; ++i) { |
- int diff = rgb_converted_bytes[i] - rgb_bytes[i]; |
- if (diff < 0) |
- diff = -diff; |
- error += diff; |
- } |
- |
- // Make sure error is within bound. |
- DVLOG(1) << "Average error per channel: " << error / kRGBSize; |
- EXPECT_GT(5, error / kRGBSize); |
-} |
- |
-TEST(YUVConvertTest, DownScaleYUVToRGB32WithRect) { |
- // Read YUV reference data from file. |
- base::FilePath yuv_url; |
- EXPECT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT, &yuv_url)); |
- yuv_url = yuv_url.Append(FILE_PATH_LITERAL("media")) |
- .Append(FILE_PATH_LITERAL("test")) |
- .Append(FILE_PATH_LITERAL("data")) |
- .Append(FILE_PATH_LITERAL("bali_640x360_P420.yuv")); |
- const size_t size_of_yuv = kSourceYSize * 12 / 8; // 12 bpp. |
- std::unique_ptr<uint8_t[]> yuv_bytes(new uint8_t[size_of_yuv]); |
- EXPECT_EQ(static_cast<int>(size_of_yuv), |
- base::ReadFile(yuv_url, |
- reinterpret_cast<char*>(yuv_bytes.get()), |
- static_cast<int>(size_of_yuv))); |
- |
- // Scale the full frame of YUV to 32 bit ARGB. |
- // The API currently only supports down-scaling, so we don't test up-scaling. |
- const size_t size_of_rgb_scaled = kDownScaledWidth * kDownScaledHeight * kBpp; |
- std::unique_ptr<uint8_t[]> rgb_scaled_bytes(new uint8_t[size_of_rgb_scaled]); |
- gfx::Rect sub_rect(0, 0, kDownScaledWidth, kDownScaledHeight); |
- |
- // We can't compare with the full-frame scaler because it uses slightly |
- // different sampling coordinates. |
- media::ScaleYUVToRGB32WithRect( |
- yuv_bytes.get(), // Y |
- yuv_bytes.get() + kSourceUOffset, // U |
- yuv_bytes.get() + kSourceVOffset, // V |
- rgb_scaled_bytes.get(), // Rgb output |
- kSourceWidth, kSourceHeight, // Dimensions |
- kDownScaledWidth, kDownScaledHeight, // Dimensions |
- sub_rect.x(), sub_rect.y(), // Dest rect |
- sub_rect.right(), sub_rect.bottom(), // Dest rect |
- kSourceWidth, // YStride |
- kSourceWidth / 2, // UvStride |
- kDownScaledWidth * kBpp); // RgbStride |
- |
- uint32_t rgb_hash_full_rect = |
- DJB2Hash(rgb_scaled_bytes.get(), size_of_rgb_scaled, kDJB2HashSeed); |
- |
- // Re-scale sub-rectangles and verify the results are the same. |
- int next_sub_rect = 0; |
- while (!sub_rect.IsEmpty()) { |
- // Scale a partial rectangle. |
- media::ScaleYUVToRGB32WithRect( |
- yuv_bytes.get(), // Y |
- yuv_bytes.get() + kSourceUOffset, // U |
- yuv_bytes.get() + kSourceVOffset, // V |
- rgb_scaled_bytes.get(), // Rgb output |
- kSourceWidth, kSourceHeight, // Dimensions |
- kDownScaledWidth, kDownScaledHeight, // Dimensions |
- sub_rect.x(), sub_rect.y(), // Dest rect |
- sub_rect.right(), sub_rect.bottom(), // Dest rect |
- kSourceWidth, // YStride |
- kSourceWidth / 2, // UvStride |
- kDownScaledWidth * kBpp); // RgbStride |
- uint32_t rgb_hash_sub_rect = |
- DJB2Hash(rgb_scaled_bytes.get(), size_of_rgb_scaled, kDJB2HashSeed); |
- |
- EXPECT_EQ(rgb_hash_full_rect, rgb_hash_sub_rect); |
- |
- // Now pick choose a quarter rect of this sub-rect. |
- if (next_sub_rect & 1) |
- sub_rect.set_x(sub_rect.x() + sub_rect.width() / 2); |
- if (next_sub_rect & 2) |
- sub_rect.set_y(sub_rect.y() + sub_rect.height() / 2); |
- sub_rect.set_width(sub_rect.width() / 2); |
- sub_rect.set_height(sub_rect.height() / 2); |
- next_sub_rect++; |
- } |
-} |
- |
-#if !defined(ARCH_CPU_ARM_FAMILY) && !defined(ARCH_CPU_MIPS_FAMILY) |
-#if !defined(OS_ANDROID) |
-TEST(YUVConvertTest, YUVAtoARGB_MMX_MatchReference) { |
- // Allocate all surfaces. |
- std::unique_ptr<uint8_t[]> yuv_bytes; |
- std::unique_ptr<uint8_t[]> rgb_bytes(new uint8_t[kRGBSize]); |
- std::unique_ptr<uint8_t[]> rgb_converted_bytes( |
- new uint8_t[kRGBSizeConverted]); |
- std::unique_ptr<uint8_t[]> rgb_converted_bytes_ref( |
- new uint8_t[kRGBSizeConverted]); |
- |
- // Read YUV reference data from file. |
- ReadYV12AData(&yuv_bytes); |
- |
- // Convert a frame of YUV to 32 bit ARGB using both C and MMX versions. |
- media::ConvertYUVAToARGB_C(yuv_bytes.get(), |
- yuv_bytes.get() + kSourceUOffset, |
- yuv_bytes.get() + kSourceVOffset, |
- yuv_bytes.get() + kSourceAOffset, |
- rgb_converted_bytes_ref.get(), |
- kSourceWidth, |
- kSourceHeight, |
- kSourceWidth, |
- kSourceWidth / 2, |
- kSourceWidth, |
- kSourceWidth * kBpp, |
- media::YV12); |
- media::ConvertYUVAToARGB_MMX(yuv_bytes.get(), |
- yuv_bytes.get() + kSourceUOffset, |
- yuv_bytes.get() + kSourceVOffset, |
- yuv_bytes.get() + kSourceAOffset, |
- rgb_converted_bytes.get(), |
- kSourceWidth, |
- kSourceHeight, |
- kSourceWidth, |
- kSourceWidth / 2, |
- kSourceWidth, |
- kSourceWidth * kBpp, |
- media::YV12); |
- |
- EXPECT_EQ(0, |
- memcmp(rgb_converted_bytes.get(), |
- rgb_converted_bytes_ref.get(), |
- kRGBSizeConverted)); |
-} |
-#endif // !defined(OS_ANDROID) |
- |
-TEST(YUVConvertTest, RGB32ToYUV_SSE2_MatchReference) { |
- base::CPU cpu; |
- if (!cpu.has_sse2()) { |
- LOG(WARNING) << "System doesn't support SSE2, test not executed."; |
- return; |
- } |
- |
- // Allocate all surfaces. |
- std::unique_ptr<uint8_t[]> yuv_bytes(new uint8_t[kYUV12Size]); |
- std::unique_ptr<uint8_t[]> rgb_bytes(new uint8_t[kRGBSize]); |
- std::unique_ptr<uint8_t[]> yuv_converted_bytes(new uint8_t[kYUV12Size]); |
- std::unique_ptr<uint8_t[]> yuv_reference_bytes(new uint8_t[kYUV12Size]); |
- |
- ReadYV12Data(&yuv_bytes); |
- |
- // Convert a frame of YUV to 32 bit ARGB. |
- media::ConvertYUVToRGB32( |
- yuv_bytes.get(), |
- yuv_bytes.get() + kSourceUOffset, |
- yuv_bytes.get() + kSourceVOffset, |
- rgb_bytes.get(), // RGB output |
- kSourceWidth, kSourceHeight, // Dimensions |
- kSourceWidth, // YStride |
- kSourceWidth / 2, // UVStride |
- kSourceWidth * kBpp, // RGBStride |
- media::YV12); |
- |
- // Convert RGB32 to YV12 with SSE2 version. |
- media::ConvertRGB32ToYUV_SSE2( |
- rgb_bytes.get(), |
- yuv_converted_bytes.get(), |
- yuv_converted_bytes.get() + kSourceUOffset, |
- yuv_converted_bytes.get() + kSourceVOffset, |
- kSourceWidth, kSourceHeight, // Dimensions |
- kSourceWidth * 4, // RGBStride |
- kSourceWidth, // YStride |
- kSourceWidth / 2); // UVStride |
- |
- // Convert RGB32 to YV12 with reference version. |
- media::ConvertRGB32ToYUV_SSE2_Reference( |
- rgb_bytes.get(), |
- yuv_reference_bytes.get(), |
- yuv_reference_bytes.get() + kSourceUOffset, |
- yuv_reference_bytes.get() + kSourceVOffset, |
- kSourceWidth, kSourceHeight, // Dimensions |
- kSourceWidth * 4, // RGBStride |
- kSourceWidth, // YStride |
- kSourceWidth / 2); // UVStride |
- |
- // Now convert a odd width and height, this overrides part of the buffer |
- // generated above but that is fine because the point of this test is to |
- // match the result with the reference code. |
- |
- // Convert RGB32 to YV12 with SSE2 version. |
- media::ConvertRGB32ToYUV_SSE2( |
- rgb_bytes.get(), |
- yuv_converted_bytes.get(), |
- yuv_converted_bytes.get() + kSourceUOffset, |
- yuv_converted_bytes.get() + kSourceVOffset, |
- 7, 7, // Dimensions |
- kSourceWidth * 4, // RGBStride |
- kSourceWidth, // YStride |
- kSourceWidth / 2); // UVStride |
- |
- // Convert RGB32 to YV12 with reference version. |
- media::ConvertRGB32ToYUV_SSE2_Reference( |
- rgb_bytes.get(), |
- yuv_reference_bytes.get(), |
- yuv_reference_bytes.get() + kSourceUOffset, |
- yuv_reference_bytes.get() + kSourceVOffset, |
- 7, 7, // Dimensions |
- kSourceWidth * 4, // RGBStride |
- kSourceWidth, // YStride |
- kSourceWidth / 2); // UVStride |
- |
- int error = 0; |
- for (int i = 0; i < kYUV12Size; ++i) { |
- int diff = yuv_reference_bytes[i] - yuv_converted_bytes[i]; |
- if (diff < 0) |
- diff = -diff; |
- error += diff; |
- } |
- |
- // Make sure there's no difference from the reference. |
- EXPECT_EQ(0, error); |
-} |
- |
-TEST(YUVConvertTest, ConvertYUVToRGB32Row_SSE) { |
- base::CPU cpu; |
- if (!cpu.has_sse()) { |
- LOG(WARNING) << "System not supported. Test skipped."; |
- return; |
- } |
- |
- std::unique_ptr<uint8_t[]> yuv_bytes(new uint8_t[kYUV12Size]); |
- std::unique_ptr<uint8_t[]> rgb_bytes_reference(new uint8_t[kRGBSize]); |
- std::unique_ptr<uint8_t[]> rgb_bytes_converted(new uint8_t[kRGBSize]); |
- ReadYV12Data(&yuv_bytes); |
- |
- const int kWidth = 167; |
- ConvertYUVToRGB32Row_C(yuv_bytes.get(), |
- yuv_bytes.get() + kSourceUOffset, |
- yuv_bytes.get() + kSourceVOffset, |
- rgb_bytes_reference.get(), |
- kWidth, |
- GetLookupTable(YV12)); |
- ConvertYUVToRGB32Row_SSE(yuv_bytes.get(), |
- yuv_bytes.get() + kSourceUOffset, |
- yuv_bytes.get() + kSourceVOffset, |
- rgb_bytes_converted.get(), |
- kWidth, |
- GetLookupTable(YV12)); |
- media::EmptyRegisterState(); |
- EXPECT_EQ(0, memcmp(rgb_bytes_reference.get(), |
- rgb_bytes_converted.get(), |
- kWidth * kBpp)); |
-} |
- |
-// 64-bit release + component builds on Windows are too smart and optimizes |
-// away the function being tested. |
-#if defined(OS_WIN) && (defined(ARCH_CPU_X86) || !defined(COMPONENT_BUILD)) |
-TEST(YUVConvertTest, ScaleYUVToRGB32Row_SSE) { |
- base::CPU cpu; |
- if (!cpu.has_sse()) { |
- LOG(WARNING) << "System not supported. Test skipped."; |
- return; |
- } |
- |
- std::unique_ptr<uint8_t[]> yuv_bytes(new uint8_t[kYUV12Size]); |
- std::unique_ptr<uint8_t[]> rgb_bytes_reference(new uint8_t[kRGBSize]); |
- std::unique_ptr<uint8_t[]> rgb_bytes_converted(new uint8_t[kRGBSize]); |
- ReadYV12Data(&yuv_bytes); |
- |
- const int kWidth = 167; |
- const int kSourceDx = 80000; // This value means a scale down. |
- ScaleYUVToRGB32Row_C(yuv_bytes.get(), |
- yuv_bytes.get() + kSourceUOffset, |
- yuv_bytes.get() + kSourceVOffset, |
- rgb_bytes_reference.get(), |
- kWidth, |
- kSourceDx, |
- GetLookupTable(YV12)); |
- ScaleYUVToRGB32Row_SSE(yuv_bytes.get(), |
- yuv_bytes.get() + kSourceUOffset, |
- yuv_bytes.get() + kSourceVOffset, |
- rgb_bytes_converted.get(), |
- kWidth, |
- kSourceDx, |
- GetLookupTable(YV12)); |
- media::EmptyRegisterState(); |
- EXPECT_EQ(0, memcmp(rgb_bytes_reference.get(), |
- rgb_bytes_converted.get(), |
- kWidth * kBpp)); |
-} |
- |
-TEST(YUVConvertTest, LinearScaleYUVToRGB32Row_SSE) { |
- base::CPU cpu; |
- if (!cpu.has_sse()) { |
- LOG(WARNING) << "System not supported. Test skipped."; |
- return; |
- } |
- |
- std::unique_ptr<uint8_t[]> yuv_bytes(new uint8_t[kYUV12Size]); |
- std::unique_ptr<uint8_t[]> rgb_bytes_reference(new uint8_t[kRGBSize]); |
- std::unique_ptr<uint8_t[]> rgb_bytes_converted(new uint8_t[kRGBSize]); |
- ReadYV12Data(&yuv_bytes); |
- |
- const int kWidth = 167; |
- const int kSourceDx = 80000; // This value means a scale down. |
- LinearScaleYUVToRGB32Row_C(yuv_bytes.get(), |
- yuv_bytes.get() + kSourceUOffset, |
- yuv_bytes.get() + kSourceVOffset, |
- rgb_bytes_reference.get(), |
- kWidth, |
- kSourceDx, |
- GetLookupTable(YV12)); |
- LinearScaleYUVToRGB32Row_SSE(yuv_bytes.get(), |
- yuv_bytes.get() + kSourceUOffset, |
- yuv_bytes.get() + kSourceVOffset, |
- rgb_bytes_converted.get(), |
- kWidth, |
- kSourceDx, |
- GetLookupTable(YV12)); |
- media::EmptyRegisterState(); |
- EXPECT_EQ(0, memcmp(rgb_bytes_reference.get(), |
- rgb_bytes_converted.get(), |
- kWidth * kBpp)); |
-} |
-#endif // defined(OS_WIN) && (ARCH_CPU_X86 || COMPONENT_BUILD) |
- |
-TEST(YUVConvertTest, FilterYUVRows_C_OutOfBounds) { |
- std::unique_ptr<uint8_t[]> src(new uint8_t[16]); |
- std::unique_ptr<uint8_t[]> dst(new uint8_t[16]); |
- |
- memset(src.get(), 0xff, 16); |
- memset(dst.get(), 0, 16); |
- |
- media::FilterYUVRows_C(dst.get(), src.get(), src.get(), 1, 255); |
- |
- EXPECT_EQ(255u, dst[0]); |
- for (int i = 1; i < 16; ++i) { |
- EXPECT_EQ(0u, dst[i]) << " not equal at " << i; |
- } |
-} |
- |
-TEST(YUVConvertTest, FilterYUVRows_SSE2_OutOfBounds) { |
- base::CPU cpu; |
- if (!cpu.has_sse2()) { |
- LOG(WARNING) << "System not supported. Test skipped."; |
- return; |
- } |
- |
- std::unique_ptr<uint8_t[]> src(new uint8_t[16]); |
- std::unique_ptr<uint8_t[]> dst(new uint8_t[16]); |
- |
- memset(src.get(), 0xff, 16); |
- memset(dst.get(), 0, 16); |
- |
- media::FilterYUVRows_SSE2(dst.get(), src.get(), src.get(), 1, 255); |
- |
- EXPECT_EQ(255u, dst[0]); |
- for (int i = 1; i < 16; ++i) { |
- EXPECT_EQ(0u, dst[i]); |
- } |
-} |
- |
-TEST(YUVConvertTest, FilterYUVRows_SSE2_UnalignedDestination) { |
- base::CPU cpu; |
- if (!cpu.has_sse2()) { |
- LOG(WARNING) << "System not supported. Test skipped."; |
- return; |
- } |
- |
- const int kSize = 64; |
- std::unique_ptr<uint8_t[]> src(new uint8_t[kSize]); |
- std::unique_ptr<uint8_t[]> dst_sample(new uint8_t[kSize]); |
- std::unique_ptr<uint8_t[]> dst(new uint8_t[kSize]); |
- |
- memset(dst_sample.get(), 0, kSize); |
- memset(dst.get(), 0, kSize); |
- for (int i = 0; i < kSize; ++i) |
- src[i] = 100 + i; |
- |
- media::FilterYUVRows_C(dst_sample.get(), |
- src.get(), src.get(), 37, 128); |
- |
- // Generate an unaligned output address. |
- uint8_t* dst_ptr = reinterpret_cast<uint8_t*>( |
- (reinterpret_cast<uintptr_t>(dst.get() + 16) & ~15) + 1); |
- media::FilterYUVRows_SSE2(dst_ptr, src.get(), src.get(), 37, 128); |
- media::EmptyRegisterState(); |
- |
- EXPECT_EQ(0, memcmp(dst_sample.get(), dst_ptr, 37)); |
-} |
- |
-#if defined(ARCH_CPU_X86_64) |
- |
-TEST(YUVConvertTest, ScaleYUVToRGB32Row_SSE2_X64) { |
- std::unique_ptr<uint8_t[]> yuv_bytes(new uint8_t[kYUV12Size]); |
- std::unique_ptr<uint8_t[]> rgb_bytes_reference(new uint8_t[kRGBSize]); |
- std::unique_ptr<uint8_t[]> rgb_bytes_converted(new uint8_t[kRGBSize]); |
- ReadYV12Data(&yuv_bytes); |
- |
- const int kWidth = 167; |
- const int kSourceDx = 80000; // This value means a scale down. |
- ScaleYUVToRGB32Row_C(yuv_bytes.get(), |
- yuv_bytes.get() + kSourceUOffset, |
- yuv_bytes.get() + kSourceVOffset, |
- rgb_bytes_reference.get(), |
- kWidth, |
- kSourceDx, |
- GetLookupTable(YV12)); |
- ScaleYUVToRGB32Row_SSE2_X64(yuv_bytes.get(), |
- yuv_bytes.get() + kSourceUOffset, |
- yuv_bytes.get() + kSourceVOffset, |
- rgb_bytes_converted.get(), |
- kWidth, |
- kSourceDx, |
- GetLookupTable(YV12)); |
- media::EmptyRegisterState(); |
- EXPECT_EQ(0, memcmp(rgb_bytes_reference.get(), |
- rgb_bytes_converted.get(), |
- kWidth * kBpp)); |
-} |
- |
-TEST(YUVConvertTest, LinearScaleYUVToRGB32Row_MMX_X64) { |
- std::unique_ptr<uint8_t[]> yuv_bytes(new uint8_t[kYUV12Size]); |
- std::unique_ptr<uint8_t[]> rgb_bytes_reference(new uint8_t[kRGBSize]); |
- std::unique_ptr<uint8_t[]> rgb_bytes_converted(new uint8_t[kRGBSize]); |
- ReadYV12Data(&yuv_bytes); |
- |
- const int kWidth = 167; |
- const int kSourceDx = 80000; // This value means a scale down. |
- LinearScaleYUVToRGB32Row_C(yuv_bytes.get(), |
- yuv_bytes.get() + kSourceUOffset, |
- yuv_bytes.get() + kSourceVOffset, |
- rgb_bytes_reference.get(), |
- kWidth, |
- kSourceDx, |
- GetLookupTable(YV12)); |
- LinearScaleYUVToRGB32Row_MMX_X64(yuv_bytes.get(), |
- yuv_bytes.get() + kSourceUOffset, |
- yuv_bytes.get() + kSourceVOffset, |
- rgb_bytes_converted.get(), |
- kWidth, |
- kSourceDx, |
- GetLookupTable(YV12)); |
- media::EmptyRegisterState(); |
- EXPECT_EQ(0, memcmp(rgb_bytes_reference.get(), |
- rgb_bytes_converted.get(), |
- kWidth * kBpp)); |
-} |
- |
-#endif // defined(ARCH_CPU_X86_64) |
- |
-#endif // defined(ARCH_CPU_X86_FAMILY) |
- |
-} // namespace media |