avformat/utils: factorize h264/hevc checks out in compute_pkt_fields()

Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
This commit is contained in:
Michael Niedermayer 2014-02-22 12:21:28 +01:00
parent 8bcacd9f42
commit f288e1b67c

View File

@ -1072,6 +1072,8 @@ static void compute_pkt_fields(AVFormatContext *s, AVStream *st,
int num, den, presentation_delayed, delay, i; int num, den, presentation_delayed, delay, i;
int64_t offset; int64_t offset;
AVRational duration; AVRational duration;
int onein_oneout = st->codec->codec_id != AV_CODEC_ID_H264 &&
st->codec->codec_id != AV_CODEC_ID_HEVC;
if (s->flags & AVFMT_FLAG_NOFILLIN) if (s->flags & AVFMT_FLAG_NOFILLIN)
return; return;
@ -1154,8 +1156,7 @@ static void compute_pkt_fields(AVFormatContext *s, AVStream *st,
/* Interpolate PTS and DTS if they are not present. We skip H264 /* Interpolate PTS and DTS if they are not present. We skip H264
* currently because delay and has_b_frames are not reliably set. */ * currently because delay and has_b_frames are not reliably set. */
if ((delay == 0 || (delay == 1 && pc)) && if ((delay == 0 || (delay == 1 && pc)) &&
st->codec->codec_id != AV_CODEC_ID_H264 && onein_oneout) {
st->codec->codec_id != AV_CODEC_ID_HEVC) {
if (presentation_delayed) { if (presentation_delayed) {
/* DTS = decompression timestamp */ /* DTS = decompression timestamp */
/* PTS = presentation timestamp */ /* PTS = presentation timestamp */
@ -1200,8 +1201,7 @@ static void compute_pkt_fields(AVFormatContext *s, AVStream *st,
pkt->dts = st->pts_buffer[0]; pkt->dts = st->pts_buffer[0];
} }
// We skipped it above so we try here. // We skipped it above so we try here.
if (st->codec->codec_id == AV_CODEC_ID_H264 || if (!onein_oneout)
st->codec->codec_id == AV_CODEC_ID_HEVC)
// This should happen on the first packet // This should happen on the first packet
update_initial_timestamps(s, pkt->stream_index, pkt->dts, pkt->pts, pkt); update_initial_timestamps(s, pkt->stream_index, pkt->dts, pkt->pts, pkt);
if (pkt->dts > st->cur_dts) if (pkt->dts > st->cur_dts)