| // 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/framer.h" |
| |
| #include "base/logging.h" |
| |
| namespace media { |
| namespace cast { |
| |
| typedef FrameList::const_iterator ConstFrameIterator; |
| |
| Framer::Framer(base::TickClock* clock, |
| RtpPayloadFeedback* incoming_payload_feedback, |
| uint32 ssrc, |
| bool decoder_faster_than_max_frame_rate, |
| int max_unacked_frames) |
| : decoder_faster_than_max_frame_rate_(decoder_faster_than_max_frame_rate), |
| cast_msg_builder_(new CastMessageBuilder(clock, incoming_payload_feedback, |
| &frame_id_map_, ssrc, decoder_faster_than_max_frame_rate, |
| max_unacked_frames)) { |
| DCHECK(incoming_payload_feedback) << "Invalid argument"; |
| } |
| |
| Framer::~Framer() {} |
| |
| bool Framer::InsertPacket(const uint8* payload_data, |
| size_t payload_size, |
| const RtpCastHeader& rtp_header) { |
| bool complete = false; |
| if (!frame_id_map_.InsertPacket(rtp_header, &complete)) return false; |
| |
| // Does this packet belong to a new frame? |
| FrameList::iterator it = frames_.find(rtp_header.frame_id); |
| if (it == frames_.end()) { |
| // New frame. |
| linked_ptr<FrameBuffer> frame_buffer(new FrameBuffer()); |
| frame_buffer->InsertPacket(payload_data, payload_size, rtp_header); |
| frames_.insert(std::make_pair(rtp_header.frame_id, frame_buffer)); |
| } else { |
| // Insert packet to existing frame buffer. |
| it->second->InsertPacket(payload_data, payload_size, rtp_header); |
| } |
| |
| if (complete) { |
| // ACK as soon as possible. |
| VLOG(1) << "Complete frame " << static_cast<int>(rtp_header.frame_id); |
| cast_msg_builder_->CompleteFrameReceived(rtp_header.frame_id, |
| rtp_header.is_key_frame); |
| } |
| return complete; |
| } |
| |
| // This does not release the frame. |
| bool Framer::GetEncodedAudioFrame(EncodedAudioFrame* audio_frame, |
| uint32* rtp_timestamp, |
| bool* next_frame) { |
| uint8 frame_id; |
| // Find frame id. |
| if (frame_id_map_.NextContinuousFrame(&frame_id)) { |
| // We have our next frame. |
| *next_frame = true; |
| } else { |
| if (!frame_id_map_.NextAudioFrameAllowingMissingFrames(&frame_id)) { |
| return false; |
| } |
| *next_frame = false; |
| } |
| |
| ConstFrameIterator it = frames_.find(frame_id); |
| DCHECK(it != frames_.end()); |
| if (it == frames_.end()) return false; |
| |
| return it->second->GetEncodedAudioFrame(audio_frame, rtp_timestamp); |
| } |
| |
| // This does not release the frame. |
| bool Framer::GetEncodedVideoFrame(EncodedVideoFrame* video_frame, |
| uint32* rtp_timestamp, |
| bool* next_frame) { |
| uint8 frame_id; |
| // Find frame id. |
| if (frame_id_map_.NextContinuousFrame(&frame_id)) { |
| // We have our next frame. |
| *next_frame = true; |
| } else { |
| // Check if we can skip frames when our decoder is too slow. |
| if (!decoder_faster_than_max_frame_rate_) return false; |
| |
| if (!frame_id_map_.NextVideoFrameAllowingSkippingFrames(&frame_id)) { |
| return false; |
| } |
| *next_frame = false; |
| } |
| |
| ConstFrameIterator it = frames_.find(frame_id); |
| DCHECK(it != frames_.end()); |
| if (it == frames_.end()) return false; |
| |
| return it->second->GetEncodedVideoFrame(video_frame, rtp_timestamp); |
| } |
| |
| void Framer::Reset() { |
| frame_id_map_.Clear(); |
| frames_.clear(); |
| cast_msg_builder_->Reset(); |
| } |
| |
| void Framer::ReleaseFrame(uint8 frame_id) { |
| frame_id_map_.RemoveOldFrames(frame_id); |
| frames_.erase(frame_id); |
| |
| // We have a frame - remove all frames with lower frame id. |
| bool skipped_old_frame = false; |
| FrameList::iterator it; |
| for (it = frames_.begin(); it != frames_.end(); ) { |
| if (IsOlderFrameId(it->first, frame_id)) { |
| frames_.erase(it++); |
| skipped_old_frame = true; |
| } else { |
| ++it; |
| } |
| } |
| if (skipped_old_frame) { |
| cast_msg_builder_->UpdateCastMessage(); |
| } |
| } |
| |
| bool Framer::TimeToSendNextCastMessage(base::TimeTicks* time_to_send) { |
| return cast_msg_builder_->TimeToSendNextCastMessage(time_to_send); |
| } |
| |
| void Framer::SendCastMessage() { |
| cast_msg_builder_->UpdateCastMessage(); |
| } |
| |
| } // namespace cast |
| } // namespace media |