#include <dxva2api.h>
#include <initguid.h>
+#include "avassert.h"
#include "common.h"
#include "hwcontext.h"
#include "hwcontext_dxva2.h"
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);
HRESULT hr;
int download = !!src->hw_frames_ctx;
+ int bytes_per_component;
+
+ switch (ctx->sw_format) {
+ case AV_PIX_FMT_NV12:
+ bytes_per_component = 1;
+ break;
+ case AV_PIX_FMT_P010:
+ bytes_per_component = 2;
+ break;
+ default:
+ av_assert0(0);
+ }
surface = (IDirect3DSurface9*)(download ? src->data[3] : dst->data[3]);
if (download) {
av_image_copy_plane(dst->data[0], dst->linesize[0],
(uint8_t*)LockedRect.pBits, LockedRect.Pitch,
- src->width, src->height);
+ src->width * bytes_per_component, 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);
+ LockedRect.Pitch, src->width * bytes_per_component, src->height / 2);
} else {
av_image_copy_plane((uint8_t*)LockedRect.pBits, LockedRect.Pitch,
dst->data[0], dst->linesize[0],
- src->width, src->height);
+ src->width * bytes_per_component, 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);
+ src->width * bytes_per_component, src->height / 2);
}
IDirect3DSurface9_UnlockRect(surface);