diff options
author | Timo Rothenpieler <timo@rothenpieler.org> | 2018-05-09 18:59:05 +0200 |
---|---|---|
committer | Timo Rothenpieler <timo@rothenpieler.org> | 2018-05-10 00:34:22 +0200 |
commit | 41a18982d092505ffb6af7d02a3ff62eaf2d6d63 (patch) | |
tree | cedb666e0b3d68dfa6158308f15e7692f0615140 /libavutil/hwcontext_cuda.c | |
parent | 93d1756af2908150f7c8c0590b9ed246951d474a (diff) | |
download | ffmpeg-41a18982d092505ffb6af7d02a3ff62eaf2d6d63.tar.gz |
avutil/hwcontext_cuda: add support for nvenc rgb formats
Diffstat (limited to 'libavutil/hwcontext_cuda.c')
-rw-r--r-- | libavutil/hwcontext_cuda.c | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/libavutil/hwcontext_cuda.c b/libavutil/hwcontext_cuda.c index 668293bffc..b90c29a51a 100644 --- a/libavutil/hwcontext_cuda.c +++ b/libavutil/hwcontext_cuda.c @@ -38,6 +38,8 @@ static const enum AVPixelFormat supported_formats[] = { AV_PIX_FMT_P010, AV_PIX_FMT_P016, AV_PIX_FMT_YUV444P16, + AV_PIX_FMT_0RGB32, + AV_PIX_FMT_0BGR32, }; static int cuda_frames_get_constraints(AVHWDeviceContext *ctx, @@ -146,6 +148,10 @@ static int cuda_frames_init(AVHWFramesContext *ctx) case AV_PIX_FMT_YUV444P16: size = aligned_width * ctx->height * 6; break; + case AV_PIX_FMT_0RGB32: + case AV_PIX_FMT_0BGR32: + size = aligned_width * ctx->height * 4; + break; default: av_log(ctx, AV_LOG_ERROR, "BUG: Pixel format missing from size calculation."); return AVERROR_BUG; @@ -201,6 +207,11 @@ static int cuda_get_buffer(AVHWFramesContext *ctx, AVFrame *frame) frame->linesize[1] = aligned_width; frame->linesize[2] = aligned_width; break; + case AV_PIX_FMT_0BGR32: + case AV_PIX_FMT_0RGB32: + frame->data[0] = frame->buf[0]->data; + frame->linesize[0] = aligned_width * 4; + break; default: av_frame_unref(frame); return AVERROR_BUG; |