Move Xiph's CodecPrivate writing code to its own function

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

1 files changed, 31 insertions, 26 deletions

diff --git a/libavformat/matroskaenc.c b/libavformat/matroskaenc.c
index c284c47fa4..acb527c66d 100644
--- a/libavformat/matroskaenc.c
+++ b/libavformat/matroskaenc.c
@@ -128,6 +128,36 @@ static void end_ebml_master(ByteIOContext *pb, offset_t start)
     url_fseek(pb, pos, SEEK_SET);
 }
 
+static int put_xiph_codecpriv(ByteIOContext *pb, AVCodecContext *codec)
+{
+    offset_t codecprivate;
+    uint8_t *header_start[3];
+    int header_len[3];
+    int first_header_size;
+    int j, k;
+
+    if (codec->codec_id == CODEC_ID_VORBIS)
+        first_header_size = 30;
+    else
+        first_header_size = 42;
+
+    if (ff_split_xiph_headers(codec->extradata, codec->extradata_size,
+                              first_header_size, header_start, header_len) < 0) {
+        av_log(codec, AV_LOG_ERROR, "Extradata corrupt.\n");
+        return -1;
+    }
+
+    codecprivate = start_ebml_master(pb, MATROSKA_ID_CODECPRIVATE);
+    put_byte(pb, 2);                    // number packets - 1
+    for (j = 0; j < 2; j++) {
+        for (k = 0; k < header_len[j] / 255; k++)
+            put_byte(pb, 255);
+        put_byte(pb, header_len[j] % 255);
+    }
+    for (j = 0; j < 3; j++)
+        put_buffer(pb, header_start[j], header_len[j]);
+    end_ebml_master(pb, codecprivate);
+}
 
 static int mkv_write_header(AVFormatContext *s)
 {
@@ -190,34 +220,9 @@ static int mkv_write_header(AVFormatContext *s)
 
         // XXX: CodecPrivate for vorbis, theora, aac, native mpeg4, ...
         if (native_id) {
-            offset_t codecprivate;
-
             if (codec->codec_id == CODEC_ID_VORBIS || codec->codec_id == CODEC_ID_THEORA) {
-                uint8_t *header_start[3];
-                int header_len[3];
-                int first_header_size;
-
-                if (codec->codec_id == CODEC_ID_VORBIS)
-                    first_header_size = 30;
-                else
-                    first_header_size = 42;
-
-                if (ff_split_xiph_headers(codec->extradata, codec->extradata_size,
-                                          first_header_size, header_start, header_len) < 0) {
-                    av_log(s, AV_LOG_ERROR, "Extradata corrupt.\n");
+                if (put_xiph_codecpriv(pb, codec) < 0)
                     return -1;
-                }
-
-                codecprivate = start_ebml_master(pb, MATROSKA_ID_CODECPRIVATE);
-                put_byte(pb, 2);                    // number packets - 1
-                for (j = 0; j < 2; j++) {
-                    for (k = 0; k < header_len[j] / 255; k++)
-                        put_byte(pb, 255);
-                    put_byte(pb, header_len[j] % 255);
-                }
-                for (j = 0; j < 3; j++)
-                    put_buffer(pb, header_start[j], header_len[j]);
-                end_ebml_master(pb, codecprivate);
             } else {
                 put_ebml_binary(pb, MATROSKA_ID_CODECPRIVATE, codec->extradata, codec->extradata_size);
             }