retrieve flags from mxf_parse_mpeg2_frame using a param

Originally committed as revision 17123 to svn://svn.ffmpeg.org/ffmpeg/trunk
This commit is contained in:
Baptiste Coudurier 2009-02-10 08:31:48 +00:00
parent a784cc679f
commit 4a15f3ee99
1 changed files with 9 additions and 8 deletions

View File

@ -1101,14 +1101,13 @@ static const UID *mxf_get_mpeg2_codec_ul(AVCodecContext *avctx)
return NULL; 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; MXFStreamContext *sc = st->priv_data;
uint32_t c = -1; uint32_t c = -1;
int i; int i;
mxf->index_entries[mxf->edit_units_count].flags = 0; *flags = 0;
for(i = 0; i < pkt->size - 4; i++) { for(i = 0; i < pkt->size - 4; i++) {
c = (c<<8) + pkt->data[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 } else if (c == 0x1b8) { // gop
if (i + 4 < pkt->size && pkt->data[i+4]>>6 & 0x01) // closed 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 } else if (c == 0x1b3) { // seq
mxf->index_entries[mxf->edit_units_count].flags |= 0x40; *flags |= 0x40;
} else if (c == 0x100) { // pic } else if (c == 0x100) { // pic
int pict_type = (pkt->data[i+2]>>3) & 0x07; int pict_type = (pkt->data[i+2]>>3) & 0x07;
if (pict_type == 2) { // P frame if (pict_type == 2) { // P frame
mxf->index_entries[mxf->edit_units_count].flags |= 0x22; *flags |= 0x22;
st->codec->gop_size = 1; st->codec->gop_size = 1;
} else if (pict_type == 3) { // B frame } else if (pict_type == 3) { // B frame
mxf->index_entries[mxf->edit_units_count].flags |= 0x33; *flags |= 0x33;
sc->temporal_reordering = -1; sc->temporal_reordering = -1;
} else if (!pict_type) { } else if (!pict_type) {
av_log(s, AV_LOG_ERROR, "error parsing mpeg2 frame\n"); 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; ByteIOContext *pb = s->pb;
AVStream *st = s->streams[pkt->stream_index]; AVStream *st = s->streams[pkt->stream_index];
MXFStreamContext *sc = st->priv_data; MXFStreamContext *sc = st->priv_data;
int flags = 0;
if (!(mxf->edit_units_count % MXF_INDEX_CLUSTER_SIZE)) { if (!(mxf->edit_units_count % MXF_INDEX_CLUSTER_SIZE)) {
mxf->index_entries = av_realloc(mxf->index_entries, 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 (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"); av_log(s, AV_LOG_ERROR, "could not get mpeg2 profile and level\n");
return -1; return -1;
} }
@ -1332,6 +1332,7 @@ static int mxf_write_packet(AVFormatContext *s, AVPacket *pkt)
if (st->index == 0) { if (st->index == 0) {
mxf_write_klv_fill(s); mxf_write_klv_fill(s);
mxf->index_entries[mxf->edit_units_count].offset = url_ftell(pb); 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_write_system_item(s);
mxf->edit_units_count++; mxf->edit_units_count++;