OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #include "webrtc/video_encoder.h" | 11 #include "webrtc/video_encoder.h" |
12 | 12 |
13 #include "webrtc/base/checks.h" | 13 #include "webrtc/base/checks.h" |
14 #include "webrtc/base/logging.h" | 14 #include "webrtc/base/logging.h" |
15 #include "webrtc/modules/video_coding/codecs/h264/include/h264.h" | 15 #include "webrtc/modules/video_coding/codecs/h264/include/h264.h" |
16 #include "webrtc/modules/video_coding/codecs/vp8/include/vp8.h" | 16 #include "webrtc/modules/video_coding/codecs/vp8/include/vp8.h" |
17 #include "webrtc/modules/video_coding/codecs/vp9/include/vp9.h" | 17 #include "webrtc/modules/video_coding/codecs/vp9/include/vp9.h" |
18 | 18 |
19 namespace webrtc { | 19 namespace webrtc { |
20 VideoEncoder* VideoEncoder::Create(VideoEncoder::EncoderType codec_type) { | 20 VideoEncoder* VideoEncoder::Create(VideoEncoder::EncoderType codec_type) { |
| 21 RTC_DCHECK(IsSupportedSoftware(codec_type)); |
21 switch (codec_type) { | 22 switch (codec_type) { |
22 case kH264: | 23 case kH264: |
23 RTC_DCHECK(H264Encoder::IsSupported()); | |
24 return H264Encoder::Create(); | 24 return H264Encoder::Create(); |
25 case kVp8: | 25 case kVp8: |
26 return VP8Encoder::Create(); | 26 return VP8Encoder::Create(); |
27 case kVp9: | 27 case kVp9: |
28 RTC_DCHECK(VP9Encoder::IsSupported()); | |
29 return VP9Encoder::Create(); | 28 return VP9Encoder::Create(); |
30 case kUnsupportedCodec: | 29 case kUnsupportedCodec: |
31 RTC_NOTREACHED(); | 30 RTC_NOTREACHED(); |
32 return nullptr; | 31 return nullptr; |
33 } | 32 } |
34 RTC_NOTREACHED(); | 33 RTC_NOTREACHED(); |
35 return nullptr; | 34 return nullptr; |
36 } | 35 } |
37 | 36 |
38 VideoEncoder::EncoderType CodecToEncoderType(VideoCodecType codec_type) { | 37 bool VideoEncoder::IsSupportedSoftware(EncoderType codec_type) { |
| 38 switch (codec_type) { |
| 39 case kH264: |
| 40 return H264Encoder::IsSupported(); |
| 41 case kVp8: |
| 42 return true; |
| 43 case kVp9: |
| 44 return VP9Encoder::IsSupported(); |
| 45 case kUnsupportedCodec: |
| 46 RTC_NOTREACHED(); |
| 47 return false; |
| 48 } |
| 49 RTC_NOTREACHED(); |
| 50 return false; |
| 51 } |
| 52 |
| 53 VideoEncoder::EncoderType VideoEncoder::CodecToEncoderType( |
| 54 VideoCodecType codec_type) { |
39 switch (codec_type) { | 55 switch (codec_type) { |
40 case kVideoCodecH264: | 56 case kVideoCodecH264: |
41 return VideoEncoder::kH264; | 57 return VideoEncoder::kH264; |
42 case kVideoCodecVP8: | 58 case kVideoCodecVP8: |
43 return VideoEncoder::kVp8; | 59 return VideoEncoder::kVp8; |
44 case kVideoCodecVP9: | 60 case kVideoCodecVP9: |
45 return VideoEncoder::kVp9; | 61 return VideoEncoder::kVp9; |
46 default: | 62 default: |
47 return VideoEncoder::kUnsupportedCodec; | 63 return VideoEncoder::kUnsupportedCodec; |
48 } | 64 } |
49 } | 65 } |
50 | 66 |
51 VideoEncoderSoftwareFallbackWrapper::VideoEncoderSoftwareFallbackWrapper( | 67 VideoEncoderSoftwareFallbackWrapper::VideoEncoderSoftwareFallbackWrapper( |
52 VideoCodecType codec_type, | 68 VideoCodecType codec_type, |
53 webrtc::VideoEncoder* encoder) | 69 webrtc::VideoEncoder* encoder) |
54 : rates_set_(false), | 70 : rates_set_(false), |
55 channel_parameters_set_(false), | 71 channel_parameters_set_(false), |
56 encoder_type_(CodecToEncoderType(codec_type)), | 72 encoder_type_(CodecToEncoderType(codec_type)), |
57 encoder_(encoder), | 73 encoder_(encoder), |
58 callback_(nullptr) {} | 74 callback_(nullptr) {} |
59 | 75 |
60 bool VideoEncoderSoftwareFallbackWrapper::InitFallbackEncoder() { | 76 bool VideoEncoderSoftwareFallbackWrapper::InitFallbackEncoder() { |
61 RTC_CHECK(encoder_type_ != kUnsupportedCodec) | 77 if (!VideoEncoder::IsSupportedSoftware(encoder_type_)) { |
62 << "Encoder requesting fallback to codec not supported in software."; | 78 LOG(LS_WARNING) |
| 79 << "Encoder requesting fallback to codec not supported in software."; |
| 80 return false; |
| 81 } |
63 fallback_encoder_.reset(VideoEncoder::Create(encoder_type_)); | 82 fallback_encoder_.reset(VideoEncoder::Create(encoder_type_)); |
64 if (fallback_encoder_->InitEncode(&codec_settings_, number_of_cores_, | 83 if (fallback_encoder_->InitEncode(&codec_settings_, number_of_cores_, |
65 max_payload_size_) != | 84 max_payload_size_) != |
66 WEBRTC_VIDEO_CODEC_OK) { | 85 WEBRTC_VIDEO_CODEC_OK) { |
67 LOG(LS_ERROR) << "Failed to initialize software-encoder fallback."; | 86 LOG(LS_ERROR) << "Failed to initialize software-encoder fallback."; |
68 fallback_encoder_->Release(); | 87 fallback_encoder_->Release(); |
69 fallback_encoder_.reset(); | 88 fallback_encoder_.reset(); |
70 return false; | 89 return false; |
71 } | 90 } |
72 // Replay callback, rates, and channel parameters. | 91 // Replay callback, rates, and channel parameters. |
(...skipping 65 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
138 | 157 |
139 int32_t VideoEncoderSoftwareFallbackWrapper::Encode( | 158 int32_t VideoEncoderSoftwareFallbackWrapper::Encode( |
140 const VideoFrame& frame, | 159 const VideoFrame& frame, |
141 const CodecSpecificInfo* codec_specific_info, | 160 const CodecSpecificInfo* codec_specific_info, |
142 const std::vector<FrameType>* frame_types) { | 161 const std::vector<FrameType>* frame_types) { |
143 if (fallback_encoder_) | 162 if (fallback_encoder_) |
144 return fallback_encoder_->Encode(frame, codec_specific_info, frame_types); | 163 return fallback_encoder_->Encode(frame, codec_specific_info, frame_types); |
145 int32_t ret = encoder_->Encode(frame, codec_specific_info, frame_types); | 164 int32_t ret = encoder_->Encode(frame, codec_specific_info, frame_types); |
146 // If requested, try a software fallback. | 165 // If requested, try a software fallback. |
147 if (ret == WEBRTC_VIDEO_CODEC_FALLBACK_SOFTWARE && InitFallbackEncoder()) { | 166 if (ret == WEBRTC_VIDEO_CODEC_FALLBACK_SOFTWARE && InitFallbackEncoder()) { |
| 167 if (frame.video_frame_buffer()->native_handle() && |
| 168 !fallback_encoder_->SupportsNativeHandle()) { |
| 169 LOG(LS_WARNING) << "Fallback encoder doesn't support native frames, " |
| 170 << "dropping one frame."; |
| 171 return WEBRTC_VIDEO_CODEC_ERROR; |
| 172 } |
| 173 |
148 // Fallback was successful, so start using it with this frame. | 174 // Fallback was successful, so start using it with this frame. |
149 return fallback_encoder_->Encode(frame, codec_specific_info, frame_types); | 175 return fallback_encoder_->Encode(frame, codec_specific_info, frame_types); |
150 } | 176 } |
151 return ret; | 177 return ret; |
152 } | 178 } |
153 | 179 |
154 int32_t VideoEncoderSoftwareFallbackWrapper::SetChannelParameters( | 180 int32_t VideoEncoderSoftwareFallbackWrapper::SetChannelParameters( |
155 uint32_t packet_loss, | 181 uint32_t packet_loss, |
156 int64_t rtt) { | 182 int64_t rtt) { |
157 channel_parameters_set_ = true; | 183 channel_parameters_set_ = true; |
(...skipping 22 matching lines...) Expand all Loading... |
180 return encoder_->OnDroppedFrame(); | 206 return encoder_->OnDroppedFrame(); |
181 } | 207 } |
182 | 208 |
183 bool VideoEncoderSoftwareFallbackWrapper::SupportsNativeHandle() const { | 209 bool VideoEncoderSoftwareFallbackWrapper::SupportsNativeHandle() const { |
184 if (fallback_encoder_) | 210 if (fallback_encoder_) |
185 return fallback_encoder_->SupportsNativeHandle(); | 211 return fallback_encoder_->SupportsNativeHandle(); |
186 return encoder_->SupportsNativeHandle(); | 212 return encoder_->SupportsNativeHandle(); |
187 } | 213 } |
188 | 214 |
189 } // namespace webrtc | 215 } // namespace webrtc |
OLD | NEW |