Index: webrtc/video/encoder_state_feedback.cc |
diff --git a/webrtc/video/encoder_state_feedback.cc b/webrtc/video/encoder_state_feedback.cc |
index 52f51114fa8b5710b231908e542e15602fe95556..4330b41c1c68d8c335ff0abf57176d0cdfb3d313 100644 |
--- a/webrtc/video/encoder_state_feedback.cc |
+++ b/webrtc/video/encoder_state_feedback.cc |
@@ -13,16 +13,18 @@ |
#include "webrtc/base/checks.h" |
#include "webrtc/video/vie_encoder.h" |
-namespace webrtc { |
+static const int kMinKeyFrameRequestIntervalMs = 300; |
-EncoderStateFeedback::EncoderStateFeedback() : vie_encoder_(nullptr) {} |
+namespace webrtc { |
-void EncoderStateFeedback::Init(const std::vector<uint32_t>& ssrcs, |
- ViEEncoder* encoder) { |
+EncoderStateFeedback::EncoderStateFeedback(Clock* clock, |
+ const std::vector<uint32_t>& ssrcs, |
+ ViEEncoder* encoder) |
+ : clock_(clock), |
+ ssrcs_(ssrcs), |
+ vie_encoder_(encoder), |
+ time_last_intra_request_ms_(ssrcs.size(), -1) { |
RTC_DCHECK(!ssrcs.empty()); |
- rtc::CritScope lock(&crit_); |
- ssrcs_ = ssrcs; |
- vie_encoder_ = encoder; |
} |
bool EncoderStateFeedback::HasSsrc(uint32_t ssrc) { |
@@ -33,41 +35,61 @@ bool EncoderStateFeedback::HasSsrc(uint32_t ssrc) { |
return false; |
} |
+size_t EncoderStateFeedback::GetStreamIndex(uint32_t ssrc) { |
+ for (size_t i = 0; i < ssrcs_.size(); ++i) { |
+ if (ssrcs_[i] == ssrc) |
+ return i; |
+ } |
+ RTC_NOTREACHED() << "Unknown ssrc " << ssrc; |
+ return 0; |
+} |
+ |
void EncoderStateFeedback::OnReceivedIntraFrameRequest(uint32_t ssrc) { |
- rtc::CritScope lock(&crit_); |
if (!HasSsrc(ssrc)) |
return; |
- RTC_DCHECK(vie_encoder_); |
- vie_encoder_->OnReceivedIntraFrameRequest(ssrc); |
+ size_t index = GetStreamIndex(ssrc); |
+ { |
+ int64_t now_ms = clock_->TimeInMilliseconds(); |
+ rtc::CritScope lock(&crit_); |
+ if (time_last_intra_request_ms_[index] + kMinKeyFrameRequestIntervalMs > |
+ now_ms) { |
+ return; |
+ } |
+ time_last_intra_request_ms_[index] = now_ms; |
+ } |
+ |
+ vie_encoder_->OnReceivedIntraFrameRequest(index); |
} |
void EncoderStateFeedback::OnReceivedSLI(uint32_t ssrc, uint8_t picture_id) { |
- rtc::CritScope lock(&crit_); |
if (!HasSsrc(ssrc)) |
return; |
RTC_DCHECK(vie_encoder_); |
pbos-webrtc
2016/05/04 14:10:36
Remove all of these DCHECKs, if you want them keep
perkj_webrtc
2016/05/04 15:25:38
Done.
|
- vie_encoder_->OnReceivedSLI(ssrc, picture_id); |
+ vie_encoder_->OnReceivedSLI(picture_id); |
} |
void EncoderStateFeedback::OnReceivedRPSI(uint32_t ssrc, uint64_t picture_id) { |
- rtc::CritScope lock(&crit_); |
if (!HasSsrc(ssrc)) |
return; |
RTC_DCHECK(vie_encoder_); |
- vie_encoder_->OnReceivedRPSI(ssrc, picture_id); |
+ vie_encoder_->OnReceivedRPSI(picture_id); |
} |
// Sending SSRCs for this encoder should never change since they are configured |
-// once and not reconfigured. |
+// once and not reconfigured, however, OnLocalSsrcChanged is called when the |
+// RtpModules are created with a different SSRC than what will be used in the |
+// end. |
+// TODO(perkj): Can we make sure the RTP module is created with the right SSRC |
+// from the beginning so this method is not triggered during creation ? |
void EncoderStateFeedback::OnLocalSsrcChanged(uint32_t old_ssrc, |
uint32_t new_ssrc) { |
if (!RTC_DCHECK_IS_ON) |
return; |
- rtc::CritScope lock(&crit_); |
- if (ssrcs_.empty()) // Encoder not yet attached (or detached for teardown). |
+ |
+ if (old_ssrc == 0) // old_ssrc == 0 during creation. |
return; |
// SSRC shouldn't change to something we haven't already registered with the |
// encoder. |