summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorTimo Rothenpieler <timo@rothenpieler.org>2018-05-09 18:59:05 +0200
committerTimo Rothenpieler <timo@rothenpieler.org>2018-05-10 00:34:22 +0200
commit41a18982d092505ffb6af7d02a3ff62eaf2d6d63 (patch)
treecedb666e0b3d68dfa6158308f15e7692f0615140
parent93d1756af2908150f7c8c0590b9ed246951d474a (diff)
downloadffmpeg-streaming-41a18982d092505ffb6af7d02a3ff62eaf2d6d63.zip
ffmpeg-streaming-41a18982d092505ffb6af7d02a3ff62eaf2d6d63.tar.gz
avutil/hwcontext_cuda: add support for nvenc rgb formats
-rw-r--r--libavutil/hwcontext_cuda.c11
-rw-r--r--libavutil/version.h2
2 files changed, 12 insertions, 1 deletions
diff --git a/libavutil/hwcontext_cuda.c b/libavutil/hwcontext_cuda.c
index 668293b..b90c29a 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;
diff --git a/libavutil/version.h b/libavutil/version.h
index 186fe07..44bdebd 100644
--- a/libavutil/version.h
+++ b/libavutil/version.h
@@ -80,7 +80,7 @@
#define LIBAVUTIL_VERSION_MAJOR 56
#define LIBAVUTIL_VERSION_MINOR 18
-#define LIBAVUTIL_VERSION_MICRO 101
+#define LIBAVUTIL_VERSION_MICRO 102
#define LIBAVUTIL_VERSION_INT AV_VERSION_INT(LIBAVUTIL_VERSION_MAJOR, \
LIBAVUTIL_VERSION_MINOR, \
OpenPOWER on IntegriCloud