Index: webrtc/video/screenshare_loopback.cc |
diff --git a/webrtc/video/screenshare_loopback.cc b/webrtc/video/screenshare_loopback.cc |
index f2133a211b8a947fbcecb543c103561e3cdef8cd..bfa0e6415e70ef7c685fa0a28f9b195404d9cb45 100644 |
--- a/webrtc/video/screenshare_loopback.cc |
+++ b/webrtc/video/screenshare_loopback.cc |
@@ -15,6 +15,7 @@ |
#include "gflags/gflags.h" |
#include "testing/gtest/include/gtest/gtest.h" |
+#include "webrtc/base/checks.h" |
#include "webrtc/test/field_trial.h" |
#include "webrtc/test/frame_generator.h" |
#include "webrtc/test/frame_generator_capturer.h" |
@@ -114,6 +115,11 @@ class ScreenshareLoopback : public test::Loopback { |
vp8_settings_.denoisingOn = false; |
vp8_settings_.frameDroppingOn = false; |
vp8_settings_.numberOfTemporalLayers = 2; |
+ |
+ vp9_settings_ = VideoEncoder::GetDefaultVp9Settings(); |
+ vp9_settings_.denoisingOn = false; |
+ vp9_settings_.frameDroppingOn = false; |
+ vp9_settings_.numberOfTemporalLayers = 2; |
} |
virtual ~ScreenshareLoopback() {} |
@@ -123,7 +129,14 @@ class ScreenshareLoopback : public test::Loopback { |
VideoStream* stream = &encoder_config.streams[0]; |
encoder_config.content_type = VideoEncoderConfig::ContentType::kScreen; |
encoder_config.min_transmit_bitrate_bps = flags::MinTransmitBitrate(); |
- encoder_config.encoder_specific_settings = &vp8_settings_; |
+ if (config_.codec == "VP8") { |
+ encoder_config.encoder_specific_settings = &vp8_settings_; |
+ } else if (config_.codec == "VP9") { |
+ encoder_config.encoder_specific_settings = &vp9_settings_; |
+ } else { |
+ RTC_NOTREACHED() << "Codec not supported!"; |
+ abort(); |
+ } |
stream->temporal_layer_thresholds_bps.clear(); |
stream->target_bitrate_bps = |
static_cast<int>(config_.start_bitrate_kbps) * 1000; |
@@ -148,6 +161,7 @@ class ScreenshareLoopback : public test::Loopback { |
} |
VideoCodecVP8 vp8_settings_; |
+ VideoCodecVP9 vp9_settings_; |
}; |
void Loopback() { |