pbos@webrtc.org | f577ae9 | 2014-03-19 08:43:57 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2014 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 Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 10 | #include "test/encoder_settings.h" |
pbos@webrtc.org | f577ae9 | 2014-03-19 08:43:57 +0000 | [diff] [blame] | 11 | |
perkj | fa10b55 | 2016-10-02 23:45:26 -0700 | [diff] [blame] | 12 | #include <algorithm> |
| 13 | #include <string> |
pbos@webrtc.org | f577ae9 | 2014-03-19 08:43:57 +0000 | [diff] [blame] | 14 | |
Emircan Uysaler | 03e6ec9 | 2018-03-09 15:03:26 -0800 | [diff] [blame] | 15 | #include "media/engine/internaldecoderfactory.h" |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 16 | #include "modules/video_coding/codecs/h264/include/h264.h" |
Emircan Uysaler | 03e6ec9 | 2018-03-09 15:03:26 -0800 | [diff] [blame] | 17 | #include "modules/video_coding/codecs/multiplex/include/multiplex_decoder_adapter.h" |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 18 | #include "modules/video_coding/codecs/vp8/include/vp8.h" |
| 19 | #include "modules/video_coding/codecs/vp9/include/vp9.h" |
Niels Möller | 84255bb | 2017-10-06 13:43:23 +0200 | [diff] [blame] | 20 | #include "rtc_base/refcountedobject.h" |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 21 | #include "test/fake_decoder.h" |
pbos@webrtc.org | f577ae9 | 2014-03-19 08:43:57 +0000 | [diff] [blame] | 22 | |
| 23 | namespace webrtc { |
| 24 | namespace test { |
pbos@webrtc.org | f577ae9 | 2014-03-19 08:43:57 +0000 | [diff] [blame] | 25 | |
perkj | fa10b55 | 2016-10-02 23:45:26 -0700 | [diff] [blame] | 26 | const size_t DefaultVideoStreamFactory::kMaxNumberOfStreams; |
| 27 | const int DefaultVideoStreamFactory::kMaxBitratePerStream[] = {150000, 450000, |
| 28 | 1500000}; |
| 29 | const int DefaultVideoStreamFactory::kDefaultMinBitratePerStream[] = { |
philipel | 5ef2bc1 | 2017-02-21 07:28:31 -0800 | [diff] [blame] | 30 | 30000, 200000, 700000}; |
pbos@webrtc.org | f577ae9 | 2014-03-19 08:43:57 +0000 | [diff] [blame] | 31 | |
perkj | fa10b55 | 2016-10-02 23:45:26 -0700 | [diff] [blame] | 32 | // static |
| 33 | std::vector<VideoStream> CreateVideoStreams( |
| 34 | int width, |
| 35 | int height, |
| 36 | const webrtc::VideoEncoderConfig& encoder_config) { |
| 37 | RTC_DCHECK(encoder_config.number_of_streams <= |
| 38 | DefaultVideoStreamFactory::kMaxNumberOfStreams); |
pbos@webrtc.org | f577ae9 | 2014-03-19 08:43:57 +0000 | [diff] [blame] | 39 | |
perkj | fa10b55 | 2016-10-02 23:45:26 -0700 | [diff] [blame] | 40 | std::vector<VideoStream> stream_settings(encoder_config.number_of_streams); |
| 41 | int bitrate_left_bps = encoder_config.max_bitrate_bps; |
pbos@webrtc.org | f577ae9 | 2014-03-19 08:43:57 +0000 | [diff] [blame] | 42 | |
perkj | fa10b55 | 2016-10-02 23:45:26 -0700 | [diff] [blame] | 43 | for (size_t i = 0; i < encoder_config.number_of_streams; ++i) { |
Danil Chapovalov | 350531e | 2018-06-08 11:04:04 +0000 | [diff] [blame^] | 44 | stream_settings[i].width = |
| 45 | (i + 1) * width / encoder_config.number_of_streams; |
| 46 | stream_settings[i].height = |
| 47 | (i + 1) * height / encoder_config.number_of_streams; |
perkj | fa10b55 | 2016-10-02 23:45:26 -0700 | [diff] [blame] | 48 | stream_settings[i].max_framerate = 30; |
| 49 | stream_settings[i].min_bitrate_bps = |
| 50 | DefaultVideoStreamFactory::kDefaultMinBitratePerStream[i]; |
philipel | 0676f22 | 2018-04-17 16:12:21 +0200 | [diff] [blame] | 51 | |
| 52 | int target_bitrate_bps = -1; |
| 53 | int max_bitrate_bps = -1; |
| 54 | // Use configured values instead of default values if values has been |
| 55 | // configured. |
| 56 | if (i < encoder_config.simulcast_layers.size()) { |
| 57 | const VideoStream& stream = encoder_config.simulcast_layers[i]; |
| 58 | |
| 59 | max_bitrate_bps = |
| 60 | stream.max_bitrate_bps > 0 |
| 61 | ? stream.max_bitrate_bps |
| 62 | : DefaultVideoStreamFactory::kMaxBitratePerStream[i]; |
| 63 | max_bitrate_bps = std::min(bitrate_left_bps, max_bitrate_bps); |
| 64 | |
| 65 | target_bitrate_bps = |
| 66 | stream.target_bitrate_bps > 0 |
| 67 | ? stream.target_bitrate_bps |
| 68 | : DefaultVideoStreamFactory::kMaxBitratePerStream[i]; |
| 69 | target_bitrate_bps = std::min(max_bitrate_bps, target_bitrate_bps); |
| 70 | } else { |
| 71 | max_bitrate_bps = std::min( |
| 72 | bitrate_left_bps, DefaultVideoStreamFactory::kMaxBitratePerStream[i]); |
| 73 | target_bitrate_bps = max_bitrate_bps; |
| 74 | } |
| 75 | |
| 76 | RTC_DCHECK_NE(target_bitrate_bps, -1); |
| 77 | RTC_DCHECK_NE(max_bitrate_bps, -1); |
| 78 | stream_settings[i].target_bitrate_bps = target_bitrate_bps; |
| 79 | stream_settings[i].max_bitrate_bps = max_bitrate_bps; |
perkj | fa10b55 | 2016-10-02 23:45:26 -0700 | [diff] [blame] | 80 | stream_settings[i].max_qp = 56; |
philipel | 0676f22 | 2018-04-17 16:12:21 +0200 | [diff] [blame] | 81 | |
Seth Hampson | cc7125f | 2018-02-02 08:46:16 -0800 | [diff] [blame] | 82 | if (i < encoder_config.simulcast_layers.size()) { |
| 83 | // Higher level controls are setting the active configuration for the |
| 84 | // VideoStream. |
| 85 | stream_settings[i].active = encoder_config.simulcast_layers[i].active; |
| 86 | } else { |
| 87 | stream_settings[i].active = true; |
| 88 | } |
perkj | fa10b55 | 2016-10-02 23:45:26 -0700 | [diff] [blame] | 89 | bitrate_left_bps -= stream_settings[i].target_bitrate_bps; |
| 90 | } |
pbos@webrtc.org | f577ae9 | 2014-03-19 08:43:57 +0000 | [diff] [blame] | 91 | |
perkj | fa10b55 | 2016-10-02 23:45:26 -0700 | [diff] [blame] | 92 | stream_settings[encoder_config.number_of_streams - 1].max_bitrate_bps += |
| 93 | bitrate_left_bps; |
Seth Hampson | 24722b3 | 2017-12-22 09:36:42 -0800 | [diff] [blame] | 94 | stream_settings[0].bitrate_priority = encoder_config.bitrate_priority; |
perkj | fa10b55 | 2016-10-02 23:45:26 -0700 | [diff] [blame] | 95 | |
pbos@webrtc.org | 6ae48c6 | 2014-06-06 10:49:19 +0000 | [diff] [blame] | 96 | return stream_settings; |
pbos@webrtc.org | f577ae9 | 2014-03-19 08:43:57 +0000 | [diff] [blame] | 97 | } |
| 98 | |
perkj | fa10b55 | 2016-10-02 23:45:26 -0700 | [diff] [blame] | 99 | DefaultVideoStreamFactory::DefaultVideoStreamFactory() {} |
| 100 | |
| 101 | std::vector<VideoStream> DefaultVideoStreamFactory::CreateEncoderStreams( |
| 102 | int width, |
| 103 | int height, |
| 104 | const webrtc::VideoEncoderConfig& encoder_config) { |
| 105 | return CreateVideoStreams(width, height, encoder_config); |
| 106 | } |
| 107 | |
Niels Möller | 259a497 | 2018-04-05 15:36:51 +0200 | [diff] [blame] | 108 | void FillEncoderConfiguration(VideoCodecType codec_type, |
| 109 | size_t num_streams, |
perkj | fa10b55 | 2016-10-02 23:45:26 -0700 | [diff] [blame] | 110 | VideoEncoderConfig* configuration) { |
| 111 | RTC_DCHECK_LE(num_streams, DefaultVideoStreamFactory::kMaxNumberOfStreams); |
| 112 | |
Niels Möller | 259a497 | 2018-04-05 15:36:51 +0200 | [diff] [blame] | 113 | configuration->codec_type = codec_type; |
perkj | fa10b55 | 2016-10-02 23:45:26 -0700 | [diff] [blame] | 114 | configuration->number_of_streams = num_streams; |
| 115 | configuration->video_stream_factory = |
| 116 | new rtc::RefCountedObject<DefaultVideoStreamFactory>(); |
| 117 | configuration->max_bitrate_bps = 0; |
Seth Hampson | cc7125f | 2018-02-02 08:46:16 -0800 | [diff] [blame] | 118 | configuration->simulcast_layers = std::vector<VideoStream>(num_streams); |
perkj | fa10b55 | 2016-10-02 23:45:26 -0700 | [diff] [blame] | 119 | for (size_t i = 0; i < num_streams; ++i) { |
| 120 | configuration->max_bitrate_bps += |
| 121 | DefaultVideoStreamFactory::kMaxBitratePerStream[i]; |
| 122 | } |
| 123 | } |
| 124 | |
pbos@webrtc.org | 776e6f2 | 2014-10-29 15:28:39 +0000 | [diff] [blame] | 125 | VideoReceiveStream::Decoder CreateMatchingDecoder( |
Niels Möller | 259a497 | 2018-04-05 15:36:51 +0200 | [diff] [blame] | 126 | int payload_type, const std::string& payload_name) { |
pbos@webrtc.org | 776e6f2 | 2014-10-29 15:28:39 +0000 | [diff] [blame] | 127 | VideoReceiveStream::Decoder decoder; |
Niels Möller | 259a497 | 2018-04-05 15:36:51 +0200 | [diff] [blame] | 128 | decoder.payload_type = payload_type; |
| 129 | decoder.payload_name = payload_name; |
| 130 | if (payload_name == "H264") { |
Magnus Jedvert | 46a2765 | 2017-11-13 14:10:02 +0100 | [diff] [blame] | 131 | decoder.decoder = H264Decoder::Create().release(); |
Niels Möller | 259a497 | 2018-04-05 15:36:51 +0200 | [diff] [blame] | 132 | } else if (payload_name == "VP8") { |
Magnus Jedvert | 46a2765 | 2017-11-13 14:10:02 +0100 | [diff] [blame] | 133 | decoder.decoder = VP8Decoder::Create().release(); |
Niels Möller | 259a497 | 2018-04-05 15:36:51 +0200 | [diff] [blame] | 134 | } else if (payload_name == "VP9") { |
Magnus Jedvert | 46a2765 | 2017-11-13 14:10:02 +0100 | [diff] [blame] | 135 | decoder.decoder = VP9Decoder::Create().release(); |
Niels Möller | 259a497 | 2018-04-05 15:36:51 +0200 | [diff] [blame] | 136 | } else if (payload_name == "multiplex") { |
Emircan Uysaler | 03e6ec9 | 2018-03-09 15:03:26 -0800 | [diff] [blame] | 137 | decoder.decoder = new MultiplexDecoderAdapter( |
| 138 | new InternalDecoderFactory(), SdpVideoFormat(cricket::kVp9CodecName)); |
stefan@webrtc.org | b9f5453 | 2014-07-04 12:42:07 +0000 | [diff] [blame] | 139 | } else { |
pbos@webrtc.org | 776e6f2 | 2014-10-29 15:28:39 +0000 | [diff] [blame] | 140 | decoder.decoder = new FakeDecoder(); |
stefan@webrtc.org | b9f5453 | 2014-07-04 12:42:07 +0000 | [diff] [blame] | 141 | } |
pbos@webrtc.org | 776e6f2 | 2014-10-29 15:28:39 +0000 | [diff] [blame] | 142 | return decoder; |
pbos@webrtc.org | f577ae9 | 2014-03-19 08:43:57 +0000 | [diff] [blame] | 143 | } |
Niels Möller | 259a497 | 2018-04-05 15:36:51 +0200 | [diff] [blame] | 144 | |
| 145 | VideoReceiveStream::Decoder CreateMatchingDecoder( |
| 146 | const VideoSendStream::Config& config) { |
| 147 | return CreateMatchingDecoder(config.rtp.payload_type, |
| 148 | config.rtp.payload_name); |
| 149 | } |
| 150 | |
pbos@webrtc.org | f577ae9 | 2014-03-19 08:43:57 +0000 | [diff] [blame] | 151 | } // namespace test |
| 152 | } // namespace webrtc |