diff options
| author | Yogender Kumar Gupta <yogender.gupta@gmail.com> | 2016-09-19 20:01:10 +0530 |
|---|---|---|
| committer | Anton Khirnov <anton@khirnov.net> | 2016-09-22 09:47:48 +0200 |
| commit | 340f12f71207513672b5165d810cb6c8622c6b21 (patch) | |
| tree | 6e9533eaa743968f98c04e1ce3a3feff68f2da77 | |
| parent | 75d98e30afab61542faab3c0f11880834653bd6b (diff) | |
hwcontext_cuda: Add P010 and YUV444P16 pixel format
Signed-off-by: Anton Khirnov <anton@khirnov.net>
| -rw-r--r-- | libavutil/hwcontext_cuda.c | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/libavutil/hwcontext_cuda.c b/libavutil/hwcontext_cuda.c index b8781cec1b..260783426a 100644 --- a/libavutil/hwcontext_cuda.c +++ b/libavutil/hwcontext_cuda.c @@ -32,7 +32,9 @@ typedef struct CUDAFramesContext { static const enum AVPixelFormat supported_formats[] = { AV_PIX_FMT_NV12, AV_PIX_FMT_YUV420P, + AV_PIX_FMT_P010, AV_PIX_FMT_YUV444P, + AV_PIX_FMT_YUV444P16, }; static void cuda_buffer_free(void *opaque, uint8_t *data) @@ -105,9 +107,15 @@ static int cuda_frames_init(AVHWFramesContext *ctx) case AV_PIX_FMT_YUV420P: size = ctx->width * ctx->height * 3 / 2; break; + case AV_PIX_FMT_P010: + size = ctx->width * ctx->height * 3; + break; case AV_PIX_FMT_YUV444P: size = ctx->width * ctx->height * 3; break; + case AV_PIX_FMT_YUV444P16: + size = ctx->width * ctx->height * 6; + break; } ctx->internal->pool_internal = av_buffer_pool_init2(size, ctx, cuda_pool_alloc, NULL); @@ -139,6 +147,12 @@ static int cuda_get_buffer(AVHWFramesContext *ctx, AVFrame *frame) frame->linesize[1] = ctx->width / 2; frame->linesize[2] = ctx->width / 2; break; + case AV_PIX_FMT_P010: + frame->data[0] = frame->buf[0]->data; + frame->data[1] = frame->data[0] + 2 * ctx->width * ctx->height; + frame->linesize[0] = 2 * ctx->width; + frame->linesize[1] = 2 * ctx->width; + break; case AV_PIX_FMT_YUV444P: frame->data[0] = frame->buf[0]->data; frame->data[1] = frame->data[0] + ctx->width * ctx->height; @@ -147,6 +161,14 @@ static int cuda_get_buffer(AVHWFramesContext *ctx, AVFrame *frame) frame->linesize[1] = ctx->width; frame->linesize[2] = ctx->width; break; + case AV_PIX_FMT_YUV444P16: + frame->data[0] = frame->buf[0]->data; + frame->data[1] = frame->data[0] + 2 * ctx->width * ctx->height; + frame->data[2] = frame->data[1] + 2 * ctx->width * ctx->height; + frame->linesize[0] = 2 * ctx->width; + frame->linesize[1] = 2 * ctx->width; + frame->linesize[2] = 2 * ctx->width; + break; default: av_frame_unref(frame); return AVERROR_BUG; |