nvenc: allow setting the number of slices

Based on a patch by Agatha Hu <ahu@nvidia.com>

diff --git a/libavcodec/nvenc.c b/libavcodec/nvenc.c
index 2593ba4cff93df8e8343710dfbf8b1c272158704..1ff27a19327fafc445339e91d71471812a7f186d 100644 (file)
--- a/libavcodec/nvenc.c
+++ b/libavcodec/nvenc.c
@@ -618,6 +618,9 @@ static int nvenc_setup_h264_config(AVCodecContext *avctx)
     h264->maxNumRefFrames = avctx->refs;
     h264->idrPeriod       = cc->gopLength;
 
+    h264->sliceMode     = 3;
+    h264->sliceModeData = FFMAX(avctx->slices, 1);
+
     if (ctx->flags & NVENC_LOSSLESS)
         h264->qpPrimeYZeroTransformBypassFlag = 1;
 
@@ -693,6 +696,9 @@ static int nvenc_setup_hevc_config(AVCodecContext *avctx)
     cc->profileGUID = NV_ENC_HEVC_PROFILE_MAIN_GUID;
     avctx->profile  = FF_PROFILE_HEVC_MAIN;
 
+    hevc->sliceMode     = 3;
+    hevc->sliceModeData = FFMAX(avctx->slices, 1);
+
     if (ctx->level) {
         hevc->level = ctx->level;
     } else {