libavutil/hwcontext_opencl: fix a bug for mapping qsv frame to opencl

mfxHDLPair was added to qsv, so modify qsv->opencl map function as well.
Now the following commandline works:

ffmpeg -v verbose -init_hw_device vaapi=va:/dev/dri/renderD128 \
-init_hw_device qsv=qs@va -init_hw_device opencl=ocl@va -filter_hw_device ocl \
-hwaccel qsv -hwaccel_output_format qsv -hwaccel_device qs -c:v h264_qsv \
-i input.264 -vf "hwmap=derive_device=opencl,format=opencl,avgblur_opencl, \
hwmap=derive_device=qsv:reverse=1:extra_hw_frames=32,format=qsv" \
-c:v h264_qsv output.264

Signed-off-by: nyanmisaka <nst799610810@gmail.com>
Signed-off-by: Wenbin Chen <wenbin.chen@intel.com>
This commit is contained in:
nyanmisaka 2022-01-11 14:55:38 +08:00 committed by Haihao Xiang
parent a5b1e632c9
commit 4cc7239d8b
2 changed files with 47 additions and 1 deletions

View File

@ -72,6 +72,12 @@
#include "hwcontext_drm.h"
#endif
#if HAVE_OPENCL_VAAPI_INTEL_MEDIA && CONFIG_LIBMFX
extern int ff_qsv_get_surface_base_handle(mfxFrameSurface1 *surf,
enum AVHWDeviceType base_dev_typ,
void **base_handle);
#endif
typedef struct OpenCLDeviceContext {
// Default command queue to use for transfer/mapping operations on
@ -2248,8 +2254,14 @@ static int opencl_map_from_qsv(AVHWFramesContext *dst_fc, AVFrame *dst,
#if CONFIG_LIBMFX
if (src->format == AV_PIX_FMT_QSV) {
void *base_handle;
mfxFrameSurface1 *mfx_surface = (mfxFrameSurface1*)src->data[3];
va_surface = *(VASurfaceID*)mfx_surface->Data.MemId;
err = ff_qsv_get_surface_base_handle(mfx_surface,
AV_HWDEVICE_TYPE_VAAPI,
&base_handle);
if (err < 0)
return err;
va_surface = *(VASurfaceID *)base_handle;
} else
#endif
if (src->format == AV_PIX_FMT_VAAPI) {

View File

@ -112,6 +112,40 @@ static const struct {
#endif
};
extern int ff_qsv_get_surface_base_handle(mfxFrameSurface1 *surf,
enum AVHWDeviceType base_dev_type,
void **base_handle);
/**
* Caller needs to allocate enough space for base_handle pointer.
**/
int ff_qsv_get_surface_base_handle(mfxFrameSurface1 *surf,
enum AVHWDeviceType base_dev_type,
void **base_handle)
{
mfxHDLPair *handle_pair;
handle_pair = surf->Data.MemId;
switch (base_dev_type) {
#if CONFIG_VAAPI
case AV_HWDEVICE_TYPE_VAAPI:
base_handle[0] = handle_pair->first;
return 0;
#endif
#if CONFIG_D3D11VA
case AV_HWDEVICE_TYPE_D3D11VA:
base_handle[0] = handle_pair->first;
base_handle[1] = handle_pair->secode;
return 0;
#endif
#if CONFIG_DXVA2
case AV_HWDEVICE_TYPE_DXVA2:
base_handle[0] = handle_pair->first;
return 0;
#endif
}
return AVERROR(EINVAL);
}
static uint32_t qsv_fourcc_from_pix_fmt(enum AVPixelFormat pix_fmt)
{
int i;