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..437c159d3bcadb09d45649d89493bdfdbfb5ceda 100644 |
--- a/webrtc/video/encoder_state_feedback.cc |
+++ b/webrtc/video/encoder_state_feedback.cc |
@@ -13,9 +13,12 @@ |
#include "webrtc/base/checks.h" |
#include "webrtc/video/vie_encoder.h" |
+static const int kMinKeyFrameRequestIntervalMs = 300; |
+ |
namespace webrtc { |
-EncoderStateFeedback::EncoderStateFeedback() : vie_encoder_(nullptr) {} |
+EncoderStateFeedback::EncoderStateFeedback(Clock* clock) |
+ : clock_(clock), vie_encoder_(nullptr) {} |
void EncoderStateFeedback::Init(const std::vector<uint32_t>& ssrcs, |
ViEEncoder* encoder) { |
@@ -23,6 +26,7 @@ void EncoderStateFeedback::Init(const std::vector<uint32_t>& ssrcs, |
rtc::CritScope lock(&crit_); |
ssrcs_ = ssrcs; |
vie_encoder_ = encoder; |
+ time_last_intra_request_ms_.resize(ssrcs.size(), 0); |
pbos-webrtc
2016/05/03 18:19:59
I think -1 is the usual no-timestamp timestamp.
perkj_webrtc
2016/05/04 08:31:03
Done.
|
} |
bool EncoderStateFeedback::HasSsrc(uint32_t ssrc) { |
@@ -33,13 +37,30 @@ 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(); |
+ if (time_last_intra_request_ms_[index] + kMinKeyFrameRequestIntervalMs > |
+ now_ms) { |
+ return; |
+ } |
+ time_last_intra_request_ms_[index] = now_ms; |
+ |
+ vie_encoder_->OnReceivedIntraFrameRequest(index); |
pbos-webrtc
2016/05/03 18:19:59
I think you can drop crit_ during this callback.
perkj_webrtc
2016/05/04 08:31:03
Done.
|
} |
void EncoderStateFeedback::OnReceivedSLI(uint32_t ssrc, uint8_t picture_id) { |
@@ -48,7 +69,7 @@ void EncoderStateFeedback::OnReceivedSLI(uint32_t ssrc, uint8_t picture_id) { |
return; |
RTC_DCHECK(vie_encoder_); |
- vie_encoder_->OnReceivedSLI(ssrc, picture_id); |
+ vie_encoder_->OnReceivedSLI(picture_id); |
} |
void EncoderStateFeedback::OnReceivedRPSI(uint32_t ssrc, uint64_t picture_id) { |
@@ -57,7 +78,7 @@ void EncoderStateFeedback::OnReceivedRPSI(uint32_t ssrc, uint64_t picture_id) { |
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 |