From e28e42b685349add668d21c02f34cb8779894643 Mon Sep 17 00:00:00 2001
From: Rodeo <tdskywalker@gmail.com>
Date: Sun, 6 Apr 2014 17:06:32 +0000
Subject: QSV: reset the encoder at chapter points to force a keyframe

This replaces the mfxEncodeCtrl-based solution used up to this point,
as it's not guaranteed to work with some parameter combinations.


git-svn-id: svn://svn.handbrake.fr/HandBrake/trunk@6153 b64f7644-9d1e-0410-96f1-a4d463321fa5
---
 libhb/enc_qsv.c | 50 +++++++++++++++++++++++++++++++++++++-------------
 1 file changed, 37 insertions(+), 13 deletions(-)

(limited to 'libhb')

diff --git a/libhb/enc_qsv.c b/libhb/enc_qsv.c
index d71cca6c1..a545c8960 100644
--- a/libhb/enc_qsv.c
+++ b/libhb/enc_qsv.c
@@ -830,16 +830,6 @@ int encqsvInit(hb_work_object_t *w, hb_job_t *job)
         {
             gop_ref_dist *= 2;
         }
-        /*
-         * XXX: B-pyramid + forced keyframes will cause visual artifacts,
-         *      force-disable B-pyramid until we insert keyframes properly
-         */
-        if (pv->param.gop.b_pyramid && job->chapter_markers)
-        {
-            pv->param.gop.b_pyramid = 0;
-            hb_log("encqsvInit: chapter markers enabled, disabling B-pyramid "
-                   "to work around a bug in our keyframe insertion code");
-        }
         if ((pv->param.gop.b_pyramid) &&
             (pv->param.videoParam->mfx.GopPicSize == 0 ||
              pv->param.videoParam->mfx.GopPicSize > gop_ref_dist))
@@ -1813,12 +1803,46 @@ int encqsvWork(hb_work_object_t *w, hb_buffer_t **buf_in, hb_buffer_t **buf_out)
     }
     pv->frames_in++;
 
+    /*
+     * Chapters have to start with a keyframe, so request one here.
+     *
+     * Using an mfxEncodeCtrl structure to force key frame generation is not
+     * possible when using a lookahead and frame reordering, so instead do
+     * the following before encoding the frame attached to the chapter:
+     *
+     * - flush the encoder to encode and retrieve any buffered frames
+     *
+     * - do a hard reset (MFXVideoENCODE_Close, then Init) of
+     *   the encoder to make sure the next frame is a keyframe
+     *
+     * The hard reset ensures encoding resumes with a clean state, avoiding
+     * miscellaneous hard-to-disagnose issues that may occur when resuming
+     * an encode after flushing the encoder or using MFXVideoENCODE_Reset.
+     */
     if (in->s.new_chap > 0 && job->chapter_markers)
     {
-        save_chapter(pv, in);
+        mfxStatus sts;
 
-        /* Chapters have to start with a keyframe, so request an IDR */
-        ctrl = &pv->force_keyframe;
+        if (encode_loop(pv, NULL, NULL) < 0)
+        {
+            goto fail;
+        }
+
+        sts = MFXVideoENCODE_Close(qsv_ctx->mfx_session);
+        if (sts != MFX_ERR_NONE)
+        {
+            hb_error("encqsv: MFXVideoENCODE_Close failed (%d)", sts);
+            goto fail;
+        }
+
+        sts = MFXVideoENCODE_Init(qsv_ctx->mfx_session, pv->param.videoParam);
+        if (sts < MFX_ERR_NONE)
+        {
+            hb_error("encqsv: MFXVideoENCODE_Init failed (%d)", sts);
+            goto fail;
+        }
+
+        save_chapter(pv, in);
     }
 
     /*
-- 
cgit v1.2.3