From 0ed260c7a3e57f9407602d4c78e23f7b8b675690 Mon Sep 17 00:00:00 2001 From: Ivan Schreter Date: Sat, 21 Feb 2009 19:59:25 +0000 Subject: Set context variable key_frame in H264 parser. Patch by Ivan Schreter, schreter gmx net Originally committed as revision 17489 to svn://svn.ffmpeg.org/ffmpeg/trunk --- libavcodec/h264_parser.c | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/libavcodec/h264_parser.c b/libavcodec/h264_parser.c index e00c15bb60..09f764aeae 100644 --- a/libavcodec/h264_parser.c +++ b/libavcodec/h264_parser.c @@ -118,8 +118,10 @@ static inline int parse_nal_units(AVCodecParserContext *s, /* set some sane default values */ s->pict_type = FF_I_TYPE; + s->key_frame = 0; h->s.avctx= avctx; + h->sei_recovery_frame_cnt = -1; for(;;) { int src_length, dst_length, consumed; @@ -152,10 +154,16 @@ static inline int parse_nal_units(AVCodecParserContext *s, ff_h264_decode_sei(h); break; case NAL_IDR_SLICE: + s->key_frame = 1; + /* fall through */ case NAL_SLICE: get_ue_golomb(&h->s.gb); // skip first_mb_in_slice slice_type = get_ue_golomb_31(&h->s.gb); s->pict_type = golomb_to_pict_type[slice_type % 5]; + if (h->sei_recovery_frame_cnt >= 0) { + /* key frame, since recovery_frame_cnt is set */ + s->key_frame = 1; + } return 0; /* no need to evaluate the rest */ } buf += consumed; -- cgit v1.2.3