src/media/cast/framer/frame_buffer.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/frame_buffer.h"
   6
   7 #include "base/logging.h"
   8
   9 namespace media {
  10 namespace cast {
  11
  12 FrameBuffer::FrameBuffer()
  13     : frame_id_(0),
  14       max_packet_id_(0),
  15       num_packets_received_(0),
  16       is_key_frame_(false),
  17       total_data_size_(0),
  18       last_referenced_frame_id_(0),
  19       packets_() {}
  20
  21 FrameBuffer::~FrameBuffer() {}
  22
  23 void FrameBuffer::InsertPacket(const uint8* payload_data,
  24                                size_t payload_size,
  25                                const RtpCastHeader& rtp_header) {
  26   // Is this the first packet in the frame?
  27   if (packets_.empty()) {
  28     frame_id_ = rtp_header.frame_id;
  29     max_packet_id_ = rtp_header.max_packet_id;
  30     is_key_frame_ = rtp_header.is_key_frame;
  31     if (rtp_header.is_reference) {
  32       last_referenced_frame_id_ = rtp_header.reference_frame_id;
  33     } else {
  34       last_referenced_frame_id_ = rtp_header.frame_id - 1;
  35     }
  36
  37     rtp_timestamp_ = rtp_header.webrtc.header.timestamp;
  38   }
  39   // Is this the correct frame?
  40   if (rtp_header.frame_id != frame_id_) return;
  41
  42   // Insert every packet only once.
  43   if (packets_.find(rtp_header.packet_id) != packets_.end()) {
  44     return;
  45   }
  46
  47   std::vector<uint8> data;
  48   std::pair<PacketMap::iterator, bool> retval =
  49       packets_.insert(make_pair(rtp_header.packet_id, data));
  50
  51   // Insert the packet.
  52   retval.first->second.resize(payload_size);
  53   std::copy(payload_data, payload_data + payload_size,
  54             retval.first->second.begin());
  55
  56   ++num_packets_received_;
  57   total_data_size_ += payload_size;
  58 }
  59
  60 bool FrameBuffer::Complete() const {
  61   return num_packets_received_ - 1 == max_packet_id_;
  62 }
  63
  64 bool FrameBuffer::GetEncodedAudioFrame(
  65     transport::EncodedAudioFrame* audio_frame,
  66     uint32* rtp_timestamp) const {
  67   if (!Complete()) return false;
  68
  69   *rtp_timestamp = rtp_timestamp_;
  70
  71   // Frame is complete -> construct.
  72   audio_frame->frame_id = frame_id_;
  73
  74   // Build the data vector.
  75   audio_frame->data.clear();
  76   audio_frame->data.reserve(total_data_size_);
  77   PacketMap::const_iterator it;
  78   for (it = packets_.begin(); it != packets_.end(); ++it) {
  79     audio_frame->data.insert(audio_frame->data.end(),
  80                              it->second.begin(), it->second.end());
  81   }
  82   return true;
  83 }
  84
  85 bool FrameBuffer::GetEncodedVideoFrame(
  86     transport::EncodedVideoFrame* video_frame,
  87     uint32* rtp_timestamp) const {
  88   if (!Complete()) return false;
  89
  90   *rtp_timestamp = rtp_timestamp_;
  91
  92   // Frame is complete -> construct.
  93   video_frame->key_frame = is_key_frame_;
  94   video_frame->frame_id = frame_id_;
  95   video_frame->last_referenced_frame_id = last_referenced_frame_id_;
  96
  97   // Build the data vector.
  98   video_frame->data.clear();
  99   video_frame->data.reserve(total_data_size_);
 100   PacketMap::const_iterator it;
 101   for (it = packets_.begin(); it != packets_.end(); ++it) {
 102     video_frame->data.insert(video_frame->data.end(),
 103                              it->second.begin(), it->second.end());
 104   }
 105   return true;
 106 }
 107
 108 }  // namespace cast
 109 }  // namespace media