Permalink
Browse files

avcodec/nvdec: add support for 12 bit formats

  • Loading branch information...
BtbN committed Nov 11, 2017
1 parent c60bc02 commit 3f6294a53d58ececeacb42ce2ba8349a8b0a4abb
Showing with 14 additions and 2 deletions.
  1. +14 −2 libavcodec/nvdec.c
View
@@ -489,9 +489,21 @@ int ff_nvdec_frame_params(AVCodecContext *avctx,
frames_ctx->format = AV_PIX_FMT_CUDA;
frames_ctx->width = avctx->coded_width;
frames_ctx->height = avctx->coded_height;
frames_ctx->sw_format = sw_desc->comp[0].depth > 8 ?
AV_PIX_FMT_P010 : AV_PIX_FMT_NV12;
frames_ctx->initial_pool_size = dpb_size;
switch (sw_desc->comp[0].depth) {
case 8:
frames_ctx->sw_format = AV_PIX_FMT_NV12;
break;
case 10:
frames_ctx->sw_format = AV_PIX_FMT_P010;
break;
case 12:
frames_ctx->sw_format = AV_PIX_FMT_P016;
break;
default:
return AVERROR(EINVAL);
}
return 0;
}

0 comments on commit 3f6294a

Please sign in to comment.