kwiberg@webrtc.org | ac2d27d | 2015-02-26 13:59:22 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. |
| 3 | * |
| 4 | * Use of this source code is governed by a BSD-style license |
| 5 | * that can be found in the LICENSE file in the root of the source |
| 6 | * tree. An additional intellectual property rights grant can be found |
| 7 | * in the file PATENTS. All contributing project authors may |
| 8 | * be found in the AUTHORS file in the root of the source tree. |
| 9 | */ |
| 10 | |
Mirko Bonadei | 7120742 | 2017-09-15 13:58:09 +0200 | [diff] [blame] | 11 | #include "common_types.h" // NOLINT(build/include) |
kwiberg@webrtc.org | ac2d27d | 2015-02-26 13:59:22 +0000 | [diff] [blame] | 12 | |
| 13 | #include <string.h> |
eladalon | d0244c2 | 2017-06-08 04:19:13 -0700 | [diff] [blame] | 14 | #include <algorithm> |
danilchap | ef8d773 | 2017-04-19 02:59:48 -0700 | [diff] [blame] | 15 | #include <limits> |
| 16 | #include <type_traits> |
kwiberg@webrtc.org | ac2d27d | 2015-02-26 13:59:22 +0000 | [diff] [blame] | 17 | |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 18 | #include "rtc_base/checks.h" |
| 19 | #include "rtc_base/stringutils.h" |
magjed | 10165ab | 2016-11-22 10:16:57 -0800 | [diff] [blame] | 20 | |
kwiberg@webrtc.org | ac2d27d | 2015-02-26 13:59:22 +0000 | [diff] [blame] | 21 | namespace webrtc { |
| 22 | |
hta | 257dc39 | 2016-10-25 09:05:06 -0700 | [diff] [blame] | 23 | VideoCodec::VideoCodec() |
| 24 | : codecType(kVideoCodecUnknown), |
| 25 | plName(), |
| 26 | plType(0), |
| 27 | width(0), |
| 28 | height(0), |
| 29 | startBitrate(0), |
| 30 | maxBitrate(0), |
| 31 | minBitrate(0), |
| 32 | targetBitrate(0), |
| 33 | maxFramerate(0), |
Seth Hampson | f6464c9 | 2018-01-17 13:55:14 -0800 | [diff] [blame] | 34 | active(true), |
hta | 257dc39 | 2016-10-25 09:05:06 -0700 | [diff] [blame] | 35 | qpMax(0), |
| 36 | numberOfSimulcastStreams(0), |
| 37 | simulcastStream(), |
| 38 | spatialLayers(), |
| 39 | mode(kRealtimeVideo), |
Erik Språng | 08127a9 | 2016-11-16 16:41:30 +0100 | [diff] [blame] | 40 | expect_encode_from_texture(false), |
ilnik | 04f4d12 | 2017-06-19 07:18:55 -0700 | [diff] [blame] | 41 | timing_frame_thresholds({0, 0}), |
hta | 527d347 | 2016-11-16 23:23:04 -0800 | [diff] [blame] | 42 | codec_specific_() {} |
hta | 257dc39 | 2016-10-25 09:05:06 -0700 | [diff] [blame] | 43 | |
| 44 | VideoCodecVP8* VideoCodec::VP8() { |
| 45 | RTC_DCHECK_EQ(codecType, kVideoCodecVP8); |
hta | 527d347 | 2016-11-16 23:23:04 -0800 | [diff] [blame] | 46 | return &codec_specific_.VP8; |
hta | 257dc39 | 2016-10-25 09:05:06 -0700 | [diff] [blame] | 47 | } |
| 48 | |
| 49 | const VideoCodecVP8& VideoCodec::VP8() const { |
| 50 | RTC_DCHECK_EQ(codecType, kVideoCodecVP8); |
hta | 527d347 | 2016-11-16 23:23:04 -0800 | [diff] [blame] | 51 | return codec_specific_.VP8; |
hta | 257dc39 | 2016-10-25 09:05:06 -0700 | [diff] [blame] | 52 | } |
| 53 | |
| 54 | VideoCodecVP9* VideoCodec::VP9() { |
| 55 | RTC_DCHECK_EQ(codecType, kVideoCodecVP9); |
hta | 527d347 | 2016-11-16 23:23:04 -0800 | [diff] [blame] | 56 | return &codec_specific_.VP9; |
hta | 257dc39 | 2016-10-25 09:05:06 -0700 | [diff] [blame] | 57 | } |
| 58 | |
| 59 | const VideoCodecVP9& VideoCodec::VP9() const { |
| 60 | RTC_DCHECK_EQ(codecType, kVideoCodecVP9); |
hta | 527d347 | 2016-11-16 23:23:04 -0800 | [diff] [blame] | 61 | return codec_specific_.VP9; |
hta | 257dc39 | 2016-10-25 09:05:06 -0700 | [diff] [blame] | 62 | } |
| 63 | |
| 64 | VideoCodecH264* VideoCodec::H264() { |
| 65 | RTC_DCHECK_EQ(codecType, kVideoCodecH264); |
hta | 527d347 | 2016-11-16 23:23:04 -0800 | [diff] [blame] | 66 | return &codec_specific_.H264; |
hta | 257dc39 | 2016-10-25 09:05:06 -0700 | [diff] [blame] | 67 | } |
| 68 | |
| 69 | const VideoCodecH264& VideoCodec::H264() const { |
| 70 | RTC_DCHECK_EQ(codecType, kVideoCodecH264); |
hta | 527d347 | 2016-11-16 23:23:04 -0800 | [diff] [blame] | 71 | return codec_specific_.H264; |
hta | 257dc39 | 2016-10-25 09:05:06 -0700 | [diff] [blame] | 72 | } |
| 73 | |
Erik Språng | 08127a9 | 2016-11-16 16:41:30 +0100 | [diff] [blame] | 74 | static const char* kPayloadNameVp8 = "VP8"; |
| 75 | static const char* kPayloadNameVp9 = "VP9"; |
| 76 | static const char* kPayloadNameH264 = "H264"; |
| 77 | static const char* kPayloadNameI420 = "I420"; |
| 78 | static const char* kPayloadNameRED = "RED"; |
| 79 | static const char* kPayloadNameULPFEC = "ULPFEC"; |
| 80 | static const char* kPayloadNameGeneric = "Generic"; |
Emircan Uysaler | 0a37547 | 2017-12-11 12:21:02 +0530 | [diff] [blame] | 81 | static const char* kPayloadNameStereo = "Stereo"; |
Erik Språng | 08127a9 | 2016-11-16 16:41:30 +0100 | [diff] [blame] | 82 | |
magjed | 10165ab | 2016-11-22 10:16:57 -0800 | [diff] [blame] | 83 | static bool CodecNamesEq(const char* name1, const char* name2) { |
| 84 | return _stricmp(name1, name2) == 0; |
| 85 | } |
| 86 | |
kthelgason | 1cdddc9 | 2017-08-24 03:52:48 -0700 | [diff] [blame] | 87 | const char* CodecTypeToPayloadString(VideoCodecType type) { |
Erik Språng | 08127a9 | 2016-11-16 16:41:30 +0100 | [diff] [blame] | 88 | switch (type) { |
| 89 | case kVideoCodecVP8: |
kthelgason | 1cdddc9 | 2017-08-24 03:52:48 -0700 | [diff] [blame] | 90 | return kPayloadNameVp8; |
Erik Språng | 08127a9 | 2016-11-16 16:41:30 +0100 | [diff] [blame] | 91 | case kVideoCodecVP9: |
kthelgason | 1cdddc9 | 2017-08-24 03:52:48 -0700 | [diff] [blame] | 92 | return kPayloadNameVp9; |
Erik Språng | 08127a9 | 2016-11-16 16:41:30 +0100 | [diff] [blame] | 93 | case kVideoCodecH264: |
kthelgason | 1cdddc9 | 2017-08-24 03:52:48 -0700 | [diff] [blame] | 94 | return kPayloadNameH264; |
Erik Språng | 08127a9 | 2016-11-16 16:41:30 +0100 | [diff] [blame] | 95 | case kVideoCodecI420: |
kthelgason | 1cdddc9 | 2017-08-24 03:52:48 -0700 | [diff] [blame] | 96 | return kPayloadNameI420; |
Erik Språng | 08127a9 | 2016-11-16 16:41:30 +0100 | [diff] [blame] | 97 | case kVideoCodecRED: |
kthelgason | 1cdddc9 | 2017-08-24 03:52:48 -0700 | [diff] [blame] | 98 | return kPayloadNameRED; |
Erik Språng | 08127a9 | 2016-11-16 16:41:30 +0100 | [diff] [blame] | 99 | case kVideoCodecULPFEC: |
kthelgason | 1cdddc9 | 2017-08-24 03:52:48 -0700 | [diff] [blame] | 100 | return kPayloadNameULPFEC; |
Emircan Uysaler | 0a37547 | 2017-12-11 12:21:02 +0530 | [diff] [blame] | 101 | // Other codecs default to generic. |
| 102 | case kVideoCodecStereo: |
| 103 | case kVideoCodecFlexfec: |
| 104 | case kVideoCodecGeneric: |
| 105 | case kVideoCodecUnknown: |
kthelgason | 1cdddc9 | 2017-08-24 03:52:48 -0700 | [diff] [blame] | 106 | return kPayloadNameGeneric; |
Erik Språng | 08127a9 | 2016-11-16 16:41:30 +0100 | [diff] [blame] | 107 | } |
Emircan Uysaler | 0a37547 | 2017-12-11 12:21:02 +0530 | [diff] [blame] | 108 | return kPayloadNameGeneric; |
Erik Språng | 08127a9 | 2016-11-16 16:41:30 +0100 | [diff] [blame] | 109 | } |
| 110 | |
kthelgason | 1cdddc9 | 2017-08-24 03:52:48 -0700 | [diff] [blame] | 111 | VideoCodecType PayloadStringToCodecType(const std::string& name) { |
magjed | 10165ab | 2016-11-22 10:16:57 -0800 | [diff] [blame] | 112 | if (CodecNamesEq(name.c_str(), kPayloadNameVp8)) |
kthelgason | 1cdddc9 | 2017-08-24 03:52:48 -0700 | [diff] [blame] | 113 | return kVideoCodecVP8; |
magjed | 10165ab | 2016-11-22 10:16:57 -0800 | [diff] [blame] | 114 | if (CodecNamesEq(name.c_str(), kPayloadNameVp9)) |
kthelgason | 1cdddc9 | 2017-08-24 03:52:48 -0700 | [diff] [blame] | 115 | return kVideoCodecVP9; |
magjed | 10165ab | 2016-11-22 10:16:57 -0800 | [diff] [blame] | 116 | if (CodecNamesEq(name.c_str(), kPayloadNameH264)) |
kthelgason | 1cdddc9 | 2017-08-24 03:52:48 -0700 | [diff] [blame] | 117 | return kVideoCodecH264; |
magjed | 10165ab | 2016-11-22 10:16:57 -0800 | [diff] [blame] | 118 | if (CodecNamesEq(name.c_str(), kPayloadNameI420)) |
kthelgason | 1cdddc9 | 2017-08-24 03:52:48 -0700 | [diff] [blame] | 119 | return kVideoCodecI420; |
magjed | 10165ab | 2016-11-22 10:16:57 -0800 | [diff] [blame] | 120 | if (CodecNamesEq(name.c_str(), kPayloadNameRED)) |
kthelgason | 1cdddc9 | 2017-08-24 03:52:48 -0700 | [diff] [blame] | 121 | return kVideoCodecRED; |
magjed | 10165ab | 2016-11-22 10:16:57 -0800 | [diff] [blame] | 122 | if (CodecNamesEq(name.c_str(), kPayloadNameULPFEC)) |
kthelgason | 1cdddc9 | 2017-08-24 03:52:48 -0700 | [diff] [blame] | 123 | return kVideoCodecULPFEC; |
Emircan Uysaler | 0a37547 | 2017-12-11 12:21:02 +0530 | [diff] [blame] | 124 | if (CodecNamesEq(name.c_str(), kPayloadNameStereo)) |
| 125 | return kVideoCodecStereo; |
kthelgason | 1cdddc9 | 2017-08-24 03:52:48 -0700 | [diff] [blame] | 126 | return kVideoCodecGeneric; |
| 127 | } |
| 128 | |
Erik Språng | 08127a9 | 2016-11-16 16:41:30 +0100 | [diff] [blame] | 129 | const uint32_t BitrateAllocation::kMaxBitrateBps = |
| 130 | std::numeric_limits<uint32_t>::max(); |
| 131 | |
erikvarga@webrtc.org | 01f2ec3 | 2017-11-15 14:58:23 +0100 | [diff] [blame] | 132 | BitrateAllocation::BitrateAllocation() : sum_(0), bitrates_{}, has_bitrate_{} {} |
Erik Språng | 08127a9 | 2016-11-16 16:41:30 +0100 | [diff] [blame] | 133 | |
| 134 | bool BitrateAllocation::SetBitrate(size_t spatial_index, |
| 135 | size_t temporal_index, |
| 136 | uint32_t bitrate_bps) { |
sprang | 6d314c7 | 2016-12-06 06:08:53 -0800 | [diff] [blame] | 137 | RTC_CHECK_LT(spatial_index, kMaxSpatialLayers); |
| 138 | RTC_CHECK_LT(temporal_index, kMaxTemporalStreams); |
| 139 | RTC_CHECK_LE(bitrates_[spatial_index][temporal_index], sum_); |
Erik Språng | 08127a9 | 2016-11-16 16:41:30 +0100 | [diff] [blame] | 140 | uint64_t new_bitrate_sum_bps = sum_; |
| 141 | new_bitrate_sum_bps -= bitrates_[spatial_index][temporal_index]; |
| 142 | new_bitrate_sum_bps += bitrate_bps; |
| 143 | if (new_bitrate_sum_bps > kMaxBitrateBps) |
| 144 | return false; |
| 145 | |
| 146 | bitrates_[spatial_index][temporal_index] = bitrate_bps; |
erikvarga@webrtc.org | 01f2ec3 | 2017-11-15 14:58:23 +0100 | [diff] [blame] | 147 | has_bitrate_[spatial_index][temporal_index] = true; |
Erik Språng | 08127a9 | 2016-11-16 16:41:30 +0100 | [diff] [blame] | 148 | sum_ = static_cast<uint32_t>(new_bitrate_sum_bps); |
| 149 | return true; |
| 150 | } |
| 151 | |
erikvarga@webrtc.org | 01f2ec3 | 2017-11-15 14:58:23 +0100 | [diff] [blame] | 152 | bool BitrateAllocation::HasBitrate(size_t spatial_index, |
| 153 | size_t temporal_index) const { |
| 154 | RTC_CHECK_LT(spatial_index, kMaxSpatialLayers); |
| 155 | RTC_CHECK_LT(temporal_index, kMaxTemporalStreams); |
| 156 | return has_bitrate_[spatial_index][temporal_index]; |
| 157 | } |
| 158 | |
Erik Språng | 08127a9 | 2016-11-16 16:41:30 +0100 | [diff] [blame] | 159 | uint32_t BitrateAllocation::GetBitrate(size_t spatial_index, |
| 160 | size_t temporal_index) const { |
sprang | 6d314c7 | 2016-12-06 06:08:53 -0800 | [diff] [blame] | 161 | RTC_CHECK_LT(spatial_index, kMaxSpatialLayers); |
| 162 | RTC_CHECK_LT(temporal_index, kMaxTemporalStreams); |
Erik Språng | 08127a9 | 2016-11-16 16:41:30 +0100 | [diff] [blame] | 163 | return bitrates_[spatial_index][temporal_index]; |
| 164 | } |
| 165 | |
erikvarga@webrtc.org | 01f2ec3 | 2017-11-15 14:58:23 +0100 | [diff] [blame] | 166 | // Whether the specific spatial layers has the bitrate set in any of its |
| 167 | // temporal layers. |
| 168 | bool BitrateAllocation::IsSpatialLayerUsed(size_t spatial_index) const { |
| 169 | RTC_CHECK_LT(spatial_index, kMaxSpatialLayers); |
| 170 | for (int i = 0; i < kMaxTemporalStreams; ++i) { |
| 171 | if (has_bitrate_[spatial_index][i]) |
| 172 | return true; |
| 173 | } |
| 174 | return false; |
| 175 | } |
| 176 | |
Erik Språng | 08127a9 | 2016-11-16 16:41:30 +0100 | [diff] [blame] | 177 | // Get the sum of all the temporal layer for a specific spatial layer. |
| 178 | uint32_t BitrateAllocation::GetSpatialLayerSum(size_t spatial_index) const { |
sprang | 6d314c7 | 2016-12-06 06:08:53 -0800 | [diff] [blame] | 179 | RTC_CHECK_LT(spatial_index, kMaxSpatialLayers); |
Erik Språng | 08127a9 | 2016-11-16 16:41:30 +0100 | [diff] [blame] | 180 | uint32_t sum = 0; |
| 181 | for (int i = 0; i < kMaxTemporalStreams; ++i) |
| 182 | sum += bitrates_[spatial_index][i]; |
| 183 | return sum; |
| 184 | } |
| 185 | |
sprang | d0fc37a | 2017-06-22 05:40:25 -0700 | [diff] [blame] | 186 | std::string BitrateAllocation::ToString() const { |
| 187 | if (sum_ == 0) |
| 188 | return "BitrateAllocation [ [] ]"; |
| 189 | |
| 190 | // TODO(sprang): Replace this stringstream with something cheaper. |
| 191 | std::ostringstream oss; |
| 192 | oss << "BitrateAllocation ["; |
| 193 | uint32_t spatial_cumulator = 0; |
| 194 | for (int si = 0; si < kMaxSpatialLayers; ++si) { |
| 195 | RTC_DCHECK_LE(spatial_cumulator, sum_); |
| 196 | if (spatial_cumulator == sum_) |
| 197 | break; |
| 198 | |
| 199 | const uint32_t layer_sum = GetSpatialLayerSum(si); |
| 200 | if (layer_sum == sum_) { |
| 201 | oss << " ["; |
| 202 | } else { |
| 203 | if (si > 0) |
| 204 | oss << ","; |
| 205 | oss << std::endl << " ["; |
| 206 | } |
| 207 | spatial_cumulator += layer_sum; |
| 208 | |
| 209 | uint32_t temporal_cumulator = 0; |
| 210 | for (int ti = 0; ti < kMaxTemporalStreams; ++ti) { |
| 211 | RTC_DCHECK_LE(temporal_cumulator, layer_sum); |
| 212 | if (temporal_cumulator == layer_sum) |
| 213 | break; |
| 214 | |
| 215 | if (ti > 0) |
| 216 | oss << ", "; |
| 217 | |
| 218 | uint32_t bitrate = bitrates_[si][ti]; |
| 219 | oss << bitrate; |
| 220 | temporal_cumulator += bitrate; |
| 221 | } |
| 222 | oss << "]"; |
| 223 | } |
| 224 | |
| 225 | RTC_DCHECK_EQ(spatial_cumulator, sum_); |
| 226 | oss << " ]"; |
| 227 | return oss.str(); |
| 228 | } |
| 229 | |
| 230 | std::ostream& BitrateAllocation::operator<<(std::ostream& os) const { |
| 231 | os << ToString(); |
| 232 | return os; |
| 233 | } |
| 234 | |
kwiberg@webrtc.org | ac2d27d | 2015-02-26 13:59:22 +0000 | [diff] [blame] | 235 | } // namespace webrtc |