Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 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 "media/base/video_frame.h" | 5 #include "media/base/video_frame.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/callback_helpers.h" | 8 #include "base/callback_helpers.h" |
| 9 #include "base/format_macros.h" | 9 #include "base/format_macros.h" |
| 10 #include "base/memory/aligned_memory.h" | |
| 10 #include "base/memory/scoped_ptr.h" | 11 #include "base/memory/scoped_ptr.h" |
| 11 #include "base/strings/stringprintf.h" | 12 #include "base/strings/stringprintf.h" |
| 12 #include "media/base/buffers.h" | 13 #include "media/base/buffers.h" |
| 13 #include "media/base/yuv_convert.h" | 14 #include "media/base/yuv_convert.h" |
| 14 #include "testing/gtest/include/gtest/gtest.h" | 15 #include "testing/gtest/include/gtest/gtest.h" |
| 15 | 16 |
| 16 namespace media { | 17 namespace media { |
| 17 | 18 |
| 18 using base::MD5DigestToBase16; | 19 using base::MD5DigestToBase16; |
| 19 | 20 |
| 21 static inline size_t RoundUp(size_t value, size_t alignment) { | |
|
rileya (GONE FROM CHROMIUM)
2013/11/04 19:25:53
This was copied from video_frame.cc... in the inte
scherkus (not reviewing)
2013/11/04 22:54:34
either:
1) dump this in media/base/video_util.h
rileya (GONE FROM CHROMIUM)
2013/11/04 23:51:41
Alright, I did 2).
| |
| 22 // Check that |alignment| is a power of 2. | |
| 23 DCHECK((alignment + (alignment - 1)) == (alignment | (alignment - 1))); | |
| 24 return ((value + (alignment - 1)) & ~(alignment-1)); | |
| 25 } | |
| 26 | |
| 20 // Helper function that initializes a YV12 frame with white and black scan | 27 // Helper function that initializes a YV12 frame with white and black scan |
| 21 // lines based on the |white_to_black| parameter. If 0, then the entire | 28 // lines based on the |white_to_black| parameter. If 0, then the entire |
| 22 // frame will be black, if 1 then the entire frame will be white. | 29 // frame will be black, if 1 then the entire frame will be white. |
| 23 void InitializeYV12Frame(VideoFrame* frame, double white_to_black) { | 30 void InitializeYV12Frame(VideoFrame* frame, double white_to_black) { |
| 24 EXPECT_EQ(VideoFrame::YV12, frame->format()); | 31 EXPECT_EQ(VideoFrame::YV12, frame->format()); |
| 25 int first_black_row = static_cast<int>(frame->coded_size().height() * | 32 int first_black_row = static_cast<int>(frame->coded_size().height() * |
| 26 white_to_black); | 33 white_to_black); |
| 27 uint8* y_plane = frame->data(VideoFrame::kYPlane); | 34 uint8* y_plane = frame->data(VideoFrame::kYPlane); |
| 28 for (int row = 0; row < frame->coded_size().height(); ++row) { | 35 for (int row = 0; row < frame->coded_size().height(); ++row) { |
| 29 int color = (row < first_black_row) ? 0xFF : 0x00; | 36 int color = (row < first_black_row) ? 0xFF : 0x00; |
| (...skipping 10 matching lines...) Expand all Loading... | |
| 40 } | 47 } |
| 41 } | 48 } |
| 42 | 49 |
| 43 // Given a |yv12_frame| this method converts the YV12 frame to RGBA and | 50 // Given a |yv12_frame| this method converts the YV12 frame to RGBA and |
| 44 // makes sure that all the pixels of the RBG frame equal |expect_rgb_color|. | 51 // makes sure that all the pixels of the RBG frame equal |expect_rgb_color|. |
| 45 void ExpectFrameColor(media::VideoFrame* yv12_frame, uint32 expect_rgb_color) { | 52 void ExpectFrameColor(media::VideoFrame* yv12_frame, uint32 expect_rgb_color) { |
| 46 ASSERT_EQ(VideoFrame::YV12, yv12_frame->format()); | 53 ASSERT_EQ(VideoFrame::YV12, yv12_frame->format()); |
| 47 ASSERT_EQ(yv12_frame->stride(VideoFrame::kUPlane), | 54 ASSERT_EQ(yv12_frame->stride(VideoFrame::kUPlane), |
| 48 yv12_frame->stride(VideoFrame::kVPlane)); | 55 yv12_frame->stride(VideoFrame::kVPlane)); |
| 49 | 56 |
| 50 scoped_refptr<media::VideoFrame> rgb_frame; | 57 size_t bytes_per_row = RoundUp(yv12_frame->coded_size().width(), |
| 51 rgb_frame = media::VideoFrame::CreateFrame(VideoFrame::RGB32, | 58 VideoFrame::kFrameSizeAlignment) * 4u; |
| 52 yv12_frame->coded_size(), | 59 size_t aligned_height = RoundUp(yv12_frame->coded_size().height(), |
| 53 yv12_frame->visible_rect(), | 60 VideoFrame::kFrameSizeAlignment); |
| 54 yv12_frame->natural_size(), | 61 uint8* data = reinterpret_cast<uint8*>(base::AlignedAlloc( |
|
scherkus (not reviewing)
2013/11/04 22:54:34
s/data/rgb_data/
rileya (GONE FROM CHROMIUM)
2013/11/04 23:51:41
Done.
| |
| 55 yv12_frame->GetTimestamp()); | 62 bytes_per_row * aligned_height + VideoFrame::kFrameSizePadding, |
| 56 | 63 VideoFrame::kFrameAddressAlignment)); |
| 57 ASSERT_EQ(yv12_frame->coded_size().width(), | |
| 58 rgb_frame->coded_size().width()); | |
| 59 ASSERT_EQ(yv12_frame->coded_size().height(), | |
| 60 rgb_frame->coded_size().height()); | |
| 61 | 64 |
| 62 media::ConvertYUVToRGB32(yv12_frame->data(VideoFrame::kYPlane), | 65 media::ConvertYUVToRGB32(yv12_frame->data(VideoFrame::kYPlane), |
| 63 yv12_frame->data(VideoFrame::kUPlane), | 66 yv12_frame->data(VideoFrame::kUPlane), |
| 64 yv12_frame->data(VideoFrame::kVPlane), | 67 yv12_frame->data(VideoFrame::kVPlane), |
| 65 rgb_frame->data(VideoFrame::kRGBPlane), | 68 data, |
| 66 rgb_frame->coded_size().width(), | 69 yv12_frame->coded_size().width(), |
| 67 rgb_frame->coded_size().height(), | 70 yv12_frame->coded_size().height(), |
| 68 yv12_frame->stride(VideoFrame::kYPlane), | 71 yv12_frame->stride(VideoFrame::kYPlane), |
| 69 yv12_frame->stride(VideoFrame::kUPlane), | 72 yv12_frame->stride(VideoFrame::kUPlane), |
| 70 rgb_frame->stride(VideoFrame::kRGBPlane), | 73 bytes_per_row, |
| 71 media::YV12); | 74 media::YV12); |
| 72 | 75 |
| 73 for (int row = 0; row < rgb_frame->coded_size().height(); ++row) { | 76 for (int row = 0; row < yv12_frame->coded_size().height(); ++row) { |
| 74 uint32* rgb_row_data = reinterpret_cast<uint32*>( | 77 uint32* rgb_row_data = reinterpret_cast<uint32*>( |
| 75 rgb_frame->data(VideoFrame::kRGBPlane) + | 78 data + (bytes_per_row * row)); |
| 76 (rgb_frame->stride(VideoFrame::kRGBPlane) * row)); | 79 for (int col = 0; col < yv12_frame->coded_size().width(); ++col) { |
| 77 for (int col = 0; col < rgb_frame->coded_size().width(); ++col) { | |
| 78 SCOPED_TRACE( | 80 SCOPED_TRACE( |
| 79 base::StringPrintf("Checking (%d, %d)", row, col)); | 81 base::StringPrintf("Checking (%d, %d)", row, col)); |
| 80 EXPECT_EQ(expect_rgb_color, rgb_row_data[col]); | 82 EXPECT_EQ(expect_rgb_color, rgb_row_data[col]); |
| 81 } | 83 } |
| 82 } | 84 } |
| 85 | |
| 86 base::AlignedFree(data); | |
| 83 } | 87 } |
| 84 | 88 |
| 85 // Fill each plane to its reported extents and verify accessors report non | 89 // Fill each plane to its reported extents and verify accessors report non |
| 86 // zero values. Additionally, for the first plane verify the rows and | 90 // zero values. Additionally, for the first plane verify the rows and |
| 87 // row_bytes values are correct. | 91 // row_bytes values are correct. |
| 88 void ExpectFrameExtents(VideoFrame::Format format, int planes, | 92 void ExpectFrameExtents(VideoFrame::Format format, int planes, |
| 89 int bytes_per_pixel, const char* expected_hash) { | 93 int bytes_per_pixel, const char* expected_hash) { |
| 90 const unsigned char kFillByte = 0x80; | 94 const unsigned char kFillByte = 0x80; |
| 91 const int kWidth = 61; | 95 const int kWidth = 61; |
| 92 const int kHeight = 31; | 96 const int kHeight = 31; |
| (...skipping 104 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 197 } | 201 } |
| 198 } | 202 } |
| 199 | 203 |
| 200 // Ensure each frame is properly sized and allocated. Will trigger OOB reads | 204 // Ensure each frame is properly sized and allocated. Will trigger OOB reads |
| 201 // and writes as well as incorrect frame hashes otherwise. | 205 // and writes as well as incorrect frame hashes otherwise. |
| 202 TEST(VideoFrame, CheckFrameExtents) { | 206 TEST(VideoFrame, CheckFrameExtents) { |
| 203 // Each call consists of a VideoFrame::Format, # of planes, bytes per pixel, | 207 // Each call consists of a VideoFrame::Format, # of planes, bytes per pixel, |
| 204 // and the expected hash of all planes if filled with kFillByte (defined in | 208 // and the expected hash of all planes if filled with kFillByte (defined in |
| 205 // ExpectFrameExtents). | 209 // ExpectFrameExtents). |
| 206 ExpectFrameExtents( | 210 ExpectFrameExtents( |
| 207 VideoFrame::RGB32, 1, 4, "de6d3d567e282f6a38d478f04fc81fb0"); | |
| 208 ExpectFrameExtents( | |
| 209 VideoFrame::YV12, 3, 1, "71113bdfd4c0de6cf62f48fb74f7a0b1"); | 211 VideoFrame::YV12, 3, 1, "71113bdfd4c0de6cf62f48fb74f7a0b1"); |
| 210 ExpectFrameExtents( | 212 ExpectFrameExtents( |
| 211 VideoFrame::YV16, 3, 1, "9bb99ac3ff350644ebff4d28dc01b461"); | 213 VideoFrame::YV16, 3, 1, "9bb99ac3ff350644ebff4d28dc01b461"); |
| 212 } | 214 } |
| 213 | 215 |
| 214 static void TextureCallback(uint32* called_sync_point, uint32 sync_point) { | 216 static void TextureCallback(uint32* called_sync_point, uint32 sync_point) { |
| 215 *called_sync_point = sync_point; | 217 *called_sync_point = sync_point; |
| 216 } | 218 } |
| 217 | 219 |
| 218 // Verify the TextureNoLongerNeededCallback is called when VideoFrame is | 220 // Verify the TextureNoLongerNeededCallback is called when VideoFrame is |
| (...skipping 197 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 416 | 418 |
| 417 EXPECT_EQ(0u, called_sync_point); | 419 EXPECT_EQ(0u, called_sync_point); |
| 418 | 420 |
| 419 // Don't use the mailbox at all and drop our ref on it. | 421 // Don't use the mailbox at all and drop our ref on it. |
| 420 } | 422 } |
| 421 // The VideoFrame is destroyed, it should call the callback. | 423 // The VideoFrame is destroyed, it should call the callback. |
| 422 EXPECT_EQ(sync_point, called_sync_point); | 424 EXPECT_EQ(sync_point, called_sync_point); |
| 423 } | 425 } |
| 424 | 426 |
| 425 } // namespace media | 427 } // namespace media |
| OLD | NEW |