1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
5 #include "base/base_paths.h"
7 #include "base/file_util.h"
8 #include "base/logging.h"
9 #include "base/path_service.h"
10 #include "media/base/djb2.h"
11 #include "media/base/simd/convert_rgb_to_yuv.h"
12 #include "media/base/simd/convert_yuv_to_rgb.h"
13 #include "media/base/simd/filter_yuv.h"
14 #include "media/base/simd/yuv_to_rgb_table.h"
15 #include "media/base/yuv_convert.h"
16 #include "testing/gtest/include/gtest/gtest.h"
17 #include "ui/gfx/rect.h"
20 static const int kSourceWidth
= 640;
21 static const int kSourceHeight
= 360;
22 static const int kSourceYSize
= kSourceWidth
* kSourceHeight
;
23 static const int kSourceUOffset
= kSourceYSize
;
24 static const int kSourceVOffset
= kSourceYSize
* 5 / 4;
25 static const int kScaledWidth
= 1024;
26 static const int kScaledHeight
= 768;
27 static const int kDownScaledWidth
= 512;
28 static const int kDownScaledHeight
= 320;
29 static const int kBpp
= 4;
31 // Surface sizes for various test files.
32 static const int kYUV12Size
= kSourceYSize
* 12 / 8;
33 static const int kYUV16Size
= kSourceYSize
* 16 / 8;
34 static const int kYUY2Size
= kSourceYSize
* 16 / 8;
35 static const int kRGBSize
= kSourceYSize
* kBpp
;
36 static const int kRGBSizeScaled
= kScaledWidth
* kScaledHeight
* kBpp
;
37 static const int kRGB24Size
= kSourceYSize
* 3;
38 static const int kRGBSizeConverted
= kSourceYSize
* kBpp
;
40 #if !defined(ARCH_CPU_ARM_FAMILY) && !defined(ARCH_CPU_MIPS_FAMILY)
41 static const int kSourceAOffset
= kSourceYSize
* 12 / 8;
42 static const int kYUVA12Size
= kSourceYSize
* 20 / 8;
45 // Helper for reading test data into a scoped_ptr<uint8[]>.
46 static void ReadData(const base::FilePath::CharType
* filename
,
48 scoped_ptr
<uint8
[]>* data
) {
49 data
->reset(new uint8
[expected_size
]);
52 CHECK(PathService::Get(base::DIR_SOURCE_ROOT
, &path
));
53 path
= path
.Append(FILE_PATH_LITERAL("media"))
54 .Append(FILE_PATH_LITERAL("test"))
55 .Append(FILE_PATH_LITERAL("data"))
58 // Verify file size is correct.
59 int64 actual_size
= 0;
60 base::GetFileSize(path
, &actual_size
);
61 CHECK_EQ(actual_size
, expected_size
);
63 // Verify bytes read are correct.
64 int bytes_read
= base::ReadFile(
65 path
, reinterpret_cast<char*>(data
->get()), expected_size
);
66 CHECK_EQ(bytes_read
, expected_size
);
69 static void ReadYV12Data(scoped_ptr
<uint8
[]>* data
) {
70 ReadData(FILE_PATH_LITERAL("bali_640x360_P420.yuv"), kYUV12Size
, data
);
73 static void ReadYV16Data(scoped_ptr
<uint8
[]>* data
) {
74 ReadData(FILE_PATH_LITERAL("bali_640x360_P422.yuv"), kYUV16Size
, data
);
77 #if !defined(ARCH_CPU_ARM_FAMILY) && !defined(ARCH_CPU_MIPS_FAMILY)
78 static void ReadYV12AData(scoped_ptr
<uint8
[]>* data
) {
79 ReadData(FILE_PATH_LITERAL("bali_640x360_P420_alpha.yuv"), kYUVA12Size
, data
);
83 static void ReadRGB24Data(scoped_ptr
<uint8
[]>* data
) {
84 ReadData(FILE_PATH_LITERAL("bali_640x360_RGB24.rgb"), kRGB24Size
, data
);
87 static void ReadYUY2Data(scoped_ptr
<uint8
[]>* data
) {
88 ReadData(FILE_PATH_LITERAL("bali_640x360_YUY2.yuv"), kYUY2Size
, data
);
91 #if defined(OS_ANDROID)
92 // Helper for swapping red and blue channels of RGBA or BGRA.
93 static void SwapRedAndBlueChannels(unsigned char* pixels
, size_t buffer_size
) {
94 for (size_t i
= 0; i
< buffer_size
; i
+= 4) {
95 std::swap(pixels
[i
], pixels
[i
+ 2]);
102 TEST(YUVConvertTest
, YV12
) {
103 // Allocate all surfaces.
104 scoped_ptr
<uint8
[]> yuv_bytes
;
105 scoped_ptr
<uint8
[]> rgb_bytes(new uint8
[kRGBSize
]);
106 scoped_ptr
<uint8
[]> rgb_converted_bytes(new uint8
[kRGBSizeConverted
]);
108 // Read YUV reference data from file.
109 ReadYV12Data(&yuv_bytes
);
111 // Convert a frame of YUV to 32 bit ARGB.
112 media::ConvertYUVToRGB32(yuv_bytes
.get(),
113 yuv_bytes
.get() + kSourceUOffset
,
114 yuv_bytes
.get() + kSourceVOffset
,
115 rgb_converted_bytes
.get(), // RGB output
116 kSourceWidth
, kSourceHeight
, // Dimensions
117 kSourceWidth
, // YStride
118 kSourceWidth
/ 2, // UVStride
119 kSourceWidth
* kBpp
, // RGBStride
122 #if defined(OS_ANDROID)
123 SwapRedAndBlueChannels(rgb_converted_bytes
.get(), kRGBSizeConverted
);
126 uint32 rgb_hash
= DJB2Hash(rgb_converted_bytes
.get(), kRGBSizeConverted
,
128 EXPECT_EQ(2413171226u, rgb_hash
);
131 TEST(YUVConvertTest
, YV16
) {
132 // Allocate all surfaces.
133 scoped_ptr
<uint8
[]> yuv_bytes
;
134 scoped_ptr
<uint8
[]> rgb_bytes(new uint8
[kRGBSize
]);
135 scoped_ptr
<uint8
[]> rgb_converted_bytes(new uint8
[kRGBSizeConverted
]);
137 // Read YUV reference data from file.
138 ReadYV16Data(&yuv_bytes
);
140 // Convert a frame of YUV to 32 bit ARGB.
141 media::ConvertYUVToRGB32(yuv_bytes
.get(), // Y
142 yuv_bytes
.get() + kSourceUOffset
, // U
143 yuv_bytes
.get() + kSourceYSize
* 3 / 2, // V
144 rgb_converted_bytes
.get(), // RGB output
145 kSourceWidth
, kSourceHeight
, // Dimensions
146 kSourceWidth
, // YStride
147 kSourceWidth
/ 2, // UVStride
148 kSourceWidth
* kBpp
, // RGBStride
151 #if defined(OS_ANDROID)
152 SwapRedAndBlueChannels(rgb_converted_bytes
.get(), kRGBSizeConverted
);
155 uint32 rgb_hash
= DJB2Hash(rgb_converted_bytes
.get(), kRGBSizeConverted
,
157 EXPECT_EQ(4222342047u, rgb_hash
);
160 struct YUVScaleTestData
{
161 YUVScaleTestData(media::YUVType y
, media::ScaleFilter s
, uint32 r
)
167 media::YUVType yuv_type
;
168 media::ScaleFilter scale_filter
;
172 class YUVScaleTest
: public ::testing::TestWithParam
<YUVScaleTestData
> {
175 switch (GetParam().yuv_type
) {
178 ReadYV12Data(&yuv_bytes_
);
181 ReadYV16Data(&yuv_bytes_
);
185 rgb_bytes_
.reset(new uint8
[kRGBSizeScaled
]);
188 // Helpers for getting the proper Y, U and V plane offsets.
189 uint8
* y_plane() { return yuv_bytes_
.get(); }
190 uint8
* u_plane() { return yuv_bytes_
.get() + kSourceYSize
; }
192 switch (GetParam().yuv_type
) {
195 return yuv_bytes_
.get() + kSourceVOffset
;
197 return yuv_bytes_
.get() + kSourceYSize
* 3 / 2;
202 scoped_ptr
<uint8
[]> yuv_bytes_
;
203 scoped_ptr
<uint8
[]> rgb_bytes_
;
206 TEST_P(YUVScaleTest
, NoScale
) {
207 media::ScaleYUVToRGB32(y_plane(), // Y
210 rgb_bytes_
.get(), // RGB output
211 kSourceWidth
, kSourceHeight
, // Dimensions
212 kSourceWidth
, kSourceHeight
, // Dimensions
213 kSourceWidth
, // YStride
214 kSourceWidth
/ 2, // UvStride
215 kSourceWidth
* kBpp
, // RgbStride
218 GetParam().scale_filter
);
220 uint32 yuv_hash
= DJB2Hash(rgb_bytes_
.get(), kRGBSize
, kDJB2HashSeed
);
222 media::ConvertYUVToRGB32(y_plane(), // Y
225 rgb_bytes_
.get(), // RGB output
226 kSourceWidth
, kSourceHeight
, // Dimensions
227 kSourceWidth
, // YStride
228 kSourceWidth
/ 2, // UVStride
229 kSourceWidth
* kBpp
, // RGBStride
230 GetParam().yuv_type
);
232 uint32 rgb_hash
= DJB2Hash(rgb_bytes_
.get(), kRGBSize
, kDJB2HashSeed
);
234 EXPECT_EQ(yuv_hash
, rgb_hash
);
237 TEST_P(YUVScaleTest
, Normal
) {
238 media::ScaleYUVToRGB32(y_plane(), // Y
241 rgb_bytes_
.get(), // RGB output
242 kSourceWidth
, kSourceHeight
, // Dimensions
243 kScaledWidth
, kScaledHeight
, // Dimensions
244 kSourceWidth
, // YStride
245 kSourceWidth
/ 2, // UvStride
246 kScaledWidth
* kBpp
, // RgbStride
249 GetParam().scale_filter
);
251 #if defined(OS_ANDROID)
252 SwapRedAndBlueChannels(rgb_bytes_
.get(), kRGBSizeScaled
);
255 uint32 rgb_hash
= DJB2Hash(rgb_bytes_
.get(), kRGBSizeScaled
, kDJB2HashSeed
);
256 EXPECT_EQ(GetParam().rgb_hash
, rgb_hash
);
259 TEST_P(YUVScaleTest
, ZeroSourceSize
) {
260 media::ScaleYUVToRGB32(y_plane(), // Y
263 rgb_bytes_
.get(), // RGB output
265 kScaledWidth
, kScaledHeight
, // Dimensions
266 kSourceWidth
, // YStride
267 kSourceWidth
/ 2, // UvStride
268 kScaledWidth
* kBpp
, // RgbStride
271 GetParam().scale_filter
);
273 // Testing for out-of-bound read/writes with AddressSanitizer.
276 TEST_P(YUVScaleTest
, ZeroDestinationSize
) {
277 media::ScaleYUVToRGB32(y_plane(), // Y
280 rgb_bytes_
.get(), // RGB output
281 kSourceWidth
, kSourceHeight
, // Dimensions
283 kSourceWidth
, // YStride
284 kSourceWidth
/ 2, // UvStride
285 kScaledWidth
* kBpp
, // RgbStride
288 GetParam().scale_filter
);
290 // Testing for out-of-bound read/writes with AddressSanitizer.
293 TEST_P(YUVScaleTest
, OddWidthAndHeightNotCrash
) {
294 media::ScaleYUVToRGB32(y_plane(), // Y
297 rgb_bytes_
.get(), // RGB output
298 kSourceWidth
, kSourceHeight
, // Dimensions
300 kSourceWidth
, // YStride
301 kSourceWidth
/ 2, // UvStride
302 kScaledWidth
* kBpp
, // RgbStride
305 GetParam().scale_filter
);
308 INSTANTIATE_TEST_CASE_P(
309 YUVScaleFormats
, YUVScaleTest
,
311 YUVScaleTestData(media::YV12
, media::FILTER_NONE
, 4136904952u),
312 YUVScaleTestData(media::YV16
, media::FILTER_NONE
, 1501777547u),
313 YUVScaleTestData(media::YV12
, media::FILTER_BILINEAR
, 3164274689u),
314 YUVScaleTestData(media::YV16
, media::FILTER_BILINEAR
, 3095878046u)));
316 // This tests a known worst case YUV value, and for overflow.
317 TEST(YUVConvertTest
, Clamp
) {
318 // Allocate all surfaces.
319 scoped_ptr
<uint8
[]> yuv_bytes(new uint8
[1]);
320 scoped_ptr
<uint8
[]> rgb_bytes(new uint8
[1]);
321 scoped_ptr
<uint8
[]> rgb_converted_bytes(new uint8
[1]);
323 // Values that failed previously in bug report.
324 unsigned char y
= 255u;
325 unsigned char u
= 255u;
326 unsigned char v
= 19u;
328 // Prefill extra large destination buffer to test for overflow.
329 unsigned char rgb
[8] = { 0, 1, 2, 3, 4, 5, 6, 7 };
330 unsigned char expected
[8] = { 255, 255, 104, 255, 4, 5, 6, 7 };
331 // Convert a frame of YUV to 32 bit ARGB.
332 media::ConvertYUVToRGB32(&y
, // Y
335 &rgb
[0], // RGB output
342 #if defined(OS_ANDROID)
343 SwapRedAndBlueChannels(rgb
, kBpp
);
346 int expected_test
= memcmp(rgb
, expected
, sizeof(expected
));
347 EXPECT_EQ(0, expected_test
);
350 TEST(YUVConvertTest
, RGB24ToYUV
) {
351 // Allocate all surfaces.
352 scoped_ptr
<uint8
[]> rgb_bytes
;
353 scoped_ptr
<uint8
[]> yuv_converted_bytes(new uint8
[kYUV12Size
]);
355 // Read RGB24 reference data from file.
356 ReadRGB24Data(&rgb_bytes
);
359 media::ConvertRGB24ToYUV(rgb_bytes
.get(),
360 yuv_converted_bytes
.get(),
361 yuv_converted_bytes
.get() + kSourceUOffset
,
362 yuv_converted_bytes
.get() + kSourceVOffset
,
363 kSourceWidth
, kSourceHeight
, // Dimensions
364 kSourceWidth
* 3, // RGBStride
365 kSourceWidth
, // YStride
366 kSourceWidth
/ 2); // UVStride
368 uint32 rgb_hash
= DJB2Hash(yuv_converted_bytes
.get(), kYUV12Size
,
370 EXPECT_EQ(320824432u, rgb_hash
);
373 TEST(YUVConvertTest
, RGB32ToYUV
) {
374 // Allocate all surfaces.
375 scoped_ptr
<uint8
[]> yuv_bytes(new uint8
[kYUV12Size
]);
376 scoped_ptr
<uint8
[]> rgb_bytes(new uint8
[kRGBSize
]);
377 scoped_ptr
<uint8
[]> yuv_converted_bytes(new uint8
[kYUV12Size
]);
378 scoped_ptr
<uint8
[]> rgb_converted_bytes(new uint8
[kRGBSize
]);
380 // Read YUV reference data from file.
381 base::FilePath yuv_url
;
382 EXPECT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT
, &yuv_url
));
383 yuv_url
= yuv_url
.Append(FILE_PATH_LITERAL("media"))
384 .Append(FILE_PATH_LITERAL("test"))
385 .Append(FILE_PATH_LITERAL("data"))
386 .Append(FILE_PATH_LITERAL("bali_640x360_P420.yuv"));
387 EXPECT_EQ(static_cast<int>(kYUV12Size
),
388 base::ReadFile(yuv_url
,
389 reinterpret_cast<char*>(yuv_bytes
.get()),
390 static_cast<int>(kYUV12Size
)));
392 // Convert a frame of YUV to 32 bit ARGB.
393 media::ConvertYUVToRGB32(yuv_bytes
.get(),
394 yuv_bytes
.get() + kSourceUOffset
,
395 yuv_bytes
.get() + kSourceVOffset
,
396 rgb_bytes
.get(), // RGB output
397 kSourceWidth
, kSourceHeight
, // Dimensions
398 kSourceWidth
, // YStride
399 kSourceWidth
/ 2, // UVStride
400 kSourceWidth
* kBpp
, // RGBStride
403 // Convert RGB32 to YV12.
404 media::ConvertRGB32ToYUV(rgb_bytes
.get(),
405 yuv_converted_bytes
.get(),
406 yuv_converted_bytes
.get() + kSourceUOffset
,
407 yuv_converted_bytes
.get() + kSourceVOffset
,
408 kSourceWidth
, kSourceHeight
, // Dimensions
409 kSourceWidth
* 4, // RGBStride
410 kSourceWidth
, // YStride
411 kSourceWidth
/ 2); // UVStride
413 // Convert YV12 back to RGB32.
414 media::ConvertYUVToRGB32(yuv_converted_bytes
.get(),
415 yuv_converted_bytes
.get() + kSourceUOffset
,
416 yuv_converted_bytes
.get() + kSourceVOffset
,
417 rgb_converted_bytes
.get(), // RGB output
418 kSourceWidth
, kSourceHeight
, // Dimensions
419 kSourceWidth
, // YStride
420 kSourceWidth
/ 2, // UVStride
421 kSourceWidth
* kBpp
, // RGBStride
425 for (int i
= 0; i
< kRGBSize
; ++i
) {
426 int diff
= rgb_converted_bytes
[i
] - rgb_bytes
[i
];
432 // Make sure error is within bound.
433 DVLOG(1) << "Average error per channel: " << error
/ kRGBSize
;
434 EXPECT_GT(5, error
/ kRGBSize
);
437 TEST(YUVConvertTest
, YUY2ToYUV
) {
438 // Allocate all surfaces.
439 scoped_ptr
<uint8
[]> yuy_bytes
;
440 scoped_ptr
<uint8
[]> yuv_converted_bytes(new uint8
[kYUV12Size
]);
442 // Read YUY reference data from file.
443 ReadYUY2Data(&yuy_bytes
);
446 media::ConvertYUY2ToYUV(yuy_bytes
.get(),
447 yuv_converted_bytes
.get(),
448 yuv_converted_bytes
.get() + kSourceUOffset
,
449 yuv_converted_bytes
.get() + kSourceVOffset
,
450 kSourceWidth
, kSourceHeight
);
452 uint32 yuy_hash
= DJB2Hash(yuv_converted_bytes
.get(), kYUV12Size
,
454 EXPECT_EQ(666823187u, yuy_hash
);
457 TEST(YUVConvertTest
, DownScaleYUVToRGB32WithRect
) {
458 // Read YUV reference data from file.
459 base::FilePath yuv_url
;
460 EXPECT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT
, &yuv_url
));
461 yuv_url
= yuv_url
.Append(FILE_PATH_LITERAL("media"))
462 .Append(FILE_PATH_LITERAL("test"))
463 .Append(FILE_PATH_LITERAL("data"))
464 .Append(FILE_PATH_LITERAL("bali_640x360_P420.yuv"));
465 const size_t size_of_yuv
= kSourceYSize
* 12 / 8; // 12 bpp.
466 scoped_ptr
<uint8
[]> yuv_bytes(new uint8
[size_of_yuv
]);
467 EXPECT_EQ(static_cast<int>(size_of_yuv
),
468 base::ReadFile(yuv_url
,
469 reinterpret_cast<char*>(yuv_bytes
.get()),
470 static_cast<int>(size_of_yuv
)));
472 // Scale the full frame of YUV to 32 bit ARGB.
473 // The API currently only supports down-scaling, so we don't test up-scaling.
474 const size_t size_of_rgb_scaled
= kDownScaledWidth
* kDownScaledHeight
* kBpp
;
475 scoped_ptr
<uint8
[]> rgb_scaled_bytes(new uint8
[size_of_rgb_scaled
]);
476 gfx::Rect
sub_rect(0, 0, kDownScaledWidth
, kDownScaledHeight
);
478 // We can't compare with the full-frame scaler because it uses slightly
479 // different sampling coordinates.
480 media::ScaleYUVToRGB32WithRect(
481 yuv_bytes
.get(), // Y
482 yuv_bytes
.get() + kSourceUOffset
, // U
483 yuv_bytes
.get() + kSourceVOffset
, // V
484 rgb_scaled_bytes
.get(), // Rgb output
485 kSourceWidth
, kSourceHeight
, // Dimensions
486 kDownScaledWidth
, kDownScaledHeight
, // Dimensions
487 sub_rect
.x(), sub_rect
.y(), // Dest rect
488 sub_rect
.right(), sub_rect
.bottom(), // Dest rect
489 kSourceWidth
, // YStride
490 kSourceWidth
/ 2, // UvStride
491 kDownScaledWidth
* kBpp
); // RgbStride
493 uint32 rgb_hash_full_rect
= DJB2Hash(rgb_scaled_bytes
.get(),
497 // Re-scale sub-rectangles and verify the results are the same.
498 int next_sub_rect
= 0;
499 while (!sub_rect
.IsEmpty()) {
500 // Scale a partial rectangle.
501 media::ScaleYUVToRGB32WithRect(
502 yuv_bytes
.get(), // Y
503 yuv_bytes
.get() + kSourceUOffset
, // U
504 yuv_bytes
.get() + kSourceVOffset
, // V
505 rgb_scaled_bytes
.get(), // Rgb output
506 kSourceWidth
, kSourceHeight
, // Dimensions
507 kDownScaledWidth
, kDownScaledHeight
, // Dimensions
508 sub_rect
.x(), sub_rect
.y(), // Dest rect
509 sub_rect
.right(), sub_rect
.bottom(), // Dest rect
510 kSourceWidth
, // YStride
511 kSourceWidth
/ 2, // UvStride
512 kDownScaledWidth
* kBpp
); // RgbStride
513 uint32 rgb_hash_sub_rect
= DJB2Hash(rgb_scaled_bytes
.get(),
517 EXPECT_EQ(rgb_hash_full_rect
, rgb_hash_sub_rect
);
519 // Now pick choose a quarter rect of this sub-rect.
520 if (next_sub_rect
& 1)
521 sub_rect
.set_x(sub_rect
.x() + sub_rect
.width() / 2);
522 if (next_sub_rect
& 2)
523 sub_rect
.set_y(sub_rect
.y() + sub_rect
.height() / 2);
524 sub_rect
.set_width(sub_rect
.width() / 2);
525 sub_rect
.set_height(sub_rect
.height() / 2);
530 #if !defined(ARCH_CPU_ARM_FAMILY) && !defined(ARCH_CPU_MIPS_FAMILY)
531 TEST(YUVConvertTest
, YUVAtoARGB_MMX_MatchReference
) {
532 // Allocate all surfaces.
533 scoped_ptr
<uint8
[]> yuv_bytes
;
534 scoped_ptr
<uint8
[]> rgb_bytes(new uint8
[kRGBSize
]);
535 scoped_ptr
<uint8
[]> rgb_converted_bytes(new uint8
[kRGBSizeConverted
]);
536 scoped_ptr
<uint8
[]> rgb_converted_bytes_ref(new uint8
[kRGBSizeConverted
]);
538 // Read YUV reference data from file.
539 ReadYV12AData(&yuv_bytes
);
541 // Convert a frame of YUV to 32 bit ARGB using both C and MMX versions.
542 media::ConvertYUVAToARGB_C(yuv_bytes
.get(),
543 yuv_bytes
.get() + kSourceUOffset
,
544 yuv_bytes
.get() + kSourceVOffset
,
545 yuv_bytes
.get() + kSourceAOffset
,
546 rgb_converted_bytes_ref
.get(),
554 media::ConvertYUVAToARGB_MMX(yuv_bytes
.get(),
555 yuv_bytes
.get() + kSourceUOffset
,
556 yuv_bytes
.get() + kSourceVOffset
,
557 yuv_bytes
.get() + kSourceAOffset
,
558 rgb_converted_bytes
.get(),
568 memcmp(rgb_converted_bytes
.get(),
569 rgb_converted_bytes_ref
.get(),
573 TEST(YUVConvertTest
, RGB32ToYUV_SSE2_MatchReference
) {
575 if (!cpu
.has_sse2()) {
576 LOG(WARNING
) << "System doesn't support SSE2, test not executed.";
580 // Allocate all surfaces.
581 scoped_ptr
<uint8
[]> yuv_bytes(new uint8
[kYUV12Size
]);
582 scoped_ptr
<uint8
[]> rgb_bytes(new uint8
[kRGBSize
]);
583 scoped_ptr
<uint8
[]> yuv_converted_bytes(new uint8
[kYUV12Size
]);
584 scoped_ptr
<uint8
[]> yuv_reference_bytes(new uint8
[kYUV12Size
]);
586 ReadYV12Data(&yuv_bytes
);
588 // Convert a frame of YUV to 32 bit ARGB.
589 media::ConvertYUVToRGB32(
591 yuv_bytes
.get() + kSourceUOffset
,
592 yuv_bytes
.get() + kSourceVOffset
,
593 rgb_bytes
.get(), // RGB output
594 kSourceWidth
, kSourceHeight
, // Dimensions
595 kSourceWidth
, // YStride
596 kSourceWidth
/ 2, // UVStride
597 kSourceWidth
* kBpp
, // RGBStride
600 // Convert RGB32 to YV12 with SSE2 version.
601 media::ConvertRGB32ToYUV_SSE2(
603 yuv_converted_bytes
.get(),
604 yuv_converted_bytes
.get() + kSourceUOffset
,
605 yuv_converted_bytes
.get() + kSourceVOffset
,
606 kSourceWidth
, kSourceHeight
, // Dimensions
607 kSourceWidth
* 4, // RGBStride
608 kSourceWidth
, // YStride
609 kSourceWidth
/ 2); // UVStride
611 // Convert RGB32 to YV12 with reference version.
612 media::ConvertRGB32ToYUV_SSE2_Reference(
614 yuv_reference_bytes
.get(),
615 yuv_reference_bytes
.get() + kSourceUOffset
,
616 yuv_reference_bytes
.get() + kSourceVOffset
,
617 kSourceWidth
, kSourceHeight
, // Dimensions
618 kSourceWidth
* 4, // RGBStride
619 kSourceWidth
, // YStride
620 kSourceWidth
/ 2); // UVStride
622 // Now convert a odd width and height, this overrides part of the buffer
623 // generated above but that is fine because the point of this test is to
624 // match the result with the reference code.
626 // Convert RGB32 to YV12 with SSE2 version.
627 media::ConvertRGB32ToYUV_SSE2(
629 yuv_converted_bytes
.get(),
630 yuv_converted_bytes
.get() + kSourceUOffset
,
631 yuv_converted_bytes
.get() + kSourceVOffset
,
633 kSourceWidth
* 4, // RGBStride
634 kSourceWidth
, // YStride
635 kSourceWidth
/ 2); // UVStride
637 // Convert RGB32 to YV12 with reference version.
638 media::ConvertRGB32ToYUV_SSE2_Reference(
640 yuv_reference_bytes
.get(),
641 yuv_reference_bytes
.get() + kSourceUOffset
,
642 yuv_reference_bytes
.get() + kSourceVOffset
,
644 kSourceWidth
* 4, // RGBStride
645 kSourceWidth
, // YStride
646 kSourceWidth
/ 2); // UVStride
649 for (int i
= 0; i
< kYUV12Size
; ++i
) {
650 int diff
= yuv_reference_bytes
[i
] - yuv_converted_bytes
[i
];
656 // Make sure there's no difference from the reference.
660 TEST(YUVConvertTest
, ConvertYUVToRGB32Row_MMX
) {
662 if (!cpu
.has_mmx()) {
663 LOG(WARNING
) << "System not supported. Test skipped.";
667 scoped_ptr
<uint8
[]> yuv_bytes(new uint8
[kYUV12Size
]);
668 scoped_ptr
<uint8
[]> rgb_bytes_reference(new uint8
[kRGBSize
]);
669 scoped_ptr
<uint8
[]> rgb_bytes_converted(new uint8
[kRGBSize
]);
670 ReadYV12Data(&yuv_bytes
);
672 const int kWidth
= 167;
673 ConvertYUVToRGB32Row_C(yuv_bytes
.get(),
674 yuv_bytes
.get() + kSourceUOffset
,
675 yuv_bytes
.get() + kSourceVOffset
,
676 rgb_bytes_reference
.get(),
678 GetLookupTable(YV12
));
679 ConvertYUVToRGB32Row_MMX(yuv_bytes
.get(),
680 yuv_bytes
.get() + kSourceUOffset
,
681 yuv_bytes
.get() + kSourceVOffset
,
682 rgb_bytes_converted
.get(),
684 GetLookupTable(YV12
));
685 media::EmptyRegisterState();
686 EXPECT_EQ(0, memcmp(rgb_bytes_reference
.get(),
687 rgb_bytes_converted
.get(),
691 TEST(YUVConvertTest
, ConvertYUVToRGB32Row_SSE
) {
693 if (!cpu
.has_sse()) {
694 LOG(WARNING
) << "System not supported. Test skipped.";
698 scoped_ptr
<uint8
[]> yuv_bytes(new uint8
[kYUV12Size
]);
699 scoped_ptr
<uint8
[]> rgb_bytes_reference(new uint8
[kRGBSize
]);
700 scoped_ptr
<uint8
[]> rgb_bytes_converted(new uint8
[kRGBSize
]);
701 ReadYV12Data(&yuv_bytes
);
703 const int kWidth
= 167;
704 ConvertYUVToRGB32Row_C(yuv_bytes
.get(),
705 yuv_bytes
.get() + kSourceUOffset
,
706 yuv_bytes
.get() + kSourceVOffset
,
707 rgb_bytes_reference
.get(),
709 GetLookupTable(YV12
));
710 ConvertYUVToRGB32Row_SSE(yuv_bytes
.get(),
711 yuv_bytes
.get() + kSourceUOffset
,
712 yuv_bytes
.get() + kSourceVOffset
,
713 rgb_bytes_converted
.get(),
715 GetLookupTable(YV12
));
716 media::EmptyRegisterState();
717 EXPECT_EQ(0, memcmp(rgb_bytes_reference
.get(),
718 rgb_bytes_converted
.get(),
722 TEST(YUVConvertTest
, ScaleYUVToRGB32Row_MMX
) {
724 if (!cpu
.has_mmx()) {
725 LOG(WARNING
) << "System not supported. Test skipped.";
729 scoped_ptr
<uint8
[]> yuv_bytes(new uint8
[kYUV12Size
]);
730 scoped_ptr
<uint8
[]> rgb_bytes_reference(new uint8
[kRGBSize
]);
731 scoped_ptr
<uint8
[]> rgb_bytes_converted(new uint8
[kRGBSize
]);
732 ReadYV12Data(&yuv_bytes
);
734 const int kWidth
= 167;
735 const int kSourceDx
= 80000; // This value means a scale down.
736 ScaleYUVToRGB32Row_C(yuv_bytes
.get(),
737 yuv_bytes
.get() + kSourceUOffset
,
738 yuv_bytes
.get() + kSourceVOffset
,
739 rgb_bytes_reference
.get(),
742 GetLookupTable(YV12
));
743 ScaleYUVToRGB32Row_MMX(yuv_bytes
.get(),
744 yuv_bytes
.get() + kSourceUOffset
,
745 yuv_bytes
.get() + kSourceVOffset
,
746 rgb_bytes_converted
.get(),
749 GetLookupTable(YV12
));
750 media::EmptyRegisterState();
751 EXPECT_EQ(0, memcmp(rgb_bytes_reference
.get(),
752 rgb_bytes_converted
.get(),
756 TEST(YUVConvertTest
, ScaleYUVToRGB32Row_SSE
) {
758 if (!cpu
.has_sse()) {
759 LOG(WARNING
) << "System not supported. Test skipped.";
763 scoped_ptr
<uint8
[]> yuv_bytes(new uint8
[kYUV12Size
]);
764 scoped_ptr
<uint8
[]> rgb_bytes_reference(new uint8
[kRGBSize
]);
765 scoped_ptr
<uint8
[]> rgb_bytes_converted(new uint8
[kRGBSize
]);
766 ReadYV12Data(&yuv_bytes
);
768 const int kWidth
= 167;
769 const int kSourceDx
= 80000; // This value means a scale down.
770 ScaleYUVToRGB32Row_C(yuv_bytes
.get(),
771 yuv_bytes
.get() + kSourceUOffset
,
772 yuv_bytes
.get() + kSourceVOffset
,
773 rgb_bytes_reference
.get(),
776 GetLookupTable(YV12
));
777 ScaleYUVToRGB32Row_SSE(yuv_bytes
.get(),
778 yuv_bytes
.get() + kSourceUOffset
,
779 yuv_bytes
.get() + kSourceVOffset
,
780 rgb_bytes_converted
.get(),
783 GetLookupTable(YV12
));
784 media::EmptyRegisterState();
785 EXPECT_EQ(0, memcmp(rgb_bytes_reference
.get(),
786 rgb_bytes_converted
.get(),
790 TEST(YUVConvertTest
, LinearScaleYUVToRGB32Row_MMX
) {
792 if (!cpu
.has_mmx()) {
793 LOG(WARNING
) << "System not supported. Test skipped.";
797 scoped_ptr
<uint8
[]> yuv_bytes(new uint8
[kYUV12Size
]);
798 scoped_ptr
<uint8
[]> rgb_bytes_reference(new uint8
[kRGBSize
]);
799 scoped_ptr
<uint8
[]> rgb_bytes_converted(new uint8
[kRGBSize
]);
800 ReadYV12Data(&yuv_bytes
);
802 const int kWidth
= 167;
803 const int kSourceDx
= 80000; // This value means a scale down.
804 LinearScaleYUVToRGB32Row_C(yuv_bytes
.get(),
805 yuv_bytes
.get() + kSourceUOffset
,
806 yuv_bytes
.get() + kSourceVOffset
,
807 rgb_bytes_reference
.get(),
810 GetLookupTable(YV12
));
811 LinearScaleYUVToRGB32Row_MMX(yuv_bytes
.get(),
812 yuv_bytes
.get() + kSourceUOffset
,
813 yuv_bytes
.get() + kSourceVOffset
,
814 rgb_bytes_converted
.get(),
817 GetLookupTable(YV12
));
818 media::EmptyRegisterState();
819 EXPECT_EQ(0, memcmp(rgb_bytes_reference
.get(),
820 rgb_bytes_converted
.get(),
824 TEST(YUVConvertTest
, LinearScaleYUVToRGB32Row_SSE
) {
826 if (!cpu
.has_sse()) {
827 LOG(WARNING
) << "System not supported. Test skipped.";
831 scoped_ptr
<uint8
[]> yuv_bytes(new uint8
[kYUV12Size
]);
832 scoped_ptr
<uint8
[]> rgb_bytes_reference(new uint8
[kRGBSize
]);
833 scoped_ptr
<uint8
[]> rgb_bytes_converted(new uint8
[kRGBSize
]);
834 ReadYV12Data(&yuv_bytes
);
836 const int kWidth
= 167;
837 const int kSourceDx
= 80000; // This value means a scale down.
838 LinearScaleYUVToRGB32Row_C(yuv_bytes
.get(),
839 yuv_bytes
.get() + kSourceUOffset
,
840 yuv_bytes
.get() + kSourceVOffset
,
841 rgb_bytes_reference
.get(),
844 GetLookupTable(YV12
));
845 LinearScaleYUVToRGB32Row_SSE(yuv_bytes
.get(),
846 yuv_bytes
.get() + kSourceUOffset
,
847 yuv_bytes
.get() + kSourceVOffset
,
848 rgb_bytes_converted
.get(),
851 GetLookupTable(YV12
));
852 media::EmptyRegisterState();
853 EXPECT_EQ(0, memcmp(rgb_bytes_reference
.get(),
854 rgb_bytes_converted
.get(),
858 TEST(YUVConvertTest
, FilterYUVRows_C_OutOfBounds
) {
859 scoped_ptr
<uint8
[]> src(new uint8
[16]);
860 scoped_ptr
<uint8
[]> dst(new uint8
[16]);
862 memset(src
.get(), 0xff, 16);
863 memset(dst
.get(), 0, 16);
865 media::FilterYUVRows_C(dst
.get(), src
.get(), src
.get(), 1, 255);
867 EXPECT_EQ(255u, dst
[0]);
868 for (int i
= 1; i
< 16; ++i
) {
869 EXPECT_EQ(0u, dst
[i
]) << " not equal at " << i
;
873 #if defined(MEDIA_MMX_INTRINSICS_AVAILABLE)
874 TEST(YUVConvertTest
, FilterYUVRows_MMX_OutOfBounds
) {
876 if (!cpu
.has_mmx()) {
877 LOG(WARNING
) << "System not supported. Test skipped.";
881 scoped_ptr
<uint8
[]> src(new uint8
[16]);
882 scoped_ptr
<uint8
[]> dst(new uint8
[16]);
884 memset(src
.get(), 0xff, 16);
885 memset(dst
.get(), 0, 16);
887 media::FilterYUVRows_MMX(dst
.get(), src
.get(), src
.get(), 1, 255);
888 media::EmptyRegisterState();
890 EXPECT_EQ(255u, dst
[0]);
891 for (int i
= 1; i
< 16; ++i
) {
892 EXPECT_EQ(0u, dst
[i
]);
895 #endif // defined(MEDIA_MMX_INTRINSICS_AVAILABLE)
897 TEST(YUVConvertTest
, FilterYUVRows_SSE2_OutOfBounds
) {
899 if (!cpu
.has_sse2()) {
900 LOG(WARNING
) << "System not supported. Test skipped.";
904 scoped_ptr
<uint8
[]> src(new uint8
[16]);
905 scoped_ptr
<uint8
[]> dst(new uint8
[16]);
907 memset(src
.get(), 0xff, 16);
908 memset(dst
.get(), 0, 16);
910 media::FilterYUVRows_SSE2(dst
.get(), src
.get(), src
.get(), 1, 255);
912 EXPECT_EQ(255u, dst
[0]);
913 for (int i
= 1; i
< 16; ++i
) {
914 EXPECT_EQ(0u, dst
[i
]);
918 #if defined(MEDIA_MMX_INTRINSICS_AVAILABLE)
919 TEST(YUVConvertTest
, FilterYUVRows_MMX_UnalignedDestination
) {
921 if (!cpu
.has_mmx()) {
922 LOG(WARNING
) << "System not supported. Test skipped.";
926 const int kSize
= 32;
927 scoped_ptr
<uint8
[]> src(new uint8
[kSize
]);
928 scoped_ptr
<uint8
[]> dst_sample(new uint8
[kSize
]);
929 scoped_ptr
<uint8
[]> dst(new uint8
[kSize
]);
931 memset(dst_sample
.get(), 0, kSize
);
932 memset(dst
.get(), 0, kSize
);
933 for (int i
= 0; i
< kSize
; ++i
)
936 media::FilterYUVRows_C(dst_sample
.get(),
937 src
.get(), src
.get(), 17, 128);
939 // Generate an unaligned output address.
941 reinterpret_cast<uint8
*>(
942 (reinterpret_cast<uintptr_t>(dst
.get() + 8) & ~7) + 1);
943 media::FilterYUVRows_MMX(dst_ptr
, src
.get(), src
.get(), 17, 128);
944 media::EmptyRegisterState();
946 EXPECT_EQ(0, memcmp(dst_sample
.get(), dst_ptr
, 17));
948 #endif // defined(MEDIA_MMX_INTRINSICS_AVAILABLE)
950 TEST(YUVConvertTest
, FilterYUVRows_SSE2_UnalignedDestination
) {
952 if (!cpu
.has_sse2()) {
953 LOG(WARNING
) << "System not supported. Test skipped.";
957 const int kSize
= 64;
958 scoped_ptr
<uint8
[]> src(new uint8
[kSize
]);
959 scoped_ptr
<uint8
[]> dst_sample(new uint8
[kSize
]);
960 scoped_ptr
<uint8
[]> dst(new uint8
[kSize
]);
962 memset(dst_sample
.get(), 0, kSize
);
963 memset(dst
.get(), 0, kSize
);
964 for (int i
= 0; i
< kSize
; ++i
)
967 media::FilterYUVRows_C(dst_sample
.get(),
968 src
.get(), src
.get(), 37, 128);
970 // Generate an unaligned output address.
972 reinterpret_cast<uint8
*>(
973 (reinterpret_cast<uintptr_t>(dst
.get() + 16) & ~15) + 1);
974 media::FilterYUVRows_SSE2(dst_ptr
, src
.get(), src
.get(), 37, 128);
975 media::EmptyRegisterState();
977 EXPECT_EQ(0, memcmp(dst_sample
.get(), dst_ptr
, 37));
980 #if defined(ARCH_CPU_X86_64)
982 TEST(YUVConvertTest
, ScaleYUVToRGB32Row_SSE2_X64
) {
983 scoped_ptr
<uint8
[]> yuv_bytes(new uint8
[kYUV12Size
]);
984 scoped_ptr
<uint8
[]> rgb_bytes_reference(new uint8
[kRGBSize
]);
985 scoped_ptr
<uint8
[]> rgb_bytes_converted(new uint8
[kRGBSize
]);
986 ReadYV12Data(&yuv_bytes
);
988 const int kWidth
= 167;
989 const int kSourceDx
= 80000; // This value means a scale down.
990 ScaleYUVToRGB32Row_C(yuv_bytes
.get(),
991 yuv_bytes
.get() + kSourceUOffset
,
992 yuv_bytes
.get() + kSourceVOffset
,
993 rgb_bytes_reference
.get(),
996 GetLookupTable(YV12
));
997 ScaleYUVToRGB32Row_SSE2_X64(yuv_bytes
.get(),
998 yuv_bytes
.get() + kSourceUOffset
,
999 yuv_bytes
.get() + kSourceVOffset
,
1000 rgb_bytes_converted
.get(),
1003 GetLookupTable(YV12
));
1004 media::EmptyRegisterState();
1005 EXPECT_EQ(0, memcmp(rgb_bytes_reference
.get(),
1006 rgb_bytes_converted
.get(),
1010 TEST(YUVConvertTest
, LinearScaleYUVToRGB32Row_MMX_X64
) {
1011 scoped_ptr
<uint8
[]> yuv_bytes(new uint8
[kYUV12Size
]);
1012 scoped_ptr
<uint8
[]> rgb_bytes_reference(new uint8
[kRGBSize
]);
1013 scoped_ptr
<uint8
[]> rgb_bytes_converted(new uint8
[kRGBSize
]);
1014 ReadYV12Data(&yuv_bytes
);
1016 const int kWidth
= 167;
1017 const int kSourceDx
= 80000; // This value means a scale down.
1018 LinearScaleYUVToRGB32Row_C(yuv_bytes
.get(),
1019 yuv_bytes
.get() + kSourceUOffset
,
1020 yuv_bytes
.get() + kSourceVOffset
,
1021 rgb_bytes_reference
.get(),
1024 GetLookupTable(YV12
));
1025 LinearScaleYUVToRGB32Row_MMX_X64(yuv_bytes
.get(),
1026 yuv_bytes
.get() + kSourceUOffset
,
1027 yuv_bytes
.get() + kSourceVOffset
,
1028 rgb_bytes_converted
.get(),
1031 GetLookupTable(YV12
));
1032 media::EmptyRegisterState();
1033 EXPECT_EQ(0, memcmp(rgb_bytes_reference
.get(),
1034 rgb_bytes_converted
.get(),
1038 #endif // defined(ARCH_CPU_X86_64)
1040 #endif // defined(ARCH_CPU_X86_FAMILY)
1042 } // namespace media