summaryrefslogtreecommitdiffstats
path: root/video/decode/vaapi.c
diff options
context:
space:
mode:
Diffstat (limited to 'video/decode/vaapi.c')
-rw-r--r--video/decode/vaapi.c239
1 files changed, 239 insertions, 0 deletions
diff --git a/video/decode/vaapi.c b/video/decode/vaapi.c
new file mode 100644
index 0000000000..d1099dabf5
--- /dev/null
+++ b/video/decode/vaapi.c
@@ -0,0 +1,239 @@
+/*
+ * This file is part of mpv.
+ *
+ * mpv 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.
+ *
+ * mpv 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 mpv. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include <stddef.h>
+#include <assert.h>
+
+#include <libavcodec/avcodec.h>
+#include <libavutil/common.h>
+#include <libavutil/hwcontext.h>
+#include <libavutil/hwcontext_vaapi.h>
+
+#include "config.h"
+
+#include "lavc.h"
+#include "common/common.h"
+#include "common/av_common.h"
+#include "video/fmt-conversion.h"
+#include "video/vaapi.h"
+#include "video/mp_image_pool.h"
+#include "video/hwdec.h"
+#include "video/filter/vf.h"
+
+#define ADDITIONAL_SURFACES (HWDEC_EXTRA_SURFACES + HWDEC_DELAY_QUEUE_COUNT)
+
+struct priv {
+ struct mp_log *log;
+ struct mp_vaapi_ctx *ctx;
+ bool own_ctx;
+
+ AVBufferRef *device_ref;
+ AVBufferRef *frames_ref;
+
+ struct mp_image_pool *sw_pool;
+};
+
+
+static int init_decoder(struct lavc_ctx *ctx, int w, int h)
+{
+ struct priv *p = ctx->hwdec_priv;
+ // From avconv_vaapi.c. Disgusting, but apparently this is the best we get.
+ int required_sw_format = ctx->avctx->sw_pix_fmt == AV_PIX_FMT_YUV420P10 ?
+ AV_PIX_FMT_P010 : AV_PIX_FMT_NV12;
+
+ assert(!ctx->avctx->hw_frames_ctx);
+
+ if (p->frames_ref) {
+ AVHWFramesContext *fctx = (void *)p->frames_ref->data;
+ if (fctx->width != w || fctx->height != h ||
+ fctx->sw_format != required_sw_format)
+ {
+ av_buffer_unref(&p->frames_ref);
+ }
+ }
+
+ if (!p->frames_ref) {
+ p->frames_ref = av_hwframe_ctx_alloc(p->device_ref);
+ if (!p->frames_ref)
+ return -1;
+
+ AVHWFramesContext *fctx = (void *)p->frames_ref->data;
+
+ fctx->format = AV_PIX_FMT_VAAPI;
+ fctx->sw_format = required_sw_format;
+ fctx->width = w;
+ fctx->height = h;
+
+ fctx->initial_pool_size = hwdec_get_max_refs(ctx) + ADDITIONAL_SURFACES;
+
+ va_lock(p->ctx);
+ int res = av_hwframe_ctx_init(p->frames_ref);
+ va_unlock(p->ctx);
+
+ if (res > 0) {
+ MP_ERR(ctx, "Failed to allocate hw frames.\n");
+ av_buffer_unref(&p->frames_ref);
+ return -1;
+ }
+ }
+
+ ctx->avctx->hw_frames_ctx = av_buffer_ref(p->frames_ref);
+ if (!ctx->avctx->hw_frames_ctx)
+ return -1;
+
+ return 0;
+}
+
+static struct mp_image *update_format(struct lavc_ctx *ctx, struct mp_image *img)
+{
+ va_surface_init_subformat(img);
+ return img;
+}
+
+static void uninit(struct lavc_ctx *ctx)
+{
+ struct priv *p = ctx->hwdec_priv;
+
+ if (!p)
+ return;
+
+ av_buffer_unref(&p->frames_ref);
+ av_buffer_unref(&p->device_ref);
+
+ if (p->own_ctx)
+ va_destroy(p->ctx);
+
+ talloc_free(p);
+ ctx->hwdec_priv = NULL;
+}
+
+static int init(struct lavc_ctx *ctx, bool direct)
+{
+ struct priv *p = talloc_ptrtype(NULL, p);
+ *p = (struct priv) {
+ .log = mp_log_new(p, ctx->log, "vaapi"),
+ .sw_pool = talloc_steal(p, mp_image_pool_new(17)),
+ };
+
+ if (direct) {
+ p->ctx = hwdec_devices_get(ctx->hwdec_devs, HWDEC_VAAPI)->ctx;
+ } else {
+ p->ctx = va_create_standalone(ctx->log, false);
+ if (!p->ctx) {
+ talloc_free(p);
+ return -1;
+ }
+ p->own_ctx = true;
+ }
+
+ ctx->hwdec_priv = p;
+
+ p->device_ref = av_hwdevice_ctx_alloc(AV_HWDEVICE_TYPE_VAAPI);
+ if (!p->device_ref)
+ return -1;
+
+ AVHWDeviceContext *hwctx = (void *)p->device_ref->data;
+ AVVAAPIDeviceContext *vactx = hwctx->hwctx;
+
+ vactx->display = p->ctx->display;
+
+ // Some mpv downstream code uses this.
+ hwctx->user_opaque = p->ctx;
+
+ if (av_hwdevice_ctx_init(p->device_ref) < 0)
+ return -1;
+
+ return 0;
+}
+
+static int init_direct(struct lavc_ctx *ctx)
+{
+ return init(ctx, true);
+}
+
+static int probe(struct lavc_ctx *ctx, struct vd_lavc_hwdec *hwdec,
+ const char *codec)
+{
+ if (!hwdec_devices_load(ctx->hwdec_devs, HWDEC_VAAPI))
+ return HWDEC_ERR_NO_CTX;
+ return 0;
+}
+
+static int probe_copy(struct lavc_ctx *ctx, struct vd_lavc_hwdec *hwdec,
+ const char *codec)
+{
+ struct mp_vaapi_ctx *dummy = va_create_standalone(ctx->log, true);
+ if (!dummy)
+ return HWDEC_ERR_NO_CTX;
+ bool emulated = va_guess_if_emulated(dummy);
+ va_destroy(dummy);
+ if (emulated)
+ return HWDEC_ERR_EMULATED;
+ return 0;
+}
+
+static int init_copy(struct lavc_ctx *ctx)
+{
+ return init(ctx, false);
+}
+
+static struct mp_image *copy_image(struct lavc_ctx *ctx, struct mp_image *img)
+{
+ struct priv *p = ctx->hwdec_priv;
+
+ struct mp_image *simg = va_surface_download(img, p->sw_pool);
+ talloc_free(img);
+ return simg;
+}
+
+static void intel_shit_lock(struct lavc_ctx *ctx)
+{
+ struct priv *p = ctx->hwdec_priv;
+ va_lock(p->ctx);
+}
+
+static void intel_crap_unlock(struct lavc_ctx *ctx)
+{
+ struct priv *p = ctx->hwdec_priv;
+ va_unlock(p->ctx);
+}
+
+const struct vd_lavc_hwdec mp_vd_lavc_vaapi = {
+ .type = HWDEC_VAAPI,
+ .image_format = IMGFMT_VAAPI,
+ .volatile_context = true,
+ .probe = probe,
+ .init = init_direct,
+ .uninit = uninit,
+ .init_decoder = init_decoder,
+ .lock = intel_shit_lock,
+ .unlock = intel_crap_unlock,
+ .process_image = update_format,
+};
+
+const struct vd_lavc_hwdec mp_vd_lavc_vaapi_copy = {
+ .type = HWDEC_VAAPI_COPY,
+ .copying = true,
+ .image_format = IMGFMT_VAAPI,
+ .volatile_context = true,
+ .probe = probe_copy,
+ .init = init_copy,
+ .uninit = uninit,
+ .init_decoder = init_decoder,
+ .process_image = copy_image,
+ .delay_queue = HWDEC_DELAY_QUEUE_COUNT,
+};