commit 86e9dba8fa5be26e10a096b675616b9ddb950031
author Timo Rothenpieler <timo@rothenpieler.org> Wed Apr 11 12:22:20 2018
committer Timo Rothenpieler <timo@rothenpieler.org> Wed Apr 11 12:55:28 2018
tree 41a4d90553ffd00b7a47676e5ed8a69c474596f8
parent 3e9d676192fc6ef719a904f8a2d114efec03d7c1

avcodec/nvenc: add support for B frames as ref

libavcodec/nvenc.c

diff --git a/libavcodec/nvenc.c b/libavcodec/nvenc.c
index 1f601a6..e2d1100 100644
--- a/libavcodec/nvenc.c
+++ b/libavcodec/nvenc.c
@@ -370,6 +370,22 @@
         return AVERROR(ENOSYS);
     }
 
+#ifdef NVENC_HAVE_BFRAME_REF_MODE
+    ret = nvenc_check_cap(avctx, NV_ENC_CAPS_SUPPORT_BFRAME_REF_MODE);
+    if (ctx->b_ref_mode == NV_ENC_BFRAME_REF_MODE_EACH && ret != 1) {
+        av_log(avctx, AV_LOG_VERBOSE, "Each B frame as reference is not supported\n");
+        return AVERROR(ENOSYS);
+    } else if (ctx->b_ref_mode != NV_ENC_BFRAME_REF_MODE_DISABLED && ret == 0) {
+        av_log(avctx, AV_LOG_VERBOSE, "B frames as references are not supported\n");
+        return AVERROR(ENOSYS);
+    }
+#else
+    if (ctx->b_ref_mode != 0) {
+        av_log(avctx, AV_LOG_VERBOSE, "B frames as references need SDK 8.1 at build time\n");
+        return AVERROR(ENOSYS);
+    }
+#endif
+
     return 0;
 }
 
@@ -988,6 +1004,10 @@
     if (ctx->coder >= 0)
         h264->entropyCodingMode = ctx->coder;
 
+#ifdef NVENC_HAVE_BFRAME_REF_MODE
+    h264->useBFramesAsRef = ctx->b_ref_mode;
+#endif
+
     return 0;
 }