place SPS/PPS in extradata if GLOBAL_HEADER flag is set

Originally committed as revision 4872 to svn://svn.ffmpeg.org/ffmpeg/trunk

1 files changed, 18 insertions, 0 deletions

diff --git a/libavcodec/x264.c b/libavcodec/x264.c
index 68f1638569..31fe5cc65b 100644
--- a/libavcodec/x264.c
+++ b/libavcodec/x264.c
@@ -247,12 +247,30 @@ X264_init(AVCodecContext *avctx)
 
     x4->params.i_threads = avctx->thread_count;
 
+    if(avctx->flags & CODEC_FLAG_GLOBAL_HEADER){
+        x4->params.b_repeat_headers = 0;
+    }
+
     x4->enc = x264_encoder_open(&x4->params);
     if(!x4->enc)
         return -1;
 
     avctx->coded_frame = &x4->out_pic;
 
+    if(avctx->flags & CODEC_FLAG_GLOBAL_HEADER){
+	x264_nal_t *nal;
+	int nnal, i, s = 0;
+
+	x264_encoder_headers(x4->enc, &nal, &nnal);
+
+	/* 5 bytes NAL header + worst case escaping */
+	for(i = 0; i < nnal; i++)
+	    s += 5 + nal[i].i_payload * 4 / 3;
+
+        avctx->extradata = av_malloc(s);
+        avctx->extradata_size = encode_nals(avctx->extradata, s, nal, nnal);
+    }
+
     return 0;
 }