mpv/video/out/vo_opengl_cb.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <stdbool.h>
#include <limits.h>
#include <pthread.h>
#include <assert.h>

#include "config.h"

#include "talloc.h"
#include "common/common.h"
#include "misc/bstr.h"
#include "common/msg.h"
#include "options/m_config.h"
#include "options/options.h"
#include "aspect.h"
#include "vo.h"
#include "video/vfcap.h"
#include "video/mp_image.h"
#include "sub/osd.h"

#include "common/global.h"
#include "player/client.h"

#include "gl_common.h"
#include "gl_video.h"
#include "gl_hwdec.h"

#include "video/decode/lavc.h" // HWDEC_* values

#include "libmpv/opengl_cb.h"

/*
 * mpv_opengl_cb_context is created by the host application - the host application
 * can access it any time, even if the VO is destroyed (or not created yet).
 * The OpenGL object allows initializing the renderer etc. The VO object is only
 * here to transfer the video frames somehow.
 */

struct vo_priv {
    struct vo *vo;

    struct mpv_opengl_cb_context *ctx;
};

struct mpv_opengl_cb_context {
    struct mp_log *log;

    pthread_mutex_t lock;

    // --- Protected by lock
    mpv_opengl_cb_update_fn update_cb;
    void *update_cb_ctx;
    struct mp_image *waiting_frame;
    struct mp_image *next_frame;
    struct mp_image_params img_params;
    bool reconfigured;
    struct mp_rect wnd;
    bool flip;
    bool force_update;
    bool imgfmt_supported[IMGFMT_END - IMGFMT_START];
    struct mp_vo_opts vo_opts;

    // --- All of these can only be accessed from the thread where the host
    //     application's OpenGL context is current - i.e. only while the
    //     host application is calling certain mpv_opengl_cb_* APIs.
    GL *gl;
    struct gl_video *renderer;
    struct gl_hwdec *hwdec;

    // --- Immutable or semi-threadsafe.

    struct osd_state *osd;
    struct mp_hwdec_info hwdec_info;
    const char *hwapi;

    struct vo *active;
};

struct mpv_opengl_cb_context *mp_opengl_create(struct mpv_global *g,
                                               struct osd_state *osd)
{
    mpv_opengl_cb_context *ctx = talloc_zero(NULL, mpv_opengl_cb_context);
    ctx->log = mp_log_new(ctx, g->log, "opengl-cb");
    pthread_mutex_init(&ctx->lock, NULL);

    ctx->gl = talloc_zero(ctx, GL);

    ctx->osd = osd;

    switch (g->opts->hwdec_api) {
    case HWDEC_AUTO:    ctx->hwapi = "auto"; break;
    case HWDEC_VDPAU:   ctx->hwapi = "vdpau"; break;
    case HWDEC_VDA:     ctx->hwapi = "vda"; break;
    case HWDEC_VAAPI:   ctx->hwapi = "vaapi"; break;
    default:            ctx->hwapi = "";
    }

    return ctx;
}

// To be called from VO thread, with p->ctx->lock held.
static void copy_vo_opts(struct vo *vo)
{
    struct vo_priv *p = vo->priv;

    // We're being lazy: none of the options we need use dynamic data, so
    // copy the struct with an assignment.
    // Just remove all the dynamic data to avoid confusion.
    struct mp_vo_opts opts = *vo->opts;
    opts.video_driver_list = opts.vo_defs = NULL;
    opts.winname = NULL;
    opts.sws_opts = NULL;
    p->ctx->vo_opts = opts;
}

void mpv_opengl_cb_set_update_callback(struct mpv_opengl_cb_context *ctx,
                                      mpv_opengl_cb_update_fn callback,
                                      void *callback_ctx)
{
    pthread_mutex_lock(&ctx->lock);
    ctx->update_cb = callback;
    ctx->update_cb_ctx = callback_ctx;
    pthread_mutex_unlock(&ctx->lock);
}

