| /* |
| * Copyright (c) 2012 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 "video_coding_impl.h" |
| #include "common_types.h" |
| #include "encoded_frame.h" |
| #include "jitter_buffer.h" |
| #include "packet.h" |
| #include "trace.h" |
| #include "video_codec_interface.h" |
| #include "modules/video_coding/main/source/tick_time_base.h" |
| |
| namespace webrtc |
| { |
| |
| //#define DEBUG_DECODER_BIT_STREAM |
| |
| WebRtc_UWord32 |
| VCMProcessTimer::Period() const |
| { |
| return _periodMs; |
| } |
| |
| WebRtc_UWord32 |
| VCMProcessTimer::TimeUntilProcess() const |
| { |
| return static_cast<WebRtc_UWord32>( |
| VCM_MAX(static_cast<WebRtc_Word64>(_periodMs) - |
| (_clock->MillisecondTimestamp() - _latestMs), 0)); |
| } |
| |
| void |
| VCMProcessTimer::Processed() |
| { |
| _latestMs = _clock->MillisecondTimestamp(); |
| } |
| |
| VideoCodingModuleImpl::VideoCodingModuleImpl(const WebRtc_Word32 id, |
| TickTimeBase* clock, |
| bool delete_clock_on_destroy) |
| : |
| _id(id), |
| clock_(clock), |
| delete_clock_on_destroy_(delete_clock_on_destroy), |
| _receiveCritSect(CriticalSectionWrapper::CreateCriticalSection()), |
| _receiverInited(false), |
| _timing(clock_, id, 1), |
| _dualTiming(clock_, id, 2, &_timing), |
| _receiver(_timing, clock_, id, 1), |
| _dualReceiver(_dualTiming, clock_, id, 2, false), |
| _decodedFrameCallback(_timing, clock_), |
| _dualDecodedFrameCallback(_dualTiming, clock_), |
| _frameTypeCallback(NULL), |
| _frameStorageCallback(NULL), |
| _receiveStatsCallback(NULL), |
| _packetRequestCallback(NULL), |
| _decoder(NULL), |
| _dualDecoder(NULL), |
| #ifdef DEBUG_DECODER_BIT_STREAM |
| _bitStreamBeforeDecoder(NULL), |
| #endif |
| _frameFromFile(), |
| _keyRequestMode(kKeyOnError), |
| _scheduleKeyRequest(false), |
| |
| _sendCritSect(CriticalSectionWrapper::CreateCriticalSection()), |
| _encoder(), |
| _encodedFrameCallback(), |
| _nextFrameTypes(1, kVideoFrameDelta), |
| _mediaOpt(id, clock_), |
| _sendCodecType(kVideoCodecUnknown), |
| _sendStatsCallback(NULL), |
| _encoderInputFile(NULL), |
| _codecDataBase(id), |
| _receiveStatsTimer(1000, clock_), |
| _sendStatsTimer(1000, clock_), |
| _retransmissionTimer(10, clock_), |
| _keyRequestTimer(500, clock_) |
| { |
| assert(clock_); |
| #ifdef DEBUG_DECODER_BIT_STREAM |
| _bitStreamBeforeDecoder = fopen("decoderBitStream.bit", "wb"); |
| #endif |
| } |
| |
| VideoCodingModuleImpl::~VideoCodingModuleImpl() |
| { |
| if (_dualDecoder != NULL) |
| { |
| _codecDataBase.ReleaseDecoder(_dualDecoder); |
| } |
| delete _receiveCritSect; |
| delete _sendCritSect; |
| if (delete_clock_on_destroy_) delete clock_; |
| #ifdef DEBUG_DECODER_BIT_STREAM |
| fclose(_bitStreamBeforeDecoder); |
| #endif |
| if (_encoderInputFile != NULL) |
| { |
| fclose(_encoderInputFile); |
| } |
| } |
| |
| VideoCodingModule* |
| VideoCodingModule::Create(const WebRtc_Word32 id) |
| { |
| return new VideoCodingModuleImpl(id, new TickTimeBase(), true); |
| } |
| |
| VideoCodingModule* |
| VideoCodingModule::Create(const WebRtc_Word32 id, TickTimeBase* clock) |
| { |
| assert(clock); |
| return new VideoCodingModuleImpl(id, clock, false); |
| } |
| |
| void |
| VideoCodingModule::Destroy(VideoCodingModule* module) |
| { |
| if (module != NULL) |
| { |
| delete static_cast<VideoCodingModuleImpl*>(module); |
| } |
| } |
| |
| WebRtc_Word32 |
| VideoCodingModuleImpl::Process() |
| { |
| WebRtc_Word32 returnValue = VCM_OK; |
| |
| // Receive-side statistics |
| if (_receiveStatsTimer.TimeUntilProcess() == 0) |
| { |
| _receiveStatsTimer.Processed(); |
| if (_receiveStatsCallback != NULL) |
| { |
| WebRtc_UWord32 bitRate; |
| WebRtc_UWord32 frameRate; |
| const WebRtc_Word32 ret = _receiver.ReceiveStatistics(bitRate, |
| frameRate); |
| if (ret == 0) |
| { |
| _receiveStatsCallback->ReceiveStatistics(bitRate, frameRate); |
| } |
| else if (returnValue == VCM_OK) |
| { |
| returnValue = ret; |
| } |
| } |
| } |
| |
| // Send-side statistics |
| if (_sendStatsTimer.TimeUntilProcess() == 0) |
| { |
| _sendStatsTimer.Processed(); |
| if (_sendStatsCallback != NULL) |
| { |
| WebRtc_UWord32 bitRate; |
| WebRtc_UWord32 frameRate; |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| bitRate = static_cast<WebRtc_UWord32>( |
| _mediaOpt.SentBitRate() + 0.5f); |
| frameRate = static_cast<WebRtc_UWord32>( |
| _mediaOpt.SentFrameRate() + 0.5f); |
| } |
| _sendStatsCallback->SendStatistics(bitRate, frameRate); |
| } |
| } |
| |
| // Packet retransmission requests |
| if (_retransmissionTimer.TimeUntilProcess() == 0) |
| { |
| _retransmissionTimer.Processed(); |
| if (_packetRequestCallback != NULL) |
| { |
| WebRtc_UWord16 nackList[kNackHistoryLength]; |
| WebRtc_UWord16 length = kNackHistoryLength; |
| const WebRtc_Word32 ret = NackList(nackList, length); |
| if (ret != VCM_OK && returnValue == VCM_OK) |
| { |
| returnValue = ret; |
| } |
| if (length > 0) |
| { |
| _packetRequestCallback->ResendPackets(nackList, length); |
| } |
| } |
| } |
| |
| // Key frame requests |
| if (_keyRequestTimer.TimeUntilProcess() == 0) |
| { |
| _keyRequestTimer.Processed(); |
| if (_scheduleKeyRequest && _frameTypeCallback != NULL) |
| { |
| const WebRtc_Word32 ret = RequestKeyFrame(); |
| if (ret != VCM_OK && returnValue == VCM_OK) |
| { |
| returnValue = ret; |
| } |
| } |
| } |
| |
| return returnValue; |
| } |
| |
| WebRtc_Word32 |
| VideoCodingModuleImpl::Id() const |
| { |
| CriticalSectionScoped receiveCs(_receiveCritSect); |
| { |
| CriticalSectionScoped sendCs(_sendCritSect); |
| return _id; |
| } |
| } |
| |
| // Change the unique identifier of this object |
| WebRtc_Word32 |
| VideoCodingModuleImpl::ChangeUniqueId(const WebRtc_Word32 id) |
| { |
| CriticalSectionScoped receiveCs(_receiveCritSect); |
| { |
| CriticalSectionScoped sendCs(_sendCritSect); |
| _id = id; |
| return VCM_OK; |
| } |
| } |
| |
| // Returns the number of milliseconds until the module wants a worker thread to |
| // call Process |
| WebRtc_Word32 |
| VideoCodingModuleImpl::TimeUntilNextProcess() |
| { |
| WebRtc_UWord32 timeUntilNextProcess = VCM_MIN( |
| _receiveStatsTimer.TimeUntilProcess(), |
| _sendStatsTimer.TimeUntilProcess()); |
| if ((_receiver.NackMode() != kNoNack) || |
| (_dualReceiver.State() != kPassive)) |
| { |
| // We need a Process call more often if we are relying on |
| // retransmissions |
| timeUntilNextProcess = VCM_MIN(timeUntilNextProcess, |
| _retransmissionTimer.TimeUntilProcess()); |
| } |
| timeUntilNextProcess = VCM_MIN(timeUntilNextProcess, |
| _keyRequestTimer.TimeUntilProcess()); |
| |
| return timeUntilNextProcess; |
| } |
| |
| // Get number of supported codecs |
| WebRtc_UWord8 |
| VideoCodingModule::NumberOfCodecs() |
| { |
| return VCMCodecDataBase::NumberOfCodecs(); |
| } |
| |
| // Get supported codec with id |
| WebRtc_Word32 |
| VideoCodingModule::Codec(WebRtc_UWord8 listId, VideoCodec* codec) |
| { |
| if (codec == NULL) |
| { |
| return VCM_PARAMETER_ERROR; |
| } |
| return VCMCodecDataBase::Codec(listId, codec) ? 0 : -1; |
| } |
| |
| // Get supported codec with type |
| WebRtc_Word32 |
| VideoCodingModule::Codec(VideoCodecType codecType, VideoCodec* codec) |
| { |
| if (codec == NULL) |
| { |
| return VCM_PARAMETER_ERROR; |
| } |
| return VCMCodecDataBase::Codec(codecType, codec) ? 0 : -1; |
| } |
| |
| /* |
| * Sender |
| */ |
| |
| // Reset send side to initial state - all components |
| WebRtc_Word32 |
| VideoCodingModuleImpl::InitializeSender() |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| _codecDataBase.ResetSender(); |
| _encoder = NULL; |
| _encodedFrameCallback.SetTransportCallback(NULL); |
| // setting default bitRate and frameRate to 0 |
| _mediaOpt.SetEncodingData(kVideoCodecUnknown, 0, 0, 0, 0, 0, 0); |
| _mediaOpt.Reset(); // Resetting frame dropper |
| return VCM_OK; |
| } |
| |
| // Register the send codec to be used. |
| WebRtc_Word32 |
| VideoCodingModuleImpl::RegisterSendCodec(const VideoCodec* sendCodec, |
| WebRtc_UWord32 numberOfCores, |
| WebRtc_UWord32 maxPayloadSize) |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| if (sendCodec == NULL) |
| { |
| return VCM_PARAMETER_ERROR; |
| } |
| bool ret = _codecDataBase.RegisterSendCodec(sendCodec, numberOfCores, |
| maxPayloadSize); |
| if (!ret) |
| { |
| return -1; |
| } |
| |
| _encoder = _codecDataBase.GetEncoder(sendCodec, &_encodedFrameCallback); |
| if (_encoder == NULL) |
| { |
| WEBRTC_TRACE(webrtc::kTraceError, |
| webrtc::kTraceVideoCoding, |
| VCMId(_id), |
| "Failed to initialize encoder"); |
| return VCM_CODEC_ERROR; |
| } |
| _sendCodecType = sendCodec->codecType; |
| int numLayers = (_sendCodecType != kVideoCodecVP8) ? 1 : |
| sendCodec->codecSpecific.VP8.numberOfTemporalLayers; |
| _nextFrameTypes.clear(); |
| _nextFrameTypes.resize(VCM_MAX(sendCodec->numberOfSimulcastStreams, 1), |
| kVideoFrameDelta); |
| |
| _mediaOpt.SetEncodingData(_sendCodecType, |
| sendCodec->maxBitrate, |
| sendCodec->maxFramerate, |
| sendCodec->startBitrate, |
| sendCodec->width, |
| sendCodec->height, |
| numLayers); |
| _mediaOpt.SetMtu(maxPayloadSize); |
| |
| return VCM_OK; |
| } |
| |
| // Get current send codec |
| WebRtc_Word32 |
| VideoCodingModuleImpl::SendCodec(VideoCodec* currentSendCodec) const |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| |
| if (currentSendCodec == NULL) |
| { |
| return VCM_PARAMETER_ERROR; |
| } |
| return _codecDataBase.SendCodec(currentSendCodec) ? 0 : -1; |
| } |
| |
| // Get the current send codec type |
| VideoCodecType |
| VideoCodingModuleImpl::SendCodec() const |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| |
| return _codecDataBase.SendCodec(); |
| } |
| |
| // Register an external decoder object. |
| // This can not be used together with external decoder callbacks. |
| WebRtc_Word32 |
| VideoCodingModuleImpl::RegisterExternalEncoder(VideoEncoder* externalEncoder, |
| WebRtc_UWord8 payloadType, |
| bool internalSource /*= false*/) |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| |
| if (externalEncoder == NULL) |
| { |
| bool wasSendCodec = false; |
| const bool ret = _codecDataBase.DeregisterExternalEncoder( |
| payloadType, &wasSendCodec); |
| if (wasSendCodec) |
| { |
| // Make sure the VCM doesn't use the de-registered codec |
| _encoder = NULL; |
| } |
| return ret ? 0 : -1; |
| } |
| _codecDataBase.RegisterExternalEncoder(externalEncoder, payloadType, |
| internalSource); |
| return 0; |
| } |
| |
| // Get codec config parameters |
| WebRtc_Word32 |
| VideoCodingModuleImpl::CodecConfigParameters(WebRtc_UWord8* buffer, |
| WebRtc_Word32 size) |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| if (_encoder != NULL) |
| { |
| return _encoder->CodecConfigParameters(buffer, size); |
| } |
| return VCM_UNINITIALIZED; |
| } |
| |
| // Get encode bitrate |
| int VideoCodingModuleImpl::Bitrate(unsigned int* bitrate) const |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| // return the bit rate which the encoder is set to |
| if (!_encoder) { |
| return VCM_UNINITIALIZED; |
| } |
| *bitrate = _encoder->BitRate(); |
| return 0; |
| } |
| |
| // Get encode frame rate |
| int VideoCodingModuleImpl::FrameRate(unsigned int* framerate) const |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| // input frame rate, not compensated |
| if (!_encoder) { |
| return VCM_UNINITIALIZED; |
| } |
| *framerate = _encoder->FrameRate(); |
| return 0; |
| } |
| |
| // Set channel parameters |
| WebRtc_Word32 |
| VideoCodingModuleImpl::SetChannelParameters(WebRtc_UWord32 availableBandWidth, |
| WebRtc_UWord8 lossRate, |
| WebRtc_UWord32 rtt) |
| { |
| WebRtc_Word32 ret = 0; |
| { |
| CriticalSectionScoped sendCs(_sendCritSect); |
| WebRtc_UWord32 targetRate = _mediaOpt.SetTargetRates(availableBandWidth, |
| lossRate, |
| rtt); |
| if (_encoder != NULL) |
| { |
| ret = _encoder->SetChannelParameters(lossRate, rtt); |
| if (ret < 0 ) |
| { |
| return ret; |
| } |
| ret = (WebRtc_Word32)_encoder->SetRates(targetRate, |
| _mediaOpt.InputFrameRate()); |
| if (ret < 0) |
| { |
| return ret; |
| } |
| } |
| else |
| { |
| return VCM_UNINITIALIZED; |
| } // encoder |
| }// send side |
| return VCM_OK; |
| } |
| |
| WebRtc_Word32 |
| VideoCodingModuleImpl::SetReceiveChannelParameters(WebRtc_UWord32 rtt) |
| { |
| CriticalSectionScoped receiveCs(_receiveCritSect); |
| _receiver.UpdateRtt(rtt); |
| return 0; |
| } |
| |
| // Register a transport callback which will be called to deliver the encoded |
| // buffers |
| WebRtc_Word32 |
| VideoCodingModuleImpl::RegisterTransportCallback( |
| VCMPacketizationCallback* transport) |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| _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. |
| WebRtc_Word32 |
| VideoCodingModuleImpl::RegisterSendStatisticsCallback( |
| VCMSendStatisticsCallback* sendStats) |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| _sendStatsCallback = sendStats; |
| return VCM_OK; |
| } |
| |
| // Register a video quality settings callback which will be called when frame |
| // rate/dimensions need to be updated for video quality optimization |
| WebRtc_Word32 |
| VideoCodingModuleImpl::RegisterVideoQMCallback( |
| VCMQMSettingsCallback* videoQMSettings) |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| return _mediaOpt.RegisterVideoQMCallback(videoQMSettings); |
| } |
| |
| |
| // Register a video protection callback which will be called to deliver the |
| // requested FEC rate and NACK status (on/off). |
| WebRtc_Word32 |
| VideoCodingModuleImpl::RegisterProtectionCallback( |
| VCMProtectionCallback* protection) |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| _mediaOpt.RegisterProtectionCallback(protection); |
| return VCM_OK; |
| } |
| |
| // Enable or disable a video protection method. |
| WebRtc_Word32 |
| VideoCodingModuleImpl::SetVideoProtection(VCMVideoProtection videoProtection, |
| bool enable) |
| { |
| |
| switch (videoProtection) |
| { |
| |
| case kProtectionNack: |
| { |
| // Both send-side and receive-side |
| SetVideoProtection(kProtectionNackSender, enable); |
| SetVideoProtection(kProtectionNackReceiver, enable); |
| break; |
| } |
| |
| case kProtectionNackSender: |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| _mediaOpt.EnableProtectionMethod(enable, kNack); |
| break; |
| } |
| |
| case kProtectionNackReceiver: |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| if (enable) |
| { |
| _receiver.SetNackMode(kNackInfinite); |
| } |
| else |
| { |
| _receiver.SetNackMode(kNoNack); |
| } |
| break; |
| } |
| |
| case kProtectionDualDecoder: |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| if (enable) |
| { |
| _receiver.SetNackMode(kNoNack); |
| _dualReceiver.SetNackMode(kNackInfinite); |
| } |
| else |
| { |
| _dualReceiver.SetNackMode(kNoNack); |
| } |
| break; |
| } |
| |
| case kProtectionKeyOnLoss: |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| if (enable) |
| { |
| _keyRequestMode = kKeyOnLoss; |
| } |
| else if (_keyRequestMode == kKeyOnLoss) |
| { |
| _keyRequestMode = kKeyOnError; // default mode |
| } |
| else |
| { |
| return VCM_PARAMETER_ERROR; |
| } |
| break; |
| } |
| |
| case kProtectionKeyOnKeyLoss: |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| if (enable) |
| { |
| _keyRequestMode = kKeyOnKeyLoss; |
| } |
| else if (_keyRequestMode == kKeyOnKeyLoss) |
| { |
| _keyRequestMode = kKeyOnError; // default mode |
| } |
| else |
| { |
| return VCM_PARAMETER_ERROR; |
| } |
| break; |
| } |
| |
| case kProtectionNackFEC: |
| { |
| { |
| // Receive side |
| CriticalSectionScoped cs(_receiveCritSect); |
| if (enable) |
| { |
| _receiver.SetNackMode(kNackHybrid); |
| } |
| else |
| { |
| _receiver.SetNackMode(kNoNack); |
| } |
| } |
| // Send Side |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| _mediaOpt.EnableProtectionMethod(enable, kNackFec); |
| } |
| break; |
| } |
| |
| case kProtectionFEC: |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| _mediaOpt.EnableProtectionMethod(enable, kFec); |
| break; |
| } |
| |
| case kProtectionPeriodicKeyFrames: |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| return _codecDataBase.SetPeriodicKeyFrames(enable) ? 0 : -1; |
| break; |
| } |
| } |
| return VCM_OK; |
| } |
| |
| // Add one raw video frame to the encoder, blocking. |
| WebRtc_Word32 |
| VideoCodingModuleImpl::AddVideoFrame(const VideoFrame& videoFrame, |
| const VideoContentMetrics* contentMetrics, |
| const CodecSpecificInfo* codecSpecificInfo) |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| |
| if (_encoder == NULL) |
| { |
| return VCM_UNINITIALIZED; |
| } |
| // TODO(holmer): Add support for dropping frames per stream. Currently we |
| // only have one frame dropper for all streams. |
| if (_nextFrameTypes[0] == kFrameEmpty) |
| { |
| return VCM_OK; |
| } |
| _mediaOpt.UpdateIncomingFrameRate(); |
| |
| if (_mediaOpt.DropFrame()) |
| { |
| WEBRTC_TRACE(webrtc::kTraceStream, |
| webrtc::kTraceVideoCoding, |
| VCMId(_id), |
| "Drop frame due to bitrate"); |
| } |
| else |
| { |
| _mediaOpt.updateContentData(contentMetrics); |
| WebRtc_Word32 ret = _encoder->Encode(videoFrame, |
| codecSpecificInfo, |
| &_nextFrameTypes); |
| if (_encoderInputFile != NULL) |
| { |
| if (fwrite(videoFrame.Buffer(), 1, videoFrame.Length(), |
| _encoderInputFile) != videoFrame.Length()) { |
| return -1; |
| } |
| } |
| if (ret < 0) |
| { |
| WEBRTC_TRACE(webrtc::kTraceError, |
| webrtc::kTraceVideoCoding, |
| VCMId(_id), |
| "Encode error: %d", ret); |
| return ret; |
| } |
| for (size_t i = 0; i < _nextFrameTypes.size(); ++i) { |
| _nextFrameTypes[i] = kVideoFrameDelta; // Default frame type. |
| } |
| } |
| return VCM_OK; |
| } |
| |
| WebRtc_Word32 VideoCodingModuleImpl::IntraFrameRequest(int stream_index) { |
| assert(stream_index >= 0); |
| CriticalSectionScoped cs(_sendCritSect); |
| _nextFrameTypes[stream_index] = kVideoFrameKey; |
| if (_encoder != NULL && _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; |
| } |
| |
| WebRtc_Word32 |
| VideoCodingModuleImpl::EnableFrameDropper(bool enable) |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| _mediaOpt.EnableFrameDropper(enable); |
| return VCM_OK; |
| } |
| |
| |
| WebRtc_Word32 |
| VideoCodingModuleImpl::SentFrameCount(VCMFrameCount &frameCount) const |
| { |
| CriticalSectionScoped cs(_sendCritSect); |
| return _mediaOpt.SentFrameCount(frameCount); |
| } |
| |
| // Initialize receiver, resets codec database etc |
| WebRtc_Word32 |
| VideoCodingModuleImpl::InitializeReceiver() |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| WebRtc_Word32 ret = _receiver.Initialize(); |
| if (ret < 0) |
| { |
| return ret; |
| } |
| |
| ret = _dualReceiver.Initialize(); |
| if (ret < 0) |
| { |
| return ret; |
| } |
| _codecDataBase.ResetReceiver(); |
| _timing.Reset(); |
| |
| _decoder = NULL; |
| _decodedFrameCallback.SetUserReceiveCallback(NULL); |
| _receiverInited = true; |
| _frameTypeCallback = NULL; |
| _frameStorageCallback = NULL; |
| _receiveStatsCallback = NULL; |
| _packetRequestCallback = NULL; |
| _keyRequestMode = kKeyOnError; |
| _scheduleKeyRequest = false; |
| |
| return VCM_OK; |
| } |
| |
| // Register a receive callback. Will be called whenever there is a new frame |
| // ready for rendering. |
| WebRtc_Word32 |
| VideoCodingModuleImpl::RegisterReceiveCallback( |
| VCMReceiveCallback* receiveCallback) |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| _decodedFrameCallback.SetUserReceiveCallback(receiveCallback); |
| return VCM_OK; |
| } |
| |
| WebRtc_Word32 |
| VideoCodingModuleImpl::RegisterReceiveStatisticsCallback( |
| VCMReceiveStatisticsCallback* receiveStats) |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| _receiveStatsCallback = receiveStats; |
| return VCM_OK; |
| } |
| |
| // Register an externally defined decoder/render object. |
| // Can be a decoder only or a decoder coupled with a renderer. |
| WebRtc_Word32 |
| VideoCodingModuleImpl::RegisterExternalDecoder(VideoDecoder* externalDecoder, |
| WebRtc_UWord8 payloadType, |
| bool internalRenderTiming) |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| if (externalDecoder == NULL) |
| { |
| // Make sure the VCM updates the decoder next time it decodes. |
| _decoder = NULL; |
| return _codecDataBase.DeregisterExternalDecoder(payloadType) ? 0 : -1; |
| } |
| return _codecDataBase.RegisterExternalDecoder( |
| externalDecoder, payloadType, internalRenderTiming) ? 0 : -1; |
| } |
| |
| // Register a frame type request callback. |
| WebRtc_Word32 |
| VideoCodingModuleImpl::RegisterFrameTypeCallback( |
| VCMFrameTypeCallback* frameTypeCallback) |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| _frameTypeCallback = frameTypeCallback; |
| return VCM_OK; |
| } |
| |
| WebRtc_Word32 |
| VideoCodingModuleImpl::RegisterFrameStorageCallback( |
| VCMFrameStorageCallback* frameStorageCallback) |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| _frameStorageCallback = frameStorageCallback; |
| return VCM_OK; |
| } |
| |
| WebRtc_Word32 |
| VideoCodingModuleImpl::RegisterPacketRequestCallback( |
| VCMPacketRequestCallback* callback) |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| _packetRequestCallback = callback; |
| return VCM_OK; |
| } |
| |
| // Decode next frame, blocking. |
| // Should be called as often as possible to get the most out of the decoder. |
| WebRtc_Word32 |
| VideoCodingModuleImpl::Decode(WebRtc_UWord16 maxWaitTimeMs) |
| { |
| WebRtc_Word64 nextRenderTimeMs; |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| if (!_receiverInited) |
| { |
| return VCM_UNINITIALIZED; |
| } |
| if (!_codecDataBase.DecoderRegistered()) |
| { |
| return VCM_NO_CODEC_REGISTERED; |
| } |
| } |
| |
| const bool dualReceiverEnabledNotReceiving = |
| (_dualReceiver.State() != kReceiving && |
| _dualReceiver.NackMode() == kNackInfinite); |
| |
| VCMEncodedFrame* frame = _receiver.FrameForDecoding( |
| maxWaitTimeMs, |
| nextRenderTimeMs, |
| _codecDataBase.SupportsRenderScheduling(), |
| &_dualReceiver); |
| |
| if (dualReceiverEnabledNotReceiving && _dualReceiver.State() == kReceiving) |
| { |
| // Dual receiver is enabled (kNACK enabled), but was not receiving |
| // before the call to FrameForDecoding(). After the call the state |
| // changed to receiving, and therefore we must copy the primary decoder |
| // state to the dual decoder to make it possible for the dual decoder to |
| // start decoding retransmitted frames and recover. |
| CriticalSectionScoped cs(_receiveCritSect); |
| if (_dualDecoder != NULL) |
| { |
| _codecDataBase.ReleaseDecoder(_dualDecoder); |
| } |
| _dualDecoder = _codecDataBase.CreateDecoderCopy(); |
| if (_dualDecoder != NULL) |
| { |
| _dualDecoder->RegisterDecodeCompleteCallback( |
| &_dualDecodedFrameCallback); |
| } |
| else |
| { |
| _dualReceiver.Reset(); |
| } |
| } |
| |
| if (frame == NULL) |
| return VCM_FRAME_NOT_READY; |
| else |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| |
| // If this frame was too late, we should adjust the delay accordingly |
| _timing.UpdateCurrentDelay(frame->RenderTimeMs(), |
| clock_->MillisecondTimestamp()); |
| |
| #ifdef DEBUG_DECODER_BIT_STREAM |
| if (_bitStreamBeforeDecoder != NULL) |
| { |
| // Write bit stream to file for debugging purposes |
| if (fwrite(frame->Buffer(), 1, frame->Length(), |
| _bitStreamBeforeDecoder) != frame->Length()) { |
| return -1; |
| } |
| } |
| #endif |
| if (_frameStorageCallback != NULL) |
| { |
| WebRtc_Word32 ret = frame->Store(*_frameStorageCallback); |
| if (ret < 0) |
| { |
| return ret; |
| } |
| } |
| |
| const WebRtc_Word32 ret = Decode(*frame); |
| _receiver.ReleaseFrame(frame); |
| frame = NULL; |
| if (ret != VCM_OK) |
| { |
| return ret; |
| } |
| } |
| return VCM_OK; |
| } |
| |
| WebRtc_Word32 |
| VideoCodingModuleImpl::RequestSliceLossIndication( |
| const WebRtc_UWord64 pictureID) const |
| { |
| if (_frameTypeCallback != NULL) |
| { |
| const WebRtc_Word32 ret = |
| _frameTypeCallback->SliceLossIndicationRequest(pictureID); |
| if (ret < 0) |
| { |
| WEBRTC_TRACE(webrtc::kTraceError, |
| webrtc::kTraceVideoCoding, |
| VCMId(_id), |
| "Failed to request key frame"); |
| return ret; |
| } |
| } else |
| { |
| WEBRTC_TRACE(webrtc::kTraceWarning, |
| webrtc::kTraceVideoCoding, |
| VCMId(_id), |
| "No frame type request callback registered"); |
| return VCM_MISSING_CALLBACK; |
| } |
| return VCM_OK; |
| } |
| |
| WebRtc_Word32 |
| VideoCodingModuleImpl::RequestKeyFrame() |
| { |
| if (_frameTypeCallback != NULL) |
| { |
| const WebRtc_Word32 ret = _frameTypeCallback->RequestKeyFrame(); |
| if (ret < 0) |
| { |
| WEBRTC_TRACE(webrtc::kTraceError, |
| webrtc::kTraceVideoCoding, |
| VCMId(_id), |
| "Failed to request key frame"); |
| return ret; |
| } |
| _scheduleKeyRequest = false; |
| } |
| else |
| { |
| WEBRTC_TRACE(webrtc::kTraceWarning, |
| webrtc::kTraceVideoCoding, |
| VCMId(_id), |
| "No frame type request callback registered"); |
| return VCM_MISSING_CALLBACK; |
| } |
| return VCM_OK; |
| } |
| |
| WebRtc_Word32 |
| VideoCodingModuleImpl::DecodeDualFrame(WebRtc_UWord16 maxWaitTimeMs) |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| if (_dualReceiver.State() != kReceiving || |
| _dualReceiver.NackMode() != kNackInfinite) |
| { |
| // The dual receiver is currently not receiving or |
| // dual decoder mode is disabled. |
| return VCM_OK; |
| } |
| WebRtc_Word64 dummyRenderTime; |
| WebRtc_Word32 decodeCount = 0; |
| VCMEncodedFrame* dualFrame = _dualReceiver.FrameForDecoding( |
| maxWaitTimeMs, |
| dummyRenderTime); |
| if (dualFrame != NULL && _dualDecoder != NULL) |
| { |
| WEBRTC_TRACE(webrtc::kTraceStream, |
| webrtc::kTraceVideoCoding, |
| VCMId(_id), |
| "Decoding frame %u with dual decoder", |
| dualFrame->TimeStamp()); |
| // Decode dualFrame and try to catch up |
| WebRtc_Word32 ret = _dualDecoder->Decode(*dualFrame, |
| clock_->MillisecondTimestamp()); |
| if (ret != WEBRTC_VIDEO_CODEC_OK) |
| { |
| WEBRTC_TRACE(webrtc::kTraceWarning, |
| webrtc::kTraceVideoCoding, |
| VCMId(_id), |
| "Failed to decode frame with dual decoder"); |
| _dualReceiver.ReleaseFrame(dualFrame); |
| return VCM_CODEC_ERROR; |
| } |
| if (_receiver.DualDecoderCaughtUp(dualFrame, _dualReceiver)) |
| { |
| // Copy the complete decoder state of the dual decoder |
| // to the primary decoder. |
| WEBRTC_TRACE(webrtc::kTraceStream, |
| webrtc::kTraceVideoCoding, |
| VCMId(_id), |
| "Dual decoder caught up"); |
| _codecDataBase.CopyDecoder(*_dualDecoder); |
| _codecDataBase.ReleaseDecoder(_dualDecoder); |
| _dualDecoder = NULL; |
| } |
| decodeCount++; |
| } |
| _dualReceiver.ReleaseFrame(dualFrame); |
| return decodeCount; |
| } |
| |
| |
| // Must be called from inside the receive side critical section. |
| WebRtc_Word32 |
| VideoCodingModuleImpl::Decode(const VCMEncodedFrame& frame) |
| { |
| // Change decoder if payload type has changed |
| const bool renderTimingBefore = _codecDataBase.SupportsRenderScheduling(); |
| _decoder = _codecDataBase.GetDecoder(frame.PayloadType(), |
| &_decodedFrameCallback); |
| if (renderTimingBefore != _codecDataBase.SupportsRenderScheduling()) |
| { |
| // Make sure we reset the decode time estimate since it will |
| // be zero for codecs without render timing. |
| _timing.ResetDecodeTime(); |
| } |
| if (_decoder == NULL) |
| { |
| return VCM_NO_CODEC_REGISTERED; |
| } |
| // Decode a frame |
| WebRtc_Word32 ret = _decoder->Decode(frame, clock_->MillisecondTimestamp()); |
| |
| // Check for failed decoding, run frame type request callback if needed. |
| if (ret < 0) |
| { |
| if (ret == VCM_ERROR_REQUEST_SLI) |
| { |
| return RequestSliceLossIndication( |
| _decodedFrameCallback.LastReceivedPictureID() + 1); |
| } |
| else |
| { |
| WEBRTC_TRACE(webrtc::kTraceError, |
| webrtc::kTraceVideoCoding, |
| VCMId(_id), |
| "Failed to decode frame %u, requesting key frame", |
| frame.TimeStamp()); |
| ret = RequestKeyFrame(); |
| } |
| } |
| else if (ret == VCM_REQUEST_SLI) |
| { |
| ret = RequestSliceLossIndication( |
| _decodedFrameCallback.LastReceivedPictureID() + 1); |
| } |
| if (!frame.Complete() || frame.MissingFrame()) |
| { |
| switch (_keyRequestMode) |
| { |
| case kKeyOnKeyLoss: |
| { |
| if (frame.FrameType() == kVideoFrameKey) |
| { |
| _scheduleKeyRequest = true; |
| return VCM_OK; |
| } |
| break; |
| } |
| case kKeyOnLoss: |
| { |
| _scheduleKeyRequest = true; |
| return VCM_OK; |
| } |
| default: |
| break; |
| } |
| } |
| return ret; |
| } |
| |
| WebRtc_Word32 |
| VideoCodingModuleImpl::DecodeFromStorage( |
| const EncodedVideoData& frameFromStorage) |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| WebRtc_Word32 ret = _frameFromFile.ExtractFromStorage(frameFromStorage); |
| if (ret < 0) |
| { |
| return ret; |
| } |
| return Decode(_frameFromFile); |
| } |
| |
| // Reset the decoder state |
| WebRtc_Word32 |
| VideoCodingModuleImpl::ResetDecoder() |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| if (_decoder != NULL) |
| { |
| _receiver.Initialize(); |
| _timing.Reset(); |
| _scheduleKeyRequest = false; |
| _decoder->Reset(); |
| } |
| if (_dualReceiver.State() != kPassive) |
| { |
| _dualReceiver.Initialize(); |
| } |
| if (_dualDecoder != NULL) |
| { |
| _codecDataBase.ReleaseDecoder(_dualDecoder); |
| _dualDecoder = NULL; |
| } |
| return VCM_OK; |
| } |
| |
| // Register possible receive codecs, can be called multiple times |
| WebRtc_Word32 |
| VideoCodingModuleImpl::RegisterReceiveCodec(const VideoCodec* receiveCodec, |
| WebRtc_Word32 numberOfCores, |
| bool requireKeyFrame) |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| if (receiveCodec == NULL) |
| { |
| return VCM_PARAMETER_ERROR; |
| } |
| if (!_codecDataBase.RegisterReceiveCodec(receiveCodec, numberOfCores, |
| requireKeyFrame)) { |
| return -1; |
| } |
| return 0; |
| } |
| |
| // Get current received codec |
| WebRtc_Word32 |
| VideoCodingModuleImpl::ReceiveCodec(VideoCodec* currentReceiveCodec) const |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| if (currentReceiveCodec == NULL) |
| { |
| return VCM_PARAMETER_ERROR; |
| } |
| return _codecDataBase.ReceiveCodec(currentReceiveCodec) ? 0 : -1; |
| } |
| |
| // Get current received codec |
| VideoCodecType |
| VideoCodingModuleImpl::ReceiveCodec() const |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| return _codecDataBase.ReceiveCodec(); |
| } |
| |
| // Incoming packet from network parsed and ready for decode, non blocking. |
| WebRtc_Word32 |
| VideoCodingModuleImpl::IncomingPacket(const WebRtc_UWord8* incomingPayload, |
| WebRtc_UWord32 payloadLength, |
| const WebRtcRTPHeader& rtpInfo) |
| { |
| if (incomingPayload == NULL) { |
| // The jitter buffer doesn't handle non-zero payload lengths for packets |
| // without payload. |
| // TODO(holmer): We should fix this in the jitter buffer. |
| payloadLength = 0; |
| } |
| const VCMPacket packet(incomingPayload, payloadLength, rtpInfo); |
| WebRtc_Word32 ret; |
| if (_dualReceiver.State() != kPassive) |
| { |
| ret = _dualReceiver.InsertPacket(packet, |
| rtpInfo.type.Video.width, |
| rtpInfo.type.Video.height); |
| if (ret == VCM_FLUSH_INDICATOR) { |
| RequestKeyFrame(); |
| ResetDecoder(); |
| } else if (ret < 0) { |
| return ret; |
| } |
| } |
| ret = _receiver.InsertPacket(packet, rtpInfo.type.Video.width, |
| rtpInfo.type.Video.height); |
| if (ret == VCM_FLUSH_INDICATOR) { |
| RequestKeyFrame(); |
| ResetDecoder(); |
| } else if (ret < 0) { |
| return ret; |
| } |
| return VCM_OK; |
| } |
| |
| // Minimum playout delay (used for lip-sync). This is the minimum delay required |
| // to sync with audio. Not included in VideoCodingModule::Delay() |
| // Defaults to 0 ms. |
| WebRtc_Word32 |
| VideoCodingModuleImpl::SetMinimumPlayoutDelay(WebRtc_UWord32 minPlayoutDelayMs) |
| { |
| _timing.SetMinimumTotalDelay(minPlayoutDelayMs); |
| return VCM_OK; |
| } |
| |
| // The estimated delay caused by rendering, defaults to |
| // kDefaultRenderDelayMs = 10 ms |
| WebRtc_Word32 |
| VideoCodingModuleImpl::SetRenderDelay(WebRtc_UWord32 timeMS) |
| { |
| _timing.SetRenderDelay(timeMS); |
| return VCM_OK; |
| } |
| |
| // Current video delay |
| WebRtc_Word32 |
| VideoCodingModuleImpl::Delay() const |
| { |
| return _timing.TargetVideoDelay(); |
| } |
| |
| // Nack list |
| WebRtc_Word32 |
| VideoCodingModuleImpl::NackList(WebRtc_UWord16* nackList, WebRtc_UWord16& size) |
| { |
| VCMNackStatus nackStatus = kNackOk; |
| // Collect sequence numbers from the default receiver |
| // if in normal nack mode. Otherwise collect them from |
| // the dual receiver if the dual receiver is receiving. |
| if (_receiver.NackMode() != kNoNack) |
| { |
| nackStatus = _receiver.NackList(nackList, size); |
| } |
| else if (_dualReceiver.State() != kPassive) |
| { |
| nackStatus = _dualReceiver.NackList(nackList, size); |
| } |
| else |
| { |
| size = 0; |
| } |
| |
| switch (nackStatus) |
| { |
| case kNackNeedMoreMemory: |
| { |
| WEBRTC_TRACE(webrtc::kTraceError, |
| webrtc::kTraceVideoCoding, |
| VCMId(_id), |
| "Out of memory"); |
| return VCM_MEMORY; |
| } |
| case kNackKeyFrameRequest: |
| { |
| CriticalSectionScoped cs(_receiveCritSect); |
| WEBRTC_TRACE(webrtc::kTraceWarning, |
| webrtc::kTraceVideoCoding, |
| VCMId(_id), |
| "Failed to get NACK list, requesting key frame"); |
| return RequestKeyFrame(); |
| } |
| default: |
| break; |
| } |
| return VCM_OK; |
| } |
| |
| WebRtc_Word32 |
| VideoCodingModuleImpl::ReceivedFrameCount(VCMFrameCount& frameCount) const |
| { |
| return _receiver.ReceivedFrameCount(frameCount); |
| } |
| |
| WebRtc_UWord32 VideoCodingModuleImpl::DiscardedPackets() const { |
| return _receiver.DiscardedPackets(); |
| } |
| |
| int VideoCodingModuleImpl::SetSenderNackMode(SenderNackMode mode) { |
| CriticalSectionScoped cs(_sendCritSect); |
| |
| switch (mode) { |
| case kNackNone: |
| _mediaOpt.EnableProtectionMethod(false, kNack); |
| break; |
| case kNackAll: |
| _mediaOpt.EnableProtectionMethod(true, kNack); |
| break; |
| case kNackSelective: |
| return VCM_NOT_IMPLEMENTED; |
| break; |
| } |
| return VCM_OK; |
| } |
| |
| int VideoCodingModuleImpl::SetSenderReferenceSelection(bool enable) { |
| return VCM_NOT_IMPLEMENTED; |
| } |
| |
| int VideoCodingModuleImpl::SetSenderFEC(bool enable) { |
| CriticalSectionScoped cs(_sendCritSect); |
| _mediaOpt.EnableProtectionMethod(enable, kFec); |
| return VCM_OK; |
| } |
| |
| int VideoCodingModuleImpl::SetSenderKeyFramePeriod(int periodMs) { |
| return VCM_NOT_IMPLEMENTED; |
| } |
| |
| int VideoCodingModuleImpl::SetReceiverRobustnessMode( |
| ReceiverRobustness robustnessMode, |
| DecodeErrors errorMode) { |
| CriticalSectionScoped cs(_receiveCritSect); |
| switch (robustnessMode) { |
| case kNone: |
| _receiver.SetNackMode(kNoNack); |
| _dualReceiver.SetNackMode(kNoNack); |
| if (errorMode == kNoDecodeErrors) { |
| _keyRequestMode = kKeyOnLoss; |
| } else { |
| _keyRequestMode = kKeyOnError; |
| } |
| break; |
| case kHardNack: |
| if (errorMode == kAllowDecodeErrors) { |
| return VCM_PARAMETER_ERROR; |
| } |
| _receiver.SetNackMode(kNackInfinite); |
| _dualReceiver.SetNackMode(kNoNack); |
| _keyRequestMode = kKeyOnError; // TODO(hlundin): On long NACK list? |
| break; |
| case kSoftNack: |
| assert(false); // TODO(hlundin): Not completed. |
| return VCM_NOT_IMPLEMENTED; |
| _receiver.SetNackMode(kNackHybrid); |
| _dualReceiver.SetNackMode(kNoNack); |
| _keyRequestMode = kKeyOnError; |
| break; |
| case kDualDecoder: |
| if (errorMode == kNoDecodeErrors) { |
| return VCM_PARAMETER_ERROR; |
| } |
| _receiver.SetNackMode(kNoNack); |
| _dualReceiver.SetNackMode(kNackInfinite); |
| _keyRequestMode = kKeyOnError; |
| break; |
| case kReferenceSelection: |
| assert(false); // TODO(hlundin): Not completed. |
| return VCM_NOT_IMPLEMENTED; |
| if (errorMode == kNoDecodeErrors) { |
| return VCM_PARAMETER_ERROR; |
| } |
| _receiver.SetNackMode(kNoNack); |
| _dualReceiver.SetNackMode(kNoNack); |
| break; |
| } |
| return VCM_OK; |
| } |
| |
| int VideoCodingModuleImpl::StartDebugRecording(const char* file_name_utf8) { |
| CriticalSectionScoped cs(_sendCritSect); |
| _encoderInputFile = fopen(file_name_utf8, "wb"); |
| if (_encoderInputFile == NULL) |
| return VCM_GENERAL_ERROR; |
| return VCM_OK; |
| } |
| |
| int VideoCodingModuleImpl::StopDebugRecording(){ |
| CriticalSectionScoped cs(_sendCritSect); |
| if (_encoderInputFile != NULL) { |
| fclose(_encoderInputFile); |
| _encoderInputFile = NULL; |
| } |
| return VCM_OK; |
| } |
| |
| } // namespace webrtc |
