[FFmpeg-cvslog] nvenc: allow setting the number of slices

[Anton Khirnov]

ffmpeg | branch: master | Anton Khirnov <anton at khirnov.net> | Sat May 14 16:47:55 2016 +0200| [3399a26d3f57d462e839c0ee51223ae9aca20852] | committer: Anton Khirnov

nvenc: allow setting the number of slices

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

> http://git.videolan.org/gitweb.cgi/ffmpeg.git/?a=commit;h=3399a26d3f57d462e839c0ee51223ae9aca20852
---

 libavcodec/nvenc.c |    6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/libavcodec/nvenc.c b/libavcodec/nvenc.c
index 2593ba4..1ff27a1 100644
--- 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 {