RepoMirrors
/
ffmpeg
mirror of https://git.ffmpeg.org/ffmpeg.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

lavu/hwcontext_qsv: add support for dynamic frame pool in qsv_frames_derive_to 

Make it work with the source which has a dynamic frame pool.

Signed-off-by: Haihao Xiang <haihao.xiang@intel.com>
This commit is contained in:
Haihao Xiang 2024-05-08 14:03:10 +08:00
parent 96db4a62e0
commit 932f78c4e5
1 changed files with 55 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
libavutil/hwcontext_qsv.c
View File

@ -1972,19 +1972,53 @@ static int qsv_transfer_data_to(AVHWFramesContext *ctx, AVFrame *dst,
return 0; return 0;
} }
static int qsv_frames_derive_to(AVHWFramesContext *dst_ctx, static int qsv_dynamic_frames_derive_to(AVHWFramesContext *dst_ctx,
AVHWFramesContext *src_ctx, int flags) AVHWFramesContext *src_ctx, int flags)
{
QSVFramesContext *s = dst_ctx->hwctx;
AVQSVFramesContext *dst_hwctx = &s->p;
mfxFrameSurface1 mfx_surf1;
switch (src_ctx->device_ctx->type) {
#if CONFIG_VAAPI
case AV_HWDEVICE_TYPE_VAAPI:
dst_hwctx->frame_type = MFX_MEMTYPE_VIDEO_MEMORY_DECODER_TARGET;
break;
#endif
#if CONFIG_D3D11VA
case AV_HWDEVICE_TYPE_D3D11VA:
{
AVD3D11VAFramesContext *src_hwctx = src_ctx->hwctx;
if (src_hwctx->BindFlags & D3D11_BIND_RENDER_TARGET) {
dst_hwctx->frame_type |= MFX_MEMTYPE_VIDEO_MEMORY_PROCESSOR_TARGET;
} else {
dst_hwctx->frame_type |= MFX_MEMTYPE_VIDEO_MEMORY_DECODER_TARGET;
}
}
break;
#endif
default:
return AVERROR(ENOSYS);
}
memset(&mfx_surf1, 0, sizeof(mfx_surf1));
qsv_init_surface(dst_ctx, &mfx_surf1);
s->frame_info = mfx_surf1.Info;
dst_hwctx->info = &s->frame_info;
dst_hwctx->nb_surfaces = 0;
return 0;
}
static int qsv_fixed_frames_derive_to(AVHWFramesContext *dst_ctx,
AVHWFramesContext *src_ctx, int flags)
{ {
QSVFramesContext *s = dst_ctx->hwctx; QSVFramesContext *s = dst_ctx->hwctx;
AVQSVFramesContext *dst_hwctx = &s->p; AVQSVFramesContext *dst_hwctx = &s->p;
int i; int i;
if (src_ctx->initial_pool_size == 0) {
av_log(dst_ctx, AV_LOG_ERROR, "Only fixed-size pools can be "
"mapped to QSV frames.\n");
return AVERROR(EINVAL);
}
switch (src_ctx->device_ctx->type) { switch (src_ctx->device_ctx->type) {
#if CONFIG_VAAPI #if CONFIG_VAAPI
case AV_HWDEVICE_TYPE_VAAPI: case AV_HWDEVICE_TYPE_VAAPI:
@ -2075,6 +2109,19 @@ static int qsv_frames_derive_to(AVHWFramesContext *dst_ctx,
return 0; return 0;
} }
static int qsv_frames_derive_to(AVHWFramesContext *dst_ctx,
AVHWFramesContext *src_ctx, int flags)
{
if (src_ctx->initial_pool_size < 0) {
av_log(dst_ctx, AV_LOG_ERROR, "Invalid src frame pool. \n");
return AVERROR(EINVAL);
} else if (src_ctx->initial_pool_size == 0) {
return qsv_dynamic_frames_derive_to(dst_ctx, src_ctx, flags);
} else {
return qsv_fixed_frames_derive_to(dst_ctx, src_ctx, flags);
}
}
static int qsv_map_to(AVHWFramesContext *dst_ctx, static int qsv_map_to(AVHWFramesContext *dst_ctx,
AVFrame *dst, const AVFrame *src, int flags) AVFrame *dst, const AVFrame *src, int flags)
{ {