mpv/video/out/hwdec/hwdec_drmprime_overlay.c

335 lines
11 KiB
C
Raw Normal View History

/*
* 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 <assert.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <stdbool.h>
#include <libavutil/hwcontext.h>
#include <libavutil/hwcontext_drm.h>
#include "video/hwdec.h"
#include "common/msg.h"
#include "options/m_config.h"
#include "libmpv/render_gl.h"
#include "video/out/drm_atomic.h"
#include "video/out/drm_common.h"
#include "video/out/drm_prime.h"
#include "video/out/gpu/hwdec.h"
#include "video/mp_image.h"
extern const struct m_sub_options drm_conf;
struct drm_frame {
struct drm_prime_framebuffer fb;
struct mp_image *image; // associated mpv image
};
struct priv {
struct mp_log *log;
struct mp_hwdec_ctx hwctx;
struct mp_image_params params;
struct drm_atomic_context *ctx;
struct drm_frame current_frame, last_frame, old_frame;
struct mp_rect src, dst;
int display_w, display_h;
2020-03-04 11:48:40 +00:00
struct drm_prime_handle_refs handle_refs;
};
static void set_current_frame(struct ra_hwdec *hw, struct drm_frame *frame)
{
struct priv *p = hw->priv;
// frame will be on screen after next vsync
// current_frame is currently the displayed frame and will be replaced
// by frame after next vsync.
// We used old frame as triple buffering to make sure that the drm framebuffer
// is not being displayed when we release it.
if (p->ctx) {
2020-03-04 11:48:40 +00:00
drm_prime_destroy_framebuffer(p->log, p->ctx->fd, &p->old_frame.fb, &p->handle_refs);
}
mp_image_setrefp(&p->old_frame.image, p->last_frame.image);
p->old_frame.fb = p->last_frame.fb;
mp_image_setrefp(&p->last_frame.image, p->current_frame.image);
p->last_frame.fb = p->current_frame.fb;
if (frame) {
p->current_frame.fb = frame->fb;
mp_image_setrefp(&p->current_frame.image, frame->image);
} else {
memset(&p->current_frame.fb, 0, sizeof(p->current_frame.fb));
mp_image_setrefp(&p->current_frame.image, NULL);
}
}
static void scale_dst_rect(struct ra_hwdec *hw, int source_w, int source_h ,struct mp_rect *src, struct mp_rect *dst)
{
struct priv *p = hw->priv;
// drm can allow to have a layer that has a different size from framebuffer
// we scale here the destination size to video mode
double hratio = p->display_w / (double)source_w;
double vratio = p->display_h / (double)source_h;
double ratio = hratio <= vratio ? hratio : vratio;
dst->x0 = src->x0 * ratio;
dst->x1 = src->x1 * ratio;
dst->y0 = src->y0 * ratio;
dst->y1 = src->y1 * ratio;
int offset_x = (p->display_w - ratio * source_w) / 2;
int offset_y = (p->display_h - ratio * source_h) / 2;
dst->x0 += offset_x;
dst->x1 += offset_x;
dst->y0 += offset_y;
dst->y1 += offset_y;
}
static void disable_video_plane(struct ra_hwdec *hw)
{
struct priv *p = hw->priv;
if (!p->ctx)
return;
if (!p->ctx->drmprime_video_plane)
return;
// Disabling the drmprime video plane is needed on some devices when using
// the primary plane for video. Primary buffer can't be active with no
// framebuffer associated. So we need this function to commit it right away
// as mpv will free all framebuffers on playback end.
drmModeAtomicReqPtr request = drmModeAtomicAlloc();
if (request) {
drm_object_set_property(request, p->ctx->drmprime_video_plane, "FB_ID", 0);
drm_object_set_property(request, p->ctx->drmprime_video_plane, "CRTC_ID", 0);
int ret = drmModeAtomicCommit(p->ctx->fd, request,
0, NULL);
if (ret)
MP_ERR(hw, "Failed to commit disable plane request (code %d)", ret);
drmModeAtomicFree(request);
}
}
static int overlay_frame(struct ra_hwdec *hw, struct mp_image *hw_image,
struct mp_rect *src, struct mp_rect *dst, bool newframe)
{
struct priv *p = hw->priv;
AVDRMFrameDescriptor *desc = NULL;
drmModeAtomicReq *request = NULL;
struct drm_frame next_frame = {0};
int ret;
struct ra *ra = hw->ra_ctx->ra;
// grab atomic request from native resources
if (p->ctx) {
struct mpv_opengl_drm_params_v2 *drm_params;
drm_params = (mpv_opengl_drm_params_v2 *)ra_get_native_resource(ra, "drm_params_v2");
if (!drm_params) {
MP_ERR(hw, "Failed to retrieve drm params from native resources\n");
return -1;
}
if (drm_params->atomic_request_ptr) {
request = *drm_params->atomic_request_ptr;
} else {
MP_ERR(hw, "drm params pointer to atomic request is invalid\n");
return -1;
}
}
if (hw_image) {
// grab draw plane windowing info to eventually upscale the overlay
// as egl windows could be upscaled to draw plane.
struct mpv_opengl_drm_draw_surface_size *draw_surface_size = ra_get_native_resource(ra, "drm_draw_surface_size");
if (draw_surface_size) {
scale_dst_rect(hw, draw_surface_size->width, draw_surface_size->height, dst, &p->dst);
} else {
p->dst = *dst;
}
p->src = *src;
next_frame.image = hw_image;
desc = (AVDRMFrameDescriptor *)hw_image->planes[0];
if (desc) {
int srcw = p->src.x1 - p->src.x0;
int srch = p->src.y1 - p->src.y0;
int dstw = MP_ALIGN_UP(p->dst.x1 - p->dst.x0, 2);
int dsth = MP_ALIGN_UP(p->dst.y1 - p->dst.y0, 2);
2020-03-04 11:48:40 +00:00
if (drm_prime_create_framebuffer(p->log, p->ctx->fd, desc, srcw, srch, &next_frame.fb, &p->handle_refs)) {
ret = -1;
goto fail;
}
if (request) {
drm_object_set_property(request, p->ctx->drmprime_video_plane, "FB_ID", next_frame.fb.fb_id);
drm_object_set_property(request, p->ctx->drmprime_video_plane, "CRTC_ID", p->ctx->crtc->id);
drm_object_set_property(request, p->ctx->drmprime_video_plane, "SRC_X", p->src.x0 << 16);
drm_object_set_property(request, p->ctx->drmprime_video_plane, "SRC_Y", p->src.y0 << 16);
drm_object_set_property(request, p->ctx->drmprime_video_plane, "SRC_W", srcw << 16);
drm_object_set_property(request, p->ctx->drmprime_video_plane, "SRC_H", srch << 16);
drm_object_set_property(request, p->ctx->drmprime_video_plane, "CRTC_X", MP_ALIGN_DOWN(p->dst.x0, 2));
drm_object_set_property(request, p->ctx->drmprime_video_plane, "CRTC_Y", MP_ALIGN_DOWN(p->dst.y0, 2));
drm_object_set_property(request, p->ctx->drmprime_video_plane, "CRTC_W", dstw);
drm_object_set_property(request, p->ctx->drmprime_video_plane, "CRTC_H", dsth);
drm_object_set_property(request, p->ctx->drmprime_video_plane, "ZPOS", 0);
} else {
ret = drmModeSetPlane(p->ctx->fd, p->ctx->drmprime_video_plane->id, p->ctx->crtc->id, next_frame.fb.fb_id, 0,
MP_ALIGN_DOWN(p->dst.x0, 2), MP_ALIGN_DOWN(p->dst.y0, 2), dstw, dsth,
p->src.x0 << 16, p->src.y0 << 16 , srcw << 16, srch << 16);
if (ret < 0) {
MP_ERR(hw, "Failed to set the drmprime video plane %d (buffer %d).\n",
p->ctx->drmprime_video_plane->id, next_frame.fb.fb_id);
goto fail;
}
}
}
} else {
disable_video_plane(hw);
while (p->old_frame.fb.fb_id)
set_current_frame(hw, NULL);
}
set_current_frame(hw, &next_frame);
return 0;
fail:
2020-03-04 11:48:40 +00:00
drm_prime_destroy_framebuffer(p->log, p->ctx->fd, &next_frame.fb, &p->handle_refs);
return ret;
}
static void uninit(struct ra_hwdec *hw)
{
struct priv *p = hw->priv;
disable_video_plane(hw);
set_current_frame(hw, NULL);
hwdec_devices_remove(hw->devs, &p->hwctx);
av_buffer_unref(&p->hwctx.av_device_ref);
if (p->ctx) {
drm_atomic_destroy_context(p->ctx);
p->ctx = NULL;
}
}
static int init(struct ra_hwdec *hw)
{
struct priv *p = hw->priv;
int draw_plane, drmprime_video_plane;
p->log = hw->log;
void *tmp = talloc_new(NULL);
struct drm_opts *opts = mp_get_config_group(tmp, hw->global, &drm_conf);
draw_plane = opts->draw_plane;
drmprime_video_plane = opts->drmprime_video_plane;
talloc_free(tmp);
struct mpv_opengl_drm_params_v2 *drm_params;
drm_params = ra_get_native_resource(hw->ra_ctx->ra, "drm_params_v2");
if (drm_params) {
p->ctx = drm_atomic_create_context(p->log, drm_params->fd, drm_params->crtc_id,
drm_params->connector_id, draw_plane, drmprime_video_plane);
if (!p->ctx) {
mp_err(p->log, "Failed to retrieve DRM atomic context.\n");
goto err;
}
if (!p->ctx->drmprime_video_plane) {
mp_warn(p->log, "No drmprime video plane. You might need to specify it manually using --drm-drmprime-video-plane\n");
goto err;
}
} else {
mp_verbose(p->log, "Failed to retrieve DRM fd from native display.\n");
goto err;
}
drmModeCrtcPtr crtc;
crtc = drmModeGetCrtc(p->ctx->fd, p->ctx->crtc->id);
if (crtc) {
p->display_w = crtc->mode.hdisplay;
p->display_h = crtc->mode.vdisplay;
drmModeFreeCrtc(crtc);
}
uint64_t has_prime;
if (drmGetCap(p->ctx->fd, DRM_CAP_PRIME, &has_prime) < 0) {
MP_ERR(hw, "Card does not support prime handles.\n");
goto err;
}
2020-03-04 11:48:40 +00:00
if (has_prime) {
drm_prime_init_handle_ref_count(p, &p->handle_refs);
}
disable_video_plane(hw);
p->hwctx = (struct mp_hwdec_ctx) {
.driver_name = hw->driver->name,
.hw_imgfmt = IMGFMT_DRMPRIME,
};
char *device = drmGetDeviceNameFromFd2(p->ctx->fd);
int ret = av_hwdevice_ctx_create(&p->hwctx.av_device_ref,
AV_HWDEVICE_TYPE_DRM, device, NULL, 0);
if (device)
free(device);
if (ret != 0) {
MP_VERBOSE(hw, "Failed to create hwdevice_ctx: %s\n", av_err2str(ret));
goto err;
}
hwdec_devices_add(hw->devs, &p->hwctx);
return 0;
err:
uninit(hw);
return -1;
}
const struct ra_hwdec_driver ra_hwdec_drmprime_overlay = {
.name = "drmprime-overlay",
.priv_size = sizeof(struct priv),
.imgfmts = {IMGFMT_DRMPRIME, 0},
.init = init,
.overlay_frame = overlay_frame,
.uninit = uninit,
};