vdpau: VDPAU hardware decoding with avcodec

This requires bleeding-edge libavcodec.

configure.ac

@@ -2970,6 +2970,44 @@ AS_IF([test "${enable_xcb}" != "no"], [
 ])
 AM_CONDITIONAL([HAVE_XCB], [test "${have_xcb}" = "yes"])
 
+
+dnl
+dnl VDPAU needs X11 and avcodec
+dnl
+AC_ARG_ENABLE(vdpau,
+  [AS_HELP_STRING([--enable-vdpau],
+    [VDPAU hardware decoder support (default auto)])])
+have_vdpau="no"
+AS_IF([test "${enable_vdpau}" != "no"], [
+  PKG_CHECK_MODULES([VDPAU], [vdpau], [
+    have_vdpau="yes"
+    AC_MSG_NOTICE([VDPAU acceleration activated])
+  ], [
+    AS_IF([test -n "${enable_vdpau}"], [
+      AC_MSG_ERROR([${VDPAU_PKG_ERRORS}.])
+    ])
+  ])
+  AS_IF([test "${no_x}" = "yes"], [
+    AC_MSG_ERROR([VDPAU requires Xlib (X11).])
+  ])
+])
+dnl AM_CONDITIONAL([HAVE_VDPAU], [test "${have_vdpau}" = "yes"])
+
+have_avcodec_vdpau="no"
+AS_IF([test "${have_vdpau}" = "yes"], [
+  PKG_CHECK_EXISTS([libavutil >= 0.52.4 libavcodec >= 54.36.0], [
+    have_avcodec_vdpau="yes"
+  ], [
+    AS_IF([test -n "${enable_vdpau}"], [
+      AC_MSG_ERROR([libavutil >= 0.52.4 and libavcodec >= 54.36.0 are required for VDPAU decoding.])
+    ], [
+      AC_MSG_WARN([libavutil >= 0.52.4 and libavcodec >= 54.36.0 are required for VDPAU decoding.])
+    ])
+  ])
+])
+AM_CONDITIONAL([HAVE_AVCODEC_VDPAU], [test "${have_avcodec_vdpau}" = "yes"])
+
+
 dnl
 dnl  SDL module
 dnl

modules/LIST

@@ -351,6 +351,7 @@ $Id$
  * vc1: VC-1 Video demuxer
  * vcd: input module for accessing Video CDs
  * vcdx: input module for accessing Video CDs with navigation & stills
+ * vdpau: VDPAU hardware-accelerated decoding
  * vdummy: dummy video display
  * visual: visualisation system
  * vmem: memory video driver

modules/codec/Modules.am

@@ -153,6 +153,16 @@ if HAVE_AVCODEC_VDA
 libvlc_LTLIBRARIES += libvda_plugin.la
 endif
 
+libvdpau_plugin_la_SOURCES = avcodec/vdpau.c
+libvdpau_plugin_la_CFLAGS = $(AM_CFLAGS) $(VDPAU_CFLAGS) \
+	$(X_CFLAGS) $(CFLAGS_avcodec)
+libvdpau_plugin_la_LIBADD = $(AM_LIBADD) $(VDPAU_LIBS) \
+	 $(X_LIBS) $(X_PRE_LIBS) -lX11 $(LIBS_avcodec)
+libvdpau_plugin_la_LDFLAGS = $(AM_LDFLAGS) $(LDFLAGS_avcodec)
+if HAVE_AVCODEC_VDPAU
+libvlc_LTLIBRARIES += libvdpau_plugin.la
+endif
+
 ### XWD ###
 libxwd_plugin_la_SOURCES = xwd.c
 libxwd_plugin_la_CFLAGS = $(AM_CFLAGS) $(XPROTO_CFLAGS)

modules/codec/avcodec/vdpau.c

