retrieve flags from mxf_parse_mpeg2_frame using a param

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

1 files changed, 9 insertions, 8 deletions

diff --git a/libavformat/mxfenc.c b/libavformat/mxfenc.c
index f12a72b17c..639891f62c 100644
--- a/libavformat/mxfenc.c
+++ b/libavformat/mxfenc.c
@@ -1101,14 +1101,13 @@ static const UID *mxf_get_mpeg2_codec_ul(AVCodecContext *avctx)
     return NULL;
 }
 
-static int mxf_parse_mpeg2_frame(AVFormatContext *s, AVStream *st, AVPacket *pkt)
+static int mxf_parse_mpeg2_frame(AVFormatContext *s, AVStream *st, AVPacket *pkt, int *flags)
 {
-    MXFContext *mxf = s->priv_data;
     MXFStreamContext *sc = st->priv_data;
     uint32_t c = -1;
     int i;
 
-    mxf->index_entries[mxf->edit_units_count].flags = 0;
+    *flags = 0;
 
     for(i = 0; i < pkt->size - 4; i++) {
         c = (c<<8) + pkt->data[i];
@@ -1122,16 +1121,16 @@ static int mxf_parse_mpeg2_frame(AVFormatContext *s, AVStream *st, AVPacket *pkt
             }
         } else if (c == 0x1b8) { // gop
             if (i + 4 < pkt->size && pkt->data[i+4]>>6 & 0x01) // closed
-                mxf->index_entries[mxf->edit_units_count].flags |= 0x80; // random access
+                *flags |= 0x80; // random access
         } else if (c == 0x1b3) { // seq
-            mxf->index_entries[mxf->edit_units_count].flags |= 0x40;
+            *flags |= 0x40;
         } else if (c == 0x100) { // pic
             int pict_type = (pkt->data[i+2]>>3) & 0x07;
             if (pict_type == 2) { // P frame
-                mxf->index_entries[mxf->edit_units_count].flags |= 0x22;
+                *flags |= 0x22;
                 st->codec->gop_size = 1;
             } else if (pict_type == 3) { // B frame
-                mxf->index_entries[mxf->edit_units_count].flags |= 0x33;
+                *flags |= 0x33;
                 sc->temporal_reordering = -1;
             } else if (!pict_type) {
                 av_log(s, AV_LOG_ERROR, "error parsing mpeg2 frame\n");
@@ -1306,6 +1305,7 @@ static int mxf_write_packet(AVFormatContext *s, AVPacket *pkt)
     ByteIOContext *pb = s->pb;
     AVStream *st = s->streams[pkt->stream_index];
     MXFStreamContext *sc = st->priv_data;
+    int flags = 0;
 
     if (!(mxf->edit_units_count % MXF_INDEX_CLUSTER_SIZE)) {
         mxf->index_entries = av_realloc(mxf->index_entries,
@@ -1318,7 +1318,7 @@ static int mxf_write_packet(AVFormatContext *s, AVPacket *pkt)
     }
 
     if (st->codec->codec_id == CODEC_ID_MPEG2VIDEO) {
-        if (!mxf_parse_mpeg2_frame(s, st, pkt)) {
+        if (!mxf_parse_mpeg2_frame(s, st, pkt, &flags)) {
             av_log(s, AV_LOG_ERROR, "could not get mpeg2 profile and level\n");
             return -1;
         }
@@ -1332,6 +1332,7 @@ static int mxf_write_packet(AVFormatContext *s, AVPacket *pkt)
     if (st->index == 0) {
         mxf_write_klv_fill(s);
         mxf->index_entries[mxf->edit_units_count].offset = url_ftell(pb);
+        mxf->index_entries[mxf->edit_units_count].flags = flags;
         mxf_write_system_item(s);
 
         mxf->edit_units_count++;