blob: 35439d264e61c35fbc673a5421f3bff59f362c89 [file] [log] [blame]
perkj26091b12016-09-01 01:17:40 -07001/*
2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
Erik Språng4529fbc2018-10-12 10:30:31 +020011#include "video/video_stream_encoder.h"
12
sprangfe627f32017-03-29 08:24:59 -070013#include <algorithm>
perkj803d97f2016-11-01 11:45:46 -070014#include <limits>
Danil Chapovalovd3ba2362019-04-10 17:01:23 +020015#include <memory>
Per512ecb32016-09-23 15:52:06 +020016#include <utility>
17
Mirta Dvornicic28f0eb22019-05-28 16:30:16 +020018#include "absl/memory/memory.h"
Danil Chapovalovd3ba2362019-04-10 17:01:23 +020019#include "api/task_queue/default_task_queue_factory.h"
Elad Alon45befc52019-07-02 11:20:09 +020020#include "api/test/mock_fec_controller_override.h"
Jiawei Ouc2ebe212018-11-08 10:02:56 -080021#include "api/video/builtin_video_bitrate_allocator_factory.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020022#include "api/video/i420_buffer.h"
Erik Språngf93eda12019-01-16 17:10:57 +010023#include "api/video/video_bitrate_allocation.h"
Elad Alon370f93a2019-06-11 14:57:57 +020024#include "api/video_codecs/video_encoder.h"
Erik Språng4529fbc2018-10-12 10:30:31 +020025#include "api/video_codecs/vp8_temporal_layers.h"
Elad Aloncde8ab22019-03-20 11:56:20 +010026#include "api/video_codecs/vp8_temporal_layers_factory.h"
Mirta Dvornicic28f0eb22019-05-28 16:30:16 +020027#include "common_video/h264/h264_common.h"
Noah Richards51db4212019-06-12 06:59:12 -070028#include "common_video/include/video_frame_buffer.h"
Steve Anton10542f22019-01-11 09:11:00 -080029#include "media/base/video_adapter.h"
Sergey Silkin86684962018-03-28 19:32:37 +020030#include "modules/video_coding/codecs/vp9/include/vp9_globals.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020031#include "modules/video_coding/utility/default_video_bitrate_allocator.h"
Åsa Perssonc29cb2c2019-03-25 12:06:59 +010032#include "modules/video_coding/utility/simulcast_rate_allocator.h"
Steve Anton10542f22019-01-11 09:11:00 -080033#include "rtc_base/fake_clock.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020034#include "rtc_base/logging.h"
Steve Anton10542f22019-01-11 09:11:00 -080035#include "rtc_base/ref_counted_object.h"
Erik Språng7ca375c2019-02-06 16:20:17 +010036#include "system_wrappers/include/field_trial.h"
Mirko Bonadei17f48782018-09-28 08:51:10 +020037#include "system_wrappers/include/metrics.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020038#include "system_wrappers/include/sleep.h"
39#include "test/encoder_settings.h"
40#include "test/fake_encoder.h"
Kári Tristan Helgason639602a2018-08-02 10:51:40 +020041#include "test/field_trial.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020042#include "test/frame_generator.h"
43#include "test/gmock.h"
44#include "test/gtest.h"
Niels Möllercbcbc222018-09-28 09:07:24 +020045#include "test/video_encoder_proxy_factory.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020046#include "video/send_statistics_proxy.h"
perkj26091b12016-09-01 01:17:40 -070047
48namespace webrtc {
49
sprangb1ca0732017-02-01 08:38:12 -080050using ScaleReason = AdaptationObserverInterface::AdaptReason;
sprang57c2fff2017-01-16 06:24:02 -080051using ::testing::_;
kthelgason876222f2016-11-29 01:44:11 -080052
perkj803d97f2016-11-01 11:45:46 -070053namespace {
Åsa Persson8c1bf952018-09-13 10:42:19 +020054const int kMinPixelsPerFrame = 320 * 180;
55const int kMinFramerateFps = 2;
56const int kMinBalancedFramerateFps = 7;
57const int64_t kFrameTimeoutMs = 100;
asapersson5f7226f2016-11-25 04:37:00 -080058const size_t kMaxPayloadLength = 1440;
Erik Språngd7329ca2019-02-21 21:19:53 +010059const uint32_t kTargetBitrateBps = 1000000;
Sergey Silkin5ee69672019-07-02 14:18:34 +020060const uint32_t kStartBitrateBps = 600000;
Erik Språngd7329ca2019-02-21 21:19:53 +010061const uint32_t kSimulcastTargetBitrateBps = 3150000;
62const uint32_t kLowTargetBitrateBps = kTargetBitrateBps / 10;
kthelgason2bc68642017-02-07 07:02:22 -080063const int kMaxInitialFramedrop = 4;
sprangfda496a2017-06-15 04:21:07 -070064const int kDefaultFramerate = 30;
Åsa Persson8c1bf952018-09-13 10:42:19 +020065const int64_t kFrameIntervalMs = rtc::kNumMillisecsPerSec / kDefaultFramerate;
asapersson5f7226f2016-11-25 04:37:00 -080066
Mirta Dvornicic28f0eb22019-05-28 16:30:16 +020067uint8_t optimal_sps[] = {0, 0, 0, 1, H264::NaluType::kSps,
68 0x00, 0x00, 0x03, 0x03, 0xF4,
69 0x05, 0x03, 0xC7, 0xE0, 0x1B,
70 0x41, 0x10, 0x8D, 0x00};
71
perkj803d97f2016-11-01 11:45:46 -070072class TestBuffer : public webrtc::I420Buffer {
73 public:
74 TestBuffer(rtc::Event* event, int width, int height)
75 : I420Buffer(width, height), event_(event) {}
76
77 private:
78 friend class rtc::RefCountedObject<TestBuffer>;
79 ~TestBuffer() override {
80 if (event_)
81 event_->Set();
82 }
83 rtc::Event* const event_;
84};
85
Noah Richards51db4212019-06-12 06:59:12 -070086// A fake native buffer that can't be converted to I420.
87class FakeNativeBuffer : public webrtc::VideoFrameBuffer {
88 public:
89 FakeNativeBuffer(rtc::Event* event, int width, int height)
90 : event_(event), width_(width), height_(height) {}
91 webrtc::VideoFrameBuffer::Type type() const override { return Type::kNative; }
92 int width() const override { return width_; }
93 int height() const override { return height_; }
94 rtc::scoped_refptr<webrtc::I420BufferInterface> ToI420() override {
95 return nullptr;
96 }
97
98 private:
99 friend class rtc::RefCountedObject<FakeNativeBuffer>;
100 ~FakeNativeBuffer() override {
101 if (event_)
102 event_->Set();
103 }
104 rtc::Event* const event_;
105 const int width_;
106 const int height_;
107};
108
Niels Möller7dc26b72017-12-06 10:27:48 +0100109class CpuOveruseDetectorProxy : public OveruseFrameDetector {
110 public:
Niels Möllerd1f7eb62018-03-28 16:40:58 +0200111 explicit CpuOveruseDetectorProxy(CpuOveruseMetricsObserver* metrics_observer)
112 : OveruseFrameDetector(metrics_observer),
Niels Möller7dc26b72017-12-06 10:27:48 +0100113 last_target_framerate_fps_(-1) {}
114 virtual ~CpuOveruseDetectorProxy() {}
115
116 void OnTargetFramerateUpdated(int framerate_fps) override {
117 rtc::CritScope cs(&lock_);
118 last_target_framerate_fps_ = framerate_fps;
119 OveruseFrameDetector::OnTargetFramerateUpdated(framerate_fps);
120 }
121
122 int GetLastTargetFramerate() {
123 rtc::CritScope cs(&lock_);
124 return last_target_framerate_fps_;
125 }
126
Niels Möller4db138e2018-04-19 09:04:13 +0200127 CpuOveruseOptions GetOptions() { return options_; }
128
Niels Möller7dc26b72017-12-06 10:27:48 +0100129 private:
130 rtc::CriticalSection lock_;
131 int last_target_framerate_fps_ RTC_GUARDED_BY(lock_);
132};
133
mflodmancc3d4422017-08-03 08:27:51 -0700134class VideoStreamEncoderUnderTest : public VideoStreamEncoder {
perkj803d97f2016-11-01 11:45:46 -0700135 public:
Niels Möller213618e2018-07-24 09:29:58 +0200136 VideoStreamEncoderUnderTest(SendStatisticsProxy* stats_proxy,
Danil Chapovalovd3ba2362019-04-10 17:01:23 +0200137 const VideoStreamEncoderSettings& settings,
138 TaskQueueFactory* task_queue_factory)
Sebastian Jansson572c60f2019-03-04 18:30:41 +0100139 : VideoStreamEncoder(Clock::GetRealTimeClock(),
140 1 /* number_of_cores */,
Yves Gerey665174f2018-06-19 15:03:05 +0200141 stats_proxy,
142 settings,
Yves Gerey665174f2018-06-19 15:03:05 +0200143 std::unique_ptr<OveruseFrameDetector>(
144 overuse_detector_proxy_ =
Sebastian Jansson74682c12019-03-01 11:50:20 +0100145 new CpuOveruseDetectorProxy(stats_proxy)),
Danil Chapovalovd3ba2362019-04-10 17:01:23 +0200146 task_queue_factory) {}
perkj803d97f2016-11-01 11:45:46 -0700147
sprangb1ca0732017-02-01 08:38:12 -0800148 void PostTaskAndWait(bool down, AdaptReason reason) {
Niels Möllerc572ff32018-11-07 08:43:50 +0100149 rtc::Event event;
kthelgason876222f2016-11-29 01:44:11 -0800150 encoder_queue()->PostTask([this, &event, reason, down] {
sprangb1ca0732017-02-01 08:38:12 -0800151 down ? AdaptDown(reason) : AdaptUp(reason);
perkj803d97f2016-11-01 11:45:46 -0700152 event.Set();
153 });
perkj070ba852017-02-16 15:46:27 -0800154 ASSERT_TRUE(event.Wait(5000));
perkj803d97f2016-11-01 11:45:46 -0700155 }
156
kthelgason2fc52542017-03-03 00:24:41 -0800157 // This is used as a synchronisation mechanism, to make sure that the
158 // encoder queue is not blocked before we start sending it frames.
159 void WaitUntilTaskQueueIsIdle() {
Niels Möllerc572ff32018-11-07 08:43:50 +0100160 rtc::Event event;
Yves Gerey665174f2018-06-19 15:03:05 +0200161 encoder_queue()->PostTask([&event] { event.Set(); });
kthelgason2fc52542017-03-03 00:24:41 -0800162 ASSERT_TRUE(event.Wait(5000));
163 }
164
sprangb1ca0732017-02-01 08:38:12 -0800165 void TriggerCpuOveruse() { PostTaskAndWait(true, AdaptReason::kCpu); }
kthelgason876222f2016-11-29 01:44:11 -0800166
sprangb1ca0732017-02-01 08:38:12 -0800167 void TriggerCpuNormalUsage() { PostTaskAndWait(false, AdaptReason::kCpu); }
kthelgason876222f2016-11-29 01:44:11 -0800168
sprangb1ca0732017-02-01 08:38:12 -0800169 void TriggerQualityLow() { PostTaskAndWait(true, AdaptReason::kQuality); }
kthelgason876222f2016-11-29 01:44:11 -0800170
sprangb1ca0732017-02-01 08:38:12 -0800171 void TriggerQualityHigh() { PostTaskAndWait(false, AdaptReason::kQuality); }
sprangfda496a2017-06-15 04:21:07 -0700172
Niels Möller7dc26b72017-12-06 10:27:48 +0100173 CpuOveruseDetectorProxy* overuse_detector_proxy_;
perkj803d97f2016-11-01 11:45:46 -0700174};
175
asapersson5f7226f2016-11-25 04:37:00 -0800176class VideoStreamFactory
177 : public VideoEncoderConfig::VideoStreamFactoryInterface {
178 public:
sprangfda496a2017-06-15 04:21:07 -0700179 explicit VideoStreamFactory(size_t num_temporal_layers, int framerate)
180 : num_temporal_layers_(num_temporal_layers), framerate_(framerate) {
asapersson5f7226f2016-11-25 04:37:00 -0800181 EXPECT_GT(num_temporal_layers, 0u);
sprangfda496a2017-06-15 04:21:07 -0700182 EXPECT_GT(framerate, 0);
asapersson5f7226f2016-11-25 04:37:00 -0800183 }
184
185 private:
186 std::vector<VideoStream> CreateEncoderStreams(
187 int width,
188 int height,
189 const VideoEncoderConfig& encoder_config) override {
190 std::vector<VideoStream> streams =
191 test::CreateVideoStreams(width, height, encoder_config);
192 for (VideoStream& stream : streams) {
Sergey Silkina796a7e2018-03-01 15:11:29 +0100193 stream.num_temporal_layers = num_temporal_layers_;
sprangfda496a2017-06-15 04:21:07 -0700194 stream.max_framerate = framerate_;
asapersson5f7226f2016-11-25 04:37:00 -0800195 }
196 return streams;
197 }
sprangfda496a2017-06-15 04:21:07 -0700198
asapersson5f7226f2016-11-25 04:37:00 -0800199 const size_t num_temporal_layers_;
sprangfda496a2017-06-15 04:21:07 -0700200 const int framerate_;
asapersson5f7226f2016-11-25 04:37:00 -0800201};
202
Noah Richards51db4212019-06-12 06:59:12 -0700203// Simulates simulcast behavior and makes highest stream resolutions divisible
204// by 4.
205class CroppingVideoStreamFactory
206 : public VideoEncoderConfig::VideoStreamFactoryInterface {
207 public:
208 explicit CroppingVideoStreamFactory(size_t num_temporal_layers, int framerate)
209 : num_temporal_layers_(num_temporal_layers), framerate_(framerate) {
210 EXPECT_GT(num_temporal_layers, 0u);
211 EXPECT_GT(framerate, 0);
212 }
213
214 private:
215 std::vector<VideoStream> CreateEncoderStreams(
216 int width,
217 int height,
218 const VideoEncoderConfig& encoder_config) override {
219 std::vector<VideoStream> streams = test::CreateVideoStreams(
220 width - width % 4, height - height % 4, encoder_config);
221 for (VideoStream& stream : streams) {
222 stream.num_temporal_layers = num_temporal_layers_;
223 stream.max_framerate = framerate_;
224 }
225 return streams;
226 }
227
228 const size_t num_temporal_layers_;
229 const int framerate_;
230};
231
sprangb1ca0732017-02-01 08:38:12 -0800232class AdaptingFrameForwarder : public test::FrameForwarder {
233 public:
234 AdaptingFrameForwarder() : adaptation_enabled_(false) {}
asaperssonfab67072017-04-04 05:51:49 -0700235 ~AdaptingFrameForwarder() override {}
sprangb1ca0732017-02-01 08:38:12 -0800236
237 void set_adaptation_enabled(bool enabled) {
238 rtc::CritScope cs(&crit_);
239 adaptation_enabled_ = enabled;
240 }
241
asaperssonfab67072017-04-04 05:51:49 -0700242 bool adaption_enabled() const {
sprangb1ca0732017-02-01 08:38:12 -0800243 rtc::CritScope cs(&crit_);
244 return adaptation_enabled_;
245 }
246
asapersson09f05612017-05-15 23:40:18 -0700247 rtc::VideoSinkWants last_wants() const {
248 rtc::CritScope cs(&crit_);
249 return last_wants_;
250 }
251
Danil Chapovalovb9b146c2018-06-15 12:28:07 +0200252 absl::optional<int> last_sent_width() const { return last_width_; }
253 absl::optional<int> last_sent_height() const { return last_height_; }
Jonathan Yubc771b72017-12-08 17:04:29 -0800254
sprangb1ca0732017-02-01 08:38:12 -0800255 void IncomingCapturedFrame(const VideoFrame& video_frame) override {
256 int cropped_width = 0;
257 int cropped_height = 0;
258 int out_width = 0;
259 int out_height = 0;
sprangc5d62e22017-04-02 23:53:04 -0700260 if (adaption_enabled()) {
261 if (adapter_.AdaptFrameResolution(
262 video_frame.width(), video_frame.height(),
263 video_frame.timestamp_us() * 1000, &cropped_width,
264 &cropped_height, &out_width, &out_height)) {
Artem Titov1ebfb6a2019-01-03 23:49:37 +0100265 VideoFrame adapted_frame =
266 VideoFrame::Builder()
267 .set_video_frame_buffer(new rtc::RefCountedObject<TestBuffer>(
268 nullptr, out_width, out_height))
269 .set_timestamp_rtp(99)
270 .set_timestamp_ms(99)
271 .set_rotation(kVideoRotation_0)
272 .build();
sprangc5d62e22017-04-02 23:53:04 -0700273 adapted_frame.set_ntp_time_ms(video_frame.ntp_time_ms());
274 test::FrameForwarder::IncomingCapturedFrame(adapted_frame);
Jonathan Yubc771b72017-12-08 17:04:29 -0800275 last_width_.emplace(adapted_frame.width());
276 last_height_.emplace(adapted_frame.height());
277 } else {
Danil Chapovalovb9b146c2018-06-15 12:28:07 +0200278 last_width_ = absl::nullopt;
279 last_height_ = absl::nullopt;
sprangc5d62e22017-04-02 23:53:04 -0700280 }
sprangb1ca0732017-02-01 08:38:12 -0800281 } else {
282 test::FrameForwarder::IncomingCapturedFrame(video_frame);
Jonathan Yubc771b72017-12-08 17:04:29 -0800283 last_width_.emplace(video_frame.width());
284 last_height_.emplace(video_frame.height());
sprangb1ca0732017-02-01 08:38:12 -0800285 }
286 }
287
288 void AddOrUpdateSink(rtc::VideoSinkInterface<VideoFrame>* sink,
289 const rtc::VideoSinkWants& wants) override {
290 rtc::CritScope cs(&crit_);
asapersson09f05612017-05-15 23:40:18 -0700291 last_wants_ = sink_wants();
sprangc5d62e22017-04-02 23:53:04 -0700292 adapter_.OnResolutionFramerateRequest(wants.target_pixel_count,
293 wants.max_pixel_count,
294 wants.max_framerate_fps);
sprangb1ca0732017-02-01 08:38:12 -0800295 test::FrameForwarder::AddOrUpdateSink(sink, wants);
296 }
sprangb1ca0732017-02-01 08:38:12 -0800297 cricket::VideoAdapter adapter_;
danilchapa37de392017-09-09 04:17:22 -0700298 bool adaptation_enabled_ RTC_GUARDED_BY(crit_);
299 rtc::VideoSinkWants last_wants_ RTC_GUARDED_BY(crit_);
Danil Chapovalovb9b146c2018-06-15 12:28:07 +0200300 absl::optional<int> last_width_;
301 absl::optional<int> last_height_;
sprangb1ca0732017-02-01 08:38:12 -0800302};
sprangc5d62e22017-04-02 23:53:04 -0700303
Niels Möller213618e2018-07-24 09:29:58 +0200304// TODO(nisse): Mock only VideoStreamEncoderObserver.
sprangc5d62e22017-04-02 23:53:04 -0700305class MockableSendStatisticsProxy : public SendStatisticsProxy {
306 public:
307 MockableSendStatisticsProxy(Clock* clock,
308 const VideoSendStream::Config& config,
309 VideoEncoderConfig::ContentType content_type)
310 : SendStatisticsProxy(clock, config, content_type) {}
311
312 VideoSendStream::Stats GetStats() override {
313 rtc::CritScope cs(&lock_);
314 if (mock_stats_)
315 return *mock_stats_;
316 return SendStatisticsProxy::GetStats();
317 }
318
Niels Möller213618e2018-07-24 09:29:58 +0200319 int GetInputFrameRate() const override {
320 rtc::CritScope cs(&lock_);
321 if (mock_stats_)
322 return mock_stats_->input_frame_rate;
323 return SendStatisticsProxy::GetInputFrameRate();
324 }
sprangc5d62e22017-04-02 23:53:04 -0700325 void SetMockStats(const VideoSendStream::Stats& stats) {
326 rtc::CritScope cs(&lock_);
327 mock_stats_.emplace(stats);
328 }
329
330 void ResetMockStats() {
331 rtc::CritScope cs(&lock_);
332 mock_stats_.reset();
333 }
334
335 private:
336 rtc::CriticalSection lock_;
Danil Chapovalovb9b146c2018-06-15 12:28:07 +0200337 absl::optional<VideoSendStream::Stats> mock_stats_ RTC_GUARDED_BY(lock_);
sprangc5d62e22017-04-02 23:53:04 -0700338};
339
sprang4847ae62017-06-27 07:06:52 -0700340class MockBitrateObserver : public VideoBitrateAllocationObserver {
341 public:
Erik Språng566124a2018-04-23 12:32:22 +0200342 MOCK_METHOD1(OnBitrateAllocationUpdated, void(const VideoBitrateAllocation&));
sprang4847ae62017-06-27 07:06:52 -0700343};
344
perkj803d97f2016-11-01 11:45:46 -0700345} // namespace
346
mflodmancc3d4422017-08-03 08:27:51 -0700347class VideoStreamEncoderTest : public ::testing::Test {
perkj26091b12016-09-01 01:17:40 -0700348 public:
349 static const int kDefaultTimeoutMs = 30 * 1000;
350
mflodmancc3d4422017-08-03 08:27:51 -0700351 VideoStreamEncoderTest()
perkj26091b12016-09-01 01:17:40 -0700352 : video_send_config_(VideoSendStream::Config(nullptr)),
perkjfa10b552016-10-02 23:45:26 -0700353 codec_width_(320),
354 codec_height_(240),
Åsa Persson8c1bf952018-09-13 10:42:19 +0200355 max_framerate_(kDefaultFramerate),
Danil Chapovalovd3ba2362019-04-10 17:01:23 +0200356 task_queue_factory_(CreateDefaultTaskQueueFactory()),
perkj26091b12016-09-01 01:17:40 -0700357 fake_encoder_(),
Niels Möller4db138e2018-04-19 09:04:13 +0200358 encoder_factory_(&fake_encoder_),
sprangc5d62e22017-04-02 23:53:04 -0700359 stats_proxy_(new MockableSendStatisticsProxy(
perkj803d97f2016-11-01 11:45:46 -0700360 Clock::GetRealTimeClock(),
361 video_send_config_,
362 webrtc::VideoEncoderConfig::ContentType::kRealtimeVideo)),
perkj26091b12016-09-01 01:17:40 -0700363 sink_(&fake_encoder_) {}
364
365 void SetUp() override {
perkj803d97f2016-11-01 11:45:46 -0700366 metrics::Reset();
perkj26091b12016-09-01 01:17:40 -0700367 video_send_config_ = VideoSendStream::Config(nullptr);
Niels Möller4db138e2018-04-19 09:04:13 +0200368 video_send_config_.encoder_settings.encoder_factory = &encoder_factory_;
Jiawei Ouc2ebe212018-11-08 10:02:56 -0800369 video_send_config_.encoder_settings.bitrate_allocator_factory =
Sergey Silkin5ee69672019-07-02 14:18:34 +0200370 &bitrate_allocator_factory_;
Niels Möller259a4972018-04-05 15:36:51 +0200371 video_send_config_.rtp.payload_name = "FAKE";
372 video_send_config_.rtp.payload_type = 125;
perkj26091b12016-09-01 01:17:40 -0700373
Per512ecb32016-09-23 15:52:06 +0200374 VideoEncoderConfig video_encoder_config;
Niels Möller259a4972018-04-05 15:36:51 +0200375 test::FillEncoderConfiguration(kVideoCodecVP8, 1, &video_encoder_config);
sprang4847ae62017-06-27 07:06:52 -0700376 video_encoder_config.video_stream_factory =
377 new rtc::RefCountedObject<VideoStreamFactory>(1, max_framerate_);
Erik Språng08127a92016-11-16 16:41:30 +0100378 video_encoder_config_ = video_encoder_config.Copy();
sprang4847ae62017-06-27 07:06:52 -0700379
380 // Framerate limit is specified by the VideoStreamFactory.
381 std::vector<VideoStream> streams =
382 video_encoder_config.video_stream_factory->CreateEncoderStreams(
383 codec_width_, codec_height_, video_encoder_config);
384 max_framerate_ = streams[0].max_framerate;
Sebastian Jansson40889f32019-04-17 12:11:20 +0200385 fake_clock_.SetTime(Timestamp::us(1234));
sprang4847ae62017-06-27 07:06:52 -0700386
Niels Möllerf1338562018-04-26 09:51:47 +0200387 ConfigureEncoder(std::move(video_encoder_config));
asapersson5f7226f2016-11-25 04:37:00 -0800388 }
389
Niels Möllerf1338562018-04-26 09:51:47 +0200390 void ConfigureEncoder(VideoEncoderConfig video_encoder_config) {
mflodmancc3d4422017-08-03 08:27:51 -0700391 if (video_stream_encoder_)
392 video_stream_encoder_->Stop();
393 video_stream_encoder_.reset(new VideoStreamEncoderUnderTest(
Danil Chapovalovd3ba2362019-04-10 17:01:23 +0200394 stats_proxy_.get(), video_send_config_.encoder_settings,
395 task_queue_factory_.get()));
mflodmancc3d4422017-08-03 08:27:51 -0700396 video_stream_encoder_->SetSink(&sink_, false /* rotation_applied */);
397 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -0700398 &video_source_, webrtc::DegradationPreference::MAINTAIN_FRAMERATE);
mflodmancc3d4422017-08-03 08:27:51 -0700399 video_stream_encoder_->SetStartBitrate(kTargetBitrateBps);
400 video_stream_encoder_->ConfigureEncoder(std::move(video_encoder_config),
Niels Möllerf1338562018-04-26 09:51:47 +0200401 kMaxPayloadLength);
mflodmancc3d4422017-08-03 08:27:51 -0700402 video_stream_encoder_->WaitUntilTaskQueueIsIdle();
asapersson5f7226f2016-11-25 04:37:00 -0800403 }
404
405 void ResetEncoder(const std::string& payload_name,
406 size_t num_streams,
407 size_t num_temporal_layers,
emircanbbcc3562017-08-18 00:28:40 -0700408 unsigned char num_spatial_layers,
sprang4847ae62017-06-27 07:06:52 -0700409 bool screenshare) {
Niels Möller259a4972018-04-05 15:36:51 +0200410 video_send_config_.rtp.payload_name = payload_name;
asapersson5f7226f2016-11-25 04:37:00 -0800411
412 VideoEncoderConfig video_encoder_config;
Niels Möller259a4972018-04-05 15:36:51 +0200413 video_encoder_config.codec_type = PayloadStringToCodecType(payload_name);
asapersson5f7226f2016-11-25 04:37:00 -0800414 video_encoder_config.number_of_streams = num_streams;
Erik Språngd7329ca2019-02-21 21:19:53 +0100415 video_encoder_config.max_bitrate_bps =
416 num_streams == 1 ? kTargetBitrateBps : kSimulcastTargetBitrateBps;
asapersson5f7226f2016-11-25 04:37:00 -0800417 video_encoder_config.video_stream_factory =
sprangfda496a2017-06-15 04:21:07 -0700418 new rtc::RefCountedObject<VideoStreamFactory>(num_temporal_layers,
419 kDefaultFramerate);
sprang4847ae62017-06-27 07:06:52 -0700420 video_encoder_config.content_type =
421 screenshare ? VideoEncoderConfig::ContentType::kScreen
422 : VideoEncoderConfig::ContentType::kRealtimeVideo;
emircanbbcc3562017-08-18 00:28:40 -0700423 if (payload_name == "VP9") {
424 VideoCodecVP9 vp9_settings = VideoEncoder::GetDefaultVp9Settings();
425 vp9_settings.numberOfSpatialLayers = num_spatial_layers;
426 video_encoder_config.encoder_specific_settings =
427 new rtc::RefCountedObject<
428 VideoEncoderConfig::Vp9EncoderSpecificSettings>(vp9_settings);
429 }
Niels Möllerf1338562018-04-26 09:51:47 +0200430 ConfigureEncoder(std::move(video_encoder_config));
perkj26091b12016-09-01 01:17:40 -0700431 }
432
sprang57c2fff2017-01-16 06:24:02 -0800433 VideoFrame CreateFrame(int64_t ntp_time_ms,
434 rtc::Event* destruction_event) const {
Artem Titov1ebfb6a2019-01-03 23:49:37 +0100435 VideoFrame frame =
436 VideoFrame::Builder()
437 .set_video_frame_buffer(new rtc::RefCountedObject<TestBuffer>(
438 destruction_event, codec_width_, codec_height_))
439 .set_timestamp_rtp(99)
440 .set_timestamp_ms(99)
441 .set_rotation(kVideoRotation_0)
442 .build();
sprang57c2fff2017-01-16 06:24:02 -0800443 frame.set_ntp_time_ms(ntp_time_ms);
perkj26091b12016-09-01 01:17:40 -0700444 return frame;
445 }
446
Ilya Nikolaevskiy71aee3a2019-02-18 13:01:26 +0100447 VideoFrame CreateFrameWithUpdatedPixel(int64_t ntp_time_ms,
448 rtc::Event* destruction_event,
449 int offset_x) const {
450 VideoFrame frame =
451 VideoFrame::Builder()
452 .set_video_frame_buffer(new rtc::RefCountedObject<TestBuffer>(
453 destruction_event, codec_width_, codec_height_))
454 .set_timestamp_rtp(99)
455 .set_timestamp_ms(99)
456 .set_rotation(kVideoRotation_0)
457 .set_update_rect({offset_x, 0, 1, 1})
458 .build();
459 frame.set_ntp_time_ms(ntp_time_ms);
460 return frame;
461 }
462
sprang57c2fff2017-01-16 06:24:02 -0800463 VideoFrame CreateFrame(int64_t ntp_time_ms, int width, int height) const {
Artem Titov1ebfb6a2019-01-03 23:49:37 +0100464 VideoFrame frame =
465 VideoFrame::Builder()
466 .set_video_frame_buffer(
467 new rtc::RefCountedObject<TestBuffer>(nullptr, width, height))
468 .set_timestamp_rtp(99)
469 .set_timestamp_ms(99)
470 .set_rotation(kVideoRotation_0)
471 .build();
sprang57c2fff2017-01-16 06:24:02 -0800472 frame.set_ntp_time_ms(ntp_time_ms);
sprangc5d62e22017-04-02 23:53:04 -0700473 frame.set_timestamp_us(ntp_time_ms * 1000);
perkj803d97f2016-11-01 11:45:46 -0700474 return frame;
475 }
476
Noah Richards51db4212019-06-12 06:59:12 -0700477 VideoFrame CreateFakeNativeFrame(int64_t ntp_time_ms,
478 rtc::Event* destruction_event,
479 int width,
480 int height) const {
481 VideoFrame frame =
482 VideoFrame::Builder()
483 .set_video_frame_buffer(new rtc::RefCountedObject<FakeNativeBuffer>(
484 destruction_event, width, height))
485 .set_timestamp_rtp(99)
486 .set_timestamp_ms(99)
487 .set_rotation(kVideoRotation_0)
488 .build();
489 frame.set_ntp_time_ms(ntp_time_ms);
490 return frame;
491 }
492
493 VideoFrame CreateFakeNativeFrame(int64_t ntp_time_ms,
494 rtc::Event* destruction_event) const {
495 return CreateFakeNativeFrame(ntp_time_ms, destruction_event, codec_width_,
496 codec_height_);
497 }
498
Åsa Perssonc29cb2c2019-03-25 12:06:59 +0100499 void VerifyAllocatedBitrate(const VideoBitrateAllocation& expected_bitrate) {
500 MockBitrateObserver bitrate_observer;
501 video_stream_encoder_->SetBitrateAllocationObserver(&bitrate_observer);
502
503 EXPECT_CALL(bitrate_observer, OnBitrateAllocationUpdated(expected_bitrate))
504 .Times(1);
505 video_stream_encoder_->OnBitrateUpdated(DataRate::bps(kTargetBitrateBps),
Erik Språng4c6ca302019-04-08 15:14:01 +0200506 DataRate::bps(kTargetBitrateBps), 0,
507 0);
Åsa Perssonc29cb2c2019-03-25 12:06:59 +0100508
509 video_source_.IncomingCapturedFrame(
510 CreateFrame(1, codec_width_, codec_height_));
511 WaitForEncodedFrame(1);
512 }
513
asapersson02465b82017-04-10 01:12:52 -0700514 void VerifyNoLimitation(const rtc::VideoSinkWants& wants) {
asapersson02465b82017-04-10 01:12:52 -0700515 EXPECT_EQ(std::numeric_limits<int>::max(), wants.max_framerate_fps);
asapersson09f05612017-05-15 23:40:18 -0700516 EXPECT_EQ(std::numeric_limits<int>::max(), wants.max_pixel_count);
517 EXPECT_FALSE(wants.target_pixel_count);
asapersson02465b82017-04-10 01:12:52 -0700518 }
519
asapersson09f05612017-05-15 23:40:18 -0700520 void VerifyFpsEqResolutionEq(const rtc::VideoSinkWants& wants1,
521 const rtc::VideoSinkWants& wants2) {
522 EXPECT_EQ(wants1.max_framerate_fps, wants2.max_framerate_fps);
523 EXPECT_EQ(wants1.max_pixel_count, wants2.max_pixel_count);
524 }
525
Åsa Persson8c1bf952018-09-13 10:42:19 +0200526 void VerifyFpsMaxResolutionMax(const rtc::VideoSinkWants& wants) {
527 EXPECT_EQ(kDefaultFramerate, wants.max_framerate_fps);
528 EXPECT_EQ(std::numeric_limits<int>::max(), wants.max_pixel_count);
529 EXPECT_FALSE(wants.target_pixel_count);
530 }
531
asapersson09f05612017-05-15 23:40:18 -0700532 void VerifyFpsMaxResolutionLt(const rtc::VideoSinkWants& wants1,
533 const rtc::VideoSinkWants& wants2) {
Åsa Persson8c1bf952018-09-13 10:42:19 +0200534 EXPECT_EQ(kDefaultFramerate, wants1.max_framerate_fps);
asapersson09f05612017-05-15 23:40:18 -0700535 EXPECT_LT(wants1.max_pixel_count, wants2.max_pixel_count);
536 EXPECT_GT(wants1.max_pixel_count, 0);
537 }
538
539 void VerifyFpsMaxResolutionGt(const rtc::VideoSinkWants& wants1,
540 const rtc::VideoSinkWants& wants2) {
Åsa Persson8c1bf952018-09-13 10:42:19 +0200541 EXPECT_EQ(kDefaultFramerate, wants1.max_framerate_fps);
asapersson09f05612017-05-15 23:40:18 -0700542 EXPECT_GT(wants1.max_pixel_count, wants2.max_pixel_count);
543 }
544
asaperssonf7e294d2017-06-13 23:25:22 -0700545 void VerifyFpsMaxResolutionEq(const rtc::VideoSinkWants& wants1,
546 const rtc::VideoSinkWants& wants2) {
Åsa Persson8c1bf952018-09-13 10:42:19 +0200547 EXPECT_EQ(kDefaultFramerate, wants1.max_framerate_fps);
asaperssonf7e294d2017-06-13 23:25:22 -0700548 EXPECT_EQ(wants1.max_pixel_count, wants2.max_pixel_count);
549 }
550
551 void VerifyFpsLtResolutionEq(const rtc::VideoSinkWants& wants1,
552 const rtc::VideoSinkWants& wants2) {
553 EXPECT_LT(wants1.max_framerate_fps, wants2.max_framerate_fps);
554 EXPECT_EQ(wants1.max_pixel_count, wants2.max_pixel_count);
555 }
556
557 void VerifyFpsGtResolutionEq(const rtc::VideoSinkWants& wants1,
558 const rtc::VideoSinkWants& wants2) {
559 EXPECT_GT(wants1.max_framerate_fps, wants2.max_framerate_fps);
560 EXPECT_EQ(wants1.max_pixel_count, wants2.max_pixel_count);
561 }
562
563 void VerifyFpsEqResolutionLt(const rtc::VideoSinkWants& wants1,
564 const rtc::VideoSinkWants& wants2) {
565 EXPECT_EQ(wants1.max_framerate_fps, wants2.max_framerate_fps);
566 EXPECT_LT(wants1.max_pixel_count, wants2.max_pixel_count);
567 EXPECT_GT(wants1.max_pixel_count, 0);
568 }
569
570 void VerifyFpsEqResolutionGt(const rtc::VideoSinkWants& wants1,
571 const rtc::VideoSinkWants& wants2) {
572 EXPECT_EQ(wants1.max_framerate_fps, wants2.max_framerate_fps);
573 EXPECT_GT(wants1.max_pixel_count, wants2.max_pixel_count);
574 }
575
asapersson09f05612017-05-15 23:40:18 -0700576 void VerifyFpsMaxResolutionLt(const rtc::VideoSinkWants& wants,
577 int pixel_count) {
Åsa Persson8c1bf952018-09-13 10:42:19 +0200578 EXPECT_EQ(kDefaultFramerate, wants.max_framerate_fps);
asapersson02465b82017-04-10 01:12:52 -0700579 EXPECT_LT(wants.max_pixel_count, pixel_count);
580 EXPECT_GT(wants.max_pixel_count, 0);
asapersson09f05612017-05-15 23:40:18 -0700581 }
582
583 void VerifyFpsLtResolutionMax(const rtc::VideoSinkWants& wants, int fps) {
584 EXPECT_LT(wants.max_framerate_fps, fps);
585 EXPECT_EQ(std::numeric_limits<int>::max(), wants.max_pixel_count);
586 EXPECT_FALSE(wants.target_pixel_count);
asapersson02465b82017-04-10 01:12:52 -0700587 }
588
asaperssonf7e294d2017-06-13 23:25:22 -0700589 void VerifyFpsEqResolutionMax(const rtc::VideoSinkWants& wants,
590 int expected_fps) {
591 EXPECT_EQ(expected_fps, wants.max_framerate_fps);
592 EXPECT_EQ(std::numeric_limits<int>::max(), wants.max_pixel_count);
593 EXPECT_FALSE(wants.target_pixel_count);
594 }
595
Jonathan Yubc771b72017-12-08 17:04:29 -0800596 void VerifyBalancedModeFpsRange(const rtc::VideoSinkWants& wants,
597 int last_frame_pixels) {
598 // Balanced mode should always scale FPS to the desired range before
599 // attempting to scale resolution.
600 int fps_limit = wants.max_framerate_fps;
601 if (last_frame_pixels <= 320 * 240) {
602 EXPECT_TRUE(7 <= fps_limit && fps_limit <= 10);
603 } else if (last_frame_pixels <= 480 * 270) {
604 EXPECT_TRUE(10 <= fps_limit && fps_limit <= 15);
605 } else if (last_frame_pixels <= 640 * 480) {
606 EXPECT_LE(15, fps_limit);
607 } else {
Åsa Persson8c1bf952018-09-13 10:42:19 +0200608 EXPECT_EQ(kDefaultFramerate, fps_limit);
Jonathan Yubc771b72017-12-08 17:04:29 -0800609 }
610 }
611
sprang4847ae62017-06-27 07:06:52 -0700612 void WaitForEncodedFrame(int64_t expected_ntp_time) {
613 sink_.WaitForEncodedFrame(expected_ntp_time);
Sebastian Jansson40889f32019-04-17 12:11:20 +0200614 fake_clock_.AdvanceTime(TimeDelta::seconds(1) / max_framerate_);
sprang4847ae62017-06-27 07:06:52 -0700615 }
616
617 bool TimedWaitForEncodedFrame(int64_t expected_ntp_time, int64_t timeout_ms) {
618 bool ok = sink_.TimedWaitForEncodedFrame(expected_ntp_time, timeout_ms);
Sebastian Jansson40889f32019-04-17 12:11:20 +0200619 fake_clock_.AdvanceTime(TimeDelta::seconds(1) / max_framerate_);
sprang4847ae62017-06-27 07:06:52 -0700620 return ok;
621 }
622
623 void WaitForEncodedFrame(uint32_t expected_width, uint32_t expected_height) {
624 sink_.WaitForEncodedFrame(expected_width, expected_height);
Sebastian Jansson40889f32019-04-17 12:11:20 +0200625 fake_clock_.AdvanceTime(TimeDelta::seconds(1) / max_framerate_);
sprang4847ae62017-06-27 07:06:52 -0700626 }
627
628 void ExpectDroppedFrame() {
629 sink_.ExpectDroppedFrame();
Sebastian Jansson40889f32019-04-17 12:11:20 +0200630 fake_clock_.AdvanceTime(TimeDelta::seconds(1) / max_framerate_);
sprang4847ae62017-06-27 07:06:52 -0700631 }
632
633 bool WaitForFrame(int64_t timeout_ms) {
634 bool ok = sink_.WaitForFrame(timeout_ms);
Sebastian Jansson40889f32019-04-17 12:11:20 +0200635 fake_clock_.AdvanceTime(TimeDelta::seconds(1) / max_framerate_);
sprang4847ae62017-06-27 07:06:52 -0700636 return ok;
637 }
638
perkj26091b12016-09-01 01:17:40 -0700639 class TestEncoder : public test::FakeEncoder {
640 public:
Niels Möllerc572ff32018-11-07 08:43:50 +0100641 TestEncoder() : FakeEncoder(Clock::GetRealTimeClock()) {}
perkj26091b12016-09-01 01:17:40 -0700642
asaperssonfab67072017-04-04 05:51:49 -0700643 VideoCodec codec_config() const {
brandtre78d2662017-01-16 05:57:16 -0800644 rtc::CritScope lock(&crit_sect_);
perkjfa10b552016-10-02 23:45:26 -0700645 return config_;
646 }
647
648 void BlockNextEncode() {
brandtre78d2662017-01-16 05:57:16 -0800649 rtc::CritScope lock(&local_crit_sect_);
perkjfa10b552016-10-02 23:45:26 -0700650 block_next_encode_ = true;
651 }
652
Erik Språngaed30702018-11-05 12:57:17 +0100653 VideoEncoder::EncoderInfo GetEncoderInfo() const override {
kthelgason2fc52542017-03-03 00:24:41 -0800654 rtc::CritScope lock(&local_crit_sect_);
Mirta Dvornicicccc1b572019-01-15 12:42:18 +0100655 EncoderInfo info;
Erik Språngb7cb7b52019-02-26 15:52:33 +0100656 if (initialized_ == EncoderState::kInitialized) {
Mirta Dvornicicccc1b572019-01-15 12:42:18 +0100657 if (quality_scaling_) {
658 info.scaling_settings =
659 VideoEncoder::ScalingSettings(1, 2, kMinPixelsPerFrame);
660 }
661 info.is_hardware_accelerated = is_hardware_accelerated_;
Åsa Perssonc29cb2c2019-03-25 12:06:59 +0100662 for (int i = 0; i < kMaxSpatialLayers; ++i) {
663 if (temporal_layers_supported_[i]) {
664 int num_layers = temporal_layers_supported_[i].value() ? 2 : 1;
665 info.fps_allocation[i].resize(num_layers);
666 }
667 }
Erik Språngaed30702018-11-05 12:57:17 +0100668 }
Sergey Silkin6456e352019-07-08 17:56:40 +0200669
670 info.resolution_bitrate_limits = resolution_bitrate_limits_;
Erik Språngaed30702018-11-05 12:57:17 +0100671 return info;
kthelgason876222f2016-11-29 01:44:11 -0800672 }
673
Erik Språngb7cb7b52019-02-26 15:52:33 +0100674 int32_t RegisterEncodeCompleteCallback(
675 EncodedImageCallback* callback) override {
676 rtc::CritScope lock(&local_crit_sect_);
677 encoded_image_callback_ = callback;
678 return FakeEncoder::RegisterEncodeCompleteCallback(callback);
679 }
680
perkjfa10b552016-10-02 23:45:26 -0700681 void ContinueEncode() { continue_encode_event_.Set(); }
682
683 void CheckLastTimeStampsMatch(int64_t ntp_time_ms,
684 uint32_t timestamp) const {
brandtre78d2662017-01-16 05:57:16 -0800685 rtc::CritScope lock(&local_crit_sect_);
perkjfa10b552016-10-02 23:45:26 -0700686 EXPECT_EQ(timestamp_, timestamp);
687 EXPECT_EQ(ntp_time_ms_, ntp_time_ms);
688 }
689
kthelgason2fc52542017-03-03 00:24:41 -0800690 void SetQualityScaling(bool b) {
691 rtc::CritScope lock(&local_crit_sect_);
692 quality_scaling_ = b;
693 }
kthelgasonad9010c2017-02-14 00:46:51 -0800694
Mirta Dvornicicccc1b572019-01-15 12:42:18 +0100695 void SetIsHardwareAccelerated(bool is_hardware_accelerated) {
696 rtc::CritScope lock(&local_crit_sect_);
697 is_hardware_accelerated_ = is_hardware_accelerated;
698 }
699
Åsa Perssonc29cb2c2019-03-25 12:06:59 +0100700 void SetTemporalLayersSupported(size_t spatial_idx, bool supported) {
701 RTC_DCHECK_LT(spatial_idx, kMaxSpatialLayers);
702 rtc::CritScope lock(&local_crit_sect_);
703 temporal_layers_supported_[spatial_idx] = supported;
704 }
705
Sergey Silkin6456e352019-07-08 17:56:40 +0200706 void SetResolutionBitrateLimits(
707 std::vector<ResolutionBitrateLimits> thresholds) {
708 rtc::CritScope cs(&local_crit_sect_);
709 resolution_bitrate_limits_ = thresholds;
710 }
711
sprangfe627f32017-03-29 08:24:59 -0700712 void ForceInitEncodeFailure(bool force_failure) {
713 rtc::CritScope lock(&local_crit_sect_);
714 force_init_encode_failed_ = force_failure;
715 }
716
Niels Möller6bb5ab92019-01-11 11:11:10 +0100717 void SimulateOvershoot(double rate_factor) {
718 rtc::CritScope lock(&local_crit_sect_);
719 rate_factor_ = rate_factor;
720 }
721
Erik Språngd7329ca2019-02-21 21:19:53 +0100722 uint32_t GetLastFramerate() const {
Niels Möller6bb5ab92019-01-11 11:11:10 +0100723 rtc::CritScope lock(&local_crit_sect_);
724 return last_framerate_;
725 }
726
Erik Språngd7329ca2019-02-21 21:19:53 +0100727 VideoFrame::UpdateRect GetLastUpdateRect() const {
Ilya Nikolaevskiy71aee3a2019-02-18 13:01:26 +0100728 rtc::CritScope lock(&local_crit_sect_);
729 return last_update_rect_;
730 }
731
Niels Möller87e2d782019-03-07 10:18:23 +0100732 const std::vector<VideoFrameType>& LastFrameTypes() const {
Erik Språngd7329ca2019-02-21 21:19:53 +0100733 rtc::CritScope lock(&local_crit_sect_);
734 return last_frame_types_;
735 }
736
737 void InjectFrame(const VideoFrame& input_image, bool keyframe) {
Niels Möller87e2d782019-03-07 10:18:23 +0100738 const std::vector<VideoFrameType> frame_type = {
Niels Möller8f7ce222019-03-21 15:43:58 +0100739 keyframe ? VideoFrameType::kVideoFrameKey
740 : VideoFrameType::kVideoFrameDelta};
Erik Språngd7329ca2019-02-21 21:19:53 +0100741 {
742 rtc::CritScope lock(&local_crit_sect_);
743 last_frame_types_ = frame_type;
744 }
Niels Möllerb859b322019-03-07 12:40:01 +0100745 FakeEncoder::Encode(input_image, &frame_type);
Erik Språngd7329ca2019-02-21 21:19:53 +0100746 }
747
Erik Språngb7cb7b52019-02-26 15:52:33 +0100748 void InjectEncodedImage(const EncodedImage& image) {
749 rtc::CritScope lock(&local_crit_sect_);
750 encoded_image_callback_->OnEncodedImage(image, nullptr, nullptr);
751 }
752
Mirta Dvornicic28f0eb22019-05-28 16:30:16 +0200753 void InjectEncodedImage(const EncodedImage& image,
754 const CodecSpecificInfo* codec_specific_info,
755 const RTPFragmentationHeader* fragmentation) {
756 rtc::CritScope lock(&local_crit_sect_);
757 encoded_image_callback_->OnEncodedImage(image, codec_specific_info,
758 fragmentation);
759 }
760
Erik Språngd7329ca2019-02-21 21:19:53 +0100761 void ExpectNullFrame() {
762 rtc::CritScope lock(&local_crit_sect_);
763 expect_null_frame_ = true;
764 }
765
766 absl::optional<VideoBitrateAllocation> GetAndResetLastBitrateAllocation() {
767 auto allocation = last_bitrate_allocation_;
768 last_bitrate_allocation_.reset();
769 return allocation;
770 }
771
Sergey Silkin5ee69672019-07-02 14:18:34 +0200772 int GetNumEncoderInitializations() const {
773 rtc::CritScope lock(&local_crit_sect_);
774 return num_encoder_initializations_;
775 }
776
perkjfa10b552016-10-02 23:45:26 -0700777 private:
perkj26091b12016-09-01 01:17:40 -0700778 int32_t Encode(const VideoFrame& input_image,
Niels Möller87e2d782019-03-07 10:18:23 +0100779 const std::vector<VideoFrameType>* frame_types) override {
perkj26091b12016-09-01 01:17:40 -0700780 bool block_encode;
781 {
brandtre78d2662017-01-16 05:57:16 -0800782 rtc::CritScope lock(&local_crit_sect_);
Erik Språngd7329ca2019-02-21 21:19:53 +0100783 if (expect_null_frame_) {
784 EXPECT_EQ(input_image.timestamp(), 0u);
785 EXPECT_EQ(input_image.width(), 1);
786 last_frame_types_ = *frame_types;
787 expect_null_frame_ = false;
788 } else {
789 EXPECT_GT(input_image.timestamp(), timestamp_);
790 EXPECT_GT(input_image.ntp_time_ms(), ntp_time_ms_);
791 EXPECT_EQ(input_image.timestamp(), input_image.ntp_time_ms() * 90);
792 }
perkj26091b12016-09-01 01:17:40 -0700793
794 timestamp_ = input_image.timestamp();
795 ntp_time_ms_ = input_image.ntp_time_ms();
perkj803d97f2016-11-01 11:45:46 -0700796 last_input_width_ = input_image.width();
797 last_input_height_ = input_image.height();
perkj26091b12016-09-01 01:17:40 -0700798 block_encode = block_next_encode_;
799 block_next_encode_ = false;
Ilya Nikolaevskiy71aee3a2019-02-18 13:01:26 +0100800 last_update_rect_ = input_image.update_rect();
Erik Språngd7329ca2019-02-21 21:19:53 +0100801 last_frame_types_ = *frame_types;
perkj26091b12016-09-01 01:17:40 -0700802 }
Niels Möllerb859b322019-03-07 12:40:01 +0100803 int32_t result = FakeEncoder::Encode(input_image, frame_types);
perkj26091b12016-09-01 01:17:40 -0700804 if (block_encode)
perkja49cbd32016-09-16 07:53:41 -0700805 EXPECT_TRUE(continue_encode_event_.Wait(kDefaultTimeoutMs));
perkj26091b12016-09-01 01:17:40 -0700806 return result;
807 }
808
sprangfe627f32017-03-29 08:24:59 -0700809 int32_t InitEncode(const VideoCodec* config,
Elad Alon370f93a2019-06-11 14:57:57 +0200810 const Settings& settings) override {
811 int res = FakeEncoder::InitEncode(config, settings);
Sergey Silkin5ee69672019-07-02 14:18:34 +0200812
sprangfe627f32017-03-29 08:24:59 -0700813 rtc::CritScope lock(&local_crit_sect_);
Erik Språngb7cb7b52019-02-26 15:52:33 +0100814 EXPECT_EQ(initialized_, EncoderState::kUninitialized);
Sergey Silkin5ee69672019-07-02 14:18:34 +0200815
816 ++num_encoder_initializations_;
817
Erik Språng82fad3d2018-03-21 09:57:23 +0100818 if (config->codecType == kVideoCodecVP8) {
sprangfe627f32017-03-29 08:24:59 -0700819 // Simulate setting up temporal layers, in order to validate the life
820 // cycle of these objects.
Elad Aloncde8ab22019-03-20 11:56:20 +0100821 Vp8TemporalLayersFactory factory;
Elad Alon45befc52019-07-02 11:20:09 +0200822 frame_buffer_controller_ =
823 factory.Create(*config, settings, &fec_controller_override_);
sprangfe627f32017-03-29 08:24:59 -0700824 }
Erik Språngb7cb7b52019-02-26 15:52:33 +0100825 if (force_init_encode_failed_) {
826 initialized_ = EncoderState::kInitializationFailed;
sprangfe627f32017-03-29 08:24:59 -0700827 return -1;
Erik Språngb7cb7b52019-02-26 15:52:33 +0100828 }
Mirta Dvornicicccc1b572019-01-15 12:42:18 +0100829
Erik Språngb7cb7b52019-02-26 15:52:33 +0100830 initialized_ = EncoderState::kInitialized;
sprangfe627f32017-03-29 08:24:59 -0700831 return res;
832 }
833
Erik Språngb7cb7b52019-02-26 15:52:33 +0100834 int32_t Release() override {
835 rtc::CritScope lock(&local_crit_sect_);
836 EXPECT_NE(initialized_, EncoderState::kUninitialized);
837 initialized_ = EncoderState::kUninitialized;
838 return FakeEncoder::Release();
839 }
840
Erik Språng16cb8f52019-04-12 13:59:09 +0200841 void SetRates(const RateControlParameters& parameters) {
Niels Möller6bb5ab92019-01-11 11:11:10 +0100842 rtc::CritScope lock(&local_crit_sect_);
843 VideoBitrateAllocation adjusted_rate_allocation;
844 for (size_t si = 0; si < kMaxSpatialLayers; ++si) {
845 for (size_t ti = 0; ti < kMaxTemporalStreams; ++ti) {
Erik Språng16cb8f52019-04-12 13:59:09 +0200846 if (parameters.bitrate.HasBitrate(si, ti)) {
Niels Möller6bb5ab92019-01-11 11:11:10 +0100847 adjusted_rate_allocation.SetBitrate(
848 si, ti,
Erik Språng16cb8f52019-04-12 13:59:09 +0200849 static_cast<uint32_t>(parameters.bitrate.GetBitrate(si, ti) *
Niels Möller6bb5ab92019-01-11 11:11:10 +0100850 rate_factor_));
851 }
852 }
853 }
Erik Språng16cb8f52019-04-12 13:59:09 +0200854 last_framerate_ = static_cast<uint32_t>(parameters.framerate_fps + 0.5);
855 last_bitrate_allocation_ = parameters.bitrate;
856 RateControlParameters adjusted_paramters = parameters;
857 adjusted_paramters.bitrate = adjusted_rate_allocation;
858 FakeEncoder::SetRates(adjusted_paramters);
Niels Möller6bb5ab92019-01-11 11:11:10 +0100859 }
860
brandtre78d2662017-01-16 05:57:16 -0800861 rtc::CriticalSection local_crit_sect_;
Erik Språngb7cb7b52019-02-26 15:52:33 +0100862 enum class EncoderState {
863 kUninitialized,
864 kInitializationFailed,
865 kInitialized
866 } initialized_ RTC_GUARDED_BY(local_crit_sect_) =
867 EncoderState::kUninitialized;
danilchapa37de392017-09-09 04:17:22 -0700868 bool block_next_encode_ RTC_GUARDED_BY(local_crit_sect_) = false;
perkj26091b12016-09-01 01:17:40 -0700869 rtc::Event continue_encode_event_;
danilchapa37de392017-09-09 04:17:22 -0700870 uint32_t timestamp_ RTC_GUARDED_BY(local_crit_sect_) = 0;
871 int64_t ntp_time_ms_ RTC_GUARDED_BY(local_crit_sect_) = 0;
872 int last_input_width_ RTC_GUARDED_BY(local_crit_sect_) = 0;
873 int last_input_height_ RTC_GUARDED_BY(local_crit_sect_) = 0;
874 bool quality_scaling_ RTC_GUARDED_BY(local_crit_sect_) = true;
Mirta Dvornicicccc1b572019-01-15 12:42:18 +0100875 bool is_hardware_accelerated_ RTC_GUARDED_BY(local_crit_sect_) = false;
Elad Aloncde8ab22019-03-20 11:56:20 +0100876 std::unique_ptr<Vp8FrameBufferController> frame_buffer_controller_
danilchapa37de392017-09-09 04:17:22 -0700877 RTC_GUARDED_BY(local_crit_sect_);
Åsa Perssonc29cb2c2019-03-25 12:06:59 +0100878 absl::optional<bool>
879 temporal_layers_supported_[kMaxSpatialLayers] RTC_GUARDED_BY(
880 local_crit_sect_);
danilchapa37de392017-09-09 04:17:22 -0700881 bool force_init_encode_failed_ RTC_GUARDED_BY(local_crit_sect_) = false;
Niels Möller6bb5ab92019-01-11 11:11:10 +0100882 double rate_factor_ RTC_GUARDED_BY(local_crit_sect_) = 1.0;
883 uint32_t last_framerate_ RTC_GUARDED_BY(local_crit_sect_) = 0;
Erik Språngd7329ca2019-02-21 21:19:53 +0100884 absl::optional<VideoBitrateAllocation> last_bitrate_allocation_;
Ilya Nikolaevskiy71aee3a2019-02-18 13:01:26 +0100885 VideoFrame::UpdateRect last_update_rect_
886 RTC_GUARDED_BY(local_crit_sect_) = {0, 0, 0, 0};
Niels Möller87e2d782019-03-07 10:18:23 +0100887 std::vector<VideoFrameType> last_frame_types_;
Erik Språngd7329ca2019-02-21 21:19:53 +0100888 bool expect_null_frame_ = false;
Erik Språngb7cb7b52019-02-26 15:52:33 +0100889 EncodedImageCallback* encoded_image_callback_
890 RTC_GUARDED_BY(local_crit_sect_) = nullptr;
Elad Alon45befc52019-07-02 11:20:09 +0200891 MockFecControllerOverride fec_controller_override_;
Sergey Silkin5ee69672019-07-02 14:18:34 +0200892 int num_encoder_initializations_ RTC_GUARDED_BY(local_crit_sect_) = 0;
Sergey Silkin6456e352019-07-08 17:56:40 +0200893 std::vector<ResolutionBitrateLimits> resolution_bitrate_limits_
894 RTC_GUARDED_BY(local_crit_sect_);
perkj26091b12016-09-01 01:17:40 -0700895 };
896
mflodmancc3d4422017-08-03 08:27:51 -0700897 class TestSink : public VideoStreamEncoder::EncoderSink {
perkj26091b12016-09-01 01:17:40 -0700898 public:
899 explicit TestSink(TestEncoder* test_encoder)
Niels Möllerc572ff32018-11-07 08:43:50 +0100900 : test_encoder_(test_encoder) {}
perkj26091b12016-09-01 01:17:40 -0700901
perkj26091b12016-09-01 01:17:40 -0700902 void WaitForEncodedFrame(int64_t expected_ntp_time) {
sprang4847ae62017-06-27 07:06:52 -0700903 EXPECT_TRUE(
904 TimedWaitForEncodedFrame(expected_ntp_time, kDefaultTimeoutMs));
905 }
906
907 bool TimedWaitForEncodedFrame(int64_t expected_ntp_time,
908 int64_t timeout_ms) {
perkj26091b12016-09-01 01:17:40 -0700909 uint32_t timestamp = 0;
sprang4847ae62017-06-27 07:06:52 -0700910 if (!encoded_frame_event_.Wait(timeout_ms))
911 return false;
perkj26091b12016-09-01 01:17:40 -0700912 {
913 rtc::CritScope lock(&crit_);
sprangb1ca0732017-02-01 08:38:12 -0800914 timestamp = last_timestamp_;
perkj26091b12016-09-01 01:17:40 -0700915 }
916 test_encoder_->CheckLastTimeStampsMatch(expected_ntp_time, timestamp);
sprang4847ae62017-06-27 07:06:52 -0700917 return true;
perkj26091b12016-09-01 01:17:40 -0700918 }
919
sprangb1ca0732017-02-01 08:38:12 -0800920 void WaitForEncodedFrame(uint32_t expected_width,
921 uint32_t expected_height) {
sprangc5d62e22017-04-02 23:53:04 -0700922 EXPECT_TRUE(encoded_frame_event_.Wait(kDefaultTimeoutMs));
Åsa Perssonc74d8da2017-12-04 14:13:56 +0100923 CheckLastFrameSizeMatches(expected_width, expected_height);
sprangc5d62e22017-04-02 23:53:04 -0700924 }
925
Åsa Perssonc74d8da2017-12-04 14:13:56 +0100926 void CheckLastFrameSizeMatches(uint32_t expected_width,
sprangc5d62e22017-04-02 23:53:04 -0700927 uint32_t expected_height) {
sprangb1ca0732017-02-01 08:38:12 -0800928 uint32_t width = 0;
929 uint32_t height = 0;
sprangb1ca0732017-02-01 08:38:12 -0800930 {
931 rtc::CritScope lock(&crit_);
932 width = last_width_;
933 height = last_height_;
934 }
935 EXPECT_EQ(expected_height, height);
936 EXPECT_EQ(expected_width, width);
937 }
938
Ilya Nikolaevskiyba96e2f2019-06-04 15:15:14 +0200939 void CheckLastFrameRotationMatches(VideoRotation expected_rotation) {
940 VideoRotation rotation;
941 {
942 rtc::CritScope lock(&crit_);
943 rotation = last_rotation_;
944 }
945 EXPECT_EQ(expected_rotation, rotation);
946 }
947
kthelgason2fc52542017-03-03 00:24:41 -0800948 void ExpectDroppedFrame() { EXPECT_FALSE(encoded_frame_event_.Wait(100)); }
kthelgason2bc68642017-02-07 07:02:22 -0800949
sprangc5d62e22017-04-02 23:53:04 -0700950 bool WaitForFrame(int64_t timeout_ms) {
951 return encoded_frame_event_.Wait(timeout_ms);
952 }
953
perkj26091b12016-09-01 01:17:40 -0700954 void SetExpectNoFrames() {
955 rtc::CritScope lock(&crit_);
956 expect_frames_ = false;
957 }
958
asaperssonfab67072017-04-04 05:51:49 -0700959 int number_of_reconfigurations() const {
Per512ecb32016-09-23 15:52:06 +0200960 rtc::CritScope lock(&crit_);
961 return number_of_reconfigurations_;
962 }
963
asaperssonfab67072017-04-04 05:51:49 -0700964 int last_min_transmit_bitrate() const {
Per512ecb32016-09-23 15:52:06 +0200965 rtc::CritScope lock(&crit_);
966 return min_transmit_bitrate_bps_;
967 }
968
Erik Språngd7329ca2019-02-21 21:19:53 +0100969 void SetNumExpectedLayers(size_t num_layers) {
970 rtc::CritScope lock(&crit_);
971 num_expected_layers_ = num_layers;
972 }
973
Erik Språngb7cb7b52019-02-26 15:52:33 +0100974 int64_t GetLastCaptureTimeMs() const {
975 rtc::CritScope lock(&crit_);
976 return last_capture_time_ms_;
977 }
978
Mirta Dvornicic28f0eb22019-05-28 16:30:16 +0200979 std::vector<uint8_t> GetLastEncodedImageData() {
980 rtc::CritScope lock(&crit_);
981 return std::move(last_encoded_image_data_);
982 }
983
984 RTPFragmentationHeader GetLastFragmentation() {
985 rtc::CritScope lock(&crit_);
986 return std::move(last_fragmentation_);
987 }
988
perkj26091b12016-09-01 01:17:40 -0700989 private:
sergeyu2cb155a2016-11-04 11:39:29 -0700990 Result OnEncodedImage(
991 const EncodedImage& encoded_image,
992 const CodecSpecificInfo* codec_specific_info,
993 const RTPFragmentationHeader* fragmentation) override {
Per512ecb32016-09-23 15:52:06 +0200994 rtc::CritScope lock(&crit_);
995 EXPECT_TRUE(expect_frames_);
Mirta Dvornicic28f0eb22019-05-28 16:30:16 +0200996 last_encoded_image_data_ = std::vector<uint8_t>(
997 encoded_image.data(), encoded_image.data() + encoded_image.size());
998 if (fragmentation) {
999 last_fragmentation_.CopyFrom(*fragmentation);
1000 }
Erik Språngd7329ca2019-02-21 21:19:53 +01001001 uint32_t timestamp = encoded_image.Timestamp();
1002 if (last_timestamp_ != timestamp) {
1003 num_received_layers_ = 1;
1004 } else {
1005 ++num_received_layers_;
1006 }
1007 last_timestamp_ = timestamp;
Erik Språngb7cb7b52019-02-26 15:52:33 +01001008 last_capture_time_ms_ = encoded_image.capture_time_ms_;
sprangb1ca0732017-02-01 08:38:12 -08001009 last_width_ = encoded_image._encodedWidth;
1010 last_height_ = encoded_image._encodedHeight;
Ilya Nikolaevskiyba96e2f2019-06-04 15:15:14 +02001011 last_rotation_ = encoded_image.rotation_;
Erik Språngd7329ca2019-02-21 21:19:53 +01001012 if (num_received_layers_ == num_expected_layers_) {
1013 encoded_frame_event_.Set();
1014 }
sprangb1ca0732017-02-01 08:38:12 -08001015 return Result(Result::OK, last_timestamp_);
Per512ecb32016-09-23 15:52:06 +02001016 }
1017
Rasmus Brandtc402dbe2019-02-04 11:09:46 +01001018 void OnEncoderConfigurationChanged(
1019 std::vector<VideoStream> streams,
1020 VideoEncoderConfig::ContentType content_type,
1021 int min_transmit_bitrate_bps) override {
Sergey Silkin5ee69672019-07-02 14:18:34 +02001022 rtc::CritScope lock(&crit_);
Per512ecb32016-09-23 15:52:06 +02001023 ++number_of_reconfigurations_;
1024 min_transmit_bitrate_bps_ = min_transmit_bitrate_bps;
1025 }
1026
perkj26091b12016-09-01 01:17:40 -07001027 rtc::CriticalSection crit_;
1028 TestEncoder* test_encoder_;
1029 rtc::Event encoded_frame_event_;
Mirta Dvornicic28f0eb22019-05-28 16:30:16 +02001030 std::vector<uint8_t> last_encoded_image_data_;
1031 RTPFragmentationHeader last_fragmentation_;
sprangb1ca0732017-02-01 08:38:12 -08001032 uint32_t last_timestamp_ = 0;
Erik Språngb7cb7b52019-02-26 15:52:33 +01001033 int64_t last_capture_time_ms_ = 0;
sprangb1ca0732017-02-01 08:38:12 -08001034 uint32_t last_height_ = 0;
1035 uint32_t last_width_ = 0;
Ilya Nikolaevskiyba96e2f2019-06-04 15:15:14 +02001036 VideoRotation last_rotation_ = kVideoRotation_0;
Erik Språngd7329ca2019-02-21 21:19:53 +01001037 size_t num_expected_layers_ = 1;
1038 size_t num_received_layers_ = 0;
perkj26091b12016-09-01 01:17:40 -07001039 bool expect_frames_ = true;
Per512ecb32016-09-23 15:52:06 +02001040 int number_of_reconfigurations_ = 0;
1041 int min_transmit_bitrate_bps_ = 0;
perkj26091b12016-09-01 01:17:40 -07001042 };
1043
Sergey Silkin5ee69672019-07-02 14:18:34 +02001044 class VideoBitrateAllocatorProxyFactory
1045 : public VideoBitrateAllocatorFactory {
1046 public:
1047 VideoBitrateAllocatorProxyFactory()
1048 : bitrate_allocator_factory_(
1049 CreateBuiltinVideoBitrateAllocatorFactory()) {}
1050
1051 std::unique_ptr<VideoBitrateAllocator> CreateVideoBitrateAllocator(
1052 const VideoCodec& codec) override {
1053 rtc::CritScope lock(&crit_);
1054 codec_config_ = codec;
1055 return bitrate_allocator_factory_->CreateVideoBitrateAllocator(codec);
1056 }
1057
1058 VideoCodec codec_config() const {
1059 rtc::CritScope lock(&crit_);
1060 return codec_config_;
1061 }
1062
1063 private:
1064 std::unique_ptr<VideoBitrateAllocatorFactory> bitrate_allocator_factory_;
1065
1066 rtc::CriticalSection crit_;
1067 VideoCodec codec_config_ RTC_GUARDED_BY(crit_);
1068 };
1069
perkj26091b12016-09-01 01:17:40 -07001070 VideoSendStream::Config video_send_config_;
Erik Språng08127a92016-11-16 16:41:30 +01001071 VideoEncoderConfig video_encoder_config_;
Per512ecb32016-09-23 15:52:06 +02001072 int codec_width_;
1073 int codec_height_;
sprang4847ae62017-06-27 07:06:52 -07001074 int max_framerate_;
Danil Chapovalovd3ba2362019-04-10 17:01:23 +02001075 const std::unique_ptr<TaskQueueFactory> task_queue_factory_;
perkj26091b12016-09-01 01:17:40 -07001076 TestEncoder fake_encoder_;
Niels Möllercbcbc222018-09-28 09:07:24 +02001077 test::VideoEncoderProxyFactory encoder_factory_;
Sergey Silkin5ee69672019-07-02 14:18:34 +02001078 VideoBitrateAllocatorProxyFactory bitrate_allocator_factory_;
sprangc5d62e22017-04-02 23:53:04 -07001079 std::unique_ptr<MockableSendStatisticsProxy> stats_proxy_;
perkj26091b12016-09-01 01:17:40 -07001080 TestSink sink_;
sprangb1ca0732017-02-01 08:38:12 -08001081 AdaptingFrameForwarder video_source_;
mflodmancc3d4422017-08-03 08:27:51 -07001082 std::unique_ptr<VideoStreamEncoderUnderTest> video_stream_encoder_;
sprang4847ae62017-06-27 07:06:52 -07001083 rtc::ScopedFakeClock fake_clock_;
perkj26091b12016-09-01 01:17:40 -07001084};
1085
mflodmancc3d4422017-08-03 08:27:51 -07001086TEST_F(VideoStreamEncoderTest, EncodeOneFrame) {
Erik Språng4c6ca302019-04-08 15:14:01 +02001087 video_stream_encoder_->OnBitrateUpdated(
1088 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
Niels Möllerc572ff32018-11-07 08:43:50 +01001089 rtc::Event frame_destroyed_event;
perkja49cbd32016-09-16 07:53:41 -07001090 video_source_.IncomingCapturedFrame(CreateFrame(1, &frame_destroyed_event));
sprang4847ae62017-06-27 07:06:52 -07001091 WaitForEncodedFrame(1);
perkja49cbd32016-09-16 07:53:41 -07001092 EXPECT_TRUE(frame_destroyed_event.Wait(kDefaultTimeoutMs));
mflodmancc3d4422017-08-03 08:27:51 -07001093 video_stream_encoder_->Stop();
perkj26091b12016-09-01 01:17:40 -07001094}
1095
mflodmancc3d4422017-08-03 08:27:51 -07001096TEST_F(VideoStreamEncoderTest, DropsFramesBeforeFirstOnBitrateUpdated) {
perkj26091b12016-09-01 01:17:40 -07001097 // Dropped since no target bitrate has been set.
Niels Möllerc572ff32018-11-07 08:43:50 +01001098 rtc::Event frame_destroyed_event;
Sebastian Janssona3177052018-04-10 13:05:49 +02001099 // The encoder will cache up to one frame for a short duration. Adding two
1100 // frames means that the first frame will be dropped and the second frame will
1101 // be sent when the encoder is enabled.
perkja49cbd32016-09-16 07:53:41 -07001102 video_source_.IncomingCapturedFrame(CreateFrame(1, &frame_destroyed_event));
Sebastian Janssona3177052018-04-10 13:05:49 +02001103 video_source_.IncomingCapturedFrame(CreateFrame(2, nullptr));
perkja49cbd32016-09-16 07:53:41 -07001104 EXPECT_TRUE(frame_destroyed_event.Wait(kDefaultTimeoutMs));
perkj26091b12016-09-01 01:17:40 -07001105
Erik Språng4c6ca302019-04-08 15:14:01 +02001106 video_stream_encoder_->OnBitrateUpdated(
1107 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
perkj26091b12016-09-01 01:17:40 -07001108
Sebastian Janssona3177052018-04-10 13:05:49 +02001109 // The pending frame should be received.
sprang4847ae62017-06-27 07:06:52 -07001110 WaitForEncodedFrame(2);
Sebastian Janssona3177052018-04-10 13:05:49 +02001111 video_source_.IncomingCapturedFrame(CreateFrame(3, nullptr));
1112
1113 WaitForEncodedFrame(3);
mflodmancc3d4422017-08-03 08:27:51 -07001114 video_stream_encoder_->Stop();
perkj26091b12016-09-01 01:17:40 -07001115}
1116
mflodmancc3d4422017-08-03 08:27:51 -07001117TEST_F(VideoStreamEncoderTest, DropsFramesWhenRateSetToZero) {
Erik Språng4c6ca302019-04-08 15:14:01 +02001118 video_stream_encoder_->OnBitrateUpdated(
1119 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
perkja49cbd32016-09-16 07:53:41 -07001120 video_source_.IncomingCapturedFrame(CreateFrame(1, nullptr));
sprang4847ae62017-06-27 07:06:52 -07001121 WaitForEncodedFrame(1);
perkj26091b12016-09-01 01:17:40 -07001122
Erik Språng4c6ca302019-04-08 15:14:01 +02001123 video_stream_encoder_->OnBitrateUpdated(DataRate::bps(0), DataRate::bps(0), 0,
Erik Språng610c7632019-03-06 15:37:33 +01001124 0);
Sebastian Janssona3177052018-04-10 13:05:49 +02001125 // The encoder will cache up to one frame for a short duration. Adding two
1126 // frames means that the first frame will be dropped and the second frame will
1127 // be sent when the encoder is resumed.
perkja49cbd32016-09-16 07:53:41 -07001128 video_source_.IncomingCapturedFrame(CreateFrame(2, nullptr));
Sebastian Janssona3177052018-04-10 13:05:49 +02001129 video_source_.IncomingCapturedFrame(CreateFrame(3, nullptr));
perkj26091b12016-09-01 01:17:40 -07001130
Erik Språng4c6ca302019-04-08 15:14:01 +02001131 video_stream_encoder_->OnBitrateUpdated(
1132 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
sprang4847ae62017-06-27 07:06:52 -07001133 WaitForEncodedFrame(3);
Sebastian Janssona3177052018-04-10 13:05:49 +02001134 video_source_.IncomingCapturedFrame(CreateFrame(4, nullptr));
1135 WaitForEncodedFrame(4);
mflodmancc3d4422017-08-03 08:27:51 -07001136 video_stream_encoder_->Stop();
perkj26091b12016-09-01 01:17:40 -07001137}
1138
mflodmancc3d4422017-08-03 08:27:51 -07001139TEST_F(VideoStreamEncoderTest, DropsFramesWithSameOrOldNtpTimestamp) {
Erik Språng4c6ca302019-04-08 15:14:01 +02001140 video_stream_encoder_->OnBitrateUpdated(
1141 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
perkja49cbd32016-09-16 07:53:41 -07001142 video_source_.IncomingCapturedFrame(CreateFrame(1, nullptr));
sprang4847ae62017-06-27 07:06:52 -07001143 WaitForEncodedFrame(1);
perkj26091b12016-09-01 01:17:40 -07001144
1145 // This frame will be dropped since it has the same ntp timestamp.
perkja49cbd32016-09-16 07:53:41 -07001146 video_source_.IncomingCapturedFrame(CreateFrame(1, nullptr));
perkj26091b12016-09-01 01:17:40 -07001147
perkja49cbd32016-09-16 07:53:41 -07001148 video_source_.IncomingCapturedFrame(CreateFrame(2, nullptr));
sprang4847ae62017-06-27 07:06:52 -07001149 WaitForEncodedFrame(2);
mflodmancc3d4422017-08-03 08:27:51 -07001150 video_stream_encoder_->Stop();
perkj26091b12016-09-01 01:17:40 -07001151}
1152
mflodmancc3d4422017-08-03 08:27:51 -07001153TEST_F(VideoStreamEncoderTest, DropsFrameAfterStop) {
Erik Språng4c6ca302019-04-08 15:14:01 +02001154 video_stream_encoder_->OnBitrateUpdated(
1155 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
perkj26091b12016-09-01 01:17:40 -07001156
perkja49cbd32016-09-16 07:53:41 -07001157 video_source_.IncomingCapturedFrame(CreateFrame(1, nullptr));
sprang4847ae62017-06-27 07:06:52 -07001158 WaitForEncodedFrame(1);
perkj26091b12016-09-01 01:17:40 -07001159
mflodmancc3d4422017-08-03 08:27:51 -07001160 video_stream_encoder_->Stop();
perkj26091b12016-09-01 01:17:40 -07001161 sink_.SetExpectNoFrames();
Niels Möllerc572ff32018-11-07 08:43:50 +01001162 rtc::Event frame_destroyed_event;
perkja49cbd32016-09-16 07:53:41 -07001163 video_source_.IncomingCapturedFrame(CreateFrame(2, &frame_destroyed_event));
1164 EXPECT_TRUE(frame_destroyed_event.Wait(kDefaultTimeoutMs));
perkj26091b12016-09-01 01:17:40 -07001165}
1166
mflodmancc3d4422017-08-03 08:27:51 -07001167TEST_F(VideoStreamEncoderTest, DropsPendingFramesOnSlowEncode) {
Erik Språng4c6ca302019-04-08 15:14:01 +02001168 video_stream_encoder_->OnBitrateUpdated(
1169 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
perkj26091b12016-09-01 01:17:40 -07001170
1171 fake_encoder_.BlockNextEncode();
perkja49cbd32016-09-16 07:53:41 -07001172 video_source_.IncomingCapturedFrame(CreateFrame(1, nullptr));
sprang4847ae62017-06-27 07:06:52 -07001173 WaitForEncodedFrame(1);
perkj26091b12016-09-01 01:17:40 -07001174 // Here, the encoder thread will be blocked in the TestEncoder waiting for a
1175 // call to ContinueEncode.
perkja49cbd32016-09-16 07:53:41 -07001176 video_source_.IncomingCapturedFrame(CreateFrame(2, nullptr));
1177 video_source_.IncomingCapturedFrame(CreateFrame(3, nullptr));
perkj26091b12016-09-01 01:17:40 -07001178 fake_encoder_.ContinueEncode();
sprang4847ae62017-06-27 07:06:52 -07001179 WaitForEncodedFrame(3);
perkj26091b12016-09-01 01:17:40 -07001180
mflodmancc3d4422017-08-03 08:27:51 -07001181 video_stream_encoder_->Stop();
perkj26091b12016-09-01 01:17:40 -07001182}
1183
Noah Richards51db4212019-06-12 06:59:12 -07001184TEST_F(VideoStreamEncoderTest, DropFrameWithFailedI420Conversion) {
1185 video_stream_encoder_->OnBitrateUpdated(
1186 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
1187
1188 rtc::Event frame_destroyed_event;
1189 video_source_.IncomingCapturedFrame(
1190 CreateFakeNativeFrame(1, &frame_destroyed_event));
1191 ExpectDroppedFrame();
1192 EXPECT_TRUE(frame_destroyed_event.Wait(kDefaultTimeoutMs));
1193 video_stream_encoder_->Stop();
1194}
1195
1196TEST_F(VideoStreamEncoderTest, DropFrameWithFailedI420ConversionWithCrop) {
1197 // Use the cropping factory.
1198 video_encoder_config_.video_stream_factory =
1199 new rtc::RefCountedObject<CroppingVideoStreamFactory>(1, 30);
1200 video_stream_encoder_->ConfigureEncoder(std::move(video_encoder_config_),
1201 kMaxPayloadLength);
1202 video_stream_encoder_->WaitUntilTaskQueueIsIdle();
1203
1204 // Capture a frame at codec_width_/codec_height_.
1205 video_stream_encoder_->OnBitrateUpdated(
1206 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
1207 video_source_.IncomingCapturedFrame(CreateFrame(1, nullptr));
1208 WaitForEncodedFrame(1);
1209 // The encoder will have been configured once.
1210 EXPECT_EQ(1, sink_.number_of_reconfigurations());
1211 EXPECT_EQ(codec_width_, fake_encoder_.codec_config().width);
1212 EXPECT_EQ(codec_height_, fake_encoder_.codec_config().height);
1213
1214 // Now send in a fake frame that needs to be cropped as the width/height
1215 // aren't divisible by 4 (see CreateEncoderStreams above).
1216 rtc::Event frame_destroyed_event;
1217 video_source_.IncomingCapturedFrame(CreateFakeNativeFrame(
1218 2, &frame_destroyed_event, codec_width_ + 1, codec_height_ + 1));
1219 ExpectDroppedFrame();
1220 EXPECT_TRUE(frame_destroyed_event.Wait(kDefaultTimeoutMs));
1221 video_stream_encoder_->Stop();
1222}
1223
mflodmancc3d4422017-08-03 08:27:51 -07001224TEST_F(VideoStreamEncoderTest,
1225 ConfigureEncoderTriggersOnEncoderConfigurationChanged) {
Erik Språng4c6ca302019-04-08 15:14:01 +02001226 video_stream_encoder_->OnBitrateUpdated(
1227 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
Per21d45d22016-10-30 21:37:57 +01001228 EXPECT_EQ(0, sink_.number_of_reconfigurations());
Per512ecb32016-09-23 15:52:06 +02001229
1230 // Capture a frame and wait for it to synchronize with the encoder thread.
Perf8c5f2b2016-09-23 16:24:55 +02001231 video_source_.IncomingCapturedFrame(CreateFrame(1, nullptr));
sprang4847ae62017-06-27 07:06:52 -07001232 WaitForEncodedFrame(1);
Per21d45d22016-10-30 21:37:57 +01001233 // The encoder will have been configured once when the first frame is
1234 // received.
1235 EXPECT_EQ(1, sink_.number_of_reconfigurations());
Per512ecb32016-09-23 15:52:06 +02001236
1237 VideoEncoderConfig video_encoder_config;
Niels Möller259a4972018-04-05 15:36:51 +02001238 test::FillEncoderConfiguration(kVideoCodecVP8, 1, &video_encoder_config);
Per512ecb32016-09-23 15:52:06 +02001239 video_encoder_config.min_transmit_bitrate_bps = 9999;
mflodmancc3d4422017-08-03 08:27:51 -07001240 video_stream_encoder_->ConfigureEncoder(std::move(video_encoder_config),
Niels Möllerf1338562018-04-26 09:51:47 +02001241 kMaxPayloadLength);
Per512ecb32016-09-23 15:52:06 +02001242
1243 // Capture a frame and wait for it to synchronize with the encoder thread.
Perf8c5f2b2016-09-23 16:24:55 +02001244 video_source_.IncomingCapturedFrame(CreateFrame(2, nullptr));
sprang4847ae62017-06-27 07:06:52 -07001245 WaitForEncodedFrame(2);
Per21d45d22016-10-30 21:37:57 +01001246 EXPECT_EQ(2, sink_.number_of_reconfigurations());
perkj3b703ed2016-09-29 23:25:40 -07001247 EXPECT_EQ(9999, sink_.last_min_transmit_bitrate());
perkj26105b42016-09-29 22:39:10 -07001248
mflodmancc3d4422017-08-03 08:27:51 -07001249 video_stream_encoder_->Stop();
perkj26105b42016-09-29 22:39:10 -07001250}
1251
mflodmancc3d4422017-08-03 08:27:51 -07001252TEST_F(VideoStreamEncoderTest, FrameResolutionChangeReconfigureEncoder) {
Erik Språng4c6ca302019-04-08 15:14:01 +02001253 video_stream_encoder_->OnBitrateUpdated(
1254 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
perkjfa10b552016-10-02 23:45:26 -07001255
1256 // Capture a frame and wait for it to synchronize with the encoder thread.
1257 video_source_.IncomingCapturedFrame(CreateFrame(1, nullptr));
sprang4847ae62017-06-27 07:06:52 -07001258 WaitForEncodedFrame(1);
Per21d45d22016-10-30 21:37:57 +01001259 // The encoder will have been configured once.
1260 EXPECT_EQ(1, sink_.number_of_reconfigurations());
perkjfa10b552016-10-02 23:45:26 -07001261 EXPECT_EQ(codec_width_, fake_encoder_.codec_config().width);
1262 EXPECT_EQ(codec_height_, fake_encoder_.codec_config().height);
1263
1264 codec_width_ *= 2;
1265 codec_height_ *= 2;
1266 // Capture a frame with a higher resolution and wait for it to synchronize
1267 // with the encoder thread.
1268 video_source_.IncomingCapturedFrame(CreateFrame(2, nullptr));
sprang4847ae62017-06-27 07:06:52 -07001269 WaitForEncodedFrame(2);
perkjfa10b552016-10-02 23:45:26 -07001270 EXPECT_EQ(codec_width_, fake_encoder_.codec_config().width);
1271 EXPECT_EQ(codec_height_, fake_encoder_.codec_config().height);
Per21d45d22016-10-30 21:37:57 +01001272 EXPECT_EQ(2, sink_.number_of_reconfigurations());
perkjfa10b552016-10-02 23:45:26 -07001273
mflodmancc3d4422017-08-03 08:27:51 -07001274 video_stream_encoder_->Stop();
perkjfa10b552016-10-02 23:45:26 -07001275}
1276
Sergey Silkin443b7ee2019-06-28 12:53:07 +02001277TEST_F(VideoStreamEncoderTest,
1278 EncoderInstanceDestroyedBeforeAnotherInstanceCreated) {
1279 video_stream_encoder_->OnBitrateUpdated(
1280 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
1281
1282 // Capture a frame and wait for it to synchronize with the encoder thread.
1283 video_source_.IncomingCapturedFrame(CreateFrame(1, nullptr));
1284 WaitForEncodedFrame(1);
1285
1286 VideoEncoderConfig video_encoder_config;
1287 test::FillEncoderConfiguration(kVideoCodecVP8, 1, &video_encoder_config);
1288 // Changing the max payload data length recreates encoder.
1289 video_stream_encoder_->ConfigureEncoder(std::move(video_encoder_config),
1290 kMaxPayloadLength / 2);
1291
1292 // Capture a frame and wait for it to synchronize with the encoder thread.
1293 video_source_.IncomingCapturedFrame(CreateFrame(2, nullptr));
1294 WaitForEncodedFrame(2);
1295 EXPECT_EQ(1, encoder_factory_.GetMaxNumberOfSimultaneousEncoderInstances());
1296
1297 video_stream_encoder_->Stop();
1298}
1299
Sergey Silkin5ee69672019-07-02 14:18:34 +02001300TEST_F(VideoStreamEncoderTest, BitrateLimitsChangeReconfigureRateAllocator) {
1301 video_stream_encoder_->OnBitrateUpdated(
1302 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
1303
1304 VideoEncoderConfig video_encoder_config;
1305 test::FillEncoderConfiguration(kVideoCodecVP8, 1, &video_encoder_config);
1306 video_encoder_config.max_bitrate_bps = kTargetBitrateBps;
1307 video_stream_encoder_->SetStartBitrate(kStartBitrateBps);
1308 video_stream_encoder_->ConfigureEncoder(video_encoder_config.Copy(),
1309 kMaxPayloadLength);
1310
1311 // Capture a frame and wait for it to synchronize with the encoder thread.
1312 video_source_.IncomingCapturedFrame(CreateFrame(1, nullptr));
1313 WaitForEncodedFrame(1);
1314 // The encoder will have been configured once when the first frame is
1315 // received.
1316 EXPECT_EQ(1, sink_.number_of_reconfigurations());
1317 EXPECT_EQ(kTargetBitrateBps,
1318 bitrate_allocator_factory_.codec_config().maxBitrate * 1000);
1319 EXPECT_EQ(kStartBitrateBps,
1320 bitrate_allocator_factory_.codec_config().startBitrate * 1000);
1321
Sergey Silkin6456e352019-07-08 17:56:40 +02001322 test::FillEncoderConfiguration(kVideoCodecVP8, 1,
1323 &video_encoder_config); //???
Sergey Silkin5ee69672019-07-02 14:18:34 +02001324 video_encoder_config.max_bitrate_bps = kTargetBitrateBps * 2;
1325 video_stream_encoder_->SetStartBitrate(kStartBitrateBps * 2);
1326 video_stream_encoder_->ConfigureEncoder(std::move(video_encoder_config),
1327 kMaxPayloadLength);
1328
1329 // Capture a frame and wait for it to synchronize with the encoder thread.
1330 video_source_.IncomingCapturedFrame(CreateFrame(2, nullptr));
1331 WaitForEncodedFrame(2);
1332 EXPECT_EQ(2, sink_.number_of_reconfigurations());
1333 // Bitrate limits have changed - rate allocator should be reconfigured,
1334 // encoder should not be reconfigured.
1335 EXPECT_EQ(kTargetBitrateBps * 2,
1336 bitrate_allocator_factory_.codec_config().maxBitrate * 1000);
1337 EXPECT_EQ(kStartBitrateBps * 2,
1338 bitrate_allocator_factory_.codec_config().startBitrate * 1000);
1339 EXPECT_EQ(1, fake_encoder_.GetNumEncoderInitializations());
1340
1341 video_stream_encoder_->Stop();
1342}
1343
Sergey Silkin6456e352019-07-08 17:56:40 +02001344TEST_F(VideoStreamEncoderTest,
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001345 EncoderRecommendedBitrateLimitsDoNotOverrideAppBitrateLimits) {
Sergey Silkin6456e352019-07-08 17:56:40 +02001346 video_stream_encoder_->OnBitrateUpdated(
1347 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
1348
Sergey Silkin6456e352019-07-08 17:56:40 +02001349 VideoEncoderConfig video_encoder_config;
1350 test::FillEncoderConfiguration(kVideoCodecVP8, 1, &video_encoder_config);
1351 video_encoder_config.max_bitrate_bps = 0;
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001352 video_encoder_config.simulcast_layers[0].min_bitrate_bps = 0;
Sergey Silkin6456e352019-07-08 17:56:40 +02001353 video_stream_encoder_->ConfigureEncoder(video_encoder_config.Copy(),
1354 kMaxPayloadLength);
1355
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001356 video_source_.IncomingCapturedFrame(CreateFrame(1, 360, 180));
Sergey Silkin6456e352019-07-08 17:56:40 +02001357 WaitForEncodedFrame(1);
Sergey Silkin6456e352019-07-08 17:56:40 +02001358
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001359 // Get the default bitrate limits and use them as baseline for custom
1360 // application and encoder recommended limits.
1361 const uint32_t kDefaultMinBitrateKbps =
1362 bitrate_allocator_factory_.codec_config().minBitrate;
1363 const uint32_t kDefaultMaxBitrateKbps =
1364 bitrate_allocator_factory_.codec_config().maxBitrate;
1365 const uint32_t kEncMinBitrateKbps = kDefaultMinBitrateKbps * 2;
1366 const uint32_t kEncMaxBitrateKbps = kDefaultMaxBitrateKbps * 2;
1367 const uint32_t kAppMinBitrateKbps = kDefaultMinBitrateKbps * 3;
1368 const uint32_t kAppMaxBitrateKbps = kDefaultMaxBitrateKbps * 3;
1369
1370 const VideoEncoder::ResolutionBitrateLimits encoder_bitrate_limits(
1371 codec_width_ * codec_height_, kEncMinBitrateKbps * 1000,
1372 kEncMinBitrateKbps * 1000, kEncMaxBitrateKbps * 1000);
1373 fake_encoder_.SetResolutionBitrateLimits({encoder_bitrate_limits});
1374
1375 // Change resolution. This will trigger encoder re-configuration and video
1376 // stream encoder will pick up the bitrate limits recommended by encoder.
1377 video_source_.IncomingCapturedFrame(CreateFrame(2, 640, 360));
1378 WaitForEncodedFrame(2);
1379 video_source_.IncomingCapturedFrame(CreateFrame(3, 360, 180));
1380 WaitForEncodedFrame(3);
1381
1382 // App bitrate limits are not set - bitrate limits recommended by encoder
1383 // should be used.
1384 EXPECT_EQ(kEncMaxBitrateKbps,
1385 bitrate_allocator_factory_.codec_config().maxBitrate);
1386 EXPECT_EQ(kEncMinBitrateKbps,
1387 bitrate_allocator_factory_.codec_config().minBitrate);
1388
1389 video_encoder_config.max_bitrate_bps = kAppMaxBitrateKbps * 1000;
1390 video_encoder_config.simulcast_layers[0].min_bitrate_bps = 0;
1391 video_stream_encoder_->ConfigureEncoder(video_encoder_config.Copy(),
1392 kMaxPayloadLength);
1393 video_source_.IncomingCapturedFrame(CreateFrame(4, nullptr));
1394 WaitForEncodedFrame(4);
1395
1396 // App limited the max bitrate - bitrate limits recommended by encoder should
1397 // not be applied.
1398 EXPECT_EQ(kAppMaxBitrateKbps,
1399 bitrate_allocator_factory_.codec_config().maxBitrate);
1400 EXPECT_EQ(kDefaultMinBitrateKbps,
1401 bitrate_allocator_factory_.codec_config().minBitrate);
1402
1403 video_encoder_config.max_bitrate_bps = 0;
1404 video_encoder_config.simulcast_layers[0].min_bitrate_bps =
1405 kAppMinBitrateKbps * 1000;
1406 video_stream_encoder_->ConfigureEncoder(video_encoder_config.Copy(),
1407 kMaxPayloadLength);
1408 video_source_.IncomingCapturedFrame(CreateFrame(5, nullptr));
1409 WaitForEncodedFrame(5);
1410
1411 // App limited the min bitrate - bitrate limits recommended by encoder should
1412 // not be applied.
1413 EXPECT_EQ(kDefaultMaxBitrateKbps,
1414 bitrate_allocator_factory_.codec_config().maxBitrate);
1415 EXPECT_EQ(kAppMinBitrateKbps,
1416 bitrate_allocator_factory_.codec_config().minBitrate);
1417
1418 video_encoder_config.max_bitrate_bps = kAppMaxBitrateKbps * 1000;
1419 video_encoder_config.simulcast_layers[0].min_bitrate_bps =
1420 kAppMinBitrateKbps * 1000;
Sergey Silkin6456e352019-07-08 17:56:40 +02001421 video_stream_encoder_->ConfigureEncoder(std::move(video_encoder_config),
1422 kMaxPayloadLength);
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001423 video_source_.IncomingCapturedFrame(CreateFrame(6, nullptr));
1424 WaitForEncodedFrame(6);
Sergey Silkin6456e352019-07-08 17:56:40 +02001425
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001426 // App limited both min and max bitrates - bitrate limits recommended by
1427 // encoder should not be applied.
1428 EXPECT_EQ(kAppMaxBitrateKbps,
1429 bitrate_allocator_factory_.codec_config().maxBitrate);
1430 EXPECT_EQ(kAppMinBitrateKbps,
1431 bitrate_allocator_factory_.codec_config().minBitrate);
Sergey Silkin6456e352019-07-08 17:56:40 +02001432
1433 video_stream_encoder_->Stop();
1434}
1435
1436TEST_F(VideoStreamEncoderTest,
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001437 EncoderRecommendedMaxAndMinBitratesUsedForGivenResolution) {
Sergey Silkin6456e352019-07-08 17:56:40 +02001438 video_stream_encoder_->OnBitrateUpdated(
1439 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
1440
1441 const VideoEncoder::ResolutionBitrateLimits encoder_bitrate_limits_270p(
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001442 480 * 270, 34 * 1000, 12 * 1000, 1234 * 1000);
Sergey Silkin6456e352019-07-08 17:56:40 +02001443 const VideoEncoder::ResolutionBitrateLimits encoder_bitrate_limits_360p(
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001444 640 * 360, 43 * 1000, 21 * 1000, 2345 * 1000);
Sergey Silkin6456e352019-07-08 17:56:40 +02001445 fake_encoder_.SetResolutionBitrateLimits(
1446 {encoder_bitrate_limits_270p, encoder_bitrate_limits_360p});
1447
1448 VideoEncoderConfig video_encoder_config;
1449 test::FillEncoderConfiguration(kVideoCodecVP8, 1, &video_encoder_config);
1450 video_encoder_config.max_bitrate_bps = 0;
1451 video_stream_encoder_->ConfigureEncoder(video_encoder_config.Copy(),
1452 kMaxPayloadLength);
1453
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001454 // 270p. The bitrate limits recommended by encoder for 270p should be used.
Sergey Silkin6456e352019-07-08 17:56:40 +02001455 video_source_.IncomingCapturedFrame(CreateFrame(1, 480, 270));
1456 WaitForEncodedFrame(1);
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001457 EXPECT_EQ(static_cast<uint32_t>(encoder_bitrate_limits_270p.min_bitrate_bps),
1458 bitrate_allocator_factory_.codec_config().minBitrate * 1000);
Sergey Silkin6456e352019-07-08 17:56:40 +02001459 EXPECT_EQ(static_cast<uint32_t>(encoder_bitrate_limits_270p.max_bitrate_bps),
1460 bitrate_allocator_factory_.codec_config().maxBitrate * 1000);
1461
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001462 // 360p. The bitrate limits recommended by encoder for 360p should be used.
Sergey Silkin6456e352019-07-08 17:56:40 +02001463 video_source_.IncomingCapturedFrame(CreateFrame(2, 640, 360));
1464 WaitForEncodedFrame(2);
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001465 EXPECT_EQ(static_cast<uint32_t>(encoder_bitrate_limits_360p.min_bitrate_bps),
1466 bitrate_allocator_factory_.codec_config().minBitrate * 1000);
Sergey Silkin6456e352019-07-08 17:56:40 +02001467 EXPECT_EQ(static_cast<uint32_t>(encoder_bitrate_limits_360p.max_bitrate_bps),
1468 bitrate_allocator_factory_.codec_config().maxBitrate * 1000);
1469
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001470 // Resolution between 270p and 360p. The bitrate limits recommended by
Sergey Silkin6456e352019-07-08 17:56:40 +02001471 // encoder for 360p should be used.
1472 video_source_.IncomingCapturedFrame(
1473 CreateFrame(3, (640 + 480) / 2, (360 + 270) / 2));
1474 WaitForEncodedFrame(3);
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001475 EXPECT_EQ(static_cast<uint32_t>(encoder_bitrate_limits_360p.min_bitrate_bps),
1476 bitrate_allocator_factory_.codec_config().minBitrate * 1000);
Sergey Silkin6456e352019-07-08 17:56:40 +02001477 EXPECT_EQ(static_cast<uint32_t>(encoder_bitrate_limits_360p.max_bitrate_bps),
1478 bitrate_allocator_factory_.codec_config().maxBitrate * 1000);
1479
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001480 // Resolution higher than 360p. The caps recommended by encoder should be
Sergey Silkin6456e352019-07-08 17:56:40 +02001481 // ignored.
1482 video_source_.IncomingCapturedFrame(CreateFrame(4, 960, 540));
1483 WaitForEncodedFrame(4);
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001484 EXPECT_NE(static_cast<uint32_t>(encoder_bitrate_limits_270p.min_bitrate_bps),
1485 bitrate_allocator_factory_.codec_config().minBitrate * 1000);
Sergey Silkin6456e352019-07-08 17:56:40 +02001486 EXPECT_NE(static_cast<uint32_t>(encoder_bitrate_limits_270p.max_bitrate_bps),
1487 bitrate_allocator_factory_.codec_config().maxBitrate * 1000);
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001488 EXPECT_NE(static_cast<uint32_t>(encoder_bitrate_limits_360p.min_bitrate_bps),
1489 bitrate_allocator_factory_.codec_config().minBitrate * 1000);
Sergey Silkin6456e352019-07-08 17:56:40 +02001490 EXPECT_NE(static_cast<uint32_t>(encoder_bitrate_limits_360p.max_bitrate_bps),
1491 bitrate_allocator_factory_.codec_config().maxBitrate * 1000);
1492
1493 // Resolution lower than 270p. The max bitrate limit recommended by encoder
1494 // for 270p should be used.
1495 video_source_.IncomingCapturedFrame(CreateFrame(5, 320, 180));
1496 WaitForEncodedFrame(5);
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001497 EXPECT_EQ(static_cast<uint32_t>(encoder_bitrate_limits_270p.min_bitrate_bps),
1498 bitrate_allocator_factory_.codec_config().minBitrate * 1000);
Sergey Silkin6456e352019-07-08 17:56:40 +02001499 EXPECT_EQ(static_cast<uint32_t>(encoder_bitrate_limits_270p.max_bitrate_bps),
1500 bitrate_allocator_factory_.codec_config().maxBitrate * 1000);
1501
1502 video_stream_encoder_->Stop();
1503}
1504
Sergey Silkin6b2cec12019-08-09 16:04:05 +02001505TEST_F(VideoStreamEncoderTest, EncoderRecommendedMaxBitrateCapsTargetBitrate) {
1506 video_stream_encoder_->OnBitrateUpdated(
1507 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
1508
1509 VideoEncoderConfig video_encoder_config;
1510 test::FillEncoderConfiguration(kVideoCodecVP8, 1, &video_encoder_config);
1511 video_encoder_config.max_bitrate_bps = 0;
1512 video_stream_encoder_->ConfigureEncoder(video_encoder_config.Copy(),
1513 kMaxPayloadLength);
1514
1515 // Encode 720p frame to get the default encoder target bitrate.
1516 video_source_.IncomingCapturedFrame(CreateFrame(1, 1280, 720));
1517 WaitForEncodedFrame(1);
1518 const uint32_t kDefaultTargetBitrateFor720pKbps =
1519 bitrate_allocator_factory_.codec_config()
1520 .simulcastStream[0]
1521 .targetBitrate;
1522
1523 // Set the max recommended encoder bitrate to something lower than the default
1524 // target bitrate.
1525 const VideoEncoder::ResolutionBitrateLimits encoder_bitrate_limits(
1526 1280 * 720, 10 * 1000, 10 * 1000,
1527 kDefaultTargetBitrateFor720pKbps / 2 * 1000);
1528 fake_encoder_.SetResolutionBitrateLimits({encoder_bitrate_limits});
1529
1530 // Change resolution to trigger encoder reinitialization.
1531 video_source_.IncomingCapturedFrame(CreateFrame(2, 640, 360));
1532 WaitForEncodedFrame(2);
1533 video_source_.IncomingCapturedFrame(CreateFrame(3, 1280, 720));
1534 WaitForEncodedFrame(3);
1535
1536 // Ensure the target bitrate is capped by the max bitrate.
1537 EXPECT_EQ(bitrate_allocator_factory_.codec_config().maxBitrate * 1000,
1538 static_cast<uint32_t>(encoder_bitrate_limits.max_bitrate_bps));
1539 EXPECT_EQ(bitrate_allocator_factory_.codec_config()
1540 .simulcastStream[0]
1541 .targetBitrate *
1542 1000,
1543 static_cast<uint32_t>(encoder_bitrate_limits.max_bitrate_bps));
1544
1545 video_stream_encoder_->Stop();
1546}
1547
mflodmancc3d4422017-08-03 08:27:51 -07001548TEST_F(VideoStreamEncoderTest, SwitchSourceDeregisterEncoderAsSink) {
perkj803d97f2016-11-01 11:45:46 -07001549 EXPECT_TRUE(video_source_.has_sinks());
1550 test::FrameForwarder new_video_source;
mflodmancc3d4422017-08-03 08:27:51 -07001551 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07001552 &new_video_source, webrtc::DegradationPreference::MAINTAIN_FRAMERATE);
perkj803d97f2016-11-01 11:45:46 -07001553 EXPECT_FALSE(video_source_.has_sinks());
1554 EXPECT_TRUE(new_video_source.has_sinks());
1555
mflodmancc3d4422017-08-03 08:27:51 -07001556 video_stream_encoder_->Stop();
perkj803d97f2016-11-01 11:45:46 -07001557}
1558
mflodmancc3d4422017-08-03 08:27:51 -07001559TEST_F(VideoStreamEncoderTest, SinkWantsRotationApplied) {
perkj803d97f2016-11-01 11:45:46 -07001560 EXPECT_FALSE(video_source_.sink_wants().rotation_applied);
mflodmancc3d4422017-08-03 08:27:51 -07001561 video_stream_encoder_->SetSink(&sink_, true /*rotation_applied*/);
perkj803d97f2016-11-01 11:45:46 -07001562 EXPECT_TRUE(video_source_.sink_wants().rotation_applied);
mflodmancc3d4422017-08-03 08:27:51 -07001563 video_stream_encoder_->Stop();
perkj803d97f2016-11-01 11:45:46 -07001564}
1565
Jonathan Yubc771b72017-12-08 17:04:29 -08001566TEST_F(VideoStreamEncoderTest, TestCpuDowngrades_BalancedMode) {
1567 const int kFramerateFps = 30;
asaperssonf7e294d2017-06-13 23:25:22 -07001568 const int kWidth = 1280;
1569 const int kHeight = 720;
Jonathan Yubc771b72017-12-08 17:04:29 -08001570
1571 // We rely on the automatic resolution adaptation, but we handle framerate
1572 // adaptation manually by mocking the stats proxy.
1573 video_source_.set_adaptation_enabled(true);
asaperssonf7e294d2017-06-13 23:25:22 -07001574
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07001575 // Enable BALANCED preference, no initial limitation.
Erik Språng4c6ca302019-04-08 15:14:01 +02001576 video_stream_encoder_->OnBitrateUpdated(
1577 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07001578 video_stream_encoder_->SetSource(&video_source_,
1579 webrtc::DegradationPreference::BALANCED);
Jonathan Yubc771b72017-12-08 17:04:29 -08001580 VerifyNoLimitation(video_source_.sink_wants());
asaperssonf7e294d2017-06-13 23:25:22 -07001581 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
Jonathan Yubc771b72017-12-08 17:04:29 -08001582 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_framerate);
asaperssonf7e294d2017-06-13 23:25:22 -07001583 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
1584
Jonathan Yubc771b72017-12-08 17:04:29 -08001585 // Adapt down as far as possible.
1586 rtc::VideoSinkWants last_wants;
1587 int64_t t = 1;
1588 int loop_count = 0;
1589 do {
1590 ++loop_count;
1591 last_wants = video_source_.sink_wants();
1592
1593 // Simulate the framerate we've been asked to adapt to.
1594 const int fps = std::min(kFramerateFps, last_wants.max_framerate_fps);
1595 const int frame_interval_ms = rtc::kNumMillisecsPerSec / fps;
1596 VideoSendStream::Stats mock_stats = stats_proxy_->GetStats();
1597 mock_stats.input_frame_rate = fps;
1598 stats_proxy_->SetMockStats(mock_stats);
1599
1600 video_source_.IncomingCapturedFrame(CreateFrame(t, kWidth, kHeight));
1601 sink_.WaitForEncodedFrame(t);
1602 t += frame_interval_ms;
1603
mflodmancc3d4422017-08-03 08:27:51 -07001604 video_stream_encoder_->TriggerCpuOveruse();
Jonathan Yubc771b72017-12-08 17:04:29 -08001605 VerifyBalancedModeFpsRange(
1606 video_source_.sink_wants(),
1607 *video_source_.last_sent_width() * *video_source_.last_sent_height());
1608 } while (video_source_.sink_wants().max_pixel_count <
1609 last_wants.max_pixel_count ||
1610 video_source_.sink_wants().max_framerate_fps <
1611 last_wants.max_framerate_fps);
asaperssonf7e294d2017-06-13 23:25:22 -07001612
Jonathan Yubc771b72017-12-08 17:04:29 -08001613 // Verify that we've adapted all the way down.
1614 stats_proxy_->ResetMockStats();
asaperssonf7e294d2017-06-13 23:25:22 -07001615 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_resolution);
Jonathan Yubc771b72017-12-08 17:04:29 -08001616 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_framerate);
1617 EXPECT_EQ(loop_count - 1,
asaperssonf7e294d2017-06-13 23:25:22 -07001618 stats_proxy_->GetStats().number_of_cpu_adapt_changes);
Jonathan Yubc771b72017-12-08 17:04:29 -08001619 EXPECT_EQ(kMinPixelsPerFrame, *video_source_.last_sent_width() *
1620 *video_source_.last_sent_height());
1621 EXPECT_EQ(kMinBalancedFramerateFps,
1622 video_source_.sink_wants().max_framerate_fps);
asaperssonf7e294d2017-06-13 23:25:22 -07001623
Jonathan Yubc771b72017-12-08 17:04:29 -08001624 // Adapt back up the same number of times we adapted down.
1625 for (int i = 0; i < loop_count - 1; ++i) {
1626 last_wants = video_source_.sink_wants();
1627
1628 // Simulate the framerate we've been asked to adapt to.
1629 const int fps = std::min(kFramerateFps, last_wants.max_framerate_fps);
1630 const int frame_interval_ms = rtc::kNumMillisecsPerSec / fps;
1631 VideoSendStream::Stats mock_stats = stats_proxy_->GetStats();
1632 mock_stats.input_frame_rate = fps;
1633 stats_proxy_->SetMockStats(mock_stats);
1634
1635 video_source_.IncomingCapturedFrame(CreateFrame(t, kWidth, kHeight));
1636 sink_.WaitForEncodedFrame(t);
1637 t += frame_interval_ms;
1638
mflodmancc3d4422017-08-03 08:27:51 -07001639 video_stream_encoder_->TriggerCpuNormalUsage();
Jonathan Yubc771b72017-12-08 17:04:29 -08001640 VerifyBalancedModeFpsRange(
1641 video_source_.sink_wants(),
1642 *video_source_.last_sent_width() * *video_source_.last_sent_height());
1643 EXPECT_TRUE(video_source_.sink_wants().max_pixel_count >
1644 last_wants.max_pixel_count ||
1645 video_source_.sink_wants().max_framerate_fps >
1646 last_wants.max_framerate_fps);
asaperssonf7e294d2017-06-13 23:25:22 -07001647 }
1648
Åsa Persson8c1bf952018-09-13 10:42:19 +02001649 VerifyFpsMaxResolutionMax(video_source_.sink_wants());
Jonathan Yubc771b72017-12-08 17:04:29 -08001650 stats_proxy_->ResetMockStats();
asaperssonf7e294d2017-06-13 23:25:22 -07001651 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
Jonathan Yubc771b72017-12-08 17:04:29 -08001652 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_framerate);
1653 EXPECT_EQ((loop_count - 1) * 2,
1654 stats_proxy_->GetStats().number_of_cpu_adapt_changes);
asaperssonf7e294d2017-06-13 23:25:22 -07001655
mflodmancc3d4422017-08-03 08:27:51 -07001656 video_stream_encoder_->Stop();
asaperssonf7e294d2017-06-13 23:25:22 -07001657}
mflodmancc3d4422017-08-03 08:27:51 -07001658TEST_F(VideoStreamEncoderTest, SinkWantsStoredByDegradationPreference) {
Erik Språng4c6ca302019-04-08 15:14:01 +02001659 video_stream_encoder_->OnBitrateUpdated(
1660 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asapersson02465b82017-04-10 01:12:52 -07001661 VerifyNoLimitation(video_source_.sink_wants());
perkj803d97f2016-11-01 11:45:46 -07001662
sprangc5d62e22017-04-02 23:53:04 -07001663 const int kFrameWidth = 1280;
1664 const int kFrameHeight = 720;
sprangc5d62e22017-04-02 23:53:04 -07001665
Åsa Persson8c1bf952018-09-13 10:42:19 +02001666 int64_t frame_timestamp = 1;
perkj803d97f2016-11-01 11:45:46 -07001667
kthelgason5e13d412016-12-01 03:59:51 -08001668 video_source_.IncomingCapturedFrame(
sprangc5d62e22017-04-02 23:53:04 -07001669 CreateFrame(frame_timestamp, kFrameWidth, kFrameHeight));
sprang4847ae62017-06-27 07:06:52 -07001670 WaitForEncodedFrame(frame_timestamp);
sprangc5d62e22017-04-02 23:53:04 -07001671 frame_timestamp += kFrameIntervalMs;
1672
perkj803d97f2016-11-01 11:45:46 -07001673 // Trigger CPU overuse.
mflodmancc3d4422017-08-03 08:27:51 -07001674 video_stream_encoder_->TriggerCpuOveruse();
perkj803d97f2016-11-01 11:45:46 -07001675 video_source_.IncomingCapturedFrame(
sprangc5d62e22017-04-02 23:53:04 -07001676 CreateFrame(frame_timestamp, kFrameWidth, kFrameHeight));
sprang4847ae62017-06-27 07:06:52 -07001677 WaitForEncodedFrame(frame_timestamp);
sprangc5d62e22017-04-02 23:53:04 -07001678 frame_timestamp += kFrameIntervalMs;
sprang3ea3c772017-03-30 07:23:48 -07001679
asapersson0944a802017-04-07 00:57:58 -07001680 // Default degradation preference is maintain-framerate, so will lower max
sprangc5d62e22017-04-02 23:53:04 -07001681 // wanted resolution.
1682 EXPECT_FALSE(video_source_.sink_wants().target_pixel_count);
1683 EXPECT_LT(video_source_.sink_wants().max_pixel_count,
1684 kFrameWidth * kFrameHeight);
Åsa Persson8c1bf952018-09-13 10:42:19 +02001685 EXPECT_EQ(kDefaultFramerate, video_source_.sink_wants().max_framerate_fps);
sprangc5d62e22017-04-02 23:53:04 -07001686
1687 // Set new source, switch to maintain-resolution.
perkj803d97f2016-11-01 11:45:46 -07001688 test::FrameForwarder new_video_source;
mflodmancc3d4422017-08-03 08:27:51 -07001689 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07001690 &new_video_source, webrtc::DegradationPreference::MAINTAIN_RESOLUTION);
perkj803d97f2016-11-01 11:45:46 -07001691
sprangc5d62e22017-04-02 23:53:04 -07001692 // Initially no degradation registered.
Åsa Persson8c1bf952018-09-13 10:42:19 +02001693 VerifyFpsMaxResolutionMax(new_video_source.sink_wants());
perkj803d97f2016-11-01 11:45:46 -07001694
sprangc5d62e22017-04-02 23:53:04 -07001695 // Force an input frame rate to be available, or the adaptation call won't
1696 // know what framerate to adapt form.
asapersson02465b82017-04-10 01:12:52 -07001697 const int kInputFps = 30;
sprangc5d62e22017-04-02 23:53:04 -07001698 VideoSendStream::Stats stats = stats_proxy_->GetStats();
asapersson02465b82017-04-10 01:12:52 -07001699 stats.input_frame_rate = kInputFps;
sprangc5d62e22017-04-02 23:53:04 -07001700 stats_proxy_->SetMockStats(stats);
1701
mflodmancc3d4422017-08-03 08:27:51 -07001702 video_stream_encoder_->TriggerCpuOveruse();
perkj803d97f2016-11-01 11:45:46 -07001703 new_video_source.IncomingCapturedFrame(
sprangc5d62e22017-04-02 23:53:04 -07001704 CreateFrame(frame_timestamp, kFrameWidth, kFrameHeight));
sprang4847ae62017-06-27 07:06:52 -07001705 WaitForEncodedFrame(frame_timestamp);
sprangc5d62e22017-04-02 23:53:04 -07001706 frame_timestamp += kFrameIntervalMs;
1707
1708 // Some framerate constraint should be set.
sprang84a37592017-02-10 07:04:27 -08001709 EXPECT_FALSE(new_video_source.sink_wants().target_pixel_count);
sprangc5d62e22017-04-02 23:53:04 -07001710 EXPECT_EQ(std::numeric_limits<int>::max(),
1711 new_video_source.sink_wants().max_pixel_count);
asapersson02465b82017-04-10 01:12:52 -07001712 EXPECT_LT(new_video_source.sink_wants().max_framerate_fps, kInputFps);
sprangc5d62e22017-04-02 23:53:04 -07001713
asapersson02465b82017-04-10 01:12:52 -07001714 // Turn off degradation completely.
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07001715 video_stream_encoder_->SetSource(&new_video_source,
1716 webrtc::DegradationPreference::DISABLED);
Åsa Persson8c1bf952018-09-13 10:42:19 +02001717 VerifyFpsMaxResolutionMax(new_video_source.sink_wants());
sprangc5d62e22017-04-02 23:53:04 -07001718
mflodmancc3d4422017-08-03 08:27:51 -07001719 video_stream_encoder_->TriggerCpuOveruse();
sprangc5d62e22017-04-02 23:53:04 -07001720 new_video_source.IncomingCapturedFrame(
1721 CreateFrame(frame_timestamp, kFrameWidth, kFrameHeight));
sprang4847ae62017-06-27 07:06:52 -07001722 WaitForEncodedFrame(frame_timestamp);
sprangc5d62e22017-04-02 23:53:04 -07001723 frame_timestamp += kFrameIntervalMs;
1724
1725 // Still no degradation.
Åsa Persson8c1bf952018-09-13 10:42:19 +02001726 VerifyFpsMaxResolutionMax(new_video_source.sink_wants());
perkj803d97f2016-11-01 11:45:46 -07001727
1728 // Calling SetSource with resolution scaling enabled apply the old SinkWants.
mflodmancc3d4422017-08-03 08:27:51 -07001729 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07001730 &new_video_source, webrtc::DegradationPreference::MAINTAIN_FRAMERATE);
sprangc5d62e22017-04-02 23:53:04 -07001731 EXPECT_LT(new_video_source.sink_wants().max_pixel_count,
1732 kFrameWidth * kFrameHeight);
sprang84a37592017-02-10 07:04:27 -08001733 EXPECT_FALSE(new_video_source.sink_wants().target_pixel_count);
Åsa Persson8c1bf952018-09-13 10:42:19 +02001734 EXPECT_EQ(kDefaultFramerate, new_video_source.sink_wants().max_framerate_fps);
sprangc5d62e22017-04-02 23:53:04 -07001735
1736 // Calling SetSource with framerate scaling enabled apply the old SinkWants.
mflodmancc3d4422017-08-03 08:27:51 -07001737 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07001738 &new_video_source, webrtc::DegradationPreference::MAINTAIN_RESOLUTION);
sprangc5d62e22017-04-02 23:53:04 -07001739 EXPECT_FALSE(new_video_source.sink_wants().target_pixel_count);
1740 EXPECT_EQ(std::numeric_limits<int>::max(),
1741 new_video_source.sink_wants().max_pixel_count);
asapersson02465b82017-04-10 01:12:52 -07001742 EXPECT_LT(new_video_source.sink_wants().max_framerate_fps, kInputFps);
perkj803d97f2016-11-01 11:45:46 -07001743
mflodmancc3d4422017-08-03 08:27:51 -07001744 video_stream_encoder_->Stop();
perkj803d97f2016-11-01 11:45:46 -07001745}
1746
mflodmancc3d4422017-08-03 08:27:51 -07001747TEST_F(VideoStreamEncoderTest, StatsTracksQualityAdaptationStats) {
Erik Språng4c6ca302019-04-08 15:14:01 +02001748 video_stream_encoder_->OnBitrateUpdated(
1749 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
perkj803d97f2016-11-01 11:45:46 -07001750
asaperssonfab67072017-04-04 05:51:49 -07001751 const int kWidth = 1280;
1752 const int kHeight = 720;
asaperssonfab67072017-04-04 05:51:49 -07001753 video_source_.IncomingCapturedFrame(CreateFrame(1, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07001754 WaitForEncodedFrame(1);
asaperssonfab67072017-04-04 05:51:49 -07001755 VideoSendStream::Stats stats = stats_proxy_->GetStats();
1756 EXPECT_FALSE(stats.bw_limited_resolution);
1757 EXPECT_EQ(0, stats.number_of_quality_adapt_changes);
1758
1759 // Trigger adapt down.
mflodmancc3d4422017-08-03 08:27:51 -07001760 video_stream_encoder_->TriggerQualityLow();
asaperssonfab67072017-04-04 05:51:49 -07001761 video_source_.IncomingCapturedFrame(CreateFrame(2, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07001762 WaitForEncodedFrame(2);
asaperssonfab67072017-04-04 05:51:49 -07001763
1764 stats = stats_proxy_->GetStats();
1765 EXPECT_TRUE(stats.bw_limited_resolution);
1766 EXPECT_EQ(1, stats.number_of_quality_adapt_changes);
1767
1768 // Trigger adapt up.
mflodmancc3d4422017-08-03 08:27:51 -07001769 video_stream_encoder_->TriggerQualityHigh();
asaperssonfab67072017-04-04 05:51:49 -07001770 video_source_.IncomingCapturedFrame(CreateFrame(3, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07001771 WaitForEncodedFrame(3);
asaperssonfab67072017-04-04 05:51:49 -07001772
1773 stats = stats_proxy_->GetStats();
1774 EXPECT_FALSE(stats.bw_limited_resolution);
1775 EXPECT_EQ(2, stats.number_of_quality_adapt_changes);
1776 EXPECT_EQ(0, stats.number_of_cpu_adapt_changes);
1777
mflodmancc3d4422017-08-03 08:27:51 -07001778 video_stream_encoder_->Stop();
asaperssonfab67072017-04-04 05:51:49 -07001779}
1780
mflodmancc3d4422017-08-03 08:27:51 -07001781TEST_F(VideoStreamEncoderTest, StatsTracksCpuAdaptationStats) {
Erik Språng4c6ca302019-04-08 15:14:01 +02001782 video_stream_encoder_->OnBitrateUpdated(
1783 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asaperssonfab67072017-04-04 05:51:49 -07001784
1785 const int kWidth = 1280;
1786 const int kHeight = 720;
asaperssonfab67072017-04-04 05:51:49 -07001787 video_source_.IncomingCapturedFrame(CreateFrame(1, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07001788 WaitForEncodedFrame(1);
perkj803d97f2016-11-01 11:45:46 -07001789 VideoSendStream::Stats stats = stats_proxy_->GetStats();
1790 EXPECT_FALSE(stats.cpu_limited_resolution);
1791 EXPECT_EQ(0, stats.number_of_cpu_adapt_changes);
1792
1793 // Trigger CPU overuse.
mflodmancc3d4422017-08-03 08:27:51 -07001794 video_stream_encoder_->TriggerCpuOveruse();
asaperssonfab67072017-04-04 05:51:49 -07001795 video_source_.IncomingCapturedFrame(CreateFrame(2, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07001796 WaitForEncodedFrame(2);
perkj803d97f2016-11-01 11:45:46 -07001797
1798 stats = stats_proxy_->GetStats();
1799 EXPECT_TRUE(stats.cpu_limited_resolution);
1800 EXPECT_EQ(1, stats.number_of_cpu_adapt_changes);
1801
1802 // Trigger CPU normal use.
mflodmancc3d4422017-08-03 08:27:51 -07001803 video_stream_encoder_->TriggerCpuNormalUsage();
asaperssonfab67072017-04-04 05:51:49 -07001804 video_source_.IncomingCapturedFrame(CreateFrame(3, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07001805 WaitForEncodedFrame(3);
perkj803d97f2016-11-01 11:45:46 -07001806
1807 stats = stats_proxy_->GetStats();
1808 EXPECT_FALSE(stats.cpu_limited_resolution);
1809 EXPECT_EQ(2, stats.number_of_cpu_adapt_changes);
asaperssonfab67072017-04-04 05:51:49 -07001810 EXPECT_EQ(0, stats.number_of_quality_adapt_changes);
perkj803d97f2016-11-01 11:45:46 -07001811
mflodmancc3d4422017-08-03 08:27:51 -07001812 video_stream_encoder_->Stop();
perkj803d97f2016-11-01 11:45:46 -07001813}
1814
mflodmancc3d4422017-08-03 08:27:51 -07001815TEST_F(VideoStreamEncoderTest, SwitchingSourceKeepsCpuAdaptation) {
Erik Språng4c6ca302019-04-08 15:14:01 +02001816 video_stream_encoder_->OnBitrateUpdated(
1817 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
kthelgason876222f2016-11-29 01:44:11 -08001818
asaperssonfab67072017-04-04 05:51:49 -07001819 const int kWidth = 1280;
1820 const int kHeight = 720;
1821 video_source_.IncomingCapturedFrame(CreateFrame(1, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07001822 WaitForEncodedFrame(1);
kthelgason876222f2016-11-29 01:44:11 -08001823 VideoSendStream::Stats stats = stats_proxy_->GetStats();
asapersson09f05612017-05-15 23:40:18 -07001824 EXPECT_FALSE(stats.bw_limited_resolution);
kthelgason876222f2016-11-29 01:44:11 -08001825 EXPECT_FALSE(stats.cpu_limited_resolution);
1826 EXPECT_EQ(0, stats.number_of_cpu_adapt_changes);
1827
asaperssonfab67072017-04-04 05:51:49 -07001828 // Trigger CPU overuse.
mflodmancc3d4422017-08-03 08:27:51 -07001829 video_stream_encoder_->TriggerCpuOveruse();
asaperssonfab67072017-04-04 05:51:49 -07001830 video_source_.IncomingCapturedFrame(CreateFrame(2, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07001831 WaitForEncodedFrame(2);
kthelgason876222f2016-11-29 01:44:11 -08001832 stats = stats_proxy_->GetStats();
asapersson09f05612017-05-15 23:40:18 -07001833 EXPECT_FALSE(stats.bw_limited_resolution);
kthelgason876222f2016-11-29 01:44:11 -08001834 EXPECT_TRUE(stats.cpu_limited_resolution);
1835 EXPECT_EQ(1, stats.number_of_cpu_adapt_changes);
1836
1837 // Set new source with adaptation still enabled.
1838 test::FrameForwarder new_video_source;
mflodmancc3d4422017-08-03 08:27:51 -07001839 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07001840 &new_video_source, webrtc::DegradationPreference::MAINTAIN_FRAMERATE);
kthelgason876222f2016-11-29 01:44:11 -08001841
asaperssonfab67072017-04-04 05:51:49 -07001842 new_video_source.IncomingCapturedFrame(CreateFrame(3, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07001843 WaitForEncodedFrame(3);
kthelgason876222f2016-11-29 01:44:11 -08001844 stats = stats_proxy_->GetStats();
asapersson09f05612017-05-15 23:40:18 -07001845 EXPECT_FALSE(stats.bw_limited_resolution);
kthelgason876222f2016-11-29 01:44:11 -08001846 EXPECT_TRUE(stats.cpu_limited_resolution);
1847 EXPECT_EQ(1, stats.number_of_cpu_adapt_changes);
1848
1849 // Set adaptation disabled.
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07001850 video_stream_encoder_->SetSource(&new_video_source,
1851 webrtc::DegradationPreference::DISABLED);
kthelgason876222f2016-11-29 01:44:11 -08001852
asaperssonfab67072017-04-04 05:51:49 -07001853 new_video_source.IncomingCapturedFrame(CreateFrame(4, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07001854 WaitForEncodedFrame(4);
kthelgason876222f2016-11-29 01:44:11 -08001855 stats = stats_proxy_->GetStats();
asapersson09f05612017-05-15 23:40:18 -07001856 EXPECT_FALSE(stats.bw_limited_resolution);
kthelgason876222f2016-11-29 01:44:11 -08001857 EXPECT_FALSE(stats.cpu_limited_resolution);
1858 EXPECT_EQ(1, stats.number_of_cpu_adapt_changes);
1859
1860 // Set adaptation back to enabled.
mflodmancc3d4422017-08-03 08:27:51 -07001861 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07001862 &new_video_source, webrtc::DegradationPreference::MAINTAIN_FRAMERATE);
kthelgason876222f2016-11-29 01:44:11 -08001863
asaperssonfab67072017-04-04 05:51:49 -07001864 new_video_source.IncomingCapturedFrame(CreateFrame(5, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07001865 WaitForEncodedFrame(5);
kthelgason876222f2016-11-29 01:44:11 -08001866 stats = stats_proxy_->GetStats();
asapersson09f05612017-05-15 23:40:18 -07001867 EXPECT_FALSE(stats.bw_limited_resolution);
kthelgason876222f2016-11-29 01:44:11 -08001868 EXPECT_TRUE(stats.cpu_limited_resolution);
1869 EXPECT_EQ(1, stats.number_of_cpu_adapt_changes);
1870
asaperssonfab67072017-04-04 05:51:49 -07001871 // Trigger CPU normal use.
mflodmancc3d4422017-08-03 08:27:51 -07001872 video_stream_encoder_->TriggerCpuNormalUsage();
asaperssonfab67072017-04-04 05:51:49 -07001873 new_video_source.IncomingCapturedFrame(CreateFrame(6, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07001874 WaitForEncodedFrame(6);
kthelgason876222f2016-11-29 01:44:11 -08001875 stats = stats_proxy_->GetStats();
asapersson09f05612017-05-15 23:40:18 -07001876 EXPECT_FALSE(stats.bw_limited_resolution);
kthelgason876222f2016-11-29 01:44:11 -08001877 EXPECT_FALSE(stats.cpu_limited_resolution);
1878 EXPECT_EQ(2, stats.number_of_cpu_adapt_changes);
asapersson02465b82017-04-10 01:12:52 -07001879 EXPECT_EQ(0, stats.number_of_quality_adapt_changes);
kthelgason876222f2016-11-29 01:44:11 -08001880
mflodmancc3d4422017-08-03 08:27:51 -07001881 video_stream_encoder_->Stop();
kthelgason876222f2016-11-29 01:44:11 -08001882}
1883
mflodmancc3d4422017-08-03 08:27:51 -07001884TEST_F(VideoStreamEncoderTest, SwitchingSourceKeepsQualityAdaptation) {
Erik Språng4c6ca302019-04-08 15:14:01 +02001885 video_stream_encoder_->OnBitrateUpdated(
1886 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
kthelgason876222f2016-11-29 01:44:11 -08001887
asaperssonfab67072017-04-04 05:51:49 -07001888 const int kWidth = 1280;
1889 const int kHeight = 720;
1890 video_source_.IncomingCapturedFrame(CreateFrame(1, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07001891 WaitForEncodedFrame(1);
kthelgason876222f2016-11-29 01:44:11 -08001892 VideoSendStream::Stats stats = stats_proxy_->GetStats();
kthelgason876222f2016-11-29 01:44:11 -08001893 EXPECT_FALSE(stats.bw_limited_resolution);
asapersson13874762017-06-07 00:01:02 -07001894 EXPECT_FALSE(stats.bw_limited_framerate);
asaperssonfab67072017-04-04 05:51:49 -07001895 EXPECT_EQ(0, stats.number_of_quality_adapt_changes);
kthelgason876222f2016-11-29 01:44:11 -08001896
1897 // Set new source with adaptation still enabled.
1898 test::FrameForwarder new_video_source;
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07001899 video_stream_encoder_->SetSource(&new_video_source,
1900 webrtc::DegradationPreference::BALANCED);
kthelgason876222f2016-11-29 01:44:11 -08001901
asaperssonfab67072017-04-04 05:51:49 -07001902 new_video_source.IncomingCapturedFrame(CreateFrame(2, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07001903 WaitForEncodedFrame(2);
kthelgason876222f2016-11-29 01:44:11 -08001904 stats = stats_proxy_->GetStats();
kthelgason876222f2016-11-29 01:44:11 -08001905 EXPECT_FALSE(stats.bw_limited_resolution);
asapersson13874762017-06-07 00:01:02 -07001906 EXPECT_FALSE(stats.bw_limited_framerate);
asaperssonfab67072017-04-04 05:51:49 -07001907 EXPECT_EQ(0, stats.number_of_quality_adapt_changes);
kthelgason876222f2016-11-29 01:44:11 -08001908
asaperssonfab67072017-04-04 05:51:49 -07001909 // Trigger adapt down.
mflodmancc3d4422017-08-03 08:27:51 -07001910 video_stream_encoder_->TriggerQualityLow();
asaperssonfab67072017-04-04 05:51:49 -07001911 new_video_source.IncomingCapturedFrame(CreateFrame(3, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07001912 WaitForEncodedFrame(3);
kthelgason876222f2016-11-29 01:44:11 -08001913 stats = stats_proxy_->GetStats();
kthelgason876222f2016-11-29 01:44:11 -08001914 EXPECT_TRUE(stats.bw_limited_resolution);
asapersson13874762017-06-07 00:01:02 -07001915 EXPECT_FALSE(stats.bw_limited_framerate);
asaperssonfab67072017-04-04 05:51:49 -07001916 EXPECT_EQ(1, stats.number_of_quality_adapt_changes);
kthelgason876222f2016-11-29 01:44:11 -08001917
asaperssonfab67072017-04-04 05:51:49 -07001918 // Set new source with adaptation still enabled.
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07001919 video_stream_encoder_->SetSource(&new_video_source,
1920 webrtc::DegradationPreference::BALANCED);
kthelgason876222f2016-11-29 01:44:11 -08001921
asaperssonfab67072017-04-04 05:51:49 -07001922 new_video_source.IncomingCapturedFrame(CreateFrame(4, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07001923 WaitForEncodedFrame(4);
kthelgason876222f2016-11-29 01:44:11 -08001924 stats = stats_proxy_->GetStats();
kthelgason876222f2016-11-29 01:44:11 -08001925 EXPECT_TRUE(stats.bw_limited_resolution);
asapersson13874762017-06-07 00:01:02 -07001926 EXPECT_FALSE(stats.bw_limited_framerate);
asaperssonfab67072017-04-04 05:51:49 -07001927 EXPECT_EQ(1, stats.number_of_quality_adapt_changes);
kthelgason876222f2016-11-29 01:44:11 -08001928
asapersson02465b82017-04-10 01:12:52 -07001929 // Disable resolution scaling.
mflodmancc3d4422017-08-03 08:27:51 -07001930 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07001931 &new_video_source, webrtc::DegradationPreference::MAINTAIN_RESOLUTION);
kthelgason876222f2016-11-29 01:44:11 -08001932
asaperssonfab67072017-04-04 05:51:49 -07001933 new_video_source.IncomingCapturedFrame(CreateFrame(5, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07001934 WaitForEncodedFrame(5);
kthelgason876222f2016-11-29 01:44:11 -08001935 stats = stats_proxy_->GetStats();
kthelgason876222f2016-11-29 01:44:11 -08001936 EXPECT_FALSE(stats.bw_limited_resolution);
asapersson13874762017-06-07 00:01:02 -07001937 EXPECT_FALSE(stats.bw_limited_framerate);
asaperssonfab67072017-04-04 05:51:49 -07001938 EXPECT_EQ(1, stats.number_of_quality_adapt_changes);
1939 EXPECT_EQ(0, stats.number_of_cpu_adapt_changes);
kthelgason876222f2016-11-29 01:44:11 -08001940
mflodmancc3d4422017-08-03 08:27:51 -07001941 video_stream_encoder_->Stop();
kthelgason876222f2016-11-29 01:44:11 -08001942}
1943
mflodmancc3d4422017-08-03 08:27:51 -07001944TEST_F(VideoStreamEncoderTest,
1945 QualityAdaptationStatsAreResetWhenScalerIsDisabled) {
Erik Språng4c6ca302019-04-08 15:14:01 +02001946 video_stream_encoder_->OnBitrateUpdated(
1947 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asapersson36e9eb42017-03-31 05:29:12 -07001948
1949 const int kWidth = 1280;
1950 const int kHeight = 720;
Åsa Persson8c1bf952018-09-13 10:42:19 +02001951 int64_t timestamp_ms = kFrameIntervalMs;
asapersson36e9eb42017-03-31 05:29:12 -07001952 video_source_.set_adaptation_enabled(true);
Åsa Persson8c1bf952018-09-13 10:42:19 +02001953 video_source_.IncomingCapturedFrame(
1954 CreateFrame(timestamp_ms, kWidth, kHeight));
1955 WaitForEncodedFrame(timestamp_ms);
asapersson36e9eb42017-03-31 05:29:12 -07001956 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
1957 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
1958 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
1959
1960 // Trigger adapt down.
mflodmancc3d4422017-08-03 08:27:51 -07001961 video_stream_encoder_->TriggerQualityLow();
Åsa Persson8c1bf952018-09-13 10:42:19 +02001962 timestamp_ms += kFrameIntervalMs;
1963 video_source_.IncomingCapturedFrame(
1964 CreateFrame(timestamp_ms, kWidth, kHeight));
1965 WaitForEncodedFrame(timestamp_ms);
asapersson36e9eb42017-03-31 05:29:12 -07001966 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
1967 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
1968 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
1969
1970 // Trigger overuse.
mflodmancc3d4422017-08-03 08:27:51 -07001971 video_stream_encoder_->TriggerCpuOveruse();
Åsa Persson8c1bf952018-09-13 10:42:19 +02001972 timestamp_ms += kFrameIntervalMs;
1973 video_source_.IncomingCapturedFrame(
1974 CreateFrame(timestamp_ms, kWidth, kHeight));
1975 WaitForEncodedFrame(timestamp_ms);
asapersson36e9eb42017-03-31 05:29:12 -07001976 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_resolution);
1977 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
1978 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
1979
Niels Möller4db138e2018-04-19 09:04:13 +02001980 // Leave source unchanged, but disable quality scaler.
asapersson36e9eb42017-03-31 05:29:12 -07001981 fake_encoder_.SetQualityScaling(false);
Niels Möller4db138e2018-04-19 09:04:13 +02001982
1983 VideoEncoderConfig video_encoder_config;
1984 test::FillEncoderConfiguration(kVideoCodecVP8, 1, &video_encoder_config);
1985 // Make format different, to force recreation of encoder.
1986 video_encoder_config.video_format.parameters["foo"] = "foo";
1987 video_stream_encoder_->ConfigureEncoder(std::move(video_encoder_config),
Niels Möllerf1338562018-04-26 09:51:47 +02001988 kMaxPayloadLength);
Åsa Persson8c1bf952018-09-13 10:42:19 +02001989 timestamp_ms += kFrameIntervalMs;
1990 video_source_.IncomingCapturedFrame(
1991 CreateFrame(timestamp_ms, kWidth, kHeight));
1992 WaitForEncodedFrame(timestamp_ms);
asapersson36e9eb42017-03-31 05:29:12 -07001993 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_resolution);
1994 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
1995 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
1996
mflodmancc3d4422017-08-03 08:27:51 -07001997 video_stream_encoder_->Stop();
asapersson36e9eb42017-03-31 05:29:12 -07001998}
1999
mflodmancc3d4422017-08-03 08:27:51 -07002000TEST_F(VideoStreamEncoderTest,
2001 StatsTracksCpuAdaptationStatsWhenSwitchingSource) {
Erik Språng4c6ca302019-04-08 15:14:01 +02002002 video_stream_encoder_->OnBitrateUpdated(
2003 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
perkj803d97f2016-11-01 11:45:46 -07002004
asapersson0944a802017-04-07 00:57:58 -07002005 const int kWidth = 1280;
2006 const int kHeight = 720;
sprang84a37592017-02-10 07:04:27 -08002007 int sequence = 1;
perkj803d97f2016-11-01 11:45:46 -07002008
asaperssonfab67072017-04-04 05:51:49 -07002009 video_source_.IncomingCapturedFrame(CreateFrame(sequence, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002010 WaitForEncodedFrame(sequence++);
perkj803d97f2016-11-01 11:45:46 -07002011 VideoSendStream::Stats stats = stats_proxy_->GetStats();
sprang84a37592017-02-10 07:04:27 -08002012 EXPECT_FALSE(stats.cpu_limited_resolution);
asapersson13874762017-06-07 00:01:02 -07002013 EXPECT_FALSE(stats.cpu_limited_framerate);
sprang84a37592017-02-10 07:04:27 -08002014 EXPECT_EQ(0, stats.number_of_cpu_adapt_changes);
2015
asapersson02465b82017-04-10 01:12:52 -07002016 // Trigger CPU overuse, should now adapt down.
mflodmancc3d4422017-08-03 08:27:51 -07002017 video_stream_encoder_->TriggerCpuOveruse();
asaperssonfab67072017-04-04 05:51:49 -07002018 video_source_.IncomingCapturedFrame(CreateFrame(sequence, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002019 WaitForEncodedFrame(sequence++);
sprang84a37592017-02-10 07:04:27 -08002020 stats = stats_proxy_->GetStats();
perkj803d97f2016-11-01 11:45:46 -07002021 EXPECT_TRUE(stats.cpu_limited_resolution);
asapersson13874762017-06-07 00:01:02 -07002022 EXPECT_FALSE(stats.cpu_limited_framerate);
perkj803d97f2016-11-01 11:45:46 -07002023 EXPECT_EQ(1, stats.number_of_cpu_adapt_changes);
2024
2025 // Set new source with adaptation still enabled.
2026 test::FrameForwarder new_video_source;
mflodmancc3d4422017-08-03 08:27:51 -07002027 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002028 &new_video_source, webrtc::DegradationPreference::MAINTAIN_FRAMERATE);
perkj803d97f2016-11-01 11:45:46 -07002029
2030 new_video_source.IncomingCapturedFrame(
asaperssonfab67072017-04-04 05:51:49 -07002031 CreateFrame(sequence, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002032 WaitForEncodedFrame(sequence++);
perkj803d97f2016-11-01 11:45:46 -07002033 stats = stats_proxy_->GetStats();
2034 EXPECT_TRUE(stats.cpu_limited_resolution);
asapersson09f05612017-05-15 23:40:18 -07002035 EXPECT_FALSE(stats.cpu_limited_framerate);
perkj803d97f2016-11-01 11:45:46 -07002036 EXPECT_EQ(1, stats.number_of_cpu_adapt_changes);
2037
sprangc5d62e22017-04-02 23:53:04 -07002038 // Set cpu adaptation by frame dropping.
mflodmancc3d4422017-08-03 08:27:51 -07002039 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002040 &new_video_source, webrtc::DegradationPreference::MAINTAIN_RESOLUTION);
perkj803d97f2016-11-01 11:45:46 -07002041 new_video_source.IncomingCapturedFrame(
asaperssonfab67072017-04-04 05:51:49 -07002042 CreateFrame(sequence, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002043 WaitForEncodedFrame(sequence++);
perkj803d97f2016-11-01 11:45:46 -07002044 stats = stats_proxy_->GetStats();
sprangc5d62e22017-04-02 23:53:04 -07002045 // Not adapted at first.
perkj803d97f2016-11-01 11:45:46 -07002046 EXPECT_FALSE(stats.cpu_limited_resolution);
asapersson09f05612017-05-15 23:40:18 -07002047 EXPECT_FALSE(stats.cpu_limited_framerate);
perkj803d97f2016-11-01 11:45:46 -07002048 EXPECT_EQ(1, stats.number_of_cpu_adapt_changes);
2049
sprangc5d62e22017-04-02 23:53:04 -07002050 // Force an input frame rate to be available, or the adaptation call won't
asapersson09f05612017-05-15 23:40:18 -07002051 // know what framerate to adapt from.
sprangc5d62e22017-04-02 23:53:04 -07002052 VideoSendStream::Stats mock_stats = stats_proxy_->GetStats();
2053 mock_stats.input_frame_rate = 30;
2054 stats_proxy_->SetMockStats(mock_stats);
mflodmancc3d4422017-08-03 08:27:51 -07002055 video_stream_encoder_->TriggerCpuOveruse();
sprangc5d62e22017-04-02 23:53:04 -07002056 stats_proxy_->ResetMockStats();
2057
2058 new_video_source.IncomingCapturedFrame(
asaperssonfab67072017-04-04 05:51:49 -07002059 CreateFrame(sequence, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002060 WaitForEncodedFrame(sequence++);
sprangc5d62e22017-04-02 23:53:04 -07002061
2062 // Framerate now adapted.
2063 stats = stats_proxy_->GetStats();
asapersson09f05612017-05-15 23:40:18 -07002064 EXPECT_FALSE(stats.cpu_limited_resolution);
2065 EXPECT_TRUE(stats.cpu_limited_framerate);
sprangc5d62e22017-04-02 23:53:04 -07002066 EXPECT_EQ(2, stats.number_of_cpu_adapt_changes);
2067
2068 // Disable CPU adaptation.
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002069 video_stream_encoder_->SetSource(&new_video_source,
2070 webrtc::DegradationPreference::DISABLED);
sprangc5d62e22017-04-02 23:53:04 -07002071 new_video_source.IncomingCapturedFrame(
asaperssonfab67072017-04-04 05:51:49 -07002072 CreateFrame(sequence, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002073 WaitForEncodedFrame(sequence++);
sprangc5d62e22017-04-02 23:53:04 -07002074
2075 stats = stats_proxy_->GetStats();
2076 EXPECT_FALSE(stats.cpu_limited_resolution);
asapersson13874762017-06-07 00:01:02 -07002077 EXPECT_FALSE(stats.cpu_limited_framerate);
sprangc5d62e22017-04-02 23:53:04 -07002078 EXPECT_EQ(2, stats.number_of_cpu_adapt_changes);
2079
2080 // Try to trigger overuse. Should not succeed.
2081 stats_proxy_->SetMockStats(mock_stats);
mflodmancc3d4422017-08-03 08:27:51 -07002082 video_stream_encoder_->TriggerCpuOveruse();
sprangc5d62e22017-04-02 23:53:04 -07002083 stats_proxy_->ResetMockStats();
2084
2085 stats = stats_proxy_->GetStats();
2086 EXPECT_FALSE(stats.cpu_limited_resolution);
asapersson13874762017-06-07 00:01:02 -07002087 EXPECT_FALSE(stats.cpu_limited_framerate);
sprangc5d62e22017-04-02 23:53:04 -07002088 EXPECT_EQ(2, stats.number_of_cpu_adapt_changes);
2089
2090 // Switch back the source with resolution adaptation enabled.
mflodmancc3d4422017-08-03 08:27:51 -07002091 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002092 &video_source_, webrtc::DegradationPreference::MAINTAIN_FRAMERATE);
asaperssonfab67072017-04-04 05:51:49 -07002093 video_source_.IncomingCapturedFrame(CreateFrame(sequence, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002094 WaitForEncodedFrame(sequence++);
perkj803d97f2016-11-01 11:45:46 -07002095 stats = stats_proxy_->GetStats();
2096 EXPECT_TRUE(stats.cpu_limited_resolution);
asapersson13874762017-06-07 00:01:02 -07002097 EXPECT_FALSE(stats.cpu_limited_framerate);
sprangc5d62e22017-04-02 23:53:04 -07002098 EXPECT_EQ(2, stats.number_of_cpu_adapt_changes);
perkj803d97f2016-11-01 11:45:46 -07002099
2100 // Trigger CPU normal usage.
mflodmancc3d4422017-08-03 08:27:51 -07002101 video_stream_encoder_->TriggerCpuNormalUsage();
asaperssonfab67072017-04-04 05:51:49 -07002102 video_source_.IncomingCapturedFrame(CreateFrame(sequence, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002103 WaitForEncodedFrame(sequence++);
perkj803d97f2016-11-01 11:45:46 -07002104 stats = stats_proxy_->GetStats();
2105 EXPECT_FALSE(stats.cpu_limited_resolution);
asapersson13874762017-06-07 00:01:02 -07002106 EXPECT_FALSE(stats.cpu_limited_framerate);
sprangc5d62e22017-04-02 23:53:04 -07002107 EXPECT_EQ(3, stats.number_of_cpu_adapt_changes);
2108
2109 // Back to the source with adaptation off, set it back to maintain-resolution.
mflodmancc3d4422017-08-03 08:27:51 -07002110 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002111 &new_video_source, webrtc::DegradationPreference::MAINTAIN_RESOLUTION);
sprangc5d62e22017-04-02 23:53:04 -07002112 new_video_source.IncomingCapturedFrame(
asaperssonfab67072017-04-04 05:51:49 -07002113 CreateFrame(sequence, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002114 WaitForEncodedFrame(sequence++);
sprangc5d62e22017-04-02 23:53:04 -07002115 stats = stats_proxy_->GetStats();
asapersson13874762017-06-07 00:01:02 -07002116 // Disabled, since we previously switched the source to disabled.
sprangc5d62e22017-04-02 23:53:04 -07002117 EXPECT_FALSE(stats.cpu_limited_resolution);
asapersson13874762017-06-07 00:01:02 -07002118 EXPECT_TRUE(stats.cpu_limited_framerate);
sprangc5d62e22017-04-02 23:53:04 -07002119 EXPECT_EQ(3, stats.number_of_cpu_adapt_changes);
2120
2121 // Trigger CPU normal usage.
mflodmancc3d4422017-08-03 08:27:51 -07002122 video_stream_encoder_->TriggerCpuNormalUsage();
sprangc5d62e22017-04-02 23:53:04 -07002123 new_video_source.IncomingCapturedFrame(
asaperssonfab67072017-04-04 05:51:49 -07002124 CreateFrame(sequence, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002125 WaitForEncodedFrame(sequence++);
sprangc5d62e22017-04-02 23:53:04 -07002126 stats = stats_proxy_->GetStats();
2127 EXPECT_FALSE(stats.cpu_limited_resolution);
asapersson13874762017-06-07 00:01:02 -07002128 EXPECT_FALSE(stats.cpu_limited_framerate);
sprangc5d62e22017-04-02 23:53:04 -07002129 EXPECT_EQ(4, stats.number_of_cpu_adapt_changes);
asapersson02465b82017-04-10 01:12:52 -07002130 EXPECT_EQ(0, stats.number_of_quality_adapt_changes);
perkj803d97f2016-11-01 11:45:46 -07002131
mflodmancc3d4422017-08-03 08:27:51 -07002132 video_stream_encoder_->Stop();
perkj803d97f2016-11-01 11:45:46 -07002133}
2134
mflodmancc3d4422017-08-03 08:27:51 -07002135TEST_F(VideoStreamEncoderTest,
2136 ScalingUpAndDownDoesNothingWithMaintainResolution) {
asaperssonfab67072017-04-04 05:51:49 -07002137 const int kWidth = 1280;
2138 const int kHeight = 720;
Erik Språng4c6ca302019-04-08 15:14:01 +02002139 video_stream_encoder_->OnBitrateUpdated(
2140 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
kthelgason876222f2016-11-29 01:44:11 -08002141
asaperssonfab67072017-04-04 05:51:49 -07002142 // Expect no scaling to begin with.
asapersson02465b82017-04-10 01:12:52 -07002143 VerifyNoLimitation(video_source_.sink_wants());
kthelgason876222f2016-11-29 01:44:11 -08002144
asaperssonfab67072017-04-04 05:51:49 -07002145 video_source_.IncomingCapturedFrame(CreateFrame(1, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002146 WaitForEncodedFrame(1);
kthelgason876222f2016-11-29 01:44:11 -08002147
asaperssonfab67072017-04-04 05:51:49 -07002148 // Trigger scale down.
mflodmancc3d4422017-08-03 08:27:51 -07002149 video_stream_encoder_->TriggerQualityLow();
kthelgason5e13d412016-12-01 03:59:51 -08002150
asaperssonfab67072017-04-04 05:51:49 -07002151 video_source_.IncomingCapturedFrame(CreateFrame(2, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002152 WaitForEncodedFrame(2);
kthelgason5e13d412016-12-01 03:59:51 -08002153
kthelgason876222f2016-11-29 01:44:11 -08002154 // Expect a scale down.
2155 EXPECT_TRUE(video_source_.sink_wants().max_pixel_count);
asaperssonfab67072017-04-04 05:51:49 -07002156 EXPECT_LT(video_source_.sink_wants().max_pixel_count, kWidth * kHeight);
kthelgason876222f2016-11-29 01:44:11 -08002157
asapersson02465b82017-04-10 01:12:52 -07002158 // Set resolution scaling disabled.
kthelgason876222f2016-11-29 01:44:11 -08002159 test::FrameForwarder new_video_source;
mflodmancc3d4422017-08-03 08:27:51 -07002160 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002161 &new_video_source, webrtc::DegradationPreference::MAINTAIN_RESOLUTION);
kthelgason876222f2016-11-29 01:44:11 -08002162
asaperssonfab67072017-04-04 05:51:49 -07002163 // Trigger scale down.
mflodmancc3d4422017-08-03 08:27:51 -07002164 video_stream_encoder_->TriggerQualityLow();
asaperssonfab67072017-04-04 05:51:49 -07002165 new_video_source.IncomingCapturedFrame(CreateFrame(3, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002166 WaitForEncodedFrame(3);
kthelgason876222f2016-11-29 01:44:11 -08002167
asaperssonfab67072017-04-04 05:51:49 -07002168 // Expect no scaling.
sprangc5d62e22017-04-02 23:53:04 -07002169 EXPECT_EQ(std::numeric_limits<int>::max(),
2170 new_video_source.sink_wants().max_pixel_count);
kthelgason876222f2016-11-29 01:44:11 -08002171
asaperssonfab67072017-04-04 05:51:49 -07002172 // Trigger scale up.
mflodmancc3d4422017-08-03 08:27:51 -07002173 video_stream_encoder_->TriggerQualityHigh();
asaperssonfab67072017-04-04 05:51:49 -07002174 new_video_source.IncomingCapturedFrame(CreateFrame(4, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002175 WaitForEncodedFrame(4);
kthelgason876222f2016-11-29 01:44:11 -08002176
asapersson02465b82017-04-10 01:12:52 -07002177 // Expect nothing to change, still no scaling.
sprangc5d62e22017-04-02 23:53:04 -07002178 EXPECT_EQ(std::numeric_limits<int>::max(),
2179 new_video_source.sink_wants().max_pixel_count);
kthelgason876222f2016-11-29 01:44:11 -08002180
mflodmancc3d4422017-08-03 08:27:51 -07002181 video_stream_encoder_->Stop();
kthelgason876222f2016-11-29 01:44:11 -08002182}
2183
mflodmancc3d4422017-08-03 08:27:51 -07002184TEST_F(VideoStreamEncoderTest,
2185 SkipsSameAdaptDownRequest_MaintainFramerateMode) {
asapersson02465b82017-04-10 01:12:52 -07002186 const int kWidth = 1280;
2187 const int kHeight = 720;
Erik Språng4c6ca302019-04-08 15:14:01 +02002188 video_stream_encoder_->OnBitrateUpdated(
2189 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asapersson02465b82017-04-10 01:12:52 -07002190
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002191 // Enable MAINTAIN_FRAMERATE preference, no initial limitation.
asapersson02465b82017-04-10 01:12:52 -07002192 test::FrameForwarder source;
mflodmancc3d4422017-08-03 08:27:51 -07002193 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002194 &source, webrtc::DegradationPreference::MAINTAIN_FRAMERATE);
asapersson02465b82017-04-10 01:12:52 -07002195
2196 source.IncomingCapturedFrame(CreateFrame(1, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002197 WaitForEncodedFrame(1);
Åsa Persson8c1bf952018-09-13 10:42:19 +02002198 VerifyFpsMaxResolutionMax(source.sink_wants());
asapersson02465b82017-04-10 01:12:52 -07002199 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
2200 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2201
2202 // Trigger adapt down, expect scaled down resolution.
mflodmancc3d4422017-08-03 08:27:51 -07002203 video_stream_encoder_->TriggerCpuOveruse();
asapersson09f05612017-05-15 23:40:18 -07002204 VerifyFpsMaxResolutionLt(source.sink_wants(), kWidth * kHeight);
asapersson02465b82017-04-10 01:12:52 -07002205 const int kLastMaxPixelCount = source.sink_wants().max_pixel_count;
2206 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_resolution);
2207 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2208
2209 // Trigger adapt down for same input resolution, expect no change.
mflodmancc3d4422017-08-03 08:27:51 -07002210 video_stream_encoder_->TriggerCpuOveruse();
asapersson02465b82017-04-10 01:12:52 -07002211 EXPECT_EQ(kLastMaxPixelCount, source.sink_wants().max_pixel_count);
2212 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_resolution);
2213 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2214
mflodmancc3d4422017-08-03 08:27:51 -07002215 video_stream_encoder_->Stop();
asapersson02465b82017-04-10 01:12:52 -07002216}
2217
mflodmancc3d4422017-08-03 08:27:51 -07002218TEST_F(VideoStreamEncoderTest, SkipsSameOrLargerAdaptDownRequest_BalancedMode) {
asaperssonf7e294d2017-06-13 23:25:22 -07002219 const int kWidth = 1280;
2220 const int kHeight = 720;
Erik Språng4c6ca302019-04-08 15:14:01 +02002221 video_stream_encoder_->OnBitrateUpdated(
2222 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asaperssonf7e294d2017-06-13 23:25:22 -07002223
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002224 // Enable BALANCED preference, no initial limitation.
asaperssonf7e294d2017-06-13 23:25:22 -07002225 test::FrameForwarder source;
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002226 video_stream_encoder_->SetSource(&source,
2227 webrtc::DegradationPreference::BALANCED);
asaperssonf7e294d2017-06-13 23:25:22 -07002228 source.IncomingCapturedFrame(CreateFrame(1, kWidth, kHeight));
2229 sink_.WaitForEncodedFrame(1);
Åsa Persson8c1bf952018-09-13 10:42:19 +02002230 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssonf7e294d2017-06-13 23:25:22 -07002231
2232 // Trigger adapt down, expect scaled down resolution.
mflodmancc3d4422017-08-03 08:27:51 -07002233 video_stream_encoder_->TriggerQualityLow();
asaperssonf7e294d2017-06-13 23:25:22 -07002234 VerifyFpsMaxResolutionLt(source.sink_wants(), kWidth * kHeight);
2235 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
2236 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2237 const int kLastMaxPixelCount = source.sink_wants().max_pixel_count;
2238
2239 // Trigger adapt down for same input resolution, expect no change.
2240 source.IncomingCapturedFrame(CreateFrame(2, kWidth, kHeight));
2241 sink_.WaitForEncodedFrame(2);
mflodmancc3d4422017-08-03 08:27:51 -07002242 video_stream_encoder_->TriggerQualityLow();
asaperssonf7e294d2017-06-13 23:25:22 -07002243 EXPECT_EQ(kLastMaxPixelCount, source.sink_wants().max_pixel_count);
2244 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
2245 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2246
2247 // Trigger adapt down for larger input resolution, expect no change.
2248 source.IncomingCapturedFrame(CreateFrame(3, kWidth + 1, kHeight + 1));
2249 sink_.WaitForEncodedFrame(3);
mflodmancc3d4422017-08-03 08:27:51 -07002250 video_stream_encoder_->TriggerQualityLow();
asaperssonf7e294d2017-06-13 23:25:22 -07002251 EXPECT_EQ(kLastMaxPixelCount, source.sink_wants().max_pixel_count);
2252 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
2253 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2254
mflodmancc3d4422017-08-03 08:27:51 -07002255 video_stream_encoder_->Stop();
asaperssonf7e294d2017-06-13 23:25:22 -07002256}
2257
mflodmancc3d4422017-08-03 08:27:51 -07002258TEST_F(VideoStreamEncoderTest,
2259 NoChangeForInitialNormalUsage_MaintainFramerateMode) {
asapersson02465b82017-04-10 01:12:52 -07002260 const int kWidth = 1280;
2261 const int kHeight = 720;
Erik Språng4c6ca302019-04-08 15:14:01 +02002262 video_stream_encoder_->OnBitrateUpdated(
2263 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asapersson02465b82017-04-10 01:12:52 -07002264
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002265 // Enable MAINTAIN_FRAMERATE preference, no initial limitation.
asapersson02465b82017-04-10 01:12:52 -07002266 test::FrameForwarder source;
mflodmancc3d4422017-08-03 08:27:51 -07002267 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002268 &source, webrtc::DegradationPreference::MAINTAIN_FRAMERATE);
asapersson02465b82017-04-10 01:12:52 -07002269
2270 source.IncomingCapturedFrame(CreateFrame(1, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002271 WaitForEncodedFrame(kWidth, kHeight);
Åsa Persson8c1bf952018-09-13 10:42:19 +02002272 VerifyFpsMaxResolutionMax(source.sink_wants());
asapersson02465b82017-04-10 01:12:52 -07002273 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
2274 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2275
2276 // Trigger adapt up, expect no change.
mflodmancc3d4422017-08-03 08:27:51 -07002277 video_stream_encoder_->TriggerCpuNormalUsage();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002278 VerifyFpsMaxResolutionMax(source.sink_wants());
asapersson02465b82017-04-10 01:12:52 -07002279 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
2280 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2281
mflodmancc3d4422017-08-03 08:27:51 -07002282 video_stream_encoder_->Stop();
asapersson02465b82017-04-10 01:12:52 -07002283}
2284
mflodmancc3d4422017-08-03 08:27:51 -07002285TEST_F(VideoStreamEncoderTest,
2286 NoChangeForInitialNormalUsage_MaintainResolutionMode) {
asapersson02465b82017-04-10 01:12:52 -07002287 const int kWidth = 1280;
2288 const int kHeight = 720;
Erik Språng4c6ca302019-04-08 15:14:01 +02002289 video_stream_encoder_->OnBitrateUpdated(
2290 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asapersson02465b82017-04-10 01:12:52 -07002291
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002292 // Enable MAINTAIN_RESOLUTION preference, no initial limitation.
asapersson02465b82017-04-10 01:12:52 -07002293 test::FrameForwarder source;
mflodmancc3d4422017-08-03 08:27:51 -07002294 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002295 &source, webrtc::DegradationPreference::MAINTAIN_RESOLUTION);
asapersson02465b82017-04-10 01:12:52 -07002296
2297 source.IncomingCapturedFrame(CreateFrame(1, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002298 WaitForEncodedFrame(kWidth, kHeight);
Åsa Persson8c1bf952018-09-13 10:42:19 +02002299 VerifyFpsMaxResolutionMax(source.sink_wants());
asapersson09f05612017-05-15 23:40:18 -07002300 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_framerate);
asapersson02465b82017-04-10 01:12:52 -07002301 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2302
2303 // Trigger adapt up, expect no change.
mflodmancc3d4422017-08-03 08:27:51 -07002304 video_stream_encoder_->TriggerCpuNormalUsage();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002305 VerifyFpsMaxResolutionMax(source.sink_wants());
asapersson09f05612017-05-15 23:40:18 -07002306 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_framerate);
asapersson02465b82017-04-10 01:12:52 -07002307 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2308
mflodmancc3d4422017-08-03 08:27:51 -07002309 video_stream_encoder_->Stop();
asapersson02465b82017-04-10 01:12:52 -07002310}
2311
mflodmancc3d4422017-08-03 08:27:51 -07002312TEST_F(VideoStreamEncoderTest, NoChangeForInitialNormalUsage_BalancedMode) {
asaperssonf7e294d2017-06-13 23:25:22 -07002313 const int kWidth = 1280;
2314 const int kHeight = 720;
Erik Språng4c6ca302019-04-08 15:14:01 +02002315 video_stream_encoder_->OnBitrateUpdated(
2316 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asaperssonf7e294d2017-06-13 23:25:22 -07002317
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002318 // Enable BALANCED preference, no initial limitation.
asaperssonf7e294d2017-06-13 23:25:22 -07002319 test::FrameForwarder source;
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002320 video_stream_encoder_->SetSource(&source,
2321 webrtc::DegradationPreference::BALANCED);
asaperssonf7e294d2017-06-13 23:25:22 -07002322
2323 source.IncomingCapturedFrame(CreateFrame(1, kWidth, kHeight));
2324 sink_.WaitForEncodedFrame(kWidth, kHeight);
Åsa Persson8c1bf952018-09-13 10:42:19 +02002325 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssonf7e294d2017-06-13 23:25:22 -07002326 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
2327 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_framerate);
2328 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2329
2330 // Trigger adapt up, expect no change.
mflodmancc3d4422017-08-03 08:27:51 -07002331 video_stream_encoder_->TriggerQualityHigh();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002332 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssonf7e294d2017-06-13 23:25:22 -07002333 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
2334 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_framerate);
2335 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2336
mflodmancc3d4422017-08-03 08:27:51 -07002337 video_stream_encoder_->Stop();
asaperssonf7e294d2017-06-13 23:25:22 -07002338}
2339
mflodmancc3d4422017-08-03 08:27:51 -07002340TEST_F(VideoStreamEncoderTest, NoChangeForInitialNormalUsage_DisabledMode) {
asapersson09f05612017-05-15 23:40:18 -07002341 const int kWidth = 1280;
2342 const int kHeight = 720;
Erik Språng4c6ca302019-04-08 15:14:01 +02002343 video_stream_encoder_->OnBitrateUpdated(
2344 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asapersson09f05612017-05-15 23:40:18 -07002345
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002346 // Enable DISABLED preference, no initial limitation.
asapersson09f05612017-05-15 23:40:18 -07002347 test::FrameForwarder source;
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002348 video_stream_encoder_->SetSource(&source,
2349 webrtc::DegradationPreference::DISABLED);
asapersson09f05612017-05-15 23:40:18 -07002350
2351 source.IncomingCapturedFrame(CreateFrame(1, kWidth, kHeight));
2352 sink_.WaitForEncodedFrame(kWidth, kHeight);
Åsa Persson8c1bf952018-09-13 10:42:19 +02002353 VerifyFpsMaxResolutionMax(source.sink_wants());
asapersson09f05612017-05-15 23:40:18 -07002354 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
2355 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_framerate);
2356 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2357
2358 // Trigger adapt up, expect no change.
mflodmancc3d4422017-08-03 08:27:51 -07002359 video_stream_encoder_->TriggerQualityHigh();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002360 VerifyFpsMaxResolutionMax(source.sink_wants());
asapersson09f05612017-05-15 23:40:18 -07002361 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
2362 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_framerate);
2363 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2364
mflodmancc3d4422017-08-03 08:27:51 -07002365 video_stream_encoder_->Stop();
asapersson09f05612017-05-15 23:40:18 -07002366}
2367
mflodmancc3d4422017-08-03 08:27:51 -07002368TEST_F(VideoStreamEncoderTest,
2369 AdaptsResolutionForLowQuality_MaintainFramerateMode) {
asapersson02465b82017-04-10 01:12:52 -07002370 const int kWidth = 1280;
2371 const int kHeight = 720;
Erik Språng4c6ca302019-04-08 15:14:01 +02002372 video_stream_encoder_->OnBitrateUpdated(
2373 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asapersson02465b82017-04-10 01:12:52 -07002374
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002375 // Enable MAINTAIN_FRAMERATE preference, no initial limitation.
asapersson02465b82017-04-10 01:12:52 -07002376 AdaptingFrameForwarder source;
2377 source.set_adaptation_enabled(true);
mflodmancc3d4422017-08-03 08:27:51 -07002378 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002379 &source, webrtc::DegradationPreference::MAINTAIN_FRAMERATE);
asapersson02465b82017-04-10 01:12:52 -07002380
2381 source.IncomingCapturedFrame(CreateFrame(1, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002382 WaitForEncodedFrame(1);
Åsa Persson8c1bf952018-09-13 10:42:19 +02002383 VerifyFpsMaxResolutionMax(source.sink_wants());
asapersson02465b82017-04-10 01:12:52 -07002384 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
2385 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2386
2387 // Trigger adapt down, expect scaled down resolution.
mflodmancc3d4422017-08-03 08:27:51 -07002388 video_stream_encoder_->TriggerQualityLow();
asapersson02465b82017-04-10 01:12:52 -07002389 source.IncomingCapturedFrame(CreateFrame(2, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002390 WaitForEncodedFrame(2);
asapersson09f05612017-05-15 23:40:18 -07002391 VerifyFpsMaxResolutionLt(source.sink_wants(), kWidth * kHeight);
asapersson02465b82017-04-10 01:12:52 -07002392 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
2393 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2394
2395 // Trigger adapt up, expect no restriction.
mflodmancc3d4422017-08-03 08:27:51 -07002396 video_stream_encoder_->TriggerQualityHigh();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002397 VerifyFpsMaxResolutionMax(source.sink_wants());
asapersson02465b82017-04-10 01:12:52 -07002398 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
2399 EXPECT_EQ(2, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2400 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2401
mflodmancc3d4422017-08-03 08:27:51 -07002402 video_stream_encoder_->Stop();
asapersson02465b82017-04-10 01:12:52 -07002403}
2404
mflodmancc3d4422017-08-03 08:27:51 -07002405TEST_F(VideoStreamEncoderTest,
2406 AdaptsFramerateForLowQuality_MaintainResolutionMode) {
asapersson09f05612017-05-15 23:40:18 -07002407 const int kWidth = 1280;
2408 const int kHeight = 720;
2409 const int kInputFps = 30;
Erik Språng4c6ca302019-04-08 15:14:01 +02002410 video_stream_encoder_->OnBitrateUpdated(
2411 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asapersson09f05612017-05-15 23:40:18 -07002412
2413 VideoSendStream::Stats stats = stats_proxy_->GetStats();
2414 stats.input_frame_rate = kInputFps;
2415 stats_proxy_->SetMockStats(stats);
2416
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002417 // Expect no scaling to begin with (preference: MAINTAIN_FRAMERATE).
asapersson09f05612017-05-15 23:40:18 -07002418 video_source_.IncomingCapturedFrame(CreateFrame(1, kWidth, kHeight));
2419 sink_.WaitForEncodedFrame(1);
Åsa Persson8c1bf952018-09-13 10:42:19 +02002420 VerifyFpsMaxResolutionMax(video_source_.sink_wants());
asapersson09f05612017-05-15 23:40:18 -07002421
2422 // Trigger adapt down, expect scaled down resolution.
mflodmancc3d4422017-08-03 08:27:51 -07002423 video_stream_encoder_->TriggerQualityLow();
asapersson09f05612017-05-15 23:40:18 -07002424 video_source_.IncomingCapturedFrame(CreateFrame(2, kWidth, kHeight));
2425 sink_.WaitForEncodedFrame(2);
2426 VerifyFpsMaxResolutionLt(video_source_.sink_wants(), kWidth * kHeight);
2427
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002428 // Enable MAINTAIN_RESOLUTION preference.
asapersson09f05612017-05-15 23:40:18 -07002429 test::FrameForwarder new_video_source;
mflodmancc3d4422017-08-03 08:27:51 -07002430 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002431 &new_video_source, webrtc::DegradationPreference::MAINTAIN_RESOLUTION);
Åsa Persson8c1bf952018-09-13 10:42:19 +02002432 VerifyFpsMaxResolutionMax(new_video_source.sink_wants());
asapersson09f05612017-05-15 23:40:18 -07002433
2434 // Trigger adapt down, expect reduced framerate.
mflodmancc3d4422017-08-03 08:27:51 -07002435 video_stream_encoder_->TriggerQualityLow();
asapersson09f05612017-05-15 23:40:18 -07002436 new_video_source.IncomingCapturedFrame(CreateFrame(3, kWidth, kHeight));
2437 sink_.WaitForEncodedFrame(3);
2438 VerifyFpsLtResolutionMax(new_video_source.sink_wants(), kInputFps);
2439
2440 // Trigger adapt up, expect no restriction.
mflodmancc3d4422017-08-03 08:27:51 -07002441 video_stream_encoder_->TriggerQualityHigh();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002442 VerifyFpsMaxResolutionMax(new_video_source.sink_wants());
asapersson09f05612017-05-15 23:40:18 -07002443
mflodmancc3d4422017-08-03 08:27:51 -07002444 video_stream_encoder_->Stop();
asapersson09f05612017-05-15 23:40:18 -07002445}
2446
mflodmancc3d4422017-08-03 08:27:51 -07002447TEST_F(VideoStreamEncoderTest, DoesNotScaleBelowSetResolutionLimit) {
asaperssond0de2952017-04-21 01:47:31 -07002448 const int kWidth = 1280;
2449 const int kHeight = 720;
2450 const size_t kNumFrames = 10;
2451
Erik Språng4c6ca302019-04-08 15:14:01 +02002452 video_stream_encoder_->OnBitrateUpdated(
2453 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
kthelgason5e13d412016-12-01 03:59:51 -08002454
asaperssond0de2952017-04-21 01:47:31 -07002455 // Enable adapter, expected input resolutions when downscaling:
asapersson142fcc92017-08-17 08:58:54 -07002456 // 1280x720 -> 960x540 -> 640x360 -> 480x270 -> 320x180 (kMinPixelsPerFrame)
asaperssond0de2952017-04-21 01:47:31 -07002457 video_source_.set_adaptation_enabled(true);
2458
2459 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
2460 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2461
2462 int downscales = 0;
2463 for (size_t i = 1; i <= kNumFrames; i++) {
Åsa Persson8c1bf952018-09-13 10:42:19 +02002464 video_source_.IncomingCapturedFrame(
2465 CreateFrame(i * kFrameIntervalMs, kWidth, kHeight));
2466 WaitForEncodedFrame(i * kFrameIntervalMs);
asaperssond0de2952017-04-21 01:47:31 -07002467
asaperssonfab67072017-04-04 05:51:49 -07002468 // Trigger scale down.
asaperssond0de2952017-04-21 01:47:31 -07002469 rtc::VideoSinkWants last_wants = video_source_.sink_wants();
mflodmancc3d4422017-08-03 08:27:51 -07002470 video_stream_encoder_->TriggerQualityLow();
sprangc5d62e22017-04-02 23:53:04 -07002471 EXPECT_GE(video_source_.sink_wants().max_pixel_count, kMinPixelsPerFrame);
asaperssond0de2952017-04-21 01:47:31 -07002472
2473 if (video_source_.sink_wants().max_pixel_count < last_wants.max_pixel_count)
2474 ++downscales;
2475
2476 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
2477 EXPECT_EQ(downscales,
2478 stats_proxy_->GetStats().number_of_quality_adapt_changes);
2479 EXPECT_GT(downscales, 0);
kthelgason5e13d412016-12-01 03:59:51 -08002480 }
mflodmancc3d4422017-08-03 08:27:51 -07002481 video_stream_encoder_->Stop();
asaperssond0de2952017-04-21 01:47:31 -07002482}
2483
mflodmancc3d4422017-08-03 08:27:51 -07002484TEST_F(VideoStreamEncoderTest,
asaperssond0de2952017-04-21 01:47:31 -07002485 AdaptsResolutionUpAndDownTwiceOnOveruse_MaintainFramerateMode) {
2486 const int kWidth = 1280;
2487 const int kHeight = 720;
Erik Språng4c6ca302019-04-08 15:14:01 +02002488 video_stream_encoder_->OnBitrateUpdated(
2489 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asaperssond0de2952017-04-21 01:47:31 -07002490
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002491 // Enable MAINTAIN_FRAMERATE preference, no initial limitation.
asaperssond0de2952017-04-21 01:47:31 -07002492 AdaptingFrameForwarder source;
2493 source.set_adaptation_enabled(true);
mflodmancc3d4422017-08-03 08:27:51 -07002494 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002495 &source, webrtc::DegradationPreference::MAINTAIN_FRAMERATE);
asaperssond0de2952017-04-21 01:47:31 -07002496
Åsa Persson8c1bf952018-09-13 10:42:19 +02002497 int64_t timestamp_ms = kFrameIntervalMs;
2498 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002499 WaitForEncodedFrame(kWidth, kHeight);
Åsa Persson8c1bf952018-09-13 10:42:19 +02002500 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssond0de2952017-04-21 01:47:31 -07002501 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
2502 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2503
2504 // Trigger adapt down, expect scaled down resolution.
mflodmancc3d4422017-08-03 08:27:51 -07002505 video_stream_encoder_->TriggerCpuOveruse();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002506 timestamp_ms += kFrameIntervalMs;
2507 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
2508 WaitForEncodedFrame(timestamp_ms);
asapersson09f05612017-05-15 23:40:18 -07002509 VerifyFpsMaxResolutionLt(source.sink_wants(), kWidth * kHeight);
asaperssond0de2952017-04-21 01:47:31 -07002510 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_resolution);
2511 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2512
2513 // Trigger adapt up, expect no restriction.
mflodmancc3d4422017-08-03 08:27:51 -07002514 video_stream_encoder_->TriggerCpuNormalUsage();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002515 timestamp_ms += kFrameIntervalMs;
2516 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002517 WaitForEncodedFrame(kWidth, kHeight);
Åsa Persson8c1bf952018-09-13 10:42:19 +02002518 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssond0de2952017-04-21 01:47:31 -07002519 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
2520 EXPECT_EQ(2, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2521
2522 // Trigger adapt down, expect scaled down resolution.
mflodmancc3d4422017-08-03 08:27:51 -07002523 video_stream_encoder_->TriggerCpuOveruse();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002524 timestamp_ms += kFrameIntervalMs;
2525 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
2526 WaitForEncodedFrame(timestamp_ms);
asapersson09f05612017-05-15 23:40:18 -07002527 VerifyFpsMaxResolutionLt(source.sink_wants(), kWidth * kHeight);
asaperssond0de2952017-04-21 01:47:31 -07002528 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_resolution);
2529 EXPECT_EQ(3, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2530
2531 // Trigger adapt up, expect no restriction.
mflodmancc3d4422017-08-03 08:27:51 -07002532 video_stream_encoder_->TriggerCpuNormalUsage();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002533 timestamp_ms += kFrameIntervalMs;
2534 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
asapersson09f05612017-05-15 23:40:18 -07002535 sink_.WaitForEncodedFrame(kWidth, kHeight);
Åsa Persson8c1bf952018-09-13 10:42:19 +02002536 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssond0de2952017-04-21 01:47:31 -07002537 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
2538 EXPECT_EQ(4, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2539
mflodmancc3d4422017-08-03 08:27:51 -07002540 video_stream_encoder_->Stop();
asaperssond0de2952017-04-21 01:47:31 -07002541}
2542
mflodmancc3d4422017-08-03 08:27:51 -07002543TEST_F(VideoStreamEncoderTest,
asaperssonf7e294d2017-06-13 23:25:22 -07002544 AdaptsResolutionUpAndDownTwiceForLowQuality_BalancedMode_NoFpsLimit) {
2545 const int kWidth = 1280;
2546 const int kHeight = 720;
Erik Språng4c6ca302019-04-08 15:14:01 +02002547 video_stream_encoder_->OnBitrateUpdated(
2548 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asaperssonf7e294d2017-06-13 23:25:22 -07002549
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002550 // Enable BALANCED preference, no initial limitation.
asaperssonf7e294d2017-06-13 23:25:22 -07002551 AdaptingFrameForwarder source;
2552 source.set_adaptation_enabled(true);
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002553 video_stream_encoder_->SetSource(&source,
2554 webrtc::DegradationPreference::BALANCED);
asaperssonf7e294d2017-06-13 23:25:22 -07002555
Åsa Persson8c1bf952018-09-13 10:42:19 +02002556 int64_t timestamp_ms = kFrameIntervalMs;
2557 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
asaperssonf7e294d2017-06-13 23:25:22 -07002558 sink_.WaitForEncodedFrame(kWidth, kHeight);
Åsa Persson8c1bf952018-09-13 10:42:19 +02002559 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssonf7e294d2017-06-13 23:25:22 -07002560 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
2561 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2562
2563 // Trigger adapt down, expect scaled down resolution.
mflodmancc3d4422017-08-03 08:27:51 -07002564 video_stream_encoder_->TriggerQualityLow();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002565 timestamp_ms += kFrameIntervalMs;
2566 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
2567 sink_.WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07002568 VerifyFpsMaxResolutionLt(source.sink_wants(), kWidth * kHeight);
2569 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
2570 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2571
2572 // Trigger adapt up, expect no restriction.
mflodmancc3d4422017-08-03 08:27:51 -07002573 video_stream_encoder_->TriggerQualityHigh();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002574 timestamp_ms += kFrameIntervalMs;
2575 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
asaperssonf7e294d2017-06-13 23:25:22 -07002576 sink_.WaitForEncodedFrame(kWidth, kHeight);
Åsa Persson8c1bf952018-09-13 10:42:19 +02002577 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssonf7e294d2017-06-13 23:25:22 -07002578 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
2579 EXPECT_EQ(2, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2580
2581 // Trigger adapt down, expect scaled down resolution.
mflodmancc3d4422017-08-03 08:27:51 -07002582 video_stream_encoder_->TriggerQualityLow();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002583 timestamp_ms += kFrameIntervalMs;
2584 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
2585 sink_.WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07002586 VerifyFpsMaxResolutionLt(source.sink_wants(), kWidth * kHeight);
2587 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
2588 EXPECT_EQ(3, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2589
2590 // Trigger adapt up, expect no restriction.
mflodmancc3d4422017-08-03 08:27:51 -07002591 video_stream_encoder_->TriggerQualityHigh();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002592 timestamp_ms += kFrameIntervalMs;
2593 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
asaperssonf7e294d2017-06-13 23:25:22 -07002594 sink_.WaitForEncodedFrame(kWidth, kHeight);
Åsa Persson8c1bf952018-09-13 10:42:19 +02002595 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssonf7e294d2017-06-13 23:25:22 -07002596 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
2597 EXPECT_EQ(4, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2598
mflodmancc3d4422017-08-03 08:27:51 -07002599 video_stream_encoder_->Stop();
asaperssonf7e294d2017-06-13 23:25:22 -07002600}
2601
Åsa Persson1b247f12019-08-14 17:26:39 +02002602TEST_F(VideoStreamEncoderTest, NoAdaptUpIfBwEstimateIsLessThanMinBitrate) {
2603 webrtc::test::ScopedFieldTrials field_trials(
2604 "WebRTC-Video-BalancedDegradationSettings/"
2605 "pixels:57600|129600|230400,fps:7|10|14,kbps:0|0|425/");
2606 // Reset encoder for field trials to take effect.
2607 ConfigureEncoder(video_encoder_config_.Copy());
2608
2609 const int kWidth = 640; // pixels:640x360=230400
2610 const int kHeight = 360;
2611 const int64_t kFrameIntervalMs = 150;
2612 const int kMinBitrateBps = 425000;
2613 const int kTooLowMinBitrateBps = 424000;
2614 video_stream_encoder_->OnBitrateUpdated(DataRate::bps(kTooLowMinBitrateBps),
2615 DataRate::bps(kTooLowMinBitrateBps),
2616 0, 0);
2617
2618 // Enable BALANCED preference, no initial limitation.
2619 AdaptingFrameForwarder source;
2620 source.set_adaptation_enabled(true);
2621 video_stream_encoder_->SetSource(&source,
2622 webrtc::DegradationPreference::BALANCED);
2623
2624 int64_t timestamp_ms = kFrameIntervalMs;
2625 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
2626 sink_.WaitForEncodedFrame(kWidth, kHeight);
2627 VerifyFpsMaxResolutionMax(source.sink_wants());
2628 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_framerate);
2629 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
2630 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2631
2632 // Trigger adapt down, expect scaled down framerate (640x360@14fps).
2633 video_stream_encoder_->TriggerQualityLow();
2634 timestamp_ms += kFrameIntervalMs;
2635 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
2636 sink_.WaitForEncodedFrame(timestamp_ms);
2637 VerifyFpsEqResolutionMax(source.sink_wants(), 14);
2638 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_framerate);
2639 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
2640 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2641
2642 // Trigger adapt down, expect scaled down resolution (480x270@14fps).
2643 video_stream_encoder_->TriggerQualityLow();
2644 timestamp_ms += kFrameIntervalMs;
2645 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
2646 sink_.WaitForEncodedFrame(timestamp_ms);
2647 VerifyFpsEqResolutionLt(source.sink_wants(), source.last_wants());
2648 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_framerate);
2649 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
2650 EXPECT_EQ(2, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2651
2652 // Trigger adapt up, expect no upscale (target bitrate < min bitrate).
2653 video_stream_encoder_->TriggerQualityHigh();
2654 timestamp_ms += kFrameIntervalMs;
2655 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
2656 sink_.WaitForEncodedFrame(timestamp_ms);
2657 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_framerate);
2658 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
2659 EXPECT_EQ(2, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2660
2661 // Trigger adapt up, expect upscaled resolution (target bitrate == min
2662 // bitrate).
2663 video_stream_encoder_->OnBitrateUpdated(DataRate::bps(kMinBitrateBps),
2664 DataRate::bps(kMinBitrateBps), 0, 0);
2665 video_stream_encoder_->TriggerQualityHigh();
2666 timestamp_ms += kFrameIntervalMs;
2667 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
2668 sink_.WaitForEncodedFrame(timestamp_ms);
2669 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_framerate);
2670 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
2671 EXPECT_EQ(3, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2672
2673 video_stream_encoder_->Stop();
2674}
2675
mflodmancc3d4422017-08-03 08:27:51 -07002676TEST_F(VideoStreamEncoderTest,
asaperssond0de2952017-04-21 01:47:31 -07002677 AdaptsResolutionOnOveruseAndLowQuality_MaintainFramerateMode) {
2678 const int kWidth = 1280;
2679 const int kHeight = 720;
Erik Språng4c6ca302019-04-08 15:14:01 +02002680 video_stream_encoder_->OnBitrateUpdated(
2681 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asaperssond0de2952017-04-21 01:47:31 -07002682
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002683 // Enable MAINTAIN_FRAMERATE preference, no initial limitation.
asaperssond0de2952017-04-21 01:47:31 -07002684 AdaptingFrameForwarder source;
2685 source.set_adaptation_enabled(true);
mflodmancc3d4422017-08-03 08:27:51 -07002686 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002687 &source, webrtc::DegradationPreference::MAINTAIN_FRAMERATE);
asaperssond0de2952017-04-21 01:47:31 -07002688
Åsa Persson8c1bf952018-09-13 10:42:19 +02002689 int64_t timestamp_ms = kFrameIntervalMs;
2690 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002691 WaitForEncodedFrame(kWidth, kHeight);
Åsa Persson8c1bf952018-09-13 10:42:19 +02002692 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssond0de2952017-04-21 01:47:31 -07002693 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
2694 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
2695 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2696 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2697
2698 // Trigger cpu adapt down, expect scaled down resolution (960x540).
mflodmancc3d4422017-08-03 08:27:51 -07002699 video_stream_encoder_->TriggerCpuOveruse();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002700 timestamp_ms += kFrameIntervalMs;
2701 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
2702 WaitForEncodedFrame(timestamp_ms);
asapersson09f05612017-05-15 23:40:18 -07002703 VerifyFpsMaxResolutionLt(source.sink_wants(), kWidth * kHeight);
asaperssond0de2952017-04-21 01:47:31 -07002704 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_resolution);
2705 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
2706 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2707 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2708
2709 // Trigger cpu adapt down, expect scaled down resolution (640x360).
mflodmancc3d4422017-08-03 08:27:51 -07002710 video_stream_encoder_->TriggerCpuOveruse();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002711 timestamp_ms += kFrameIntervalMs;
2712 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
2713 WaitForEncodedFrame(timestamp_ms);
asapersson09f05612017-05-15 23:40:18 -07002714 VerifyFpsMaxResolutionLt(source.sink_wants(), source.last_wants());
asaperssond0de2952017-04-21 01:47:31 -07002715 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_resolution);
2716 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
2717 EXPECT_EQ(2, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2718 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2719
Jonathan Yubc771b72017-12-08 17:04:29 -08002720 // Trigger cpu adapt down, expect scaled down resolution (480x270).
mflodmancc3d4422017-08-03 08:27:51 -07002721 video_stream_encoder_->TriggerCpuOveruse();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002722 timestamp_ms += kFrameIntervalMs;
2723 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
2724 WaitForEncodedFrame(timestamp_ms);
Jonathan Yubc771b72017-12-08 17:04:29 -08002725 VerifyFpsMaxResolutionLt(source.sink_wants(), source.last_wants());
asaperssond0de2952017-04-21 01:47:31 -07002726 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_resolution);
2727 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
Jonathan Yubc771b72017-12-08 17:04:29 -08002728 EXPECT_EQ(3, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
asaperssond0de2952017-04-21 01:47:31 -07002729 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2730
Jonathan Yubc771b72017-12-08 17:04:29 -08002731 // Trigger quality adapt down, expect scaled down resolution (320x180).
mflodmancc3d4422017-08-03 08:27:51 -07002732 video_stream_encoder_->TriggerQualityLow();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002733 timestamp_ms += kFrameIntervalMs;
2734 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
2735 WaitForEncodedFrame(timestamp_ms);
asapersson09f05612017-05-15 23:40:18 -07002736 VerifyFpsMaxResolutionLt(source.sink_wants(), source.last_wants());
Jonathan Yubc771b72017-12-08 17:04:29 -08002737 rtc::VideoSinkWants last_wants = source.sink_wants();
asaperssond0de2952017-04-21 01:47:31 -07002738 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_resolution);
2739 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
2740 EXPECT_EQ(3, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2741 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2742
Jonathan Yubc771b72017-12-08 17:04:29 -08002743 // Trigger quality adapt down, expect no change (min resolution reached).
2744 video_stream_encoder_->TriggerQualityLow();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002745 timestamp_ms += kFrameIntervalMs;
2746 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
2747 WaitForEncodedFrame(timestamp_ms);
Jonathan Yubc771b72017-12-08 17:04:29 -08002748 VerifyFpsMaxResolutionEq(source.sink_wants(), last_wants);
2749 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_resolution);
2750 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
2751 EXPECT_EQ(3, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2752 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2753
2754 // Trigger cpu adapt up, expect upscaled resolution (480x270).
mflodmancc3d4422017-08-03 08:27:51 -07002755 video_stream_encoder_->TriggerCpuNormalUsage();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002756 timestamp_ms += kFrameIntervalMs;
2757 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
2758 WaitForEncodedFrame(timestamp_ms);
asapersson09f05612017-05-15 23:40:18 -07002759 VerifyFpsMaxResolutionGt(source.sink_wants(), source.last_wants());
Jonathan Yubc771b72017-12-08 17:04:29 -08002760 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_resolution);
2761 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
2762 EXPECT_EQ(4, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2763 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2764
2765 // Trigger cpu adapt up, expect upscaled resolution (640x360).
2766 video_stream_encoder_->TriggerCpuNormalUsage();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002767 timestamp_ms += kFrameIntervalMs;
2768 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
2769 WaitForEncodedFrame(timestamp_ms);
Jonathan Yubc771b72017-12-08 17:04:29 -08002770 VerifyFpsMaxResolutionGt(source.sink_wants(), source.last_wants());
2771 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_resolution);
2772 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
2773 EXPECT_EQ(5, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
2774 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2775
2776 // Trigger cpu adapt up, expect upscaled resolution (960x540).
2777 video_stream_encoder_->TriggerCpuNormalUsage();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002778 timestamp_ms += kFrameIntervalMs;
2779 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
2780 WaitForEncodedFrame(timestamp_ms);
Jonathan Yubc771b72017-12-08 17:04:29 -08002781 VerifyFpsMaxResolutionGt(source.sink_wants(), source.last_wants());
asaperssond0de2952017-04-21 01:47:31 -07002782 last_wants = source.sink_wants();
2783 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
2784 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
Jonathan Yubc771b72017-12-08 17:04:29 -08002785 EXPECT_EQ(6, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
asaperssond0de2952017-04-21 01:47:31 -07002786 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2787
2788 // Trigger cpu adapt up, no cpu downgrades, expect no change (960x540).
mflodmancc3d4422017-08-03 08:27:51 -07002789 video_stream_encoder_->TriggerCpuNormalUsage();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002790 timestamp_ms += kFrameIntervalMs;
2791 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
2792 WaitForEncodedFrame(timestamp_ms);
asapersson09f05612017-05-15 23:40:18 -07002793 VerifyFpsEqResolutionEq(source.sink_wants(), last_wants);
asaperssond0de2952017-04-21 01:47:31 -07002794 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
2795 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
Jonathan Yubc771b72017-12-08 17:04:29 -08002796 EXPECT_EQ(6, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
asaperssond0de2952017-04-21 01:47:31 -07002797 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_quality_adapt_changes);
2798
2799 // Trigger quality adapt up, expect no restriction (1280x720).
mflodmancc3d4422017-08-03 08:27:51 -07002800 video_stream_encoder_->TriggerQualityHigh();
Åsa Persson8c1bf952018-09-13 10:42:19 +02002801 timestamp_ms += kFrameIntervalMs;
2802 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002803 WaitForEncodedFrame(kWidth, kHeight);
asapersson09f05612017-05-15 23:40:18 -07002804 VerifyFpsMaxResolutionGt(source.sink_wants(), source.last_wants());
Åsa Persson8c1bf952018-09-13 10:42:19 +02002805 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssond0de2952017-04-21 01:47:31 -07002806 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
2807 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
Jonathan Yubc771b72017-12-08 17:04:29 -08002808 EXPECT_EQ(6, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
asaperssond0de2952017-04-21 01:47:31 -07002809 EXPECT_EQ(2, stats_proxy_->GetStats().number_of_quality_adapt_changes);
kthelgason5e13d412016-12-01 03:59:51 -08002810
mflodmancc3d4422017-08-03 08:27:51 -07002811 video_stream_encoder_->Stop();
kthelgason5e13d412016-12-01 03:59:51 -08002812}
2813
mflodmancc3d4422017-08-03 08:27:51 -07002814TEST_F(VideoStreamEncoderTest, CpuLimitedHistogramIsReported) {
asaperssonfab67072017-04-04 05:51:49 -07002815 const int kWidth = 640;
2816 const int kHeight = 360;
perkj803d97f2016-11-01 11:45:46 -07002817
Erik Språng4c6ca302019-04-08 15:14:01 +02002818 video_stream_encoder_->OnBitrateUpdated(
2819 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
sprang4847ae62017-06-27 07:06:52 -07002820
perkj803d97f2016-11-01 11:45:46 -07002821 for (int i = 1; i <= SendStatisticsProxy::kMinRequiredMetricsSamples; ++i) {
asaperssonfab67072017-04-04 05:51:49 -07002822 video_source_.IncomingCapturedFrame(CreateFrame(i, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002823 WaitForEncodedFrame(i);
perkj803d97f2016-11-01 11:45:46 -07002824 }
2825
mflodmancc3d4422017-08-03 08:27:51 -07002826 video_stream_encoder_->TriggerCpuOveruse();
perkj803d97f2016-11-01 11:45:46 -07002827 for (int i = 1; i <= SendStatisticsProxy::kMinRequiredMetricsSamples; ++i) {
asaperssonfab67072017-04-04 05:51:49 -07002828 video_source_.IncomingCapturedFrame(CreateFrame(
2829 SendStatisticsProxy::kMinRequiredMetricsSamples + i, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002830 WaitForEncodedFrame(SendStatisticsProxy::kMinRequiredMetricsSamples + i);
perkj803d97f2016-11-01 11:45:46 -07002831 }
2832
mflodmancc3d4422017-08-03 08:27:51 -07002833 video_stream_encoder_->Stop();
2834 video_stream_encoder_.reset();
perkj803d97f2016-11-01 11:45:46 -07002835 stats_proxy_.reset();
sprangf8ee65e2017-02-28 08:49:33 -08002836
perkj803d97f2016-11-01 11:45:46 -07002837 EXPECT_EQ(1,
2838 metrics::NumSamples("WebRTC.Video.CpuLimitedResolutionInPercent"));
2839 EXPECT_EQ(
2840 1, metrics::NumEvents("WebRTC.Video.CpuLimitedResolutionInPercent", 50));
2841}
2842
mflodmancc3d4422017-08-03 08:27:51 -07002843TEST_F(VideoStreamEncoderTest,
2844 CpuLimitedHistogramIsNotReportedForDisabledDegradation) {
Erik Språng4c6ca302019-04-08 15:14:01 +02002845 video_stream_encoder_->OnBitrateUpdated(
2846 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asaperssonf4e44af2017-04-19 02:01:06 -07002847 const int kWidth = 640;
2848 const int kHeight = 360;
2849
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002850 video_stream_encoder_->SetSource(&video_source_,
2851 webrtc::DegradationPreference::DISABLED);
asaperssonf4e44af2017-04-19 02:01:06 -07002852
2853 for (int i = 1; i <= SendStatisticsProxy::kMinRequiredMetricsSamples; ++i) {
2854 video_source_.IncomingCapturedFrame(CreateFrame(i, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07002855 WaitForEncodedFrame(i);
asaperssonf4e44af2017-04-19 02:01:06 -07002856 }
2857
mflodmancc3d4422017-08-03 08:27:51 -07002858 video_stream_encoder_->Stop();
2859 video_stream_encoder_.reset();
asaperssonf4e44af2017-04-19 02:01:06 -07002860 stats_proxy_.reset();
2861
2862 EXPECT_EQ(0,
2863 metrics::NumSamples("WebRTC.Video.CpuLimitedResolutionInPercent"));
2864}
2865
mflodmancc3d4422017-08-03 08:27:51 -07002866TEST_F(VideoStreamEncoderTest, CallsBitrateObserver) {
sprang4847ae62017-06-27 07:06:52 -07002867 MockBitrateObserver bitrate_observer;
Niels Möller0327c2d2018-05-21 14:09:31 +02002868 video_stream_encoder_->SetBitrateAllocationObserver(&bitrate_observer);
sprang57c2fff2017-01-16 06:24:02 -08002869
2870 const int kDefaultFps = 30;
Erik Språng566124a2018-04-23 12:32:22 +02002871 const VideoBitrateAllocation expected_bitrate =
sprang57c2fff2017-01-16 06:24:02 -08002872 DefaultVideoBitrateAllocator(fake_encoder_.codec_config())
Florent Castelli8bbdb5b2019-08-02 15:16:28 +02002873 .Allocate(VideoBitrateAllocationParameters(kLowTargetBitrateBps,
2874 kDefaultFps));
sprang57c2fff2017-01-16 06:24:02 -08002875
sprang57c2fff2017-01-16 06:24:02 -08002876 EXPECT_CALL(bitrate_observer, OnBitrateAllocationUpdated(expected_bitrate))
Niels Möller6bb5ab92019-01-11 11:11:10 +01002877 .Times(1);
Erik Språng610c7632019-03-06 15:37:33 +01002878 video_stream_encoder_->OnBitrateUpdated(DataRate::bps(kLowTargetBitrateBps),
Erik Språng4c6ca302019-04-08 15:14:01 +02002879 DataRate::bps(kLowTargetBitrateBps),
2880 0, 0);
sprang57c2fff2017-01-16 06:24:02 -08002881
sprang57c2fff2017-01-16 06:24:02 -08002882 video_source_.IncomingCapturedFrame(
Erik Språngd7329ca2019-02-21 21:19:53 +01002883 CreateFrame(rtc::TimeMillis(), codec_width_, codec_height_));
2884 WaitForEncodedFrame(rtc::TimeMillis());
2885 absl::optional<VideoBitrateAllocation> bitrate_allocation =
2886 fake_encoder_.GetAndResetLastBitrateAllocation();
2887 // Check that encoder has been updated too, not just allocation observer.
2888 EXPECT_EQ(bitrate_allocation->get_sum_bps(), kLowTargetBitrateBps);
Sebastian Jansson40889f32019-04-17 12:11:20 +02002889 // TODO(srte): The use of millisecs here looks like an error, but the tests
2890 // fails using seconds, this should be investigated.
2891 fake_clock_.AdvanceTime(TimeDelta::ms(1) / kDefaultFps);
sprang57c2fff2017-01-16 06:24:02 -08002892
2893 // Not called on second frame.
2894 EXPECT_CALL(bitrate_observer, OnBitrateAllocationUpdated(expected_bitrate))
2895 .Times(0);
2896 video_source_.IncomingCapturedFrame(
Erik Språngd7329ca2019-02-21 21:19:53 +01002897 CreateFrame(rtc::TimeMillis(), codec_width_, codec_height_));
2898 WaitForEncodedFrame(rtc::TimeMillis());
Sebastian Jansson40889f32019-04-17 12:11:20 +02002899 fake_clock_.AdvanceTime(TimeDelta::ms(1) / kDefaultFps);
sprang57c2fff2017-01-16 06:24:02 -08002900
2901 // Called after a process interval.
2902 const int64_t kProcessIntervalMs =
2903 vcm::VCMProcessTimer::kDefaultProcessIntervalMs;
sprang57c2fff2017-01-16 06:24:02 -08002904 EXPECT_CALL(bitrate_observer, OnBitrateAllocationUpdated(expected_bitrate))
2905 .Times(1);
Erik Språngd7329ca2019-02-21 21:19:53 +01002906 const int64_t start_time_ms = rtc::TimeMillis();
2907 while (rtc::TimeMillis() - start_time_ms < kProcessIntervalMs) {
2908 video_source_.IncomingCapturedFrame(
2909 CreateFrame(rtc::TimeMillis(), codec_width_, codec_height_));
2910 WaitForEncodedFrame(rtc::TimeMillis());
Sebastian Jansson40889f32019-04-17 12:11:20 +02002911 fake_clock_.AdvanceTime(TimeDelta::ms(1) / kDefaultFps);
Erik Språngd7329ca2019-02-21 21:19:53 +01002912 }
2913
2914 // Since rates are unchanged, encoder should not be reconfigured.
2915 EXPECT_FALSE(fake_encoder_.GetAndResetLastBitrateAllocation().has_value());
sprang57c2fff2017-01-16 06:24:02 -08002916
mflodmancc3d4422017-08-03 08:27:51 -07002917 video_stream_encoder_->Stop();
sprang57c2fff2017-01-16 06:24:02 -08002918}
2919
Åsa Perssonc29cb2c2019-03-25 12:06:59 +01002920TEST_F(VideoStreamEncoderTest, TemporalLayersNotDisabledIfSupported) {
2921 // 2 TLs configured, temporal layers supported by encoder.
2922 const int kNumTemporalLayers = 2;
2923 ResetEncoder("VP8", 1, kNumTemporalLayers, 1, /*screenshare*/ false);
2924 fake_encoder_.SetTemporalLayersSupported(0, true);
2925
2926 // Bitrate allocated across temporal layers.
2927 const int kTl0Bps = kTargetBitrateBps *
2928 webrtc::SimulcastRateAllocator::GetTemporalRateAllocation(
2929 kNumTemporalLayers, /*temporal_id*/ 0);
2930 const int kTl1Bps = kTargetBitrateBps *
2931 webrtc::SimulcastRateAllocator::GetTemporalRateAllocation(
2932 kNumTemporalLayers, /*temporal_id*/ 1);
2933 VideoBitrateAllocation expected_bitrate;
2934 expected_bitrate.SetBitrate(/*si*/ 0, /*ti*/ 0, kTl0Bps);
2935 expected_bitrate.SetBitrate(/*si*/ 0, /*ti*/ 1, kTl1Bps - kTl0Bps);
2936
2937 VerifyAllocatedBitrate(expected_bitrate);
2938 video_stream_encoder_->Stop();
2939}
2940
2941TEST_F(VideoStreamEncoderTest, TemporalLayersDisabledIfNotSupported) {
2942 // 2 TLs configured, temporal layers not supported by encoder.
2943 ResetEncoder("VP8", 1, /*num_temporal_layers*/ 2, 1, /*screenshare*/ false);
2944 fake_encoder_.SetTemporalLayersSupported(0, false);
2945
2946 // Temporal layers not supported by the encoder.
2947 // Total bitrate should be at ti:0.
2948 VideoBitrateAllocation expected_bitrate;
2949 expected_bitrate.SetBitrate(/*si*/ 0, /*ti*/ 0, kTargetBitrateBps);
2950
2951 VerifyAllocatedBitrate(expected_bitrate);
2952 video_stream_encoder_->Stop();
2953}
2954
2955TEST_F(VideoStreamEncoderTest, VerifyBitrateAllocationForTwoStreams) {
2956 // 2 TLs configured, temporal layers only supported for first stream.
2957 ResetEncoder("VP8", 2, /*num_temporal_layers*/ 2, 1, /*screenshare*/ false);
2958 fake_encoder_.SetTemporalLayersSupported(0, true);
2959 fake_encoder_.SetTemporalLayersSupported(1, false);
2960
2961 const int kS0Bps = 150000;
2962 const int kS0Tl0Bps =
2963 kS0Bps * webrtc::SimulcastRateAllocator::GetTemporalRateAllocation(
2964 /*num_layers*/ 2, /*temporal_id*/ 0);
2965 const int kS0Tl1Bps =
2966 kS0Bps * webrtc::SimulcastRateAllocator::GetTemporalRateAllocation(
2967 /*num_layers*/ 2, /*temporal_id*/ 1);
2968 const int kS1Bps = kTargetBitrateBps - kS0Tl1Bps;
2969 // Temporal layers not supported by si:1.
2970 VideoBitrateAllocation expected_bitrate;
2971 expected_bitrate.SetBitrate(/*si*/ 0, /*ti*/ 0, kS0Tl0Bps);
2972 expected_bitrate.SetBitrate(/*si*/ 0, /*ti*/ 1, kS0Tl1Bps - kS0Tl0Bps);
2973 expected_bitrate.SetBitrate(/*si*/ 1, /*ti*/ 0, kS1Bps);
2974
2975 VerifyAllocatedBitrate(expected_bitrate);
2976 video_stream_encoder_->Stop();
2977}
2978
Niels Möller7dc26b72017-12-06 10:27:48 +01002979TEST_F(VideoStreamEncoderTest, OveruseDetectorUpdatedOnReconfigureAndAdaption) {
2980 const int kFrameWidth = 1280;
2981 const int kFrameHeight = 720;
2982 const int kFramerate = 24;
2983
Erik Språng4c6ca302019-04-08 15:14:01 +02002984 video_stream_encoder_->OnBitrateUpdated(
2985 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
Niels Möller7dc26b72017-12-06 10:27:48 +01002986 test::FrameForwarder source;
2987 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07002988 &source, webrtc::DegradationPreference::MAINTAIN_RESOLUTION);
Niels Möller7dc26b72017-12-06 10:27:48 +01002989
2990 // Insert a single frame, triggering initial configuration.
2991 source.IncomingCapturedFrame(CreateFrame(1, kFrameWidth, kFrameHeight));
2992 video_stream_encoder_->WaitUntilTaskQueueIsIdle();
2993
2994 EXPECT_EQ(
2995 video_stream_encoder_->overuse_detector_proxy_->GetLastTargetFramerate(),
2996 kDefaultFramerate);
2997
2998 // Trigger reconfigure encoder (without resetting the entire instance).
2999 VideoEncoderConfig video_encoder_config;
Niels Möller259a4972018-04-05 15:36:51 +02003000 video_encoder_config.codec_type = kVideoCodecVP8;
Niels Möller7dc26b72017-12-06 10:27:48 +01003001 video_encoder_config.max_bitrate_bps = kTargetBitrateBps;
3002 video_encoder_config.number_of_streams = 1;
3003 video_encoder_config.video_stream_factory =
3004 new rtc::RefCountedObject<VideoStreamFactory>(1, kFramerate);
3005 video_stream_encoder_->ConfigureEncoder(std::move(video_encoder_config),
Niels Möllerf1338562018-04-26 09:51:47 +02003006 kMaxPayloadLength);
Niels Möller7dc26b72017-12-06 10:27:48 +01003007 video_stream_encoder_->WaitUntilTaskQueueIsIdle();
3008
3009 // Detector should be updated with fps limit from codec config.
3010 EXPECT_EQ(
3011 video_stream_encoder_->overuse_detector_proxy_->GetLastTargetFramerate(),
3012 kFramerate);
3013
3014 // Trigger overuse, max framerate should be reduced.
3015 VideoSendStream::Stats stats = stats_proxy_->GetStats();
3016 stats.input_frame_rate = kFramerate;
3017 stats_proxy_->SetMockStats(stats);
3018 video_stream_encoder_->TriggerCpuOveruse();
3019 video_stream_encoder_->WaitUntilTaskQueueIsIdle();
3020 int adapted_framerate =
3021 video_stream_encoder_->overuse_detector_proxy_->GetLastTargetFramerate();
3022 EXPECT_LT(adapted_framerate, kFramerate);
3023
3024 // Trigger underuse, max framerate should go back to codec configured fps.
3025 // Set extra low fps, to make sure it's actually reset, not just incremented.
3026 stats = stats_proxy_->GetStats();
3027 stats.input_frame_rate = adapted_framerate / 2;
3028 stats_proxy_->SetMockStats(stats);
3029 video_stream_encoder_->TriggerCpuNormalUsage();
3030 video_stream_encoder_->WaitUntilTaskQueueIsIdle();
3031 EXPECT_EQ(
3032 video_stream_encoder_->overuse_detector_proxy_->GetLastTargetFramerate(),
3033 kFramerate);
3034
3035 video_stream_encoder_->Stop();
3036}
3037
3038TEST_F(VideoStreamEncoderTest,
3039 OveruseDetectorUpdatedRespectsFramerateAfterUnderuse) {
3040 const int kFrameWidth = 1280;
3041 const int kFrameHeight = 720;
3042 const int kLowFramerate = 15;
3043 const int kHighFramerate = 25;
3044
Erik Språng4c6ca302019-04-08 15:14:01 +02003045 video_stream_encoder_->OnBitrateUpdated(
3046 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
Niels Möller7dc26b72017-12-06 10:27:48 +01003047 test::FrameForwarder source;
3048 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07003049 &source, webrtc::DegradationPreference::MAINTAIN_RESOLUTION);
Niels Möller7dc26b72017-12-06 10:27:48 +01003050
3051 // Trigger initial configuration.
3052 VideoEncoderConfig video_encoder_config;
Niels Möller259a4972018-04-05 15:36:51 +02003053 video_encoder_config.codec_type = kVideoCodecVP8;
Niels Möller7dc26b72017-12-06 10:27:48 +01003054 video_encoder_config.max_bitrate_bps = kTargetBitrateBps;
3055 video_encoder_config.number_of_streams = 1;
3056 video_encoder_config.video_stream_factory =
3057 new rtc::RefCountedObject<VideoStreamFactory>(1, kLowFramerate);
3058 source.IncomingCapturedFrame(CreateFrame(1, kFrameWidth, kFrameHeight));
3059 video_stream_encoder_->ConfigureEncoder(std::move(video_encoder_config),
Niels Möllerf1338562018-04-26 09:51:47 +02003060 kMaxPayloadLength);
Niels Möller7dc26b72017-12-06 10:27:48 +01003061 video_stream_encoder_->WaitUntilTaskQueueIsIdle();
3062
3063 EXPECT_EQ(
3064 video_stream_encoder_->overuse_detector_proxy_->GetLastTargetFramerate(),
3065 kLowFramerate);
3066
3067 // Trigger overuse, max framerate should be reduced.
3068 VideoSendStream::Stats stats = stats_proxy_->GetStats();
3069 stats.input_frame_rate = kLowFramerate;
3070 stats_proxy_->SetMockStats(stats);
3071 video_stream_encoder_->TriggerCpuOveruse();
3072 video_stream_encoder_->WaitUntilTaskQueueIsIdle();
3073 int adapted_framerate =
3074 video_stream_encoder_->overuse_detector_proxy_->GetLastTargetFramerate();
3075 EXPECT_LT(adapted_framerate, kLowFramerate);
3076
3077 // Reconfigure the encoder with a new (higher max framerate), max fps should
3078 // still respect the adaptation.
3079 video_encoder_config.video_stream_factory =
3080 new rtc::RefCountedObject<VideoStreamFactory>(1, kHighFramerate);
3081 source.IncomingCapturedFrame(CreateFrame(1, kFrameWidth, kFrameHeight));
3082 video_stream_encoder_->ConfigureEncoder(std::move(video_encoder_config),
Niels Möllerf1338562018-04-26 09:51:47 +02003083 kMaxPayloadLength);
Niels Möller7dc26b72017-12-06 10:27:48 +01003084 video_stream_encoder_->WaitUntilTaskQueueIsIdle();
3085
3086 EXPECT_EQ(
3087 video_stream_encoder_->overuse_detector_proxy_->GetLastTargetFramerate(),
3088 adapted_framerate);
3089
3090 // Trigger underuse, max framerate should go back to codec configured fps.
3091 stats = stats_proxy_->GetStats();
3092 stats.input_frame_rate = adapted_framerate;
3093 stats_proxy_->SetMockStats(stats);
3094 video_stream_encoder_->TriggerCpuNormalUsage();
3095 video_stream_encoder_->WaitUntilTaskQueueIsIdle();
3096 EXPECT_EQ(
3097 video_stream_encoder_->overuse_detector_proxy_->GetLastTargetFramerate(),
3098 kHighFramerate);
3099
3100 video_stream_encoder_->Stop();
3101}
3102
mflodmancc3d4422017-08-03 08:27:51 -07003103TEST_F(VideoStreamEncoderTest,
3104 OveruseDetectorUpdatedOnDegradationPreferenceChange) {
sprangfda496a2017-06-15 04:21:07 -07003105 const int kFrameWidth = 1280;
3106 const int kFrameHeight = 720;
3107 const int kFramerate = 24;
3108
Erik Språng4c6ca302019-04-08 15:14:01 +02003109 video_stream_encoder_->OnBitrateUpdated(
3110 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
sprangfda496a2017-06-15 04:21:07 -07003111 test::FrameForwarder source;
mflodmancc3d4422017-08-03 08:27:51 -07003112 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07003113 &source, webrtc::DegradationPreference::MAINTAIN_RESOLUTION);
sprangfda496a2017-06-15 04:21:07 -07003114
3115 // Trigger initial configuration.
3116 VideoEncoderConfig video_encoder_config;
Niels Möller259a4972018-04-05 15:36:51 +02003117 video_encoder_config.codec_type = kVideoCodecVP8;
sprangfda496a2017-06-15 04:21:07 -07003118 video_encoder_config.max_bitrate_bps = kTargetBitrateBps;
3119 video_encoder_config.number_of_streams = 1;
3120 video_encoder_config.video_stream_factory =
3121 new rtc::RefCountedObject<VideoStreamFactory>(1, kFramerate);
3122 source.IncomingCapturedFrame(CreateFrame(1, kFrameWidth, kFrameHeight));
mflodmancc3d4422017-08-03 08:27:51 -07003123 video_stream_encoder_->ConfigureEncoder(std::move(video_encoder_config),
Niels Möllerf1338562018-04-26 09:51:47 +02003124 kMaxPayloadLength);
mflodmancc3d4422017-08-03 08:27:51 -07003125 video_stream_encoder_->WaitUntilTaskQueueIsIdle();
sprangfda496a2017-06-15 04:21:07 -07003126
Niels Möller7dc26b72017-12-06 10:27:48 +01003127 EXPECT_EQ(
3128 video_stream_encoder_->overuse_detector_proxy_->GetLastTargetFramerate(),
3129 kFramerate);
sprangfda496a2017-06-15 04:21:07 -07003130
3131 // Trigger overuse, max framerate should be reduced.
3132 VideoSendStream::Stats stats = stats_proxy_->GetStats();
3133 stats.input_frame_rate = kFramerate;
3134 stats_proxy_->SetMockStats(stats);
mflodmancc3d4422017-08-03 08:27:51 -07003135 video_stream_encoder_->TriggerCpuOveruse();
3136 video_stream_encoder_->WaitUntilTaskQueueIsIdle();
Niels Möller7dc26b72017-12-06 10:27:48 +01003137 int adapted_framerate =
3138 video_stream_encoder_->overuse_detector_proxy_->GetLastTargetFramerate();
sprangfda496a2017-06-15 04:21:07 -07003139 EXPECT_LT(adapted_framerate, kFramerate);
3140
3141 // Change degradation preference to not enable framerate scaling. Target
3142 // framerate should be changed to codec defined limit.
mflodmancc3d4422017-08-03 08:27:51 -07003143 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07003144 &source, webrtc::DegradationPreference::MAINTAIN_FRAMERATE);
mflodmancc3d4422017-08-03 08:27:51 -07003145 video_stream_encoder_->WaitUntilTaskQueueIsIdle();
Niels Möller7dc26b72017-12-06 10:27:48 +01003146 EXPECT_EQ(
3147 video_stream_encoder_->overuse_detector_proxy_->GetLastTargetFramerate(),
3148 kFramerate);
sprangfda496a2017-06-15 04:21:07 -07003149
mflodmancc3d4422017-08-03 08:27:51 -07003150 video_stream_encoder_->Stop();
sprangfda496a2017-06-15 04:21:07 -07003151}
3152
mflodmancc3d4422017-08-03 08:27:51 -07003153TEST_F(VideoStreamEncoderTest, DropsFramesAndScalesWhenBitrateIsTooLow) {
asaperssonfab67072017-04-04 05:51:49 -07003154 const int kTooLowBitrateForFrameSizeBps = 10000;
Erik Språng610c7632019-03-06 15:37:33 +01003155 video_stream_encoder_->OnBitrateUpdated(
Erik Språng4c6ca302019-04-08 15:14:01 +02003156 DataRate::bps(kTooLowBitrateForFrameSizeBps),
3157 DataRate::bps(kTooLowBitrateForFrameSizeBps), 0, 0);
asaperssonfab67072017-04-04 05:51:49 -07003158 const int kWidth = 640;
3159 const int kHeight = 360;
kthelgason2bc68642017-02-07 07:02:22 -08003160
asaperssonfab67072017-04-04 05:51:49 -07003161 video_source_.IncomingCapturedFrame(CreateFrame(1, kWidth, kHeight));
kthelgason2bc68642017-02-07 07:02:22 -08003162
3163 // Expect to drop this frame, the wait should time out.
sprang4847ae62017-06-27 07:06:52 -07003164 ExpectDroppedFrame();
kthelgason2bc68642017-02-07 07:02:22 -08003165
3166 // Expect the sink_wants to specify a scaled frame.
asapersson0944a802017-04-07 00:57:58 -07003167 EXPECT_LT(video_source_.sink_wants().max_pixel_count, kWidth * kHeight);
kthelgason2bc68642017-02-07 07:02:22 -08003168
sprangc5d62e22017-04-02 23:53:04 -07003169 int last_pixel_count = video_source_.sink_wants().max_pixel_count;
kthelgason2bc68642017-02-07 07:02:22 -08003170
asaperssonfab67072017-04-04 05:51:49 -07003171 // Next frame is scaled.
kthelgason2bc68642017-02-07 07:02:22 -08003172 video_source_.IncomingCapturedFrame(
asaperssonfab67072017-04-04 05:51:49 -07003173 CreateFrame(2, kWidth * 3 / 4, kHeight * 3 / 4));
kthelgason2bc68642017-02-07 07:02:22 -08003174
3175 // Expect to drop this frame, the wait should time out.
sprang4847ae62017-06-27 07:06:52 -07003176 ExpectDroppedFrame();
kthelgason2bc68642017-02-07 07:02:22 -08003177
sprangc5d62e22017-04-02 23:53:04 -07003178 EXPECT_LT(video_source_.sink_wants().max_pixel_count, last_pixel_count);
kthelgason2bc68642017-02-07 07:02:22 -08003179
mflodmancc3d4422017-08-03 08:27:51 -07003180 video_stream_encoder_->Stop();
kthelgason2bc68642017-02-07 07:02:22 -08003181}
3182
mflodmancc3d4422017-08-03 08:27:51 -07003183TEST_F(VideoStreamEncoderTest,
3184 NumberOfDroppedFramesLimitedWhenBitrateIsTooLow) {
asaperssonfab67072017-04-04 05:51:49 -07003185 const int kTooLowBitrateForFrameSizeBps = 10000;
Erik Språng610c7632019-03-06 15:37:33 +01003186 video_stream_encoder_->OnBitrateUpdated(
Erik Språng4c6ca302019-04-08 15:14:01 +02003187 DataRate::bps(kTooLowBitrateForFrameSizeBps),
3188 DataRate::bps(kTooLowBitrateForFrameSizeBps), 0, 0);
asaperssonfab67072017-04-04 05:51:49 -07003189 const int kWidth = 640;
3190 const int kHeight = 360;
kthelgason2bc68642017-02-07 07:02:22 -08003191
3192 // We expect the n initial frames to get dropped.
3193 int i;
3194 for (i = 1; i <= kMaxInitialFramedrop; ++i) {
asaperssonfab67072017-04-04 05:51:49 -07003195 video_source_.IncomingCapturedFrame(CreateFrame(i, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003196 ExpectDroppedFrame();
kthelgason2bc68642017-02-07 07:02:22 -08003197 }
3198 // The n+1th frame should not be dropped, even though it's size is too large.
asaperssonfab67072017-04-04 05:51:49 -07003199 video_source_.IncomingCapturedFrame(CreateFrame(i, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003200 WaitForEncodedFrame(i);
kthelgason2bc68642017-02-07 07:02:22 -08003201
3202 // Expect the sink_wants to specify a scaled frame.
asaperssonfab67072017-04-04 05:51:49 -07003203 EXPECT_LT(video_source_.sink_wants().max_pixel_count, kWidth * kHeight);
kthelgason2bc68642017-02-07 07:02:22 -08003204
mflodmancc3d4422017-08-03 08:27:51 -07003205 video_stream_encoder_->Stop();
kthelgason2bc68642017-02-07 07:02:22 -08003206}
3207
mflodmancc3d4422017-08-03 08:27:51 -07003208TEST_F(VideoStreamEncoderTest,
3209 InitialFrameDropOffWithMaintainResolutionPreference) {
asaperssonfab67072017-04-04 05:51:49 -07003210 const int kWidth = 640;
3211 const int kHeight = 360;
Erik Språng610c7632019-03-06 15:37:33 +01003212 video_stream_encoder_->OnBitrateUpdated(DataRate::bps(kLowTargetBitrateBps),
Erik Språng4c6ca302019-04-08 15:14:01 +02003213 DataRate::bps(kLowTargetBitrateBps),
3214 0, 0);
kthelgason2bc68642017-02-07 07:02:22 -08003215
3216 // Set degradation preference.
mflodmancc3d4422017-08-03 08:27:51 -07003217 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07003218 &video_source_, webrtc::DegradationPreference::MAINTAIN_RESOLUTION);
kthelgason2bc68642017-02-07 07:02:22 -08003219
asaperssonfab67072017-04-04 05:51:49 -07003220 video_source_.IncomingCapturedFrame(CreateFrame(1, kWidth, kHeight));
kthelgason2bc68642017-02-07 07:02:22 -08003221 // Frame should not be dropped, even if it's too large.
sprang4847ae62017-06-27 07:06:52 -07003222 WaitForEncodedFrame(1);
kthelgason2bc68642017-02-07 07:02:22 -08003223
mflodmancc3d4422017-08-03 08:27:51 -07003224 video_stream_encoder_->Stop();
kthelgason2bc68642017-02-07 07:02:22 -08003225}
3226
mflodmancc3d4422017-08-03 08:27:51 -07003227TEST_F(VideoStreamEncoderTest, InitialFrameDropOffWhenEncoderDisabledScaling) {
asaperssonfab67072017-04-04 05:51:49 -07003228 const int kWidth = 640;
3229 const int kHeight = 360;
kthelgasonad9010c2017-02-14 00:46:51 -08003230 fake_encoder_.SetQualityScaling(false);
Niels Möller4db138e2018-04-19 09:04:13 +02003231
3232 VideoEncoderConfig video_encoder_config;
3233 test::FillEncoderConfiguration(kVideoCodecVP8, 1, &video_encoder_config);
3234 // Make format different, to force recreation of encoder.
3235 video_encoder_config.video_format.parameters["foo"] = "foo";
3236 video_stream_encoder_->ConfigureEncoder(std::move(video_encoder_config),
Niels Möllerf1338562018-04-26 09:51:47 +02003237 kMaxPayloadLength);
Erik Språng610c7632019-03-06 15:37:33 +01003238 video_stream_encoder_->OnBitrateUpdated(DataRate::bps(kLowTargetBitrateBps),
Erik Språng4c6ca302019-04-08 15:14:01 +02003239 DataRate::bps(kLowTargetBitrateBps),
3240 0, 0);
asapersson09f05612017-05-15 23:40:18 -07003241
kthelgasonb83797b2017-02-14 11:57:25 -08003242 // Force quality scaler reconfiguration by resetting the source.
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07003243 video_stream_encoder_->SetSource(&video_source_,
3244 webrtc::DegradationPreference::BALANCED);
kthelgasonad9010c2017-02-14 00:46:51 -08003245
asaperssonfab67072017-04-04 05:51:49 -07003246 video_source_.IncomingCapturedFrame(CreateFrame(1, kWidth, kHeight));
kthelgasonad9010c2017-02-14 00:46:51 -08003247 // Frame should not be dropped, even if it's too large.
sprang4847ae62017-06-27 07:06:52 -07003248 WaitForEncodedFrame(1);
kthelgasonad9010c2017-02-14 00:46:51 -08003249
mflodmancc3d4422017-08-03 08:27:51 -07003250 video_stream_encoder_->Stop();
kthelgasonad9010c2017-02-14 00:46:51 -08003251 fake_encoder_.SetQualityScaling(true);
3252}
3253
Kári Tristan Helgason639602a2018-08-02 10:51:40 +02003254TEST_F(VideoStreamEncoderTest, InitialFrameDropActivatesWhenBWEstimateReady) {
3255 webrtc::test::ScopedFieldTrials field_trials(
3256 "WebRTC-InitialFramedrop/Enabled/");
3257 // Reset encoder for field trials to take effect.
3258 ConfigureEncoder(video_encoder_config_.Copy());
3259 const int kTooLowBitrateForFrameSizeBps = 10000;
3260 const int kWidth = 640;
3261 const int kHeight = 360;
3262
Erik Språng4c6ca302019-04-08 15:14:01 +02003263 video_stream_encoder_->OnBitrateUpdated(
3264 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
Kári Tristan Helgason639602a2018-08-02 10:51:40 +02003265 video_source_.IncomingCapturedFrame(CreateFrame(1, kWidth, kHeight));
3266 // Frame should not be dropped.
3267 WaitForEncodedFrame(1);
3268
Erik Språng610c7632019-03-06 15:37:33 +01003269 video_stream_encoder_->OnBitrateUpdated(
Erik Språng4c6ca302019-04-08 15:14:01 +02003270 DataRate::bps(kTooLowBitrateForFrameSizeBps),
3271 DataRate::bps(kTooLowBitrateForFrameSizeBps), 0, 0);
Kári Tristan Helgason639602a2018-08-02 10:51:40 +02003272 video_source_.IncomingCapturedFrame(CreateFrame(2, kWidth, kHeight));
3273 // Expect to drop this frame, the wait should time out.
3274 ExpectDroppedFrame();
3275
3276 // Expect the sink_wants to specify a scaled frame.
3277 EXPECT_LT(video_source_.sink_wants().max_pixel_count, kWidth * kHeight);
3278 video_stream_encoder_->Stop();
3279}
3280
Åsa Persson139f4dc2019-08-02 09:29:58 +02003281TEST_F(VideoStreamEncoderTest, InitialFrameDropActivatesWhenBweDrops) {
3282 webrtc::test::ScopedFieldTrials field_trials(
3283 "WebRTC-Video-QualityScalerSettings/"
3284 "initial_bitrate_interval_ms:1000,initial_bitrate_factor:0.2/");
3285 // Reset encoder for field trials to take effect.
3286 ConfigureEncoder(video_encoder_config_.Copy());
3287 const int kNotTooLowBitrateForFrameSizeBps = kTargetBitrateBps * 0.2;
3288 const int kTooLowBitrateForFrameSizeBps = kTargetBitrateBps * 0.19;
3289 const int kWidth = 640;
3290 const int kHeight = 360;
3291
3292 video_stream_encoder_->OnBitrateUpdated(
3293 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
3294 video_source_.IncomingCapturedFrame(CreateFrame(1, kWidth, kHeight));
3295 // Frame should not be dropped.
3296 WaitForEncodedFrame(1);
3297
3298 video_stream_encoder_->OnBitrateUpdated(
3299 DataRate::bps(kNotTooLowBitrateForFrameSizeBps),
3300 DataRate::bps(kNotTooLowBitrateForFrameSizeBps), 0, 0);
3301 video_source_.IncomingCapturedFrame(CreateFrame(2, kWidth, kHeight));
3302 // Frame should not be dropped.
3303 WaitForEncodedFrame(2);
3304
3305 video_stream_encoder_->OnBitrateUpdated(
3306 DataRate::bps(kTooLowBitrateForFrameSizeBps),
3307 DataRate::bps(kTooLowBitrateForFrameSizeBps), 0, 0);
3308 video_source_.IncomingCapturedFrame(CreateFrame(3, kWidth, kHeight));
3309 // Expect to drop this frame, the wait should time out.
3310 ExpectDroppedFrame();
3311
3312 // Expect the sink_wants to specify a scaled frame.
3313 EXPECT_LT(video_source_.sink_wants().max_pixel_count, kWidth * kHeight);
3314 video_stream_encoder_->Stop();
3315}
3316
mflodmancc3d4422017-08-03 08:27:51 -07003317TEST_F(VideoStreamEncoderTest,
asaperssond0de2952017-04-21 01:47:31 -07003318 ResolutionNotAdaptedForTooSmallFrame_MaintainFramerateMode) {
3319 const int kTooSmallWidth = 10;
3320 const int kTooSmallHeight = 10;
Erik Språng4c6ca302019-04-08 15:14:01 +02003321 video_stream_encoder_->OnBitrateUpdated(
3322 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asaperssond0de2952017-04-21 01:47:31 -07003323
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07003324 // Enable MAINTAIN_FRAMERATE preference, no initial limitation.
asaperssond0de2952017-04-21 01:47:31 -07003325 test::FrameForwarder source;
mflodmancc3d4422017-08-03 08:27:51 -07003326 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07003327 &source, webrtc::DegradationPreference::MAINTAIN_FRAMERATE);
asaperssond0de2952017-04-21 01:47:31 -07003328 VerifyNoLimitation(source.sink_wants());
3329 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
3330
3331 // Trigger adapt down, too small frame, expect no change.
3332 source.IncomingCapturedFrame(CreateFrame(1, kTooSmallWidth, kTooSmallHeight));
sprang4847ae62017-06-27 07:06:52 -07003333 WaitForEncodedFrame(1);
mflodmancc3d4422017-08-03 08:27:51 -07003334 video_stream_encoder_->TriggerCpuOveruse();
Åsa Persson8c1bf952018-09-13 10:42:19 +02003335 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssond0de2952017-04-21 01:47:31 -07003336 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
3337 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
3338
mflodmancc3d4422017-08-03 08:27:51 -07003339 video_stream_encoder_->Stop();
asaperssond0de2952017-04-21 01:47:31 -07003340}
3341
mflodmancc3d4422017-08-03 08:27:51 -07003342TEST_F(VideoStreamEncoderTest,
3343 ResolutionNotAdaptedForTooSmallFrame_BalancedMode) {
asaperssonf7e294d2017-06-13 23:25:22 -07003344 const int kTooSmallWidth = 10;
3345 const int kTooSmallHeight = 10;
3346 const int kFpsLimit = 7;
Erik Språng4c6ca302019-04-08 15:14:01 +02003347 video_stream_encoder_->OnBitrateUpdated(
3348 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asaperssonf7e294d2017-06-13 23:25:22 -07003349
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07003350 // Enable BALANCED preference, no initial limitation.
asaperssonf7e294d2017-06-13 23:25:22 -07003351 test::FrameForwarder source;
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07003352 video_stream_encoder_->SetSource(&source,
3353 webrtc::DegradationPreference::BALANCED);
asaperssonf7e294d2017-06-13 23:25:22 -07003354 VerifyNoLimitation(source.sink_wants());
3355 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
3356 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_framerate);
3357
3358 // Trigger adapt down, expect limited framerate.
3359 source.IncomingCapturedFrame(CreateFrame(1, kTooSmallWidth, kTooSmallHeight));
sprang4847ae62017-06-27 07:06:52 -07003360 WaitForEncodedFrame(1);
mflodmancc3d4422017-08-03 08:27:51 -07003361 video_stream_encoder_->TriggerQualityLow();
asaperssonf7e294d2017-06-13 23:25:22 -07003362 VerifyFpsEqResolutionMax(source.sink_wants(), kFpsLimit);
3363 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
3364 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_framerate);
3365 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3366
3367 // Trigger adapt down, too small frame, expect no change.
3368 source.IncomingCapturedFrame(CreateFrame(2, kTooSmallWidth, kTooSmallHeight));
sprang4847ae62017-06-27 07:06:52 -07003369 WaitForEncodedFrame(2);
mflodmancc3d4422017-08-03 08:27:51 -07003370 video_stream_encoder_->TriggerQualityLow();
asaperssonf7e294d2017-06-13 23:25:22 -07003371 VerifyFpsEqResolutionMax(source.sink_wants(), kFpsLimit);
3372 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
3373 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_framerate);
3374 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3375
mflodmancc3d4422017-08-03 08:27:51 -07003376 video_stream_encoder_->Stop();
asaperssonf7e294d2017-06-13 23:25:22 -07003377}
3378
mflodmancc3d4422017-08-03 08:27:51 -07003379TEST_F(VideoStreamEncoderTest, FailingInitEncodeDoesntCauseCrash) {
asapersson02465b82017-04-10 01:12:52 -07003380 fake_encoder_.ForceInitEncodeFailure(true);
Erik Språng4c6ca302019-04-08 15:14:01 +02003381 video_stream_encoder_->OnBitrateUpdated(
3382 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
Niels Möllerf1338562018-04-26 09:51:47 +02003383 ResetEncoder("VP8", 2, 1, 1, false);
asapersson02465b82017-04-10 01:12:52 -07003384 const int kFrameWidth = 1280;
3385 const int kFrameHeight = 720;
3386 video_source_.IncomingCapturedFrame(
3387 CreateFrame(1, kFrameWidth, kFrameHeight));
sprang4847ae62017-06-27 07:06:52 -07003388 ExpectDroppedFrame();
mflodmancc3d4422017-08-03 08:27:51 -07003389 video_stream_encoder_->Stop();
asapersson02465b82017-04-10 01:12:52 -07003390}
3391
sprangb1ca0732017-02-01 08:38:12 -08003392// TODO(sprang): Extend this with fps throttling and any "balanced" extensions.
mflodmancc3d4422017-08-03 08:27:51 -07003393TEST_F(VideoStreamEncoderTest,
3394 AdaptsResolutionOnOveruse_MaintainFramerateMode) {
Erik Språng4c6ca302019-04-08 15:14:01 +02003395 video_stream_encoder_->OnBitrateUpdated(
3396 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
sprangb1ca0732017-02-01 08:38:12 -08003397
3398 const int kFrameWidth = 1280;
3399 const int kFrameHeight = 720;
3400 // Enabled default VideoAdapter downscaling. First step is 3/4, not 3/5 as
mflodmancc3d4422017-08-03 08:27:51 -07003401 // requested by
3402 // VideoStreamEncoder::VideoSourceProxy::RequestResolutionLowerThan().
sprangb1ca0732017-02-01 08:38:12 -08003403 video_source_.set_adaptation_enabled(true);
3404
3405 video_source_.IncomingCapturedFrame(
Åsa Persson8c1bf952018-09-13 10:42:19 +02003406 CreateFrame(1 * kFrameIntervalMs, kFrameWidth, kFrameHeight));
sprang4847ae62017-06-27 07:06:52 -07003407 WaitForEncodedFrame(kFrameWidth, kFrameHeight);
sprangb1ca0732017-02-01 08:38:12 -08003408
3409 // Trigger CPU overuse, downscale by 3/4.
mflodmancc3d4422017-08-03 08:27:51 -07003410 video_stream_encoder_->TriggerCpuOveruse();
sprangb1ca0732017-02-01 08:38:12 -08003411 video_source_.IncomingCapturedFrame(
Åsa Persson8c1bf952018-09-13 10:42:19 +02003412 CreateFrame(2 * kFrameIntervalMs, kFrameWidth, kFrameHeight));
sprang4847ae62017-06-27 07:06:52 -07003413 WaitForEncodedFrame((kFrameWidth * 3) / 4, (kFrameHeight * 3) / 4);
sprangb1ca0732017-02-01 08:38:12 -08003414
asaperssonfab67072017-04-04 05:51:49 -07003415 // Trigger CPU normal use, return to original resolution.
mflodmancc3d4422017-08-03 08:27:51 -07003416 video_stream_encoder_->TriggerCpuNormalUsage();
sprangb1ca0732017-02-01 08:38:12 -08003417 video_source_.IncomingCapturedFrame(
Åsa Persson8c1bf952018-09-13 10:42:19 +02003418 CreateFrame(3 * kFrameIntervalMs, kFrameWidth, kFrameHeight));
sprang4847ae62017-06-27 07:06:52 -07003419 WaitForEncodedFrame(kFrameWidth, kFrameHeight);
sprangb1ca0732017-02-01 08:38:12 -08003420
mflodmancc3d4422017-08-03 08:27:51 -07003421 video_stream_encoder_->Stop();
sprangb1ca0732017-02-01 08:38:12 -08003422}
sprangfe627f32017-03-29 08:24:59 -07003423
mflodmancc3d4422017-08-03 08:27:51 -07003424TEST_F(VideoStreamEncoderTest,
3425 AdaptsFramerateOnOveruse_MaintainResolutionMode) {
sprangc5d62e22017-04-02 23:53:04 -07003426 const int kFrameWidth = 1280;
3427 const int kFrameHeight = 720;
sprangc5d62e22017-04-02 23:53:04 -07003428
Erik Språng4c6ca302019-04-08 15:14:01 +02003429 video_stream_encoder_->OnBitrateUpdated(
3430 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
mflodmancc3d4422017-08-03 08:27:51 -07003431 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07003432 &video_source_, webrtc::DegradationPreference::MAINTAIN_RESOLUTION);
sprangc5d62e22017-04-02 23:53:04 -07003433 video_source_.set_adaptation_enabled(true);
3434
sprang4847ae62017-06-27 07:06:52 -07003435 int64_t timestamp_ms = fake_clock_.TimeNanos() / rtc::kNumNanosecsPerMillisec;
sprangc5d62e22017-04-02 23:53:04 -07003436
3437 video_source_.IncomingCapturedFrame(
3438 CreateFrame(timestamp_ms, kFrameWidth, kFrameHeight));
sprang4847ae62017-06-27 07:06:52 -07003439 WaitForEncodedFrame(timestamp_ms);
sprangc5d62e22017-04-02 23:53:04 -07003440
3441 // Try to trigger overuse. No fps estimate available => no effect.
mflodmancc3d4422017-08-03 08:27:51 -07003442 video_stream_encoder_->TriggerCpuOveruse();
sprangc5d62e22017-04-02 23:53:04 -07003443
3444 // Insert frames for one second to get a stable estimate.
sprang4847ae62017-06-27 07:06:52 -07003445 for (int i = 0; i < max_framerate_; ++i) {
sprangc5d62e22017-04-02 23:53:04 -07003446 timestamp_ms += kFrameIntervalMs;
sprangc5d62e22017-04-02 23:53:04 -07003447 video_source_.IncomingCapturedFrame(
3448 CreateFrame(timestamp_ms, kFrameWidth, kFrameHeight));
sprang4847ae62017-06-27 07:06:52 -07003449 WaitForEncodedFrame(timestamp_ms);
sprangc5d62e22017-04-02 23:53:04 -07003450 }
3451
3452 // Trigger CPU overuse, reduce framerate by 2/3.
mflodmancc3d4422017-08-03 08:27:51 -07003453 video_stream_encoder_->TriggerCpuOveruse();
sprangc5d62e22017-04-02 23:53:04 -07003454 int num_frames_dropped = 0;
sprang4847ae62017-06-27 07:06:52 -07003455 for (int i = 0; i < max_framerate_; ++i) {
sprangc5d62e22017-04-02 23:53:04 -07003456 timestamp_ms += kFrameIntervalMs;
sprangc5d62e22017-04-02 23:53:04 -07003457 video_source_.IncomingCapturedFrame(
3458 CreateFrame(timestamp_ms, kFrameWidth, kFrameHeight));
sprang4847ae62017-06-27 07:06:52 -07003459 if (!WaitForFrame(kFrameTimeoutMs)) {
sprangc5d62e22017-04-02 23:53:04 -07003460 ++num_frames_dropped;
3461 } else {
Åsa Perssonc74d8da2017-12-04 14:13:56 +01003462 sink_.CheckLastFrameSizeMatches(kFrameWidth, kFrameHeight);
sprangc5d62e22017-04-02 23:53:04 -07003463 }
3464 }
3465
sprang4847ae62017-06-27 07:06:52 -07003466 // Add some slack to account for frames dropped by the frame dropper.
3467 const int kErrorMargin = 1;
3468 EXPECT_NEAR(num_frames_dropped, max_framerate_ - (max_framerate_ * 2 / 3),
sprangc5d62e22017-04-02 23:53:04 -07003469 kErrorMargin);
3470
3471 // Trigger CPU overuse, reduce framerate by 2/3 again.
mflodmancc3d4422017-08-03 08:27:51 -07003472 video_stream_encoder_->TriggerCpuOveruse();
sprangc5d62e22017-04-02 23:53:04 -07003473 num_frames_dropped = 0;
Åsa Persson8c1bf952018-09-13 10:42:19 +02003474 for (int i = 0; i <= max_framerate_; ++i) {
sprangc5d62e22017-04-02 23:53:04 -07003475 timestamp_ms += kFrameIntervalMs;
sprangc5d62e22017-04-02 23:53:04 -07003476 video_source_.IncomingCapturedFrame(
3477 CreateFrame(timestamp_ms, kFrameWidth, kFrameHeight));
sprang4847ae62017-06-27 07:06:52 -07003478 if (!WaitForFrame(kFrameTimeoutMs)) {
sprangc5d62e22017-04-02 23:53:04 -07003479 ++num_frames_dropped;
3480 } else {
Åsa Perssonc74d8da2017-12-04 14:13:56 +01003481 sink_.CheckLastFrameSizeMatches(kFrameWidth, kFrameHeight);
sprangc5d62e22017-04-02 23:53:04 -07003482 }
3483 }
sprang4847ae62017-06-27 07:06:52 -07003484 EXPECT_NEAR(num_frames_dropped, max_framerate_ - (max_framerate_ * 4 / 9),
sprangc5d62e22017-04-02 23:53:04 -07003485 kErrorMargin);
3486
3487 // Go back up one step.
mflodmancc3d4422017-08-03 08:27:51 -07003488 video_stream_encoder_->TriggerCpuNormalUsage();
sprangc5d62e22017-04-02 23:53:04 -07003489 num_frames_dropped = 0;
sprang4847ae62017-06-27 07:06:52 -07003490 for (int i = 0; i < max_framerate_; ++i) {
sprangc5d62e22017-04-02 23:53:04 -07003491 timestamp_ms += kFrameIntervalMs;
sprangc5d62e22017-04-02 23:53:04 -07003492 video_source_.IncomingCapturedFrame(
3493 CreateFrame(timestamp_ms, kFrameWidth, kFrameHeight));
sprang4847ae62017-06-27 07:06:52 -07003494 if (!WaitForFrame(kFrameTimeoutMs)) {
sprangc5d62e22017-04-02 23:53:04 -07003495 ++num_frames_dropped;
3496 } else {
Åsa Perssonc74d8da2017-12-04 14:13:56 +01003497 sink_.CheckLastFrameSizeMatches(kFrameWidth, kFrameHeight);
sprangc5d62e22017-04-02 23:53:04 -07003498 }
3499 }
sprang4847ae62017-06-27 07:06:52 -07003500 EXPECT_NEAR(num_frames_dropped, max_framerate_ - (max_framerate_ * 2 / 3),
sprangc5d62e22017-04-02 23:53:04 -07003501 kErrorMargin);
3502
3503 // Go back up to original mode.
mflodmancc3d4422017-08-03 08:27:51 -07003504 video_stream_encoder_->TriggerCpuNormalUsage();
sprangc5d62e22017-04-02 23:53:04 -07003505 num_frames_dropped = 0;
sprang4847ae62017-06-27 07:06:52 -07003506 for (int i = 0; i < max_framerate_; ++i) {
sprangc5d62e22017-04-02 23:53:04 -07003507 timestamp_ms += kFrameIntervalMs;
sprangc5d62e22017-04-02 23:53:04 -07003508 video_source_.IncomingCapturedFrame(
3509 CreateFrame(timestamp_ms, kFrameWidth, kFrameHeight));
sprang4847ae62017-06-27 07:06:52 -07003510 if (!WaitForFrame(kFrameTimeoutMs)) {
sprangc5d62e22017-04-02 23:53:04 -07003511 ++num_frames_dropped;
3512 } else {
Åsa Perssonc74d8da2017-12-04 14:13:56 +01003513 sink_.CheckLastFrameSizeMatches(kFrameWidth, kFrameHeight);
sprangc5d62e22017-04-02 23:53:04 -07003514 }
3515 }
3516 EXPECT_NEAR(num_frames_dropped, 0, kErrorMargin);
3517
mflodmancc3d4422017-08-03 08:27:51 -07003518 video_stream_encoder_->Stop();
sprangc5d62e22017-04-02 23:53:04 -07003519}
3520
mflodmancc3d4422017-08-03 08:27:51 -07003521TEST_F(VideoStreamEncoderTest, DoesntAdaptDownPastMinFramerate) {
sprangc5d62e22017-04-02 23:53:04 -07003522 const int kFramerateFps = 5;
3523 const int kFrameIntervalMs = rtc::kNumMillisecsPerSec / kFramerateFps;
sprangc5d62e22017-04-02 23:53:04 -07003524 const int kFrameWidth = 1280;
3525 const int kFrameHeight = 720;
3526
sprang4847ae62017-06-27 07:06:52 -07003527 // Reconfigure encoder with two temporal layers and screensharing, which will
3528 // disable frame dropping and make testing easier.
Niels Möllerf1338562018-04-26 09:51:47 +02003529 ResetEncoder("VP8", 1, 2, 1, true);
sprang4847ae62017-06-27 07:06:52 -07003530
Erik Språng4c6ca302019-04-08 15:14:01 +02003531 video_stream_encoder_->OnBitrateUpdated(
3532 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
mflodmancc3d4422017-08-03 08:27:51 -07003533 video_stream_encoder_->SetSource(
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07003534 &video_source_, webrtc::DegradationPreference::MAINTAIN_RESOLUTION);
sprangc5d62e22017-04-02 23:53:04 -07003535 video_source_.set_adaptation_enabled(true);
3536
sprang4847ae62017-06-27 07:06:52 -07003537 int64_t timestamp_ms = fake_clock_.TimeNanos() / rtc::kNumNanosecsPerMillisec;
sprangc5d62e22017-04-02 23:53:04 -07003538
3539 // Trigger overuse as much as we can.
Jonathan Yubc771b72017-12-08 17:04:29 -08003540 rtc::VideoSinkWants last_wants;
3541 do {
3542 last_wants = video_source_.sink_wants();
3543
sprangc5d62e22017-04-02 23:53:04 -07003544 // Insert frames to get a new fps estimate...
3545 for (int j = 0; j < kFramerateFps; ++j) {
3546 video_source_.IncomingCapturedFrame(
3547 CreateFrame(timestamp_ms, kFrameWidth, kFrameHeight));
Jonathan Yubc771b72017-12-08 17:04:29 -08003548 if (video_source_.last_sent_width()) {
3549 sink_.WaitForEncodedFrame(timestamp_ms);
3550 }
sprangc5d62e22017-04-02 23:53:04 -07003551 timestamp_ms += kFrameIntervalMs;
Sebastian Jansson40889f32019-04-17 12:11:20 +02003552 fake_clock_.AdvanceTime(TimeDelta::ms(kFrameIntervalMs));
sprangc5d62e22017-04-02 23:53:04 -07003553 }
3554 // ...and then try to adapt again.
mflodmancc3d4422017-08-03 08:27:51 -07003555 video_stream_encoder_->TriggerCpuOveruse();
Jonathan Yubc771b72017-12-08 17:04:29 -08003556 } while (video_source_.sink_wants().max_framerate_fps <
3557 last_wants.max_framerate_fps);
sprangc5d62e22017-04-02 23:53:04 -07003558
Jonathan Yubc771b72017-12-08 17:04:29 -08003559 VerifyFpsEqResolutionMax(video_source_.sink_wants(), kMinFramerateFps);
asaperssonf7e294d2017-06-13 23:25:22 -07003560
mflodmancc3d4422017-08-03 08:27:51 -07003561 video_stream_encoder_->Stop();
sprangc5d62e22017-04-02 23:53:04 -07003562}
asaperssonf7e294d2017-06-13 23:25:22 -07003563
mflodmancc3d4422017-08-03 08:27:51 -07003564TEST_F(VideoStreamEncoderTest,
3565 AdaptsResolutionAndFramerateForLowQuality_BalancedMode) {
asaperssonf7e294d2017-06-13 23:25:22 -07003566 const int kWidth = 1280;
3567 const int kHeight = 720;
3568 const int64_t kFrameIntervalMs = 150;
3569 int64_t timestamp_ms = kFrameIntervalMs;
Erik Språng4c6ca302019-04-08 15:14:01 +02003570 video_stream_encoder_->OnBitrateUpdated(
3571 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asaperssonf7e294d2017-06-13 23:25:22 -07003572
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07003573 // Enable BALANCED preference, no initial limitation.
asaperssonf7e294d2017-06-13 23:25:22 -07003574 AdaptingFrameForwarder source;
3575 source.set_adaptation_enabled(true);
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07003576 video_stream_encoder_->SetSource(&source,
3577 webrtc::DegradationPreference::BALANCED);
asaperssonf7e294d2017-06-13 23:25:22 -07003578 timestamp_ms += kFrameIntervalMs;
3579 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003580 WaitForEncodedFrame(kWidth, kHeight);
Åsa Persson8c1bf952018-09-13 10:42:19 +02003581 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssonf7e294d2017-06-13 23:25:22 -07003582 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
3583 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_framerate);
3584 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3585
3586 // Trigger adapt down, expect scaled down resolution (960x540@30fps).
mflodmancc3d4422017-08-03 08:27:51 -07003587 video_stream_encoder_->TriggerQualityLow();
asaperssonf7e294d2017-06-13 23:25:22 -07003588 timestamp_ms += kFrameIntervalMs;
3589 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003590 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003591 VerifyFpsMaxResolutionLt(source.sink_wants(), kWidth * kHeight);
3592 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
3593 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_framerate);
3594 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3595
3596 // Trigger adapt down, expect scaled down resolution (640x360@30fps).
mflodmancc3d4422017-08-03 08:27:51 -07003597 video_stream_encoder_->TriggerQualityLow();
asaperssonf7e294d2017-06-13 23:25:22 -07003598 timestamp_ms += kFrameIntervalMs;
3599 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003600 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003601 VerifyFpsMaxResolutionLt(source.sink_wants(), source.last_wants());
3602 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
3603 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_framerate);
3604 EXPECT_EQ(2, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3605
3606 // Trigger adapt down, expect reduced fps (640x360@15fps).
mflodmancc3d4422017-08-03 08:27:51 -07003607 video_stream_encoder_->TriggerQualityLow();
asaperssonf7e294d2017-06-13 23:25:22 -07003608 timestamp_ms += kFrameIntervalMs;
3609 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003610 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003611 VerifyFpsLtResolutionEq(source.sink_wants(), source.last_wants());
3612 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
3613 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_framerate);
3614 EXPECT_EQ(3, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3615
3616 // Trigger adapt down, expect scaled down resolution (480x270@15fps).
mflodmancc3d4422017-08-03 08:27:51 -07003617 video_stream_encoder_->TriggerQualityLow();
asaperssonf7e294d2017-06-13 23:25:22 -07003618 timestamp_ms += kFrameIntervalMs;
3619 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003620 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003621 VerifyFpsEqResolutionLt(source.sink_wants(), source.last_wants());
3622 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
3623 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_framerate);
3624 EXPECT_EQ(4, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3625
3626 // Restrict bitrate, trigger adapt down, expect reduced fps (480x270@10fps).
mflodmancc3d4422017-08-03 08:27:51 -07003627 video_stream_encoder_->TriggerQualityLow();
asaperssonf7e294d2017-06-13 23:25:22 -07003628 timestamp_ms += kFrameIntervalMs;
3629 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003630 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003631 VerifyFpsLtResolutionEq(source.sink_wants(), source.last_wants());
3632 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
3633 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_framerate);
3634 EXPECT_EQ(5, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3635
3636 // Trigger adapt down, expect scaled down resolution (320x180@10fps).
mflodmancc3d4422017-08-03 08:27:51 -07003637 video_stream_encoder_->TriggerQualityLow();
asaperssonf7e294d2017-06-13 23:25:22 -07003638 timestamp_ms += kFrameIntervalMs;
3639 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003640 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003641 VerifyFpsEqResolutionLt(source.sink_wants(), source.last_wants());
3642 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
3643 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_framerate);
3644 EXPECT_EQ(6, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3645
3646 // Trigger adapt down, expect reduced fps (320x180@7fps).
mflodmancc3d4422017-08-03 08:27:51 -07003647 video_stream_encoder_->TriggerQualityLow();
asaperssonf7e294d2017-06-13 23:25:22 -07003648 timestamp_ms += kFrameIntervalMs;
3649 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003650 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003651 VerifyFpsLtResolutionEq(source.sink_wants(), source.last_wants());
3652 rtc::VideoSinkWants last_wants = source.sink_wants();
3653 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
3654 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_framerate);
3655 EXPECT_EQ(7, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3656
3657 // Trigger adapt down, min resolution reached, expect no change.
mflodmancc3d4422017-08-03 08:27:51 -07003658 video_stream_encoder_->TriggerQualityLow();
asaperssonf7e294d2017-06-13 23:25:22 -07003659 timestamp_ms += kFrameIntervalMs;
3660 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003661 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003662 VerifyFpsEqResolutionEq(source.sink_wants(), last_wants);
3663 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
3664 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_framerate);
3665 EXPECT_EQ(7, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3666
3667 // Trigger adapt down, expect expect increased fps (320x180@10fps).
mflodmancc3d4422017-08-03 08:27:51 -07003668 video_stream_encoder_->TriggerQualityHigh();
asaperssonf7e294d2017-06-13 23:25:22 -07003669 timestamp_ms += kFrameIntervalMs;
3670 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003671 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003672 VerifyFpsGtResolutionEq(source.sink_wants(), source.last_wants());
3673 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
3674 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_framerate);
3675 EXPECT_EQ(8, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3676
3677 // Trigger adapt up, expect upscaled resolution (480x270@10fps).
mflodmancc3d4422017-08-03 08:27:51 -07003678 video_stream_encoder_->TriggerQualityHigh();
asaperssonf7e294d2017-06-13 23:25:22 -07003679 timestamp_ms += kFrameIntervalMs;
3680 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003681 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003682 VerifyFpsEqResolutionGt(source.sink_wants(), source.last_wants());
3683 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
3684 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_framerate);
3685 EXPECT_EQ(9, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3686
3687 // Increase bitrate, trigger adapt up, expect increased fps (480x270@15fps).
mflodmancc3d4422017-08-03 08:27:51 -07003688 video_stream_encoder_->TriggerQualityHigh();
asaperssonf7e294d2017-06-13 23:25:22 -07003689 timestamp_ms += kFrameIntervalMs;
3690 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003691 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003692 VerifyFpsGtResolutionEq(source.sink_wants(), source.last_wants());
3693 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
3694 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_framerate);
3695 EXPECT_EQ(10, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3696
3697 // Trigger adapt up, expect upscaled resolution (640x360@15fps).
mflodmancc3d4422017-08-03 08:27:51 -07003698 video_stream_encoder_->TriggerQualityHigh();
asaperssonf7e294d2017-06-13 23:25:22 -07003699 timestamp_ms += kFrameIntervalMs;
3700 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003701 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003702 VerifyFpsEqResolutionGt(source.sink_wants(), source.last_wants());
3703 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
3704 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_framerate);
3705 EXPECT_EQ(11, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3706
3707 // Trigger adapt up, expect increased fps (640x360@30fps).
mflodmancc3d4422017-08-03 08:27:51 -07003708 video_stream_encoder_->TriggerQualityHigh();
asaperssonf7e294d2017-06-13 23:25:22 -07003709 timestamp_ms += kFrameIntervalMs;
3710 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003711 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003712 VerifyFpsMaxResolutionEq(source.sink_wants(), source.last_wants());
3713 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
3714 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_framerate);
3715 EXPECT_EQ(12, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3716
3717 // Trigger adapt up, expect upscaled resolution (960x540@30fps).
mflodmancc3d4422017-08-03 08:27:51 -07003718 video_stream_encoder_->TriggerQualityHigh();
asaperssonf7e294d2017-06-13 23:25:22 -07003719 timestamp_ms += kFrameIntervalMs;
3720 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003721 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003722 VerifyFpsMaxResolutionGt(source.sink_wants(), source.last_wants());
3723 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
3724 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_framerate);
3725 EXPECT_EQ(13, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3726
3727 // Trigger adapt up, expect no restriction (1280x720fps@30fps).
mflodmancc3d4422017-08-03 08:27:51 -07003728 video_stream_encoder_->TriggerQualityHigh();
asaperssonf7e294d2017-06-13 23:25:22 -07003729 timestamp_ms += kFrameIntervalMs;
3730 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003731 WaitForEncodedFrame(kWidth, kHeight);
asaperssonf7e294d2017-06-13 23:25:22 -07003732 VerifyFpsMaxResolutionGt(source.sink_wants(), source.last_wants());
Åsa Persson8c1bf952018-09-13 10:42:19 +02003733 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssonf7e294d2017-06-13 23:25:22 -07003734 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
3735 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_framerate);
3736 EXPECT_EQ(14, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3737
3738 // Trigger adapt up, expect no change.
mflodmancc3d4422017-08-03 08:27:51 -07003739 video_stream_encoder_->TriggerQualityHigh();
Åsa Persson8c1bf952018-09-13 10:42:19 +02003740 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssonf7e294d2017-06-13 23:25:22 -07003741 EXPECT_EQ(14, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3742
mflodmancc3d4422017-08-03 08:27:51 -07003743 video_stream_encoder_->Stop();
asaperssonf7e294d2017-06-13 23:25:22 -07003744}
3745
mflodmancc3d4422017-08-03 08:27:51 -07003746TEST_F(VideoStreamEncoderTest, AdaptWithTwoReasonsAndDifferentOrder_Framerate) {
asaperssonf7e294d2017-06-13 23:25:22 -07003747 const int kWidth = 1280;
3748 const int kHeight = 720;
3749 const int64_t kFrameIntervalMs = 150;
3750 int64_t timestamp_ms = kFrameIntervalMs;
Erik Språng4c6ca302019-04-08 15:14:01 +02003751 video_stream_encoder_->OnBitrateUpdated(
3752 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asaperssonf7e294d2017-06-13 23:25:22 -07003753
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07003754 // Enable BALANCED preference, no initial limitation.
asaperssonf7e294d2017-06-13 23:25:22 -07003755 AdaptingFrameForwarder source;
3756 source.set_adaptation_enabled(true);
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07003757 video_stream_encoder_->SetSource(&source,
3758 webrtc::DegradationPreference::BALANCED);
asaperssonf7e294d2017-06-13 23:25:22 -07003759 timestamp_ms += kFrameIntervalMs;
3760 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003761 WaitForEncodedFrame(kWidth, kHeight);
Åsa Persson8c1bf952018-09-13 10:42:19 +02003762 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssonf7e294d2017-06-13 23:25:22 -07003763 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
3764 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_framerate);
3765 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
3766 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_framerate);
3767 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
3768 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3769
3770 // Trigger cpu adapt down, expect scaled down resolution (960x540@30fps).
mflodmancc3d4422017-08-03 08:27:51 -07003771 video_stream_encoder_->TriggerCpuOveruse();
asaperssonf7e294d2017-06-13 23:25:22 -07003772 timestamp_ms += kFrameIntervalMs;
3773 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003774 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003775 VerifyFpsMaxResolutionLt(source.sink_wants(), kWidth * kHeight);
3776 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
3777 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_framerate);
3778 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_resolution);
3779 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_framerate);
3780 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
3781 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3782
3783 // Trigger cpu adapt down, expect scaled down resolution (640x360@30fps).
mflodmancc3d4422017-08-03 08:27:51 -07003784 video_stream_encoder_->TriggerCpuOveruse();
asaperssonf7e294d2017-06-13 23:25:22 -07003785 timestamp_ms += kFrameIntervalMs;
3786 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003787 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003788 VerifyFpsMaxResolutionLt(source.sink_wants(), source.last_wants());
3789 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
3790 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_framerate);
3791 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_resolution);
3792 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_framerate);
3793 EXPECT_EQ(2, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
3794 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3795
3796 // Trigger quality adapt down, expect reduced fps (640x360@15fps).
mflodmancc3d4422017-08-03 08:27:51 -07003797 video_stream_encoder_->TriggerQualityLow();
asaperssonf7e294d2017-06-13 23:25:22 -07003798 timestamp_ms += kFrameIntervalMs;
3799 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003800 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003801 VerifyFpsLtResolutionEq(source.sink_wants(), source.last_wants());
3802 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
3803 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_framerate);
3804 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_resolution);
3805 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_framerate);
3806 EXPECT_EQ(2, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
3807 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3808
3809 // Trigger cpu adapt up, expect increased fps (640x360@30fps).
mflodmancc3d4422017-08-03 08:27:51 -07003810 video_stream_encoder_->TriggerCpuNormalUsage();
asaperssonf7e294d2017-06-13 23:25:22 -07003811 timestamp_ms += kFrameIntervalMs;
3812 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003813 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003814 VerifyFpsMaxResolutionEq(source.sink_wants(), source.last_wants());
3815 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
3816 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_framerate);
3817 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_resolution);
3818 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_framerate);
3819 EXPECT_EQ(3, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
3820 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3821
3822 // Trigger quality adapt up, expect upscaled resolution (960x540@30fps).
mflodmancc3d4422017-08-03 08:27:51 -07003823 video_stream_encoder_->TriggerQualityHigh();
asaperssonf7e294d2017-06-13 23:25:22 -07003824 timestamp_ms += kFrameIntervalMs;
3825 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003826 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003827 VerifyFpsMaxResolutionGt(source.sink_wants(), source.last_wants());
3828 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
3829 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_framerate);
3830 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_resolution);
3831 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_framerate);
3832 EXPECT_EQ(3, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
3833 EXPECT_EQ(2, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3834
3835 // Trigger cpu adapt up, expect no restriction (1280x720fps@30fps).
mflodmancc3d4422017-08-03 08:27:51 -07003836 video_stream_encoder_->TriggerCpuNormalUsage();
asaperssonf7e294d2017-06-13 23:25:22 -07003837 timestamp_ms += kFrameIntervalMs;
3838 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003839 WaitForEncodedFrame(kWidth, kHeight);
asaperssonf7e294d2017-06-13 23:25:22 -07003840 VerifyFpsMaxResolutionGt(source.sink_wants(), source.last_wants());
Åsa Persson8c1bf952018-09-13 10:42:19 +02003841 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssonf7e294d2017-06-13 23:25:22 -07003842 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
3843 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_framerate);
3844 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
3845 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_framerate);
3846 EXPECT_EQ(4, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
3847 EXPECT_EQ(2, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3848
3849 // Trigger adapt up, expect no change.
mflodmancc3d4422017-08-03 08:27:51 -07003850 video_stream_encoder_->TriggerQualityHigh();
Åsa Persson8c1bf952018-09-13 10:42:19 +02003851 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssonf7e294d2017-06-13 23:25:22 -07003852 EXPECT_EQ(4, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
3853 EXPECT_EQ(2, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3854
mflodmancc3d4422017-08-03 08:27:51 -07003855 video_stream_encoder_->Stop();
asaperssonf7e294d2017-06-13 23:25:22 -07003856}
3857
mflodmancc3d4422017-08-03 08:27:51 -07003858TEST_F(VideoStreamEncoderTest,
3859 AdaptWithTwoReasonsAndDifferentOrder_Resolution) {
asaperssonf7e294d2017-06-13 23:25:22 -07003860 const int kWidth = 640;
3861 const int kHeight = 360;
3862 const int kFpsLimit = 15;
3863 const int64_t kFrameIntervalMs = 150;
3864 int64_t timestamp_ms = kFrameIntervalMs;
Erik Språng4c6ca302019-04-08 15:14:01 +02003865 video_stream_encoder_->OnBitrateUpdated(
3866 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
asaperssonf7e294d2017-06-13 23:25:22 -07003867
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07003868 // Enable BALANCED preference, no initial limitation.
asaperssonf7e294d2017-06-13 23:25:22 -07003869 AdaptingFrameForwarder source;
3870 source.set_adaptation_enabled(true);
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07003871 video_stream_encoder_->SetSource(&source,
3872 webrtc::DegradationPreference::BALANCED);
asaperssonf7e294d2017-06-13 23:25:22 -07003873 timestamp_ms += kFrameIntervalMs;
3874 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003875 WaitForEncodedFrame(kWidth, kHeight);
Åsa Persson8c1bf952018-09-13 10:42:19 +02003876 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssonf7e294d2017-06-13 23:25:22 -07003877 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
3878 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_framerate);
3879 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
3880 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_framerate);
3881 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
3882 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3883
3884 // Trigger cpu adapt down, expect scaled down framerate (640x360@15fps).
mflodmancc3d4422017-08-03 08:27:51 -07003885 video_stream_encoder_->TriggerCpuOveruse();
asaperssonf7e294d2017-06-13 23:25:22 -07003886 timestamp_ms += kFrameIntervalMs;
3887 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003888 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003889 VerifyFpsEqResolutionMax(source.sink_wants(), kFpsLimit);
3890 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
3891 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_framerate);
3892 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
3893 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_framerate);
3894 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
3895 EXPECT_EQ(0, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3896
3897 // Trigger quality adapt down, expect scaled down resolution (480x270@15fps).
mflodmancc3d4422017-08-03 08:27:51 -07003898 video_stream_encoder_->TriggerQualityLow();
asaperssonf7e294d2017-06-13 23:25:22 -07003899 timestamp_ms += kFrameIntervalMs;
3900 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003901 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003902 VerifyFpsEqResolutionLt(source.sink_wants(), source.last_wants());
3903 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_resolution);
3904 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_framerate);
3905 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
3906 EXPECT_TRUE(stats_proxy_->GetStats().cpu_limited_framerate);
3907 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
3908 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3909
3910 // Trigger cpu adapt up, expect upscaled resolution (640x360@15fps).
mflodmancc3d4422017-08-03 08:27:51 -07003911 video_stream_encoder_->TriggerCpuNormalUsage();
asaperssonf7e294d2017-06-13 23:25:22 -07003912 timestamp_ms += kFrameIntervalMs;
3913 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003914 WaitForEncodedFrame(timestamp_ms);
asaperssonf7e294d2017-06-13 23:25:22 -07003915 VerifyFpsEqResolutionGt(source.sink_wants(), source.last_wants());
3916 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
3917 EXPECT_TRUE(stats_proxy_->GetStats().bw_limited_framerate);
3918 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
3919 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_framerate);
3920 EXPECT_EQ(2, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
3921 EXPECT_EQ(1, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3922
3923 // Trigger quality adapt up, expect increased fps (640x360@30fps).
mflodmancc3d4422017-08-03 08:27:51 -07003924 video_stream_encoder_->TriggerQualityHigh();
asaperssonf7e294d2017-06-13 23:25:22 -07003925 timestamp_ms += kFrameIntervalMs;
3926 source.IncomingCapturedFrame(CreateFrame(timestamp_ms, kWidth, kHeight));
sprang4847ae62017-06-27 07:06:52 -07003927 WaitForEncodedFrame(timestamp_ms);
Åsa Persson8c1bf952018-09-13 10:42:19 +02003928 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssonf7e294d2017-06-13 23:25:22 -07003929 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_resolution);
3930 EXPECT_FALSE(stats_proxy_->GetStats().bw_limited_framerate);
3931 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_resolution);
3932 EXPECT_FALSE(stats_proxy_->GetStats().cpu_limited_framerate);
3933 EXPECT_EQ(2, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
3934 EXPECT_EQ(2, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3935
3936 // Trigger adapt up, expect no change.
mflodmancc3d4422017-08-03 08:27:51 -07003937 video_stream_encoder_->TriggerQualityHigh();
Åsa Persson8c1bf952018-09-13 10:42:19 +02003938 VerifyFpsMaxResolutionMax(source.sink_wants());
asaperssonf7e294d2017-06-13 23:25:22 -07003939 EXPECT_EQ(2, stats_proxy_->GetStats().number_of_cpu_adapt_changes);
3940 EXPECT_EQ(2, stats_proxy_->GetStats().number_of_quality_adapt_changes);
3941
mflodmancc3d4422017-08-03 08:27:51 -07003942 video_stream_encoder_->Stop();
asaperssonf7e294d2017-06-13 23:25:22 -07003943}
3944
mflodmancc3d4422017-08-03 08:27:51 -07003945TEST_F(VideoStreamEncoderTest, AcceptsFullHdAdaptedDownSimulcastFrames) {
ilnik6b826ef2017-06-16 06:53:48 -07003946 const int kFrameWidth = 1920;
3947 const int kFrameHeight = 1080;
3948 // 3/4 of 1920.
3949 const int kAdaptedFrameWidth = 1440;
3950 // 3/4 of 1080 rounded down to multiple of 4.
3951 const int kAdaptedFrameHeight = 808;
3952 const int kFramerate = 24;
3953
Erik Språng4c6ca302019-04-08 15:14:01 +02003954 video_stream_encoder_->OnBitrateUpdated(
3955 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
ilnik6b826ef2017-06-16 06:53:48 -07003956 // Trigger reconfigure encoder (without resetting the entire instance).
3957 VideoEncoderConfig video_encoder_config;
Niels Möller259a4972018-04-05 15:36:51 +02003958 video_encoder_config.codec_type = kVideoCodecVP8;
ilnik6b826ef2017-06-16 06:53:48 -07003959 video_encoder_config.max_bitrate_bps = kTargetBitrateBps;
3960 video_encoder_config.number_of_streams = 1;
3961 video_encoder_config.video_stream_factory =
3962 new rtc::RefCountedObject<CroppingVideoStreamFactory>(1, kFramerate);
mflodmancc3d4422017-08-03 08:27:51 -07003963 video_stream_encoder_->ConfigureEncoder(std::move(video_encoder_config),
Niels Möllerf1338562018-04-26 09:51:47 +02003964 kMaxPayloadLength);
mflodmancc3d4422017-08-03 08:27:51 -07003965 video_stream_encoder_->WaitUntilTaskQueueIsIdle();
ilnik6b826ef2017-06-16 06:53:48 -07003966
3967 video_source_.set_adaptation_enabled(true);
3968
3969 video_source_.IncomingCapturedFrame(
3970 CreateFrame(1, kFrameWidth, kFrameHeight));
sprang4847ae62017-06-27 07:06:52 -07003971 WaitForEncodedFrame(kFrameWidth, kFrameHeight);
ilnik6b826ef2017-06-16 06:53:48 -07003972
3973 // Trigger CPU overuse, downscale by 3/4.
mflodmancc3d4422017-08-03 08:27:51 -07003974 video_stream_encoder_->TriggerCpuOveruse();
ilnik6b826ef2017-06-16 06:53:48 -07003975 video_source_.IncomingCapturedFrame(
3976 CreateFrame(2, kFrameWidth, kFrameHeight));
sprang4847ae62017-06-27 07:06:52 -07003977 WaitForEncodedFrame(kAdaptedFrameWidth, kAdaptedFrameHeight);
ilnik6b826ef2017-06-16 06:53:48 -07003978
mflodmancc3d4422017-08-03 08:27:51 -07003979 video_stream_encoder_->Stop();
ilnik6b826ef2017-06-16 06:53:48 -07003980}
3981
mflodmancc3d4422017-08-03 08:27:51 -07003982TEST_F(VideoStreamEncoderTest, PeriodicallyUpdatesChannelParameters) {
sprang4847ae62017-06-27 07:06:52 -07003983 const int kFrameWidth = 1280;
3984 const int kFrameHeight = 720;
3985 const int kLowFps = 2;
3986 const int kHighFps = 30;
3987
Erik Språng4c6ca302019-04-08 15:14:01 +02003988 video_stream_encoder_->OnBitrateUpdated(
3989 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
sprang4847ae62017-06-27 07:06:52 -07003990
3991 int64_t timestamp_ms = fake_clock_.TimeNanos() / rtc::kNumNanosecsPerMillisec;
3992 max_framerate_ = kLowFps;
3993
3994 // Insert 2 seconds of 2fps video.
3995 for (int i = 0; i < kLowFps * 2; ++i) {
3996 video_source_.IncomingCapturedFrame(
3997 CreateFrame(timestamp_ms, kFrameWidth, kFrameHeight));
3998 WaitForEncodedFrame(timestamp_ms);
3999 timestamp_ms += 1000 / kLowFps;
4000 }
4001
4002 // Make sure encoder is updated with new target.
Erik Språng4c6ca302019-04-08 15:14:01 +02004003 video_stream_encoder_->OnBitrateUpdated(
4004 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
sprang4847ae62017-06-27 07:06:52 -07004005 video_source_.IncomingCapturedFrame(
4006 CreateFrame(timestamp_ms, kFrameWidth, kFrameHeight));
4007 WaitForEncodedFrame(timestamp_ms);
4008 timestamp_ms += 1000 / kLowFps;
4009
4010 EXPECT_EQ(kLowFps, fake_encoder_.GetConfiguredInputFramerate());
4011
4012 // Insert 30fps frames for just a little more than the forced update period.
4013 const int kVcmTimerIntervalFrames =
4014 (vcm::VCMProcessTimer::kDefaultProcessIntervalMs * kHighFps) / 1000;
4015 const int kFrameIntervalMs = 1000 / kHighFps;
4016 max_framerate_ = kHighFps;
4017 for (int i = 0; i < kVcmTimerIntervalFrames + 2; ++i) {
4018 video_source_.IncomingCapturedFrame(
4019 CreateFrame(timestamp_ms, kFrameWidth, kFrameHeight));
4020 // Wait for encoded frame, but skip ahead if it doesn't arrive as it might
4021 // be dropped if the encoder hans't been updated with the new higher target
4022 // framerate yet, causing it to overshoot the target bitrate and then
4023 // suffering the wrath of the media optimizer.
4024 TimedWaitForEncodedFrame(timestamp_ms, 2 * kFrameIntervalMs);
4025 timestamp_ms += kFrameIntervalMs;
4026 }
4027
4028 // Don expect correct measurement just yet, but it should be higher than
4029 // before.
4030 EXPECT_GT(fake_encoder_.GetConfiguredInputFramerate(), kLowFps);
4031
mflodmancc3d4422017-08-03 08:27:51 -07004032 video_stream_encoder_->Stop();
sprang4847ae62017-06-27 07:06:52 -07004033}
4034
mflodmancc3d4422017-08-03 08:27:51 -07004035TEST_F(VideoStreamEncoderTest, DoesNotUpdateBitrateAllocationWhenSuspended) {
sprang4847ae62017-06-27 07:06:52 -07004036 const int kFrameWidth = 1280;
4037 const int kFrameHeight = 720;
4038 const int kTargetBitrateBps = 1000000;
4039
4040 MockBitrateObserver bitrate_observer;
Niels Möller0327c2d2018-05-21 14:09:31 +02004041 video_stream_encoder_->SetBitrateAllocationObserver(&bitrate_observer);
Erik Språng4c6ca302019-04-08 15:14:01 +02004042 video_stream_encoder_->OnBitrateUpdated(
4043 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
mflodmancc3d4422017-08-03 08:27:51 -07004044 video_stream_encoder_->WaitUntilTaskQueueIsIdle();
sprang4847ae62017-06-27 07:06:52 -07004045
4046 // Insert a first video frame, causes another bitrate update.
4047 int64_t timestamp_ms = fake_clock_.TimeNanos() / rtc::kNumNanosecsPerMillisec;
4048 EXPECT_CALL(bitrate_observer, OnBitrateAllocationUpdated(_)).Times(1);
4049 video_source_.IncomingCapturedFrame(
4050 CreateFrame(timestamp_ms, kFrameWidth, kFrameHeight));
4051 WaitForEncodedFrame(timestamp_ms);
4052
4053 // Next, simulate video suspension due to pacer queue overrun.
Erik Språng4c6ca302019-04-08 15:14:01 +02004054 video_stream_encoder_->OnBitrateUpdated(DataRate::bps(0), DataRate::bps(0), 0,
Erik Språng610c7632019-03-06 15:37:33 +01004055 1);
sprang4847ae62017-06-27 07:06:52 -07004056
4057 // Skip ahead until a new periodic parameter update should have occured.
4058 timestamp_ms += vcm::VCMProcessTimer::kDefaultProcessIntervalMs;
Sebastian Jansson40889f32019-04-17 12:11:20 +02004059 fake_clock_.AdvanceTime(
4060 TimeDelta::ms(vcm::VCMProcessTimer::kDefaultProcessIntervalMs));
sprang4847ae62017-06-27 07:06:52 -07004061
4062 // Bitrate observer should not be called.
4063 EXPECT_CALL(bitrate_observer, OnBitrateAllocationUpdated(_)).Times(0);
4064 video_source_.IncomingCapturedFrame(
4065 CreateFrame(timestamp_ms, kFrameWidth, kFrameHeight));
4066 ExpectDroppedFrame();
4067
mflodmancc3d4422017-08-03 08:27:51 -07004068 video_stream_encoder_->Stop();
sprang4847ae62017-06-27 07:06:52 -07004069}
ilnik6b826ef2017-06-16 06:53:48 -07004070
Niels Möller4db138e2018-04-19 09:04:13 +02004071TEST_F(VideoStreamEncoderTest,
4072 DefaultCpuAdaptationThresholdsForSoftwareEncoder) {
4073 const int kFrameWidth = 1280;
4074 const int kFrameHeight = 720;
4075 const CpuOveruseOptions default_options;
Erik Språng4c6ca302019-04-08 15:14:01 +02004076 video_stream_encoder_->OnBitrateUpdated(
4077 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
Niels Möller4db138e2018-04-19 09:04:13 +02004078 video_source_.IncomingCapturedFrame(
4079 CreateFrame(1, kFrameWidth, kFrameHeight));
4080 WaitForEncodedFrame(1);
4081 EXPECT_EQ(video_stream_encoder_->overuse_detector_proxy_->GetOptions()
4082 .low_encode_usage_threshold_percent,
4083 default_options.low_encode_usage_threshold_percent);
4084 EXPECT_EQ(video_stream_encoder_->overuse_detector_proxy_->GetOptions()
4085 .high_encode_usage_threshold_percent,
4086 default_options.high_encode_usage_threshold_percent);
4087 video_stream_encoder_->Stop();
4088}
4089
4090TEST_F(VideoStreamEncoderTest,
4091 HigherCpuAdaptationThresholdsForHardwareEncoder) {
4092 const int kFrameWidth = 1280;
4093 const int kFrameHeight = 720;
4094 CpuOveruseOptions hardware_options;
4095 hardware_options.low_encode_usage_threshold_percent = 150;
4096 hardware_options.high_encode_usage_threshold_percent = 200;
Mirta Dvornicicccc1b572019-01-15 12:42:18 +01004097 fake_encoder_.SetIsHardwareAccelerated(true);
Niels Möller4db138e2018-04-19 09:04:13 +02004098
Erik Språng4c6ca302019-04-08 15:14:01 +02004099 video_stream_encoder_->OnBitrateUpdated(
4100 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
Niels Möller4db138e2018-04-19 09:04:13 +02004101 video_source_.IncomingCapturedFrame(
4102 CreateFrame(1, kFrameWidth, kFrameHeight));
4103 WaitForEncodedFrame(1);
4104 EXPECT_EQ(video_stream_encoder_->overuse_detector_proxy_->GetOptions()
4105 .low_encode_usage_threshold_percent,
4106 hardware_options.low_encode_usage_threshold_percent);
4107 EXPECT_EQ(video_stream_encoder_->overuse_detector_proxy_->GetOptions()
4108 .high_encode_usage_threshold_percent,
4109 hardware_options.high_encode_usage_threshold_percent);
4110 video_stream_encoder_->Stop();
4111}
4112
Niels Möller6bb5ab92019-01-11 11:11:10 +01004113TEST_F(VideoStreamEncoderTest, DropsFramesWhenEncoderOvershoots) {
4114 const int kFrameWidth = 320;
4115 const int kFrameHeight = 240;
4116 const int kFps = 30;
4117 const int kTargetBitrateBps = 120000;
4118 const int kNumFramesInRun = kFps * 5; // Runs of five seconds.
4119
Erik Språng4c6ca302019-04-08 15:14:01 +02004120 video_stream_encoder_->OnBitrateUpdated(
4121 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
Niels Möller6bb5ab92019-01-11 11:11:10 +01004122
4123 int64_t timestamp_ms = fake_clock_.TimeNanos() / rtc::kNumNanosecsPerMillisec;
4124 max_framerate_ = kFps;
4125
4126 // Insert 3 seconds of video, verify number of drops with normal bitrate.
4127 fake_encoder_.SimulateOvershoot(1.0);
4128 int num_dropped = 0;
4129 for (int i = 0; i < kNumFramesInRun; ++i) {
4130 video_source_.IncomingCapturedFrame(
4131 CreateFrame(timestamp_ms, kFrameWidth, kFrameHeight));
4132 // Wait up to two frame durations for a frame to arrive.
4133 if (!TimedWaitForEncodedFrame(timestamp_ms, 2 * 1000 / kFps)) {
4134 ++num_dropped;
4135 }
4136 timestamp_ms += 1000 / kFps;
4137 }
4138
Erik Språnga8d48ab2019-02-08 14:17:40 +01004139 // Framerate should be measured to be near the expected target rate.
4140 EXPECT_NEAR(fake_encoder_.GetLastFramerate(), kFps, 1);
4141
4142 // Frame drops should be within 5% of expected 0%.
4143 EXPECT_NEAR(num_dropped, 0, 5 * kNumFramesInRun / 100);
Niels Möller6bb5ab92019-01-11 11:11:10 +01004144
4145 // Make encoder produce frames at double the expected bitrate during 3 seconds
4146 // of video, verify number of drops. Rate needs to be slightly changed in
4147 // order to force the rate to be reconfigured.
Erik Språng7ca375c2019-02-06 16:20:17 +01004148 double overshoot_factor = 2.0;
4149 if (RateControlSettings::ParseFromFieldTrials().UseEncoderBitrateAdjuster()) {
4150 // With bitrate adjuster, when need to overshoot even more to trigger
4151 // frame dropping.
4152 overshoot_factor *= 2;
4153 }
4154 fake_encoder_.SimulateOvershoot(overshoot_factor);
Erik Språng610c7632019-03-06 15:37:33 +01004155 video_stream_encoder_->OnBitrateUpdated(
Erik Språng4c6ca302019-04-08 15:14:01 +02004156 DataRate::bps(kTargetBitrateBps + 1000),
4157 DataRate::bps(kTargetBitrateBps + 1000), 0, 0);
Niels Möller6bb5ab92019-01-11 11:11:10 +01004158 num_dropped = 0;
4159 for (int i = 0; i < kNumFramesInRun; ++i) {
4160 video_source_.IncomingCapturedFrame(
4161 CreateFrame(timestamp_ms, kFrameWidth, kFrameHeight));
4162 // Wait up to two frame durations for a frame to arrive.
4163 if (!TimedWaitForEncodedFrame(timestamp_ms, 2 * 1000 / kFps)) {
4164 ++num_dropped;
4165 }
4166 timestamp_ms += 1000 / kFps;
4167 }
4168
Erik Språng4c6ca302019-04-08 15:14:01 +02004169 video_stream_encoder_->OnBitrateUpdated(
4170 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
Erik Språnga8d48ab2019-02-08 14:17:40 +01004171
4172 // Target framerate should be still be near the expected target, despite
4173 // the frame drops.
4174 EXPECT_NEAR(fake_encoder_.GetLastFramerate(), kFps, 1);
4175
4176 // Frame drops should be within 5% of expected 50%.
4177 EXPECT_NEAR(num_dropped, kNumFramesInRun / 2, 5 * kNumFramesInRun / 100);
Niels Möller6bb5ab92019-01-11 11:11:10 +01004178
4179 video_stream_encoder_->Stop();
4180}
4181
4182TEST_F(VideoStreamEncoderTest, ConfiguresCorrectFrameRate) {
4183 const int kFrameWidth = 320;
4184 const int kFrameHeight = 240;
4185 const int kActualInputFps = 24;
4186 const int kTargetBitrateBps = 120000;
4187
4188 ASSERT_GT(max_framerate_, kActualInputFps);
4189
4190 int64_t timestamp_ms = fake_clock_.TimeNanos() / rtc::kNumNanosecsPerMillisec;
4191 max_framerate_ = kActualInputFps;
Erik Språng4c6ca302019-04-08 15:14:01 +02004192 video_stream_encoder_->OnBitrateUpdated(
4193 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
Niels Möller6bb5ab92019-01-11 11:11:10 +01004194
4195 // Insert 3 seconds of video, with an input fps lower than configured max.
4196 for (int i = 0; i < kActualInputFps * 3; ++i) {
4197 video_source_.IncomingCapturedFrame(
4198 CreateFrame(timestamp_ms, kFrameWidth, kFrameHeight));
4199 // Wait up to two frame durations for a frame to arrive.
4200 WaitForEncodedFrame(timestamp_ms);
4201 timestamp_ms += 1000 / kActualInputFps;
4202 }
4203
4204 EXPECT_NEAR(kActualInputFps, fake_encoder_.GetLastFramerate(), 1);
4205
4206 video_stream_encoder_->Stop();
4207}
4208
Ilya Nikolaevskiy71aee3a2019-02-18 13:01:26 +01004209TEST_F(VideoStreamEncoderTest, AccumulatesUpdateRectOnDroppedFrames) {
4210 VideoFrame::UpdateRect rect;
Erik Språng4c6ca302019-04-08 15:14:01 +02004211 video_stream_encoder_->OnBitrateUpdated(
4212 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
Ilya Nikolaevskiy71aee3a2019-02-18 13:01:26 +01004213
4214 fake_encoder_.BlockNextEncode();
4215 video_source_.IncomingCapturedFrame(
4216 CreateFrameWithUpdatedPixel(1, nullptr, 0));
4217 WaitForEncodedFrame(1);
4218 // On the very first frame full update should be forced.
4219 rect = fake_encoder_.GetLastUpdateRect();
4220 EXPECT_EQ(rect.offset_x, 0);
4221 EXPECT_EQ(rect.offset_y, 0);
4222 EXPECT_EQ(rect.height, codec_height_);
4223 EXPECT_EQ(rect.width, codec_width_);
4224 // Here, the encoder thread will be blocked in the TestEncoder waiting for a
4225 // call to ContinueEncode.
4226 video_source_.IncomingCapturedFrame(
4227 CreateFrameWithUpdatedPixel(2, nullptr, 1));
4228 ExpectDroppedFrame();
4229 video_source_.IncomingCapturedFrame(
4230 CreateFrameWithUpdatedPixel(3, nullptr, 10));
4231 ExpectDroppedFrame();
4232 fake_encoder_.ContinueEncode();
4233 WaitForEncodedFrame(3);
4234 // Updates to pixels 1 and 10 should be accumulated to one 10x1 rect.
4235 rect = fake_encoder_.GetLastUpdateRect();
4236 EXPECT_EQ(rect.offset_x, 1);
4237 EXPECT_EQ(rect.offset_y, 0);
4238 EXPECT_EQ(rect.width, 10);
4239 EXPECT_EQ(rect.height, 1);
4240
4241 video_source_.IncomingCapturedFrame(
4242 CreateFrameWithUpdatedPixel(4, nullptr, 0));
4243 WaitForEncodedFrame(4);
4244 // Previous frame was encoded, so no accumulation should happen.
4245 rect = fake_encoder_.GetLastUpdateRect();
4246 EXPECT_EQ(rect.offset_x, 0);
4247 EXPECT_EQ(rect.offset_y, 0);
4248 EXPECT_EQ(rect.width, 1);
4249 EXPECT_EQ(rect.height, 1);
4250
4251 video_stream_encoder_->Stop();
4252}
4253
Erik Språngd7329ca2019-02-21 21:19:53 +01004254TEST_F(VideoStreamEncoderTest, SetsFrameTypes) {
Erik Språng4c6ca302019-04-08 15:14:01 +02004255 video_stream_encoder_->OnBitrateUpdated(
4256 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
Erik Språngd7329ca2019-02-21 21:19:53 +01004257
4258 // First frame is always keyframe.
4259 video_source_.IncomingCapturedFrame(CreateFrame(1, nullptr));
4260 WaitForEncodedFrame(1);
Niels Möller8f7ce222019-03-21 15:43:58 +01004261 EXPECT_THAT(
4262 fake_encoder_.LastFrameTypes(),
Mirko Bonadei6a489f22019-04-09 15:11:12 +02004263 ::testing::ElementsAre(VideoFrameType{VideoFrameType::kVideoFrameKey}));
Erik Språngd7329ca2019-02-21 21:19:53 +01004264
4265 // Insert delta frame.
4266 video_source_.IncomingCapturedFrame(CreateFrame(2, nullptr));
4267 WaitForEncodedFrame(2);
Niels Möller8f7ce222019-03-21 15:43:58 +01004268 EXPECT_THAT(
4269 fake_encoder_.LastFrameTypes(),
Mirko Bonadei6a489f22019-04-09 15:11:12 +02004270 ::testing::ElementsAre(VideoFrameType{VideoFrameType::kVideoFrameDelta}));
Erik Språngd7329ca2019-02-21 21:19:53 +01004271
4272 // Request next frame be a key-frame.
4273 video_stream_encoder_->SendKeyFrame();
4274 video_source_.IncomingCapturedFrame(CreateFrame(3, nullptr));
4275 WaitForEncodedFrame(3);
Niels Möller8f7ce222019-03-21 15:43:58 +01004276 EXPECT_THAT(
4277 fake_encoder_.LastFrameTypes(),
Mirko Bonadei6a489f22019-04-09 15:11:12 +02004278 ::testing::ElementsAre(VideoFrameType{VideoFrameType::kVideoFrameKey}));
Erik Språngd7329ca2019-02-21 21:19:53 +01004279
4280 video_stream_encoder_->Stop();
4281}
4282
4283TEST_F(VideoStreamEncoderTest, SetsFrameTypesSimulcast) {
4284 // Setup simulcast with three streams.
4285 ResetEncoder("VP8", 3, 1, 1, false);
Erik Språng610c7632019-03-06 15:37:33 +01004286 video_stream_encoder_->OnBitrateUpdated(
Erik Språng4c6ca302019-04-08 15:14:01 +02004287 DataRate::bps(kSimulcastTargetBitrateBps),
4288 DataRate::bps(kSimulcastTargetBitrateBps), 0, 0);
Erik Språngd7329ca2019-02-21 21:19:53 +01004289 // Wait for all three layers before triggering event.
4290 sink_.SetNumExpectedLayers(3);
4291
4292 // First frame is always keyframe.
4293 video_source_.IncomingCapturedFrame(CreateFrame(1, nullptr));
4294 WaitForEncodedFrame(1);
4295 EXPECT_THAT(fake_encoder_.LastFrameTypes(),
Mirko Bonadei6a489f22019-04-09 15:11:12 +02004296 ::testing::ElementsAreArray({VideoFrameType::kVideoFrameKey,
4297 VideoFrameType::kVideoFrameKey,
4298 VideoFrameType::kVideoFrameKey}));
Erik Språngd7329ca2019-02-21 21:19:53 +01004299
4300 // Insert delta frame.
4301 video_source_.IncomingCapturedFrame(CreateFrame(2, nullptr));
4302 WaitForEncodedFrame(2);
4303 EXPECT_THAT(fake_encoder_.LastFrameTypes(),
Mirko Bonadei6a489f22019-04-09 15:11:12 +02004304 ::testing::ElementsAreArray({VideoFrameType::kVideoFrameDelta,
4305 VideoFrameType::kVideoFrameDelta,
4306 VideoFrameType::kVideoFrameDelta}));
Erik Språngd7329ca2019-02-21 21:19:53 +01004307
4308 // Request next frame be a key-frame.
4309 // Only first stream is configured to produce key-frame.
4310 video_stream_encoder_->SendKeyFrame();
4311 video_source_.IncomingCapturedFrame(CreateFrame(3, nullptr));
4312 WaitForEncodedFrame(3);
Sergey Silkine62a08a2019-05-13 13:45:39 +02004313
4314 // TODO(webrtc:10615): Map keyframe request to spatial layer. Currently
4315 // keyframe request on any layer triggers keyframe on all layers.
Erik Språngd7329ca2019-02-21 21:19:53 +01004316 EXPECT_THAT(fake_encoder_.LastFrameTypes(),
Mirko Bonadei6a489f22019-04-09 15:11:12 +02004317 ::testing::ElementsAreArray({VideoFrameType::kVideoFrameKey,
Sergey Silkine62a08a2019-05-13 13:45:39 +02004318 VideoFrameType::kVideoFrameKey,
4319 VideoFrameType::kVideoFrameKey}));
Erik Språngd7329ca2019-02-21 21:19:53 +01004320
4321 video_stream_encoder_->Stop();
4322}
4323
4324TEST_F(VideoStreamEncoderTest, RequestKeyframeInternalSource) {
4325 // Configure internal source factory and setup test again.
4326 encoder_factory_.SetHasInternalSource(true);
4327 ResetEncoder("VP8", 1, 1, 1, false);
Erik Språng4c6ca302019-04-08 15:14:01 +02004328 video_stream_encoder_->OnBitrateUpdated(
4329 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
Erik Språngd7329ca2019-02-21 21:19:53 +01004330
4331 // Call encoder directly, simulating internal source where encoded frame
4332 // callback in VideoStreamEncoder is called despite no OnFrame().
4333 fake_encoder_.InjectFrame(CreateFrame(1, nullptr), true);
4334 EXPECT_TRUE(WaitForFrame(kDefaultTimeoutMs));
Niels Möller8f7ce222019-03-21 15:43:58 +01004335 EXPECT_THAT(
4336 fake_encoder_.LastFrameTypes(),
Mirko Bonadei6a489f22019-04-09 15:11:12 +02004337 ::testing::ElementsAre(VideoFrameType{VideoFrameType::kVideoFrameKey}));
Erik Språngd7329ca2019-02-21 21:19:53 +01004338
Niels Möller8f7ce222019-03-21 15:43:58 +01004339 const std::vector<VideoFrameType> kDeltaFrame = {
4340 VideoFrameType::kVideoFrameDelta};
Erik Språngd7329ca2019-02-21 21:19:53 +01004341 // Need to set timestamp manually since manually for injected frame.
4342 VideoFrame frame = CreateFrame(101, nullptr);
4343 frame.set_timestamp(101);
4344 fake_encoder_.InjectFrame(frame, false);
4345 EXPECT_TRUE(WaitForFrame(kDefaultTimeoutMs));
Niels Möller8f7ce222019-03-21 15:43:58 +01004346 EXPECT_THAT(
4347 fake_encoder_.LastFrameTypes(),
Mirko Bonadei6a489f22019-04-09 15:11:12 +02004348 ::testing::ElementsAre(VideoFrameType{VideoFrameType::kVideoFrameDelta}));
Erik Språngd7329ca2019-02-21 21:19:53 +01004349
4350 // Request key-frame. The forces a dummy frame down into the encoder.
4351 fake_encoder_.ExpectNullFrame();
4352 video_stream_encoder_->SendKeyFrame();
4353 EXPECT_TRUE(WaitForFrame(kDefaultTimeoutMs));
Niels Möller8f7ce222019-03-21 15:43:58 +01004354 EXPECT_THAT(
4355 fake_encoder_.LastFrameTypes(),
Mirko Bonadei6a489f22019-04-09 15:11:12 +02004356 ::testing::ElementsAre(VideoFrameType{VideoFrameType::kVideoFrameKey}));
Erik Språngd7329ca2019-02-21 21:19:53 +01004357
4358 video_stream_encoder_->Stop();
4359}
Erik Språngb7cb7b52019-02-26 15:52:33 +01004360
4361TEST_F(VideoStreamEncoderTest, AdjustsTimestampInternalSource) {
4362 // Configure internal source factory and setup test again.
4363 encoder_factory_.SetHasInternalSource(true);
4364 ResetEncoder("VP8", 1, 1, 1, false);
Erik Språng4c6ca302019-04-08 15:14:01 +02004365 video_stream_encoder_->OnBitrateUpdated(
4366 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
Erik Språngb7cb7b52019-02-26 15:52:33 +01004367
4368 int64_t timestamp = 1;
4369 EncodedImage image;
Niels Möller4d504c72019-06-18 15:56:56 +02004370 image.SetEncodedData(
4371 EncodedImageBuffer::Create(kTargetBitrateBps / kDefaultFramerate / 8));
Erik Språngb7cb7b52019-02-26 15:52:33 +01004372 image.capture_time_ms_ = ++timestamp;
4373 image.SetTimestamp(static_cast<uint32_t>(timestamp * 90));
4374 const int64_t kEncodeFinishDelayMs = 10;
4375 image.timing_.encode_start_ms = timestamp;
4376 image.timing_.encode_finish_ms = timestamp + kEncodeFinishDelayMs;
4377 fake_encoder_.InjectEncodedImage(image);
4378 // Wait for frame without incrementing clock.
4379 EXPECT_TRUE(sink_.WaitForFrame(kDefaultTimeoutMs));
4380 // Frame is captured kEncodeFinishDelayMs before it's encoded, so restored
4381 // capture timestamp should be kEncodeFinishDelayMs in the past.
4382 EXPECT_EQ(sink_.GetLastCaptureTimeMs(),
4383 fake_clock_.TimeNanos() / rtc::kNumNanosecsPerMillisec -
4384 kEncodeFinishDelayMs);
4385
4386 video_stream_encoder_->Stop();
4387}
Mirta Dvornicic28f0eb22019-05-28 16:30:16 +02004388
4389TEST_F(VideoStreamEncoderTest, DoesNotRewriteH264BitstreamWithOptimalSps) {
4390 // Configure internal source factory and setup test again.
4391 encoder_factory_.SetHasInternalSource(true);
4392 ResetEncoder("H264", 1, 1, 1, false);
4393
4394 EncodedImage image(optimal_sps, sizeof(optimal_sps), sizeof(optimal_sps));
4395 image._frameType = VideoFrameType::kVideoFrameKey;
4396
4397 CodecSpecificInfo codec_specific_info;
4398 codec_specific_info.codecType = kVideoCodecH264;
4399
4400 RTPFragmentationHeader fragmentation;
4401 fragmentation.VerifyAndAllocateFragmentationHeader(1);
4402 fragmentation.fragmentationOffset[0] = 4;
4403 fragmentation.fragmentationLength[0] = sizeof(optimal_sps) - 4;
4404
4405 fake_encoder_.InjectEncodedImage(image, &codec_specific_info, &fragmentation);
4406 EXPECT_TRUE(sink_.WaitForFrame(kDefaultTimeoutMs));
4407
4408 EXPECT_THAT(sink_.GetLastEncodedImageData(),
4409 testing::ElementsAreArray(optimal_sps));
4410 RTPFragmentationHeader last_fragmentation = sink_.GetLastFragmentation();
4411 ASSERT_THAT(last_fragmentation.fragmentationVectorSize, 1U);
4412 EXPECT_EQ(last_fragmentation.fragmentationOffset[0], 4U);
4413 EXPECT_EQ(last_fragmentation.fragmentationLength[0], sizeof(optimal_sps) - 4);
4414
4415 video_stream_encoder_->Stop();
4416}
4417
4418TEST_F(VideoStreamEncoderTest, RewritesH264BitstreamWithNonOptimalSps) {
4419 uint8_t original_sps[] = {0, 0, 0, 1, H264::NaluType::kSps,
4420 0x00, 0x00, 0x03, 0x03, 0xF4,
4421 0x05, 0x03, 0xC7, 0xC0};
4422
4423 // Configure internal source factory and setup test again.
4424 encoder_factory_.SetHasInternalSource(true);
4425 ResetEncoder("H264", 1, 1, 1, false);
4426
4427 EncodedImage image(original_sps, sizeof(original_sps), sizeof(original_sps));
4428 image._frameType = VideoFrameType::kVideoFrameKey;
4429
4430 CodecSpecificInfo codec_specific_info;
4431 codec_specific_info.codecType = kVideoCodecH264;
4432
4433 RTPFragmentationHeader fragmentation;
4434 fragmentation.VerifyAndAllocateFragmentationHeader(1);
4435 fragmentation.fragmentationOffset[0] = 4;
4436 fragmentation.fragmentationLength[0] = sizeof(original_sps) - 4;
4437
4438 fake_encoder_.InjectEncodedImage(image, &codec_specific_info, &fragmentation);
4439 EXPECT_TRUE(sink_.WaitForFrame(kDefaultTimeoutMs));
4440
4441 EXPECT_THAT(sink_.GetLastEncodedImageData(),
4442 testing::ElementsAreArray(optimal_sps));
4443 RTPFragmentationHeader last_fragmentation = sink_.GetLastFragmentation();
4444 ASSERT_THAT(last_fragmentation.fragmentationVectorSize, 1U);
4445 EXPECT_EQ(last_fragmentation.fragmentationOffset[0], 4U);
4446 EXPECT_EQ(last_fragmentation.fragmentationLength[0], sizeof(optimal_sps) - 4);
4447
4448 video_stream_encoder_->Stop();
4449}
4450
Ilya Nikolaevskiyba96e2f2019-06-04 15:15:14 +02004451TEST_F(VideoStreamEncoderTest, CopiesVideoFrameMetadataAfterDownscale) {
4452 const int kFrameWidth = 1280;
4453 const int kFrameHeight = 720;
4454 const int kTargetBitrateBps = 300000; // To low for HD resolution.
4455
4456 video_stream_encoder_->OnBitrateUpdated(
4457 DataRate::bps(kTargetBitrateBps), DataRate::bps(kTargetBitrateBps), 0, 0);
4458 video_stream_encoder_->WaitUntilTaskQueueIsIdle();
4459
4460 // Insert a first video frame. It should be dropped because of downscale in
4461 // resolution.
4462 int64_t timestamp_ms = fake_clock_.TimeNanos() / rtc::kNumNanosecsPerMillisec;
4463 VideoFrame frame = CreateFrame(timestamp_ms, kFrameWidth, kFrameHeight);
4464 frame.set_rotation(kVideoRotation_270);
4465 video_source_.IncomingCapturedFrame(frame);
4466
4467 ExpectDroppedFrame();
4468
4469 // Second frame is downscaled.
4470 timestamp_ms = fake_clock_.TimeNanos() / rtc::kNumNanosecsPerMillisec;
4471 frame = CreateFrame(timestamp_ms, kFrameWidth / 2, kFrameHeight / 2);
4472 frame.set_rotation(kVideoRotation_90);
4473 video_source_.IncomingCapturedFrame(frame);
4474
4475 WaitForEncodedFrame(timestamp_ms);
4476 sink_.CheckLastFrameRotationMatches(kVideoRotation_90);
4477
4478 // Insert another frame, also downscaled.
4479 timestamp_ms = fake_clock_.TimeNanos() / rtc::kNumNanosecsPerMillisec;
4480 frame = CreateFrame(timestamp_ms, kFrameWidth / 2, kFrameHeight / 2);
4481 frame.set_rotation(kVideoRotation_180);
4482 video_source_.IncomingCapturedFrame(frame);
4483
4484 WaitForEncodedFrame(timestamp_ms);
4485 sink_.CheckLastFrameRotationMatches(kVideoRotation_180);
4486
4487 video_stream_encoder_->Stop();
4488}
4489
perkj26091b12016-09-01 01:17:40 -07004490} // namespace webrtc