nvenc: Propagate desired number of reference frames.
authorPhilip Langdale <philipl@overt.org>
Sat, 24 Jan 2015 20:52:58 +0000 (12:52 -0800)
committerMichael Niedermayer <michaelni@gmx.at>
Sat, 24 Jan 2015 21:59:54 +0000 (22:59 +0100)
Reviewed-by: Timo Rothenpieler <timo@rothenpieler.org>
Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
libavcodec/nvenc.c

index 7d82d58..16ef93f 100644 (file)
@@ -590,6 +590,11 @@ static av_cold int nvenc_encode_init(AVCodecContext *avctx)
     memcpy(&ctx->encode_config, &preset_config.presetCfg, sizeof(ctx->encode_config));
     ctx->encode_config.version = NV_ENC_CONFIG_VER;
 
+    if (avctx->refs >= 0) {
+        /* 0 means "let the hardware decide" */
+        ctx->encode_config.encodeCodecConfig.h264Config.maxNumRefFrames = avctx->refs;
+    }
+
     if (avctx->gop_size >= 0) {
         ctx->encode_config.gopLength = avctx->gop_size;
         ctx->encode_config.encodeCodecConfig.h264Config.idrPeriod = avctx->gop_size;