blob: a7e0867fd6a6f590376018505ae31c4dadabd498 [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
Elad Alon83ccca12017-10-04 13:18:26 +020020#include "logging/rtc_event_log/output/rtc_event_log_output_file.h"
Magnus Jedvert46a27652017-11-13 14:10:02 +010021#include "media/engine/internalencoderfactory.h"
Niels Möller19a68d42018-04-13 13:37:52 +020022#include "media/engine/vp8_encoder_simulcast_proxy.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020023#include "media/engine/webrtcvideoengine.h"
24#include "modules/audio_mixer/audio_mixer_impl.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020025#include "modules/rtp_rtcp/source/rtp_format.h"
26#include "modules/rtp_rtcp/source/rtp_utility.h"
27#include "modules/video_coding/codecs/h264/include/h264.h"
Emircan Uysaler03e6ec92018-03-09 15:03:26 -080028#include "modules/video_coding/codecs/multiplex/include/multiplex_encoder_adapter.h"
Niels Möller4db138e2018-04-19 09:04:13 +020029#include "modules/video_coding/codecs/vp8/include/vp8.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020030#include "modules/video_coding/codecs/vp9/include/vp9.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020031#include "rtc_base/cpu_time.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020032#include "rtc_base/flags.h"
33#include "rtc_base/format_macros.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020034#include "rtc_base/memory_usage.h"
35#include "rtc_base/pathutils.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020036#include "system_wrappers/include/cpu_info.h"
Mirko Bonadei5f99e652017-12-19 13:40:40 +010037#include "test/rtp_file_writer.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020038#include "test/run_loop.h"
39#include "test/statistics.h"
40#include "test/testsupport/fileutils.h"
41#include "test/testsupport/frame_writer.h"
Edward Lemurd1d67432018-01-02 11:54:37 +010042#include "test/testsupport/perf_test.h"
Edward Lemuraf8659a2017-09-27 14:46:24 +020043#include "test/testsupport/test_artifacts.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020044#include "test/vcm_capturer.h"
45#include "test/video_renderer.h"
minyue73208662016-08-18 06:28:55 -070046
ilnikee42d192017-08-22 07:16:20 -070047DEFINE_bool(save_worst_frame,
48 false,
49 "Enable saving a frame with the lowest PSNR to a jpeg file in the "
Edward Lemuraf8659a2017-09-27 14:46:24 +020050 "test_artifacts_dir");
ilnikee42d192017-08-22 07:16:20 -070051
minyue73208662016-08-18 06:28:55 -070052namespace {
53
54constexpr int kSendStatsPollingIntervalMs = 1000;
minyue20c84cc2017-04-10 16:57:57 -070055
minyue73208662016-08-18 06:28:55 -070056constexpr size_t kMaxComparisons = 10;
57constexpr char kSyncGroup[] = "av_sync";
minyue10cbb462016-11-07 09:29:22 -080058constexpr int kOpusMinBitrateBps = 6000;
59constexpr int kOpusBitrateFbBps = 32000;
ilnik9ae0d762017-02-15 00:53:12 -080060constexpr int kFramesSentInQuickTest = 1;
ilnika014cc52017-03-07 04:21:04 -080061constexpr uint32_t kThumbnailSendSsrcStart = 0xE0000;
62constexpr uint32_t kThumbnailRtxSsrcStart = 0xF0000;
minyue73208662016-08-18 06:28:55 -070063
sprang1168fd42017-06-21 09:00:17 -070064constexpr int kDefaultMaxQp = cricket::WebRtcVideoChannel::kDefaultQpMax;
65
perkjfa10b552016-10-02 23:45:26 -070066class VideoStreamFactory
67 : public webrtc::VideoEncoderConfig::VideoStreamFactoryInterface {
68 public:
69 explicit VideoStreamFactory(const std::vector<webrtc::VideoStream>& streams)
70 : streams_(streams) {}
71
72 private:
73 std::vector<webrtc::VideoStream> CreateEncoderStreams(
74 int width,
75 int height,
76 const webrtc::VideoEncoderConfig& encoder_config) override {
mflodmand79f97b2016-12-15 07:24:33 -080077 // The highest layer must match the incoming resolution.
78 std::vector<webrtc::VideoStream> streams = streams_;
79 streams[streams_.size() - 1].height = height;
80 streams[streams_.size() - 1].width = width;
Seth Hampson24722b32017-12-22 09:36:42 -080081
82 streams[0].bitrate_priority = encoder_config.bitrate_priority;
mflodmand79f97b2016-12-15 07:24:33 -080083 return streams;
perkjfa10b552016-10-02 23:45:26 -070084 }
85
86 std::vector<webrtc::VideoStream> streams_;
87};
88
brandtr504b95e2016-12-21 02:54:35 -080089bool IsFlexfec(int payload_type) {
90 return payload_type == webrtc::VideoQualityTest::kFlexfecPayloadType;
91}
92
minyue73208662016-08-18 06:28:55 -070093} // namespace
ivica5d6a06c2015-09-17 05:30:24 -070094
95namespace webrtc {
96
ivica5d6a06c2015-09-17 05:30:24 -070097class VideoAnalyzer : public PacketReceiver,
pbos2d566682015-09-28 09:59:31 -070098 public Transport,
Niels Möllerf88a22c2018-06-19 17:05:03 +020099 public rtc::VideoSinkInterface<VideoFrame>,
100 public EncodedFrameObserver {
ivica5d6a06c2015-09-17 05:30:24 -0700101 public:
sprangce4aef12015-11-02 07:23:20 -0800102 VideoAnalyzer(test::LayerFilteringTransport* transport,
ivica5d6a06c2015-09-17 05:30:24 -0700103 const std::string& test_label,
104 double avg_psnr_threshold,
105 double avg_ssim_threshold,
106 int duration_frames,
sprangce4aef12015-11-02 07:23:20 -0800107 FILE* graph_data_output_file,
108 const std::string& graph_title,
ilnik3dd5ad92017-02-09 04:58:53 -0800109 uint32_t ssrc_to_analyze,
ilnik46a00212017-02-10 09:16:05 -0800110 uint32_t rtx_ssrc_to_analyze,
ilnikcb8c1462017-03-09 09:23:30 -0800111 size_t selected_stream,
ilnik1e7732c2017-02-23 05:07:56 -0800112 int selected_sl,
113 int selected_tl,
ilnik6b826ef2017-06-16 06:53:48 -0700114 bool is_quick_test_enabled,
ilnik98436952017-07-13 00:47:03 -0700115 Clock* clock,
116 std::string rtp_dump_name)
perkja49cbd32016-09-16 07:53:41 -0700117 : transport_(transport),
ivica5d6a06c2015-09-17 05:30:24 -0700118 receiver_(nullptr),
stefan889d9652017-07-05 03:03:02 -0700119 call_(nullptr),
ivica5d6a06c2015-09-17 05:30:24 -0700120 send_stream_(nullptr),
philipelfd870db2017-01-23 03:22:15 -0800121 receive_stream_(nullptr),
ilnik6b826ef2017-06-16 06:53:48 -0700122 captured_frame_forwarder_(this, clock),
ivica5d6a06c2015-09-17 05:30:24 -0700123 test_label_(test_label),
124 graph_data_output_file_(graph_data_output_file),
sprangce4aef12015-11-02 07:23:20 -0800125 graph_title_(graph_title),
126 ssrc_to_analyze_(ssrc_to_analyze),
ilnik46a00212017-02-10 09:16:05 -0800127 rtx_ssrc_to_analyze_(rtx_ssrc_to_analyze),
ilnikcb8c1462017-03-09 09:23:30 -0800128 selected_stream_(selected_stream),
ilnik1e7732c2017-02-23 05:07:56 -0800129 selected_sl_(selected_sl),
130 selected_tl_(selected_tl),
pbos14fe7082016-04-20 06:35:56 -0700131 pre_encode_proxy_(this),
stefan889d9652017-07-05 03:03:02 -0700132 last_fec_bytes_(0),
ivica5d6a06c2015-09-17 05:30:24 -0700133 frames_to_process_(duration_frames),
134 frames_recorded_(0),
135 frames_processed_(0),
136 dropped_frames_(0),
pbos14fe7082016-04-20 06:35:56 -0700137 dropped_frames_before_first_encode_(0),
138 dropped_frames_before_rendering_(0),
ivica5d6a06c2015-09-17 05:30:24 -0700139 last_render_time_(0),
Stefan Holmer32e930f2018-02-09 09:43:45 +0100140 last_render_delta_ms_(0),
141 last_unfreeze_time_ms_(0),
ivica5d6a06c2015-09-17 05:30:24 -0700142 rtp_timestamp_delta_(0),
ilnik1e7732c2017-02-23 05:07:56 -0800143 total_media_bytes_(0),
144 first_sending_time_(0),
145 last_sending_time_(0),
ilnikdf92c5c2017-02-23 02:08:44 -0800146 cpu_time_(0),
147 wallclock_time_(0),
ivica5d6a06c2015-09-17 05:30:24 -0700148 avg_psnr_threshold_(avg_psnr_threshold),
149 avg_ssim_threshold_(avg_ssim_threshold),
ilnik9ae0d762017-02-15 00:53:12 -0800150 is_quick_test_enabled_(is_quick_test_enabled),
Peter Boström8c38e8b2015-11-26 17:45:47 +0100151 stats_polling_thread_(&PollStatsThread, this, "StatsPoller"),
Peter Boström5811a392015-12-10 13:02:50 +0100152 comparison_available_event_(false, false),
ilnik98436952017-07-13 00:47:03 -0700153 done_(true, false),
154 clock_(clock),
155 start_ms_(clock->TimeInMilliseconds()) {
ivica5d6a06c2015-09-17 05:30:24 -0700156 // Create thread pool for CPU-expensive PSNR/SSIM calculations.
157
158 // Try to use about as many threads as cores, but leave kMinCoresLeft alone,
159 // so that we don't accidentally starve "real" worker threads (codec etc).
160 // Also, don't allocate more than kMaxComparisonThreads, even if there are
161 // spare cores.
162
163 uint32_t num_cores = CpuInfo::DetectNumberOfCores();
kwibergaf476c72016-11-28 15:21:39 -0800164 RTC_DCHECK_GE(num_cores, 1);
ivica5d6a06c2015-09-17 05:30:24 -0700165 static const uint32_t kMinCoresLeft = 4;
166 static const uint32_t kMaxComparisonThreads = 8;
167
168 if (num_cores <= kMinCoresLeft) {
169 num_cores = 1;
170 } else {
171 num_cores -= kMinCoresLeft;
172 num_cores = std::min(num_cores, kMaxComparisonThreads);
173 }
174
175 for (uint32_t i = 0; i < num_cores; ++i) {
Peter Boström8c38e8b2015-11-26 17:45:47 +0100176 rtc::PlatformThread* thread =
177 new rtc::PlatformThread(&FrameComparisonThread, this, "Analyzer");
178 thread->Start();
179 comparison_thread_pool_.push_back(thread);
ivica5d6a06c2015-09-17 05:30:24 -0700180 }
ilnik98436952017-07-13 00:47:03 -0700181
182 if (!rtp_dump_name.empty()) {
183 fprintf(stdout, "Writing rtp dump to %s\n", rtp_dump_name.c_str());
184 rtp_file_writer_.reset(test::RtpFileWriter::Create(
185 test::RtpFileWriter::kRtpDump, rtp_dump_name));
186 }
ivica5d6a06c2015-09-17 05:30:24 -0700187 }
188
189 ~VideoAnalyzer() {
Peter Boström8c38e8b2015-11-26 17:45:47 +0100190 for (rtc::PlatformThread* thread : comparison_thread_pool_) {
191 thread->Stop();
ivica5d6a06c2015-09-17 05:30:24 -0700192 delete thread;
193 }
194 }
195
196 virtual void SetReceiver(PacketReceiver* receiver) { receiver_ = receiver; }
197
ilnik6b826ef2017-06-16 06:53:48 -0700198 void SetSource(test::VideoCapturer* video_capturer, bool respect_sink_wants) {
199 if (respect_sink_wants)
200 captured_frame_forwarder_.SetSource(video_capturer);
201 rtc::VideoSinkWants wants;
202 video_capturer->AddOrUpdateSink(InputInterface(), wants);
203 }
204
stefan889d9652017-07-05 03:03:02 -0700205 void SetCall(Call* call) {
206 rtc::CritScope lock(&crit_);
207 RTC_DCHECK(!call_);
208 call_ = call;
209 }
210
perkja49cbd32016-09-16 07:53:41 -0700211 void SetSendStream(VideoSendStream* stream) {
212 rtc::CritScope lock(&crit_);
213 RTC_DCHECK(!send_stream_);
214 send_stream_ = stream;
215 }
216
philipelfd870db2017-01-23 03:22:15 -0800217 void SetReceiveStream(VideoReceiveStream* stream) {
218 rtc::CritScope lock(&crit_);
219 RTC_DCHECK(!receive_stream_);
220 receive_stream_ = stream;
221 }
222
perkja49cbd32016-09-16 07:53:41 -0700223 rtc::VideoSinkInterface<VideoFrame>* InputInterface() {
224 return &captured_frame_forwarder_;
225 }
226 rtc::VideoSourceInterface<VideoFrame>* OutputInterface() {
227 return &captured_frame_forwarder_;
228 }
229
ivica5d6a06c2015-09-17 05:30:24 -0700230 DeliveryStatus DeliverPacket(MediaType media_type,
Danil Chapovalov292a73e2017-12-07 17:00:40 +0100231 rtc::CopyOnWriteBuffer packet,
ivica5d6a06c2015-09-17 05:30:24 -0700232 const PacketTime& packet_time) override {
Taylor Brandstetter433b95a2016-03-18 11:41:03 -0700233 // Ignore timestamps of RTCP packets. They're not synchronized with
234 // RTP packet timestamps and so they would confuse wrap_handler_.
Danil Chapovalov292a73e2017-12-07 17:00:40 +0100235 if (RtpHeaderParser::IsRtcp(packet.cdata(), packet.size())) {
236 return receiver_->DeliverPacket(media_type, std::move(packet),
237 packet_time);
Taylor Brandstetter433b95a2016-03-18 11:41:03 -0700238 }
ilnik98436952017-07-13 00:47:03 -0700239
240 if (rtp_file_writer_) {
241 test::RtpPacket p;
Danil Chapovalov292a73e2017-12-07 17:00:40 +0100242 memcpy(p.data, packet.cdata(), packet.size());
243 p.length = packet.size();
244 p.original_length = packet.size();
ilnik98436952017-07-13 00:47:03 -0700245 p.time_ms = clock_->TimeInMilliseconds() - start_ms_;
246 rtp_file_writer_->WritePacket(&p);
247 }
248
Danil Chapovalov292a73e2017-12-07 17:00:40 +0100249 RtpUtility::RtpHeaderParser parser(packet.cdata(), packet.size());
ivica5d6a06c2015-09-17 05:30:24 -0700250 RTPHeader header;
danilchapf6975f42015-12-28 10:18:46 -0800251 parser.Parse(&header);
ilnik46a00212017-02-10 09:16:05 -0800252 if (!IsFlexfec(header.payloadType) &&
253 (header.ssrc == ssrc_to_analyze_ ||
254 header.ssrc == rtx_ssrc_to_analyze_)) {
brandtr504b95e2016-12-21 02:54:35 -0800255 // Ignore FlexFEC timestamps, to avoid collisions with media timestamps.
256 // (FlexFEC and media are sent on different SSRCs, which have different
257 // timestamps spaces.)
ilnik46a00212017-02-10 09:16:05 -0800258 // Also ignore packets from wrong SSRC, but include retransmits.
ivica5d6a06c2015-09-17 05:30:24 -0700259 rtc::CritScope lock(&crit_);
sprang16daaa52016-03-09 01:30:24 -0800260 int64_t timestamp =
261 wrap_handler_.Unwrap(header.timestamp - rtp_timestamp_delta_);
262 recv_times_[timestamp] =
ivica5d6a06c2015-09-17 05:30:24 -0700263 Clock::GetRealTimeClock()->CurrentNtpInMilliseconds();
264 }
265
Danil Chapovalov292a73e2017-12-07 17:00:40 +0100266 return receiver_->DeliverPacket(media_type, std::move(packet), packet_time);
ivica5d6a06c2015-09-17 05:30:24 -0700267 }
268
pbos14fe7082016-04-20 06:35:56 -0700269 void PreEncodeOnFrame(const VideoFrame& video_frame) {
270 rtc::CritScope lock(&crit_);
ilnik3dd5ad92017-02-09 04:58:53 -0800271 if (!first_encoded_timestamp_) {
pbos14fe7082016-04-20 06:35:56 -0700272 while (frames_.front().timestamp() != video_frame.timestamp()) {
273 ++dropped_frames_before_first_encode_;
274 frames_.pop_front();
275 RTC_CHECK(!frames_.empty());
276 }
Oskar Sundbom8e07c132018-01-08 16:45:42 +0100277 first_encoded_timestamp_ = video_frame.timestamp();
ilnik3dd5ad92017-02-09 04:58:53 -0800278 }
279 }
280
Niels Möllerf88a22c2018-06-19 17:05:03 +0200281 // EncodedFrameObserver implementation, wired to post_encode_callback.
282 void EncodedFrameCallback(const EncodedFrame& encoded_frame) override {
ilnik3dd5ad92017-02-09 04:58:53 -0800283 rtc::CritScope lock(&crit_);
284 if (!first_sent_timestamp_ &&
ilnikcb8c1462017-03-09 09:23:30 -0800285 encoded_frame.stream_id_ == selected_stream_) {
Oskar Sundbom8e07c132018-01-08 16:45:42 +0100286 first_sent_timestamp_ = encoded_frame.timestamp_;
pbos14fe7082016-04-20 06:35:56 -0700287 }
288 }
289
stefan1d8a5062015-10-02 03:39:33 -0700290 bool SendRtp(const uint8_t* packet,
291 size_t length,
292 const PacketOptions& options) override {
sprangce4aef12015-11-02 07:23:20 -0800293 RtpUtility::RtpHeaderParser parser(packet, length);
ivica5d6a06c2015-09-17 05:30:24 -0700294 RTPHeader header;
danilchapf6975f42015-12-28 10:18:46 -0800295 parser.Parse(&header);
ivica5d6a06c2015-09-17 05:30:24 -0700296
sprangce4aef12015-11-02 07:23:20 -0800297 int64_t current_time =
298 Clock::GetRealTimeClock()->CurrentNtpInMilliseconds();
ilnik2a8c2f52017-02-15 02:23:28 -0800299
sprangce4aef12015-11-02 07:23:20 -0800300 bool result = transport_->SendRtp(packet, length, options);
ivica5d6a06c2015-09-17 05:30:24 -0700301 {
302 rtc::CritScope lock(&crit_);
ilnik3dd5ad92017-02-09 04:58:53 -0800303 if (rtp_timestamp_delta_ == 0 && header.ssrc == ssrc_to_analyze_) {
ilnik1e1c84d2017-02-09 08:32:53 -0800304 RTC_CHECK(static_cast<bool>(first_sent_timestamp_));
ilnik3dd5ad92017-02-09 04:58:53 -0800305 rtp_timestamp_delta_ = header.timestamp - *first_sent_timestamp_;
ilnike67c59e2017-02-09 04:08:56 -0800306 }
ilnik3dd5ad92017-02-09 04:58:53 -0800307
308 if (!IsFlexfec(header.payloadType) && header.ssrc == ssrc_to_analyze_) {
brandtr504b95e2016-12-21 02:54:35 -0800309 // Ignore FlexFEC timestamps, to avoid collisions with media timestamps.
310 // (FlexFEC and media are sent on different SSRCs, which have different
311 // timestamps spaces.)
ilnik46a00212017-02-10 09:16:05 -0800312 // Also ignore packets from wrong SSRC and retransmits.
brandtr504b95e2016-12-21 02:54:35 -0800313 int64_t timestamp =
314 wrap_handler_.Unwrap(header.timestamp - rtp_timestamp_delta_);
315 send_times_[timestamp] = current_time;
ilnik1e7732c2017-02-23 05:07:56 -0800316
317 if (IsInSelectedSpatialAndTemporalLayer(packet, length, header)) {
brandtr504b95e2016-12-21 02:54:35 -0800318 encoded_frame_sizes_[timestamp] +=
319 length - (header.headerLength + header.paddingLength);
ilnik1e7732c2017-02-23 05:07:56 -0800320 total_media_bytes_ +=
321 length - (header.headerLength + header.paddingLength);
brandtr504b95e2016-12-21 02:54:35 -0800322 }
ilnik1e7732c2017-02-23 05:07:56 -0800323 if (first_sending_time_ == 0)
324 first_sending_time_ = current_time;
325 last_sending_time_ = current_time;
sprangce4aef12015-11-02 07:23:20 -0800326 }
ivica5d6a06c2015-09-17 05:30:24 -0700327 }
sprangce4aef12015-11-02 07:23:20 -0800328 return result;
ivica5d6a06c2015-09-17 05:30:24 -0700329 }
330
331 bool SendRtcp(const uint8_t* packet, size_t length) override {
332 return transport_->SendRtcp(packet, length);
333 }
334
nisseeb83a1a2016-03-21 01:27:56 -0700335 void OnFrame(const VideoFrame& video_frame) override {
ivica5d6a06c2015-09-17 05:30:24 -0700336 int64_t render_time_ms =
337 Clock::GetRealTimeClock()->CurrentNtpInMilliseconds();
ivica5d6a06c2015-09-17 05:30:24 -0700338
339 rtc::CritScope lock(&crit_);
ilnikdf92c5c2017-02-23 02:08:44 -0800340
341 StartExcludingCpuThreadTime();
342
Taylor Brandstetter433b95a2016-03-18 11:41:03 -0700343 int64_t send_timestamp =
sprang16daaa52016-03-09 01:30:24 -0800344 wrap_handler_.Unwrap(video_frame.timestamp() - rtp_timestamp_delta_);
ivica5d6a06c2015-09-17 05:30:24 -0700345
sprang16daaa52016-03-09 01:30:24 -0800346 while (wrap_handler_.Unwrap(frames_.front().timestamp()) < send_timestamp) {
nisse97f0b932016-05-26 09:44:40 -0700347 if (!last_rendered_frame_) {
pbos14fe7082016-04-20 06:35:56 -0700348 // No previous frame rendered, this one was dropped after sending but
349 // before rendering.
350 ++dropped_frames_before_rendering_;
kthelgason2bc68642017-02-07 07:02:22 -0800351 } else {
352 AddFrameComparison(frames_.front(), *last_rendered_frame_, true,
353 render_time_ms);
pbos14fe7082016-04-20 06:35:56 -0700354 }
ivica5d6a06c2015-09-17 05:30:24 -0700355 frames_.pop_front();
pbos14fe7082016-04-20 06:35:56 -0700356 RTC_DCHECK(!frames_.empty());
ivica5d6a06c2015-09-17 05:30:24 -0700357 }
358
359 VideoFrame reference_frame = frames_.front();
360 frames_.pop_front();
sprang16daaa52016-03-09 01:30:24 -0800361 int64_t reference_timestamp =
362 wrap_handler_.Unwrap(reference_frame.timestamp());
363 if (send_timestamp == reference_timestamp - 1) {
sprangce4aef12015-11-02 07:23:20 -0800364 // TODO(ivica): Make this work for > 2 streams.
Peter Boströme4499152016-02-05 11:13:28 +0100365 // Look at RTPSender::BuildRTPHeader.
sprangce4aef12015-11-02 07:23:20 -0800366 ++send_timestamp;
367 }
sprang16daaa52016-03-09 01:30:24 -0800368 ASSERT_EQ(reference_timestamp, send_timestamp);
ivica5d6a06c2015-09-17 05:30:24 -0700369
370 AddFrameComparison(reference_frame, video_frame, false, render_time_ms);
371
Oskar Sundbom8e07c132018-01-08 16:45:42 +0100372 last_rendered_frame_ = video_frame;
ilnikdf92c5c2017-02-23 02:08:44 -0800373
374 StopExcludingCpuThreadTime();
ivica5d6a06c2015-09-17 05:30:24 -0700375 }
376
ivica5d6a06c2015-09-17 05:30:24 -0700377 void Wait() {
378 // Frame comparisons can be very expensive. Wait for test to be done, but
379 // at time-out check if frames_processed is going up. If so, give it more
380 // time, otherwise fail. Hopefully this will reduce test flakiness.
381
Peter Boström8c38e8b2015-11-26 17:45:47 +0100382 stats_polling_thread_.Start();
sprangce4aef12015-11-02 07:23:20 -0800383
ivica5d6a06c2015-09-17 05:30:24 -0700384 int last_frames_processed = -1;
ivica5d6a06c2015-09-17 05:30:24 -0700385 int iteration = 0;
Peter Boström5811a392015-12-10 13:02:50 +0100386 while (!done_.Wait(VideoQualityTest::kDefaultTimeoutMs)) {
ivica5d6a06c2015-09-17 05:30:24 -0700387 int frames_processed;
388 {
389 rtc::CritScope crit(&comparison_lock_);
390 frames_processed = frames_processed_;
391 }
392
393 // Print some output so test infrastructure won't think we've crashed.
394 const char* kKeepAliveMessages[3] = {
395 "Uh, I'm-I'm not quite dead, sir.",
396 "Uh, I-I think uh, I could pull through, sir.",
397 "Actually, I think I'm all right to come with you--"};
398 printf("- %s\n", kKeepAliveMessages[iteration++ % 3]);
399
400 if (last_frames_processed == -1) {
401 last_frames_processed = frames_processed;
402 continue;
403 }
Peter Boströmdd45eb62016-01-19 15:22:32 +0100404 if (frames_processed == last_frames_processed) {
405 EXPECT_GT(frames_processed, last_frames_processed)
406 << "Analyzer stalled while waiting for test to finish.";
407 done_.Set();
408 break;
409 }
ivica5d6a06c2015-09-17 05:30:24 -0700410 last_frames_processed = frames_processed;
411 }
412
413 if (iteration > 0)
414 printf("- Farewell, sweet Concorde!\n");
415
Peter Boström8c38e8b2015-11-26 17:45:47 +0100416 stats_polling_thread_.Stop();
ivica5d6a06c2015-09-17 05:30:24 -0700417 }
418
pbos14fe7082016-04-20 06:35:56 -0700419 rtc::VideoSinkInterface<VideoFrame>* pre_encode_proxy() {
420 return &pre_encode_proxy_;
421 }
Peter Boströme4499152016-02-05 11:13:28 +0100422
ilnikdf92c5c2017-02-23 02:08:44 -0800423 void StartMeasuringCpuProcessTime() {
424 rtc::CritScope lock(&cpu_measurement_lock_);
425 cpu_time_ -= rtc::GetProcessCpuTimeNanos();
426 wallclock_time_ -= rtc::SystemTimeNanos();
427 }
428
429 void StopMeasuringCpuProcessTime() {
430 rtc::CritScope lock(&cpu_measurement_lock_);
431 cpu_time_ += rtc::GetProcessCpuTimeNanos();
432 wallclock_time_ += rtc::SystemTimeNanos();
433 }
434
435 void StartExcludingCpuThreadTime() {
436 rtc::CritScope lock(&cpu_measurement_lock_);
437 cpu_time_ += rtc::GetThreadCpuTimeNanos();
438 }
439
440 void StopExcludingCpuThreadTime() {
441 rtc::CritScope lock(&cpu_measurement_lock_);
442 cpu_time_ -= rtc::GetThreadCpuTimeNanos();
443 }
444
445 double GetCpuUsagePercent() {
446 rtc::CritScope lock(&cpu_measurement_lock_);
447 return static_cast<double>(cpu_time_) / wallclock_time_ * 100.0;
448 }
449
sprangce4aef12015-11-02 07:23:20 -0800450 test::LayerFilteringTransport* const transport_;
ivica5d6a06c2015-09-17 05:30:24 -0700451 PacketReceiver* receiver_;
ivica5d6a06c2015-09-17 05:30:24 -0700452
453 private:
454 struct FrameComparison {
455 FrameComparison()
456 : dropped(false),
nissedf2ceb82016-12-15 06:29:53 -0800457 input_time_ms(0),
ivica5d6a06c2015-09-17 05:30:24 -0700458 send_time_ms(0),
459 recv_time_ms(0),
460 render_time_ms(0),
461 encoded_frame_size(0) {}
462
463 FrameComparison(const VideoFrame& reference,
464 const VideoFrame& render,
465 bool dropped,
nissedf2ceb82016-12-15 06:29:53 -0800466 int64_t input_time_ms,
ivica5d6a06c2015-09-17 05:30:24 -0700467 int64_t send_time_ms,
468 int64_t recv_time_ms,
469 int64_t render_time_ms,
470 size_t encoded_frame_size)
471 : reference(reference),
472 render(render),
473 dropped(dropped),
nissedf2ceb82016-12-15 06:29:53 -0800474 input_time_ms(input_time_ms),
ivica5d6a06c2015-09-17 05:30:24 -0700475 send_time_ms(send_time_ms),
476 recv_time_ms(recv_time_ms),
477 render_time_ms(render_time_ms),
478 encoded_frame_size(encoded_frame_size) {}
479
nissedf2ceb82016-12-15 06:29:53 -0800480 FrameComparison(bool dropped,
481 int64_t input_time_ms,
482 int64_t send_time_ms,
483 int64_t recv_time_ms,
484 int64_t render_time_ms,
485 size_t encoded_frame_size)
486 : dropped(dropped),
487 input_time_ms(input_time_ms),
488 send_time_ms(send_time_ms),
489 recv_time_ms(recv_time_ms),
490 render_time_ms(render_time_ms),
491 encoded_frame_size(encoded_frame_size) {}
492
Danil Chapovalovb9b146c2018-06-15 12:28:07 +0200493 absl::optional<VideoFrame> reference;
494 absl::optional<VideoFrame> render;
ivica5d6a06c2015-09-17 05:30:24 -0700495 bool dropped;
nissedf2ceb82016-12-15 06:29:53 -0800496 int64_t input_time_ms;
ivica5d6a06c2015-09-17 05:30:24 -0700497 int64_t send_time_ms;
498 int64_t recv_time_ms;
499 int64_t render_time_ms;
500 size_t encoded_frame_size;
501 };
502
503 struct Sample {
ivica8d15bd62015-10-07 02:43:12 -0700504 Sample(int dropped,
505 int64_t input_time_ms,
506 int64_t send_time_ms,
507 int64_t recv_time_ms,
508 int64_t render_time_ms,
509 size_t encoded_frame_size,
ivica5d6a06c2015-09-17 05:30:24 -0700510 double psnr,
ivica8d15bd62015-10-07 02:43:12 -0700511 double ssim)
ivica5d6a06c2015-09-17 05:30:24 -0700512 : dropped(dropped),
513 input_time_ms(input_time_ms),
514 send_time_ms(send_time_ms),
515 recv_time_ms(recv_time_ms),
ivica8d15bd62015-10-07 02:43:12 -0700516 render_time_ms(render_time_ms),
ivica5d6a06c2015-09-17 05:30:24 -0700517 encoded_frame_size(encoded_frame_size),
518 psnr(psnr),
ivica8d15bd62015-10-07 02:43:12 -0700519 ssim(ssim) {}
ivica5d6a06c2015-09-17 05:30:24 -0700520
ivica8d15bd62015-10-07 02:43:12 -0700521 int dropped;
522 int64_t input_time_ms;
523 int64_t send_time_ms;
524 int64_t recv_time_ms;
525 int64_t render_time_ms;
526 size_t encoded_frame_size;
ivica5d6a06c2015-09-17 05:30:24 -0700527 double psnr;
528 double ssim;
ivica5d6a06c2015-09-17 05:30:24 -0700529 };
530
pbos14fe7082016-04-20 06:35:56 -0700531 // This class receives the send-side OnFrame callback and is provided to not
532 // conflict with the receiver-side renderer callback.
533 class PreEncodeProxy : public rtc::VideoSinkInterface<VideoFrame> {
534 public:
535 explicit PreEncodeProxy(VideoAnalyzer* parent) : parent_(parent) {}
536
537 void OnFrame(const VideoFrame& video_frame) override {
538 parent_->PreEncodeOnFrame(video_frame);
539 }
540
541 private:
542 VideoAnalyzer* const parent_;
543 };
544
ilnik1e7732c2017-02-23 05:07:56 -0800545 bool IsInSelectedSpatialAndTemporalLayer(const uint8_t* packet,
546 size_t length,
547 const RTPHeader& header) {
ilnik863f03b2017-07-11 02:38:36 -0700548 if (header.payloadType != test::CallTest::kPayloadTypeVP9 &&
549 header.payloadType != test::CallTest::kPayloadTypeVP8) {
ilnik1e7732c2017-02-23 05:07:56 -0800550 return true;
551 } else {
552 // Get VP8 and VP9 specific header to check layers indexes.
553 const uint8_t* payload = packet + header.headerLength;
554 const size_t payload_length = length - header.headerLength;
555 const size_t payload_data_length = payload_length - header.paddingLength;
ilnik863f03b2017-07-11 02:38:36 -0700556 const bool is_vp8 = header.payloadType == test::CallTest::kPayloadTypeVP8;
ilnik1e7732c2017-02-23 05:07:56 -0800557 std::unique_ptr<RtpDepacketizer> depacketizer(
Niels Möller520ca4e2018-06-04 11:14:38 +0200558 RtpDepacketizer::Create(is_vp8 ? kVideoCodecVP8 : kVideoCodecVP9));
ilnik1e7732c2017-02-23 05:07:56 -0800559 RtpDepacketizer::ParsedPayload parsed_payload;
560 bool result =
561 depacketizer->Parse(&parsed_payload, payload, payload_data_length);
562 RTC_DCHECK(result);
563 const int temporal_idx = static_cast<int>(
564 is_vp8 ? parsed_payload.type.Video.codecHeader.VP8.temporalIdx
565 : parsed_payload.type.Video.codecHeader.VP9.temporal_idx);
566 const int spatial_idx = static_cast<int>(
567 is_vp8 ? kNoSpatialIdx
568 : parsed_payload.type.Video.codecHeader.VP9.spatial_idx);
569 return (selected_tl_ < 0 || temporal_idx == kNoTemporalIdx ||
570 temporal_idx <= selected_tl_) &&
571 (selected_sl_ < 0 || spatial_idx == kNoSpatialIdx ||
572 spatial_idx <= selected_sl_);
573 }
574 }
575
ivica5d6a06c2015-09-17 05:30:24 -0700576 void AddFrameComparison(const VideoFrame& reference,
577 const VideoFrame& render,
578 bool dropped,
579 int64_t render_time_ms)
danilchapa37de392017-09-09 04:17:22 -0700580 RTC_EXCLUSIVE_LOCKS_REQUIRED(crit_) {
sprange1f2f1f2016-02-01 02:04:52 -0800581 int64_t reference_timestamp = wrap_handler_.Unwrap(reference.timestamp());
582 int64_t send_time_ms = send_times_[reference_timestamp];
583 send_times_.erase(reference_timestamp);
584 int64_t recv_time_ms = recv_times_[reference_timestamp];
585 recv_times_.erase(reference_timestamp);
ivica5d6a06c2015-09-17 05:30:24 -0700586
sprangce4aef12015-11-02 07:23:20 -0800587 // TODO(ivica): Make this work for > 2 streams.
sprange1f2f1f2016-02-01 02:04:52 -0800588 auto it = encoded_frame_sizes_.find(reference_timestamp);
sprangce4aef12015-11-02 07:23:20 -0800589 if (it == encoded_frame_sizes_.end())
sprange1f2f1f2016-02-01 02:04:52 -0800590 it = encoded_frame_sizes_.find(reference_timestamp - 1);
sprangce4aef12015-11-02 07:23:20 -0800591 size_t encoded_size = it == encoded_frame_sizes_.end() ? 0 : it->second;
592 if (it != encoded_frame_sizes_.end())
593 encoded_frame_sizes_.erase(it);
ivica5d6a06c2015-09-17 05:30:24 -0700594
ivica5d6a06c2015-09-17 05:30:24 -0700595 rtc::CritScope crit(&comparison_lock_);
stefanb1797672016-08-11 07:00:57 -0700596 if (comparisons_.size() < kMaxComparisons) {
Yves Gerey665174f2018-06-19 15:03:05 +0200597 comparisons_.push_back(FrameComparison(
598 reference, render, dropped, reference.ntp_time_ms(), send_time_ms,
599 recv_time_ms, render_time_ms, encoded_size));
stefanb1797672016-08-11 07:00:57 -0700600 } else {
Yves Gerey665174f2018-06-19 15:03:05 +0200601 comparisons_.push_back(FrameComparison(dropped, reference.ntp_time_ms(),
nissedf2ceb82016-12-15 06:29:53 -0800602 send_time_ms, recv_time_ms,
603 render_time_ms, encoded_size));
stefanb1797672016-08-11 07:00:57 -0700604 }
Peter Boström5811a392015-12-10 13:02:50 +0100605 comparison_available_event_.Set();
ivica5d6a06c2015-09-17 05:30:24 -0700606 }
607
tommi0f8b4032017-02-22 11:22:05 -0800608 static void PollStatsThread(void* obj) {
609 static_cast<VideoAnalyzer*>(obj)->PollStats();
ivica5d6a06c2015-09-17 05:30:24 -0700610 }
611
tommi0f8b4032017-02-22 11:22:05 -0800612 void PollStats() {
613 while (!done_.Wait(kSendStatsPollingIntervalMs)) {
614 rtc::CritScope crit(&comparison_lock_);
ivica5d6a06c2015-09-17 05:30:24 -0700615
stefan889d9652017-07-05 03:03:02 -0700616 Call::Stats call_stats = call_->GetStats();
617 send_bandwidth_bps_.AddSample(call_stats.send_bandwidth_bps);
618
tommi0f8b4032017-02-22 11:22:05 -0800619 VideoSendStream::Stats send_stats = send_stream_->GetStats();
620 // It's not certain that we yet have estimates for any of these stats.
621 // Check that they are positive before mixing them in.
622 if (send_stats.encode_frame_rate > 0)
623 encode_frame_rate_.AddSample(send_stats.encode_frame_rate);
624 if (send_stats.avg_encode_time_ms > 0)
625 encode_time_ms_.AddSample(send_stats.avg_encode_time_ms);
626 if (send_stats.encode_usage_percent > 0)
627 encode_usage_percent_.AddSample(send_stats.encode_usage_percent);
628 if (send_stats.media_bitrate_bps > 0)
629 media_bitrate_bps_.AddSample(send_stats.media_bitrate_bps);
stefan889d9652017-07-05 03:03:02 -0700630 size_t fec_bytes = 0;
631 for (auto kv : send_stats.substreams) {
632 fec_bytes += kv.second.rtp_stats.fec.payload_bytes +
633 kv.second.rtp_stats.fec.padding_bytes;
634 }
635 fec_bitrate_bps_.AddSample((fec_bytes - last_fec_bytes_) * 8);
636 last_fec_bytes_ = fec_bytes;
philipelfd870db2017-01-23 03:22:15 -0800637
tommi0f8b4032017-02-22 11:22:05 -0800638 if (receive_stream_ != nullptr) {
639 VideoReceiveStream::Stats receive_stats = receive_stream_->GetStats();
640 if (receive_stats.decode_ms > 0)
641 decode_time_ms_.AddSample(receive_stats.decode_ms);
642 if (receive_stats.max_decode_ms > 0)
643 decode_time_max_ms_.AddSample(receive_stats.max_decode_ms);
644 }
ilnikdaa574d2017-03-01 06:46:05 -0800645
646 memory_usage_.AddSample(rtc::GetProcessResidentSizeBytes());
philipelfd870db2017-01-23 03:22:15 -0800647 }
ivica5d6a06c2015-09-17 05:30:24 -0700648 }
649
650 static bool FrameComparisonThread(void* obj) {
651 return static_cast<VideoAnalyzer*>(obj)->CompareFrames();
652 }
653
654 bool CompareFrames() {
655 if (AllFramesRecorded())
656 return false;
657
ivica5d6a06c2015-09-17 05:30:24 -0700658 FrameComparison comparison;
659
660 if (!PopComparison(&comparison)) {
661 // Wait until new comparison task is available, or test is done.
662 // If done, wake up remaining threads waiting.
Peter Boström5811a392015-12-10 13:02:50 +0100663 comparison_available_event_.Wait(1000);
ivica5d6a06c2015-09-17 05:30:24 -0700664 if (AllFramesRecorded()) {
Peter Boström5811a392015-12-10 13:02:50 +0100665 comparison_available_event_.Set();
ivica5d6a06c2015-09-17 05:30:24 -0700666 return false;
667 }
668 return true; // Try again.
669 }
670
ilnikdf92c5c2017-02-23 02:08:44 -0800671 StartExcludingCpuThreadTime();
672
ivica5d6a06c2015-09-17 05:30:24 -0700673 PerformFrameComparison(comparison);
674
ilnikdf92c5c2017-02-23 02:08:44 -0800675 StopExcludingCpuThreadTime();
676
ivica5d6a06c2015-09-17 05:30:24 -0700677 if (FrameProcessed()) {
678 PrintResults();
679 if (graph_data_output_file_)
680 PrintSamplesToFile();
Peter Boström5811a392015-12-10 13:02:50 +0100681 done_.Set();
682 comparison_available_event_.Set();
ivica5d6a06c2015-09-17 05:30:24 -0700683 return false;
684 }
685
686 return true;
687 }
688
689 bool PopComparison(FrameComparison* comparison) {
690 rtc::CritScope crit(&comparison_lock_);
691 // If AllFramesRecorded() is true, it means we have already popped
692 // frames_to_process_ frames from comparisons_, so there is no more work
693 // for this thread to be done. frames_processed_ might still be lower if
694 // all comparisons are not done, but those frames are currently being
695 // worked on by other threads.
696 if (comparisons_.empty() || AllFramesRecorded())
697 return false;
698
699 *comparison = comparisons_.front();
700 comparisons_.pop_front();
701
702 FrameRecorded();
703 return true;
704 }
705
706 // Increment counter for number of frames received for comparison.
707 void FrameRecorded() {
708 rtc::CritScope crit(&comparison_lock_);
709 ++frames_recorded_;
710 }
711
712 // Returns true if all frames to be compared have been taken from the queue.
713 bool AllFramesRecorded() {
714 rtc::CritScope crit(&comparison_lock_);
715 assert(frames_recorded_ <= frames_to_process_);
716 return frames_recorded_ == frames_to_process_;
717 }
718
719 // Increase count of number of frames processed. Returns true if this was the
720 // last frame to be processed.
721 bool FrameProcessed() {
722 rtc::CritScope crit(&comparison_lock_);
723 ++frames_processed_;
724 assert(frames_processed_ <= frames_to_process_);
725 return frames_processed_ == frames_to_process_;
726 }
727
728 void PrintResults() {
ilnikdf92c5c2017-02-23 02:08:44 -0800729 StopMeasuringCpuProcessTime();
ivica5d6a06c2015-09-17 05:30:24 -0700730 rtc::CritScope crit(&comparison_lock_);
Stefan Holmer32e930f2018-02-09 09:43:45 +0100731 // Record the time from the last freeze until the last rendered frame to
732 // ensure we cover the full timespan of the session. Otherwise the metric
733 // would penalize an early freeze followed by no freezes until the end.
734 time_between_freezes_.AddSample(last_render_time_ - last_unfreeze_time_ms_);
ivica5d6a06c2015-09-17 05:30:24 -0700735 PrintResult("psnr", psnr_, " dB");
tnakamura3123cbc2016-02-10 11:21:51 -0800736 PrintResult("ssim", ssim_, " score");
stefan2da7a242017-03-30 01:02:15 -0700737 PrintResult("sender_time", sender_time_, " ms");
738 PrintResult("receiver_time", receiver_time_, " ms");
Ying Wang338bd622018-02-09 16:21:26 +0100739 PrintResult("network_time", network_time_, " ms");
ivica5d6a06c2015-09-17 05:30:24 -0700740 PrintResult("total_delay_incl_network", end_to_end_, " ms");
741 PrintResult("time_between_rendered_frames", rendered_delta_, " ms");
ivica5d6a06c2015-09-17 05:30:24 -0700742 PrintResult("encode_frame_rate", encode_frame_rate_, " fps");
philipelfd870db2017-01-23 03:22:15 -0800743 PrintResult("encode_time", encode_time_ms_, " ms");
philipelfd870db2017-01-23 03:22:15 -0800744 PrintResult("media_bitrate", media_bitrate_bps_, " bps");
stefan889d9652017-07-05 03:03:02 -0700745 PrintResult("fec_bitrate", fec_bitrate_bps_, " bps");
746 PrintResult("send_bandwidth", send_bandwidth_bps_, " bps");
Stefan Holmer32e930f2018-02-09 09:43:45 +0100747 PrintResult("time_between_freezes", time_between_freezes_, " ms");
philipelfd870db2017-01-23 03:22:15 -0800748
ilnik59cac992017-07-25 05:45:03 -0700749 if (worst_frame_) {
Edward Lemurd1d67432018-01-02 11:54:37 +0100750 test::PrintResult("min_psnr", "", test_label_.c_str(), worst_frame_->psnr,
Yves Gerey665174f2018-06-19 15:03:05 +0200751 "dB", false);
ilnik59cac992017-07-25 05:45:03 -0700752 }
753
philipelfd870db2017-01-23 03:22:15 -0800754 if (receive_stream_ != nullptr) {
755 PrintResult("decode_time", decode_time_ms_, " ms");
philipelfd870db2017-01-23 03:22:15 -0800756 }
ivica5d6a06c2015-09-17 05:30:24 -0700757
Edward Lemurd1d67432018-01-02 11:54:37 +0100758 test::PrintResult("dropped_frames", "", test_label_.c_str(),
Yves Gerey665174f2018-06-19 15:03:05 +0200759 dropped_frames_, "frames", false);
Edward Lemurd1d67432018-01-02 11:54:37 +0100760 test::PrintResult("cpu_usage", "", test_label_.c_str(),
761 GetCpuUsagePercent(), "%", false);
ilnikdaa574d2017-03-01 06:46:05 -0800762
763#if defined(WEBRTC_WIN)
Yves Gerey665174f2018-06-19 15:03:05 +0200764 // On Linux and Mac in Resident Set some unused pages may be counted.
765 // Therefore this metric will depend on order in which tests are run and
766 // will be flaky.
ilnikdaa574d2017-03-01 06:46:05 -0800767 PrintResult("memory_usage", memory_usage_, " bytes");
768#endif
ilnik59cac992017-07-25 05:45:03 -0700769
ilnikee42d192017-08-22 07:16:20 -0700770 // Saving only the worst frame for manual analysis. Intention here is to
771 // only detect video corruptions and not to track picture quality. Thus,
772 // jpeg is used here.
oprypin9b2f20c2017-08-29 05:51:57 -0700773 if (FLAG_save_worst_frame && worst_frame_) {
ilnikee42d192017-08-22 07:16:20 -0700774 std::string output_dir;
Edward Lemuraf8659a2017-09-27 14:46:24 +0200775 test::GetTestArtifactsDir(&output_dir);
ilnikee42d192017-08-22 07:16:20 -0700776 std::string output_path =
777 rtc::Pathname(output_dir, test_label_ + ".jpg").pathname();
Mirko Bonadei675513b2017-11-09 11:09:25 +0100778 RTC_LOG(LS_INFO) << "Saving worst frame to " << output_path;
ilnikee42d192017-08-22 07:16:20 -0700779 test::JpegFrameWriter frame_writer(output_path);
Yves Gerey665174f2018-06-19 15:03:05 +0200780 RTC_CHECK(
781 frame_writer.WriteFrame(worst_frame_->frame, 100 /*best quality*/));
ilnik59cac992017-07-25 05:45:03 -0700782 }
ilnikdaa574d2017-03-01 06:46:05 -0800783
ilnik9ae0d762017-02-15 00:53:12 -0800784 // Disable quality check for quick test, as quality checks may fail
785 // because too few samples were collected.
786 if (!is_quick_test_enabled_) {
787 EXPECT_GT(psnr_.Mean(), avg_psnr_threshold_);
788 EXPECT_GT(ssim_.Mean(), avg_ssim_threshold_);
789 }
ivica5d6a06c2015-09-17 05:30:24 -0700790 }
791
792 void PerformFrameComparison(const FrameComparison& comparison) {
793 // Perform expensive psnr and ssim calculations while not holding lock.
stefanb1797672016-08-11 07:00:57 -0700794 double psnr = -1.0;
795 double ssim = -1.0;
ilnik6b826ef2017-06-16 06:53:48 -0700796 if (comparison.reference && !comparison.dropped) {
nissedf2ceb82016-12-15 06:29:53 -0800797 psnr = I420PSNR(&*comparison.reference, &*comparison.render);
798 ssim = I420SSIM(&*comparison.reference, &*comparison.render);
stefanb1797672016-08-11 07:00:57 -0700799 }
ivica5d6a06c2015-09-17 05:30:24 -0700800
ivica5d6a06c2015-09-17 05:30:24 -0700801 rtc::CritScope crit(&comparison_lock_);
ilnik59cac992017-07-25 05:45:03 -0700802
803 if (psnr >= 0.0 && (!worst_frame_ || worst_frame_->psnr > psnr)) {
804 worst_frame_.emplace(FrameWithPsnr{psnr, *comparison.render});
805 }
806
ivica5d6a06c2015-09-17 05:30:24 -0700807 if (graph_data_output_file_) {
nissedf2ceb82016-12-15 06:29:53 -0800808 samples_.push_back(Sample(
809 comparison.dropped, comparison.input_time_ms, comparison.send_time_ms,
810 comparison.recv_time_ms, comparison.render_time_ms,
811 comparison.encoded_frame_size, psnr, ssim));
ivica5d6a06c2015-09-17 05:30:24 -0700812 }
stefanb1797672016-08-11 07:00:57 -0700813 if (psnr >= 0.0)
814 psnr_.AddSample(psnr);
815 if (ssim >= 0.0)
816 ssim_.AddSample(ssim);
ivica5d6a06c2015-09-17 05:30:24 -0700817
818 if (comparison.dropped) {
819 ++dropped_frames_;
820 return;
821 }
Stefan Holmer32e930f2018-02-09 09:43:45 +0100822 if (last_unfreeze_time_ms_ == 0)
823 last_unfreeze_time_ms_ = comparison.render_time_ms;
824 if (last_render_time_ != 0) {
825 const int64_t render_delta_ms =
826 comparison.render_time_ms - last_render_time_;
827 rendered_delta_.AddSample(render_delta_ms);
828 if (last_render_delta_ms_ != 0 &&
829 render_delta_ms - last_render_delta_ms_ > 150) {
830 time_between_freezes_.AddSample(last_render_time_ -
831 last_unfreeze_time_ms_);
832 last_unfreeze_time_ms_ = comparison.render_time_ms;
833 }
834 last_render_delta_ms_ = render_delta_ms;
835 }
ivica5d6a06c2015-09-17 05:30:24 -0700836 last_render_time_ = comparison.render_time_ms;
837
nissedf2ceb82016-12-15 06:29:53 -0800838 sender_time_.AddSample(comparison.send_time_ms - comparison.input_time_ms);
brandtr504b95e2016-12-21 02:54:35 -0800839 if (comparison.recv_time_ms > 0) {
840 // If recv_time_ms == 0, this frame consisted of a packets which were all
841 // lost in the transport. Since we were able to render the frame, however,
842 // the dropped packets were recovered by FlexFEC. The FlexFEC recovery
843 // happens internally in Call, and we can therefore here not know which
844 // FEC packets that protected the lost media packets. Consequently, we
845 // were not able to record a meaningful recv_time_ms. We therefore skip
846 // this sample.
847 //
848 // The reasoning above does not hold for ULPFEC and RTX, as for those
849 // strategies the timestamp of the received packets is set to the
850 // timestamp of the protected/retransmitted media packet. I.e., then
851 // recv_time_ms != 0, even though the media packets were lost.
852 receiver_time_.AddSample(comparison.render_time_ms -
853 comparison.recv_time_ms);
Ying Wang338bd622018-02-09 16:21:26 +0100854 network_time_.AddSample(comparison.recv_time_ms -
855 comparison.send_time_ms);
brandtr504b95e2016-12-21 02:54:35 -0800856 }
nissedf2ceb82016-12-15 06:29:53 -0800857 end_to_end_.AddSample(comparison.render_time_ms - comparison.input_time_ms);
ivica5d6a06c2015-09-17 05:30:24 -0700858 encoded_frame_size_.AddSample(comparison.encoded_frame_size);
859 }
860
861 void PrintResult(const char* result_type,
862 test::Statistics stats,
863 const char* unit) {
Edward Lemurd1d67432018-01-02 11:54:37 +0100864 test::PrintResultMeanAndError(result_type, "", test_label_.c_str(),
865 stats.Mean(), stats.StandardDeviation(), unit,
866 false);
ivica5d6a06c2015-09-17 05:30:24 -0700867 }
868
869 void PrintSamplesToFile(void) {
870 FILE* out = graph_data_output_file_;
871 rtc::CritScope crit(&comparison_lock_);
872 std::sort(samples_.begin(), samples_.end(),
873 [](const Sample& A, const Sample& B) -> bool {
874 return A.input_time_ms < B.input_time_ms;
875 });
876
sprangce4aef12015-11-02 07:23:20 -0800877 fprintf(out, "%s\n", graph_title_.c_str());
ivica5d6a06c2015-09-17 05:30:24 -0700878 fprintf(out, "%" PRIuS "\n", samples_.size());
879 fprintf(out,
880 "dropped "
881 "input_time_ms "
882 "send_time_ms "
883 "recv_time_ms "
ivica8d15bd62015-10-07 02:43:12 -0700884 "render_time_ms "
ivica5d6a06c2015-09-17 05:30:24 -0700885 "encoded_frame_size "
886 "psnr "
887 "ssim "
ivica8d15bd62015-10-07 02:43:12 -0700888 "encode_time_ms\n");
ivica5d6a06c2015-09-17 05:30:24 -0700889 for (const Sample& sample : samples_) {
Yves Gerey665174f2018-06-19 15:03:05 +0200890 fprintf(out,
891 "%d %" PRId64 " %" PRId64 " %" PRId64 " %" PRId64 " %" PRIuS
892 " %lf %lf\n",
ivica8d15bd62015-10-07 02:43:12 -0700893 sample.dropped, sample.input_time_ms, sample.send_time_ms,
894 sample.recv_time_ms, sample.render_time_ms,
Niels Möller6b642f72017-12-08 14:11:14 +0100895 sample.encoded_frame_size, sample.psnr, sample.ssim);
ivica5d6a06c2015-09-17 05:30:24 -0700896 }
897 }
898
ilnik1e7732c2017-02-23 05:07:56 -0800899 double GetAverageMediaBitrateBps() {
900 if (last_sending_time_ == first_sending_time_) {
901 return 0;
902 } else {
903 return static_cast<double>(total_media_bytes_) * 8 /
904 (last_sending_time_ - first_sending_time_) *
905 rtc::kNumMillisecsPerSec;
906 }
907 }
908
perkja49cbd32016-09-16 07:53:41 -0700909 // Implements VideoSinkInterface to receive captured frames from a
910 // FrameGeneratorCapturer. Implements VideoSourceInterface to be able to act
911 // as a source to VideoSendStream.
912 // It forwards all input frames to the VideoAnalyzer for later comparison and
913 // forwards the captured frames to the VideoSendStream.
914 class CapturedFrameForwarder : public rtc::VideoSinkInterface<VideoFrame>,
915 public rtc::VideoSourceInterface<VideoFrame> {
916 public:
ilnik6b826ef2017-06-16 06:53:48 -0700917 explicit CapturedFrameForwarder(VideoAnalyzer* analyzer, Clock* clock)
918 : analyzer_(analyzer),
919 send_stream_input_(nullptr),
920 video_capturer_(nullptr),
921 clock_(clock) {}
922
923 void SetSource(test::VideoCapturer* video_capturer) {
924 video_capturer_ = video_capturer;
925 }
perkja49cbd32016-09-16 07:53:41 -0700926
927 private:
928 void OnFrame(const VideoFrame& video_frame) override {
929 VideoFrame copy = video_frame;
ilnik3dd5ad92017-02-09 04:58:53 -0800930 // Frames from the capturer does not have a rtp timestamp.
931 // Create one so it can be used for comparison.
932 RTC_DCHECK_EQ(0, video_frame.timestamp());
ilnik6b826ef2017-06-16 06:53:48 -0700933 if (video_frame.ntp_time_ms() == 0)
934 copy.set_ntp_time_ms(clock_->CurrentNtpInMilliseconds());
perkja49cbd32016-09-16 07:53:41 -0700935 copy.set_timestamp(copy.ntp_time_ms() * 90);
ilnik3dd5ad92017-02-09 04:58:53 -0800936 analyzer_->AddCapturedFrameForComparison(copy);
perkja49cbd32016-09-16 07:53:41 -0700937 rtc::CritScope lock(&crit_);
938 if (send_stream_input_)
ilnikb82ac6a2017-05-02 00:48:41 -0700939 send_stream_input_->OnFrame(copy);
perkja49cbd32016-09-16 07:53:41 -0700940 }
941
942 // Called when |send_stream_.SetSource()| is called.
943 void AddOrUpdateSink(rtc::VideoSinkInterface<VideoFrame>* sink,
944 const rtc::VideoSinkWants& wants) override {
ilnik267041c2017-06-27 07:21:01 -0700945 {
946 rtc::CritScope lock(&crit_);
947 RTC_DCHECK(!send_stream_input_ || send_stream_input_ == sink);
948 send_stream_input_ = sink;
949 }
ilnik6b826ef2017-06-16 06:53:48 -0700950 if (video_capturer_) {
951 video_capturer_->AddOrUpdateSink(this, wants);
952 }
perkja49cbd32016-09-16 07:53:41 -0700953 }
954
955 // Called by |send_stream_| when |send_stream_.SetSource()| is called.
956 void RemoveSink(rtc::VideoSinkInterface<VideoFrame>* sink) override {
957 rtc::CritScope lock(&crit_);
958 RTC_DCHECK(sink == send_stream_input_);
959 send_stream_input_ = nullptr;
960 }
961
962 VideoAnalyzer* const analyzer_;
963 rtc::CriticalSection crit_;
danilchapa37de392017-09-09 04:17:22 -0700964 rtc::VideoSinkInterface<VideoFrame>* send_stream_input_
965 RTC_GUARDED_BY(crit_);
ilnik6b826ef2017-06-16 06:53:48 -0700966 test::VideoCapturer* video_capturer_;
967 Clock* clock_;
perkja49cbd32016-09-16 07:53:41 -0700968 };
969
970 void AddCapturedFrameForComparison(const VideoFrame& video_frame) {
971 rtc::CritScope lock(&crit_);
ilnik3dd5ad92017-02-09 04:58:53 -0800972 frames_.push_back(video_frame);
perkja49cbd32016-09-16 07:53:41 -0700973 }
974
stefan889d9652017-07-05 03:03:02 -0700975 Call* call_;
perkja49cbd32016-09-16 07:53:41 -0700976 VideoSendStream* send_stream_;
philipelfd870db2017-01-23 03:22:15 -0800977 VideoReceiveStream* receive_stream_;
perkja49cbd32016-09-16 07:53:41 -0700978 CapturedFrameForwarder captured_frame_forwarder_;
ivica5d6a06c2015-09-17 05:30:24 -0700979 const std::string test_label_;
980 FILE* const graph_data_output_file_;
sprangce4aef12015-11-02 07:23:20 -0800981 const std::string graph_title_;
982 const uint32_t ssrc_to_analyze_;
ilnik46a00212017-02-10 09:16:05 -0800983 const uint32_t rtx_ssrc_to_analyze_;
ilnikcb8c1462017-03-09 09:23:30 -0800984 const size_t selected_stream_;
ilnik1e7732c2017-02-23 05:07:56 -0800985 const int selected_sl_;
986 const int selected_tl_;
pbos14fe7082016-04-20 06:35:56 -0700987 PreEncodeProxy pre_encode_proxy_;
danilchapa37de392017-09-09 04:17:22 -0700988 std::vector<Sample> samples_ RTC_GUARDED_BY(comparison_lock_);
danilchapa37de392017-09-09 04:17:22 -0700989 test::Statistics sender_time_ RTC_GUARDED_BY(comparison_lock_);
990 test::Statistics receiver_time_ RTC_GUARDED_BY(comparison_lock_);
Ying Wang338bd622018-02-09 16:21:26 +0100991 test::Statistics network_time_ RTC_GUARDED_BY(comparison_lock_);
danilchapa37de392017-09-09 04:17:22 -0700992 test::Statistics psnr_ RTC_GUARDED_BY(comparison_lock_);
993 test::Statistics ssim_ RTC_GUARDED_BY(comparison_lock_);
994 test::Statistics end_to_end_ RTC_GUARDED_BY(comparison_lock_);
995 test::Statistics rendered_delta_ RTC_GUARDED_BY(comparison_lock_);
996 test::Statistics encoded_frame_size_ RTC_GUARDED_BY(comparison_lock_);
997 test::Statistics encode_frame_rate_ RTC_GUARDED_BY(comparison_lock_);
998 test::Statistics encode_time_ms_ RTC_GUARDED_BY(comparison_lock_);
999 test::Statistics encode_usage_percent_ RTC_GUARDED_BY(comparison_lock_);
1000 test::Statistics decode_time_ms_ RTC_GUARDED_BY(comparison_lock_);
1001 test::Statistics decode_time_max_ms_ RTC_GUARDED_BY(comparison_lock_);
1002 test::Statistics media_bitrate_bps_ RTC_GUARDED_BY(comparison_lock_);
1003 test::Statistics fec_bitrate_bps_ RTC_GUARDED_BY(comparison_lock_);
1004 test::Statistics send_bandwidth_bps_ RTC_GUARDED_BY(comparison_lock_);
1005 test::Statistics memory_usage_ RTC_GUARDED_BY(comparison_lock_);
Stefan Holmer32e930f2018-02-09 09:43:45 +01001006 test::Statistics time_between_freezes_ RTC_GUARDED_BY(comparison_lock_);
ilnikdaa574d2017-03-01 06:46:05 -08001007
ilnik59cac992017-07-25 05:45:03 -07001008 struct FrameWithPsnr {
1009 double psnr;
1010 VideoFrame frame;
1011 };
1012
1013 // Rendered frame with worst PSNR is saved for further analysis.
Danil Chapovalovb9b146c2018-06-15 12:28:07 +02001014 absl::optional<FrameWithPsnr> worst_frame_ RTC_GUARDED_BY(comparison_lock_);
ilnik59cac992017-07-25 05:45:03 -07001015
stefan889d9652017-07-05 03:03:02 -07001016 size_t last_fec_bytes_;
ivica5d6a06c2015-09-17 05:30:24 -07001017
1018 const int frames_to_process_;
1019 int frames_recorded_;
1020 int frames_processed_;
1021 int dropped_frames_;
pbos14fe7082016-04-20 06:35:56 -07001022 int dropped_frames_before_first_encode_;
1023 int dropped_frames_before_rendering_;
ivica5d6a06c2015-09-17 05:30:24 -07001024 int64_t last_render_time_;
Stefan Holmer32e930f2018-02-09 09:43:45 +01001025 int64_t last_render_delta_ms_;
1026 int64_t last_unfreeze_time_ms_;
ivica5d6a06c2015-09-17 05:30:24 -07001027 uint32_t rtp_timestamp_delta_;
ilnik1e7732c2017-02-23 05:07:56 -08001028 int64_t total_media_bytes_;
1029 int64_t first_sending_time_;
1030 int64_t last_sending_time_;
ivica5d6a06c2015-09-17 05:30:24 -07001031
danilchapa37de392017-09-09 04:17:22 -07001032 int64_t cpu_time_ RTC_GUARDED_BY(cpu_measurement_lock_);
1033 int64_t wallclock_time_ RTC_GUARDED_BY(cpu_measurement_lock_);
ilnikdf92c5c2017-02-23 02:08:44 -08001034 rtc::CriticalSection cpu_measurement_lock_;
1035
ivica5d6a06c2015-09-17 05:30:24 -07001036 rtc::CriticalSection crit_;
danilchapa37de392017-09-09 04:17:22 -07001037 std::deque<VideoFrame> frames_ RTC_GUARDED_BY(crit_);
Danil Chapovalovb9b146c2018-06-15 12:28:07 +02001038 absl::optional<VideoFrame> last_rendered_frame_ RTC_GUARDED_BY(crit_);
danilchapa37de392017-09-09 04:17:22 -07001039 rtc::TimestampWrapAroundHandler wrap_handler_ RTC_GUARDED_BY(crit_);
1040 std::map<int64_t, int64_t> send_times_ RTC_GUARDED_BY(crit_);
1041 std::map<int64_t, int64_t> recv_times_ RTC_GUARDED_BY(crit_);
1042 std::map<int64_t, size_t> encoded_frame_sizes_ RTC_GUARDED_BY(crit_);
Danil Chapovalovb9b146c2018-06-15 12:28:07 +02001043 absl::optional<uint32_t> first_encoded_timestamp_ RTC_GUARDED_BY(crit_);
1044 absl::optional<uint32_t> first_sent_timestamp_ RTC_GUARDED_BY(crit_);
ivica5d6a06c2015-09-17 05:30:24 -07001045 const double avg_psnr_threshold_;
1046 const double avg_ssim_threshold_;
ilnik9ae0d762017-02-15 00:53:12 -08001047 bool is_quick_test_enabled_;
ivica5d6a06c2015-09-17 05:30:24 -07001048
1049 rtc::CriticalSection comparison_lock_;
Peter Boström8c38e8b2015-11-26 17:45:47 +01001050 std::vector<rtc::PlatformThread*> comparison_thread_pool_;
1051 rtc::PlatformThread stats_polling_thread_;
Peter Boström5811a392015-12-10 13:02:50 +01001052 rtc::Event comparison_available_event_;
danilchapa37de392017-09-09 04:17:22 -07001053 std::deque<FrameComparison> comparisons_ RTC_GUARDED_BY(comparison_lock_);
Peter Boström5811a392015-12-10 13:02:50 +01001054 rtc::Event done_;
ilnik98436952017-07-13 00:47:03 -07001055
1056 std::unique_ptr<test::RtpFileWriter> rtp_file_writer_;
1057 Clock* const clock_;
1058 const int64_t start_ms_;
ivica5d6a06c2015-09-17 05:30:24 -07001059};
1060
Niels Möller4db138e2018-04-19 09:04:13 +02001061// Not used by these tests.
1062std::vector<SdpVideoFormat>
1063VideoQualityTest::TestVideoEncoderFactory::GetSupportedFormats() const {
1064 RTC_NOTREACHED();
1065 return {};
1066}
1067
1068VideoEncoderFactory::CodecInfo
1069VideoQualityTest::TestVideoEncoderFactory::QueryVideoEncoder(
1070 const SdpVideoFormat& format) const {
1071 CodecInfo codec_info;
1072 codec_info.is_hardware_accelerated = false;
1073 codec_info.has_internal_source = false;
1074 return codec_info;
1075}
1076
1077std::unique_ptr<VideoEncoder>
1078VideoQualityTest::TestVideoEncoderFactory::CreateVideoEncoder(
1079 const SdpVideoFormat& format) {
1080 if (format.name == "VP8") {
Ilya Nikolaevskiy97b4ee52018-05-28 10:24:22 +02001081 return rtc::MakeUnique<VP8EncoderSimulcastProxy>(&internal_encoder_factory_,
1082 format);
Niels Möller4db138e2018-04-19 09:04:13 +02001083 } else if (format.name == "multiplex") {
1084 return rtc::MakeUnique<MultiplexEncoderAdapter>(
1085 &internal_encoder_factory_, SdpVideoFormat(cricket::kVp9CodecName));
1086 } else {
1087 return internal_encoder_factory_.CreateVideoEncoder(format);
1088 }
1089}
1090
palmkviste75f2042016-09-28 06:19:48 -07001091VideoQualityTest::VideoQualityTest()
minyue20c84cc2017-04-10 16:57:57 -07001092 : clock_(Clock::GetRealTimeClock()), receive_logs_(0), send_logs_(0) {
1093 payload_type_map_ = test::CallTest::payload_type_map_;
1094 RTC_DCHECK(payload_type_map_.find(kPayloadTypeH264) ==
1095 payload_type_map_.end());
1096 RTC_DCHECK(payload_type_map_.find(kPayloadTypeVP8) ==
1097 payload_type_map_.end());
1098 RTC_DCHECK(payload_type_map_.find(kPayloadTypeVP9) ==
1099 payload_type_map_.end());
1100 payload_type_map_[kPayloadTypeH264] = webrtc::MediaType::VIDEO;
1101 payload_type_map_[kPayloadTypeVP8] = webrtc::MediaType::VIDEO;
1102 payload_type_map_[kPayloadTypeVP9] = webrtc::MediaType::VIDEO;
1103}
ivica5d6a06c2015-09-17 05:30:24 -07001104
Ying Wang3b790f32018-01-19 17:58:57 +01001105VideoQualityTest::VideoQualityTest(
Ying Wang0dd1b0a2018-02-20 12:50:27 +01001106 std::unique_ptr<FecControllerFactoryInterface> fec_controller_factory)
Ying Wang3b790f32018-01-19 17:58:57 +01001107 : VideoQualityTest() {
Ying Wang0dd1b0a2018-02-20 12:50:27 +01001108 fec_controller_factory_ = std::move(fec_controller_factory);
Ying Wang3b790f32018-01-19 17:58:57 +01001109}
1110
minyue626bc952016-10-31 05:47:02 -07001111VideoQualityTest::Params::Params()
Sebastian Janssonfc8d26b2018-02-21 09:52:06 +01001112 : call({false, BitrateConstraints(), 0}),
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001113 video{{false, 640, 480, 30, 50, 800, 800, false, "VP8", 1, -1, 0, false,
Niels Möller6aa415e2018-06-07 11:14:13 +02001114 false, false, ""},
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001115 {false, 640, 480, 30, 50, 800, 800, false, "VP8", 1, -1, 0, false,
Niels Möller6aa415e2018-06-07 11:14:13 +02001116 false, false, ""}},
minyue4c8b9422017-03-21 04:11:43 -07001117 audio({false, false, false}),
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001118 screenshare{{false, false, 10, 0}, {false, false, 10, 0}},
minyue626bc952016-10-31 05:47:02 -07001119 analyzer({"", 0.0, 0.0, 0, "", ""}),
1120 pipe(),
Sergey Silkin57027362018-05-15 09:12:05 +02001121 ss{{std::vector<VideoStream>(), 0, 0, -1, InterLayerPredMode::kOn,
1122 std::vector<SpatialLayer>()},
1123 {std::vector<VideoStream>(), 0, 0, -1, InterLayerPredMode::kOn,
1124 std::vector<SpatialLayer>()}},
ilnik98436952017-07-13 00:47:03 -07001125 logging({false, "", "", ""}) {}
minyue626bc952016-10-31 05:47:02 -07001126
1127VideoQualityTest::Params::~Params() = default;
1128
ivica5d6a06c2015-09-17 05:30:24 -07001129void VideoQualityTest::TestBody() {}
1130
sprangce4aef12015-11-02 07:23:20 -08001131std::string VideoQualityTest::GenerateGraphTitle() const {
1132 std::stringstream ss;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001133 ss << params_.video[0].codec;
1134 ss << " (" << params_.video[0].target_bitrate_bps / 1000 << "kbps";
1135 ss << ", " << params_.video[0].fps << " FPS";
1136 if (params_.screenshare[0].scroll_duration)
1137 ss << ", " << params_.screenshare[0].scroll_duration << "s scroll";
1138 if (params_.ss[0].streams.size() > 1)
1139 ss << ", Stream #" << params_.ss[0].selected_stream;
1140 if (params_.ss[0].num_spatial_layers > 1)
1141 ss << ", Layer #" << params_.ss[0].selected_sl;
sprangce4aef12015-11-02 07:23:20 -08001142 ss << ")";
1143 return ss.str();
1144}
1145
1146void VideoQualityTest::CheckParams() {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001147 for (size_t video_idx = 0; video_idx < num_video_streams_; ++video_idx) {
1148 // Iterate over primary and secondary video streams.
1149 if (!params_.video[video_idx].enabled)
1150 return;
1151 // Add a default stream in none specified.
1152 if (params_.ss[video_idx].streams.empty())
1153 params_.ss[video_idx].streams.push_back(
1154 VideoQualityTest::DefaultVideoStream(params_, video_idx));
1155 if (params_.ss[video_idx].num_spatial_layers == 0)
1156 params_.ss[video_idx].num_spatial_layers = 1;
sprangce4aef12015-11-02 07:23:20 -08001157
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001158 if (params_.pipe.loss_percent != 0 ||
1159 params_.pipe.queue_length_packets != 0) {
1160 // Since LayerFilteringTransport changes the sequence numbers, we can't
1161 // use that feature with pack loss, since the NACK request would end up
1162 // retransmitting the wrong packets.
1163 RTC_CHECK(params_.ss[video_idx].selected_sl == -1 ||
1164 params_.ss[video_idx].selected_sl ==
1165 params_.ss[video_idx].num_spatial_layers - 1);
1166 RTC_CHECK(params_.video[video_idx].selected_tl == -1 ||
1167 params_.video[video_idx].selected_tl ==
1168 params_.video[video_idx].num_temporal_layers - 1);
1169 }
sprangce4aef12015-11-02 07:23:20 -08001170
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001171 // TODO(ivica): Should max_bitrate_bps == -1 represent inf max bitrate, as
1172 // it does in some parts of the code?
1173 RTC_CHECK_GE(params_.video[video_idx].max_bitrate_bps,
1174 params_.video[video_idx].target_bitrate_bps);
1175 RTC_CHECK_GE(params_.video[video_idx].target_bitrate_bps,
1176 params_.video[video_idx].min_bitrate_bps);
1177 RTC_CHECK_LT(params_.video[video_idx].selected_tl,
1178 params_.video[video_idx].num_temporal_layers);
1179 RTC_CHECK_LE(params_.ss[video_idx].selected_stream,
1180 params_.ss[video_idx].streams.size());
1181 for (const VideoStream& stream : params_.ss[video_idx].streams) {
1182 RTC_CHECK_GE(stream.min_bitrate_bps, 0);
1183 RTC_CHECK_GE(stream.target_bitrate_bps, stream.min_bitrate_bps);
1184 RTC_CHECK_GE(stream.max_bitrate_bps, stream.target_bitrate_bps);
1185 }
1186 // TODO(ivica): Should we check if the sum of all streams/layers is equal to
1187 // the total bitrate? We anyway have to update them in the case bitrate
1188 // estimator changes the total bitrates.
1189 RTC_CHECK_GE(params_.ss[video_idx].num_spatial_layers, 1);
1190 RTC_CHECK_LE(params_.ss[video_idx].selected_sl,
1191 params_.ss[video_idx].num_spatial_layers);
1192 RTC_CHECK(
1193 params_.ss[video_idx].spatial_layers.empty() ||
1194 params_.ss[video_idx].spatial_layers.size() ==
1195 static_cast<size_t>(params_.ss[video_idx].num_spatial_layers));
1196 if (params_.video[video_idx].codec == "VP8") {
1197 RTC_CHECK_EQ(params_.ss[video_idx].num_spatial_layers, 1);
1198 } else if (params_.video[video_idx].codec == "VP9") {
1199 RTC_CHECK_EQ(params_.ss[video_idx].streams.size(), 1);
1200 }
1201 RTC_CHECK_GE(params_.call.num_thumbnails, 0);
1202 if (params_.call.num_thumbnails > 0) {
1203 RTC_CHECK_EQ(params_.ss[video_idx].num_spatial_layers, 1);
1204 RTC_CHECK_EQ(params_.ss[video_idx].streams.size(), 3);
1205 RTC_CHECK_EQ(params_.video[video_idx].num_temporal_layers, 3);
1206 RTC_CHECK_EQ(params_.video[video_idx].codec, "VP8");
1207 }
1208 // Dual streams with FEC not supported in tests yet.
1209 RTC_CHECK(!params_.video[video_idx].flexfec || num_video_streams_ == 1);
1210 RTC_CHECK(!params_.video[video_idx].ulpfec || num_video_streams_ == 1);
ilnika014cc52017-03-07 04:21:04 -08001211 }
sprangce4aef12015-11-02 07:23:20 -08001212}
1213
1214// Static.
1215std::vector<int> VideoQualityTest::ParseCSV(const std::string& str) {
1216 // Parse comma separated nonnegative integers, where some elements may be
1217 // empty. The empty values are replaced with -1.
1218 // E.g. "10,-20,,30,40" --> {10, 20, -1, 30,40}
1219 // E.g. ",,10,,20," --> {-1, -1, 10, -1, 20, -1}
1220 std::vector<int> result;
1221 if (str.empty())
1222 return result;
1223
1224 const char* p = str.c_str();
1225 int value = -1;
1226 int pos;
1227 while (*p) {
1228 if (*p == ',') {
1229 result.push_back(value);
1230 value = -1;
1231 ++p;
1232 continue;
1233 }
1234 RTC_CHECK_EQ(sscanf(p, "%d%n", &value, &pos), 1)
1235 << "Unexpected non-number value.";
1236 p += pos;
1237 }
1238 result.push_back(value);
1239 return result;
1240}
1241
1242// Static.
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001243VideoStream VideoQualityTest::DefaultVideoStream(const Params& params,
1244 size_t video_idx) {
sprangce4aef12015-11-02 07:23:20 -08001245 VideoStream stream;
Danil Chapovalov350531e2018-06-08 11:04:04 +00001246 stream.width = params.video[video_idx].width;
1247 stream.height = params.video[video_idx].height;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001248 stream.max_framerate = params.video[video_idx].fps;
1249 stream.min_bitrate_bps = params.video[video_idx].min_bitrate_bps;
1250 stream.target_bitrate_bps = params.video[video_idx].target_bitrate_bps;
1251 stream.max_bitrate_bps = params.video[video_idx].max_bitrate_bps;
sprang1168fd42017-06-21 09:00:17 -07001252 stream.max_qp = kDefaultMaxQp;
Sergey Silkina796a7e2018-03-01 15:11:29 +01001253 stream.num_temporal_layers = params.video[video_idx].num_temporal_layers;
Seth Hampson46e31ba2018-01-18 10:39:54 -08001254 stream.active = true;
ilnika014cc52017-03-07 04:21:04 -08001255 return stream;
1256}
1257
1258// Static.
1259VideoStream VideoQualityTest::DefaultThumbnailStream() {
1260 VideoStream stream;
1261 stream.width = 320;
1262 stream.height = 180;
1263 stream.max_framerate = 7;
1264 stream.min_bitrate_bps = 7500;
1265 stream.target_bitrate_bps = 37500;
1266 stream.max_bitrate_bps = 50000;
sprang1168fd42017-06-21 09:00:17 -07001267 stream.max_qp = kDefaultMaxQp;
sprangce4aef12015-11-02 07:23:20 -08001268 return stream;
1269}
1270
1271// Static.
1272void VideoQualityTest::FillScalabilitySettings(
1273 Params* params,
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001274 size_t video_idx,
sprangce4aef12015-11-02 07:23:20 -08001275 const std::vector<std::string>& stream_descriptors,
sprang1168fd42017-06-21 09:00:17 -07001276 int num_streams,
sprangce4aef12015-11-02 07:23:20 -08001277 size_t selected_stream,
1278 int num_spatial_layers,
1279 int selected_sl,
Sergey Silkin57027362018-05-15 09:12:05 +02001280 InterLayerPredMode inter_layer_pred,
sprangce4aef12015-11-02 07:23:20 -08001281 const std::vector<std::string>& sl_descriptors) {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001282 if (params->ss[video_idx].streams.empty() &&
1283 params->ss[video_idx].infer_streams) {
sprang1168fd42017-06-21 09:00:17 -07001284 webrtc::VideoEncoderConfig encoder_config;
Niels Möller259a4972018-04-05 15:36:51 +02001285 encoder_config.codec_type =
1286 PayloadStringToCodecType(params->video[video_idx].codec);
sprang1168fd42017-06-21 09:00:17 -07001287 encoder_config.content_type =
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001288 params->screenshare[video_idx].enabled
sprang1168fd42017-06-21 09:00:17 -07001289 ? webrtc::VideoEncoderConfig::ContentType::kScreen
1290 : webrtc::VideoEncoderConfig::ContentType::kRealtimeVideo;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001291 encoder_config.max_bitrate_bps = params->video[video_idx].max_bitrate_bps;
1292 encoder_config.min_transmit_bitrate_bps =
1293 params->video[video_idx].min_transmit_bps;
sprang1168fd42017-06-21 09:00:17 -07001294 encoder_config.number_of_streams = num_streams;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001295 encoder_config.spatial_layers = params->ss[video_idx].spatial_layers;
Seth Hampson8234ead2018-02-02 15:16:24 -08001296 encoder_config.simulcast_layers = std::vector<VideoStream>(num_streams);
sprang1168fd42017-06-21 09:00:17 -07001297 encoder_config.video_stream_factory =
1298 new rtc::RefCountedObject<cricket::EncoderStreamFactory>(
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001299 params->video[video_idx].codec, kDefaultMaxQp,
1300 params->video[video_idx].fps,
1301 params->screenshare[video_idx].enabled, true);
1302 params->ss[video_idx].streams =
sprang1168fd42017-06-21 09:00:17 -07001303 encoder_config.video_stream_factory->CreateEncoderStreams(
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001304 static_cast<int>(params->video[video_idx].width),
1305 static_cast<int>(params->video[video_idx].height), encoder_config);
sprang1168fd42017-06-21 09:00:17 -07001306 } else {
1307 // Read VideoStream and SpatialLayer elements from a list of comma separated
1308 // lists. To use a default value for an element, use -1 or leave empty.
1309 // Validity checks performed in CheckParams.
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001310 RTC_CHECK(params->ss[video_idx].streams.empty());
sprang1168fd42017-06-21 09:00:17 -07001311 for (auto descriptor : stream_descriptors) {
1312 if (descriptor.empty())
1313 continue;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001314 VideoStream stream =
1315 VideoQualityTest::DefaultVideoStream(*params, video_idx);
sprang1168fd42017-06-21 09:00:17 -07001316 std::vector<int> v = VideoQualityTest::ParseCSV(descriptor);
1317 if (v[0] != -1)
1318 stream.width = static_cast<size_t>(v[0]);
1319 if (v[1] != -1)
1320 stream.height = static_cast<size_t>(v[1]);
1321 if (v[2] != -1)
1322 stream.max_framerate = v[2];
1323 if (v[3] != -1)
1324 stream.min_bitrate_bps = v[3];
1325 if (v[4] != -1)
1326 stream.target_bitrate_bps = v[4];
1327 if (v[5] != -1)
1328 stream.max_bitrate_bps = v[5];
1329 if (v.size() > 6 && v[6] != -1)
1330 stream.max_qp = v[6];
Sergey Silkina796a7e2018-03-01 15:11:29 +01001331 if (v.size() > 7 && v[7] != -1) {
1332 stream.num_temporal_layers = v[7];
sprang1168fd42017-06-21 09:00:17 -07001333 } else {
1334 // Automatic TL thresholds for more than two layers not supported.
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001335 RTC_CHECK_LE(params->video[video_idx].num_temporal_layers, 2);
sprang1168fd42017-06-21 09:00:17 -07001336 }
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001337 params->ss[video_idx].streams.push_back(stream);
sprangce4aef12015-11-02 07:23:20 -08001338 }
sprangce4aef12015-11-02 07:23:20 -08001339 }
sprangce4aef12015-11-02 07:23:20 -08001340
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001341 params->ss[video_idx].num_spatial_layers = std::max(1, num_spatial_layers);
1342 params->ss[video_idx].selected_stream = selected_stream;
sprang1168fd42017-06-21 09:00:17 -07001343
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001344 params->ss[video_idx].selected_sl = selected_sl;
Sergey Silkin57027362018-05-15 09:12:05 +02001345 params->ss[video_idx].inter_layer_pred = inter_layer_pred;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001346 RTC_CHECK(params->ss[video_idx].spatial_layers.empty());
sprangce4aef12015-11-02 07:23:20 -08001347 for (auto descriptor : sl_descriptors) {
1348 if (descriptor.empty())
1349 continue;
1350 std::vector<int> v = VideoQualityTest::ParseCSV(descriptor);
Sergey Silkin13e74342018-03-02 12:28:00 +01001351 RTC_CHECK_EQ(v.size(), 7);
sprangce4aef12015-11-02 07:23:20 -08001352
Sergey Silkin13e74342018-03-02 12:28:00 +01001353 SpatialLayer layer = {0};
1354 layer.width = v[0];
1355 layer.height = v[1];
1356 layer.numberOfTemporalLayers = v[2];
1357 layer.maxBitrate = v[3];
1358 layer.minBitrate = v[4];
1359 layer.targetBitrate = v[5];
1360 layer.qpMax = v[6];
1361 layer.active = true;
1362
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001363 params->ss[video_idx].spatial_layers.push_back(layer);
sprangce4aef12015-11-02 07:23:20 -08001364 }
1365}
1366
minyuea27172d2016-11-01 05:59:29 -07001367void VideoQualityTest::SetupVideo(Transport* send_transport,
1368 Transport* recv_transport) {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001369 size_t total_streams_used = 0;
1370 size_t num_flexfec_streams = params_.video[0].flexfec ? 1 : 0;
1371 CreateAudioAndFecSendConfigs(0, num_flexfec_streams, send_transport);
1372 CreateMatchingAudioAndFecConfigs(recv_transport);
1373 video_receive_configs_.clear();
1374 video_send_configs_.clear();
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001375 video_encoder_configs_.clear();
1376 allocated_decoders_.clear();
1377 bool decode_all_receive_streams = true;
1378 size_t num_video_substreams = params_.ss[0].streams.size();
1379 RTC_CHECK(num_video_streams_ > 0);
1380 video_encoder_configs_.resize(num_video_streams_);
1381 for (size_t video_idx = 0; video_idx < num_video_streams_; ++video_idx) {
1382 video_send_configs_.push_back(VideoSendStream::Config(send_transport));
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001383 video_encoder_configs_.push_back(VideoEncoderConfig());
1384 num_video_substreams = params_.ss[video_idx].streams.size();
1385 RTC_CHECK_GT(num_video_substreams, 0);
1386 CreateVideoSendConfig(&video_send_configs_[video_idx], num_video_substreams,
1387 total_streams_used, send_transport);
ivica5d6a06c2015-09-17 05:30:24 -07001388
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001389 int payload_type;
1390 if (params_.video[video_idx].codec == "H264") {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001391 payload_type = kPayloadTypeH264;
1392 } else if (params_.video[video_idx].codec == "VP8") {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001393 payload_type = kPayloadTypeVP8;
1394 } else if (params_.video[video_idx].codec == "VP9") {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001395 payload_type = kPayloadTypeVP9;
Emircan Uysaler03e6ec92018-03-09 15:03:26 -08001396 } else if (params_.video[video_idx].codec == "multiplex") {
Emircan Uysaler03e6ec92018-03-09 15:03:26 -08001397 payload_type = kPayloadTypeVP9;
ilnikcb8c1462017-03-09 09:23:30 -08001398 } else {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001399 RTC_NOTREACHED() << "Codec not supported!";
1400 return;
ilnikcb8c1462017-03-09 09:23:30 -08001401 }
Niels Möller4db138e2018-04-19 09:04:13 +02001402 video_send_configs_[video_idx].encoder_settings.encoder_factory =
1403 &video_encoder_factory_;
1404
Niels Möller259a4972018-04-05 15:36:51 +02001405 video_send_configs_[video_idx].rtp.payload_name =
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001406 params_.video[video_idx].codec;
Niels Möller259a4972018-04-05 15:36:51 +02001407 video_send_configs_[video_idx].rtp.payload_type = payload_type;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001408 video_send_configs_[video_idx].rtp.nack.rtp_history_ms = kNackRtpHistoryMs;
1409 video_send_configs_[video_idx].rtp.rtx.payload_type = kSendRtxPayloadType;
1410 for (size_t i = 0; i < num_video_substreams; ++i) {
1411 video_send_configs_[video_idx].rtp.rtx.ssrcs.push_back(
1412 kSendRtxSsrcs[i + total_streams_used]);
ilnik9fd9f6c2017-03-02 08:10:10 -08001413 }
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001414 video_send_configs_[video_idx].rtp.extensions.clear();
1415 if (params_.call.send_side_bwe) {
1416 video_send_configs_[video_idx].rtp.extensions.push_back(
1417 RtpExtension(RtpExtension::kTransportSequenceNumberUri,
1418 test::kTransportSequenceNumberExtensionId));
1419 } else {
1420 video_send_configs_[video_idx].rtp.extensions.push_back(RtpExtension(
1421 RtpExtension::kAbsSendTimeUri, test::kAbsSendTimeExtensionId));
1422 }
1423 video_send_configs_[video_idx].rtp.extensions.push_back(
1424 RtpExtension(RtpExtension::kVideoContentTypeUri,
1425 test::kVideoContentTypeExtensionId));
1426 video_send_configs_[video_idx].rtp.extensions.push_back(RtpExtension(
1427 RtpExtension::kVideoTimingUri, test::kVideoTimingExtensionId));
1428
Niels Möller4db138e2018-04-19 09:04:13 +02001429 video_encoder_configs_[video_idx].video_format.name =
1430 params_.video[video_idx].codec;
1431
Niels Möller259a4972018-04-05 15:36:51 +02001432 video_encoder_configs_[video_idx].codec_type =
1433 PayloadStringToCodecType(params_.video[video_idx].codec);
1434
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001435 video_encoder_configs_[video_idx].min_transmit_bitrate_bps =
1436 params_.video[video_idx].min_transmit_bps;
1437
1438 video_send_configs_[video_idx].suspend_below_min_bitrate =
1439 params_.video[video_idx].suspend_below_min_bitrate;
1440
1441 video_encoder_configs_[video_idx].number_of_streams =
1442 params_.ss[video_idx].streams.size();
1443 video_encoder_configs_[video_idx].max_bitrate_bps = 0;
1444 for (size_t i = 0; i < params_.ss[video_idx].streams.size(); ++i) {
1445 video_encoder_configs_[video_idx].max_bitrate_bps +=
1446 params_.ss[video_idx].streams[i].max_bitrate_bps;
1447 }
1448 if (params_.ss[video_idx].infer_streams) {
Seth Hampson8234ead2018-02-02 15:16:24 -08001449 video_encoder_configs_[video_idx].simulcast_layers =
1450 std::vector<VideoStream>(params_.ss[video_idx].streams.size());
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001451 video_encoder_configs_[video_idx].video_stream_factory =
1452 new rtc::RefCountedObject<cricket::EncoderStreamFactory>(
1453 params_.video[video_idx].codec,
1454 params_.ss[video_idx].streams[0].max_qp,
1455 params_.video[video_idx].fps,
1456 params_.screenshare[video_idx].enabled, true);
1457 } else {
1458 video_encoder_configs_[video_idx].video_stream_factory =
1459 new rtc::RefCountedObject<VideoStreamFactory>(
1460 params_.ss[video_idx].streams);
1461 }
1462
1463 video_encoder_configs_[video_idx].spatial_layers =
1464 params_.ss[video_idx].spatial_layers;
1465
1466 std::vector<VideoReceiveStream::Config> new_receive_configs =
1467 CreateMatchingVideoReceiveConfigs(video_send_configs_[video_idx],
1468 recv_transport);
1469
1470 decode_all_receive_streams = params_.ss[video_idx].selected_stream ==
1471 params_.ss[video_idx].streams.size();
1472
1473 for (size_t i = 0; i < num_video_substreams; ++i) {
1474 new_receive_configs[i].rtp.nack.rtp_history_ms = kNackRtpHistoryMs;
1475 new_receive_configs[i].rtp.rtx_ssrc =
1476 kSendRtxSsrcs[i + total_streams_used];
1477 new_receive_configs[i]
1478 .rtp.rtx_associated_payload_types[kSendRtxPayloadType] = payload_type;
1479 new_receive_configs[i].rtp.transport_cc = params_.call.send_side_bwe;
1480 new_receive_configs[i].rtp.remb = !params_.call.send_side_bwe;
1481 // Enable RTT calculation so NTP time estimator will work.
1482 new_receive_configs[i].rtp.rtcp_xr.receiver_reference_time_report = true;
1483 // Force fake decoders on non-selected simulcast streams.
1484 if (!decode_all_receive_streams &&
1485 i != params_.ss[video_idx].selected_stream) {
1486 VideoReceiveStream::Decoder decoder;
1487 decoder.decoder = new test::FakeDecoder();
Niels Möller259a4972018-04-05 15:36:51 +02001488 decoder.payload_type = video_send_configs_[video_idx].rtp.payload_type;
1489 decoder.payload_name = video_send_configs_[video_idx].rtp.payload_name;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001490 new_receive_configs[i].decoders.clear();
1491 allocated_decoders_.emplace_back(decoder.decoder);
1492 new_receive_configs[i].decoders.push_back(decoder);
1493 }
1494 }
1495
1496 for (VideoReceiveStream::Config& config : new_receive_configs) {
1497 video_receive_configs_.push_back(config.Copy());
1498 }
1499
1500 if (params_.screenshare[video_idx].enabled) {
1501 // Fill out codec settings.
1502 video_encoder_configs_[video_idx].content_type =
1503 VideoEncoderConfig::ContentType::kScreen;
Taylor Brandstetter49fcc102018-05-16 14:20:41 -07001504 degradation_preference_ = DegradationPreference::MAINTAIN_RESOLUTION;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001505 if (params_.video[video_idx].codec == "VP8") {
1506 VideoCodecVP8 vp8_settings = VideoEncoder::GetDefaultVp8Settings();
1507 vp8_settings.denoisingOn = false;
1508 vp8_settings.frameDroppingOn = false;
1509 vp8_settings.numberOfTemporalLayers = static_cast<unsigned char>(
1510 params_.video[video_idx].num_temporal_layers);
1511 video_encoder_configs_[video_idx].encoder_specific_settings =
1512 new rtc::RefCountedObject<
1513 VideoEncoderConfig::Vp8EncoderSpecificSettings>(vp8_settings);
1514 } else if (params_.video[video_idx].codec == "VP9") {
1515 VideoCodecVP9 vp9_settings = VideoEncoder::GetDefaultVp9Settings();
1516 vp9_settings.denoisingOn = false;
1517 vp9_settings.frameDroppingOn = false;
1518 vp9_settings.numberOfTemporalLayers = static_cast<unsigned char>(
1519 params_.video[video_idx].num_temporal_layers);
1520 vp9_settings.numberOfSpatialLayers = static_cast<unsigned char>(
1521 params_.ss[video_idx].num_spatial_layers);
Sergey Silkin57027362018-05-15 09:12:05 +02001522 vp9_settings.interLayerPred = params_.ss[video_idx].inter_layer_pred;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001523 video_encoder_configs_[video_idx].encoder_specific_settings =
1524 new rtc::RefCountedObject<
1525 VideoEncoderConfig::Vp9EncoderSpecificSettings>(vp9_settings);
1526 }
1527 } else if (params_.ss[video_idx].num_spatial_layers > 1) {
1528 // If SVC mode without screenshare, still need to set codec specifics.
1529 RTC_CHECK(params_.video[video_idx].codec == "VP9");
1530 VideoCodecVP9 vp9_settings = VideoEncoder::GetDefaultVp9Settings();
1531 vp9_settings.numberOfTemporalLayers = static_cast<unsigned char>(
1532 params_.video[video_idx].num_temporal_layers);
1533 vp9_settings.numberOfSpatialLayers =
1534 static_cast<unsigned char>(params_.ss[video_idx].num_spatial_layers);
Sergey Silkin57027362018-05-15 09:12:05 +02001535 vp9_settings.interLayerPred = params_.ss[video_idx].inter_layer_pred;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001536 video_encoder_configs_[video_idx].encoder_specific_settings =
1537 new rtc::RefCountedObject<
1538 VideoEncoderConfig::Vp9EncoderSpecificSettings>(vp9_settings);
Niels Möller6aa415e2018-06-07 11:14:13 +02001539 } else if (params_.video[video_idx].automatic_scaling) {
1540 if (params_.video[video_idx].codec == "VP8") {
1541 VideoCodecVP8 vp8_settings = VideoEncoder::GetDefaultVp8Settings();
1542 vp8_settings.automaticResizeOn = true;
1543 video_encoder_configs_[video_idx].encoder_specific_settings =
1544 new rtc::RefCountedObject<
1545 VideoEncoderConfig::Vp8EncoderSpecificSettings>(vp8_settings);
1546 } else if (params_.video[video_idx].codec == "VP9") {
1547 VideoCodecVP9 vp9_settings = VideoEncoder::GetDefaultVp9Settings();
1548 vp9_settings.automaticResizeOn = true;
1549 video_encoder_configs_[video_idx].encoder_specific_settings =
1550 new rtc::RefCountedObject<
1551 VideoEncoderConfig::Vp9EncoderSpecificSettings>(vp9_settings);
1552 } else {
1553 RTC_NOTREACHED() << "Automatic scaling not supported for codec "
Yves Gerey665174f2018-06-19 15:03:05 +02001554 << params_.video[video_idx].codec << ", stream "
1555 << video_idx;
Niels Möller6aa415e2018-06-07 11:14:13 +02001556 }
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001557 }
1558 total_streams_used += num_video_substreams;
sprangce4aef12015-11-02 07:23:20 -08001559 }
brandtr1293aca2016-11-16 22:47:29 -08001560
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001561 // FEC supported only for single video stream mode yet.
1562 if (params_.video[0].flexfec) {
Ilya Nikolaevskiy38aaf692017-12-22 14:39:09 +01001563 video_send_configs_[0].rtp.flexfec.payload_type = kFlexfecPayloadType;
1564 video_send_configs_[0].rtp.flexfec.ssrc = kFlexfecSendSsrc;
sprang1168fd42017-06-21 09:00:17 -07001565 if (decode_all_receive_streams) {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001566 for (uint32_t media_ssrc : video_send_configs_[0].rtp.ssrcs) {
1567 video_send_configs_[0].rtp.flexfec.protected_media_ssrcs.push_back(
sprang1168fd42017-06-21 09:00:17 -07001568 media_ssrc);
1569 }
1570 } else {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001571 video_send_configs_[0].rtp.flexfec.protected_media_ssrcs = {
1572 kVideoSendSsrcs[params_.ss[0].selected_stream]};
sprang1168fd42017-06-21 09:00:17 -07001573 }
brandtr1293aca2016-11-16 22:47:29 -08001574
brandtr8313a6f2017-01-13 07:41:19 -08001575 // The matching receive config is _not_ created by
1576 // CreateMatchingReceiveConfigs, since VideoQualityTest is not a BaseTest.
1577 // Set up the receive config manually instead.
1578 FlexfecReceiveStream::Config flexfec_receive_config(recv_transport);
brandtr1cfbd602016-12-08 04:17:53 -08001579 flexfec_receive_config.payload_type =
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001580 video_send_configs_[0].rtp.flexfec.payload_type;
1581 flexfec_receive_config.remote_ssrc =
1582 video_send_configs_[0].rtp.flexfec.ssrc;
brandtr1293aca2016-11-16 22:47:29 -08001583 flexfec_receive_config.protected_media_ssrcs =
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001584 video_send_configs_[0].rtp.flexfec.protected_media_ssrcs;
brandtrfa5a3682017-01-17 01:33:54 -08001585 flexfec_receive_config.local_ssrc = kReceiverLocalVideoSsrc;
brandtrb29e6522016-12-21 06:37:18 -08001586 flexfec_receive_config.transport_cc = params_.call.send_side_bwe;
1587 if (params_.call.send_side_bwe) {
1588 flexfec_receive_config.rtp_header_extensions.push_back(
1589 RtpExtension(RtpExtension::kTransportSequenceNumberUri,
1590 test::kTransportSequenceNumberExtensionId));
1591 } else {
1592 flexfec_receive_config.rtp_header_extensions.push_back(RtpExtension(
1593 RtpExtension::kAbsSendTimeUri, test::kAbsSendTimeExtensionId));
1594 }
brandtr1293aca2016-11-16 22:47:29 -08001595 flexfec_receive_configs_.push_back(flexfec_receive_config);
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001596 if (num_video_substreams > 0) {
brandtr7cd28b92017-09-22 00:26:25 -07001597 video_receive_configs_[0].rtp.protected_by_flexfec = true;
1598 }
brandtr1293aca2016-11-16 22:47:29 -08001599 }
1600
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001601 if (params_.video[0].ulpfec) {
1602 video_send_configs_[0].rtp.ulpfec.red_payload_type = kRedPayloadType;
1603 video_send_configs_[0].rtp.ulpfec.ulpfec_payload_type = kUlpfecPayloadType;
1604 video_send_configs_[0].rtp.ulpfec.red_rtx_payload_type = kRtxRedPayloadType;
brandtr1293aca2016-11-16 22:47:29 -08001605
sprang1168fd42017-06-21 09:00:17 -07001606 if (decode_all_receive_streams) {
1607 for (auto it = video_receive_configs_.begin();
1608 it != video_receive_configs_.end(); ++it) {
nisse3b3622f2017-09-26 02:49:21 -07001609 it->rtp.red_payload_type =
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001610 video_send_configs_[0].rtp.ulpfec.red_payload_type;
nisse3b3622f2017-09-26 02:49:21 -07001611 it->rtp.ulpfec_payload_type =
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001612 video_send_configs_[0].rtp.ulpfec.ulpfec_payload_type;
1613 it->rtp.rtx_associated_payload_types
1614 [video_send_configs_[0].rtp.ulpfec.red_rtx_payload_type] =
1615 video_send_configs_[0].rtp.ulpfec.red_payload_type;
sprang1168fd42017-06-21 09:00:17 -07001616 }
1617 } else {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001618 video_receive_configs_[params_.ss[0].selected_stream]
1619 .rtp.red_payload_type =
1620 video_send_configs_[0].rtp.ulpfec.red_payload_type;
1621 video_receive_configs_[params_.ss[0].selected_stream]
nisse3b3622f2017-09-26 02:49:21 -07001622 .rtp.ulpfec_payload_type =
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001623 video_send_configs_[0].rtp.ulpfec.ulpfec_payload_type;
1624 video_receive_configs_[params_.ss[0].selected_stream]
1625 .rtp.rtx_associated_payload_types
1626 [video_send_configs_[0].rtp.ulpfec.red_rtx_payload_type] =
1627 video_send_configs_[0].rtp.ulpfec.red_payload_type;
sprang1168fd42017-06-21 09:00:17 -07001628 }
brandtr1293aca2016-11-16 22:47:29 -08001629 }
ivica5d6a06c2015-09-17 05:30:24 -07001630}
1631
ilnika014cc52017-03-07 04:21:04 -08001632void VideoQualityTest::SetupThumbnails(Transport* send_transport,
1633 Transport* recv_transport) {
ilnik98436952017-07-13 00:47:03 -07001634 for (int i = 0; i < params_.call.num_thumbnails; ++i) {
ilnika014cc52017-03-07 04:21:04 -08001635 // Thumbnails will be send in the other way: from receiver_call to
1636 // sender_call.
1637 VideoSendStream::Config thumbnail_send_config(recv_transport);
1638 thumbnail_send_config.rtp.ssrcs.push_back(kThumbnailSendSsrcStart + i);
Niels Möller4db138e2018-04-19 09:04:13 +02001639 // TODO(nisse): Could use a simpler VP8-only encoder factory.
1640 thumbnail_send_config.encoder_settings.encoder_factory =
1641 &video_encoder_factory_;
Niels Möller259a4972018-04-05 15:36:51 +02001642 thumbnail_send_config.rtp.payload_name = params_.video[0].codec;
1643 thumbnail_send_config.rtp.payload_type = kPayloadTypeVP8;
ilnika014cc52017-03-07 04:21:04 -08001644 thumbnail_send_config.rtp.nack.rtp_history_ms = kNackRtpHistoryMs;
1645 thumbnail_send_config.rtp.rtx.payload_type = kSendRtxPayloadType;
1646 thumbnail_send_config.rtp.rtx.ssrcs.push_back(kThumbnailRtxSsrcStart + i);
1647 thumbnail_send_config.rtp.extensions.clear();
1648 if (params_.call.send_side_bwe) {
1649 thumbnail_send_config.rtp.extensions.push_back(
1650 RtpExtension(RtpExtension::kTransportSequenceNumberUri,
1651 test::kTransportSequenceNumberExtensionId));
1652 } else {
1653 thumbnail_send_config.rtp.extensions.push_back(RtpExtension(
1654 RtpExtension::kAbsSendTimeUri, test::kAbsSendTimeExtensionId));
1655 }
1656
1657 VideoEncoderConfig thumbnail_encoder_config;
Niels Möller259a4972018-04-05 15:36:51 +02001658 thumbnail_encoder_config.codec_type = kVideoCodecVP8;
Niels Möller4db138e2018-04-19 09:04:13 +02001659 thumbnail_encoder_config.video_format.name = "VP8";
ilnika014cc52017-03-07 04:21:04 -08001660 thumbnail_encoder_config.min_transmit_bitrate_bps = 7500;
1661 thumbnail_send_config.suspend_below_min_bitrate =
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001662 params_.video[0].suspend_below_min_bitrate;
ilnika014cc52017-03-07 04:21:04 -08001663 thumbnail_encoder_config.number_of_streams = 1;
1664 thumbnail_encoder_config.max_bitrate_bps = 50000;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001665 if (params_.ss[0].infer_streams) {
ilnik6b826ef2017-06-16 06:53:48 -07001666 thumbnail_encoder_config.video_stream_factory =
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001667 new rtc::RefCountedObject<VideoStreamFactory>(params_.ss[0].streams);
ilnik6b826ef2017-06-16 06:53:48 -07001668 } else {
Seth Hampson8234ead2018-02-02 15:16:24 -08001669 thumbnail_encoder_config.simulcast_layers = std::vector<VideoStream>(1);
ilnik6b826ef2017-06-16 06:53:48 -07001670 thumbnail_encoder_config.video_stream_factory =
1671 new rtc::RefCountedObject<cricket::EncoderStreamFactory>(
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001672 params_.video[0].codec, params_.ss[0].streams[0].max_qp,
1673 params_.video[0].fps, params_.screenshare[0].enabled, true);
ilnik6b826ef2017-06-16 06:53:48 -07001674 }
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001675 thumbnail_encoder_config.spatial_layers = params_.ss[0].spatial_layers;
ilnika014cc52017-03-07 04:21:04 -08001676
1677 VideoReceiveStream::Config thumbnail_receive_config(send_transport);
1678 thumbnail_receive_config.rtp.remb = false;
1679 thumbnail_receive_config.rtp.transport_cc = true;
1680 thumbnail_receive_config.rtp.local_ssrc = kReceiverLocalVideoSsrc;
1681 for (const RtpExtension& extension : thumbnail_send_config.rtp.extensions)
1682 thumbnail_receive_config.rtp.extensions.push_back(extension);
1683 thumbnail_receive_config.renderer = &fake_renderer_;
1684
1685 VideoReceiveStream::Decoder decoder =
Niels Möller259a4972018-04-05 15:36:51 +02001686 test::CreateMatchingDecoder(thumbnail_send_config);
ilnika014cc52017-03-07 04:21:04 -08001687 allocated_decoders_.push_back(
1688 std::unique_ptr<VideoDecoder>(decoder.decoder));
1689 thumbnail_receive_config.decoders.clear();
1690 thumbnail_receive_config.decoders.push_back(decoder);
1691 thumbnail_receive_config.rtp.remote_ssrc =
1692 thumbnail_send_config.rtp.ssrcs[0];
1693
1694 thumbnail_receive_config.rtp.nack.rtp_history_ms = kNackRtpHistoryMs;
1695 thumbnail_receive_config.rtp.rtx_ssrc = kThumbnailRtxSsrcStart + i;
nisse26e3abb2017-08-25 04:44:25 -07001696 thumbnail_receive_config.rtp
1697 .rtx_associated_payload_types[kSendRtxPayloadType] = kPayloadTypeVP8;
ilnika014cc52017-03-07 04:21:04 -08001698 thumbnail_receive_config.rtp.transport_cc = params_.call.send_side_bwe;
1699 thumbnail_receive_config.rtp.remb = !params_.call.send_side_bwe;
1700
1701 thumbnail_encoder_configs_.push_back(thumbnail_encoder_config.Copy());
1702 thumbnail_send_configs_.push_back(thumbnail_send_config.Copy());
1703 thumbnail_receive_configs_.push_back(thumbnail_receive_config.Copy());
1704 }
1705
ilnik98436952017-07-13 00:47:03 -07001706 for (int i = 0; i < params_.call.num_thumbnails; ++i) {
ilnika014cc52017-03-07 04:21:04 -08001707 thumbnail_send_streams_.push_back(receiver_call_->CreateVideoSendStream(
1708 thumbnail_send_configs_[i].Copy(),
1709 thumbnail_encoder_configs_[i].Copy()));
1710 thumbnail_receive_streams_.push_back(sender_call_->CreateVideoReceiveStream(
1711 thumbnail_receive_configs_[i].Copy()));
1712 }
1713}
1714
1715void VideoQualityTest::DestroyThumbnailStreams() {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001716 for (VideoSendStream* thumbnail_send_stream : thumbnail_send_streams_) {
ilnika014cc52017-03-07 04:21:04 -08001717 receiver_call_->DestroyVideoSendStream(thumbnail_send_stream);
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001718 }
ilnika014cc52017-03-07 04:21:04 -08001719 thumbnail_send_streams_.clear();
1720 for (VideoReceiveStream* thumbnail_receive_stream :
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001721 thumbnail_receive_streams_) {
ilnika014cc52017-03-07 04:21:04 -08001722 sender_call_->DestroyVideoReceiveStream(thumbnail_receive_stream);
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001723 }
ilnika014cc52017-03-07 04:21:04 -08001724 thumbnail_send_streams_.clear();
1725 thumbnail_receive_streams_.clear();
eladalon413ee9a2017-08-22 04:02:52 -07001726 for (std::unique_ptr<test::VideoCapturer>& video_caputurer :
1727 thumbnail_capturers_) {
1728 video_caputurer.reset();
1729 }
ilnika014cc52017-03-07 04:21:04 -08001730}
1731
ilnika014cc52017-03-07 04:21:04 -08001732void VideoQualityTest::SetupThumbnailCapturers(size_t num_thumbnail_streams) {
1733 VideoStream thumbnail = DefaultThumbnailStream();
1734 for (size_t i = 0; i < num_thumbnail_streams; ++i) {
1735 thumbnail_capturers_.emplace_back(test::FrameGeneratorCapturer::Create(
1736 static_cast<int>(thumbnail.width), static_cast<int>(thumbnail.height),
Danil Chapovalovb9b146c2018-06-15 12:28:07 +02001737 absl::nullopt, absl::nullopt, thumbnail.max_framerate, clock_));
ilnikf89a7382017-03-07 06:15:27 -08001738 RTC_DCHECK(thumbnail_capturers_.back());
ilnika014cc52017-03-07 04:21:04 -08001739 }
1740}
1741
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001742std::unique_ptr<test::FrameGenerator> VideoQualityTest::CreateFrameGenerator(
1743 size_t video_idx) {
1744 // Setup frame generator.
1745 const size_t kWidth = 1850;
1746 const size_t kHeight = 1110;
1747 std::unique_ptr<test::FrameGenerator> frame_generator;
1748 if (params_.screenshare[video_idx].generate_slides) {
1749 frame_generator = test::FrameGenerator::CreateSlideGenerator(
1750 kWidth, kHeight,
1751 params_.screenshare[video_idx].slide_change_interval *
1752 params_.video[video_idx].fps);
ivica5d6a06c2015-09-17 05:30:24 -07001753 } else {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001754 std::vector<std::string> slides = params_.screenshare[video_idx].slides;
1755 if (slides.size() == 0) {
1756 slides.push_back(test::ResourcePath("web_screenshot_1850_1110", "yuv"));
1757 slides.push_back(test::ResourcePath("presentation_1850_1110", "yuv"));
1758 slides.push_back(test::ResourcePath("photo_1850_1110", "yuv"));
1759 slides.push_back(test::ResourcePath("difficult_photo_1850_1110", "yuv"));
1760 }
1761 if (params_.screenshare[video_idx].scroll_duration == 0) {
1762 // Cycle image every slide_change_interval seconds.
1763 frame_generator = test::FrameGenerator::CreateFromYuvFile(
1764 slides, kWidth, kHeight,
1765 params_.screenshare[video_idx].slide_change_interval *
1766 params_.video[video_idx].fps);
ivica5d6a06c2015-09-17 05:30:24 -07001767 } else {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001768 RTC_CHECK_LE(params_.video[video_idx].width, kWidth);
1769 RTC_CHECK_LE(params_.video[video_idx].height, kHeight);
1770 RTC_CHECK_GT(params_.screenshare[video_idx].slide_change_interval, 0);
1771 const int kPauseDurationMs =
1772 (params_.screenshare[video_idx].slide_change_interval -
1773 params_.screenshare[video_idx].scroll_duration) *
1774 1000;
1775 RTC_CHECK_LE(params_.screenshare[video_idx].scroll_duration,
1776 params_.screenshare[video_idx].slide_change_interval);
1777
1778 frame_generator = test::FrameGenerator::CreateScrollingInputFromYuvFiles(
1779 clock_, slides, kWidth, kHeight, params_.video[video_idx].width,
1780 params_.video[video_idx].height,
1781 params_.screenshare[video_idx].scroll_duration * 1000,
1782 kPauseDurationMs);
ivica5d6a06c2015-09-17 05:30:24 -07001783 }
1784 }
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001785 return frame_generator;
1786}
1787
1788void VideoQualityTest::CreateCapturers() {
1789 video_capturers_.resize(num_video_streams_);
1790 for (size_t video_idx = 0; video_idx < num_video_streams_; ++video_idx) {
1791 if (params_.screenshare[video_idx].enabled) {
1792 std::unique_ptr<test::FrameGenerator> frame_generator =
1793 CreateFrameGenerator(video_idx);
1794 test::FrameGeneratorCapturer* frame_generator_capturer =
1795 new test::FrameGeneratorCapturer(clock_, std::move(frame_generator),
1796 params_.video[video_idx].fps);
1797 EXPECT_TRUE(frame_generator_capturer->Init());
1798 video_capturers_[video_idx].reset(frame_generator_capturer);
1799 } else {
1800 if (params_.video[video_idx].clip_name == "Generator") {
1801 video_capturers_[video_idx].reset(test::FrameGeneratorCapturer::Create(
1802 static_cast<int>(params_.video[video_idx].width),
Danil Chapovalovb9b146c2018-06-15 12:28:07 +02001803 static_cast<int>(params_.video[video_idx].height), absl::nullopt,
1804 absl::nullopt, params_.video[video_idx].fps, clock_));
Emircan Uysaler03e6ec92018-03-09 15:03:26 -08001805 } else if (params_.video[video_idx].clip_name == "GeneratorI420A") {
1806 video_capturers_[video_idx].reset(test::FrameGeneratorCapturer::Create(
1807 static_cast<int>(params_.video[video_idx].width),
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001808 static_cast<int>(params_.video[video_idx].height),
Danil Chapovalovb9b146c2018-06-15 12:28:07 +02001809 test::FrameGenerator::OutputType::I420A, absl::nullopt,
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001810 params_.video[video_idx].fps, clock_));
1811 } else if (params_.video[video_idx].clip_name.empty()) {
1812 video_capturers_[video_idx].reset(test::VcmCapturer::Create(
1813 params_.video[video_idx].width, params_.video[video_idx].height,
1814 params_.video[video_idx].fps,
1815 params_.video[video_idx].capture_device_index));
1816 if (!video_capturers_[video_idx]) {
1817 // Failed to get actual camera, use chroma generator as backup.
1818 video_capturers_[video_idx].reset(
1819 test::FrameGeneratorCapturer::Create(
1820 static_cast<int>(params_.video[video_idx].width),
1821 static_cast<int>(params_.video[video_idx].height),
Danil Chapovalovb9b146c2018-06-15 12:28:07 +02001822 absl::nullopt, absl::nullopt, params_.video[video_idx].fps,
Emircan Uysaler03e6ec92018-03-09 15:03:26 -08001823 clock_));
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001824 }
1825 } else {
1826 video_capturers_[video_idx].reset(
1827 test::FrameGeneratorCapturer::CreateFromYuvFile(
1828 test::ResourcePath(params_.video[video_idx].clip_name, "yuv"),
1829 params_.video[video_idx].width, params_.video[video_idx].height,
1830 params_.video[video_idx].fps, clock_));
1831 ASSERT_TRUE(video_capturers_[video_idx])
1832 << "Could not create capturer for "
1833 << params_.video[video_idx].clip_name
1834 << ".yuv. Is this resource file present?";
1835 }
1836 }
1837 RTC_DCHECK(video_capturers_[video_idx].get());
1838 }
ivica5d6a06c2015-09-17 05:30:24 -07001839}
1840
Christoffer Rodbrob4bb4eb2017-11-13 13:03:52 +01001841std::unique_ptr<test::LayerFilteringTransport>
1842VideoQualityTest::CreateSendTransport() {
1843 return rtc::MakeUnique<test::LayerFilteringTransport>(
1844 &task_queue_, params_.pipe, sender_call_.get(), kPayloadTypeVP8,
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001845 kPayloadTypeVP9, params_.video[0].selected_tl, params_.ss[0].selected_sl,
1846 payload_type_map_, kVideoSendSsrcs[0],
1847 static_cast<uint32_t>(kVideoSendSsrcs[0] + params_.ss[0].streams.size() -
1848 1));
Christoffer Rodbrob4bb4eb2017-11-13 13:03:52 +01001849}
1850
1851std::unique_ptr<test::DirectTransport>
1852VideoQualityTest::CreateReceiveTransport() {
1853 return rtc::MakeUnique<test::DirectTransport>(
1854 &task_queue_, params_.pipe, receiver_call_.get(), payload_type_map_);
1855}
1856
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001857void VideoQualityTest::CreateVideoStreams() {
1858 RTC_DCHECK(video_send_streams_.empty());
1859 RTC_DCHECK(video_receive_streams_.empty());
1860 RTC_DCHECK_EQ(video_send_configs_.size(), num_video_streams_);
Ying Wang0dd1b0a2018-02-20 12:50:27 +01001861
Stefan Holmerc6b224a2018-02-06 13:52:44 +01001862 // We currently only support testing external fec controllers with a single
1863 // VideoSendStream.
Ying Wang0dd1b0a2018-02-20 12:50:27 +01001864 if (fec_controller_factory_.get()) {
Stefan Holmerc6b224a2018-02-06 13:52:44 +01001865 RTC_DCHECK_LE(video_send_configs_.size(), 1);
1866 }
Ilya Nikolaevskiy4c09d7a2018-03-02 13:37:00 +01001867
1868 // TODO(http://crbug/818127):
1869 // Remove this workaround when ALR is not screenshare-specific.
1870 std::list<size_t> streams_creation_order;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001871 for (size_t i = 0; i < video_send_configs_.size(); ++i) {
Ilya Nikolaevskiy4c09d7a2018-03-02 13:37:00 +01001872 // If dual streams are created, add the screenshare stream last.
1873 if (video_encoder_configs_[i].content_type ==
1874 VideoEncoderConfig::ContentType::kScreen) {
1875 streams_creation_order.push_back(i);
Stefan Holmerc6b224a2018-02-06 13:52:44 +01001876 } else {
Ilya Nikolaevskiy4c09d7a2018-03-02 13:37:00 +01001877 streams_creation_order.push_front(i);
1878 }
1879 }
1880
1881 video_send_streams_.resize(video_send_configs_.size(), nullptr);
1882
1883 for (size_t i : streams_creation_order) {
1884 if (fec_controller_factory_.get()) {
1885 video_send_streams_[i] = sender_call_->CreateVideoSendStream(
1886 video_send_configs_[i].Copy(), video_encoder_configs_[i].Copy(),
1887 fec_controller_factory_->CreateFecController());
1888 } else {
1889 video_send_streams_[i] = sender_call_->CreateVideoSendStream(
1890 video_send_configs_[i].Copy(), video_encoder_configs_[i].Copy());
Stefan Holmerc6b224a2018-02-06 13:52:44 +01001891 }
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001892 }
1893 for (size_t i = 0; i < video_receive_configs_.size(); ++i) {
1894 video_receive_streams_.push_back(receiver_call_->CreateVideoReceiveStream(
1895 video_receive_configs_[i].Copy()));
1896 }
1897
1898 AssociateFlexfecStreamsWithVideoStreams();
1899}
1900
1901void VideoQualityTest::DestroyStreams() {
1902 CallTest::DestroyStreams();
1903
1904 for (VideoSendStream* video_send_stream : video_send_streams_)
1905 sender_call_->DestroyVideoSendStream(video_send_stream);
1906}
1907
sprang7a975f72015-10-12 06:33:21 -07001908void VideoQualityTest::RunWithAnalyzer(const Params& params) {
Ilya Nikolaevskiyad556762018-01-17 11:48:59 +01001909 rtc::LogMessage::SetLogToStderr(params.logging.logs);
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001910 num_video_streams_ = params.call.dual_video ? 2 : 1;
eladalon413ee9a2017-08-22 04:02:52 -07001911 std::unique_ptr<test::LayerFilteringTransport> send_transport;
1912 std::unique_ptr<test::DirectTransport> recv_transport;
1913 FILE* graph_data_output_file = nullptr;
1914 std::unique_ptr<VideoAnalyzer> analyzer;
1915
sprangce4aef12015-11-02 07:23:20 -08001916 params_ = params;
1917
minyue626bc952016-10-31 05:47:02 -07001918 RTC_CHECK(!params_.audio.enabled);
ivica5d6a06c2015-09-17 05:30:24 -07001919 // TODO(ivica): Merge with RunWithRenderer and use a flag / argument to
1920 // differentiate between the analyzer and the renderer case.
sprangce4aef12015-11-02 07:23:20 -08001921 CheckParams();
ivica5d6a06c2015-09-17 05:30:24 -07001922
sprangce4aef12015-11-02 07:23:20 -08001923 if (!params_.analyzer.graph_data_output_filename.empty()) {
ivica5d6a06c2015-09-17 05:30:24 -07001924 graph_data_output_file =
sprangce4aef12015-11-02 07:23:20 -08001925 fopen(params_.analyzer.graph_data_output_filename.c_str(), "w");
Peter Boström74f6e9e2016-04-04 17:56:10 +02001926 RTC_CHECK(graph_data_output_file)
sprangce4aef12015-11-02 07:23:20 -08001927 << "Can't open the file " << params_.analyzer.graph_data_output_filename
1928 << "!";
ivica87f83a92015-10-08 05:13:32 -07001929 }
sprang7a975f72015-10-12 06:33:21 -07001930
ilnik98436952017-07-13 00:47:03 -07001931 if (!params.logging.rtc_event_log_name.empty()) {
Bjorn Terelius0a6a2b72018-01-19 17:52:38 +01001932 send_event_log_ = RtcEventLog::Create(RtcEventLog::EncodingType::Legacy);
1933 recv_event_log_ = RtcEventLog::Create(RtcEventLog::EncodingType::Legacy);
Ilya Nikolaevskiya4259f62017-12-05 13:19:45 +01001934 std::unique_ptr<RtcEventLogOutputFile> send_output(
Bjorn Tereliusde939432017-11-20 17:38:14 +01001935 rtc::MakeUnique<RtcEventLogOutputFile>(
Ilya Nikolaevskiya4259f62017-12-05 13:19:45 +01001936 params.logging.rtc_event_log_name + "_send",
1937 RtcEventLog::kUnlimitedOutput));
1938 std::unique_ptr<RtcEventLogOutputFile> recv_output(
1939 rtc::MakeUnique<RtcEventLogOutputFile>(
1940 params.logging.rtc_event_log_name + "_recv",
1941 RtcEventLog::kUnlimitedOutput));
1942 bool event_log_started =
1943 send_event_log_->StartLogging(std::move(send_output),
1944 RtcEventLog::kImmediateOutput) &&
1945 recv_event_log_->StartLogging(std::move(recv_output),
1946 RtcEventLog::kImmediateOutput);
ilnik98436952017-07-13 00:47:03 -07001947 RTC_DCHECK(event_log_started);
Ilya Nikolaevskiya4259f62017-12-05 13:19:45 +01001948 } else {
1949 send_event_log_ = RtcEventLog::CreateNull();
1950 recv_event_log_ = RtcEventLog::CreateNull();
ilnik98436952017-07-13 00:47:03 -07001951 }
1952
Ilya Nikolaevskiya4259f62017-12-05 13:19:45 +01001953 Call::Config send_call_config(send_event_log_.get());
1954 Call::Config recv_call_config(recv_event_log_.get());
1955 send_call_config.bitrate_config = params.call.call_bitrate_config;
1956 recv_call_config.bitrate_config = params.call.call_bitrate_config;
stefanf116bd02015-10-27 08:29:42 -07001957
Ilya Nikolaevskiya4259f62017-12-05 13:19:45 +01001958 task_queue_.SendTask([this, &send_call_config, &recv_call_config,
1959 &send_transport, &recv_transport]() {
1960 CreateCalls(send_call_config, recv_call_config);
1961 send_transport = CreateSendTransport();
1962 recv_transport = CreateReceiveTransport();
1963 });
stefanf116bd02015-10-27 08:29:42 -07001964
sprangce4aef12015-11-02 07:23:20 -08001965 std::string graph_title = params_.analyzer.graph_title;
1966 if (graph_title.empty())
1967 graph_title = VideoQualityTest::GenerateGraphTitle();
sprangc1b57a12017-02-28 08:50:47 -08001968 bool is_quick_test_enabled = field_trial::IsEnabled("WebRTC-QuickPerfTest");
eladalon413ee9a2017-08-22 04:02:52 -07001969 analyzer = rtc::MakeUnique<VideoAnalyzer>(
1970 send_transport.get(), params_.analyzer.test_label,
ilnik2a8c2f52017-02-15 02:23:28 -08001971 params_.analyzer.avg_psnr_threshold, params_.analyzer.avg_ssim_threshold,
ilnik9ae0d762017-02-15 00:53:12 -08001972 is_quick_test_enabled
1973 ? kFramesSentInQuickTest
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001974 : params_.analyzer.test_durations_secs * params_.video[0].fps,
sprangce4aef12015-11-02 07:23:20 -08001975 graph_data_output_file, graph_title,
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001976 kVideoSendSsrcs[params_.ss[0].selected_stream],
1977 kSendRtxSsrcs[params_.ss[0].selected_stream],
1978 static_cast<size_t>(params_.ss[0].selected_stream),
1979 params.ss[0].selected_sl, params_.video[0].selected_tl,
1980 is_quick_test_enabled, clock_, params_.logging.rtp_dump_name);
ivica5d6a06c2015-09-17 05:30:24 -07001981
eladalon413ee9a2017-08-22 04:02:52 -07001982 task_queue_.SendTask([&]() {
1983 analyzer->SetCall(sender_call_.get());
1984 analyzer->SetReceiver(receiver_call_->Receiver());
1985 send_transport->SetReceiver(analyzer.get());
1986 recv_transport->SetReceiver(sender_call_->Receiver());
ivica5d6a06c2015-09-17 05:30:24 -07001987
eladalon413ee9a2017-08-22 04:02:52 -07001988 SetupVideo(analyzer.get(), recv_transport.get());
1989 SetupThumbnails(analyzer.get(), recv_transport.get());
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001990 video_receive_configs_[params_.ss[0].selected_stream].renderer =
eladalon413ee9a2017-08-22 04:02:52 -07001991 analyzer.get();
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001992 video_send_configs_[0].pre_encode_callback = analyzer->pre_encode_proxy();
1993 RTC_DCHECK(!video_send_configs_[0].post_encode_callback);
Niels Möllerf88a22c2018-06-19 17:05:03 +02001994 video_send_configs_[0].post_encode_callback = analyzer.get();
kthelgason2bc68642017-02-07 07:02:22 -08001995
eladalon413ee9a2017-08-22 04:02:52 -07001996 CreateFlexfecStreams();
1997 CreateVideoStreams();
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01001998 analyzer->SetSendStream(video_send_streams_[0]);
eladalon413ee9a2017-08-22 04:02:52 -07001999 if (video_receive_streams_.size() == 1)
2000 analyzer->SetReceiveStream(video_receive_streams_[0]);
ivica5d6a06c2015-09-17 05:30:24 -07002001
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002002 video_send_streams_[0]->SetSource(analyzer->OutputInterface(),
2003 degradation_preference_);
eladalon413ee9a2017-08-22 04:02:52 -07002004 SetupThumbnailCapturers(params_.call.num_thumbnails);
2005 for (size_t i = 0; i < thumbnail_send_streams_.size(); ++i) {
2006 thumbnail_send_streams_[i]->SetSource(thumbnail_capturers_[i].get(),
2007 degradation_preference_);
2008 }
ilnika014cc52017-03-07 04:21:04 -08002009
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002010 CreateCapturers();
ivicac1cc8542015-10-08 03:44:06 -07002011
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002012 analyzer->SetSource(video_capturers_[0].get(), params_.ss[0].infer_streams);
ilnika014cc52017-03-07 04:21:04 -08002013
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002014 for (size_t video_idx = 1; video_idx < num_video_streams_; ++video_idx) {
2015 video_send_streams_[video_idx]->SetSource(
2016 video_capturers_[video_idx].get(), degradation_preference_);
2017 }
2018
2019 StartEncodedFrameLogs(video_send_streams_[0]);
2020 StartEncodedFrameLogs(
2021 video_receive_streams_[params_.ss[0].selected_stream]);
2022 for (VideoSendStream* video_send_stream : video_send_streams_)
2023 video_send_stream->Start();
eladalon413ee9a2017-08-22 04:02:52 -07002024 for (VideoSendStream* thumbnail_send_stream : thumbnail_send_streams_)
2025 thumbnail_send_stream->Start();
2026 for (VideoReceiveStream* receive_stream : video_receive_streams_)
2027 receive_stream->Start();
2028 for (VideoReceiveStream* thumbnail_receive_stream :
2029 thumbnail_receive_streams_)
2030 thumbnail_receive_stream->Start();
ilnika014cc52017-03-07 04:21:04 -08002031
eladalon413ee9a2017-08-22 04:02:52 -07002032 analyzer->StartMeasuringCpuProcessTime();
ivica5d6a06c2015-09-17 05:30:24 -07002033
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002034 for (size_t video_idx = 0; video_idx < num_video_streams_; ++video_idx) {
2035 video_capturers_[video_idx]->Start();
2036 }
eladalon413ee9a2017-08-22 04:02:52 -07002037 for (std::unique_ptr<test::VideoCapturer>& video_caputurer :
2038 thumbnail_capturers_) {
2039 video_caputurer->Start();
2040 }
2041 });
ivica5d6a06c2015-09-17 05:30:24 -07002042
eladalon413ee9a2017-08-22 04:02:52 -07002043 analyzer->Wait();
ivica5d6a06c2015-09-17 05:30:24 -07002044
Ilya Nikolaevskiy2c72fe82017-10-02 13:01:04 +02002045 event_log_->StopLogging();
2046
eladalon413ee9a2017-08-22 04:02:52 -07002047 task_queue_.SendTask([&]() {
2048 for (std::unique_ptr<test::VideoCapturer>& video_caputurer :
2049 thumbnail_capturers_)
2050 video_caputurer->Stop();
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002051 for (size_t video_idx = 0; video_idx < num_video_streams_; ++video_idx) {
2052 video_capturers_[video_idx]->Stop();
2053 }
eladalon413ee9a2017-08-22 04:02:52 -07002054 for (VideoReceiveStream* thumbnail_receive_stream :
2055 thumbnail_receive_streams_)
2056 thumbnail_receive_stream->Stop();
2057 for (VideoReceiveStream* receive_stream : video_receive_streams_)
2058 receive_stream->Stop();
2059 for (VideoSendStream* thumbnail_send_stream : thumbnail_send_streams_)
2060 thumbnail_send_stream->Stop();
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002061 for (VideoSendStream* video_send_stream : video_send_streams_)
2062 video_send_stream->Stop();
ivica5d6a06c2015-09-17 05:30:24 -07002063
eladalon413ee9a2017-08-22 04:02:52 -07002064 DestroyStreams();
2065 DestroyThumbnailStreams();
ivica5d6a06c2015-09-17 05:30:24 -07002066
eladalon413ee9a2017-08-22 04:02:52 -07002067 if (graph_data_output_file)
2068 fclose(graph_data_output_file);
2069
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002070 video_capturers_.clear();
eladalon413ee9a2017-08-22 04:02:52 -07002071 send_transport.reset();
2072 recv_transport.reset();
2073
2074 DestroyCalls();
2075 });
ivica5d6a06c2015-09-17 05:30:24 -07002076}
2077
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +01002078void VideoQualityTest::SetupAudio(Transport* transport,
minyuea27172d2016-11-01 05:59:29 -07002079 AudioReceiveStream** audio_receive_stream) {
2080 audio_send_config_ = AudioSendStream::Config(transport);
minyuea27172d2016-11-01 05:59:29 -07002081 audio_send_config_.rtp.ssrc = kAudioSendSsrc;
2082
2083 // Add extension to enable audio send side BWE, and allow audio bit rate
2084 // adaptation.
2085 audio_send_config_.rtp.extensions.clear();
2086 if (params_.call.send_side_bwe) {
2087 audio_send_config_.rtp.extensions.push_back(
2088 webrtc::RtpExtension(webrtc::RtpExtension::kTransportSequenceNumberUri,
2089 test::kTransportSequenceNumberExtensionId));
minyue10cbb462016-11-07 09:29:22 -08002090 audio_send_config_.min_bitrate_bps = kOpusMinBitrateBps;
2091 audio_send_config_.max_bitrate_bps = kOpusBitrateFbBps;
minyuea27172d2016-11-01 05:59:29 -07002092 }
Oskar Sundbom8e07c132018-01-08 16:45:42 +01002093 audio_send_config_.send_codec_spec = AudioSendStream::Config::SendCodecSpec(
2094 kAudioSendPayloadType,
Yves Gerey665174f2018-06-19 15:03:05 +02002095 {"OPUS",
2096 48000,
2097 2,
2098 {{"usedtx", (params_.audio.dtx ? "1" : "0")}, {"stereo", "1"}}});
Niels Möller2784a032018-03-28 14:16:04 +02002099 audio_send_config_.encoder_factory = audio_encoder_factory_;
sprang1168fd42017-06-21 09:00:17 -07002100 audio_send_stream_ = sender_call_->CreateAudioSendStream(audio_send_config_);
minyuea27172d2016-11-01 05:59:29 -07002101
2102 AudioReceiveStream::Config audio_config;
2103 audio_config.rtp.local_ssrc = kReceiverLocalAudioSsrc;
2104 audio_config.rtcp_send_transport = transport;
minyuea27172d2016-11-01 05:59:29 -07002105 audio_config.rtp.remote_ssrc = audio_send_config_.rtp.ssrc;
2106 audio_config.rtp.transport_cc = params_.call.send_side_bwe;
2107 audio_config.rtp.extensions = audio_send_config_.rtp.extensions;
Niels Möller2784a032018-03-28 14:16:04 +02002108 audio_config.decoder_factory = audio_decoder_factory_;
minyue20c84cc2017-04-10 16:57:57 -07002109 audio_config.decoder_map = {{kAudioSendPayloadType, {"OPUS", 48000, 2}}};
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002110 if (params_.video[0].enabled && params_.audio.sync_video)
minyuea27172d2016-11-01 05:59:29 -07002111 audio_config.sync_group = kSyncGroup;
2112
sprang1168fd42017-06-21 09:00:17 -07002113 *audio_receive_stream =
2114 receiver_call_->CreateAudioReceiveStream(audio_config);
minyuea27172d2016-11-01 05:59:29 -07002115}
2116
minyue73208662016-08-18 06:28:55 -07002117void VideoQualityTest::RunWithRenderers(const Params& params) {
Ilya Nikolaevskiyad556762018-01-17 11:48:59 +01002118 rtc::LogMessage::SetLogToStderr(params.logging.logs);
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002119 num_video_streams_ = params.call.dual_video ? 2 : 1;
eladalon413ee9a2017-08-22 04:02:52 -07002120 std::unique_ptr<test::LayerFilteringTransport> send_transport;
2121 std::unique_ptr<test::DirectTransport> recv_transport;
minyuea27172d2016-11-01 05:59:29 -07002122 std::unique_ptr<test::VideoRenderer> local_preview;
eladalon413ee9a2017-08-22 04:02:52 -07002123 std::vector<std::unique_ptr<test::VideoRenderer>> loopback_renderers;
minyue73208662016-08-18 06:28:55 -07002124 AudioReceiveStream* audio_receive_stream = nullptr;
minyue73208662016-08-18 06:28:55 -07002125
eladalon413ee9a2017-08-22 04:02:52 -07002126 task_queue_.SendTask([&]() {
2127 params_ = params;
2128 CheckParams();
palmkviste75f2042016-09-28 06:19:48 -07002129
eladalon413ee9a2017-08-22 04:02:52 -07002130 // TODO(ivica): Remove bitrate_config and use the default Call::Config(), to
2131 // match the full stack tests.
2132 Call::Config call_config(event_log_.get());
2133 call_config.bitrate_config = params_.call.call_bitrate_config;
Niels Möller2bf9e732017-08-14 11:26:16 +02002134
Artem Titov3faa8322018-03-07 14:44:00 +01002135 rtc::scoped_refptr<TestAudioDeviceModule> fake_audio_device =
2136 TestAudioDeviceModule::CreateTestAudioDeviceModule(
2137 TestAudioDeviceModule::CreatePulsedNoiseCapturer(32000, 48000),
2138 TestAudioDeviceModule::CreateDiscardRenderer(48000), 1.f);
ivica5d6a06c2015-09-17 05:30:24 -07002139
eladalon413ee9a2017-08-22 04:02:52 -07002140 if (params_.audio.enabled) {
eladalon413ee9a2017-08-22 04:02:52 -07002141 AudioState::Config audio_state_config;
eladalon413ee9a2017-08-22 04:02:52 -07002142 audio_state_config.audio_mixer = AudioMixerImpl::Create();
Ivo Creusen62337e52018-01-09 14:17:33 +01002143 audio_state_config.audio_processing = AudioProcessingBuilder().Create();
Fredrik Solenberg2a877972017-12-15 16:42:15 +01002144 audio_state_config.audio_device_module = fake_audio_device;
eladalon413ee9a2017-08-22 04:02:52 -07002145 call_config.audio_state = AudioState::Create(audio_state_config);
Fredrik Solenbergd3195342017-11-21 20:33:05 +01002146 fake_audio_device->RegisterAudioCallback(
2147 call_config.audio_state->audio_transport());
eladalon413ee9a2017-08-22 04:02:52 -07002148 }
minyue73208662016-08-18 06:28:55 -07002149
eladalon413ee9a2017-08-22 04:02:52 -07002150 CreateCalls(call_config, call_config);
2151
2152 // TODO(minyue): consider if this is a good transport even for audio only
2153 // calls.
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002154 send_transport = CreateSendTransport();
eladalon413ee9a2017-08-22 04:02:52 -07002155
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002156 recv_transport = CreateReceiveTransport();
eladalon413ee9a2017-08-22 04:02:52 -07002157
2158 // TODO(ivica): Use two calls to be able to merge with RunWithAnalyzer or at
2159 // least share as much code as possible. That way this test would also match
2160 // the full stack tests better.
2161 send_transport->SetReceiver(receiver_call_->Receiver());
2162 recv_transport->SetReceiver(sender_call_->Receiver());
2163
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002164 if (params_.video[0].enabled) {
eladalon413ee9a2017-08-22 04:02:52 -07002165 // Create video renderers.
2166 local_preview.reset(test::VideoRenderer::Create(
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002167 "Local Preview", params_.video[0].width, params_.video[0].height));
eladalon413ee9a2017-08-22 04:02:52 -07002168
2169 SetupVideo(send_transport.get(), recv_transport.get());
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002170 video_send_configs_[0].pre_encode_callback = local_preview.get();
eladalon413ee9a2017-08-22 04:02:52 -07002171
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002172 size_t num_streams_processed = 0;
2173 for (size_t video_idx = 0; video_idx < num_video_streams_; ++video_idx) {
2174 const size_t selected_stream_id = params_.ss[video_idx].selected_stream;
2175 const size_t num_streams = params_.ss[video_idx].streams.size();
2176 if (selected_stream_id == num_streams) {
2177 for (size_t stream_id = 0; stream_id < num_streams; ++stream_id) {
2178 std::ostringstream oss;
2179 oss << "Loopback Video #" << video_idx << " - Stream #"
2180 << static_cast<int>(stream_id);
2181 loopback_renderers.emplace_back(test::VideoRenderer::Create(
2182 oss.str().c_str(),
2183 params_.ss[video_idx].streams[stream_id].width,
2184 params_.ss[video_idx].streams[stream_id].height));
2185 video_receive_configs_[stream_id + num_streams_processed].renderer =
2186 loopback_renderers.back().get();
2187 if (params_.audio.enabled && params_.audio.sync_video)
2188 video_receive_configs_[stream_id + num_streams_processed]
2189 .sync_group = kSyncGroup;
2190 }
2191 } else {
2192 std::ostringstream oss;
2193 oss << "Loopback Video #" << video_idx;
2194 loopback_renderers.emplace_back(test::VideoRenderer::Create(
2195 oss.str().c_str(),
2196 params_.ss[video_idx].streams[selected_stream_id].width,
2197 params_.ss[video_idx].streams[selected_stream_id].height));
2198 video_receive_configs_[selected_stream_id + num_streams_processed]
2199 .renderer = loopback_renderers.back().get();
eladalon413ee9a2017-08-22 04:02:52 -07002200 if (params_.audio.enabled && params_.audio.sync_video)
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002201 video_receive_configs_[num_streams_processed + selected_stream_id]
2202 .sync_group = kSyncGroup;
eladalon413ee9a2017-08-22 04:02:52 -07002203 }
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002204 num_streams_processed += num_streams;
eladalon413ee9a2017-08-22 04:02:52 -07002205 }
eladalon413ee9a2017-08-22 04:02:52 -07002206 CreateFlexfecStreams();
2207 CreateVideoStreams();
2208
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002209 CreateCapturers();
2210 for (size_t video_idx = 0; video_idx < num_video_streams_; ++video_idx) {
2211 video_send_streams_[video_idx]->SetSource(
2212 video_capturers_[video_idx].get(), degradation_preference_);
2213 }
eladalon413ee9a2017-08-22 04:02:52 -07002214 }
2215
2216 if (params_.audio.enabled) {
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +01002217 SetupAudio(send_transport.get(), &audio_receive_stream);
eladalon413ee9a2017-08-22 04:02:52 -07002218 }
2219
2220 for (VideoReceiveStream* receive_stream : video_receive_streams_)
2221 StartEncodedFrameLogs(receive_stream);
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002222 StartEncodedFrameLogs(video_send_streams_[0]);
eladalon413ee9a2017-08-22 04:02:52 -07002223
2224 // Start sending and receiving video.
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002225 if (params_.video[0].enabled) {
eladalon413ee9a2017-08-22 04:02:52 -07002226 for (VideoReceiveStream* video_receive_stream : video_receive_streams_)
2227 video_receive_stream->Start();
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002228 for (VideoSendStream* video_send_stream : video_send_streams_)
2229 video_send_stream->Start();
2230 for (auto& video_capturer : video_capturers_)
2231 video_capturer->Start();
eladalon413ee9a2017-08-22 04:02:52 -07002232 }
2233
2234 if (params_.audio.enabled) {
2235 // Start receiving audio.
2236 audio_receive_stream->Start();
eladalon413ee9a2017-08-22 04:02:52 -07002237
2238 // Start sending audio.
2239 audio_send_stream_->Start();
eladalon413ee9a2017-08-22 04:02:52 -07002240 }
2241 });
minyue73208662016-08-18 06:28:55 -07002242
ivica5d6a06c2015-09-17 05:30:24 -07002243 test::PressEnterToContinue();
2244
eladalon413ee9a2017-08-22 04:02:52 -07002245 task_queue_.SendTask([&]() {
2246 if (params_.audio.enabled) {
2247 // Stop sending audio.
eladalon413ee9a2017-08-22 04:02:52 -07002248 audio_send_stream_->Stop();
minyue73208662016-08-18 06:28:55 -07002249
eladalon413ee9a2017-08-22 04:02:52 -07002250 // Stop receiving audio.
eladalon413ee9a2017-08-22 04:02:52 -07002251 audio_receive_stream->Stop();
eladalon413ee9a2017-08-22 04:02:52 -07002252 }
minyue73208662016-08-18 06:28:55 -07002253
eladalon413ee9a2017-08-22 04:02:52 -07002254 // Stop receiving and sending video.
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002255 if (params_.video[0].enabled) {
2256 for (auto& video_capturer : video_capturers_)
2257 video_capturer->Stop();
2258 for (VideoSendStream* video_send_stream : video_send_streams_)
2259 video_send_stream->Stop();
2260 DestroyStreams();
brandtr1293aca2016-11-16 22:47:29 -08002261 }
minyue73208662016-08-18 06:28:55 -07002262
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +01002263 video_capturers_.clear();
eladalon413ee9a2017-08-22 04:02:52 -07002264 send_transport.reset();
2265 recv_transport.reset();
sprang1168fd42017-06-21 09:00:17 -07002266
eladalon413ee9a2017-08-22 04:02:52 -07002267 local_preview.reset();
2268 loopback_renderers.clear();
2269
2270 DestroyCalls();
2271 });
ivica5d6a06c2015-09-17 05:30:24 -07002272}
2273
palmkviste75f2042016-09-28 06:19:48 -07002274void VideoQualityTest::StartEncodedFrameLogs(VideoSendStream* stream) {
ilnik98436952017-07-13 00:47:03 -07002275 if (!params_.logging.encoded_frame_base_path.empty()) {
palmkviste75f2042016-09-28 06:19:48 -07002276 std::ostringstream str;
2277 str << send_logs_++;
2278 std::string prefix =
ilnik98436952017-07-13 00:47:03 -07002279 params_.logging.encoded_frame_base_path + "." + str.str() + ".send.";
palmkviste75f2042016-09-28 06:19:48 -07002280 stream->EnableEncodedFrameRecording(
2281 std::vector<rtc::PlatformFile>(
2282 {rtc::CreatePlatformFile(prefix + "1.ivf"),
2283 rtc::CreatePlatformFile(prefix + "2.ivf"),
2284 rtc::CreatePlatformFile(prefix + "3.ivf")}),
ilnik98436952017-07-13 00:47:03 -07002285 100000000);
palmkviste75f2042016-09-28 06:19:48 -07002286 }
2287}
ilnikcb8c1462017-03-09 09:23:30 -08002288
palmkviste75f2042016-09-28 06:19:48 -07002289void VideoQualityTest::StartEncodedFrameLogs(VideoReceiveStream* stream) {
ilnik98436952017-07-13 00:47:03 -07002290 if (!params_.logging.encoded_frame_base_path.empty()) {
palmkviste75f2042016-09-28 06:19:48 -07002291 std::ostringstream str;
2292 str << receive_logs_++;
2293 std::string path =
ilnik98436952017-07-13 00:47:03 -07002294 params_.logging.encoded_frame_base_path + "." + str.str() + ".recv.ivf";
palmkviste75f2042016-09-28 06:19:48 -07002295 stream->EnableEncodedFrameRecording(rtc::CreatePlatformFile(path),
ilnik98436952017-07-13 00:47:03 -07002296 100000000);
palmkviste75f2042016-09-28 06:19:48 -07002297 }
2298}
ivica5d6a06c2015-09-17 05:30:24 -07002299} // namespace webrtc