| Index: webrtc/modules/rtp_rtcp/source/rtcp_packet/rpsi.cc
|
| diff --git a/webrtc/modules/rtp_rtcp/source/rtcp_packet/rpsi.cc b/webrtc/modules/rtp_rtcp/source/rtcp_packet/rpsi.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..56766bff1babe62d5930f3c4f8663b3b5bd2c840
|
| --- /dev/null
|
| +++ b/webrtc/modules/rtp_rtcp/source/rtcp_packet/rpsi.cc
|
| @@ -0,0 +1,102 @@
|
| +/*
|
| + * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
|
| + *
|
| + * Use of this source code is governed by a BSD-style license
|
| + * that can be found in the LICENSE file in the root of the source
|
| + * tree. An additional intellectual property rights grant can be found
|
| + * in the file PATENTS. All contributing project authors may
|
| + * be found in the AUTHORS file in the root of the source tree.
|
| + */
|
| +
|
| +#include "webrtc/modules/rtp_rtcp/source/rtcp_packet/rpsi.h"
|
| +
|
| +#include "webrtc/modules/rtp_rtcp/source/byte_io.h"
|
| +
|
| +using webrtc::RTCPUtility::PT_PSFB;
|
| +using webrtc::RTCPUtility::RTCPPacketPSFBRPSI;
|
| +
|
| +namespace webrtc {
|
| +namespace rtcp {
|
| +namespace {
|
| +void AssignUWord8(uint8_t* buffer, size_t* offset, uint8_t value) {
|
| + buffer[(*offset)++] = value;
|
| +}
|
| +
|
| +void AssignUWord32(uint8_t* buffer, size_t* offset, uint32_t value) {
|
| + ByteWriter<uint32_t>::WriteBigEndian(buffer + *offset, value);
|
| + *offset += 4;
|
| +}
|
| +
|
| +// Reference picture selection indication (RPSI) (RFC 4585).
|
| +//
|
| +// FCI:
|
| +//
|
| +// 0 1 2 3
|
| +// 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
|
| +// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
| +// | PB |0| Payload Type| Native RPSI bit string |
|
| +// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
| +// | defined per codec ... | Padding (0) |
|
| +// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
| +void CreateRpsi(const RTCPPacketPSFBRPSI& rpsi,
|
| + uint8_t padding_bytes,
|
| + uint8_t* buffer,
|
| + size_t* pos) {
|
| + // Native bit string should be a multiple of 8 bits.
|
| + assert(rpsi.NumberOfValidBits % 8 == 0);
|
| + AssignUWord32(buffer, pos, rpsi.SenderSSRC);
|
| + AssignUWord32(buffer, pos, rpsi.MediaSSRC);
|
| + AssignUWord8(buffer, pos, padding_bytes * 8);
|
| + AssignUWord8(buffer, pos, rpsi.PayloadType);
|
| + memcpy(buffer + *pos, rpsi.NativeBitString, rpsi.NumberOfValidBits / 8);
|
| + *pos += rpsi.NumberOfValidBits / 8;
|
| + memset(buffer + *pos, 0, padding_bytes);
|
| + *pos += padding_bytes;
|
| +}
|
| +} // namespace
|
| +
|
| +bool Rpsi::Create(uint8_t* packet,
|
| + size_t* index,
|
| + size_t max_length,
|
| + RtcpPacket::PacketReadyCallback* callback) const {
|
| + assert(rpsi_.NumberOfValidBits > 0);
|
| + while (*index + BlockLength() > max_length) {
|
| + if (!OnBufferFull(packet, index, callback))
|
| + return false;
|
| + }
|
| + const uint8_t kFmt = 3;
|
| + CreateHeader(kFmt, PT_PSFB, HeaderLength(), packet, index);
|
| + CreateRpsi(rpsi_, padding_bytes_, packet, index);
|
| + return true;
|
| +}
|
| +
|
| +void Rpsi::WithPictureId(uint64_t picture_id) {
|
| + const uint32_t kPidBits = 7;
|
| + const uint64_t k7MsbZeroMask = 0x1ffffffffffffffULL;
|
| + uint8_t required_bytes = 0;
|
| + uint64_t shifted_pid = picture_id;
|
| + do {
|
| + ++required_bytes;
|
| + shifted_pid = (shifted_pid >> kPidBits) & k7MsbZeroMask;
|
| + } while (shifted_pid > 0);
|
| +
|
| + // Convert picture id to native bit string (natively defined by the video
|
| + // codec).
|
| + int pos = 0;
|
| + for (int i = required_bytes - 1; i > 0; i--) {
|
| + rpsi_.NativeBitString[pos++] =
|
| + 0x80 | static_cast<uint8_t>(picture_id >> (i * kPidBits));
|
| + }
|
| + rpsi_.NativeBitString[pos++] = static_cast<uint8_t>(picture_id & 0x7f);
|
| + rpsi_.NumberOfValidBits = pos * 8;
|
| +
|
| + // Calculate padding bytes (to reach next 32-bit boundary, 1, 2 or 3 bytes).
|
| + padding_bytes_ = 4 - ((2 + required_bytes) % 4);
|
| + if (padding_bytes_ == 4) {
|
| + padding_bytes_ = 0;
|
| + }
|
| +}
|
| +
|
| +
|
| +} // namespace rtcp
|
| +} // namespace webrtc
|
|
|