blob: 1912f2ba9619a7751a82d64aeedbb232373354b8 [file] [log] [blame]
ivica5d6a06c2015-09-17 05:30:24 -07001/*
2 * Copyright (c) 2015 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 */
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020010#include "video/video_quality_test.h"
perkj9fdbda62016-09-15 09:19:20 -070011
perkja49cbd32016-09-16 07:53:41 -070012#include <stdio.h>
ivica5d6a06c2015-09-17 05:30:24 -070013#include <algorithm>
14#include <deque>
15#include <map>
sprangce4aef12015-11-02 07:23:20 -080016#include <sstream>
mflodmand1590b22015-12-09 07:07:59 -080017#include <string>
ivica5d6a06c2015-09-17 05:30:24 -070018#include <vector>
19
Artem Titovdd2eebe2018-08-20 13:27:45 +020020#include "call/fake_network_pipe.h"
21#include "call/simulated_network.h"
Elad Alon83ccca12017-10-04 13:18:26 +020022#include "logging/rtc_event_log/output/rtc_event_log_output_file.h"
henrika255750b2018-08-27 16:13:37 +020023#include "media/engine/adm_helpers.h"
Magnus Jedvert46a27652017-11-13 14:10:02 +010024#include "media/engine/internalencoderfactory.h"
Niels Möller19a68d42018-04-13 13:37:52 +020025#include "media/engine/vp8_encoder_simulcast_proxy.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020026#include "media/engine/webrtcvideoengine.h"
henrika255750b2018-08-27 16:13:37 +020027#include "modules/audio_device/include/audio_device.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020028#include "modules/audio_mixer/audio_mixer_impl.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020029#include "modules/video_coding/codecs/h264/include/h264.h"
Emircan Uysaler03e6ec92018-03-09 15:03:26 -080030#include "modules/video_coding/codecs/multiplex/include/multiplex_encoder_adapter.h"
Niels Möller4db138e2018-04-19 09:04:13 +020031#include "modules/video_coding/codecs/vp8/include/vp8.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020032#include "modules/video_coding/codecs/vp9/include/vp9.h"
Niels Möller1beef1a2018-09-03 14:21:44 +020033#include "modules/video_coding/utility/ivf_file_writer.h"
34#include "rtc_base/strings/string_builder.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020035#include "test/run_loop.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020036#include "test/testsupport/fileutils.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020037#include "test/video_renderer.h"
Sebastian Janssond4c5d632018-07-10 12:57:37 +020038#include "video/video_analyzer.h"
henrika255750b2018-08-27 16:13:37 +020039#ifdef WEBRTC_WIN
40#include "modules/audio_device/include/audio_device_factory.h"
41#endif
ilnikee42d192017-08-22 07:16:20 -070042
Niels Möller1beef1a2018-09-03 14:21:44 +020043namespace webrtc {
44
minyue73208662016-08-18 06:28:55 -070045namespace {
minyue73208662016-08-18 06:28:55 -070046constexpr char kSyncGroup[] = "av_sync";
minyue10cbb462016-11-07 09:29:22 -080047constexpr int kOpusMinBitrateBps = 6000;
48constexpr int kOpusBitrateFbBps = 32000;
ilnik9ae0d762017-02-15 00:53:12 -080049constexpr int kFramesSentInQuickTest = 1;
ilnika014cc52017-03-07 04:21:04 -080050constexpr uint32_t kThumbnailSendSsrcStart = 0xE0000;
51constexpr uint32_t kThumbnailRtxSsrcStart = 0xF0000;
minyue73208662016-08-18 06:28:55 -070052
sprang1168fd42017-06-21 09:00:17 -070053constexpr int kDefaultMaxQp = cricket::WebRtcVideoChannel::kDefaultQpMax;
54
perkjfa10b552016-10-02 23:45:26 -070055class VideoStreamFactory
Niels Möller1beef1a2018-09-03 14:21:44 +020056 : public VideoEncoderConfig::VideoStreamFactoryInterface {
perkjfa10b552016-10-02 23:45:26 -070057 public:
Niels Möller1beef1a2018-09-03 14:21:44 +020058 explicit VideoStreamFactory(const std::vector<VideoStream>& streams)
perkjfa10b552016-10-02 23:45:26 -070059 : streams_(streams) {}
60
61 private:
Niels Möller1beef1a2018-09-03 14:21:44 +020062 std::vector<VideoStream> CreateEncoderStreams(
perkjfa10b552016-10-02 23:45:26 -070063 int width,
64 int height,
Niels Möller1beef1a2018-09-03 14:21:44 +020065 const VideoEncoderConfig& encoder_config) override {
mflodmand79f97b2016-12-15 07:24:33 -080066 // The highest layer must match the incoming resolution.
Niels Möller1beef1a2018-09-03 14:21:44 +020067 std::vector<VideoStream> streams = streams_;
mflodmand79f97b2016-12-15 07:24:33 -080068 streams[streams_.size() - 1].height = height;
69 streams[streams_.size() - 1].width = width;
Seth Hampson24722b32017-12-22 09:36:42 -080070
71 streams[0].bitrate_priority = encoder_config.bitrate_priority;
mflodmand79f97b2016-12-15 07:24:33 -080072 return streams;
perkjfa10b552016-10-02 23:45:26 -070073 }
74
Niels Möller1beef1a2018-09-03 14:21:44 +020075 std::vector<VideoStream> streams_;
perkjfa10b552016-10-02 23:45:26 -070076};
ivica5d6a06c2015-09-17 05:30:24 -070077
Niels Möller1beef1a2018-09-03 14:21:44 +020078// An encoder wrapper that writes the encoded frames to file, one per simulcast
79// layer.
80class FrameDumpingEncoder : public VideoEncoder, private EncodedImageCallback {
81 public:
82 FrameDumpingEncoder(std::unique_ptr<VideoEncoder> encoder,
83 std::vector<rtc::PlatformFile> files)
84 : encoder_(std::move(encoder)) {
85 for (rtc::PlatformFile file : files) {
86 writers_.push_back(
87 IvfFileWriter::Wrap(rtc::File(file), 100000000 /* byte_limit */));
88 }
89 }
90 // Implement VideoEncoder
91 int32_t InitEncode(const VideoCodec* codec_settings,
92 int32_t number_of_cores,
93 size_t max_payload_size) override {
94 return encoder_->InitEncode(codec_settings, number_of_cores,
95 max_payload_size);
96 }
97 int32_t RegisterEncodeCompleteCallback(
98 EncodedImageCallback* callback) override {
99 callback_ = callback;
100 return encoder_->RegisterEncodeCompleteCallback(this);
101 }
102 int32_t Release() override { return encoder_->Release(); }
103 int32_t Encode(const VideoFrame& frame,
104 const CodecSpecificInfo* codec_specific_info,
105 const std::vector<FrameType>* frame_types) {
106 return encoder_->Encode(frame, codec_specific_info, frame_types);
107 }
108 int32_t SetChannelParameters(uint32_t packet_loss, int64_t rtt) override {
109 return encoder_->SetChannelParameters(packet_loss, rtt);
110 }
111 int32_t SetRates(uint32_t bitrate, uint32_t framerate) override {
112 return encoder_->SetRates(bitrate, framerate);
113 }
114 int32_t SetRateAllocation(const VideoBitrateAllocation& allocation,
115 uint32_t framerate) override {
116 return encoder_->SetRateAllocation(allocation, framerate);
117 }
118 ScalingSettings GetScalingSettings() const override {
119 return encoder_->GetScalingSettings();
120 }
121 bool SupportsNativeHandle() const override {
122 return encoder_->SupportsNativeHandle();
123 }
124 const char* ImplementationName() const override {
125 return encoder_->ImplementationName();
126 }
127
128 private:
129 // Implement EncodedImageCallback
130 Result OnEncodedImage(const EncodedImage& encoded_image,
131 const CodecSpecificInfo* codec_specific_info,
132 const RTPFragmentationHeader* fragmentation) override {
133 if (codec_specific_info) {
134 int simulcast_index;
135 if (codec_specific_info->codecType == kVideoCodecVP9) {
136 simulcast_index = 0;
137 } else {
138 simulcast_index = encoded_image.SpatialIndex().value_or(0);
139 }
140 RTC_DCHECK_GE(simulcast_index, 0);
141 if (static_cast<size_t>(simulcast_index) < writers_.size()) {
142 writers_[simulcast_index]->WriteFrame(encoded_image,
143 codec_specific_info->codecType);
144 }
145 }
146
147 return callback_->OnEncodedImage(encoded_image, codec_specific_info,
148 fragmentation);
149 }
150
151 void OnDroppedFrame(DropReason reason) override {
152 callback_->OnDroppedFrame(reason);
153 }
154
155 std::unique_ptr<VideoEncoder> encoder_;
156 EncodedImageCallback* callback_ = nullptr;
157 std::vector<std::unique_ptr<IvfFileWriter>> writers_;
158};
159
160} // namespace
ivica5d6a06c2015-09-17 05:30:24 -0700161
Sebastian Janssone6d7c3e2018-07-11 15:00:41 +0200162std::unique_ptr<VideoEncoder> VideoQualityTest::CreateVideoEncoder(
Niels Möller4db138e2018-04-19 09:04:13 +0200163 const SdpVideoFormat& format) {
Niels Möller1beef1a2018-09-03 14:21:44 +0200164 std::unique_ptr<VideoEncoder> encoder;
Niels Möller4db138e2018-04-19 09:04:13 +0200165 if (format.name == "VP8") {
Niels Möller1beef1a2018-09-03 14:21:44 +0200166 encoder = absl::make_unique<VP8EncoderSimulcastProxy>(
Karl Wiberg918f50c2018-07-05 11:40:33 +0200167 &internal_encoder_factory_, format);
Niels Möller4db138e2018-04-19 09:04:13 +0200168 } else if (format.name == "multiplex") {
Niels Möller1beef1a2018-09-03 14:21:44 +0200169 encoder = absl::make_unique<MultiplexEncoderAdapter>(
Niels Möller4db138e2018-04-19 09:04:13 +0200170 &internal_encoder_factory_, SdpVideoFormat(cricket::kVp9CodecName));
Niels Möller1beef1a2018-09-03 14:21:44 +0200171 } else {
172 encoder = internal_encoder_factory_.CreateVideoEncoder(format);
Niels Möller4db138e2018-04-19 09:04:13 +0200173 }
Niels Möller1beef1a2018-09-03 14:21:44 +0200174 if (!params_.logging.encoded_frame_base_path.empty()) {
175 char ss_buf[100];
176 rtc::SimpleStringBuilder sb(ss_buf);
177 sb << send_logs_++;
178 std::string prefix =
179 params_.logging.encoded_frame_base_path + "." + sb.str() + ".send.";
180 encoder = absl::make_unique<FrameDumpingEncoder>(
181 std::move(encoder), std::vector<rtc::PlatformFile>(
182 {rtc::CreatePlatformFile(prefix + "1.ivf"),
183 rtc::CreatePlatformFile(prefix + "2.ivf"),
184 rtc::CreatePlatformFile(prefix + "3.ivf")}));
185 }
186 return encoder;
Niels Möller4db138e2018-04-19 09:04:13 +0200187}
188
Patrik Höglundb6b29e02018-06-21 16:58:01 +0200189VideoQualityTest::VideoQualityTest(
Artem Titove269cb42018-08-29 09:59:23 +0200190 std::unique_ptr<InjectionComponents> injection_components)
Sebastian Janssone6d7c3e2018-07-11 15:00:41 +0200191 : clock_(Clock::GetRealTimeClock()),
192 video_encoder_factory_([this](const SdpVideoFormat& format) {
193 return this->CreateVideoEncoder(format);
194 }),
195 receive_logs_(0),
Artem Titove269cb42018-08-29 09:59:23 +0200196 send_logs_(0),
197 injection_components_(std::move(injection_components)) {
198 if (injection_components_ == nullptr) {
199 injection_components_ = absl::make_unique<InjectionComponents>();
200 }
201
minyue20c84cc2017-04-10 16:57:57 -0700202 payload_type_map_ = test::CallTest::payload_type_map_;
203 RTC_DCHECK(payload_type_map_.find(kPayloadTypeH264) ==
204 payload_type_map_.end());
205 RTC_DCHECK(payload_type_map_.find(kPayloadTypeVP8) ==
206 payload_type_map_.end());
207 RTC_DCHECK(payload_type_map_.find(kPayloadTypeVP9) ==
208 payload_type_map_.end());
209 payload_type_map_[kPayloadTypeH264] = webrtc::MediaType::VIDEO;
210 payload_type_map_[kPayloadTypeVP8] = webrtc::MediaType::VIDEO;
211 payload_type_map_[kPayloadTypeVP9] = webrtc::MediaType::VIDEO;
ivica5d6a06c2015-09-17 05:30:24 -0700212
Artem Titove269cb42018-08-29 09:59:23 +0200213 fec_controller_factory_ =
214 std::move(injection_components_->fec_controller_factory);
Ying Wang3b790f32018-01-19 17:58:57 +0100215}
216
minyue626bc952016-10-31 05:47:02 -0700217VideoQualityTest::Params::Params()
Sebastian Janssonfc8d26b2018-02-21 09:52:06 +0100218 : call({false, BitrateConstraints(), 0}),
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100219 video{{false, 640, 480, 30, 50, 800, 800, false, "VP8", 1, -1, 0, false,
Niels Möller6aa415e2018-06-07 11:14:13 +0200220 false, false, ""},
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100221 {false, 640, 480, 30, 50, 800, 800, false, "VP8", 1, -1, 0, false,
Niels Möller6aa415e2018-06-07 11:14:13 +0200222 false, false, ""}},
henrika255750b2018-08-27 16:13:37 +0200223 audio({false, false, false, false}),
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100224 screenshare{{false, false, 10, 0}, {false, false, 10, 0}},
minyue626bc952016-10-31 05:47:02 -0700225 analyzer({"", 0.0, 0.0, 0, "", ""}),
226 pipe(),
Artem Titovf18b3522018-08-28 16:54:24 +0200227 config(absl::nullopt),
Sergey Silkin57027362018-05-15 09:12:05 +0200228 ss{{std::vector<VideoStream>(), 0, 0, -1, InterLayerPredMode::kOn,
229 std::vector<SpatialLayer>()},
230 {std::vector<VideoStream>(), 0, 0, -1, InterLayerPredMode::kOn,
231 std::vector<SpatialLayer>()}},
Mirko Bonadei45a4c412018-07-31 15:07:28 +0200232 logging({"", "", ""}) {}
minyue626bc952016-10-31 05:47:02 -0700233
234VideoQualityTest::Params::~Params() = default;
235
Artem Titove269cb42018-08-29 09:59:23 +0200236VideoQualityTest::InjectionComponents::InjectionComponents() = default;
237
238VideoQualityTest::InjectionComponents::~InjectionComponents() = default;
239
ivica5d6a06c2015-09-17 05:30:24 -0700240void VideoQualityTest::TestBody() {}
241
sprangce4aef12015-11-02 07:23:20 -0800242std::string VideoQualityTest::GenerateGraphTitle() const {
243 std::stringstream ss;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100244 ss << params_.video[0].codec;
245 ss << " (" << params_.video[0].target_bitrate_bps / 1000 << "kbps";
246 ss << ", " << params_.video[0].fps << " FPS";
247 if (params_.screenshare[0].scroll_duration)
248 ss << ", " << params_.screenshare[0].scroll_duration << "s scroll";
249 if (params_.ss[0].streams.size() > 1)
250 ss << ", Stream #" << params_.ss[0].selected_stream;
251 if (params_.ss[0].num_spatial_layers > 1)
252 ss << ", Layer #" << params_.ss[0].selected_sl;
sprangce4aef12015-11-02 07:23:20 -0800253 ss << ")";
254 return ss.str();
255}
256
Artem Titove269cb42018-08-29 09:59:23 +0200257void VideoQualityTest::CheckParamsAndInjectionComponents() {
258 if (injection_components_ == nullptr) {
259 injection_components_ = absl::make_unique<InjectionComponents>();
260 }
261 if (!params_.config && injection_components_->sender_network == nullptr &&
262 injection_components_->receiver_network == nullptr) {
Artem Titovf18b3522018-08-28 16:54:24 +0200263 // TODO(titovartem) replace with default config creation when removing
264 // pipe.
265 params_.config = params_.pipe;
266 }
Artem Titove269cb42018-08-29 09:59:23 +0200267 RTC_CHECK(
268 (params_.config && injection_components_->sender_network == nullptr &&
269 injection_components_->receiver_network == nullptr) ||
270 (!params_.config && injection_components_->sender_network != nullptr &&
271 injection_components_->receiver_network != nullptr));
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100272 for (size_t video_idx = 0; video_idx < num_video_streams_; ++video_idx) {
273 // Iterate over primary and secondary video streams.
274 if (!params_.video[video_idx].enabled)
275 return;
276 // Add a default stream in none specified.
277 if (params_.ss[video_idx].streams.empty())
278 params_.ss[video_idx].streams.push_back(
279 VideoQualityTest::DefaultVideoStream(params_, video_idx));
280 if (params_.ss[video_idx].num_spatial_layers == 0)
281 params_.ss[video_idx].num_spatial_layers = 1;
sprangce4aef12015-11-02 07:23:20 -0800282
Artem Titove269cb42018-08-29 09:59:23 +0200283 if (params_.config) {
284 if (params_.config->loss_percent != 0 ||
285 params_.config->queue_length_packets != 0) {
286 // Since LayerFilteringTransport changes the sequence numbers, we can't
287 // use that feature with pack loss, since the NACK request would end up
288 // retransmitting the wrong packets.
289 RTC_CHECK(params_.ss[video_idx].selected_sl == -1 ||
290 params_.ss[video_idx].selected_sl ==
291 params_.ss[video_idx].num_spatial_layers - 1);
292 RTC_CHECK(params_.video[video_idx].selected_tl == -1 ||
293 params_.video[video_idx].selected_tl ==
294 params_.video[video_idx].num_temporal_layers - 1);
295 }
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100296 }
sprangce4aef12015-11-02 07:23:20 -0800297
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100298 // TODO(ivica): Should max_bitrate_bps == -1 represent inf max bitrate, as
299 // it does in some parts of the code?
300 RTC_CHECK_GE(params_.video[video_idx].max_bitrate_bps,
301 params_.video[video_idx].target_bitrate_bps);
302 RTC_CHECK_GE(params_.video[video_idx].target_bitrate_bps,
303 params_.video[video_idx].min_bitrate_bps);
304 RTC_CHECK_LT(params_.video[video_idx].selected_tl,
305 params_.video[video_idx].num_temporal_layers);
306 RTC_CHECK_LE(params_.ss[video_idx].selected_stream,
307 params_.ss[video_idx].streams.size());
308 for (const VideoStream& stream : params_.ss[video_idx].streams) {
309 RTC_CHECK_GE(stream.min_bitrate_bps, 0);
310 RTC_CHECK_GE(stream.target_bitrate_bps, stream.min_bitrate_bps);
311 RTC_CHECK_GE(stream.max_bitrate_bps, stream.target_bitrate_bps);
312 }
313 // TODO(ivica): Should we check if the sum of all streams/layers is equal to
314 // the total bitrate? We anyway have to update them in the case bitrate
315 // estimator changes the total bitrates.
316 RTC_CHECK_GE(params_.ss[video_idx].num_spatial_layers, 1);
317 RTC_CHECK_LE(params_.ss[video_idx].selected_sl,
318 params_.ss[video_idx].num_spatial_layers);
319 RTC_CHECK(
320 params_.ss[video_idx].spatial_layers.empty() ||
321 params_.ss[video_idx].spatial_layers.size() ==
322 static_cast<size_t>(params_.ss[video_idx].num_spatial_layers));
323 if (params_.video[video_idx].codec == "VP8") {
324 RTC_CHECK_EQ(params_.ss[video_idx].num_spatial_layers, 1);
325 } else if (params_.video[video_idx].codec == "VP9") {
326 RTC_CHECK_EQ(params_.ss[video_idx].streams.size(), 1);
327 }
328 RTC_CHECK_GE(params_.call.num_thumbnails, 0);
329 if (params_.call.num_thumbnails > 0) {
330 RTC_CHECK_EQ(params_.ss[video_idx].num_spatial_layers, 1);
331 RTC_CHECK_EQ(params_.ss[video_idx].streams.size(), 3);
332 RTC_CHECK_EQ(params_.video[video_idx].num_temporal_layers, 3);
333 RTC_CHECK_EQ(params_.video[video_idx].codec, "VP8");
334 }
335 // Dual streams with FEC not supported in tests yet.
336 RTC_CHECK(!params_.video[video_idx].flexfec || num_video_streams_ == 1);
337 RTC_CHECK(!params_.video[video_idx].ulpfec || num_video_streams_ == 1);
ilnika014cc52017-03-07 04:21:04 -0800338 }
sprangce4aef12015-11-02 07:23:20 -0800339}
340
341// Static.
342std::vector<int> VideoQualityTest::ParseCSV(const std::string& str) {
343 // Parse comma separated nonnegative integers, where some elements may be
344 // empty. The empty values are replaced with -1.
345 // E.g. "10,-20,,30,40" --> {10, 20, -1, 30,40}
346 // E.g. ",,10,,20," --> {-1, -1, 10, -1, 20, -1}
347 std::vector<int> result;
348 if (str.empty())
349 return result;
350
351 const char* p = str.c_str();
352 int value = -1;
353 int pos;
354 while (*p) {
355 if (*p == ',') {
356 result.push_back(value);
357 value = -1;
358 ++p;
359 continue;
360 }
361 RTC_CHECK_EQ(sscanf(p, "%d%n", &value, &pos), 1)
362 << "Unexpected non-number value.";
363 p += pos;
364 }
365 result.push_back(value);
366 return result;
367}
368
369// Static.
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100370VideoStream VideoQualityTest::DefaultVideoStream(const Params& params,
371 size_t video_idx) {
sprangce4aef12015-11-02 07:23:20 -0800372 VideoStream stream;
Danil Chapovalov350531e2018-06-08 11:04:04 +0000373 stream.width = params.video[video_idx].width;
374 stream.height = params.video[video_idx].height;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100375 stream.max_framerate = params.video[video_idx].fps;
376 stream.min_bitrate_bps = params.video[video_idx].min_bitrate_bps;
377 stream.target_bitrate_bps = params.video[video_idx].target_bitrate_bps;
378 stream.max_bitrate_bps = params.video[video_idx].max_bitrate_bps;
sprang1168fd42017-06-21 09:00:17 -0700379 stream.max_qp = kDefaultMaxQp;
Sergey Silkina796a7e2018-03-01 15:11:29 +0100380 stream.num_temporal_layers = params.video[video_idx].num_temporal_layers;
Seth Hampson46e31ba2018-01-18 10:39:54 -0800381 stream.active = true;
ilnika014cc52017-03-07 04:21:04 -0800382 return stream;
383}
384
385// Static.
386VideoStream VideoQualityTest::DefaultThumbnailStream() {
387 VideoStream stream;
388 stream.width = 320;
389 stream.height = 180;
390 stream.max_framerate = 7;
391 stream.min_bitrate_bps = 7500;
392 stream.target_bitrate_bps = 37500;
393 stream.max_bitrate_bps = 50000;
sprang1168fd42017-06-21 09:00:17 -0700394 stream.max_qp = kDefaultMaxQp;
sprangce4aef12015-11-02 07:23:20 -0800395 return stream;
396}
397
398// Static.
399void VideoQualityTest::FillScalabilitySettings(
400 Params* params,
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100401 size_t video_idx,
sprangce4aef12015-11-02 07:23:20 -0800402 const std::vector<std::string>& stream_descriptors,
sprang1168fd42017-06-21 09:00:17 -0700403 int num_streams,
sprangce4aef12015-11-02 07:23:20 -0800404 size_t selected_stream,
405 int num_spatial_layers,
406 int selected_sl,
Sergey Silkin57027362018-05-15 09:12:05 +0200407 InterLayerPredMode inter_layer_pred,
sprangce4aef12015-11-02 07:23:20 -0800408 const std::vector<std::string>& sl_descriptors) {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100409 if (params->ss[video_idx].streams.empty() &&
410 params->ss[video_idx].infer_streams) {
sprang1168fd42017-06-21 09:00:17 -0700411 webrtc::VideoEncoderConfig encoder_config;
Niels Möller259a4972018-04-05 15:36:51 +0200412 encoder_config.codec_type =
413 PayloadStringToCodecType(params->video[video_idx].codec);
sprang1168fd42017-06-21 09:00:17 -0700414 encoder_config.content_type =
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100415 params->screenshare[video_idx].enabled
sprang1168fd42017-06-21 09:00:17 -0700416 ? webrtc::VideoEncoderConfig::ContentType::kScreen
417 : webrtc::VideoEncoderConfig::ContentType::kRealtimeVideo;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100418 encoder_config.max_bitrate_bps = params->video[video_idx].max_bitrate_bps;
419 encoder_config.min_transmit_bitrate_bps =
420 params->video[video_idx].min_transmit_bps;
sprang1168fd42017-06-21 09:00:17 -0700421 encoder_config.number_of_streams = num_streams;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100422 encoder_config.spatial_layers = params->ss[video_idx].spatial_layers;
Seth Hampson8234ead2018-02-02 15:16:24 -0800423 encoder_config.simulcast_layers = std::vector<VideoStream>(num_streams);
sprang1168fd42017-06-21 09:00:17 -0700424 encoder_config.video_stream_factory =
425 new rtc::RefCountedObject<cricket::EncoderStreamFactory>(
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100426 params->video[video_idx].codec, kDefaultMaxQp,
Mirko Bonadei948b7e32018-08-14 07:23:21 +0000427 params->video[video_idx].fps,
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100428 params->screenshare[video_idx].enabled, true);
429 params->ss[video_idx].streams =
sprang1168fd42017-06-21 09:00:17 -0700430 encoder_config.video_stream_factory->CreateEncoderStreams(
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100431 static_cast<int>(params->video[video_idx].width),
432 static_cast<int>(params->video[video_idx].height), encoder_config);
sprang1168fd42017-06-21 09:00:17 -0700433 } else {
434 // Read VideoStream and SpatialLayer elements from a list of comma separated
435 // lists. To use a default value for an element, use -1 or leave empty.
Artem Titove269cb42018-08-29 09:59:23 +0200436 // Validity checks performed in CheckParamsAndInjectionComponents.
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100437 RTC_CHECK(params->ss[video_idx].streams.empty());
sprang1168fd42017-06-21 09:00:17 -0700438 for (auto descriptor : stream_descriptors) {
439 if (descriptor.empty())
440 continue;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100441 VideoStream stream =
442 VideoQualityTest::DefaultVideoStream(*params, video_idx);
sprang1168fd42017-06-21 09:00:17 -0700443 std::vector<int> v = VideoQualityTest::ParseCSV(descriptor);
444 if (v[0] != -1)
445 stream.width = static_cast<size_t>(v[0]);
446 if (v[1] != -1)
447 stream.height = static_cast<size_t>(v[1]);
448 if (v[2] != -1)
449 stream.max_framerate = v[2];
450 if (v[3] != -1)
451 stream.min_bitrate_bps = v[3];
452 if (v[4] != -1)
453 stream.target_bitrate_bps = v[4];
454 if (v[5] != -1)
455 stream.max_bitrate_bps = v[5];
456 if (v.size() > 6 && v[6] != -1)
457 stream.max_qp = v[6];
Sergey Silkina796a7e2018-03-01 15:11:29 +0100458 if (v.size() > 7 && v[7] != -1) {
459 stream.num_temporal_layers = v[7];
sprang1168fd42017-06-21 09:00:17 -0700460 } else {
461 // Automatic TL thresholds for more than two layers not supported.
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100462 RTC_CHECK_LE(params->video[video_idx].num_temporal_layers, 2);
sprang1168fd42017-06-21 09:00:17 -0700463 }
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100464 params->ss[video_idx].streams.push_back(stream);
sprangce4aef12015-11-02 07:23:20 -0800465 }
sprangce4aef12015-11-02 07:23:20 -0800466 }
sprangce4aef12015-11-02 07:23:20 -0800467
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100468 params->ss[video_idx].num_spatial_layers = std::max(1, num_spatial_layers);
469 params->ss[video_idx].selected_stream = selected_stream;
sprang1168fd42017-06-21 09:00:17 -0700470
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100471 params->ss[video_idx].selected_sl = selected_sl;
Sergey Silkin57027362018-05-15 09:12:05 +0200472 params->ss[video_idx].inter_layer_pred = inter_layer_pred;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100473 RTC_CHECK(params->ss[video_idx].spatial_layers.empty());
sprangce4aef12015-11-02 07:23:20 -0800474 for (auto descriptor : sl_descriptors) {
475 if (descriptor.empty())
476 continue;
477 std::vector<int> v = VideoQualityTest::ParseCSV(descriptor);
Sergey Silkin13e74342018-03-02 12:28:00 +0100478 RTC_CHECK_EQ(v.size(), 7);
sprangce4aef12015-11-02 07:23:20 -0800479
Sergey Silkin13e74342018-03-02 12:28:00 +0100480 SpatialLayer layer = {0};
481 layer.width = v[0];
482 layer.height = v[1];
483 layer.numberOfTemporalLayers = v[2];
484 layer.maxBitrate = v[3];
485 layer.minBitrate = v[4];
486 layer.targetBitrate = v[5];
487 layer.qpMax = v[6];
488 layer.active = true;
489
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100490 params->ss[video_idx].spatial_layers.push_back(layer);
sprangce4aef12015-11-02 07:23:20 -0800491 }
492}
493
minyuea27172d2016-11-01 05:59:29 -0700494void VideoQualityTest::SetupVideo(Transport* send_transport,
495 Transport* recv_transport) {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100496 size_t total_streams_used = 0;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100497 video_receive_configs_.clear();
498 video_send_configs_.clear();
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100499 video_encoder_configs_.clear();
500 allocated_decoders_.clear();
501 bool decode_all_receive_streams = true;
502 size_t num_video_substreams = params_.ss[0].streams.size();
503 RTC_CHECK(num_video_streams_ > 0);
504 video_encoder_configs_.resize(num_video_streams_);
505 for (size_t video_idx = 0; video_idx < num_video_streams_; ++video_idx) {
506 video_send_configs_.push_back(VideoSendStream::Config(send_transport));
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100507 video_encoder_configs_.push_back(VideoEncoderConfig());
508 num_video_substreams = params_.ss[video_idx].streams.size();
509 RTC_CHECK_GT(num_video_substreams, 0);
Sebastian Jansson3bd2c792018-07-13 13:29:03 +0200510 for (size_t i = 0; i < num_video_substreams; ++i)
511 video_send_configs_[video_idx].rtp.ssrcs.push_back(
512 kVideoSendSsrcs[total_streams_used + i]);
Qingsi Wang2d82ade2018-07-11 06:04:44 +0000513
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100514 int payload_type;
515 if (params_.video[video_idx].codec == "H264") {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100516 payload_type = kPayloadTypeH264;
517 } else if (params_.video[video_idx].codec == "VP8") {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100518 payload_type = kPayloadTypeVP8;
519 } else if (params_.video[video_idx].codec == "VP9") {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100520 payload_type = kPayloadTypeVP9;
Emircan Uysaler03e6ec92018-03-09 15:03:26 -0800521 } else if (params_.video[video_idx].codec == "multiplex") {
Emircan Uysaler03e6ec92018-03-09 15:03:26 -0800522 payload_type = kPayloadTypeVP9;
ilnikcb8c1462017-03-09 09:23:30 -0800523 } else {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100524 RTC_NOTREACHED() << "Codec not supported!";
525 return;
ilnikcb8c1462017-03-09 09:23:30 -0800526 }
Niels Möller4db138e2018-04-19 09:04:13 +0200527 video_send_configs_[video_idx].encoder_settings.encoder_factory =
528 &video_encoder_factory_;
529
Niels Möller259a4972018-04-05 15:36:51 +0200530 video_send_configs_[video_idx].rtp.payload_name =
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100531 params_.video[video_idx].codec;
Niels Möller259a4972018-04-05 15:36:51 +0200532 video_send_configs_[video_idx].rtp.payload_type = payload_type;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100533 video_send_configs_[video_idx].rtp.nack.rtp_history_ms = kNackRtpHistoryMs;
534 video_send_configs_[video_idx].rtp.rtx.payload_type = kSendRtxPayloadType;
535 for (size_t i = 0; i < num_video_substreams; ++i) {
536 video_send_configs_[video_idx].rtp.rtx.ssrcs.push_back(
537 kSendRtxSsrcs[i + total_streams_used]);
ilnik9fd9f6c2017-03-02 08:10:10 -0800538 }
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100539 video_send_configs_[video_idx].rtp.extensions.clear();
540 if (params_.call.send_side_bwe) {
541 video_send_configs_[video_idx].rtp.extensions.push_back(
542 RtpExtension(RtpExtension::kTransportSequenceNumberUri,
543 test::kTransportSequenceNumberExtensionId));
544 } else {
545 video_send_configs_[video_idx].rtp.extensions.push_back(RtpExtension(
546 RtpExtension::kAbsSendTimeUri, test::kAbsSendTimeExtensionId));
547 }
548 video_send_configs_[video_idx].rtp.extensions.push_back(
549 RtpExtension(RtpExtension::kVideoContentTypeUri,
550 test::kVideoContentTypeExtensionId));
551 video_send_configs_[video_idx].rtp.extensions.push_back(RtpExtension(
552 RtpExtension::kVideoTimingUri, test::kVideoTimingExtensionId));
553
Niels Möller4db138e2018-04-19 09:04:13 +0200554 video_encoder_configs_[video_idx].video_format.name =
555 params_.video[video_idx].codec;
556
Emircan Uysaler0823eec2018-07-13 17:10:00 -0700557 video_encoder_configs_[video_idx].video_format.parameters =
558 params_.video[video_idx].sdp_params;
559
Niels Möller259a4972018-04-05 15:36:51 +0200560 video_encoder_configs_[video_idx].codec_type =
561 PayloadStringToCodecType(params_.video[video_idx].codec);
562
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100563 video_encoder_configs_[video_idx].min_transmit_bitrate_bps =
564 params_.video[video_idx].min_transmit_bps;
565
566 video_send_configs_[video_idx].suspend_below_min_bitrate =
567 params_.video[video_idx].suspend_below_min_bitrate;
568
569 video_encoder_configs_[video_idx].number_of_streams =
570 params_.ss[video_idx].streams.size();
571 video_encoder_configs_[video_idx].max_bitrate_bps = 0;
572 for (size_t i = 0; i < params_.ss[video_idx].streams.size(); ++i) {
573 video_encoder_configs_[video_idx].max_bitrate_bps +=
574 params_.ss[video_idx].streams[i].max_bitrate_bps;
575 }
576 if (params_.ss[video_idx].infer_streams) {
Seth Hampson8234ead2018-02-02 15:16:24 -0800577 video_encoder_configs_[video_idx].simulcast_layers =
578 std::vector<VideoStream>(params_.ss[video_idx].streams.size());
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100579 video_encoder_configs_[video_idx].video_stream_factory =
580 new rtc::RefCountedObject<cricket::EncoderStreamFactory>(
581 params_.video[video_idx].codec,
582 params_.ss[video_idx].streams[0].max_qp,
Mirko Bonadei948b7e32018-08-14 07:23:21 +0000583 params_.video[video_idx].fps,
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100584 params_.screenshare[video_idx].enabled, true);
585 } else {
586 video_encoder_configs_[video_idx].video_stream_factory =
587 new rtc::RefCountedObject<VideoStreamFactory>(
588 params_.ss[video_idx].streams);
589 }
590
591 video_encoder_configs_[video_idx].spatial_layers =
592 params_.ss[video_idx].spatial_layers;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100593 decode_all_receive_streams = params_.ss[video_idx].selected_stream ==
594 params_.ss[video_idx].streams.size();
Sebastian Jansson3bd2c792018-07-13 13:29:03 +0200595 absl::optional<int> decode_sub_stream;
596 if (!decode_all_receive_streams)
597 decode_sub_stream = params_.ss[video_idx].selected_stream;
598 CreateMatchingVideoReceiveConfigs(
599 video_send_configs_[video_idx], recv_transport,
600 params_.call.send_side_bwe, decode_sub_stream, true, kNackRtpHistoryMs);
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100601
602 if (params_.screenshare[video_idx].enabled) {
603 // Fill out codec settings.
604 video_encoder_configs_[video_idx].content_type =
605 VideoEncoderConfig::ContentType::kScreen;
Taylor Brandstetter49fcc102018-05-16 14:20:41 -0700606 degradation_preference_ = DegradationPreference::MAINTAIN_RESOLUTION;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100607 if (params_.video[video_idx].codec == "VP8") {
608 VideoCodecVP8 vp8_settings = VideoEncoder::GetDefaultVp8Settings();
609 vp8_settings.denoisingOn = false;
610 vp8_settings.frameDroppingOn = false;
611 vp8_settings.numberOfTemporalLayers = static_cast<unsigned char>(
612 params_.video[video_idx].num_temporal_layers);
613 video_encoder_configs_[video_idx].encoder_specific_settings =
614 new rtc::RefCountedObject<
615 VideoEncoderConfig::Vp8EncoderSpecificSettings>(vp8_settings);
616 } else if (params_.video[video_idx].codec == "VP9") {
617 VideoCodecVP9 vp9_settings = VideoEncoder::GetDefaultVp9Settings();
618 vp9_settings.denoisingOn = false;
619 vp9_settings.frameDroppingOn = false;
620 vp9_settings.numberOfTemporalLayers = static_cast<unsigned char>(
621 params_.video[video_idx].num_temporal_layers);
622 vp9_settings.numberOfSpatialLayers = static_cast<unsigned char>(
623 params_.ss[video_idx].num_spatial_layers);
Sergey Silkin57027362018-05-15 09:12:05 +0200624 vp9_settings.interLayerPred = params_.ss[video_idx].inter_layer_pred;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100625 video_encoder_configs_[video_idx].encoder_specific_settings =
626 new rtc::RefCountedObject<
627 VideoEncoderConfig::Vp9EncoderSpecificSettings>(vp9_settings);
628 }
629 } else if (params_.ss[video_idx].num_spatial_layers > 1) {
630 // If SVC mode without screenshare, still need to set codec specifics.
631 RTC_CHECK(params_.video[video_idx].codec == "VP9");
632 VideoCodecVP9 vp9_settings = VideoEncoder::GetDefaultVp9Settings();
633 vp9_settings.numberOfTemporalLayers = static_cast<unsigned char>(
634 params_.video[video_idx].num_temporal_layers);
635 vp9_settings.numberOfSpatialLayers =
636 static_cast<unsigned char>(params_.ss[video_idx].num_spatial_layers);
Sergey Silkin57027362018-05-15 09:12:05 +0200637 vp9_settings.interLayerPred = params_.ss[video_idx].inter_layer_pred;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100638 video_encoder_configs_[video_idx].encoder_specific_settings =
639 new rtc::RefCountedObject<
640 VideoEncoderConfig::Vp9EncoderSpecificSettings>(vp9_settings);
Niels Möller6aa415e2018-06-07 11:14:13 +0200641 } else if (params_.video[video_idx].automatic_scaling) {
642 if (params_.video[video_idx].codec == "VP8") {
643 VideoCodecVP8 vp8_settings = VideoEncoder::GetDefaultVp8Settings();
644 vp8_settings.automaticResizeOn = true;
645 video_encoder_configs_[video_idx].encoder_specific_settings =
646 new rtc::RefCountedObject<
647 VideoEncoderConfig::Vp8EncoderSpecificSettings>(vp8_settings);
648 } else if (params_.video[video_idx].codec == "VP9") {
649 VideoCodecVP9 vp9_settings = VideoEncoder::GetDefaultVp9Settings();
650 vp9_settings.automaticResizeOn = true;
651 video_encoder_configs_[video_idx].encoder_specific_settings =
652 new rtc::RefCountedObject<
653 VideoEncoderConfig::Vp9EncoderSpecificSettings>(vp9_settings);
654 } else {
655 RTC_NOTREACHED() << "Automatic scaling not supported for codec "
Yves Gerey665174f2018-06-19 15:03:05 +0200656 << params_.video[video_idx].codec << ", stream "
657 << video_idx;
Niels Möller6aa415e2018-06-07 11:14:13 +0200658 }
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100659 }
660 total_streams_used += num_video_substreams;
sprangce4aef12015-11-02 07:23:20 -0800661 }
brandtr1293aca2016-11-16 22:47:29 -0800662
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100663 // FEC supported only for single video stream mode yet.
664 if (params_.video[0].flexfec) {
sprang1168fd42017-06-21 09:00:17 -0700665 if (decode_all_receive_streams) {
Sebastian Jansson3bd2c792018-07-13 13:29:03 +0200666 SetSendFecConfig(GetVideoSendConfig()->rtp.ssrcs);
sprang1168fd42017-06-21 09:00:17 -0700667 } else {
Sebastian Jansson3bd2c792018-07-13 13:29:03 +0200668 SetSendFecConfig({kVideoSendSsrcs[params_.ss[0].selected_stream]});
sprang1168fd42017-06-21 09:00:17 -0700669 }
brandtr1293aca2016-11-16 22:47:29 -0800670
Sebastian Jansson3bd2c792018-07-13 13:29:03 +0200671 CreateMatchingFecConfig(recv_transport, *GetVideoSendConfig());
672 GetFlexFecConfig()->transport_cc = params_.call.send_side_bwe;
brandtrb29e6522016-12-21 06:37:18 -0800673 if (params_.call.send_side_bwe) {
Sebastian Jansson3bd2c792018-07-13 13:29:03 +0200674 GetFlexFecConfig()->rtp_header_extensions.push_back(
brandtrb29e6522016-12-21 06:37:18 -0800675 RtpExtension(RtpExtension::kTransportSequenceNumberUri,
676 test::kTransportSequenceNumberExtensionId));
677 } else {
Sebastian Jansson3bd2c792018-07-13 13:29:03 +0200678 GetFlexFecConfig()->rtp_header_extensions.push_back(RtpExtension(
brandtrb29e6522016-12-21 06:37:18 -0800679 RtpExtension::kAbsSendTimeUri, test::kAbsSendTimeExtensionId));
680 }
brandtr1293aca2016-11-16 22:47:29 -0800681 }
682
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100683 if (params_.video[0].ulpfec) {
Sebastian Jansson3bd2c792018-07-13 13:29:03 +0200684 SetSendUlpFecConfig(GetVideoSendConfig());
sprang1168fd42017-06-21 09:00:17 -0700685 if (decode_all_receive_streams) {
Sebastian Jansson3bd2c792018-07-13 13:29:03 +0200686 for (auto& receive_config : video_receive_configs_) {
687 SetReceiveUlpFecConfig(&receive_config);
sprang1168fd42017-06-21 09:00:17 -0700688 }
689 } else {
Sebastian Jansson3bd2c792018-07-13 13:29:03 +0200690 SetReceiveUlpFecConfig(
691 &video_receive_configs_[params_.ss[0].selected_stream]);
sprang1168fd42017-06-21 09:00:17 -0700692 }
brandtr1293aca2016-11-16 22:47:29 -0800693 }
ivica5d6a06c2015-09-17 05:30:24 -0700694}
695
ilnika014cc52017-03-07 04:21:04 -0800696void VideoQualityTest::SetupThumbnails(Transport* send_transport,
697 Transport* recv_transport) {
ilnik98436952017-07-13 00:47:03 -0700698 for (int i = 0; i < params_.call.num_thumbnails; ++i) {
ilnika014cc52017-03-07 04:21:04 -0800699 // Thumbnails will be send in the other way: from receiver_call to
700 // sender_call.
701 VideoSendStream::Config thumbnail_send_config(recv_transport);
702 thumbnail_send_config.rtp.ssrcs.push_back(kThumbnailSendSsrcStart + i);
Niels Möller4db138e2018-04-19 09:04:13 +0200703 // TODO(nisse): Could use a simpler VP8-only encoder factory.
704 thumbnail_send_config.encoder_settings.encoder_factory =
705 &video_encoder_factory_;
Niels Möller259a4972018-04-05 15:36:51 +0200706 thumbnail_send_config.rtp.payload_name = params_.video[0].codec;
707 thumbnail_send_config.rtp.payload_type = kPayloadTypeVP8;
ilnika014cc52017-03-07 04:21:04 -0800708 thumbnail_send_config.rtp.nack.rtp_history_ms = kNackRtpHistoryMs;
709 thumbnail_send_config.rtp.rtx.payload_type = kSendRtxPayloadType;
710 thumbnail_send_config.rtp.rtx.ssrcs.push_back(kThumbnailRtxSsrcStart + i);
711 thumbnail_send_config.rtp.extensions.clear();
712 if (params_.call.send_side_bwe) {
713 thumbnail_send_config.rtp.extensions.push_back(
714 RtpExtension(RtpExtension::kTransportSequenceNumberUri,
715 test::kTransportSequenceNumberExtensionId));
716 } else {
717 thumbnail_send_config.rtp.extensions.push_back(RtpExtension(
718 RtpExtension::kAbsSendTimeUri, test::kAbsSendTimeExtensionId));
719 }
720
721 VideoEncoderConfig thumbnail_encoder_config;
Niels Möller259a4972018-04-05 15:36:51 +0200722 thumbnail_encoder_config.codec_type = kVideoCodecVP8;
Niels Möller4db138e2018-04-19 09:04:13 +0200723 thumbnail_encoder_config.video_format.name = "VP8";
ilnika014cc52017-03-07 04:21:04 -0800724 thumbnail_encoder_config.min_transmit_bitrate_bps = 7500;
725 thumbnail_send_config.suspend_below_min_bitrate =
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100726 params_.video[0].suspend_below_min_bitrate;
ilnika014cc52017-03-07 04:21:04 -0800727 thumbnail_encoder_config.number_of_streams = 1;
728 thumbnail_encoder_config.max_bitrate_bps = 50000;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100729 if (params_.ss[0].infer_streams) {
ilnik6b826ef2017-06-16 06:53:48 -0700730 thumbnail_encoder_config.video_stream_factory =
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100731 new rtc::RefCountedObject<VideoStreamFactory>(params_.ss[0].streams);
ilnik6b826ef2017-06-16 06:53:48 -0700732 } else {
Seth Hampson8234ead2018-02-02 15:16:24 -0800733 thumbnail_encoder_config.simulcast_layers = std::vector<VideoStream>(1);
ilnik6b826ef2017-06-16 06:53:48 -0700734 thumbnail_encoder_config.video_stream_factory =
735 new rtc::RefCountedObject<cricket::EncoderStreamFactory>(
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100736 params_.video[0].codec, params_.ss[0].streams[0].max_qp,
Mirko Bonadei948b7e32018-08-14 07:23:21 +0000737 params_.video[0].fps, params_.screenshare[0].enabled, true);
ilnik6b826ef2017-06-16 06:53:48 -0700738 }
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100739 thumbnail_encoder_config.spatial_layers = params_.ss[0].spatial_layers;
ilnika014cc52017-03-07 04:21:04 -0800740
ilnika014cc52017-03-07 04:21:04 -0800741 thumbnail_encoder_configs_.push_back(thumbnail_encoder_config.Copy());
742 thumbnail_send_configs_.push_back(thumbnail_send_config.Copy());
ilnika014cc52017-03-07 04:21:04 -0800743
Sebastian Jansson3bd2c792018-07-13 13:29:03 +0200744 AddMatchingVideoReceiveConfigs(
745 &thumbnail_receive_configs_, thumbnail_send_config, send_transport,
746 params_.call.send_side_bwe, absl::nullopt, false, kNackRtpHistoryMs);
747 }
748 for (size_t i = 0; i < thumbnail_send_configs_.size(); ++i) {
ilnika014cc52017-03-07 04:21:04 -0800749 thumbnail_send_streams_.push_back(receiver_call_->CreateVideoSendStream(
750 thumbnail_send_configs_[i].Copy(),
751 thumbnail_encoder_configs_[i].Copy()));
Sebastian Jansson3bd2c792018-07-13 13:29:03 +0200752 }
753 for (size_t i = 0; i < thumbnail_receive_configs_.size(); ++i) {
ilnika014cc52017-03-07 04:21:04 -0800754 thumbnail_receive_streams_.push_back(sender_call_->CreateVideoReceiveStream(
755 thumbnail_receive_configs_[i].Copy()));
756 }
757}
758
759void VideoQualityTest::DestroyThumbnailStreams() {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100760 for (VideoSendStream* thumbnail_send_stream : thumbnail_send_streams_) {
ilnika014cc52017-03-07 04:21:04 -0800761 receiver_call_->DestroyVideoSendStream(thumbnail_send_stream);
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100762 }
ilnika014cc52017-03-07 04:21:04 -0800763 thumbnail_send_streams_.clear();
764 for (VideoReceiveStream* thumbnail_receive_stream :
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100765 thumbnail_receive_streams_) {
ilnika014cc52017-03-07 04:21:04 -0800766 sender_call_->DestroyVideoReceiveStream(thumbnail_receive_stream);
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100767 }
ilnika014cc52017-03-07 04:21:04 -0800768 thumbnail_send_streams_.clear();
769 thumbnail_receive_streams_.clear();
Sebastian Janssonf1f363f2018-08-13 14:24:58 +0200770 for (std::unique_ptr<test::TestVideoCapturer>& video_caputurer :
eladalon413ee9a2017-08-22 04:02:52 -0700771 thumbnail_capturers_) {
772 video_caputurer.reset();
773 }
ilnika014cc52017-03-07 04:21:04 -0800774}
775
ilnika014cc52017-03-07 04:21:04 -0800776void VideoQualityTest::SetupThumbnailCapturers(size_t num_thumbnail_streams) {
777 VideoStream thumbnail = DefaultThumbnailStream();
778 for (size_t i = 0; i < num_thumbnail_streams; ++i) {
779 thumbnail_capturers_.emplace_back(test::FrameGeneratorCapturer::Create(
780 static_cast<int>(thumbnail.width), static_cast<int>(thumbnail.height),
Danil Chapovalovb9b146c2018-06-15 12:28:07 +0200781 absl::nullopt, absl::nullopt, thumbnail.max_framerate, clock_));
ilnikf89a7382017-03-07 06:15:27 -0800782 RTC_DCHECK(thumbnail_capturers_.back());
ilnika014cc52017-03-07 04:21:04 -0800783 }
784}
785
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100786std::unique_ptr<test::FrameGenerator> VideoQualityTest::CreateFrameGenerator(
787 size_t video_idx) {
788 // Setup frame generator.
789 const size_t kWidth = 1850;
790 const size_t kHeight = 1110;
791 std::unique_ptr<test::FrameGenerator> frame_generator;
792 if (params_.screenshare[video_idx].generate_slides) {
793 frame_generator = test::FrameGenerator::CreateSlideGenerator(
794 kWidth, kHeight,
795 params_.screenshare[video_idx].slide_change_interval *
796 params_.video[video_idx].fps);
ivica5d6a06c2015-09-17 05:30:24 -0700797 } else {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100798 std::vector<std::string> slides = params_.screenshare[video_idx].slides;
799 if (slides.size() == 0) {
800 slides.push_back(test::ResourcePath("web_screenshot_1850_1110", "yuv"));
801 slides.push_back(test::ResourcePath("presentation_1850_1110", "yuv"));
802 slides.push_back(test::ResourcePath("photo_1850_1110", "yuv"));
803 slides.push_back(test::ResourcePath("difficult_photo_1850_1110", "yuv"));
804 }
805 if (params_.screenshare[video_idx].scroll_duration == 0) {
806 // Cycle image every slide_change_interval seconds.
807 frame_generator = test::FrameGenerator::CreateFromYuvFile(
808 slides, kWidth, kHeight,
809 params_.screenshare[video_idx].slide_change_interval *
810 params_.video[video_idx].fps);
ivica5d6a06c2015-09-17 05:30:24 -0700811 } else {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100812 RTC_CHECK_LE(params_.video[video_idx].width, kWidth);
813 RTC_CHECK_LE(params_.video[video_idx].height, kHeight);
814 RTC_CHECK_GT(params_.screenshare[video_idx].slide_change_interval, 0);
815 const int kPauseDurationMs =
816 (params_.screenshare[video_idx].slide_change_interval -
817 params_.screenshare[video_idx].scroll_duration) *
818 1000;
819 RTC_CHECK_LE(params_.screenshare[video_idx].scroll_duration,
820 params_.screenshare[video_idx].slide_change_interval);
821
822 frame_generator = test::FrameGenerator::CreateScrollingInputFromYuvFiles(
823 clock_, slides, kWidth, kHeight, params_.video[video_idx].width,
824 params_.video[video_idx].height,
825 params_.screenshare[video_idx].scroll_duration * 1000,
826 kPauseDurationMs);
ivica5d6a06c2015-09-17 05:30:24 -0700827 }
828 }
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100829 return frame_generator;
830}
831
832void VideoQualityTest::CreateCapturers() {
Sebastian Jansson3bd2c792018-07-13 13:29:03 +0200833 RTC_DCHECK(video_sources_.empty());
834 RTC_DCHECK(video_capturers_.empty());
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100835 video_capturers_.resize(num_video_streams_);
836 for (size_t video_idx = 0; video_idx < num_video_streams_; ++video_idx) {
837 if (params_.screenshare[video_idx].enabled) {
838 std::unique_ptr<test::FrameGenerator> frame_generator =
839 CreateFrameGenerator(video_idx);
840 test::FrameGeneratorCapturer* frame_generator_capturer =
841 new test::FrameGeneratorCapturer(clock_, std::move(frame_generator),
842 params_.video[video_idx].fps);
843 EXPECT_TRUE(frame_generator_capturer->Init());
844 video_capturers_[video_idx].reset(frame_generator_capturer);
845 } else {
846 if (params_.video[video_idx].clip_name == "Generator") {
847 video_capturers_[video_idx].reset(test::FrameGeneratorCapturer::Create(
848 static_cast<int>(params_.video[video_idx].width),
Danil Chapovalovb9b146c2018-06-15 12:28:07 +0200849 static_cast<int>(params_.video[video_idx].height), absl::nullopt,
850 absl::nullopt, params_.video[video_idx].fps, clock_));
Emircan Uysaler03e6ec92018-03-09 15:03:26 -0800851 } else if (params_.video[video_idx].clip_name == "GeneratorI420A") {
852 video_capturers_[video_idx].reset(test::FrameGeneratorCapturer::Create(
853 static_cast<int>(params_.video[video_idx].width),
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100854 static_cast<int>(params_.video[video_idx].height),
Danil Chapovalovb9b146c2018-06-15 12:28:07 +0200855 test::FrameGenerator::OutputType::I420A, absl::nullopt,
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100856 params_.video[video_idx].fps, clock_));
Emircan Uysaler0823eec2018-07-13 17:10:00 -0700857 } else if (params_.video[video_idx].clip_name == "GeneratorI010") {
858 video_capturers_[video_idx].reset(test::FrameGeneratorCapturer::Create(
859 static_cast<int>(params_.video[video_idx].width),
860 static_cast<int>(params_.video[video_idx].height),
861 test::FrameGenerator::OutputType::I010, absl::nullopt,
862 params_.video[video_idx].fps, clock_));
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100863 } else if (params_.video[video_idx].clip_name.empty()) {
864 video_capturers_[video_idx].reset(test::VcmCapturer::Create(
865 params_.video[video_idx].width, params_.video[video_idx].height,
866 params_.video[video_idx].fps,
867 params_.video[video_idx].capture_device_index));
868 if (!video_capturers_[video_idx]) {
869 // Failed to get actual camera, use chroma generator as backup.
870 video_capturers_[video_idx].reset(
871 test::FrameGeneratorCapturer::Create(
872 static_cast<int>(params_.video[video_idx].width),
873 static_cast<int>(params_.video[video_idx].height),
Danil Chapovalovb9b146c2018-06-15 12:28:07 +0200874 absl::nullopt, absl::nullopt, params_.video[video_idx].fps,
Emircan Uysaler03e6ec92018-03-09 15:03:26 -0800875 clock_));
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100876 }
877 } else {
878 video_capturers_[video_idx].reset(
879 test::FrameGeneratorCapturer::CreateFromYuvFile(
880 test::ResourcePath(params_.video[video_idx].clip_name, "yuv"),
881 params_.video[video_idx].width, params_.video[video_idx].height,
882 params_.video[video_idx].fps, clock_));
883 ASSERT_TRUE(video_capturers_[video_idx])
884 << "Could not create capturer for "
885 << params_.video[video_idx].clip_name
886 << ".yuv. Is this resource file present?";
887 }
888 }
889 RTC_DCHECK(video_capturers_[video_idx].get());
Sebastian Jansson3bd2c792018-07-13 13:29:03 +0200890 video_sources_.push_back(video_capturers_[video_idx].get());
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100891 }
ivica5d6a06c2015-09-17 05:30:24 -0700892}
893
Christoffer Rodbro39a44b22018-08-07 17:07:24 +0200894void VideoQualityTest::StartAudioStreams() {
895 audio_send_stream_->Start();
896 for (AudioReceiveStream* audio_recv_stream : audio_receive_streams_)
897 audio_recv_stream->Start();
898}
899
900void VideoQualityTest::StartThumbnailCapture() {
Sebastian Janssonf1f363f2018-08-13 14:24:58 +0200901 for (std::unique_ptr<test::TestVideoCapturer>& capturer :
902 thumbnail_capturers_)
Christoffer Rodbro39a44b22018-08-07 17:07:24 +0200903 capturer->Start();
904}
905
906void VideoQualityTest::StopThumbnailCapture() {
Sebastian Janssonf1f363f2018-08-13 14:24:58 +0200907 for (std::unique_ptr<test::TestVideoCapturer>& capturer :
908 thumbnail_capturers_)
Christoffer Rodbro39a44b22018-08-07 17:07:24 +0200909 capturer->Stop();
910}
911
Christoffer Rodbroc2a02882018-08-07 14:10:56 +0200912void VideoQualityTest::StartThumbnails() {
913 for (VideoSendStream* send_stream : thumbnail_send_streams_)
914 send_stream->Start();
915 for (VideoReceiveStream* receive_stream : thumbnail_receive_streams_)
916 receive_stream->Start();
Christoffer Rodbroc2a02882018-08-07 14:10:56 +0200917}
918
919void VideoQualityTest::StopThumbnails() {
Christoffer Rodbroc2a02882018-08-07 14:10:56 +0200920 for (VideoReceiveStream* receive_stream : thumbnail_receive_streams_)
921 receive_stream->Stop();
922 for (VideoSendStream* send_stream : thumbnail_send_streams_)
923 send_stream->Stop();
924}
925
Christoffer Rodbrob4bb4eb2017-11-13 13:03:52 +0100926std::unique_ptr<test::LayerFilteringTransport>
927VideoQualityTest::CreateSendTransport() {
Artem Titove269cb42018-08-29 09:59:23 +0200928 std::unique_ptr<NetworkSimulationInterface> simulated_network = nullptr;
929 if (injection_components_->sender_network == nullptr) {
930 simulated_network = absl::make_unique<SimulatedNetwork>(*params_.config);
931 } else {
932 simulated_network = std::move(injection_components_->sender_network);
933 }
Karl Wiberg918f50c2018-07-05 11:40:33 +0200934 return absl::make_unique<test::LayerFilteringTransport>(
Artem Titov4e199e92018-08-20 13:30:39 +0200935 &task_queue_,
Artem Titove269cb42018-08-29 09:59:23 +0200936 absl::make_unique<FakeNetworkPipe>(Clock::GetRealTimeClock(),
937 std::move(simulated_network)),
Artem Titov4e199e92018-08-20 13:30:39 +0200938 sender_call_.get(), kPayloadTypeVP8, kPayloadTypeVP9,
939 params_.video[0].selected_tl, params_.ss[0].selected_sl,
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100940 payload_type_map_, kVideoSendSsrcs[0],
941 static_cast<uint32_t>(kVideoSendSsrcs[0] + params_.ss[0].streams.size() -
942 1));
Christoffer Rodbrob4bb4eb2017-11-13 13:03:52 +0100943}
944
945std::unique_ptr<test::DirectTransport>
946VideoQualityTest::CreateReceiveTransport() {
Artem Titove269cb42018-08-29 09:59:23 +0200947 std::unique_ptr<NetworkSimulationInterface> simulated_network = nullptr;
948 if (injection_components_->receiver_network == nullptr) {
949 simulated_network = absl::make_unique<SimulatedNetwork>(*params_.config);
950 } else {
951 simulated_network = std::move(injection_components_->receiver_network);
952 }
Karl Wiberg918f50c2018-07-05 11:40:33 +0200953 return absl::make_unique<test::DirectTransport>(
Artem Titovdd2eebe2018-08-20 13:27:45 +0200954 &task_queue_,
Artem Titove269cb42018-08-29 09:59:23 +0200955 absl::make_unique<FakeNetworkPipe>(Clock::GetRealTimeClock(),
956 std::move(simulated_network)),
Artem Titovdd2eebe2018-08-20 13:27:45 +0200957 receiver_call_.get(), payload_type_map_);
Christoffer Rodbrob4bb4eb2017-11-13 13:03:52 +0100958}
959
sprang7a975f72015-10-12 06:33:21 -0700960void VideoQualityTest::RunWithAnalyzer(const Params& params) {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100961 num_video_streams_ = params.call.dual_video ? 2 : 1;
eladalon413ee9a2017-08-22 04:02:52 -0700962 std::unique_ptr<test::LayerFilteringTransport> send_transport;
963 std::unique_ptr<test::DirectTransport> recv_transport;
964 FILE* graph_data_output_file = nullptr;
965 std::unique_ptr<VideoAnalyzer> analyzer;
966
sprangce4aef12015-11-02 07:23:20 -0800967 params_ = params;
ivica5d6a06c2015-09-17 05:30:24 -0700968 // TODO(ivica): Merge with RunWithRenderer and use a flag / argument to
969 // differentiate between the analyzer and the renderer case.
Artem Titove269cb42018-08-29 09:59:23 +0200970 CheckParamsAndInjectionComponents();
ivica5d6a06c2015-09-17 05:30:24 -0700971
sprangce4aef12015-11-02 07:23:20 -0800972 if (!params_.analyzer.graph_data_output_filename.empty()) {
ivica5d6a06c2015-09-17 05:30:24 -0700973 graph_data_output_file =
sprangce4aef12015-11-02 07:23:20 -0800974 fopen(params_.analyzer.graph_data_output_filename.c_str(), "w");
Peter Boström74f6e9e2016-04-04 17:56:10 +0200975 RTC_CHECK(graph_data_output_file)
sprangce4aef12015-11-02 07:23:20 -0800976 << "Can't open the file " << params_.analyzer.graph_data_output_filename
977 << "!";
ivica87f83a92015-10-08 05:13:32 -0700978 }
sprang7a975f72015-10-12 06:33:21 -0700979
ilnik98436952017-07-13 00:47:03 -0700980 if (!params.logging.rtc_event_log_name.empty()) {
Bjorn Terelius0a6a2b72018-01-19 17:52:38 +0100981 send_event_log_ = RtcEventLog::Create(RtcEventLog::EncodingType::Legacy);
982 recv_event_log_ = RtcEventLog::Create(RtcEventLog::EncodingType::Legacy);
Ilya Nikolaevskiya4259f62017-12-05 13:19:45 +0100983 std::unique_ptr<RtcEventLogOutputFile> send_output(
Karl Wiberg918f50c2018-07-05 11:40:33 +0200984 absl::make_unique<RtcEventLogOutputFile>(
Ilya Nikolaevskiya4259f62017-12-05 13:19:45 +0100985 params.logging.rtc_event_log_name + "_send",
986 RtcEventLog::kUnlimitedOutput));
987 std::unique_ptr<RtcEventLogOutputFile> recv_output(
Karl Wiberg918f50c2018-07-05 11:40:33 +0200988 absl::make_unique<RtcEventLogOutputFile>(
Ilya Nikolaevskiya4259f62017-12-05 13:19:45 +0100989 params.logging.rtc_event_log_name + "_recv",
990 RtcEventLog::kUnlimitedOutput));
991 bool event_log_started =
992 send_event_log_->StartLogging(std::move(send_output),
993 RtcEventLog::kImmediateOutput) &&
994 recv_event_log_->StartLogging(std::move(recv_output),
995 RtcEventLog::kImmediateOutput);
ilnik98436952017-07-13 00:47:03 -0700996 RTC_DCHECK(event_log_started);
Ilya Nikolaevskiya4259f62017-12-05 13:19:45 +0100997 } else {
998 send_event_log_ = RtcEventLog::CreateNull();
999 recv_event_log_ = RtcEventLog::CreateNull();
ilnik98436952017-07-13 00:47:03 -07001000 }
1001
Ilya Nikolaevskiya4259f62017-12-05 13:19:45 +01001002 Call::Config send_call_config(send_event_log_.get());
1003 Call::Config recv_call_config(recv_event_log_.get());
1004 send_call_config.bitrate_config = params.call.call_bitrate_config;
1005 recv_call_config.bitrate_config = params.call.call_bitrate_config;
stefanf116bd02015-10-27 08:29:42 -07001006
Ilya Nikolaevskiya4259f62017-12-05 13:19:45 +01001007 task_queue_.SendTask([this, &send_call_config, &recv_call_config,
1008 &send_transport, &recv_transport]() {
Christoffer Rodbro39a44b22018-08-07 17:07:24 +02001009 if (params_.audio.enabled)
henrika255750b2018-08-27 16:13:37 +02001010 InitializeAudioDevice(
1011 &send_call_config, &recv_call_config, params_.audio.use_real_adm);
Christoffer Rodbroc2a02882018-08-07 14:10:56 +02001012
Ilya Nikolaevskiya4259f62017-12-05 13:19:45 +01001013 CreateCalls(send_call_config, recv_call_config);
1014 send_transport = CreateSendTransport();
1015 recv_transport = CreateReceiveTransport();
1016 });
stefanf116bd02015-10-27 08:29:42 -07001017
sprangce4aef12015-11-02 07:23:20 -08001018 std::string graph_title = params_.analyzer.graph_title;
1019 if (graph_title.empty())
1020 graph_title = VideoQualityTest::GenerateGraphTitle();
sprangc1b57a12017-02-28 08:50:47 -08001021 bool is_quick_test_enabled = field_trial::IsEnabled("WebRTC-QuickPerfTest");
Karl Wiberg918f50c2018-07-05 11:40:33 +02001022 analyzer = absl::make_unique<VideoAnalyzer>(
eladalon413ee9a2017-08-22 04:02:52 -07001023 send_transport.get(), params_.analyzer.test_label,
ilnik2a8c2f52017-02-15 02:23:28 -08001024 params_.analyzer.avg_psnr_threshold, params_.analyzer.avg_ssim_threshold,
ilnik9ae0d762017-02-15 00:53:12 -08001025 is_quick_test_enabled
1026 ? kFramesSentInQuickTest
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001027 : params_.analyzer.test_durations_secs * params_.video[0].fps,
sprangce4aef12015-11-02 07:23:20 -08001028 graph_data_output_file, graph_title,
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001029 kVideoSendSsrcs[params_.ss[0].selected_stream],
1030 kSendRtxSsrcs[params_.ss[0].selected_stream],
1031 static_cast<size_t>(params_.ss[0].selected_stream),
1032 params.ss[0].selected_sl, params_.video[0].selected_tl,
1033 is_quick_test_enabled, clock_, params_.logging.rtp_dump_name);
ivica5d6a06c2015-09-17 05:30:24 -07001034
eladalon413ee9a2017-08-22 04:02:52 -07001035 task_queue_.SendTask([&]() {
1036 analyzer->SetCall(sender_call_.get());
1037 analyzer->SetReceiver(receiver_call_->Receiver());
1038 send_transport->SetReceiver(analyzer.get());
1039 recv_transport->SetReceiver(sender_call_->Receiver());
ivica5d6a06c2015-09-17 05:30:24 -07001040
eladalon413ee9a2017-08-22 04:02:52 -07001041 SetupVideo(analyzer.get(), recv_transport.get());
1042 SetupThumbnails(analyzer.get(), recv_transport.get());
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001043 video_receive_configs_[params_.ss[0].selected_stream].renderer =
eladalon413ee9a2017-08-22 04:02:52 -07001044 analyzer.get();
Sebastian Jansson3bd2c792018-07-13 13:29:03 +02001045 GetVideoSendConfig()->pre_encode_callback = analyzer->pre_encode_proxy();
1046 RTC_DCHECK(!GetVideoSendConfig()->post_encode_callback);
1047 GetVideoSendConfig()->post_encode_callback = analyzer.get();
kthelgason2bc68642017-02-07 07:02:22 -08001048
eladalon413ee9a2017-08-22 04:02:52 -07001049 CreateFlexfecStreams();
1050 CreateVideoStreams();
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001051 analyzer->SetSendStream(video_send_streams_[0]);
eladalon413ee9a2017-08-22 04:02:52 -07001052 if (video_receive_streams_.size() == 1)
1053 analyzer->SetReceiveStream(video_receive_streams_[0]);
ivica5d6a06c2015-09-17 05:30:24 -07001054
Sebastian Jansson3bd2c792018-07-13 13:29:03 +02001055 GetVideoSendStream()->SetSource(analyzer->OutputInterface(),
1056 degradation_preference_);
eladalon413ee9a2017-08-22 04:02:52 -07001057 SetupThumbnailCapturers(params_.call.num_thumbnails);
1058 for (size_t i = 0; i < thumbnail_send_streams_.size(); ++i) {
1059 thumbnail_send_streams_[i]->SetSource(thumbnail_capturers_[i].get(),
1060 degradation_preference_);
1061 }
ilnika014cc52017-03-07 04:21:04 -08001062
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001063 CreateCapturers();
ivicac1cc8542015-10-08 03:44:06 -07001064
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001065 analyzer->SetSource(video_capturers_[0].get(), params_.ss[0].infer_streams);
ilnika014cc52017-03-07 04:21:04 -08001066
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001067 for (size_t video_idx = 1; video_idx < num_video_streams_; ++video_idx) {
1068 video_send_streams_[video_idx]->SetSource(
1069 video_capturers_[video_idx].get(), degradation_preference_);
1070 }
1071
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001072 StartEncodedFrameLogs(
1073 video_receive_streams_[params_.ss[0].selected_stream]);
ilnika014cc52017-03-07 04:21:04 -08001074
Christoffer Rodbroc2a02882018-08-07 14:10:56 +02001075 if (params_.audio.enabled) {
1076 SetupAudio(send_transport.get());
Christoffer Rodbro39a44b22018-08-07 17:07:24 +02001077 StartAudioStreams();
Christoffer Rodbroc2a02882018-08-07 14:10:56 +02001078 analyzer->SetAudioReceiveStream(audio_receive_streams_[0]);
1079 }
Christoffer Rodbro39a44b22018-08-07 17:07:24 +02001080 StartVideoStreams();
1081 StartThumbnails();
Christoffer Rodbroc2a02882018-08-07 14:10:56 +02001082 analyzer->StartMeasuringCpuProcessTime();
Christoffer Rodbro39a44b22018-08-07 17:07:24 +02001083 StartVideoCapture();
1084 StartThumbnailCapture();
eladalon413ee9a2017-08-22 04:02:52 -07001085 });
ivica5d6a06c2015-09-17 05:30:24 -07001086
eladalon413ee9a2017-08-22 04:02:52 -07001087 analyzer->Wait();
ivica5d6a06c2015-09-17 05:30:24 -07001088
eladalon413ee9a2017-08-22 04:02:52 -07001089 task_queue_.SendTask([&]() {
Christoffer Rodbro39a44b22018-08-07 17:07:24 +02001090 StopThumbnailCapture();
Christoffer Rodbroc2a02882018-08-07 14:10:56 +02001091 StopThumbnails();
1092 Stop();
ivica5d6a06c2015-09-17 05:30:24 -07001093
eladalon413ee9a2017-08-22 04:02:52 -07001094 DestroyStreams();
1095 DestroyThumbnailStreams();
ivica5d6a06c2015-09-17 05:30:24 -07001096
eladalon413ee9a2017-08-22 04:02:52 -07001097 if (graph_data_output_file)
1098 fclose(graph_data_output_file);
1099
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001100 video_capturers_.clear();
eladalon413ee9a2017-08-22 04:02:52 -07001101 send_transport.reset();
1102 recv_transport.reset();
1103
1104 DestroyCalls();
1105 });
ivica5d6a06c2015-09-17 05:30:24 -07001106}
1107
henrika255750b2018-08-27 16:13:37 +02001108rtc::scoped_refptr<AudioDeviceModule> VideoQualityTest::CreateAudioDevice() {
1109#ifdef WEBRTC_WIN
1110 RTC_LOG(INFO) << "Using latest version of ADM on Windows";
1111 // We must initialize the COM library on a thread before we calling any of
1112 // the library functions. All COM functions in the ADM will return
1113 // CO_E_NOTINITIALIZED otherwise. The legacy ADM for Windows used internal
1114 // COM initialization but the new ADM requires COM to be initialized
1115 // externally.
1116 com_initializer_ = absl::make_unique<webrtc_win::ScopedCOMInitializer>(
1117 webrtc_win::ScopedCOMInitializer::kMTA);
1118 RTC_CHECK(com_initializer_->Succeeded());
1119 RTC_CHECK(webrtc_win::core_audio_utility::IsSupported());
1120 RTC_CHECK(webrtc_win::core_audio_utility::IsMMCSSSupported());
1121 return CreateWindowsCoreAudioAudioDeviceModule();
1122#else
1123 // Use legacy factory method on all platforms except Windows.
1124 return AudioDeviceModule::Create(AudioDeviceModule::kPlatformDefaultAudio);
1125#endif
1126}
1127
Christoffer Rodbroc2a02882018-08-07 14:10:56 +02001128void VideoQualityTest::InitializeAudioDevice(Call::Config* send_call_config,
henrika255750b2018-08-27 16:13:37 +02001129 Call::Config* recv_call_config,
1130 bool use_real_adm) {
1131 rtc::scoped_refptr<AudioDeviceModule> audio_device;
1132 if (use_real_adm) {
1133 // Run test with real ADM (using default audio devices) if user has
1134 // explicitly set the --audio and --use_real_adm command-line flags.
1135 audio_device = CreateAudioDevice();
1136 } else {
1137 // By default, create a test ADM which fakes audio.
1138 audio_device = TestAudioDeviceModule::CreateTestAudioDeviceModule(
1139 TestAudioDeviceModule::CreatePulsedNoiseCapturer(32000, 48000),
1140 TestAudioDeviceModule::CreateDiscardRenderer(48000), 1.f);
1141 }
1142 RTC_CHECK(audio_device);
Christoffer Rodbroc2a02882018-08-07 14:10:56 +02001143
1144 AudioState::Config audio_state_config;
1145 audio_state_config.audio_mixer = AudioMixerImpl::Create();
1146 audio_state_config.audio_processing = AudioProcessingBuilder().Create();
henrika255750b2018-08-27 16:13:37 +02001147 audio_state_config.audio_device_module = audio_device;
Christoffer Rodbroc2a02882018-08-07 14:10:56 +02001148 send_call_config->audio_state = AudioState::Create(audio_state_config);
Christoffer Rodbroc2a02882018-08-07 14:10:56 +02001149 recv_call_config->audio_state = AudioState::Create(audio_state_config);
henrika255750b2018-08-27 16:13:37 +02001150 if (use_real_adm) {
1151 // The real ADM requires extra initialization: setting default devices,
1152 // setting up number of channels etc. Helper class also calls
1153 // AudioDeviceModule::Init().
1154 webrtc::adm_helpers::Init(audio_device.get());
1155 } else {
1156 audio_device->Init();
1157 }
1158 // Always initialize the ADM before injecting a valid audio transport.
1159 RTC_CHECK(audio_device->RegisterAudioCallback(
1160 send_call_config->audio_state->audio_transport()) == 0);
Christoffer Rodbroc2a02882018-08-07 14:10:56 +02001161}
1162
Sebastian Jansson3bd2c792018-07-13 13:29:03 +02001163void VideoQualityTest::SetupAudio(Transport* transport) {
1164 AudioSendStream::Config audio_send_config(transport);
1165 audio_send_config.rtp.ssrc = kAudioSendSsrc;
minyuea27172d2016-11-01 05:59:29 -07001166
1167 // Add extension to enable audio send side BWE, and allow audio bit rate
1168 // adaptation.
Sebastian Jansson3bd2c792018-07-13 13:29:03 +02001169 audio_send_config.rtp.extensions.clear();
1170 audio_send_config.send_codec_spec = AudioSendStream::Config::SendCodecSpec(
Oskar Sundbom8e07c132018-01-08 16:45:42 +01001171 kAudioSendPayloadType,
Yves Gerey665174f2018-06-19 15:03:05 +02001172 {"OPUS",
1173 48000,
1174 2,
1175 {{"usedtx", (params_.audio.dtx ? "1" : "0")}, {"stereo", "1"}}});
minyuea27172d2016-11-01 05:59:29 -07001176
Sebastian Jansson3bd2c792018-07-13 13:29:03 +02001177 if (params_.call.send_side_bwe) {
1178 audio_send_config.rtp.extensions.push_back(
1179 webrtc::RtpExtension(webrtc::RtpExtension::kTransportSequenceNumberUri,
1180 test::kTransportSequenceNumberExtensionId));
1181 audio_send_config.min_bitrate_bps = kOpusMinBitrateBps;
1182 audio_send_config.max_bitrate_bps = kOpusBitrateFbBps;
1183 audio_send_config.send_codec_spec->transport_cc_enabled = true;
1184 }
1185 audio_send_config.encoder_factory = audio_encoder_factory_;
1186 SetAudioConfig(audio_send_config);
1187
Christoffer Rodbroc2a02882018-08-07 14:10:56 +02001188 std::string sync_group;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001189 if (params_.video[0].enabled && params_.audio.sync_video)
Sebastian Jansson3bd2c792018-07-13 13:29:03 +02001190 sync_group = kSyncGroup;
minyuea27172d2016-11-01 05:59:29 -07001191
Sebastian Jansson3bd2c792018-07-13 13:29:03 +02001192 CreateMatchingAudioConfigs(transport, sync_group);
1193 CreateAudioStreams();
minyuea27172d2016-11-01 05:59:29 -07001194}
1195
minyue73208662016-08-18 06:28:55 -07001196void VideoQualityTest::RunWithRenderers(const Params& params) {
henrika255750b2018-08-27 16:13:37 +02001197 RTC_LOG(INFO) << __FUNCTION__;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001198 num_video_streams_ = params.call.dual_video ? 2 : 1;
eladalon413ee9a2017-08-22 04:02:52 -07001199 std::unique_ptr<test::LayerFilteringTransport> send_transport;
1200 std::unique_ptr<test::DirectTransport> recv_transport;
minyuea27172d2016-11-01 05:59:29 -07001201 std::unique_ptr<test::VideoRenderer> local_preview;
eladalon413ee9a2017-08-22 04:02:52 -07001202 std::vector<std::unique_ptr<test::VideoRenderer>> loopback_renderers;
Sebastian Jansson8e6602f2018-07-13 10:43:20 +02001203 RtcEventLogNullImpl null_event_log;
minyue73208662016-08-18 06:28:55 -07001204
eladalon413ee9a2017-08-22 04:02:52 -07001205 task_queue_.SendTask([&]() {
1206 params_ = params;
Artem Titove269cb42018-08-29 09:59:23 +02001207 CheckParamsAndInjectionComponents();
palmkviste75f2042016-09-28 06:19:48 -07001208
eladalon413ee9a2017-08-22 04:02:52 -07001209 // TODO(ivica): Remove bitrate_config and use the default Call::Config(), to
1210 // match the full stack tests.
Christoffer Rodbro39a44b22018-08-07 17:07:24 +02001211 Call::Config send_call_config(&null_event_log);
1212 send_call_config.bitrate_config = params_.call.call_bitrate_config;
Christoffer Rodbroc2a02882018-08-07 14:10:56 +02001213 Call::Config recv_call_config(&null_event_log);
ivica5d6a06c2015-09-17 05:30:24 -07001214
Christoffer Rodbro39a44b22018-08-07 17:07:24 +02001215 if (params_.audio.enabled)
henrika255750b2018-08-27 16:13:37 +02001216 InitializeAudioDevice(
1217 &send_call_config, &recv_call_config, params_.audio.use_real_adm);
minyue73208662016-08-18 06:28:55 -07001218
Christoffer Rodbro39a44b22018-08-07 17:07:24 +02001219 CreateCalls(send_call_config, recv_call_config);
eladalon413ee9a2017-08-22 04:02:52 -07001220
1221 // TODO(minyue): consider if this is a good transport even for audio only
1222 // calls.
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001223 send_transport = CreateSendTransport();
eladalon413ee9a2017-08-22 04:02:52 -07001224
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001225 recv_transport = CreateReceiveTransport();
eladalon413ee9a2017-08-22 04:02:52 -07001226
1227 // TODO(ivica): Use two calls to be able to merge with RunWithAnalyzer or at
1228 // least share as much code as possible. That way this test would also match
1229 // the full stack tests better.
1230 send_transport->SetReceiver(receiver_call_->Receiver());
1231 recv_transport->SetReceiver(sender_call_->Receiver());
1232
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001233 if (params_.video[0].enabled) {
eladalon413ee9a2017-08-22 04:02:52 -07001234 // Create video renderers.
1235 local_preview.reset(test::VideoRenderer::Create(
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001236 "Local Preview", params_.video[0].width, params_.video[0].height));
eladalon413ee9a2017-08-22 04:02:52 -07001237
1238 SetupVideo(send_transport.get(), recv_transport.get());
Sebastian Jansson3bd2c792018-07-13 13:29:03 +02001239 GetVideoSendConfig()->pre_encode_callback = local_preview.get();
eladalon413ee9a2017-08-22 04:02:52 -07001240
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001241 size_t num_streams_processed = 0;
1242 for (size_t video_idx = 0; video_idx < num_video_streams_; ++video_idx) {
1243 const size_t selected_stream_id = params_.ss[video_idx].selected_stream;
1244 const size_t num_streams = params_.ss[video_idx].streams.size();
1245 if (selected_stream_id == num_streams) {
1246 for (size_t stream_id = 0; stream_id < num_streams; ++stream_id) {
1247 std::ostringstream oss;
1248 oss << "Loopback Video #" << video_idx << " - Stream #"
1249 << static_cast<int>(stream_id);
1250 loopback_renderers.emplace_back(test::VideoRenderer::Create(
1251 oss.str().c_str(),
1252 params_.ss[video_idx].streams[stream_id].width,
1253 params_.ss[video_idx].streams[stream_id].height));
1254 video_receive_configs_[stream_id + num_streams_processed].renderer =
1255 loopback_renderers.back().get();
1256 if (params_.audio.enabled && params_.audio.sync_video)
1257 video_receive_configs_[stream_id + num_streams_processed]
1258 .sync_group = kSyncGroup;
1259 }
1260 } else {
1261 std::ostringstream oss;
1262 oss << "Loopback Video #" << video_idx;
1263 loopback_renderers.emplace_back(test::VideoRenderer::Create(
1264 oss.str().c_str(),
1265 params_.ss[video_idx].streams[selected_stream_id].width,
1266 params_.ss[video_idx].streams[selected_stream_id].height));
1267 video_receive_configs_[selected_stream_id + num_streams_processed]
1268 .renderer = loopback_renderers.back().get();
eladalon413ee9a2017-08-22 04:02:52 -07001269 if (params_.audio.enabled && params_.audio.sync_video)
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001270 video_receive_configs_[num_streams_processed + selected_stream_id]
1271 .sync_group = kSyncGroup;
eladalon413ee9a2017-08-22 04:02:52 -07001272 }
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001273 num_streams_processed += num_streams;
eladalon413ee9a2017-08-22 04:02:52 -07001274 }
eladalon413ee9a2017-08-22 04:02:52 -07001275 CreateFlexfecStreams();
1276 CreateVideoStreams();
1277
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001278 CreateCapturers();
Sebastian Jansson3bd2c792018-07-13 13:29:03 +02001279 ConnectVideoSourcesToStreams();
eladalon413ee9a2017-08-22 04:02:52 -07001280 }
1281
1282 if (params_.audio.enabled) {
Sebastian Jansson3bd2c792018-07-13 13:29:03 +02001283 SetupAudio(send_transport.get());
eladalon413ee9a2017-08-22 04:02:52 -07001284 }
1285
1286 for (VideoReceiveStream* receive_stream : video_receive_streams_)
1287 StartEncodedFrameLogs(receive_stream);
Sebastian Jansson3bd2c792018-07-13 13:29:03 +02001288 Start();
eladalon413ee9a2017-08-22 04:02:52 -07001289 });
minyue73208662016-08-18 06:28:55 -07001290
ivica5d6a06c2015-09-17 05:30:24 -07001291 test::PressEnterToContinue();
1292
eladalon413ee9a2017-08-22 04:02:52 -07001293 task_queue_.SendTask([&]() {
Sebastian Jansson3bd2c792018-07-13 13:29:03 +02001294 Stop();
1295 DestroyStreams();
minyue73208662016-08-18 06:28:55 -07001296
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001297 video_capturers_.clear();
eladalon413ee9a2017-08-22 04:02:52 -07001298 send_transport.reset();
1299 recv_transport.reset();
sprang1168fd42017-06-21 09:00:17 -07001300
eladalon413ee9a2017-08-22 04:02:52 -07001301 local_preview.reset();
1302 loopback_renderers.clear();
1303
1304 DestroyCalls();
1305 });
ivica5d6a06c2015-09-17 05:30:24 -07001306}
1307
palmkviste75f2042016-09-28 06:19:48 -07001308void VideoQualityTest::StartEncodedFrameLogs(VideoReceiveStream* stream) {
ilnik98436952017-07-13 00:47:03 -07001309 if (!params_.logging.encoded_frame_base_path.empty()) {
palmkviste75f2042016-09-28 06:19:48 -07001310 std::ostringstream str;
1311 str << receive_logs_++;
1312 std::string path =
ilnik98436952017-07-13 00:47:03 -07001313 params_.logging.encoded_frame_base_path + "." + str.str() + ".recv.ivf";
palmkviste75f2042016-09-28 06:19:48 -07001314 stream->EnableEncodedFrameRecording(rtc::CreatePlatformFile(path),
ilnik98436952017-07-13 00:47:03 -07001315 100000000);
palmkviste75f2042016-09-28 06:19:48 -07001316 }
1317}
ivica5d6a06c2015-09-17 05:30:24 -07001318} // namespace webrtc