From fb34c580bb2468b0b6664f819250d281d2f691e5 Mon Sep 17 00:00:00 2001 From: Timo Rothenpieler Date: Mon, 26 Jan 2015 13:28:22 +0100 Subject: [PATCH] avcodec/nvenc: Handle non-square pixel aspect ratios Reviewed-by: Philip Langdale Signed-off-by: Michael Niedermayer --- libavcodec/nvenc.c | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/libavcodec/nvenc.c b/libavcodec/nvenc.c index 2431c097c28..2cfc06a6b93 100644 --- a/libavcodec/nvenc.c +++ b/libavcodec/nvenc.c @@ -472,6 +472,7 @@ static av_cold int nvenc_encode_init(AVCodecContext *avctx) int i, num_mbs; int isLL = 0; int res = 0; + int dw, dh; #if NVENCAPI_MAJOR_VERSION < 5 GUID license = dummy_license; @@ -572,8 +573,20 @@ static av_cold int nvenc_encode_init(AVCodecContext *avctx) ctx->init_encode_params.encodeGUID = NV_ENC_CODEC_H264_GUID; ctx->init_encode_params.encodeHeight = avctx->height; ctx->init_encode_params.encodeWidth = avctx->width; - ctx->init_encode_params.darHeight = avctx->height; - ctx->init_encode_params.darWidth = avctx->width; + + if (avctx->sample_aspect_ratio.num && avctx->sample_aspect_ratio.den && + (avctx->sample_aspect_ratio.num != 1 || avctx->sample_aspect_ratio.num != 1)) { + av_reduce(&dw, &dh, + avctx->width * avctx->sample_aspect_ratio.num, + avctx->height * avctx->sample_aspect_ratio.den, + 1024 * 1024); + ctx->init_encode_params.darHeight = dh; + ctx->init_encode_params.darWidth = dw; + } else { + ctx->init_encode_params.darHeight = avctx->height; + ctx->init_encode_params.darWidth = avctx->width; + } + ctx->init_encode_params.frameRateNum = avctx->time_base.den; ctx->init_encode_params.frameRateDen = avctx->time_base.num * avctx->ticks_per_frame; -- 2.39.2