mirror of
https://git.ffmpeg.org/ffmpeg.git
synced 2025-02-14 10:39:38 +00:00
nvenc: Extended rate-control support as provided by SDK 7
Signed-off-by: Luca Barbato <lu_zero@gentoo.org> Signed-off-by: Diego Biurrun <diego@biurrun.de>
This commit is contained in:
parent
358c887a9f
commit
70de2ea426
@ -610,6 +610,53 @@ static void nvenc_setup_rate_control(AVCodecContext *avctx)
|
||||
|
||||
if (rc->averageBitRate > 0)
|
||||
avctx->bit_rate = rc->averageBitRate;
|
||||
|
||||
#if NVENCAPI_MAJOR_VERSION >= 7
|
||||
if (ctx->aq) {
|
||||
ctx->config.rcParams.enableAQ = 1;
|
||||
ctx->config.rcParams.aqStrength = ctx->aq_strength;
|
||||
av_log(avctx, AV_LOG_VERBOSE, "AQ enabled.\n");
|
||||
}
|
||||
|
||||
if (ctx->temporal_aq) {
|
||||
ctx->config.rcParams.enableTemporalAQ = 1;
|
||||
av_log(avctx, AV_LOG_VERBOSE, "Temporal AQ enabled.\n");
|
||||
}
|
||||
|
||||
if (ctx->rc_lookahead) {
|
||||
int lkd_bound = FFMIN(ctx->nb_surfaces, ctx->async_depth) -
|
||||
ctx->config.frameIntervalP - 4;
|
||||
|
||||
if (lkd_bound < 0) {
|
||||
av_log(avctx, AV_LOG_WARNING,
|
||||
"Lookahead not enabled. Increase buffer delay (-delay).\n");
|
||||
} else {
|
||||
ctx->config.rcParams.enableLookahead = 1;
|
||||
ctx->config.rcParams.lookaheadDepth = av_clip(ctx->rc_lookahead, 0, lkd_bound);
|
||||
ctx->config.rcParams.disableIadapt = ctx->no_scenecut;
|
||||
ctx->config.rcParams.disableBadapt = !ctx->b_adapt;
|
||||
av_log(avctx, AV_LOG_VERBOSE,
|
||||
"Lookahead enabled: depth %d, scenecut %s, B-adapt %s.\n",
|
||||
ctx->config.rcParams.lookaheadDepth,
|
||||
ctx->config.rcParams.disableIadapt ? "disabled" : "enabled",
|
||||
ctx->config.rcParams.disableBadapt ? "disabled" : "enabled");
|
||||
}
|
||||
}
|
||||
|
||||
if (ctx->strict_gop) {
|
||||
ctx->config.rcParams.strictGOPTarget = 1;
|
||||
av_log(avctx, AV_LOG_VERBOSE, "Strict GOP target enabled.\n");
|
||||
}
|
||||
|
||||
if (ctx->nonref_p)
|
||||
ctx->config.rcParams.enableNonRefP = 1;
|
||||
|
||||
if (ctx->zerolatency)
|
||||
ctx->config.rcParams.zeroReorderDelay = 1;
|
||||
|
||||
if (ctx->quality)
|
||||
ctx->config.rcParams.targetQuality = ctx->quality;
|
||||
#endif /* NVENCAPI_MAJOR_VERSION >= 7 */
|
||||
}
|
||||
|
||||
static int nvenc_setup_h264_config(AVCodecContext *avctx)
|
||||
|
@ -171,6 +171,16 @@ typedef struct NVENCContext {
|
||||
int device;
|
||||
int flags;
|
||||
int async_depth;
|
||||
int rc_lookahead;
|
||||
int aq;
|
||||
int no_scenecut;
|
||||
int b_adapt;
|
||||
int temporal_aq;
|
||||
int zerolatency;
|
||||
int nonref_p;
|
||||
int strict_gop;
|
||||
int aq_strength;
|
||||
int quality;
|
||||
} NVENCContext;
|
||||
|
||||
int ff_nvenc_encode_init(AVCodecContext *avctx);
|
||||
|
@ -77,6 +77,18 @@ static const AVOption options[] = {
|
||||
{ "list", "List the available devices", 0, AV_OPT_TYPE_CONST, { .i64 = LIST_DEVICES }, 0, 0, VE, "device" },
|
||||
{ "async_depth", "Delay frame output by the given amount of frames", OFFSET(async_depth), AV_OPT_TYPE_INT, { .i64 = INT_MAX }, 0, INT_MAX, VE },
|
||||
{ "delay", "Delay frame output by the given amount of frames", OFFSET(async_depth), AV_OPT_TYPE_INT, { .i64 = INT_MAX }, 0, INT_MAX, VE },
|
||||
#if NVENCAPI_MAJOR_VERSION >= 7
|
||||
{ "rc-lookahead", "Number of frames to look ahead for rate-control", OFFSET(rc_lookahead), AV_OPT_TYPE_INT, { .i64 = -1 }, -1, INT_MAX, VE },
|
||||
{ "no-scenecut", "When lookahead is enabled, set this to 1 to disable adaptive I-frame insertion at scene cuts", OFFSET(no_scenecut), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE },
|
||||
{ "b_adapt", "When lookahead is enabled, set this to 0 to disable adaptive B-frame decision", OFFSET(b_adapt), AV_OPT_TYPE_INT, { .i64 = 1 }, 0, 1, VE },
|
||||
{ "spatial-aq", "set to 1 to enable Spatial AQ", OFFSET(aq), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE },
|
||||
{ "temporal-aq", "set to 1 to enable Temporal AQ", OFFSET(temporal_aq), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE },
|
||||
{ "zerolatency", "Set 1 to indicate zero latency operation (no reordering delay)", OFFSET(zerolatency), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE },
|
||||
{ "nonref_p", "Set this to 1 to enable automatic insertion of non-reference P-frames", OFFSET(nonref_p), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE },
|
||||
{ "strict_gop", "Set 1 to minimize GOP-to-GOP rate fluctuations", OFFSET(strict_gop), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE },
|
||||
{ "aq-strength", "When Spatial AQ is enabled, this field is used to specify AQ strength. AQ strength scale is from 1 (low) - 15 (aggressive)", OFFSET(aq_strength), AV_OPT_TYPE_INT, { .i64 = 8 }, 1, 15, VE },
|
||||
{ "cq", "Set target quality level (0 to 51, 0 means automatic) for constant quality mode in VBR rate control", OFFSET(quality), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 51, VE },
|
||||
#endif /* NVENCAPI_MAJOR_VERSION >= 7 */
|
||||
{ NULL }
|
||||
};
|
||||
|
||||
|
@ -77,6 +77,16 @@ static const AVOption options[] = {
|
||||
{ "list", "List the available devices", 0, AV_OPT_TYPE_CONST, { .i64 = LIST_DEVICES }, 0, 0, VE, "device" },
|
||||
{ "async_depth", "Delay frame output by the given amount of frames", OFFSET(async_depth), AV_OPT_TYPE_INT, { .i64 = INT_MAX }, 0, INT_MAX, VE },
|
||||
{ "delay", "Delay frame output by the given amount of frames", OFFSET(async_depth), AV_OPT_TYPE_INT, { .i64 = INT_MAX }, 0, INT_MAX, VE },
|
||||
#if NVENCAPI_MAJOR_VERSION >= 7
|
||||
{ "rc-lookahead", "Number of frames to look ahead for rate-control", OFFSET(rc_lookahead), AV_OPT_TYPE_INT, { .i64 = -1 }, -1, INT_MAX, VE },
|
||||
{ "no-scenecut", "When lookahead is enabled, set this to 1 to disable adaptive I-frame insertion at scene cuts", OFFSET(no_scenecut), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE },
|
||||
{ "spatial_aq", "set to 1 to enable Spatial AQ", OFFSET(aq), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE },
|
||||
{ "zerolatency", "Set 1 to indicate zero latency operation (no reordering delay)", OFFSET(zerolatency), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE },
|
||||
{ "nonref_p", "Set this to 1 to enable automatic insertion of non-reference P-frames", OFFSET(nonref_p), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE },
|
||||
{ "strict_gop", "Set 1 to minimize GOP-to-GOP rate fluctuations", OFFSET(strict_gop), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE },
|
||||
{ "aq-strength", "When Spatial AQ is enabled, this field is used to specify AQ strength. AQ strength scale is from 1 (low) - 15 (aggressive)", OFFSET(aq_strength), AV_OPT_TYPE_INT, { .i64 = 8 }, 1, 15, VE },
|
||||
{ "cq", "Set target quality level (0 to 51, 0 means automatic) for constant quality mode in VBR rate control", OFFSET(quality), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 51, VE },
|
||||
#endif /* NVENCAPI_MAJOR_VERSION >= 7 */
|
||||
{ NULL }
|
||||
};
|
||||
|
||||
|
@ -29,7 +29,7 @@
|
||||
|
||||
#define LIBAVCODEC_VERSION_MAJOR 57
|
||||
#define LIBAVCODEC_VERSION_MINOR 27
|
||||
#define LIBAVCODEC_VERSION_MICRO 0
|
||||
#define LIBAVCODEC_VERSION_MICRO 1
|
||||
|
||||
#define LIBAVCODEC_VERSION_INT AV_VERSION_INT(LIBAVCODEC_VERSION_MAJOR, \
|
||||
LIBAVCODEC_VERSION_MINOR, \
|
||||
|
Loading…
Reference in New Issue
Block a user