videolan
/
vlc
mirror of https://gitea.goldendeliverer.com/videolan/vlc.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
70593 Commits
7 Branches
132 Tags
718 MiB
 
 
 
 
 
 
Tree: 42f10bb870
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '42f10bb870'
${ noResults }
vlc/modules/codec/avcodec/vaapi.c

436 lines
12 KiB
Raw Blame History

/*****************************************************************************
* vaapi.c: VAAPI helpers for the libavcodec decoder
*****************************************************************************
* Copyright (C) 2017 VLC authors and VideoLAN
* Copyright (C) 2009-2010 Laurent Aimar
* Copyright (C) 2012-2014 Rémi Denis-Courmont
*
* Authors: Laurent Aimar <fenrir_AT_ videolan _DOT_ org>
*
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation; either version 2.1 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program; if not, write to the Free Software Foundation,
* Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
*****************************************************************************/
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
#include <assert.h>
#include <vlc_common.h>
#include <vlc_plugin.h>
#include <vlc_fourcc.h>
#include <vlc_picture.h>
#include <vlc_picture_pool.h>
#ifdef VLC_VA_BACKEND_XLIB
# include <vlc_xlib.h>
# include <va/va_x11.h>
#endif
#ifdef VLC_VA_BACKEND_DRM
# include <sys/types.h>
# include <sys/stat.h>
# include <fcntl.h>
# include <vlc_fs.h>
# include <va/va_drm.h>
#endif
#include <libavcodec/avcodec.h>
#include <libavcodec/vaapi.h>
#include "avcodec.h"
#include "va.h"
#include "../../hw/vaapi/vlc_vaapi.h"
struct vlc_va_sys_t
{
#ifdef VLC_VA_BACKEND_XLIB
Display *p_display_x11;
#endif
#ifdef VLC_VA_BACKEND_DRM
int drm_fd;
#endif
struct vaapi_context hw_ctx;
#ifndef VLC_VA_BACKEND_DR /* XLIB or DRM */
picture_pool_t *pool;
#endif
};
static int GetVaProfile(AVCodecContext *ctx, VAProfile *va_profile,
unsigned *pic_count)
{
VAProfile i_profile;
unsigned count = 3;
switch(ctx->codec_id)
{
case AV_CODEC_ID_MPEG1VIDEO:
case AV_CODEC_ID_MPEG2VIDEO:
i_profile = VAProfileMPEG2Main;
count = 4;
break;
case AV_CODEC_ID_MPEG4:
i_profile = VAProfileMPEG4AdvancedSimple;
break;
case AV_CODEC_ID_WMV3:
i_profile = VAProfileVC1Main;
break;
case AV_CODEC_ID_VC1:
i_profile = VAProfileVC1Advanced;
break;
case AV_CODEC_ID_H264:
i_profile = VAProfileH264High;
count = 18;
break;
case AV_CODEC_ID_HEVC:
if (ctx->profile == FF_PROFILE_HEVC_MAIN)
i_profile = VAProfileHEVCMain;
else if (ctx->profile == FF_PROFILE_HEVC_MAIN_10)
i_profile = VAProfileHEVCMain10;
else
return VLC_EGENERIC;
count = 18;
break;
case AV_CODEC_ID_VP8:
i_profile = VAProfileVP8Version0_3;
count = 5;
break;
case AV_CODEC_ID_VP9:
if (ctx->profile == FF_PROFILE_VP9_0)
i_profile = VAProfileVP9Profile0;
#if VA_CHECK_VERSION( 0, 39, 0 )
else if (ctx->profile == FF_PROFILE_VP9_2)
i_profile = VAProfileVP9Profile2;
#endif
else
return VLC_EGENERIC;
count = 10;
break;
default:
return VLC_EGENERIC;
}
*va_profile = i_profile;
*pic_count = count + ctx->thread_count;
return VLC_SUCCESS;
}
static int Extract(vlc_va_t *va, picture_t *pic, uint8_t *data)
{
(void) va; (void) pic; (void) data;
return VLC_SUCCESS;
}
#ifdef VLC_VA_BACKEND_DR
static int GetDR(vlc_va_t *va, picture_t *pic, uint8_t **data)
{
(void) va;
vlc_vaapi_PicAttachContext(pic);
*data = (void *) (uintptr_t) vlc_vaapi_PicGetSurface(pic);
return VLC_SUCCESS;
}
static void DeleteDR(vlc_va_t *va, AVCodecContext *avctx)
{
vlc_va_sys_t *sys = va->sys;
vlc_object_t *o = VLC_OBJECT(va);
(void) avctx;
vlc_vaapi_DestroyContext(o, sys->hw_ctx.display, sys->hw_ctx.context_id);
vlc_vaapi_DestroyConfig(o, sys->hw_ctx.display, sys->hw_ctx.config_id);
vlc_vaapi_ReleaseInstance(sys->hw_ctx.display);
free(sys);
}
static int CreateDR(vlc_va_t *va, AVCodecContext *ctx, enum PixelFormat pix_fmt,
const es_format_t *fmt, picture_sys_t *p_sys)
{
if (pix_fmt != AV_PIX_FMT_VAAPI_VLD)
return VLC_EGENERIC;
(void) fmt;
vlc_object_t *o = VLC_OBJECT(va);
int ret = VLC_EGENERIC;
vlc_va_sys_t *sys = NULL;
/* The picture must be allocated by the vout */
VADisplay *va_dpy = vlc_vaapi_GetInstance();
if (va_dpy == NULL)
return VLC_EGENERIC;
VASurfaceID *render_targets;
unsigned num_render_targets =
vlc_vaapi_PicSysGetRenderTargets(p_sys, &render_targets);
if (num_render_targets == 0)
goto error;
VAProfile i_profile;
unsigned count;
if (GetVaProfile(ctx, &i_profile, &count) != VLC_SUCCESS)
goto error;
sys = malloc(sizeof *sys);
if (unlikely(sys == NULL))
{
ret = VLC_ENOMEM;
goto error;
}
memset(sys, 0, sizeof (*sys));
/* */
sys->hw_ctx.display = va_dpy;
sys->hw_ctx.config_id = VA_INVALID_ID;
sys->hw_ctx.context_id = VA_INVALID_ID;
sys->hw_ctx.config_id =
vlc_vaapi_CreateConfigChecked(o, sys->hw_ctx.display, i_profile,
VAEntrypointVLD, VA_FOURCC_NV12);
if (sys->hw_ctx.config_id == VA_INVALID_ID)
goto error;
/* Create a context */
sys->hw_ctx.context_id =
vlc_vaapi_CreateContext(o, sys->hw_ctx.display, sys->hw_ctx.config_id,
ctx->coded_width, ctx->coded_height, VA_PROGRESSIVE,
render_targets, num_render_targets);
if (sys->hw_ctx.context_id == VA_INVALID_ID)
goto error;
ctx->hwaccel_context = &sys->hw_ctx;
va->sys = sys;
va->description = vaQueryVendorString(sys->hw_ctx.display);
va->get = GetDR;
va->release = NULL;
va->extract = Extract;
return VLC_SUCCESS;
error:
if (sys != NULL)
{
if (sys->hw_ctx.context_id != VA_INVALID_ID)
vlc_vaapi_DestroyContext(o, sys->hw_ctx.display, sys->hw_ctx.context_id);
if (sys->hw_ctx.config_id != VA_INVALID_ID)
vlc_vaapi_DestroyConfig(o, sys->hw_ctx.display, sys->hw_ctx.config_id);
free(sys);
}
vlc_vaapi_ReleaseInstance(va_dpy);
return ret;
}
#else /* XLIB or DRM */
static int Get(vlc_va_t *va, picture_t *pic, uint8_t **data)
{
vlc_va_sys_t *sys = va->sys;
picture_t *vapic = picture_pool_Wait(sys->pool);
if (vapic == NULL)
return VLC_EGENERIC;
vlc_vaapi_PicAttachContext(vapic);
pic->context = vapic->context->copy(vapic->context);
picture_Release(vapic);
if (pic->context == NULL)
return VLC_EGENERIC;
*data = (void *)(uintptr_t)vlc_vaapi_PicGetSurface(pic);
return VLC_SUCCESS;
}
static void Delete(vlc_va_t *va, AVCodecContext *avctx)
{
vlc_va_sys_t *sys = va->sys;
vlc_object_t *o = VLC_OBJECT(va);
(void) avctx;
picture_pool_Release(sys->pool);
vlc_vaapi_DestroyContext(o, sys->hw_ctx.display, sys->hw_ctx.context_id);
vlc_vaapi_DestroyConfig(o, sys->hw_ctx.display, sys->hw_ctx.config_id);
vlc_vaapi_ReleaseInstance(sys->hw_ctx.display);
#ifdef VLC_VA_BACKEND_XLIB
XCloseDisplay(sys->p_display_x11);
#endif
#ifdef VLC_VA_BACKEND_DRM
vlc_close(sys->drm_fd);
#endif
free(sys);
}
static int Create(vlc_va_t *va, AVCodecContext *ctx, enum PixelFormat pix_fmt,
const es_format_t *fmt, picture_sys_t *p_sys)
{
if (pix_fmt != AV_PIX_FMT_VAAPI_VLD)
return VLC_EGENERIC;
(void) fmt;
(void) p_sys;
vlc_object_t *o = VLC_OBJECT(va);
#ifdef VLC_VA_BACKEND_XLIB
if (!vlc_xlib_init(o))
{
msg_Warn(va, "Ignoring VA-X11 API");
return VLC_EGENERIC;
}
#endif
VAProfile i_profile;
unsigned count;
if (GetVaProfile(ctx, &i_profile, &count) != VLC_SUCCESS)
return VLC_EGENERIC;
vlc_va_sys_t *sys;
sys = malloc(sizeof(vlc_va_sys_t));
if (!sys)
return VLC_ENOMEM;
memset(sys, 0, sizeof (*sys));
/* */
sys->hw_ctx.display = NULL;
sys->hw_ctx.config_id = VA_INVALID_ID;
sys->hw_ctx.context_id = VA_INVALID_ID;
sys->pool = NULL;
/* Create a VA display */
#ifdef VLC_VA_BACKEND_XLIB
sys->p_display_x11 = XOpenDisplay(NULL);
if (!sys->p_display_x11)
{
msg_Err(va, "Could not connect to X server");
goto error;
}
sys->hw_ctx.display = vaGetDisplay(sys->p_display_x11);
#endif
#ifdef VLC_VA_BACKEND_DRM
static const char const *drm_device_paths[] = {
"/dev/dri/renderD128",
"/dev/dri/card0"
};
for (int i = 0; ARRAY_SIZE(drm_device_paths); i++)
{
sys->drm_fd = vlc_open(drm_device_paths[i], O_RDWR);
if (sys->drm_fd < 0)
continue;
sys->hw_ctx.display = vaGetDisplayDRM(sys->drm_fd);
if (sys->hw_ctx.display)
break;
vlc_close(sys->drm_fd);
sys->drm_fd = -1;
}
#endif
if (sys->hw_ctx.display == NULL)
{
msg_Err(va, "Could not get a VAAPI device");
goto error;
}
if (vlc_vaapi_Initialize(o, sys->hw_ctx.display))
{
sys->hw_ctx.display = NULL;
goto error;
}
if (vlc_vaapi_SetInstance(sys->hw_ctx.display))
{
msg_Err(va, "VAAPI instance already in use");
sys->hw_ctx.display = NULL;
goto error;
}
sys->hw_ctx.config_id =
vlc_vaapi_CreateConfigChecked(o, sys->hw_ctx.display, i_profile,
VAEntrypointVLD, 0);
/* Create surfaces */
assert(ctx->coded_width > 0 && ctx->coded_height > 0);
video_format_t vfmt = {
.i_chroma = VLC_CODEC_VAAPI_420,
.i_width = ctx->coded_width,
.i_height = ctx->coded_height,
.i_visible_width = ctx->coded_width,
.i_visible_height = ctx->coded_height
};
VASurfaceID *surfaces;
sys->pool = vlc_vaapi_PoolNew(o, sys->hw_ctx.display, count,
&surfaces, &vfmt, VA_RT_FORMAT_YUV420, 0);
if (!sys->pool)
goto error;
/* Create a context */
sys->hw_ctx.context_id =
vlc_vaapi_CreateContext(o, sys->hw_ctx.display, sys->hw_ctx.config_id,
ctx->coded_width, ctx->coded_height,
VA_PROGRESSIVE, surfaces, count);
if (sys->hw_ctx.context_id == VA_INVALID_ID)
goto error;
ctx->hwaccel_context = &sys->hw_ctx;
va->sys = sys;
va->description = vaQueryVendorString(sys->hw_ctx.display);
va->get = Get;
va->release = NULL;
va->extract = Extract;
return VLC_SUCCESS;
error:
if (sys->hw_ctx.context_id != VA_INVALID_ID)
vlc_vaapi_DestroyContext(o, sys->hw_ctx.display, sys->hw_ctx.context_id);
if (sys->pool != NULL)
picture_pool_Release(sys->pool);
if (sys->hw_ctx.config_id != VA_INVALID_ID)
vlc_vaapi_DestroyConfig(o, sys->hw_ctx.display, sys->hw_ctx.config_id);
if (sys->hw_ctx.display != NULL)
vlc_vaapi_ReleaseInstance(sys->hw_ctx.display);
#ifdef VLC_VA_BACKEND_XLIB
if( sys->p_display_x11 != NULL)
XCloseDisplay(sys->p_display_x11);
#endif
#ifdef VLC_VA_BACKEND_DRM
if( sys->drm_fd != -1 )
vlc_close(sys->drm_fd);
#endif
free(sys);
return VLC_EGENERIC;
}
#endif
vlc_module_begin ()
#if defined (VLC_VA_BACKEND_XLIB)
set_description( N_("VA-API video decoder via X11") )
set_capability( "hw decoder", 0 )
set_callbacks( Create, Delete )
#elif defined (VLC_VA_BACKEND_DRM)
set_description( N_("VA-API video decoder via DRM") )
set_capability( "hw decoder", 0 )
set_callbacks( Create, Delete )
#elif defined (VLC_VA_BACKEND_DR)
set_description( N_("VA-API direct video decoder") )
set_capability( "hw decoder", 100 )
set_callbacks( CreateDR, DeleteDR )
#endif
set_category( CAT_INPUT )
set_subcategory( SUBCAT_INPUT_VCODEC )
add_shortcut( "vaapi" )
vlc_module_end ()