From 65e053271a98f7acf3ef6b412998cfcb44a8eef8 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Martin=20Storsj=C3=B6?= Date: Mon, 10 Dec 2012 01:10:42 +0200 Subject: [PATCH] rtpenc_vp8: Include the picture number in VP8 packets MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit This makes it easier for receivers to decide what to do if data is lost. Refactor calculating the max payload size, to avoid hardcoding the header size in too many places, reducing the number of lines that have to be touched if the header is adjusted further. Signed-off-by: Martin Storsjö --- libavformat/rtpenc.h | 2 ++ libavformat/rtpenc_vp8.c | 20 +++++++++++++------- 2 files changed, 15 insertions(+), 7 deletions(-) diff --git a/libavformat/rtpenc.h b/libavformat/rtpenc.h index 369b413..5fd0b9c 100644 --- a/libavformat/rtpenc.h +++ b/libavformat/rtpenc.h @@ -59,6 +59,8 @@ struct RTPMuxContext { int nal_length_size; int flags; + + unsigned int frame_count; }; typedef struct RTPMuxContext RTPMuxContext; diff --git a/libavformat/rtpenc_vp8.c b/libavformat/rtpenc_vp8.c index 1730379..d24c8bc 100644 --- a/libavformat/rtpenc_vp8.c +++ b/libavformat/rtpenc_vp8.c @@ -26,24 +26,30 @@ void ff_rtp_send_vp8(AVFormatContext *s1, const uint8_t *buf, int size) { RTPMuxContext *s = s1->priv_data; - int len, max_packet_size; + int len, max_packet_size, header_size; s->buf_ptr = s->buf; s->timestamp = s->cur_timestamp; - max_packet_size = s->max_payload_size - 1; // minus one for header byte - // no extended control bits, reference frame, start of partition, + // extended control bit set, reference frame, start of partition, // partition id 0 - *s->buf_ptr++ = 0x10; + *s->buf_ptr++ = 0x90; + *s->buf_ptr++ = 0x80; // Picture id present + *s->buf_ptr++ = s->frame_count++ & 0x7f; + // Calculate the number of remaining bytes + header_size = s->buf_ptr - s->buf; + max_packet_size = s->max_payload_size - header_size; + while (size > 0) { len = FFMIN(size, max_packet_size); memcpy(s->buf_ptr, buf, len); - ff_rtp_send_data(s1, s->buf, len+1, size == len); // marker bit is last packet in frame + // marker bit is last packet in frame + ff_rtp_send_data(s1, s->buf, len + header_size, size == len); size -= len; buf += len; - s->buf_ptr = s->buf; - *s->buf_ptr++ = 0; // payload descriptor + // Clear the partition start bit, keep the rest of the header untouched + s->buf[0] &= ~0x10; } } -- 2.7.4