Home | History | Annotate | Download | only in framer
      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_id_map.h"
      6 
      7 #include "base/logging.h"
      8 #include "media/cast/rtp_receiver/rtp_receiver_defines.h"
      9 
     10 namespace media {
     11 namespace cast {
     12 
     13 FrameInfo::FrameInfo(uint32 frame_id,
     14                      uint32 referenced_frame_id,
     15                      uint16 max_packet_id,
     16                      bool key_frame)
     17     : is_key_frame_(key_frame),
     18       frame_id_(frame_id),
     19       referenced_frame_id_(referenced_frame_id),
     20       max_received_packet_id_(0) {
     21   if (max_packet_id > 0) {
     22     // Create the set with all packets missing.
     23     for (uint16 i = 0; i <= max_packet_id; i++) {
     24       missing_packets_.insert(i);
     25     }
     26   }
     27 }
     28 
     29 FrameInfo::~FrameInfo() {}
     30 
     31 bool FrameInfo::InsertPacket(uint16 packet_id) {
     32   // Update the last received packet id.
     33   if (IsNewerPacketId(packet_id, max_received_packet_id_)) {
     34     max_received_packet_id_ = packet_id;
     35   }
     36   missing_packets_.erase(packet_id);
     37   return missing_packets_.empty();
     38 }
     39 
     40 bool FrameInfo::Complete() const {
     41   return missing_packets_.empty();
     42 }
     43 
     44 void FrameInfo::GetMissingPackets(bool newest_frame,
     45                                   PacketIdSet* missing_packets) const {
     46   if (newest_frame) {
     47     // Missing packets capped by max_received_packet_id_.
     48     PacketIdSet::const_iterator it_after_last_received =
     49         missing_packets_.lower_bound(max_received_packet_id_);
     50     missing_packets->insert(missing_packets_.begin(), it_after_last_received);
     51   } else {
     52     missing_packets->insert(missing_packets_.begin(), missing_packets_.end());
     53   }
     54 }
     55 
     56 
     57 FrameIdMap::FrameIdMap()
     58     : waiting_for_key_(true),
     59       last_released_frame_(kStartFrameId),
     60       newest_frame_id_(kStartFrameId) {
     61 }
     62 
     63 FrameIdMap::~FrameIdMap() {}
     64 
     65 bool FrameIdMap::InsertPacket(const RtpCastHeader& rtp_header, bool* complete) {
     66   uint32 frame_id = rtp_header.frame_id;
     67   uint32 reference_frame_id;
     68   if (rtp_header.is_reference) {
     69     reference_frame_id = rtp_header.reference_frame_id;
     70   } else {
     71     reference_frame_id = static_cast<uint32>(frame_id - 1);
     72   }
     73 
     74   if (rtp_header.is_key_frame && waiting_for_key_) {
     75     last_released_frame_ = static_cast<uint32>(frame_id - 1);
     76     waiting_for_key_ = false;
     77   }
     78 
     79   VLOG(1) << "InsertPacket frame:" << frame_id
     80           << " packet:" << static_cast<int>(rtp_header.packet_id)
     81           << " max packet:" << static_cast<int>(rtp_header.max_packet_id);
     82 
     83   if (IsOlderFrameId(frame_id, last_released_frame_) && !waiting_for_key_) {
     84     return false;
     85   }
     86 
     87   // Update the last received frame id.
     88   if (IsNewerFrameId(frame_id, newest_frame_id_)) {
     89     newest_frame_id_ = frame_id;
     90   }
     91 
     92   // Does this packet belong to a new frame?
     93   FrameMap::iterator it = frame_map_.find(frame_id);
     94   if (it == frame_map_.end()) {
     95     // New frame.
     96     linked_ptr<FrameInfo> frame_info(new FrameInfo(frame_id,
     97                                                    reference_frame_id,
     98                                                    rtp_header.max_packet_id,
     99                                                    rtp_header.is_key_frame));
    100     std::pair<FrameMap::iterator, bool> retval =
    101         frame_map_.insert(std::make_pair(frame_id, frame_info));
    102 
    103     *complete = retval.first->second->InsertPacket(rtp_header.packet_id);
    104   } else {
    105     // Insert packet to existing frame.
    106     *complete = it->second->InsertPacket(rtp_header.packet_id);
    107   }
    108   return true;
    109 }
    110 
    111 void FrameIdMap::RemoveOldFrames(uint32 frame_id) {
    112   FrameMap::iterator it = frame_map_.begin();
    113 
    114   while (it != frame_map_.end()) {
    115     if (IsNewerFrameId(it->first, frame_id)) {
    116       ++it;
    117     } else {
    118       // Older or equal; erase.
    119       frame_map_.erase(it++);
    120     }
    121   }
    122   last_released_frame_ = frame_id;
    123 }
    124 
    125 void FrameIdMap::Clear() {
    126   frame_map_.clear();
    127   waiting_for_key_ = true;
    128   last_released_frame_ = kStartFrameId;
    129   newest_frame_id_ = kStartFrameId;
    130 }
    131 
    132 uint32 FrameIdMap::NewestFrameId() const {
    133   return newest_frame_id_;
    134 }
    135 
    136 bool FrameIdMap::NextContinuousFrame(uint32* frame_id) const {
    137   FrameMap::const_iterator it;
    138 
    139   for (it = frame_map_.begin(); it != frame_map_.end(); ++it) {
    140     if (it->second->Complete() && ContinuousFrame(it->second.get())) {
    141       *frame_id = it->first;
    142       return true;
    143     }
    144   }
    145   return false;
    146 }
    147 
    148 uint32 FrameIdMap::LastContinuousFrame() const {
    149   uint32 last_continuous_frame_id = last_released_frame_;
    150   uint32 next_expected_frame = last_released_frame_;
    151 
    152   FrameMap::const_iterator it;
    153 
    154   do {
    155     next_expected_frame++;
    156     it = frame_map_.find(next_expected_frame);
    157     if (it == frame_map_.end()) break;
    158     if (!it->second->Complete()) break;
    159 
    160     // We found the next continuous frame.
    161     last_continuous_frame_id = it->first;
    162   } while (next_expected_frame != newest_frame_id_);
    163   return last_continuous_frame_id;
    164 }
    165 
    166 bool FrameIdMap::NextAudioFrameAllowingMissingFrames(uint32* frame_id) const {
    167   // First check if we have continuous frames.
    168   if (NextContinuousFrame(frame_id)) return true;
    169 
    170   // Find the oldest frame.
    171   FrameMap::const_iterator it_best_match = frame_map_.end();
    172   FrameMap::const_iterator it;
    173 
    174   // Find first complete frame.
    175   for (it = frame_map_.begin(); it != frame_map_.end(); ++it) {
    176     if (it->second->Complete()) {
    177       it_best_match = it;
    178       break;
    179     }
    180   }
    181   if (it_best_match == frame_map_.end()) return false;  // No complete frame.
    182 
    183   ++it;
    184   for (; it != frame_map_.end(); ++it) {
    185     if (it->second->Complete() &&
    186         IsOlderFrameId(it->first, it_best_match->first)) {
    187       it_best_match = it;
    188     }
    189   }
    190   *frame_id = it_best_match->first;
    191   return true;
    192 }
    193 
    194 bool FrameIdMap::NextVideoFrameAllowingSkippingFrames(uint32* frame_id) const {
    195   // Find the oldest decodable frame.
    196   FrameMap::const_iterator it_best_match = frame_map_.end();
    197   FrameMap::const_iterator it;
    198   for (it = frame_map_.begin(); it != frame_map_.end(); ++it) {
    199     if (it->second->Complete() && DecodableVideoFrame(it->second.get())) {
    200       it_best_match = it;
    201     }
    202   }
    203   if (it_best_match == frame_map_.end()) return false;
    204 
    205   *frame_id = it_best_match->first;
    206   return true;
    207 }
    208 
    209 bool FrameIdMap::Empty() const {
    210   return frame_map_.empty();
    211 }
    212 
    213 int FrameIdMap::NumberOfCompleteFrames() const {
    214   int count = 0;
    215   FrameMap::const_iterator it;
    216   for (it = frame_map_.begin(); it != frame_map_.end(); ++it) {
    217     if (it->second->Complete()) {
    218       ++count;
    219     }
    220   }
    221   return count;
    222 }
    223 
    224 bool FrameIdMap::FrameExists(uint32 frame_id) const {
    225   return frame_map_.end() != frame_map_.find(frame_id);
    226 }
    227 
    228 void FrameIdMap::GetMissingPackets(uint32 frame_id,
    229                                    bool last_frame,
    230                                    PacketIdSet* missing_packets) const {
    231   FrameMap::const_iterator it = frame_map_.find(frame_id);
    232   if (it == frame_map_.end()) return;
    233 
    234   it->second->GetMissingPackets(last_frame, missing_packets);
    235 }
    236 
    237 bool FrameIdMap::ContinuousFrame(FrameInfo* frame) const {
    238   DCHECK(frame);
    239   if (waiting_for_key_ && !frame->is_key_frame()) return false;
    240   return static_cast<uint32>(last_released_frame_ + 1) == frame->frame_id();
    241 }
    242 
    243 bool FrameIdMap::DecodableVideoFrame(FrameInfo* frame) const {
    244   if (frame->is_key_frame()) return true;
    245   if (waiting_for_key_ && !frame->is_key_frame()) return false;
    246 
    247   // Current frame is not necessarily referencing the last frame.
    248   // Do we have the reference frame?
    249   if (IsOlderFrameId(frame->referenced_frame_id(), last_released_frame_)) {
    250     return true;
    251   }
    252   return frame->referenced_frame_id() == last_released_frame_;
    253 }
    254 
    255 }  //  namespace cast
    256 }  //  namespace media
    257