1// Copyright 2014 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_BUFFER
6#define MEDIA_CAST_FRAMER_FRAME_BUFFER
7
8#include <map>
9#include <vector>
10
11#include "media/cast/cast_config.h"
12#include "media/cast/net/rtp/rtp_receiver_defines.h"
13
14namespace media {
15namespace cast {
16
17typedef std::map<uint16, std::vector<uint8> > PacketMap;
18
19class FrameBuffer {
20 public:
21  FrameBuffer();
22  ~FrameBuffer();
23  bool InsertPacket(const uint8* payload_data,
24                    size_t payload_size,
25                    const RtpCastHeader& rtp_header);
26  bool Complete() const;
27
28  void GetMissingPackets(bool newest_frame, PacketIdSet* missing_packets) const;
29
30  // If a frame is complete, sets the frame IDs and RTP timestamp in |frame|,
31  // and also copies the data from all packets into the data field in |frame|.
32  // Returns true if the frame was complete; false if incomplete and |frame|
33  // remains unchanged.
34  bool AssembleEncodedFrame(EncodedFrame* frame) const;
35
36  bool is_key_frame() const { return is_key_frame_; }
37  uint32 last_referenced_frame_id() const { return last_referenced_frame_id_; }
38  uint32 frame_id() const { return frame_id_; }
39
40 private:
41  uint32 frame_id_;
42  uint16 max_packet_id_;
43  uint16 num_packets_received_;
44  uint16 max_seen_packet_id_;
45  uint16 new_playout_delay_ms_;
46  bool is_key_frame_;
47  size_t total_data_size_;
48  uint32 last_referenced_frame_id_;
49  uint32 rtp_timestamp_;
50  PacketMap packets_;
51
52  DISALLOW_COPY_AND_ASSIGN(FrameBuffer);
53};
54
55}  // namespace cast
56}  // namespace media
57
58#endif  // MEDIA_CAST_FRAMER_FRAME_BUFFER
59