src/media/cast/framer/framer.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "media/cast/framer/framer.h"
   6
   7 #include "base/logging.h"
   8
   9 namespace media {
  10 namespace cast {
  11
  12 typedef FrameList::const_iterator ConstFrameIterator;
  13
  14 Framer::Framer(base::TickClock* clock,
  15                RtpPayloadFeedback* incoming_payload_feedback,
  16                uint32 ssrc,
  17                bool decoder_faster_than_max_frame_rate,
  18                int max_unacked_frames)
  19     : decoder_faster_than_max_frame_rate_(decoder_faster_than_max_frame_rate),
  20       cast_msg_builder_(new CastMessageBuilder(clock, incoming_payload_feedback,
  21           &frame_id_map_, ssrc, decoder_faster_than_max_frame_rate,
  22           max_unacked_frames)) {
  23   DCHECK(incoming_payload_feedback) << "Invalid argument";
  24 }
  25
  26 Framer::~Framer() {}
  27
  28 bool Framer::InsertPacket(const uint8* payload_data,
  29                           size_t payload_size,
  30                           const RtpCastHeader& rtp_header) {
  31   bool complete = false;
  32   if (!frame_id_map_.InsertPacket(rtp_header, &complete)) return false;
  33
  34   // Does this packet belong to a new frame?
  35   FrameList::iterator it = frames_.find(rtp_header.frame_id);
  36   if (it == frames_.end()) {
  37     // New frame.
  38     linked_ptr<FrameBuffer> frame_buffer(new FrameBuffer());
  39     frame_buffer->InsertPacket(payload_data, payload_size, rtp_header);
  40     frames_.insert(std::make_pair(rtp_header.frame_id, frame_buffer));
  41   } else {
  42     // Insert packet to existing frame buffer.
  43     it->second->InsertPacket(payload_data, payload_size, rtp_header);
  44   }
  45
  46   if (complete) {
  47     // ACK as soon as possible.
  48     VLOG(1) << "Complete frame " << static_cast<int>(rtp_header.frame_id);
  49     cast_msg_builder_->CompleteFrameReceived(rtp_header.frame_id,
  50                                              rtp_header.is_key_frame);
  51   }
  52   return complete;
  53 }
  54
  55 // This does not release the frame.
  56 bool Framer::GetEncodedAudioFrame(EncodedAudioFrame* audio_frame,
  57                                   uint32* rtp_timestamp,
  58                                   bool* next_frame) {
  59   uint8 frame_id;
  60   // Find frame id.
  61   if (frame_id_map_.NextContinuousFrame(&frame_id)) {
  62     // We have our next frame.
  63     *next_frame = true;
  64   } else {
  65     if (!frame_id_map_.NextAudioFrameAllowingMissingFrames(&frame_id)) {
  66       return false;
  67     }
  68     *next_frame = false;
  69   }
  70
  71   ConstFrameIterator it = frames_.find(frame_id);
  72   DCHECK(it != frames_.end());
  73   if (it == frames_.end()) return false;
  74
  75   return it->second->GetEncodedAudioFrame(audio_frame, rtp_timestamp);
  76 }
  77
  78 // This does not release the frame.
  79 bool Framer::GetEncodedVideoFrame(EncodedVideoFrame* video_frame,
  80                                   uint32* rtp_timestamp,
  81                                   bool* next_frame) {
  82   uint8 frame_id;
  83   // Find frame id.
  84   if (frame_id_map_.NextContinuousFrame(&frame_id)) {
  85     // We have our next frame.
  86     *next_frame = true;
  87   } else {
  88     // Check if we can skip frames when our decoder is too slow.
  89     if (!decoder_faster_than_max_frame_rate_) return false;
  90
  91     if (!frame_id_map_.NextVideoFrameAllowingSkippingFrames(&frame_id)) {
  92       return false;
  93     }
  94     *next_frame = false;
  95   }
  96
  97   ConstFrameIterator it = frames_.find(frame_id);
  98   DCHECK(it != frames_.end());
  99   if (it == frames_.end())  return false;
 100
 101   return it->second->GetEncodedVideoFrame(video_frame, rtp_timestamp);
 102 }
 103
 104 void Framer::Reset() {
 105   frame_id_map_.Clear();
 106   frames_.clear();
 107   cast_msg_builder_->Reset();
 108 }
 109
 110 void Framer::ReleaseFrame(uint8 frame_id) {
 111   frame_id_map_.RemoveOldFrames(frame_id);
 112   frames_.erase(frame_id);
 113
 114   // We have a frame - remove all frames with lower frame id.
 115   bool skipped_old_frame = false;
 116   FrameList::iterator it;
 117   for (it = frames_.begin(); it != frames_.end(); ) {
 118     if (IsOlderFrameId(it->first, frame_id)) {
 119       frames_.erase(it++);
 120       skipped_old_frame = true;
 121     } else {
 122       ++it;
 123     }
 124   }
 125   if (skipped_old_frame) {
 126     cast_msg_builder_->UpdateCastMessage();
 127   }
 128 }
 129
 130 bool Framer::TimeToSendNextCastMessage(base::TimeTicks* time_to_send) {
 131   return cast_msg_builder_->TimeToSendNextCastMessage(time_to_send);
 132 }
 133
 134 void Framer::SendCastMessage() {
 135   cast_msg_builder_->UpdateCastMessage();
 136 }
 137
 138 }  // namespace cast
 139 }  // namespace media