int mpv_opengl_cb_init_gl(struct mpv_opengl_cb_context *ctx, const char *exts,
                          mpv_opengl_cb_get_proc_address_fn get_proc_address,
                          void *get_proc_address_ctx)
{
    if (ctx->renderer)
        return MPV_ERROR_INVALID_PARAMETER;

    mpgl_load_functions2(ctx->gl, get_proc_address, get_proc_address_ctx,
                         exts, ctx->log);
    int caps = MPGL_CAP_GL21;
    if ((ctx->gl->mpgl_caps & caps) != caps) {
        MP_FATAL(ctx, "Missing OpenGL features.\n");
        return MPV_ERROR_UNSUPPORTED;
    }
    ctx->renderer = gl_video_init(ctx->gl, ctx->log, ctx->osd);
    ctx->hwdec = gl_hwdec_load_api(ctx->log, ctx->gl, ctx->hwapi, &ctx->hwdec_info);
    gl_video_set_hwdec(ctx->renderer, ctx->hwdec);

    pthread_mutex_lock(&ctx->lock);
    for (int n = IMGFMT_START; n < IMGFMT_END; n++) {
        ctx->imgfmt_supported[n - IMGFMT_START] =
            gl_video_check_format(ctx->renderer, n);
    }
    pthread_mutex_unlock(&ctx->lock);

    gl_video_unset_gl_state(ctx->renderer);
    return 0;
}

int mpv_opengl_cb_uninit_gl(struct mpv_opengl_cb_context *ctx)
{
    gl_video_uninit(ctx->renderer);
    ctx->renderer = NULL;
    gl_hwdec_uninit(ctx->hwdec);
    ctx->hwdec = NULL;
    talloc_free(ctx->gl);
    ctx->gl = NULL;
    return 0;
}

int mpv_opengl_cb_render(struct mpv_opengl_cb_context *ctx, int fbo, int vp[4])
{
    assert(ctx->renderer);

    gl_video_set_gl_state(ctx->renderer);

    pthread_mutex_lock(&ctx->lock);

    struct vo *vo = ctx->active;

    ctx->force_update |= ctx->reconfigured;

    int h = vp[3];
    bool flip = h < 0 && h > INT_MIN;
    if (flip)
        h = -h;
    struct mp_rect wnd = {vp[0], vp[1], vp[0] + vp[2], vp[1] + h};
    if (wnd.x0 != ctx->wnd.x0 || wnd.y0 != ctx->wnd.y0 ||
        wnd.x1 != ctx->wnd.x1 || wnd.y1 != ctx->wnd.y1 ||
        ctx->flip != flip)
        ctx->force_update = true;

    if (ctx->force_update && vo) {
        ctx->force_update = false;
        ctx->wnd = wnd;

        struct mp_rect src, dst;
        struct mp_osd_res osd;
        mp_get_src_dst_rects(ctx->log, &ctx->vo_opts, vo->driver->caps,
                             &ctx->img_params, wnd.x1 - wnd.x0, wnd.y1 - wnd.y0,
                             1.0, &src, &dst, &osd);

        gl_video_resize(ctx->renderer, &wnd, &src, &dst, &osd, !ctx->flip);
    }

    if (ctx->reconfigured) {
        ctx->reconfigured = false;
        gl_video_config(ctx->renderer, &ctx->img_params);
        struct gl_video_opts opts = gl_video_opts_def;
        opts.background.a = 0; // transparent
        gl_video_set_options(ctx->renderer, &opts);
    }

    struct mp_image *mpi = ctx->next_frame;
    ctx->next_frame = NULL;

    pthread_mutex_unlock(&ctx->lock);

    if (mpi)
        gl_video_upload_image(ctx->renderer, mpi);

    gl_video_render_frame(ctx->renderer, fbo);

    gl_video_unset_gl_state(ctx->renderer);

    return 0;
}

static void draw_image(struct vo *vo, mp_image_t *mpi)
{
    struct vo_priv *p = vo->priv;
    if (p->ctx) {
        pthread_mutex_lock(&p->ctx->lock);
        mp_image_setrefp(&p->ctx->waiting_frame, mpi);
        pthread_mutex_unlock(&p->ctx->lock);
    }
    talloc_free(mpi);
}

static void flip_page(struct vo *vo)
{
    struct vo_priv *p = vo->priv;
    if (p->ctx) {
        pthread_mutex_lock(&p->ctx->lock);
        mp_image_unrefp(&p->ctx->next_frame);
        p->ctx->next_frame = p->ctx->waiting_frame;
        p->ctx->waiting_frame = NULL;
        if (p->ctx->update_cb)
            p->ctx->update_cb(p->ctx->update_cb_ctx);
        pthread_mutex_unlock(&p->ctx->lock);
    }
}

