[FFmpeg-devel] [PATCH] cuda: add support for yuv420p10

[Lynne <dev@lynne.ee>]

This patch adds support for accepting yuv420p10 in encoders and cuda.
---
 libavcodec/nvenc.c         | 2 ++
 libavutil/hwcontext_cuda.c | 7 +++++--
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/libavcodec/nvenc.c b/libavcodec/nvenc.c
index 41a4dc55f4..ccf8cc6d39 100644
--- a/libavcodec/nvenc.c
+++ b/libavcodec/nvenc.c
@@ -57,6 +57,7 @@
 
 const enum AVPixelFormat ff_nvenc_pix_fmts[] = {
     AV_PIX_FMT_YUV420P,
+    AV_PIX_FMT_YUV420P10,
     AV_PIX_FMT_NV12,
     AV_PIX_FMT_P010,
     AV_PIX_FMT_YUV444P,
@@ -99,6 +100,7 @@ const AVCodecHWConfigInternal *const ff_nvenc_hw_configs[] = {
                             pix_fmt == AV_PIX_FMT_YUV444P16 || \
                             pix_fmt == AV_PIX_FMT_X2RGB10   || \
                             pix_fmt == AV_PIX_FMT_X2BGR10   || \
+                            pix_fmt == AV_PIX_FMT_YUV420P10 || \
                             pix_fmt == AV_PIX_FMT_GBRP16)
 
 #define IS_RGB(pix_fmt)    (pix_fmt == AV_PIX_FMT_0RGB32  || \
diff --git a/libavutil/hwcontext_cuda.c b/libavutil/hwcontext_cuda.c
index 5721612225..495d3687c2 100644
--- a/libavutil/hwcontext_cuda.c
+++ b/libavutil/hwcontext_cuda.c
@@ -44,6 +44,7 @@ static const enum AVPixelFormat supported_formats[] = {
     AV_PIX_FMT_NV12,
     AV_PIX_FMT_NV16,
     AV_PIX_FMT_YUV420P,
+    AV_PIX_FMT_YUV420P10,
     AV_PIX_FMT_YUVA420P,
     AV_PIX_FMT_YUV444P,
     AV_PIX_FMT_P010,
@@ -162,7 +163,8 @@ static int cuda_frames_init(AVHWFramesContext *ctx)
     // YUV420P is a special case.
     // Since nvenc expects the U/V planes to have half the linesize of the Y plane
     // alignment has to be doubled to ensure the U/V planes still end up aligned.
-    if (ctx->sw_format == AV_PIX_FMT_YUV420P)
+    if (ctx->sw_format == AV_PIX_FMT_YUV420P ||
+        ctx->sw_format == AV_PIX_FMT_YUV420P10)
         priv->tex_alignment *= 2;
 
     av_pix_fmt_get_chroma_sub_sample(ctx->sw_format, &priv->shift_width, &priv->shift_height);
@@ -197,7 +199,8 @@ static int cuda_get_buffer(AVHWFramesContext *ctx, AVFrame *frame)
 
     // YUV420P is a special case.
     // Nvenc expects the U/V planes in swapped order from how ffmpeg expects them, also chroma is half-aligned
-    if (ctx->sw_format == AV_PIX_FMT_YUV420P) {
+    if (ctx->sw_format == AV_PIX_FMT_YUV420P ||
+        ctx->sw_format == AV_PIX_FMT_YUV420P10) {
         frame->linesize[1] = frame->linesize[2] = frame->linesize[0] / 2;
         frame->data[2]     = frame->data[1];
         frame->data[1]     = frame->data[2] + frame->linesize[2] * (ctx->height / 2);
-- 
2.49.0.395.g12beb8f557c
_______________________________________________
ffmpeg-devel mailing list
ffmpeg-devel@ffmpeg.org
https://ffmpeg.org/mailman/listinfo/ffmpeg-devel

To unsubscribe, visit link above, or email
ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe".