Chromium Code Reviews| Index: media/cast/net/rtcp/receiver_rtcp_event_subscriber.cc |
| diff --git a/media/cast/net/rtcp/receiver_rtcp_event_subscriber.cc b/media/cast/net/rtcp/receiver_rtcp_event_subscriber.cc |
| index a751ff94d37cd000258d5a82b4775371e4dec1a2..88b983a3c1b114fbdba3d3c7594cac85b05a8804 100644 |
| --- a/media/cast/net/rtcp/receiver_rtcp_event_subscriber.cc |
| +++ b/media/cast/net/rtcp/receiver_rtcp_event_subscriber.cc |
| @@ -11,9 +11,14 @@ namespace cast { |
| ReceiverRtcpEventSubscriber::ReceiverRtcpEventSubscriber( |
| const size_t max_size_to_retain, EventMediaType type) |
| - : max_size_to_retain_(max_size_to_retain), type_(type) { |
| + : max_size_to_retain_( |
| + max_size_to_retain * (kResendDelay * kNumResends + 1)), |
| + type_(type) { |
| DCHECK(max_size_to_retain_ > 0u); |
| DCHECK(type_ == AUDIO_EVENT || type_ == VIDEO_EVENT); |
| + for (size_t i = 0; i < kNumResends; i++) { |
|
miu
2014/12/04 04:18:53
nit: memset(send_ptrs_, 0, sizeof(send_ptrs_));
hubbe
2014/12/05 23:57:20
Not sure that is better.
The compiler should be mu
|
| + send_ptrs_[i] = 0; |
| + } |
| } |
| ReceiverRtcpEventSubscriber::~ReceiverRtcpEventSubscriber() { |
| @@ -33,7 +38,7 @@ void ReceiverRtcpEventSubscriber::OnReceiveFrameEvent( |
| case FRAME_DECODED: |
| rtcp_event.type = frame_event.type; |
| rtcp_event.timestamp = frame_event.timestamp; |
| - rtcp_events_.insert( |
| + rtcp_events_.push_back( |
| std::make_pair(frame_event.rtp_timestamp, rtcp_event)); |
| break; |
| default: |
| @@ -56,7 +61,7 @@ void ReceiverRtcpEventSubscriber::OnReceivePacketEvent( |
| rtcp_event.type = packet_event.type; |
| rtcp_event.timestamp = packet_event.timestamp; |
| rtcp_event.packet_id = packet_event.packet_id; |
| - rtcp_events_.insert( |
| + rtcp_events_.push_back( |
| std::make_pair(packet_event.rtp_timestamp, rtcp_event)); |
| } |
| } |
| @@ -66,12 +71,50 @@ void ReceiverRtcpEventSubscriber::OnReceivePacketEvent( |
| DCHECK(rtcp_events_.size() <= max_size_to_retain_); |
|
miu
2014/12/04 04:18:53
ditto
hubbe
2014/12/05 23:57:20
Done.
|
| } |
| -void ReceiverRtcpEventSubscriber::GetRtcpEventsAndReset( |
| - RtcpEventMultiMap* rtcp_events) { |
| +struct CompareByFirst { |
| + bool operator()(const std::pair<RtpTimestamp, RtcpEvent>& a, |
| + const std::pair<RtpTimestamp, RtcpEvent>& b) { |
| + return a.first < b.first; |
| + } |
| +}; |
| + |
| +void ReceiverRtcpEventSubscriber::GetRtcpEventsWithRedundancy( |
| + RtcpEvents* rtcp_events) { |
| DCHECK(thread_checker_.CalledOnValidThread()); |
| DCHECK(rtcp_events); |
| - rtcp_events->swap(rtcp_events_); |
| - rtcp_events_.clear(); |
| + |
| + uint64 event_level = rtcp_events_.size() + popped_events_; |
| + event_levels_for_past_frames_.push_back(event_level); |
| + |
| + for (size_t i = 0; i < kNumResends; i++) { |
| + size_t resend_delay = kResendDelay * i; |
| + if (event_levels_for_past_frames_.size() < resend_delay + 1) |
| + break; |
| + |
| + uint64 send_limit = event_levels_for_past_frames_[ |
| + event_levels_for_past_frames_.size() - 1 - resend_delay]; |
| + |
| + if (send_ptrs_[i] < popped_events_) { |
| + send_ptrs_[i] = popped_events_; |
| + } |
| + |
| + while (send_ptrs_[i] < send_limit && |
| + rtcp_events->size() < kMaxEventsPerRTCP) { |
| + rtcp_events->push_back(rtcp_events_[send_ptrs_[i] - popped_events_]); |
| + send_ptrs_[i]++; |
| + } |
| + send_limit = send_ptrs_[i]; |
| + } |
| + |
| + if (event_levels_for_past_frames_.size() > kResendDelay * (kNumResends + 1)) { |
| + while (popped_events_ < event_levels_for_past_frames_[0]) { |
| + rtcp_events_.pop_front(); |
| + popped_events_++; |
| + } |
| + event_levels_for_past_frames_.pop_front(); |
| + } |
| + |
| + std::sort(rtcp_events->begin(), rtcp_events->end(), CompareByFirst()); |
| } |
| void ReceiverRtcpEventSubscriber::TruncateMapIfNeeded() { |
| @@ -81,7 +124,8 @@ void ReceiverRtcpEventSubscriber::TruncateMapIfNeeded() { |
| DVLOG(3) << "RTCP event map exceeded size limit; " |
| << "removing oldest entry"; |
| // This is fine since we only insert elements one at a time. |
| - rtcp_events_.erase(rtcp_events_.begin()); |
| + rtcp_events_.pop_front(); |
| + popped_events_++; |
| } |
| } |