| Index: media/cast/framer/frame_buffer.cc
|
| diff --git a/media/cast/framer/frame_buffer.cc b/media/cast/framer/frame_buffer.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..ed7e11f0ce023c75f86b47559084c4bcf174ad0a
|
| --- /dev/null
|
| +++ b/media/cast/framer/frame_buffer.cc
|
| @@ -0,0 +1,103 @@
|
| +// Copyright 2013 The Chromium Authors. All rights reserved.
|
| +// Use of this source code is governed by a BSD-style license that can be
|
| +// found in the LICENSE file.
|
| +
|
| +#include "media/cast/framer/frame_buffer.h"
|
| +
|
| +namespace media {
|
| +namespace cast {
|
| +
|
| +FrameBuffer::FrameBuffer()
|
| + : frame_id_(0),
|
| + max_packet_id_(0),
|
| + num_packets_received_(0),
|
| + is_key_frame_(false),
|
| + total_data_size_(0),
|
| + last_referenced_frame_id_(0),
|
| + packets_() {}
|
| +
|
| +FrameBuffer::~FrameBuffer() {}
|
| +
|
| +void FrameBuffer::InsertPacket(const uint8* payload_data,
|
| + int payload_size,
|
| + const RtpCastHeader& rtp_header) {
|
| + // Is this the first packet in the frame?
|
| + if (packets_.empty()) {
|
| + frame_id_ = rtp_header.frame_id;
|
| + max_packet_id_ = rtp_header.max_packet_id;
|
| + is_key_frame_ = rtp_header.is_key_frame;
|
| + if (rtp_header.is_reference) {
|
| + last_referenced_frame_id_ = rtp_header.reference_frame_id;
|
| + } else {
|
| + last_referenced_frame_id_ = static_cast<uint8>(rtp_header.frame_id - 1);
|
| + }
|
| +
|
| + rtp_timestamp_ = rtp_header.webrtc.header.timestamp;
|
| + }
|
| + // Is this the correct frame?
|
| + if (rtp_header.frame_id != frame_id_) return;
|
| +
|
| + // Insert every packet only once.
|
| + if (packets_.find(rtp_header.packet_id) != packets_.end()) return;
|
| +
|
| + std::vector<uint8> data;
|
| + std::pair<PacketMap::iterator, bool> retval =
|
| + packets_.insert(make_pair(rtp_header.packet_id, data));
|
| +
|
| + // Insert the packet.
|
| + retval.first->second.resize(payload_size);
|
| + std::copy(payload_data, payload_data + payload_size,
|
| + retval.first->second.begin());
|
| +
|
| + ++num_packets_received_;
|
| + total_data_size_ += payload_size;
|
| +}
|
| +
|
| +bool FrameBuffer::Complete() const {
|
| + return num_packets_received_ - 1 == max_packet_id_;
|
| +}
|
| +
|
| +bool FrameBuffer::GetEncodedAudioFrame(EncodedAudioFrame* audio_frame,
|
| + uint32* rtp_timestamp) const {
|
| + if (!Complete()) return false;
|
| +
|
| + *rtp_timestamp = rtp_timestamp_;
|
| +
|
| + // Frame is complete -> construct.
|
| + audio_frame->frame_id = frame_id_;
|
| +
|
| + // Build the data vector.
|
| + audio_frame->data.clear();
|
| + audio_frame->data.reserve(total_data_size_);
|
| + PacketMap::const_iterator it;
|
| + for (it = packets_.begin(); it != packets_.end(); ++it) {
|
| + audio_frame->data.insert(audio_frame->data.end(),
|
| + it->second.begin(), it->second.end());
|
| + }
|
| + return true;
|
| +}
|
| +
|
| +bool FrameBuffer::GetEncodedVideoFrame(EncodedVideoFrame* video_frame,
|
| + uint32* rtp_timestamp) const {
|
| + if (!Complete()) return false;
|
| +
|
| + *rtp_timestamp = rtp_timestamp_;
|
| +
|
| + // Frame is complete -> construct.
|
| + video_frame->key_frame = is_key_frame_;
|
| + video_frame->frame_id = frame_id_;
|
| + video_frame->last_referenced_frame_id = last_referenced_frame_id_;
|
| +
|
| + // Build the data vector.
|
| + video_frame->data.clear();
|
| + video_frame->data.reserve(total_data_size_);
|
| + PacketMap::const_iterator it;
|
| + for (it = packets_.begin(); it != packets_.end(); ++it) {
|
| + video_frame->data.insert(video_frame->data.end(),
|
| + it->second.begin(), it->second.end());
|
| + }
|
| + return true;
|
| +}
|
| +
|
| +} // namespace cast
|
| +} // namespace media
|
|
|