From 4aeb7a88ec69aff6490e7f94f5987f7525b7eab7 Mon Sep 17 00:00:00 2001 From: "Sven C. Dack" Date: Wed, 7 Sep 2016 15:35:14 +0200 Subject: [PATCH] avcodec/nvenc: support RGB input nvenc still encodes as yuv, but does the conversion internally which brings some performance gains. Signed-off-by: Timo Rothenpieler --- libavcodec/nvenc.c | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/libavcodec/nvenc.c b/libavcodec/nvenc.c index b3351160070..bff204d9266 100644 --- a/libavcodec/nvenc.c +++ b/libavcodec/nvenc.c @@ -81,6 +81,8 @@ const enum AVPixelFormat ff_nvenc_pix_fmts[] = { AV_PIX_FMT_P010, AV_PIX_FMT_YUV444P, AV_PIX_FMT_YUV444P16, + AV_PIX_FMT_0RGB32, + AV_PIX_FMT_0BGR32, #if CONFIG_CUDA AV_PIX_FMT_CUDA, #endif @@ -1032,6 +1034,14 @@ static av_cold int nvenc_alloc_surface(AVCodecContext *avctx, int idx) ctx->surfaces[idx].format = NV_ENC_BUFFER_FORMAT_YUV444_10BIT; break; + case AV_PIX_FMT_0RGB32: + ctx->surfaces[idx].format = NV_ENC_BUFFER_FORMAT_ARGB; + break; + + case AV_PIX_FMT_0BGR32: + ctx->surfaces[idx].format = NV_ENC_BUFFER_FORMAT_ABGR; + break; + default: av_log(avctx, AV_LOG_FATAL, "Invalid input pixel format\n"); return AVERROR(EINVAL); -- 2.39.2