| Index: webrtc/video/vie_encoder_unittest.cc
|
| diff --git a/webrtc/video/vie_encoder_unittest.cc b/webrtc/video/vie_encoder_unittest.cc
|
| deleted file mode 100644
|
| index 698dd3a54088b8873a2473ca69707adf1664c632..0000000000000000000000000000000000000000
|
| --- a/webrtc/video/vie_encoder_unittest.cc
|
| +++ /dev/null
|
| @@ -1,255 +0,0 @@
|
| -/*
|
| - * Copyright (c) 2016 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 "testing/gtest/include/gtest/gtest.h"
|
| -#include "webrtc/base/logging.h"
|
| -#include "webrtc/test/encoder_settings.h"
|
| -#include "webrtc/test/fake_encoder.h"
|
| -#include "webrtc/video/send_statistics_proxy.h"
|
| -#include "webrtc/video/vie_encoder.h"
|
| -
|
| -namespace webrtc {
|
| -
|
| -class ViEEncoderTest : public ::testing::Test {
|
| - public:
|
| - static const int kDefaultTimeoutMs = 30 * 1000;
|
| -
|
| - ViEEncoderTest()
|
| - : video_send_config_(VideoSendStream::Config(nullptr)),
|
| - fake_encoder_(),
|
| - stats_proxy_(Clock::GetRealTimeClock(),
|
| - video_send_config_,
|
| - webrtc::VideoEncoderConfig::ContentType::kRealtimeVideo),
|
| - sink_(&fake_encoder_) {}
|
| -
|
| - void SetUp() override {
|
| - video_send_config_ = VideoSendStream::Config(nullptr);
|
| - video_send_config_.encoder_settings.encoder = &fake_encoder_;
|
| - video_send_config_.encoder_settings.payload_name = "FAKE";
|
| - video_send_config_.encoder_settings.payload_type = 125;
|
| -
|
| - video_encoder_config_.streams = test::CreateVideoStreams(1);
|
| -
|
| - vie_encoder_.reset(new ViEEncoder(
|
| - 1 /* number_of_cores */, &stats_proxy_,
|
| - video_send_config_.encoder_settings, nullptr /* pre_encode_callback */,
|
| - nullptr /* overuse_callback */, nullptr /* encoder_timing */));
|
| - vie_encoder_->SetSink(&sink_);
|
| - vie_encoder_->SetStartBitrate(10000);
|
| - vie_encoder_->ConfigureEncoder(video_encoder_config_, 1440);
|
| - }
|
| -
|
| - VideoFrame CreateFrame(int64_t ntp_ts, rtc::Event* destruction_event) const {
|
| - class TestBuffer : public webrtc::I420Buffer {
|
| - public:
|
| - TestBuffer(rtc::Event* event, int width, int height)
|
| - : I420Buffer(width, height), event_(event) {}
|
| -
|
| - private:
|
| - friend class rtc::RefCountedObject<TestBuffer>;
|
| - ~TestBuffer() override {
|
| - if (event_)
|
| - event_->Set();
|
| - }
|
| - rtc::Event* const event_;
|
| - };
|
| -
|
| - VideoFrame frame(
|
| - new rtc::RefCountedObject<TestBuffer>(
|
| - destruction_event,
|
| - static_cast<int>(video_encoder_config_.streams[0].width),
|
| - static_cast<int>(video_encoder_config_.streams[0].height)),
|
| - 99, 99, kVideoRotation_0);
|
| - frame.set_ntp_time_ms(ntp_ts);
|
| - return frame;
|
| - }
|
| -
|
| - class TestEncoder : public test::FakeEncoder {
|
| - public:
|
| - TestEncoder()
|
| - : FakeEncoder(Clock::GetRealTimeClock()),
|
| - continue_encode_event_(false, false) {}
|
| -
|
| - int32_t Encode(const VideoFrame& input_image,
|
| - const CodecSpecificInfo* codec_specific_info,
|
| - const std::vector<FrameType>* frame_types) override {
|
| - bool block_encode;
|
| - {
|
| - rtc::CritScope lock(&crit_);
|
| - EXPECT_GT(input_image.timestamp(), timestamp_);
|
| - EXPECT_GT(input_image.ntp_time_ms(), ntp_time_ms_);
|
| - EXPECT_EQ(input_image.timestamp(), input_image.ntp_time_ms() * 90);
|
| -
|
| - timestamp_ = input_image.timestamp();
|
| - ntp_time_ms_ = input_image.ntp_time_ms();
|
| - block_encode = block_next_encode_;
|
| - block_next_encode_ = false;
|
| - }
|
| - int32_t result =
|
| - FakeEncoder::Encode(input_image, codec_specific_info, frame_types);
|
| - if (block_encode)
|
| - continue_encode_event_.Wait(kDefaultTimeoutMs);
|
| - return result;
|
| - }
|
| -
|
| - void BlockNextEncode() {
|
| - rtc::CritScope lock(&crit_);
|
| - block_next_encode_ = true;
|
| - }
|
| -
|
| - void ContinueEncode() { continue_encode_event_.Set(); }
|
| -
|
| - void CheckLastTimeStampsMatch(int64_t ntp_time_ms,
|
| - uint32_t timestamp) const {
|
| - rtc::CritScope lock(&crit_);
|
| - EXPECT_EQ(timestamp_, timestamp);
|
| - EXPECT_EQ(ntp_time_ms_, ntp_time_ms);
|
| - }
|
| -
|
| - private:
|
| - rtc::CriticalSection crit_;
|
| - bool block_next_encode_ = false;
|
| - rtc::Event continue_encode_event_;
|
| - uint32_t timestamp_ = 0;
|
| - int64_t ntp_time_ms_ = 0;
|
| - };
|
| -
|
| - class TestSink : public EncodedImageCallback {
|
| - public:
|
| - explicit TestSink(TestEncoder* test_encoder)
|
| - : test_encoder_(test_encoder), encoded_frame_event_(false, false) {}
|
| -
|
| - int32_t Encoded(const EncodedImage& encoded_image,
|
| - const CodecSpecificInfo* codec_specific_info,
|
| - const RTPFragmentationHeader* fragmentation) override {
|
| - rtc::CritScope lock(&crit_);
|
| - EXPECT_TRUE(expect_frames_);
|
| - timestamp_ = encoded_image._timeStamp;
|
| - encoded_frame_event_.Set();
|
| - return 0;
|
| - }
|
| -
|
| - void WaitForEncodedFrame(int64_t expected_ntp_time) {
|
| - uint32_t timestamp = 0;
|
| - encoded_frame_event_.Wait(kDefaultTimeoutMs);
|
| - {
|
| - rtc::CritScope lock(&crit_);
|
| - timestamp = timestamp_;
|
| - }
|
| - test_encoder_->CheckLastTimeStampsMatch(expected_ntp_time, timestamp);
|
| - }
|
| -
|
| - void SetExpectNoFrames() {
|
| - rtc::CritScope lock(&crit_);
|
| - expect_frames_ = false;
|
| - }
|
| -
|
| - private:
|
| - rtc::CriticalSection crit_;
|
| - TestEncoder* test_encoder_;
|
| - rtc::Event encoded_frame_event_;
|
| - uint32_t timestamp_ = 0;
|
| - bool expect_frames_ = true;
|
| - };
|
| -
|
| - VideoSendStream::Config video_send_config_;
|
| - VideoEncoderConfig video_encoder_config_;
|
| - TestEncoder fake_encoder_;
|
| - SendStatisticsProxy stats_proxy_;
|
| - TestSink sink_;
|
| - std::unique_ptr<ViEEncoder> vie_encoder_;
|
| -};
|
| -
|
| -TEST_F(ViEEncoderTest, EncodeOneFrame) {
|
| - const int kTargetBitrateBps = 100000;
|
| - vie_encoder_->OnBitrateUpdated(kTargetBitrateBps, 0, 0);
|
| - rtc::Event frame_destroyed_event(false, false);
|
| - vie_encoder_->IncomingCapturedFrame(CreateFrame(1, &frame_destroyed_event));
|
| - sink_.WaitForEncodedFrame(1);
|
| - frame_destroyed_event.Wait(kDefaultTimeoutMs);
|
| - vie_encoder_->Stop();
|
| -}
|
| -
|
| -TEST_F(ViEEncoderTest, DropsFramesBeforeFirstOnBitrateUpdated) {
|
| - // Dropped since no target bitrate has been set.
|
| - rtc::Event frame_destroyed_event(false, false);
|
| - vie_encoder_->IncomingCapturedFrame(CreateFrame(1, &frame_destroyed_event));
|
| - frame_destroyed_event.Wait(kDefaultTimeoutMs);
|
| -
|
| - const int kTargetBitrateBps = 100000;
|
| - vie_encoder_->OnBitrateUpdated(kTargetBitrateBps, 0, 0);
|
| -
|
| - vie_encoder_->IncomingCapturedFrame(CreateFrame(2, nullptr));
|
| - sink_.WaitForEncodedFrame(2);
|
| - vie_encoder_->Stop();
|
| -}
|
| -
|
| -TEST_F(ViEEncoderTest, DropsFramesWhenRateSetToZero) {
|
| - const int kTargetBitrateBps = 100000;
|
| - vie_encoder_->OnBitrateUpdated(kTargetBitrateBps, 0, 0);
|
| - vie_encoder_->IncomingCapturedFrame(CreateFrame(1, nullptr));
|
| - sink_.WaitForEncodedFrame(1);
|
| -
|
| - vie_encoder_->OnBitrateUpdated(0, 0, 0);
|
| - // Dropped since bitrate is zero.
|
| - vie_encoder_->IncomingCapturedFrame(CreateFrame(2, nullptr));
|
| -
|
| - vie_encoder_->OnBitrateUpdated(kTargetBitrateBps, 0, 0);
|
| - vie_encoder_->IncomingCapturedFrame(CreateFrame(3, nullptr));
|
| - sink_.WaitForEncodedFrame(3);
|
| - vie_encoder_->Stop();
|
| -}
|
| -
|
| -TEST_F(ViEEncoderTest, DropsFramesWithSameOrOldNtpTimestamp) {
|
| - const int kTargetBitrateBps = 100000;
|
| - vie_encoder_->OnBitrateUpdated(kTargetBitrateBps, 0, 0);
|
| - vie_encoder_->IncomingCapturedFrame(CreateFrame(1, nullptr));
|
| - sink_.WaitForEncodedFrame(1);
|
| -
|
| - // This frame will be dropped since it has the same ntp timestamp.
|
| - vie_encoder_->IncomingCapturedFrame(CreateFrame(1, nullptr));
|
| -
|
| - vie_encoder_->IncomingCapturedFrame(CreateFrame(2, nullptr));
|
| - sink_.WaitForEncodedFrame(2);
|
| - vie_encoder_->Stop();
|
| -}
|
| -
|
| -TEST_F(ViEEncoderTest, DropsFrameAfterStop) {
|
| - const int kTargetBitrateBps = 100000;
|
| - vie_encoder_->OnBitrateUpdated(kTargetBitrateBps, 0, 0);
|
| -
|
| - vie_encoder_->IncomingCapturedFrame(CreateFrame(1, nullptr));
|
| - sink_.WaitForEncodedFrame(1);
|
| -
|
| - vie_encoder_->Stop();
|
| - sink_.SetExpectNoFrames();
|
| - rtc::Event frame_destroyed_event(false, false);
|
| - vie_encoder_->IncomingCapturedFrame(CreateFrame(2, &frame_destroyed_event));
|
| - frame_destroyed_event.Wait(kDefaultTimeoutMs);
|
| -}
|
| -
|
| -TEST_F(ViEEncoderTest, DropsPendingFramesOnSlowEncode) {
|
| - const int kTargetBitrateBps = 100000;
|
| - vie_encoder_->OnBitrateUpdated(kTargetBitrateBps, 0, 0);
|
| -
|
| - fake_encoder_.BlockNextEncode();
|
| - vie_encoder_->IncomingCapturedFrame(CreateFrame(1, nullptr));
|
| - sink_.WaitForEncodedFrame(1);
|
| - // Here, the encoder thread will be blocked in the TestEncoder waiting for a
|
| - // call to ContinueEncode.
|
| - vie_encoder_->IncomingCapturedFrame(CreateFrame(2, nullptr));
|
| - vie_encoder_->IncomingCapturedFrame(CreateFrame(3, nullptr));
|
| - fake_encoder_.ContinueEncode();
|
| - sink_.WaitForEncodedFrame(3);
|
| -
|
| - vie_encoder_->Stop();
|
| -}
|
| -
|
| -} // namespace webrtc
|
|
|