static int query_format(struct vo *vo, uint32_t format)
{
    struct vo_priv *p = vo->priv;

    bool ok = false;
    if (p->ctx) {
        pthread_mutex_lock(&p->ctx->lock);
        if (format >= IMGFMT_START && format < IMGFMT_END)
            ok = p->ctx->imgfmt_supported[format - IMGFMT_START];
        pthread_mutex_unlock(&p->ctx->lock);
    }
    return ok ? VFCAP_CSP_SUPPORTED | VFCAP_CSP_SUPPORTED_BY_HW : 0;
}

static int reconfig(struct vo *vo, struct mp_image_params *params, int flags)
{
    struct vo_priv *p = vo->priv;

    if (p->ctx) {
        pthread_mutex_lock(&p->ctx->lock);
        mp_image_unrefp(&p->ctx->next_frame);
        p->ctx->img_params = *params;
        p->ctx->reconfigured = true;
        pthread_mutex_unlock(&p->ctx->lock);
    } else {
        return -1;
    }

    return 0;
}

static int control(struct vo *vo, uint32_t request, void *data)
{
    struct vo_priv *p = vo->priv;

    switch (request) {
    case VOCTRL_SET_LIBMPV_OPENGL_CB_CONTEXT: {
        if (p->ctx)
            return VO_FALSE;
        struct mpv_opengl_cb_context *nctx = data;
        if (nctx) {
            pthread_mutex_lock(&nctx->lock);
            if (nctx->active) {
                MP_FATAL(vo, "There is already a VO using the OpenGL context.\n");
            } else {
                nctx->active = vo;
                nctx->reconfigured = true;
                p->ctx = nctx;
                assert(vo->osd == p->ctx->osd);
                copy_vo_opts(vo);
            }
            pthread_mutex_unlock(&nctx->lock);
        }
        return VO_TRUE;
    }
    case VOCTRL_GET_PANSCAN:
        return VO_TRUE;
    case VOCTRL_SET_PANSCAN:
    case VOCTRL_REDRAW_FRAME:
        pthread_mutex_lock(&p->ctx->lock);
        copy_vo_opts(vo);
        p->ctx->force_update = true;
        if (p->ctx->update_cb)
            p->ctx->update_cb(p->ctx->update_cb_ctx);
        pthread_mutex_unlock(&p->ctx->lock);
        return VO_TRUE;
    case VOCTRL_GET_HWDEC_INFO: {
        // Warning: in theory, the API user could destroy the OpenGL context
        //          while the decoder uses the hwdec thing, and bad things would
        //          happen. Currently, the API user is told not to do this.
        struct mp_hwdec_info **arg = data;
        *arg = p->ctx ? &p->ctx->hwdec_info : NULL;
        return true;
    }
    }

    return VO_NOTIMPL;
}

static void uninit(struct vo *vo)
{
    struct vo_priv *p = vo->priv;

    if (p->ctx) {
        pthread_mutex_lock(&p->ctx->lock);
        mp_image_unrefp(&p->ctx->next_frame);
        mp_image_unrefp(&p->ctx->waiting_frame);
        p->ctx->img_params = (struct mp_image_params){0};
        p->ctx->reconfigured = true;
        p->ctx->active = NULL;
        pthread_mutex_unlock(&p->ctx->lock);
    }
}

static int preinit(struct vo *vo)
{
    struct vo_priv *p = vo->priv;
    p->vo = vo;
    return 0;
}

#define OPT_BASE_STRUCT struct gl_priv
static const struct m_option options[] = {
    {0},
};

const struct vo_driver video_out_opengl_cb = {
    .description = "OpenGL Callbacks for libmpv",
    .name = "opengl-cb",
    .caps = VO_CAP_ROTATE90,
    .preinit = preinit,
    .query_format = query_format,
    .reconfig = reconfig,
    .control = control,
    .draw_image = draw_image,
    .flip_page = flip_page,
    .uninit = uninit,
    .priv_size = sizeof(struct vo_priv),
    .options = options,
};