diff options
author | Chih-hung Hsieh <chh@google.com> | 2016-01-20 17:50:13 +0000 |
---|---|---|
committer | android-build-merger <android-build-merger@google.com> | 2016-01-20 17:50:13 +0000 |
commit | b3cb8ab4ede8bb77f0bdef2715efc2c1e6267072 (patch) | |
tree | 28c4cf735dd5bd9cc8f1ccd06fff8a173b20d1cb /webrtc/modules/video_coding/video_sender.cc | |
parent | a4acd9d6bc9b3b033d7d274316e75ee067df8d20 (diff) | |
parent | 9a337512d97e37afc142dee4fd50a41b741a87d2 (diff) | |
download | webrtc-nougat-mr2.2-release.tar.gz |
Merge "Merge upstream SHA 04cb763"android-cts_7.1_r1android-cts-7.1_r9android-cts-7.1_r8android-cts-7.1_r7android-cts-7.1_r6android-cts-7.1_r5android-cts-7.1_r4android-cts-7.1_r3android-cts-7.1_r29android-cts-7.1_r28android-cts-7.1_r27android-cts-7.1_r26android-cts-7.1_r25android-cts-7.1_r24android-cts-7.1_r23android-cts-7.1_r22android-cts-7.1_r21android-cts-7.1_r20android-cts-7.1_r2android-cts-7.1_r19android-cts-7.1_r18android-cts-7.1_r17android-cts-7.1_r16android-cts-7.1_r15android-cts-7.1_r14android-cts-7.1_r13android-cts-7.1_r12android-cts-7.1_r11android-cts-7.1_r10android-cts-7.1_r1android-cts-7.0_r9android-cts-7.0_r8android-cts-7.0_r7android-cts-7.0_r6android-cts-7.0_r5android-cts-7.0_r4android-cts-7.0_r33android-cts-7.0_r32android-cts-7.0_r31android-cts-7.0_r30android-cts-7.0_r3android-cts-7.0_r29android-cts-7.0_r28android-cts-7.0_r27android-cts-7.0_r26android-cts-7.0_r25android-cts-7.0_r24android-cts-7.0_r23android-cts-7.0_r22android-cts-7.0_r21android-cts-7.0_r20android-cts-7.0_r2android-cts-7.0_r19android-cts-7.0_r18android-cts-7.0_r17android-cts-7.0_r16android-cts-7.0_r15android-cts-7.0_r14android-cts-7.0_r13android-cts-7.0_r12android-cts-7.0_r11android-cts-7.0_r10android-cts-7.0_r1android-7.1.2_r9android-7.1.2_r8android-7.1.2_r6android-7.1.2_r5android-7.1.2_r4android-7.1.2_r39android-7.1.2_r38android-7.1.2_r37android-7.1.2_r36android-7.1.2_r33android-7.1.2_r32android-7.1.2_r30android-7.1.2_r3android-7.1.2_r29android-7.1.2_r28android-7.1.2_r27android-7.1.2_r25android-7.1.2_r24android-7.1.2_r23android-7.1.2_r2android-7.1.2_r19android-7.1.2_r18android-7.1.2_r17android-7.1.2_r16android-7.1.2_r15android-7.1.2_r14android-7.1.2_r13android-7.1.2_r12android-7.1.2_r11android-7.1.2_r10android-7.1.2_r1android-7.1.1_r9android-7.1.1_r8android-7.1.1_r7android-7.1.1_r61android-7.1.1_r60android-7.1.1_r6android-7.1.1_r59android-7.1.1_r58android-7.1.1_r57android-7.1.1_r56android-7.1.1_r55android-7.1.1_r54android-7.1.1_r53android-7.1.1_r52android-7.1.1_r51android-7.1.1_r50android-7.1.1_r49android-7.1.1_r48android-7.1.1_r47android-7.1.1_r46android-7.1.1_r45android-7.1.1_r44android-7.1.1_r43android-7.1.1_r42android-7.1.1_r41android-7.1.1_r40android-7.1.1_r4android-7.1.1_r39android-7.1.1_r38android-7.1.1_r35android-7.1.1_r33android-7.1.1_r32android-7.1.1_r31android-7.1.1_r3android-7.1.1_r28android-7.1.1_r27android-7.1.1_r26android-7.1.1_r25android-7.1.1_r24android-7.1.1_r23android-7.1.1_r22android-7.1.1_r21android-7.1.1_r20android-7.1.1_r2android-7.1.1_r17android-7.1.1_r16android-7.1.1_r15android-7.1.1_r14android-7.1.1_r13android-7.1.1_r12android-7.1.1_r11android-7.1.1_r10android-7.1.1_r1android-7.1.0_r7android-7.1.0_r6android-7.1.0_r5android-7.1.0_r4android-7.1.0_r3android-7.1.0_r2android-7.1.0_r1android-7.0.0_r9android-7.0.0_r8android-7.0.0_r7android-7.0.0_r6android-7.0.0_r5android-7.0.0_r4android-7.0.0_r36android-7.0.0_r35android-7.0.0_r34android-7.0.0_r33android-7.0.0_r32android-7.0.0_r31android-7.0.0_r30android-7.0.0_r3android-7.0.0_r29android-7.0.0_r28android-7.0.0_r27android-7.0.0_r24android-7.0.0_r21android-7.0.0_r19android-7.0.0_r17android-7.0.0_r15android-7.0.0_r14android-7.0.0_r13android-7.0.0_r12android-7.0.0_r11android-7.0.0_r10android-7.0.0_r1nougat-releasenougat-mr2.3-releasenougat-mr2.2-releasenougat-mr2.1-releasenougat-mr2-security-releasenougat-mr2-releasenougat-mr2-pixel-releasenougat-mr2-devnougat-mr1.8-releasenougat-mr1.7-releasenougat-mr1.6-releasenougat-mr1.5-releasenougat-mr1.4-releasenougat-mr1.3-releasenougat-mr1.2-releasenougat-mr1.1-releasenougat-mr1-volantis-releasenougat-mr1-security-releasenougat-mr1-releasenougat-mr1-flounder-releasenougat-mr1-devnougat-mr1-cts-releasenougat-mr0.5-releasenougat-dr1-releasenougat-devnougat-cts-releasenougat-bugfix-release
am: 9a337512d9
* commit '9a337512d97e37afc142dee4fd50a41b741a87d2': (797 commits)
Add tests for verifying transport feedback for audio and video.
Eliminate defines in talk/
Revert of Update with new default boringssl no-aes cipher suites. Re-enable tests. (patchset #3 id:40001 of https://codereview.webrtc.org/1550773002/ )
Remove assert which was incorrectly added to TcpPort::OnSentPacket.
Reland Connect TurnPort and TCPPort to AsyncPacketSocket::SignalSentPacket.
Update with new default boringssl no-aes cipher suites. Re-enable tests.
Revert of Connect TurnPort and TCPPort to AsyncPacketSocket::SignalSentPacket. (patchset #3 id:40001 of https://codereview.webrtc.org/1577873003/ )
Re-land: "Use an explicit identifier in Config"
Connect TurnPort and TCPPort to AsyncPacketSocket::SignalSentPacket.
Revert of Delete remnants of non-square pixel support from cricket::VideoFrame. (patchset #1 id:1 of https://codereview.webrtc.org/1586613002/ )
Remove libfuzzer trybot from default trybot set.
Add ramp-up tests for transport sequence number with and w/o audio.
Delete remnants of non-square pixel support from cricket::VideoFrame.
Fix IPAddress::ToSensitiveString() to avoid dependency on inet_ntop().
Revert of Storing raw audio sink for default audio track. (patchset #7 id:120001 of https://codereview.chromium.org/1551813002/ )
Re-enable tests that failed under Linux_Msan.
Revert of Use an explicit identifier in Config (patchset #4 id:60001 of https://codereview.webrtc.org/1538643004/ )
Roll chromium_revision 346fea9..099be58 (369082:369139)
Disable WebRtcVideoChannel2BaseTest.SendManyResizeOnce for TSan
Add build_protobuf variable.
...
Diffstat (limited to 'webrtc/modules/video_coding/video_sender.cc')
-rw-r--r-- | webrtc/modules/video_coding/video_sender.cc | 352 |
1 files changed, 352 insertions, 0 deletions
diff --git a/webrtc/modules/video_coding/video_sender.cc b/webrtc/modules/video_coding/video_sender.cc new file mode 100644 index 0000000000..ac901f95b9 --- /dev/null +++ b/webrtc/modules/video_coding/video_sender.cc @@ -0,0 +1,352 @@ +/* + * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved. + * + * Use of this source code is governed by a BSD-style license + * that can be found in the LICENSE file in the root of the source + * tree. An additional intellectual property rights grant can be found + * in the file PATENTS. All contributing project authors may + * be found in the AUTHORS file in the root of the source tree. + */ + + +#include <algorithm> // std::max + +#include "webrtc/base/checks.h" +#include "webrtc/base/logging.h" +#include "webrtc/common_types.h" +#include "webrtc/common_video/libyuv/include/webrtc_libyuv.h" +#include "webrtc/modules/video_coding/include/video_codec_interface.h" +#include "webrtc/modules/video_coding/encoded_frame.h" +#include "webrtc/modules/video_coding/utility/quality_scaler.h" +#include "webrtc/modules/video_coding/video_coding_impl.h" +#include "webrtc/system_wrappers/include/clock.h" + +namespace webrtc { +namespace vcm { + +VideoSender::VideoSender(Clock* clock, + EncodedImageCallback* post_encode_callback, + VideoEncoderRateObserver* encoder_rate_observer, + VCMQMSettingsCallback* qm_settings_callback) + : clock_(clock), + process_crit_sect_(CriticalSectionWrapper::CreateCriticalSection()), + _encoder(nullptr), + _encodedFrameCallback(post_encode_callback), + _nextFrameTypes(1, kVideoFrameDelta), + _mediaOpt(clock_), + _sendStatsCallback(nullptr), + _codecDataBase(encoder_rate_observer, &_encodedFrameCallback), + frame_dropper_enabled_(true), + _sendStatsTimer(1000, clock_), + current_codec_(), + qm_settings_callback_(qm_settings_callback), + protection_callback_(nullptr), + encoder_params_({0, 0, 0, 0}) { + // Allow VideoSender to be created on one thread but used on another, post + // construction. This is currently how this class is being used by at least + // one external project (diffractor). + _mediaOpt.EnableQM(qm_settings_callback_ != nullptr); + _mediaOpt.Reset(); + main_thread_.DetachFromThread(); +} + +VideoSender::~VideoSender() {} + +int32_t VideoSender::Process() { + int32_t returnValue = VCM_OK; + + if (_sendStatsTimer.TimeUntilProcess() == 0) { + _sendStatsTimer.Processed(); + CriticalSectionScoped cs(process_crit_sect_.get()); + if (_sendStatsCallback != nullptr) { + uint32_t bitRate = _mediaOpt.SentBitRate(); + uint32_t frameRate = _mediaOpt.SentFrameRate(); + _sendStatsCallback->SendStatistics(bitRate, frameRate); + } + } + + { + rtc::CritScope cs(¶ms_lock_); + // Force an encoder parameters update, so that incoming frame rate is + // updated even if bandwidth hasn't changed. + encoder_params_.input_frame_rate = _mediaOpt.InputFrameRate(); + } + + return returnValue; +} + +int64_t VideoSender::TimeUntilNextProcess() { + return _sendStatsTimer.TimeUntilProcess(); +} + +// Register the send codec to be used. +int32_t VideoSender::RegisterSendCodec(const VideoCodec* sendCodec, + uint32_t numberOfCores, + uint32_t maxPayloadSize) { + RTC_DCHECK(main_thread_.CalledOnValidThread()); + rtc::CritScope lock(&send_crit_); + if (sendCodec == nullptr) { + return VCM_PARAMETER_ERROR; + } + + bool ret = + _codecDataBase.SetSendCodec(sendCodec, numberOfCores, maxPayloadSize); + + // Update encoder regardless of result to make sure that we're not holding on + // to a deleted instance. + _encoder = _codecDataBase.GetEncoder(); + // Cache the current codec here so they can be fetched from this thread + // without requiring the _sendCritSect lock. + current_codec_ = *sendCodec; + + if (!ret) { + LOG(LS_ERROR) << "Failed to initialize set encoder with payload name '" + << sendCodec->plName << "'."; + return VCM_CODEC_ERROR; + } + + int numLayers; + if (sendCodec->codecType == kVideoCodecVP8) { + numLayers = sendCodec->codecSpecific.VP8.numberOfTemporalLayers; + } else if (sendCodec->codecType == kVideoCodecVP9) { + numLayers = sendCodec->codecSpecific.VP9.numberOfTemporalLayers; + } else { + numLayers = 1; + } + + // If we have screensharing and we have layers, we disable frame dropper. + bool disable_frame_dropper = + numLayers > 1 && sendCodec->mode == kScreensharing; + if (disable_frame_dropper) { + _mediaOpt.EnableFrameDropper(false); + } else if (frame_dropper_enabled_) { + _mediaOpt.EnableFrameDropper(true); + } + _nextFrameTypes.clear(); + _nextFrameTypes.resize(VCM_MAX(sendCodec->numberOfSimulcastStreams, 1), + kVideoFrameDelta); + + _mediaOpt.SetEncodingData(sendCodec->codecType, sendCodec->maxBitrate * 1000, + sendCodec->startBitrate * 1000, sendCodec->width, + sendCodec->height, sendCodec->maxFramerate, + numLayers, maxPayloadSize); + return VCM_OK; +} + +// Register an external decoder object. +// This can not be used together with external decoder callbacks. +void VideoSender::RegisterExternalEncoder(VideoEncoder* externalEncoder, + uint8_t payloadType, + bool internalSource /*= false*/) { + RTC_DCHECK(main_thread_.CalledOnValidThread()); + + rtc::CritScope lock(&send_crit_); + + if (externalEncoder == nullptr) { + bool wasSendCodec = false; + RTC_CHECK( + _codecDataBase.DeregisterExternalEncoder(payloadType, &wasSendCodec)); + if (wasSendCodec) { + // Make sure the VCM doesn't use the de-registered codec + _encoder = nullptr; + } + return; + } + _codecDataBase.RegisterExternalEncoder(externalEncoder, payloadType, + internalSource); +} + +// Get encode bitrate +int VideoSender::Bitrate(unsigned int* bitrate) const { + RTC_DCHECK(main_thread_.CalledOnValidThread()); + // Since we're running on the thread that's the only thread known to modify + // the value of _encoder, we don't need to grab the lock here. + + if (!_encoder) + return VCM_UNINITIALIZED; + *bitrate = _encoder->GetEncoderParameters().target_bitrate; + return 0; +} + +// Get encode frame rate +int VideoSender::FrameRate(unsigned int* framerate) const { + RTC_DCHECK(main_thread_.CalledOnValidThread()); + // Since we're running on the thread that's the only thread known to modify + // the value of _encoder, we don't need to grab the lock here. + + if (!_encoder) + return VCM_UNINITIALIZED; + + *framerate = _encoder->GetEncoderParameters().input_frame_rate; + return 0; +} + +int32_t VideoSender::SetChannelParameters(uint32_t target_bitrate, + uint8_t lossRate, + int64_t rtt) { + uint32_t target_rate = + _mediaOpt.SetTargetRates(target_bitrate, lossRate, rtt, + protection_callback_, qm_settings_callback_); + + uint32_t input_frame_rate = _mediaOpt.InputFrameRate(); + + rtc::CritScope cs(¶ms_lock_); + encoder_params_ = {target_rate, lossRate, rtt, input_frame_rate}; + + return VCM_OK; +} + +void VideoSender::SetEncoderParameters(EncoderParameters params) { + if (params.target_bitrate == 0) + return; + + if (params.input_frame_rate == 0) { + // No frame rate estimate available, use default. + params.input_frame_rate = current_codec_.maxFramerate; + } + if (_encoder != nullptr) + _encoder->SetEncoderParameters(params); +} + +int32_t VideoSender::RegisterTransportCallback( + VCMPacketizationCallback* transport) { + rtc::CritScope lock(&send_crit_); + _encodedFrameCallback.SetMediaOpt(&_mediaOpt); + _encodedFrameCallback.SetTransportCallback(transport); + return VCM_OK; +} + +// Register video output information callback which will be called to deliver +// information about the video stream produced by the encoder, for instance the +// average frame rate and bit rate. +int32_t VideoSender::RegisterSendStatisticsCallback( + VCMSendStatisticsCallback* sendStats) { + CriticalSectionScoped cs(process_crit_sect_.get()); + _sendStatsCallback = sendStats; + return VCM_OK; +} + +// Register a video protection callback which will be called to deliver the +// requested FEC rate and NACK status (on/off). +// Note: this callback is assumed to only be registered once and before it is +// used in this class. +int32_t VideoSender::RegisterProtectionCallback( + VCMProtectionCallback* protection_callback) { + RTC_DCHECK(protection_callback == nullptr || protection_callback_ == nullptr); + protection_callback_ = protection_callback; + return VCM_OK; +} + +// Enable or disable a video protection method. +void VideoSender::SetVideoProtection(VCMVideoProtection videoProtection) { + rtc::CritScope lock(&send_crit_); + switch (videoProtection) { + case kProtectionNone: + _mediaOpt.SetProtectionMethod(media_optimization::kNone); + break; + case kProtectionNack: + _mediaOpt.SetProtectionMethod(media_optimization::kNack); + break; + case kProtectionNackFEC: + _mediaOpt.SetProtectionMethod(media_optimization::kNackFec); + break; + case kProtectionFEC: + _mediaOpt.SetProtectionMethod(media_optimization::kFec); + break; + } +} +// Add one raw video frame to the encoder, blocking. +int32_t VideoSender::AddVideoFrame(const VideoFrame& videoFrame, + const VideoContentMetrics* contentMetrics, + const CodecSpecificInfo* codecSpecificInfo) { + EncoderParameters encoder_params; + { + rtc::CritScope lock(¶ms_lock_); + encoder_params = encoder_params_; + } + rtc::CritScope lock(&send_crit_); + if (_encoder == nullptr) + return VCM_UNINITIALIZED; + SetEncoderParameters(encoder_params); + // TODO(holmer): Add support for dropping frames per stream. Currently we + // only have one frame dropper for all streams. + if (_nextFrameTypes[0] == kEmptyFrame) { + return VCM_OK; + } + if (_mediaOpt.DropFrame()) { + _encoder->OnDroppedFrame(); + return VCM_OK; + } + _mediaOpt.UpdateContentData(contentMetrics); + // TODO(pbos): Make sure setting send codec is synchronized with video + // processing so frame size always matches. + if (!_codecDataBase.MatchesCurrentResolution(videoFrame.width(), + videoFrame.height())) { + LOG(LS_ERROR) << "Incoming frame doesn't match set resolution. Dropping."; + return VCM_PARAMETER_ERROR; + } + VideoFrame converted_frame = videoFrame; + if (converted_frame.native_handle() && !_encoder->SupportsNativeHandle()) { + // This module only supports software encoding. + // TODO(pbos): Offload conversion from the encoder thread. + converted_frame = converted_frame.ConvertNativeToI420Frame(); + RTC_CHECK(!converted_frame.IsZeroSize()) + << "Frame conversion failed, won't be able to encode frame."; + } + int32_t ret = + _encoder->Encode(converted_frame, codecSpecificInfo, _nextFrameTypes); + if (ret < 0) { + LOG(LS_ERROR) << "Failed to encode frame. Error code: " << ret; + return ret; + } + for (size_t i = 0; i < _nextFrameTypes.size(); ++i) { + _nextFrameTypes[i] = kVideoFrameDelta; // Default frame type. + } + if (qm_settings_callback_) + qm_settings_callback_->SetTargetFramerate(_encoder->GetTargetFramerate()); + return VCM_OK; +} + +int32_t VideoSender::IntraFrameRequest(int stream_index) { + rtc::CritScope lock(&send_crit_); + if (stream_index < 0 || + static_cast<unsigned int>(stream_index) >= _nextFrameTypes.size()) { + return -1; + } + _nextFrameTypes[stream_index] = kVideoFrameKey; + if (_encoder != nullptr && _encoder->InternalSource()) { + // Try to request the frame if we have an external encoder with + // internal source since AddVideoFrame never will be called. + if (_encoder->RequestFrame(_nextFrameTypes) == WEBRTC_VIDEO_CODEC_OK) { + _nextFrameTypes[stream_index] = kVideoFrameDelta; + } + } + return VCM_OK; +} + +int32_t VideoSender::EnableFrameDropper(bool enable) { + rtc::CritScope lock(&send_crit_); + frame_dropper_enabled_ = enable; + _mediaOpt.EnableFrameDropper(enable); + return VCM_OK; +} + +void VideoSender::SuspendBelowMinBitrate() { + RTC_DCHECK(main_thread_.CalledOnValidThread()); + int threshold_bps; + if (current_codec_.numberOfSimulcastStreams == 0) { + threshold_bps = current_codec_.minBitrate * 1000; + } else { + threshold_bps = current_codec_.simulcastStream[0].minBitrate * 1000; + } + // Set the hysteresis window to be at 10% of the threshold, but at least + // 10 kbps. + int window_bps = std::max(threshold_bps / 10, 10000); + _mediaOpt.SuspendBelowMinBitrate(threshold_bps, window_bps); +} + +bool VideoSender::VideoSuspended() const { + return _mediaOpt.IsVideoSuspended(); +} +} // namespace vcm +} // namespace webrtc |