summaryrefslogtreecommitdiffstats
path: root/contrib/ffmpeg/A03-h264-recovery-point.patch
diff options
context:
space:
mode:
Diffstat (limited to 'contrib/ffmpeg/A03-h264-recovery-point.patch')
-rw-r--r--contrib/ffmpeg/A03-h264-recovery-point.patch63
1 files changed, 63 insertions, 0 deletions
diff --git a/contrib/ffmpeg/A03-h264-recovery-point.patch b/contrib/ffmpeg/A03-h264-recovery-point.patch
new file mode 100644
index 000000000..d018dd6b1
--- /dev/null
+++ b/contrib/ffmpeg/A03-h264-recovery-point.patch
@@ -0,0 +1,63 @@
+diff --git a/libavcodec/h264.c b/libavcodec/h264.c
+index 6ed251e..a3242ce 100644
+--- a/libavcodec/h264.c
++++ b/libavcodec/h264.c
+@@ -2627,6 +2627,7 @@ static void flush_dpb(AVCodecContext *avctx)
+ h->s.first_field = 0;
+ ff_h264_reset_sei(h);
+ ff_mpeg_flush(avctx);
++ h->recovery_frame= -1;
+ }
+
+ static int init_poc(H264Context *h)
+@@ -4289,9 +4290,18 @@ again:
+ if ((err = decode_slice_header(hx, h)))
+ break;
+
++ if (h->sei_recovery_frame_cnt >= 0 && h->recovery_frame < 0) {
++ h->recovery_frame = (h->frame_num + h->sei_recovery_frame_cnt) %
++ (1 << h->sps.log2_max_frame_num);
++ }
++
+ s->current_picture_ptr->f.key_frame |=
+- (hx->nal_unit_type == NAL_IDR_SLICE) ||
+- (h->sei_recovery_frame_cnt >= 0);
++ (hx->nal_unit_type == NAL_IDR_SLICE);
++
++ if (h->recovery_frame == h->frame_num) {
++ s->current_picture_ptr->f.key_frame |= 1;
++ h->recovery_frame = -1;
++ }
+
+ if (h->current_slice == 1) {
+ if (!(s->flags2 & CODEC_FLAG2_CHUNKS))
+diff --git a/libavcodec/h264.h b/libavcodec/h264.h
+index 570ce2f..454b755 100644
+--- a/libavcodec/h264.h
++++ b/libavcodec/h264.h
+@@ -569,6 +569,13 @@ typedef struct H264Context {
+ * frames.
+ */
+ int sei_recovery_frame_cnt;
++ /**
++ * recovery_frame is the frame_num at which the next frame should
++ * be fully constructed.
++ *
++ * Set to -1 when not expecting a recovery point.
++ */
++ int recovery_frame;
+
+ int luma_weight_flag[2]; ///< 7.4.3.2 luma_weight_lX_flag
+ int chroma_weight_flag[2]; ///< 7.4.3.2 chroma_weight_lX_flag
+diff --git a/libavcodec/h264_sei.c b/libavcodec/h264_sei.c
+index 2e5fb65..ea33a18 100644
+--- a/libavcodec/h264_sei.c
++++ b/libavcodec/h264_sei.c
+@@ -38,6 +38,7 @@ static const uint8_t sei_num_clock_ts_table[9]={
+ };
+
+ void ff_h264_reset_sei(H264Context *h) {
++ h->recovery_frame = -1;
+ h->sei_recovery_frame_cnt = -1;
+ h->sei_dpb_output_delay = 0;
+ h->sei_cpb_removal_delay = -1;