+/*****************************************************************************
+ * vdpau.c: VDPAU decoder for libav
+ *****************************************************************************
+ * Copyright (C) 2012-2013 Rémi Denis-Courmont
+ *
+ * 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 <vdpau/vdpau.h>
+#include <libavutil/mem.h>
+#include <libavcodec/avcodec.h>
+#include <libavcodec/vdpau.h>
+#include <vlc_common.h>
+#include <vlc_plugin.h>
+#include <vlc_fourcc.h>
+#include <vlc_picture.h>
+
+#include <X11/Xlib.h>
+#include <vdpau/vdpau_x11.h>
+#include <vlc_xlib.h>
+
+#include "avcodec.h"
+#include "va.h"
+
+static int Open (vlc_va_t *, int, const es_format_t *);
+static void Close (vlc_va_t *);
+
+vlc_module_begin ()
+    set_description (N_("Video Decode and Presentation API for Unix (VDPAU)"))
+    set_capability ("hw decoder", 100)
+    set_category (CAT_INPUT)
+    set_subcategory (SUBCAT_INPUT_VCODEC)
+    set_callbacks (Open, Close)
+vlc_module_end ()
+
+#define MAX_SURFACES (16+1)
+
+struct vlc_va_sys_t
+{
+    VdpDevice device;
+    VdpDecoderProfile profile;
+    AVVDPAUContext context;
+    VdpVideoSurface surfaces[MAX_SURFACES];
+    uint32_t available;
+    uint16_t width;
+    uint16_t height;
+    void *display;
+
+    VdpGetErrorString *GetErrorString;
+    VdpGetInformationString *GetInformationString;
+    VdpDeviceDestroy *DeviceDestroy;
+    VdpVideoSurfaceCreate *VideoSurfaceCreate;
+    VdpVideoSurfaceDestroy *VideoSurfaceDestroy;
+    VdpVideoSurfaceGetBitsYCbCr *VideoSurfaceGetBitsYCbCr;
+    VdpDecoderCreate *DecoderCreate;
+    VdpDecoderDestroy *DecoderDestroy;
+    VdpDecoderRender *DecoderRender;
+};
+
+static int Lock (vlc_va_t *va, AVFrame *ff)
+{
+    vlc_va_sys_t *sys = va->sys;
+
+    for (unsigned i = 0; i < AV_NUM_DATA_POINTERS; i++)
+    {
+        ff->data[i] = NULL;
+        ff->linesize[i] = 0;
+    }
+
+    if (!sys->available)
+    {
+        msg_Err (va, "no free surfaces");
+        return VLC_EGENERIC;
+    }
+
+    unsigned idx = __builtin_ctz (sys->available);
+    sys->available &= ~(1 << idx);
+
+    VdpVideoSurface *surface = sys->surfaces + idx;
+    assert (*surface != VDP_INVALID_HANDLE);
+    ff->data[0] = (void *)surface; /* must be non-NULL */
+    ff->data[3] = (void *)(uintptr_t)*surface;
+    ff->opaque = surface;
+    return VLC_SUCCESS;
+}
+
+static void Unlock (vlc_va_t *va, AVFrame *ff)
+{
+    vlc_va_sys_t *sys = va->sys;
+    VdpVideoSurface *surface = ff->opaque;
+    unsigned idx = surface - sys->surfaces;
+
+    assert (idx < MAX_SURFACES);
+    sys->available |= (1 << idx);
+
+    ff->data[0] = ff->data[3] = NULL;
+    ff->opaque = NULL;
+}
+
+static int Copy (vlc_va_t *va, picture_t *pic, AVFrame *ff)
+{
+    vlc_va_sys_t *sys = va->sys;
+    VdpVideoSurface *surface = ff->opaque;
+    void *planes[3];
+    uint32_t pitches[3];
+    VdpStatus err;
+
+    for (unsigned i = 0; i < 3; i++)
+    {
+         planes[i] = pic->p[i].p_pixels;
+         pitches[i] = pic->p[i].i_pitch;
+    }
+
+    err = sys->VideoSurfaceGetBitsYCbCr (*surface, VDP_YCBCR_FORMAT_YV12,
+                                         planes, pitches);
+    if (err != VDP_STATUS_OK)
+    {
+        msg_Err (va, "surface copy failure: %s", sys->GetErrorString (err));
+        return VLC_EGENERIC;
+    }
+    return VLC_SUCCESS;
+}
+
+static int Init (vlc_va_t *va, void **ctxp, vlc_fourcc_t *chromap,
+                 int width, int height)
+{
+    vlc_va_sys_t *sys = va->sys;
+    VdpStatus err;
+
+    width = (width + 1) & ~1;
+    height = (height + 3) & ~3;
+
+    unsigned surfaces = 2;
+    if (sys->profile == VDP_DECODER_PROFILE_H264_HIGH)
+        surfaces = 16;
+
+    err = sys->DecoderCreate (sys->device, sys->profile, width, height,
+                              surfaces, &sys->context.decoder);
+    if (err != VDP_STATUS_OK)
+    {
+        msg_Err (va, "decoder creation failure: %s",
+                 sys->GetErrorString (err));
+        sys->context.decoder = VDP_INVALID_HANDLE;
+        return VLC_EGENERIC;
+    }
+
+    assert (width > 0 && height > 0);
+    surfaces++;
+    assert (surfaces <= MAX_SURFACES);
+    sys->available = 0;
+
+    /* TODO: select better chromas when appropriate */
+    for (unsigned i = 0; i < surfaces; i++)
+    {
+        err = sys->VideoSurfaceCreate (sys->device, VDP_CHROMA_TYPE_420, width,
+                                       height, sys->surfaces + i);
+        if (err != VDP_STATUS_OK)
+        {
+             msg_Err (va, "surface creation failure: %s",
+                      sys->GetErrorString (err));
+             sys->surfaces[i] = VDP_INVALID_HANDLE;
+             break;
+        }
+        sys->available |= (1 << i);
+    }
+
+    *ctxp = &sys->context;
+    *chromap = VLC_CODEC_YV12;
+    return VLC_SUCCESS;
+}
+
+static void Deinit (vlc_va_t *va)
+{
+    vlc_va_sys_t *sys = va->sys;
+
+    for (unsigned i = 0; i < MAX_SURFACES; i++)
+    {
+        VdpVideoSurface *surface = sys->surfaces + i;
+
+        if (*surface != VDP_INVALID_HANDLE)
+        {
+            sys->VideoSurfaceDestroy (*surface);
+            *surface = VDP_INVALID_HANDLE;
+        }
+    }
+
+    assert (sys->context.decoder != VDP_INVALID_HANDLE);
+    sys->DecoderDestroy (sys->context.decoder);
+    av_freep (&sys->context.bitstream_buffers);
+}
+
+static int Setup (vlc_va_t *va, void **ctxp, vlc_fourcc_t *chromap,
+                 int width, int height)
+{
+    vlc_va_sys_t *sys = va->sys;
+
+    if (sys->context.decoder != VDP_INVALID_HANDLE)
+    {
+        if (sys->width == width && sys->height == height)
+            return VLC_SUCCESS;
+        Deinit (va);
+        sys->context.decoder = VDP_INVALID_HANDLE;
+    }
+
+    sys->width = width;
+    sys->height = height;
+    return Init (va, ctxp, chromap, width, height);
+}
+
+static int vdp_device_Create (vlc_object_t *obj, void **sysp, VdpDevice *devp,
+                              VdpGetProcAddress **gpap)
+{
+    VdpStatus err;
+
+    if (!vlc_xlib_init (obj))
+    {
+        msg_Err (obj, "Xlib is required for VDPAU");
+        return VLC_EGENERIC;
+    }
+
+    Display *x11 = XOpenDisplay (NULL);
+    if (x11 == NULL)
+    {
+        msg_Err (obj, "windowing system failure failure");
+        return VLC_EGENERIC;
+    }
+
+    err = vdp_device_create_x11 (x11, XDefaultScreen (x11), devp, gpap);
+    if (err)
+    {
+        msg_Err (obj, "device creation failure: error %d", (int)err);
+        XCloseDisplay (x11);
+        return VLC_EGENERIC;
+    }
+    *sysp = x11;
+    return VLC_SUCCESS;
+}
+
+static int Open (vlc_va_t *va, int codec, const es_format_t *fmt)
+{
+    VdpStatus err;
+    int profile;
+
+    switch (codec)
+    {
+        case AV_CODEC_ID_MPEG1VIDEO:
+            profile = VDP_DECODER_PROFILE_MPEG1;
+            break;
+        case AV_CODEC_ID_MPEG2VIDEO:
+            profile = VDP_DECODER_PROFILE_MPEG2_MAIN;
+            break;
+        case AV_CODEC_ID_H263:
+        case AV_CODEC_ID_MPEG4:
+            profile = VDP_DECODER_PROFILE_MPEG4_PART2_ASP;
+            break;
+        case AV_CODEC_ID_H264:
+            profile = VDP_DECODER_PROFILE_H264_HIGH;
+            break;
+        case AV_CODEC_ID_WMV3:
+        case AV_CODEC_ID_VC1:
+            profile = VDP_DECODER_PROFILE_VC1_ADVANCED;
+            break;
+        default:
+            msg_Err (va, "unknown codec (%d)", codec);
+            return VLC_EGENERIC;
+    }
+
+    vlc_va_sys_t *sys = malloc (sizeof (*sys));
+    if (unlikely(sys == NULL))
+       return VLC_ENOMEM;
+
+    VdpDevice device;
+    VdpGetProcAddress *GetProcAddress;
+    if (vdp_device_Create (VLC_OBJECT(va), &sys->display, &device,
+                           &GetProcAddress))
+    {
+        free (sys);
+        return VLC_EGENERIC;
+    }
+
+    sys->device = device;
+    sys->profile = profile;
+    memset (&sys->context, 0, sizeof (sys->context));
+    sys->context.decoder = VDP_INVALID_HANDLE;
+    for (unsigned i = 0; i < MAX_SURFACES; i++)
+        sys->surfaces[i] = VDP_INVALID_HANDLE;
+
+#define PROC(id,name) \
+    do { \
+        void *ptr; \
+        err = GetProcAddress (sys->device, VDP_FUNC_ID_##id, &ptr); \
+        if (unlikely(err)) \
+            abort (); \
+        sys->name = ptr; \
+    } while (0)
+
+    /* We are really screwed if any function fails. We cannot even delete the
+     * already allocated device. */
+    PROC(GET_ERROR_STRING, GetErrorString);
+    PROC(GET_INFORMATION_STRING, GetInformationString);
+    PROC(DEVICE_DESTROY, DeviceDestroy);
+    PROC(VIDEO_SURFACE_CREATE, VideoSurfaceCreate);
+    PROC(VIDEO_SURFACE_DESTROY, VideoSurfaceDestroy);
+    PROC(VIDEO_SURFACE_GET_BITS_Y_CB_CR, VideoSurfaceGetBitsYCbCr);
+    PROC(DECODER_CREATE, DecoderCreate);
+    PROC(DECODER_DESTROY, DecoderDestroy);
+    PROC(DECODER_RENDER, DecoderRender);
+    sys->context.render = sys->DecoderRender;
+
+    /* TODO: Query cap */
+
+    const char *infos;
+    if (sys->GetInformationString (&infos) != VDP_STATUS_OK)
+        infos = "VDPAU";
+
+    va->sys = sys;
+    va->description = (char *)infos;
+    va->pix_fmt = AV_PIX_FMT_VDPAU;
+    va->setup = Setup;
+    va->get = Lock;
+    va->release = Unlock;
+    va->extract = Copy;
+    (void) fmt;
+    return VLC_SUCCESS;
+
+/*error:
+    XCloseDisplay (sys->display);
+    free (sys);
+    return VLC_EGENERIC;*/
+}
+
+static void Close (vlc_va_t *va)
+{
+    vlc_va_sys_t *sys = va->sys;
+
+    if (sys->context.decoder != VDP_INVALID_HANDLE)
+        Deinit (va);
+    sys->DeviceDestroy (sys->device);
+    XCloseDisplay (sys->display);
+    free (sys);
+}

po/POTFILES.in

@@ -352,6 +352,7 @@ modules/codec/avcodec/encoder.c
 modules/codec/avcodec/fourcc.c
 modules/codec/avcodec/vaapi.c
 modules/codec/avcodec/vda.c
+modules/codec/avcodec/vdpau.c
 modules/codec/avcodec/video.c
 modules/codec/cc.c
 modules/codec/cc.h