diff options
author | Anton Khirnov <anton@khirnov.net> | 2016-06-22 13:00:49 +0200 |
---|---|---|
committer | Anton Khirnov <anton@khirnov.net> | 2016-06-29 12:15:14 +0200 |
commit | 8bf6cfee7a9f1e67787ba1894b25bedb59a42a46 (patch) | |
tree | 508b6d79ac1263a0da35f1e0c6d57181c753a77f | |
parent | d929b0c2325d37dc845f6a83549a6411c8bb354f (diff) |
hwcontext_dxva2: add support for p010
-rw-r--r-- | libavutil/hwcontext_dxva2.c | 27 |
1 files changed, 15 insertions, 12 deletions
diff --git a/libavutil/hwcontext_dxva2.c b/libavutil/hwcontext_dxva2.c index e2c27bf160..f66c0fa740 100644 --- a/libavutil/hwcontext_dxva2.c +++ b/libavutil/hwcontext_dxva2.c @@ -65,6 +65,7 @@ static const struct { enum AVPixelFormat pix_fmt; } supported_formats[] = { { MKTAG('N', 'V', '1', '2'), AV_PIX_FMT_NV12 }, + { MKTAG('P', '0', '1', '0'), AV_PIX_FMT_P010 }, }; DEFINE_GUID(video_decoder_service, 0xfc51a551, 0xd5e7, 0x11d9, 0xaf, 0x55, 0x00, 0x05, 0x4e, 0x43, 0xff, 0x02); @@ -248,6 +249,10 @@ static int dxva2_transfer_data(AVHWFramesContext *ctx, AVFrame *dst, D3DLOCKED_RECT LockedRect; HRESULT hr; + uint8_t *surf_data[4] = { NULL }; + int surf_linesize[4] = { 0 }; + int i; + int download = !!src->hw_frames_ctx; surface = (IDirect3DSurface9*)(download ? src->data[3] : dst->data[3]); @@ -265,20 +270,18 @@ static int dxva2_transfer_data(AVHWFramesContext *ctx, AVFrame *dst, return AVERROR_UNKNOWN; } + for (i = 0; download ? dst->data[i] : src->data[i]; i++) + surf_linesize[i] = LockedRect.Pitch; + + av_image_fill_pointers(surf_data, ctx->sw_format, surfaceDesc.Height, + (uint8_t*)LockedRect.pBits, surf_linesize); + if (download) { - av_image_copy_plane(dst->data[0], dst->linesize[0], - (uint8_t*)LockedRect.pBits, LockedRect.Pitch, - src->width, src->height); - av_image_copy_plane(dst->data[1], dst->linesize[1], - (uint8_t*)LockedRect.pBits + LockedRect.Pitch * surfaceDesc.Height, - LockedRect.Pitch, src->width, src->height / 2); + av_image_copy(dst->data, dst->linesize, surf_data, surf_linesize, + ctx->sw_format, src->width, src->height); } else { - av_image_copy_plane((uint8_t*)LockedRect.pBits, LockedRect.Pitch, - dst->data[0], dst->linesize[0], - src->width, src->height); - av_image_copy_plane((uint8_t*)LockedRect.pBits + LockedRect.Pitch * surfaceDesc.Height, - LockedRect.Pitch, dst->data[1], dst->linesize[1], - src->width, src->height / 2); + av_image_copy(surf_data, surf_linesize, src->data, src->linesize, + ctx->sw_format, src->width, src->height); } IDirect3DSurface9_UnlockRect(surface); |