media/cast/framer/frame_id_map.h

   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 #ifndef MEDIA_CAST_FRAMER_FRAME_ID_MAP_H_
   6 #define MEDIA_CAST_FRAMER_FRAME_ID_MAP_H_
   7
   8 #include <map>
   9 #include <set>
  10
  11 #include "base/memory/linked_ptr.h"
  12 #include "base/memory/scoped_ptr.h"
  13 #include "media/cast/cast_config.h"
  14 #include "media/cast/rtcp/rtcp_defines.h"
  15 #include "media/cast/rtp_receiver/rtp_receiver_defines.h"
  16
  17 namespace media {
  18 namespace cast {
  19
  20 class FrameInfo {
  21  public:
  22   FrameInfo(uint32 frame_id,
  23             uint32 referenced_frame_id,
  24             uint16 max_packet_id,
  25             bool key_frame);
  26   ~FrameInfo();
  27
  28   PacketType InsertPacket(uint16 packet_id);
  29   bool Complete() const;
  30   void GetMissingPackets(bool newest_frame,
  31                          PacketIdSet* missing_packets) const;
  32
  33   bool is_key_frame() const { return is_key_frame_; }
  34   uint32 frame_id() const { return frame_id_; }
  35   uint32 referenced_frame_id() const { return referenced_frame_id_; }
  36
  37  private:
  38   const bool is_key_frame_;
  39   const uint32 frame_id_;
  40   const uint32 referenced_frame_id_;
  41
  42   uint16 max_received_packet_id_;
  43   PacketIdSet missing_packets_;
  44
  45   DISALLOW_COPY_AND_ASSIGN(FrameInfo);
  46 };
  47
  48 typedef std::map<uint32, linked_ptr<FrameInfo> > FrameMap;
  49
  50 class FrameIdMap {
  51  public:
  52   FrameIdMap();
  53   ~FrameIdMap();
  54
  55   PacketType InsertPacket(const RtpCastHeader& rtp_header);
  56
  57   bool Empty() const;
  58   bool FrameExists(uint32 frame_id) const;
  59   uint32 NewestFrameId() const;
  60
  61   void RemoveOldFrames(uint32 frame_id);
  62   void Clear();
  63
  64   // Identifies the next frame to be released (rendered).
  65   bool NextContinuousFrame(uint32* frame_id) const;
  66   uint32 LastContinuousFrame() const;
  67
  68   bool NextAudioFrameAllowingMissingFrames(uint32* frame_id) const;
  69   bool NextVideoFrameAllowingSkippingFrames(uint32* frame_id) const;
  70
  71   int NumberOfCompleteFrames() const;
  72   void GetMissingPackets(uint32 frame_id,
  73                          bool last_frame,
  74                          PacketIdSet* missing_packets) const;
  75
  76  private:
  77   bool ContinuousFrame(FrameInfo* frame) const;
  78   bool DecodableVideoFrame(FrameInfo* frame) const;
  79
  80   FrameMap frame_map_;
  81   bool waiting_for_key_;
  82   uint32 last_released_frame_;
  83   uint32 newest_frame_id_;
  84
  85   DISALLOW_COPY_AND_ASSIGN(FrameIdMap);
  86 };
  87
  88 }  //  namespace cast
  89 }  //  namespace media
  90
  91 #endif  // MEDIA_CAST_FRAMER_FRAME_ID_MAP_H_