Index: webrtc/video/video_capture_input_unittest.cc |
diff --git a/webrtc/video/video_capture_input_unittest.cc b/webrtc/video/video_capture_input_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..2da722b47d4db297b4414084a005a6d07ec918a3 |
--- /dev/null |
+++ b/webrtc/video/video_capture_input_unittest.cc |
@@ -0,0 +1,255 @@ |
+/* |
+ * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. |
+ * |
+ * Use of this source code is governed by a BSD-style license |
+ * that can be found in the LICENSE file in the root of the source |
+ * tree. An additional intellectual property rights grant can be found |
+ * in the file PATENTS. All contributing project authors may |
+ * be found in the AUTHORS file in the root of the source tree. |
+ */ |
+#include "webrtc/video/video_capture_input.h" |
+ |
+#include <memory> |
+#include <vector> |
+ |
+#include "testing/gtest/include/gtest/gtest.h" |
+#include "webrtc/base/event.h" |
+#include "webrtc/base/refcount.h" |
+#include "webrtc/test/fake_texture_frame.h" |
+#include "webrtc/test/frame_utils.h" |
+#include "webrtc/video/send_statistics_proxy.h" |
+ |
+// If an output frame does not arrive in 500ms, the test will fail. |
+#define FRAME_TIMEOUT_MS 500 |
+ |
+namespace webrtc { |
+ |
+bool EqualFramesVector(const std::vector<std::unique_ptr<VideoFrame>>& frames1, |
+ const std::vector<std::unique_ptr<VideoFrame>>& frames2); |
+std::unique_ptr<VideoFrame> CreateVideoFrame(uint8_t length); |
+ |
+class VideoCaptureInputTest : public ::testing::Test { |
+ protected: |
+ VideoCaptureInputTest() |
+ : stats_proxy_(Clock::GetRealTimeClock(), |
+ webrtc::VideoSendStream::Config(nullptr), |
+ webrtc::VideoEncoderConfig::ContentType::kRealtimeVideo), |
+ capture_event_(false, false) {} |
+ |
+ virtual void SetUp() { |
+ overuse_detector_.reset( |
+ new OveruseFrameDetector(Clock::GetRealTimeClock(), CpuOveruseOptions(), |
+ nullptr, nullptr, &stats_proxy_)); |
+ input_.reset(new internal::VideoCaptureInput( |
+ &capture_event_, nullptr, &stats_proxy_, overuse_detector_.get())); |
+ } |
+ |
+ void AddInputFrame(VideoFrame* frame) { |
+ input_->IncomingCapturedFrame(*frame); |
+ } |
+ |
+ void WaitOutputFrame() { |
+ EXPECT_TRUE(capture_event_.Wait(FRAME_TIMEOUT_MS)); |
+ VideoFrame frame; |
+ EXPECT_TRUE(input_->GetVideoFrame(&frame)); |
+ ASSERT_TRUE(frame.video_frame_buffer()); |
+ if (!frame.video_frame_buffer()->native_handle()) { |
+ output_frame_ybuffers_.push_back(frame.video_frame_buffer()->DataY()); |
+ } |
+ output_frames_.push_back( |
+ std::unique_ptr<VideoFrame>(new VideoFrame(frame))); |
+ } |
+ |
+ SendStatisticsProxy stats_proxy_; |
+ |
+ rtc::Event capture_event_; |
+ |
+ std::unique_ptr<OveruseFrameDetector> overuse_detector_; |
+ |
+ // Used to send input capture frames to VideoCaptureInput. |
+ std::unique_ptr<internal::VideoCaptureInput> input_; |
+ |
+ // Input capture frames of VideoCaptureInput. |
+ std::vector<std::unique_ptr<VideoFrame>> input_frames_; |
+ |
+ // Output delivered frames of VideoCaptureInput. |
+ std::vector<std::unique_ptr<VideoFrame>> output_frames_; |
+ |
+ // The pointers of Y plane buffers of output frames. This is used to verify |
+ // the frame are swapped and not copied. |
+ std::vector<const uint8_t*> output_frame_ybuffers_; |
+}; |
+ |
+TEST_F(VideoCaptureInputTest, DoesNotRetainHandleNorCopyBuffer) { |
+ // Indicate an output frame has arrived. |
+ rtc::Event frame_destroyed_event(false, false); |
+ class TestBuffer : public webrtc::I420Buffer { |
+ public: |
+ explicit TestBuffer(rtc::Event* event) : I420Buffer(5, 5), event_(event) {} |
+ |
+ private: |
+ friend class rtc::RefCountedObject<TestBuffer>; |
+ ~TestBuffer() override { event_->Set(); } |
+ rtc::Event* const event_; |
+ }; |
+ |
+ { |
+ VideoFrame frame( |
+ new rtc::RefCountedObject<TestBuffer>(&frame_destroyed_event), 1, 1, |
+ kVideoRotation_0); |
+ |
+ AddInputFrame(&frame); |
+ WaitOutputFrame(); |
+ |
+ EXPECT_EQ(output_frames_[0]->video_frame_buffer().get(), |
+ frame.video_frame_buffer().get()); |
+ output_frames_.clear(); |
+ } |
+ EXPECT_TRUE(frame_destroyed_event.Wait(FRAME_TIMEOUT_MS)); |
+} |
+ |
+TEST_F(VideoCaptureInputTest, TestNtpTimeStampSetIfRenderTimeSet) { |
+ input_frames_.push_back(CreateVideoFrame(0)); |
+ input_frames_[0]->set_render_time_ms(5); |
+ input_frames_[0]->set_ntp_time_ms(0); |
+ |
+ AddInputFrame(input_frames_[0].get()); |
+ WaitOutputFrame(); |
+ EXPECT_GT(output_frames_[0]->ntp_time_ms(), |
+ input_frames_[0]->render_time_ms()); |
+} |
+ |
+TEST_F(VideoCaptureInputTest, TestRtpTimeStampSet) { |
+ input_frames_.push_back(CreateVideoFrame(0)); |
+ input_frames_[0]->set_render_time_ms(0); |
+ input_frames_[0]->set_ntp_time_ms(1); |
+ input_frames_[0]->set_timestamp(0); |
+ |
+ AddInputFrame(input_frames_[0].get()); |
+ WaitOutputFrame(); |
+ EXPECT_EQ(output_frames_[0]->timestamp(), |
+ input_frames_[0]->ntp_time_ms() * 90); |
+} |
+ |
+TEST_F(VideoCaptureInputTest, DropsFramesWithSameOrOldNtpTimestamp) { |
+ input_frames_.push_back(CreateVideoFrame(0)); |
+ |
+ input_frames_[0]->set_ntp_time_ms(17); |
+ AddInputFrame(input_frames_[0].get()); |
+ WaitOutputFrame(); |
+ EXPECT_EQ(output_frames_[0]->timestamp(), |
+ input_frames_[0]->ntp_time_ms() * 90); |
+ |
+ // Repeat frame with the same NTP timestamp should drop. |
+ AddInputFrame(input_frames_[0].get()); |
+ EXPECT_FALSE(capture_event_.Wait(FRAME_TIMEOUT_MS)); |
+ |
+ // As should frames with a decreased NTP timestamp. |
+ input_frames_[0]->set_ntp_time_ms(input_frames_[0]->ntp_time_ms() - 1); |
+ AddInputFrame(input_frames_[0].get()); |
+ EXPECT_FALSE(capture_event_.Wait(FRAME_TIMEOUT_MS)); |
+ |
+ // But delivering with an increased NTP timestamp should succeed. |
+ input_frames_[0]->set_ntp_time_ms(4711); |
+ AddInputFrame(input_frames_[0].get()); |
+ WaitOutputFrame(); |
+ EXPECT_EQ(output_frames_[1]->timestamp(), |
+ input_frames_[0]->ntp_time_ms() * 90); |
+} |
+ |
+TEST_F(VideoCaptureInputTest, TestTextureFrames) { |
+ const int kNumFrame = 3; |
+ for (int i = 0 ; i < kNumFrame; ++i) { |
+ test::FakeNativeHandle* dummy_handle = new test::FakeNativeHandle(); |
+ // Add one to |i| so that width/height > 0. |
+ input_frames_.push_back(std::unique_ptr<VideoFrame>(new VideoFrame( |
+ test::FakeNativeHandle::CreateFrame(dummy_handle, i + 1, i + 1, i + 1, |
+ i + 1, webrtc::kVideoRotation_0)))); |
+ AddInputFrame(input_frames_[i].get()); |
+ WaitOutputFrame(); |
+ ASSERT_TRUE(output_frames_[i]->video_frame_buffer()); |
+ EXPECT_EQ(dummy_handle, |
+ output_frames_[i]->video_frame_buffer()->native_handle()); |
+ } |
+ |
+ EXPECT_TRUE(EqualFramesVector(input_frames_, output_frames_)); |
+} |
+ |
+TEST_F(VideoCaptureInputTest, TestI420Frames) { |
+ const int kNumFrame = 4; |
+ std::vector<const uint8_t*> ybuffer_pointers; |
+ for (int i = 0; i < kNumFrame; ++i) { |
+ input_frames_.push_back(CreateVideoFrame(static_cast<uint8_t>(i + 1))); |
+ ybuffer_pointers.push_back(input_frames_[i]->video_frame_buffer()->DataY()); |
+ AddInputFrame(input_frames_[i].get()); |
+ WaitOutputFrame(); |
+ } |
+ |
+ EXPECT_TRUE(EqualFramesVector(input_frames_, output_frames_)); |
+ // Make sure the buffer is not copied. |
+ for (int i = 0; i < kNumFrame; ++i) |
+ EXPECT_EQ(ybuffer_pointers[i], output_frame_ybuffers_[i]); |
+} |
+ |
+TEST_F(VideoCaptureInputTest, TestI420FrameAfterTextureFrame) { |
+ test::FakeNativeHandle* dummy_handle = new test::FakeNativeHandle(); |
+ input_frames_.push_back(std::unique_ptr<VideoFrame>( |
+ new VideoFrame(test::FakeNativeHandle::CreateFrame( |
+ dummy_handle, 1, 1, 1, 1, webrtc::kVideoRotation_0)))); |
+ AddInputFrame(input_frames_[0].get()); |
+ WaitOutputFrame(); |
+ ASSERT_TRUE(output_frames_[0]->video_frame_buffer()); |
+ EXPECT_EQ(dummy_handle, |
+ output_frames_[0]->video_frame_buffer()->native_handle()); |
+ |
+ input_frames_.push_back(CreateVideoFrame(2)); |
+ AddInputFrame(input_frames_[1].get()); |
+ WaitOutputFrame(); |
+ |
+ EXPECT_TRUE(EqualFramesVector(input_frames_, output_frames_)); |
+} |
+ |
+TEST_F(VideoCaptureInputTest, TestTextureFrameAfterI420Frame) { |
+ input_frames_.push_back(CreateVideoFrame(1)); |
+ AddInputFrame(input_frames_[0].get()); |
+ WaitOutputFrame(); |
+ |
+ test::FakeNativeHandle* dummy_handle = new test::FakeNativeHandle(); |
+ input_frames_.push_back(std::unique_ptr<VideoFrame>( |
+ new VideoFrame(test::FakeNativeHandle::CreateFrame( |
+ dummy_handle, 1, 1, 2, 2, webrtc::kVideoRotation_0)))); |
+ AddInputFrame(input_frames_[1].get()); |
+ WaitOutputFrame(); |
+ |
+ EXPECT_TRUE(EqualFramesVector(input_frames_, output_frames_)); |
+} |
+ |
+bool EqualFramesVector( |
+ const std::vector<std::unique_ptr<VideoFrame>>& frames1, |
+ const std::vector<std::unique_ptr<VideoFrame>>& frames2) { |
+ if (frames1.size() != frames2.size()) |
+ return false; |
+ for (size_t i = 0; i < frames1.size(); ++i) { |
+ // Compare frame buffers, since we don't care about differing timestamps. |
+ if (!test::FrameBufsEqual(frames1[i]->video_frame_buffer(), |
+ frames2[i]->video_frame_buffer())) { |
+ return false; |
+ } |
+ } |
+ return true; |
+} |
+ |
+std::unique_ptr<VideoFrame> CreateVideoFrame(uint8_t data) { |
+ std::unique_ptr<VideoFrame> frame(new VideoFrame()); |
+ const int width = 36; |
+ const int height = 24; |
+ const int kSizeY = width * height * 2; |
+ uint8_t buffer[kSizeY]; |
+ memset(buffer, data, kSizeY); |
+ frame->CreateFrame(buffer, buffer, buffer, width, height, width, width / 2, |
+ width / 2, kVideoRotation_0); |
+ frame->set_render_time_ms(data); |
+ return frame; |
+} |
+ |
+} // namespace webrtc |