Chromium Code Reviews| Index: media/cast/framer/cast_message_builder.cc |
| diff --git a/media/cast/framer/cast_message_builder.cc b/media/cast/framer/cast_message_builder.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..3c98fb39b2dad0d6c1e89e4572648da37c92efe3 |
| --- /dev/null |
| +++ b/media/cast/framer/cast_message_builder.cc |
| @@ -0,0 +1,192 @@ |
| +// Copyright 2013 The Chromium Authors. All rights reserved. |
| +// Use of this source code is governed by a BSD-style license that can be |
| +// found in the LICENSE file. |
| + |
| +#include "media/cast/framer/cast_message_builder.h" |
| + |
| +namespace media { |
| +namespace cast { |
| + |
| +static const uint16 kCompleteFrameLost = 0xffff; |
| + |
| +CastMessageBuilder::CastMessageBuilder( |
| + RtpPayloadFeedback* incoming_payload_feedback, |
| + FrameIdMap* frame_id_map, |
| + uint32 media_ssrc, |
| + bool decoder_faster_than_max_frame_rate, |
| + int max_unacked_frames) |
| + : cast_feedback_(incoming_payload_feedback), |
| + frame_id_map_(frame_id_map), |
| + media_ssrc_(media_ssrc), |
| + decoder_faster_than_max_frame_rate_(decoder_faster_than_max_frame_rate), |
| + max_unacked_frames_(max_unacked_frames), |
| + cast_msg_(media_ssrc), |
| + waiting_for_key_frame_(true), |
| + slowing_down_ack_(false), |
| + acked_last_frame_(true), |
| + last_acked_frame_id_(kStartFrameId), |
| + default_tick_clock_(new base::DefaultTickClock()), |
| + clock_(default_tick_clock_.get()) { |
| + cast_msg_.ack_frame_id_ = kStartFrameId; |
| +} |
| + |
| +CastMessageBuilder::~CastMessageBuilder() {} |
| + |
| +void CastMessageBuilder::CompleteFrameReceived(uint8 frame_id, |
| + bool is_key_frame) { |
| + if (last_update_time_.is_null()) { |
| + // Our first update. |
| + last_update_time_ = clock_->NowTicks(); |
| + } |
| + if (waiting_for_key_frame_) { |
| + if (!is_key_frame) { |
| + // Ignore that we have received this complete frame since we are |
| + // waiting on a key frame. |
| + return; |
| + } |
| + waiting_for_key_frame_ = false; |
| + cast_msg_.missing_frames_and_packets_.clear(); |
| + cast_msg_.ack_frame_id_ = frame_id; |
| + last_update_time_ = clock_->NowTicks(); |
| + // We might have other complete frames waiting after we receive the last |
| + // packet in the key-frame. |
| + UpdateAckMessage(); |
| + } else { |
| + if (!UpdateAckMessage()) return; |
|
Alpha Left Google
2013/08/28 00:27:31
nit: This looks strange, return can just move to n
pwestin
2013/08/28 16:40:44
Done.
|
| + |
| + BuildPacketList(); |
| + } |
| + // Send cast message. |
| + cast_feedback_->CastFeedback(cast_msg_); |
| +} |
| + |
| +bool CastMessageBuilder::UpdateAckMessage() { |
| + if (!decoder_faster_than_max_frame_rate_) { |
| + int complete_frame_count = frame_id_map_->NumberOfCompleteFrames(); |
| + if (complete_frame_count > max_unacked_frames_) { |
| + // We have too many frames pending in our framer; slow down ACK. |
| + slowing_down_ack_ = true; |
| + } else if (complete_frame_count <= 1) { |
| + // We are down to one or less frames in our framer; ACK normally. |
| + slowing_down_ack_ = false; |
| + } |
| + } |
| + if (slowing_down_ack_) { |
| + // We are slowing down acknowledgment by acknowledging every other frame. |
| + if (acked_last_frame_) { |
| + acked_last_frame_ = false; |
| + } else { |
| + acked_last_frame_ = true; |
| + last_acked_frame_id_++; |
| + // Note: frame skipping and slowdown ACK is not supported at the same |
| + // time; and it's not needed since we can skip frames to catch up. |
| + } |
| + } else { |
| + uint8 frame_id = frame_id_map_->LastContinuousFrame(); |
| + |
| + // Is it a new frame? |
| + if (last_acked_frame_id_ == frame_id) return false; |
|
Alpha Left Google
2013/08/28 00:27:31
nit: return false can move to next line or one spa
pwestin
2013/08/28 16:40:44
Done.
|
| + |
| + last_acked_frame_id_ = frame_id; |
| + acked_last_frame_ = true; |
| + } |
| + cast_msg_.ack_frame_id_ = last_acked_frame_id_; |
| + cast_msg_.missing_frames_and_packets_.clear(); |
| + last_update_time_ = clock_->NowTicks(); |
| + return true; |
| +} |
| + |
| +bool CastMessageBuilder::TimeToSendNextCastMessage( |
| + base::TimeTicks* time_to_send) { |
| + // We haven't received any packets. |
| + if (last_update_time_.is_null() && frame_id_map_->Empty()) return false; |
|
Alpha Left Google
2013/08/28 00:27:31
nit: return can go to next line or 1 space.
pwestin
2013/08/28 16:40:44
Done.
|
| + |
| + *time_to_send = last_update_time_ + |
| + base::TimeDelta::FromMilliseconds(kCastMessageUpdateIntervalMs); |
| + return true; |
| +} |
| + |
| +void CastMessageBuilder::UpdateCastMessage() { |
| + RtcpCastMessage message(media_ssrc_); |
| + if (!UpdateCastMessageInternal(&message)) return; |
|
Alpha Left Google
2013/08/28 00:27:31
nit: return can go to next line or 1 space.
pwestin
2013/08/28 16:40:44
Done.
|
| + // Send cast message. |
| + cast_feedback_->CastFeedback(message); |
| +} |
| + |
| +void CastMessageBuilder::Reset() { |
| + waiting_for_key_frame_ = true; |
| + cast_msg_.ack_frame_id_ = kStartFrameId; |
| + cast_msg_.missing_frames_and_packets_.clear(); |
| + time_last_nacked_map_.clear(); |
| +} |
| + |
| +bool CastMessageBuilder::UpdateCastMessageInternal(RtcpCastMessage* message) { |
| + if (last_update_time_.is_null()) { |
| + if (!frame_id_map_->Empty()) { |
| + // We have received packets. |
| + last_update_time_ = clock_->NowTicks(); |
| + } |
| + return false; |
| + } |
| + // Is it time to update the cast message? |
| + base::TimeTicks now = clock_->NowTicks(); |
| + if (now - last_update_time_ < |
| + base::TimeDelta::FromMilliseconds(kCastMessageUpdateIntervalMs)) { |
| + return false; |
| + } |
| + last_update_time_ = now; |
| + |
| + UpdateAckMessage(); // Needed to cover when a frame is skipped. |
| + BuildPacketList(); |
| + *message = cast_msg_; |
| + return true; |
| +} |
| + |
| +void CastMessageBuilder::BuildPacketList() { |
| + base::TimeTicks now = clock_->NowTicks(); |
| + |
| + // Clear message NACK list. |
| + cast_msg_.missing_frames_and_packets_.clear(); |
| + |
| + // Are we missing packets? |
| + if (frame_id_map_->Empty()) return; |
|
Alpha Left Google
2013/08/28 00:27:31
nit: return can go to next line or 1 space.
pwestin
2013/08/28 16:40:44
Done.
|
| + |
| + uint8 newest_frame_id = frame_id_map_->NewestFrameId(); |
| + uint8 next_expected_frame_id = |
| + static_cast<uint8>(cast_msg_.ack_frame_id_ + 1); |
| + |
| + // Iterate over all frames. |
| + for (; !IsNewerFrameId(next_expected_frame_id, newest_frame_id); |
| + ++next_expected_frame_id) { |
| + std::map<uint8, base::TimeTicks>::iterator it = |
|
Alpha Left Google
2013/08/28 00:27:31
Use typdef'ed type here for readability.
pwestin
2013/08/28 16:40:44
Done.
|
| + time_last_nacked_map_.find(next_expected_frame_id); |
| + if (it != time_last_nacked_map_.end()) { |
| + // We have sent a NACK in this frame before, make sure enough time have |
| + // passed. |
| + if (now - it->second < |
| + base::TimeDelta::FromMilliseconds(kNackRepeatIntervalMs)) { |
| + continue; |
| + } |
| + } |
| + |
| + std::set<uint16> missing; |
|
Alpha Left Google
2013/08/28 00:27:31
This should use a typedef'ed type like PacketIdSet
pwestin
2013/08/28 16:40:44
Done.
|
| + if (frame_id_map_->FrameExists(next_expected_frame_id)) { |
| + bool last_frame = (newest_frame_id == next_expected_frame_id); |
| + frame_id_map_->GetMissingPackets(next_expected_frame_id, last_frame, |
| + &missing); |
| + if (!missing.empty()) { |
| + time_last_nacked_map_[next_expected_frame_id] = now; |
| + cast_msg_.missing_frames_and_packets_.insert( |
| + std::pair<uint8, std::set<uint16> >( |
|
Alpha Left Google
2013/08/28 00:27:31
Use std::make_pair() instead so types are inferred
pwestin
2013/08/28 16:40:44
Done.
|
| + next_expected_frame_id, missing)); |
| + } |
| + } else { |
| + time_last_nacked_map_[next_expected_frame_id] = now; |
| + missing.insert(kCompleteFrameLost); |
| + cast_msg_.missing_frames_and_packets_[next_expected_frame_id] = missing; |
| + } |
| + } |
| +} |
| + |
| +} // namespace cast |
| +} // namespace media |