#ifdef HAVE_LIBAVCODEC_AVCODEC_H
# include <libavcodec/avcodec.h>
-# ifdef HAVE_AVCODEC_VAAPI
-# include <libavcodec/vaapi.h>
-# endif
#elif defined(HAVE_FFMPEG_AVCODEC_H)
# include <ffmpeg/avcodec.h>
#else
#endif
#include "avcodec.h"
-#include "vaapi.h"
+#include "va.h"
+#include "copy.h"
#ifdef HAVE_AVCODEC_VAAPI
+#include <libavcodec/vaapi.h>
+
#include <X11/Xlib.h>
#include <va/va_x11.h>
-
typedef struct
{
VASurfaceID i_id;
} vlc_va_surface_t;
-struct vlc_va_t
+typedef struct
{
+ vlc_va_t va;
+
/* */
Display *p_display_x11;
VADisplay p_display;
vlc_va_surface_t *p_surface;
VAImage image;
+ copy_cache_t image_cache;
-};
-
-static int VaOpen( vlc_va_t *p_va, int i_codec_id );
-static void VaClose( vlc_va_t *p_va );
+} vlc_va_vaapi_t;
-static int VaCreateSurfaces( vlc_va_t *p_va, void **pp_hw_ctx, vlc_fourcc_t *pi_chroma,
- int i_width, int i_height );
-static void VaDestroySurfaces( vlc_va_t *p_va );
-
-vlc_va_t *VaNew( int i_codec_id )
+static vlc_va_vaapi_t *vlc_va_vaapi_Get( void *p_va )
{
- vlc_va_t *p_va = calloc( 1, sizeof(*p_va) );
- if( !p_va )
- return NULL;
-
- if( VaOpen( p_va, i_codec_id ) )
- {
- free( p_va );
- return NULL;
- }
return p_va;
}
-void VaDelete( vlc_va_t *p_va )
-{
- VaClose( p_va );
- free( p_va );
-}
-int VaSetup( vlc_va_t *p_va, void **pp_hw_ctx, vlc_fourcc_t *pi_chroma,
- int i_width, int i_height )
-{
- if( p_va->i_surface_width == i_width &&
- p_va->i_surface_height == i_height )
- return VLC_SUCCESS;
-
- *pp_hw_ctx = NULL;
- *pi_chroma = 0;
- if( p_va->i_surface_width || p_va->i_surface_height )
- VaDestroySurfaces( p_va );
-
- if( i_width > 0 && i_height > 0 )
- return VaCreateSurfaces( p_va, pp_hw_ctx, pi_chroma, i_width, i_height );
-
- return VLC_EGENERIC;
-}
-void VaVersion( vlc_va_t *p_va, char *psz_version, size_t i_version )
-{
- snprintf( psz_version, i_version, "%d.%d", p_va->i_version_major, p_va->i_version_minor );
-}
-static int VaOpen( vlc_va_t *p_va, int i_codec_id )
+/* */
+static int Open( vlc_va_vaapi_t *p_va, int i_codec_id )
{
VAProfile i_profile;
int i_surface_count;
p_va->i_surface_count = i_surface_count;
+ if( asprintf( &p_va->va.description, "VA API version %d.%d",
+ p_va->i_version_major, p_va->i_version_minor ) < 0 )
+ p_va->va.description = NULL;
+
return VLC_SUCCESS;
error:
return VLC_EGENERIC;
}
-static void VaClose( vlc_va_t *p_va )
+
+static void DestroySurfaces( vlc_va_vaapi_t *p_va )
{
- if( p_va->i_surface_width || p_va->i_surface_height )
- VaDestroySurfaces( p_va );
+ if( p_va->image.image_id )
+ {
+ CopyCleanCache( &p_va->image_cache );
+ vaDestroyImage( p_va->p_display, p_va->image.image_id );
+ }
- if( p_va->i_config_id )
- vaDestroyConfig( p_va->p_display, p_va->i_config_id );
- if( p_va->p_display )
- vaTerminate( p_va->p_display );
- if( p_va->p_display_x11 )
- XCloseDisplay( p_va->p_display_x11 );
-}
+ if( p_va->i_context_id )
+ vaDestroyContext( p_va->p_display, p_va->i_context_id );
-static int VaCreateSurfaces( vlc_va_t *p_va, void **pp_hw_ctx, vlc_fourcc_t *pi_chroma,
- int i_width, int i_height )
+ for( int i = 0; i < p_va->i_surface_count && p_va->p_surface; i++ )
+ {
+ vlc_va_surface_t *p_surface = &p_va->p_surface[i];
+
+ if( p_surface->i_id != VA_INVALID_SURFACE )
+ vaDestroySurfaces( p_va->p_display, &p_surface->i_id, 1 );
+ }
+ free( p_va->p_surface );
+
+ /* */
+ p_va->image.image_id = 0;
+ p_va->i_context_id = 0;
+ p_va->p_surface = NULL;
+ p_va->i_surface_width = 0;
+ p_va->i_surface_height = 0;
+}
+static int CreateSurfaces( vlc_va_vaapi_t *p_va, void **pp_hw_ctx, vlc_fourcc_t *pi_chroma,
+ int i_width, int i_height )
{
assert( i_width > 0 && i_height > 0 );
for( int i = 0; i < i_fmt_count; i++ )
{
if( p_fmt[i].fourcc == VA_FOURCC( 'Y', 'V', '1', '2' ) ||
- p_fmt[i].fourcc == VA_FOURCC( 'I', '4', '2', '0' ) )
+ p_fmt[i].fourcc == VA_FOURCC( 'I', '4', '2', '0' ) ||
+ p_fmt[i].fourcc == VA_FOURCC( 'N', 'V', '1', '2' ) )
{
- i_chroma = VLC_FOURCC( 'I', '4', '2', '0' );
+ i_chroma = VLC_CODEC_YV12;
fmt = p_fmt[i];
+ break;
}
- /* TODO: It seems that these may also be available (but not
- * with my setup):
- * VA_FOURCC( 'N', 'V', '1', '2')
- * VA_FOURCC( 'U', 'Y', 'V', 'Y')
- * VA_FOURCC( 'Y', 'U', 'Y', 'V')
- */
}
free( p_fmt );
if( !i_chroma )
p_va->image.image_id = 0;
goto error;
}
+ CopyInitCache( &p_va->image_cache, i_width );
/* Setup the ffmpeg hardware context */
*pp_hw_ctx = &p_va->hw_ctx;
return VLC_SUCCESS;
error:
- VaDestroySurfaces( p_va );
+ DestroySurfaces( p_va );
return VLC_EGENERIC;
}
-static void VaDestroySurfaces( vlc_va_t *p_va )
-{
- if( p_va->image.image_id )
- vaDestroyImage( p_va->p_display, p_va->image.image_id );
- if( p_va->i_context_id )
- vaDestroyContext( p_va->p_display, p_va->i_context_id );
+static int Setup( vlc_va_t *p_external, void **pp_hw_ctx, vlc_fourcc_t *pi_chroma,
+ int i_width, int i_height )
+{
+ vlc_va_vaapi_t *p_va = vlc_va_vaapi_Get(p_external);
- for( int i = 0; i < p_va->i_surface_count && p_va->p_surface; i++ )
+ if( p_va->i_surface_width == i_width &&
+ p_va->i_surface_height == i_height )
{
- vlc_va_surface_t *p_surface = &p_va->p_surface[i];
-
- if( p_surface->i_id != VA_INVALID_SURFACE )
- vaDestroySurfaces( p_va->p_display, &p_surface->i_id, 1 );
+ *pp_hw_ctx = &p_va->hw_ctx;
+ *pi_chroma = p_va->i_surface_chroma;
+ return VLC_SUCCESS;
}
- free( p_va->p_surface );
- /* */
- p_va->image.image_id = 0;
- p_va->i_context_id = 0;
- p_va->p_surface = NULL;
- p_va->i_surface_width = 0;
- p_va->i_surface_height = 0;
-}
+ *pp_hw_ctx = NULL;
+ *pi_chroma = 0;
+ if( p_va->i_surface_width || p_va->i_surface_height )
+ DestroySurfaces( p_va );
+
+ if( i_width > 0 && i_height > 0 )
+ return CreateSurfaces( p_va, pp_hw_ctx, pi_chroma, i_width, i_height );
-int VaExtract( vlc_va_t *p_va, picture_t *p_picture, AVFrame *p_ff )
+ return VLC_EGENERIC;
+}
+static int Extract( vlc_va_t *p_external, picture_t *p_picture, AVFrame *p_ff )
{
+ vlc_va_vaapi_t *p_va = vlc_va_vaapi_Get(p_external);
+
+ if( !p_va->image_cache.buffer )
+ return VLC_EGENERIC;
+
VASurfaceID i_surface_id = (VASurfaceID)(uintptr_t)p_ff->data[3];
#if VA_CHECK_VERSION(0,31,0)
if( vaMapBuffer( p_va->p_display, p_va->image.buf, &p_base ) )
return VLC_EGENERIC;
- for( int i_plane = 0; i_plane < p_picture->i_planes; i_plane++ )
+ const uint32_t i_fourcc = p_va->image.format.fourcc;
+ if( i_fourcc == VA_FOURCC('Y','V','1','2') ||
+ i_fourcc == VA_FOURCC('I','4','2','0') )
{
- const int i_src_plane = ((p_va->image.format.fourcc == VA_FOURCC('Y','V','1','2' )) && i_plane != 0) ? (3 - i_plane) : i_plane;
- const uint8_t *p_src = (uint8_t*)p_base + p_va->image.offsets[i_src_plane];
- const int i_src_stride = p_va->image.pitches[i_src_plane];
-
- uint8_t *p_dst = p_picture->p[i_plane].p_pixels;
- const int i_dst_stride = p_picture->p[i_plane].i_pitch;
+ bool b_swap_uv = i_fourcc == VA_FOURCC('I','4','2','0');
+ uint8_t *pp_plane[3];
+ size_t pi_pitch[3];
- if( i_src_stride != i_dst_stride )
+ for( int i = 0; i < 3; i++ )
{
- for( int i = 0; i < p_picture->p[i_plane].i_visible_lines; i++ )
- {
- vlc_memcpy( p_dst, p_src, __MIN( i_src_stride, i_dst_stride ) );
- p_src += i_src_stride;
- p_dst += i_dst_stride;
- }
+ const int i_src_plane = (b_swap_uv && i != 0) ? (3 - i) : i;
+ pp_plane[i] = (uint8_t*)p_base + p_va->image.offsets[i_src_plane];
+ pi_pitch[i] = p_va->image.pitches[i_src_plane];
}
- else
+ CopyFromYv12( p_picture, pp_plane, pi_pitch,
+ p_va->i_surface_width,
+ p_va->i_surface_height,
+ &p_va->image_cache );
+ }
+ else
+ {
+ assert( i_fourcc == VA_FOURCC('N','V','1','2') );
+ uint8_t *pp_plane[2];
+ size_t pi_pitch[2];
+
+ for( int i = 0; i < 2; i++ )
{
- vlc_memcpy( p_dst, p_src, p_picture->p[i_plane].i_visible_lines * i_src_stride );
+ pp_plane[i] = (uint8_t*)p_base + p_va->image.offsets[i];
+ pi_pitch[i] = p_va->image.pitches[i];
}
+ CopyFromNv12( p_picture, pp_plane, pi_pitch,
+ p_va->i_surface_width,
+ p_va->i_surface_height,
+ &p_va->image_cache );
}
if( vaUnmapBuffer( p_va->p_display, p_va->image.buf ) )
return VLC_SUCCESS;
}
-int VaGrabSurface( vlc_va_t *p_va, AVFrame *p_ff )
+static int Get( vlc_va_t *p_external, AVFrame *p_ff )
{
+ vlc_va_vaapi_t *p_va = vlc_va_vaapi_Get(p_external);
int i_old;
int i;
}
return VLC_SUCCESS;
}
-void VaUngrabSurface( vlc_va_t *p_va, AVFrame *p_ff )
+static void Release( vlc_va_t *p_external, AVFrame *p_ff )
{
+ vlc_va_vaapi_t *p_va = vlc_va_vaapi_Get(p_external);
+
VASurfaceID i_surface_id = (VASurfaceID)(uintptr_t)p_ff->data[3];
for( int i = 0; i < p_va->i_surface_count; i++ )
}
}
-#else
-
-vlc_va_t *VaNew( int i_codec_id )
+static void Close( vlc_va_vaapi_t *p_va )
{
- VLC_UNUSED(i_codec_id);
- return NULL;
+ if( p_va->i_surface_width || p_va->i_surface_height )
+ DestroySurfaces( p_va );
+
+ if( p_va->i_config_id )
+ vaDestroyConfig( p_va->p_display, p_va->i_config_id );
+ if( p_va->p_display )
+ vaTerminate( p_va->p_display );
+ if( p_va->p_display_x11 )
+ XCloseDisplay( p_va->p_display_x11 );
}
-void VaDelete( vlc_va_t *p_va )
+static void Delete( vlc_va_t *p_external )
{
- VLC_UNUSED(p_va);
- assert( 0 );
+ vlc_va_vaapi_t *p_va = vlc_va_vaapi_Get(p_external);
+ Close( p_va );
+ free( p_va->va.description );
+ free( p_va );
}
-void VaVersion( vlc_va_t *p_va, char *psz_version, size_t i_version )
+/* */
+vlc_va_t *vlc_va_NewVaapi( int i_codec_id )
{
- VLC_UNUSED(p_va); VLC_UNUSED(psz_version); VLC_UNUSED(i_version);
- assert(0);
-}
+ if( !XInitThreads() )
+ return NULL;
-int VaSetup( vlc_va_t *p_va, void **pp_hw_ctx, vlc_fourcc_t *pi_chroma,
- int i_width, int i_height )
-{
- VLC_UNUSED(p_va); VLC_UNUSED(pp_hw_ctx); VLC_UNUSED(pi_chroma);
- VLC_UNUSED(i_width); VLC_UNUSED(i_height);
- assert(0);
- return -1;
-}
+ vlc_va_vaapi_t *p_va = calloc( 1, sizeof(*p_va) );
+ if( !p_va )
+ return NULL;
-int VaExtract( vlc_va_t *p_va, picture_t *p_picture, AVFrame *p_ff )
-{
- VLC_UNUSED(p_va); VLC_UNUSED(p_picture); VLC_UNUSED(p_ff);
- assert(0);
- return -1;
-}
+ if( Open( p_va, i_codec_id ) )
+ {
+ free( p_va );
+ return NULL;
+ }
-int VaGrabSurface( vlc_va_t *p_va, AVFrame *p_ff )
-{
- VLC_UNUSED(p_va); VLC_UNUSED(p_ff);
- assert(0);
- return -1;
+ /* */
+ p_va->va.setup = Setup;
+ p_va->va.get = Get;
+ p_va->va.release = Release;
+ p_va->va.extract = Extract;
+ p_va->va.close = Delete;
+ return &p_va->va;
}
-
-void VaUngrabSurface( vlc_va_t *p_va, AVFrame *p_ff )
+#else
+vlc_va_t *vlc_va_NewVaapi( int i_codec_id )
{
- VLC_UNUSED(p_va); VLC_UNUSED(p_ff);
- assert(0);
+ VLC_UNUSED( i_codec_id );
+ return NULL;
}
-
#endif