parent
f47445f142
commit
5913040110
@ -0,0 +1,29 @@
|
|||||||
|
#ifndef RENDER_GBM_ALLOCATOR_H
|
||||||
|
#define RENDER_GBM_ALLOCATOR_H
|
||||||
|
|
||||||
|
#include <gbm.h>
|
||||||
|
#include <wlr/types/wlr_buffer.h>
|
||||||
|
#include "render/allocator.h"
|
||||||
|
|
||||||
|
struct wlr_gbm_buffer {
|
||||||
|
struct wlr_buffer base;
|
||||||
|
|
||||||
|
struct gbm_bo *gbm_bo;
|
||||||
|
struct wlr_dmabuf_attributes dmabuf;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct wlr_gbm_allocator {
|
||||||
|
struct wlr_allocator base;
|
||||||
|
|
||||||
|
int fd;
|
||||||
|
struct gbm_device *gbm_device;
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Creates a new GBM allocator from a render FD.
|
||||||
|
*
|
||||||
|
* Takes ownership over the FD.
|
||||||
|
*/
|
||||||
|
struct wlr_gbm_allocator *wlr_gbm_allocator_create(int render_fd);
|
||||||
|
|
||||||
|
#endif
|
@ -0,0 +1,182 @@
|
|||||||
|
#define _POSIX_C_SOURCE 200809L
|
||||||
|
#include <assert.h>
|
||||||
|
#include <drm_fourcc.h>
|
||||||
|
#include <fcntl.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <wlr/util/log.h>
|
||||||
|
#include <xf86drm.h>
|
||||||
|
#include "render/gbm_allocator.h"
|
||||||
|
|
||||||
|
static const struct wlr_buffer_impl buffer_impl;
|
||||||
|
|
||||||
|
static struct wlr_gbm_buffer *get_gbm_buffer_from_buffer(
|
||||||
|
struct wlr_buffer *buffer) {
|
||||||
|
assert(buffer->impl == &buffer_impl);
|
||||||
|
return (struct wlr_gbm_buffer *)buffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
static struct wlr_gbm_buffer *create_buffer(struct gbm_device *gbm_device,
|
||||||
|
int width, int height, const struct wlr_drm_format *format) {
|
||||||
|
struct gbm_bo *bo = NULL;
|
||||||
|
if (format->len > 0) {
|
||||||
|
bo = gbm_bo_create_with_modifiers(gbm_device, width, height,
|
||||||
|
format->format, format->modifiers, format->len);
|
||||||
|
}
|
||||||
|
if (bo == NULL) {
|
||||||
|
uint32_t usage = GBM_BO_USE_SCANOUT | GBM_BO_USE_RENDERING;
|
||||||
|
if (format->len == 1 &&
|
||||||
|
format->modifiers[0] == DRM_FORMAT_MOD_LINEAR) {
|
||||||
|
usage |= GBM_BO_USE_LINEAR;
|
||||||
|
}
|
||||||
|
bo = gbm_bo_create(gbm_device, width, height, format->format, usage);
|
||||||
|
}
|
||||||
|
if (bo == NULL) {
|
||||||
|
wlr_log(WLR_ERROR, "gbm_bo_create failed");
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
struct wlr_gbm_buffer *buffer = calloc(1, sizeof(*buffer));
|
||||||
|
if (buffer == NULL) {
|
||||||
|
gbm_bo_destroy(bo);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
wlr_buffer_init(&buffer->base, &buffer_impl, width, height);
|
||||||
|
buffer->gbm_bo = bo;
|
||||||
|
|
||||||
|
wlr_log(WLR_DEBUG, "Allocated %dx%d GBM buffer (format 0x%"PRIX32", "
|
||||||
|
"modifier 0x%"PRIX64")", buffer->base.width, buffer->base.height,
|
||||||
|
gbm_bo_get_format(bo), gbm_bo_get_modifier(bo));
|
||||||
|
|
||||||
|
return buffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void buffer_destroy(struct wlr_buffer *wlr_buffer) {
|
||||||
|
struct wlr_gbm_buffer *buffer =
|
||||||
|
get_gbm_buffer_from_buffer(wlr_buffer);
|
||||||
|
wlr_dmabuf_attributes_finish(&buffer->dmabuf);
|
||||||
|
gbm_bo_destroy(buffer->gbm_bo);
|
||||||
|
free(buffer);
|
||||||
|
}
|
||||||
|
|
||||||
|
static bool buffer_create_dmabuf(struct wlr_gbm_buffer *buffer) {
|
||||||
|
assert(buffer->dmabuf.n_planes == 0);
|
||||||
|
|
||||||
|
struct gbm_bo *bo = buffer->gbm_bo;
|
||||||
|
struct wlr_dmabuf_attributes attribs = {0};
|
||||||
|
|
||||||
|
attribs.n_planes = gbm_bo_get_plane_count(bo);
|
||||||
|
if (attribs.n_planes > WLR_DMABUF_MAX_PLANES) {
|
||||||
|
wlr_log(WLR_ERROR, "GBM BO contains too many planes (%d)",
|
||||||
|
attribs.n_planes);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
attribs.width = gbm_bo_get_width(bo);
|
||||||
|
attribs.height = gbm_bo_get_height(bo);
|
||||||
|
attribs.format = gbm_bo_get_format(bo);
|
||||||
|
attribs.modifier = gbm_bo_get_modifier(bo);
|
||||||
|
|
||||||
|
int i;
|
||||||
|
for (i = 0; i < attribs.n_planes; ++i) {
|
||||||
|
union gbm_bo_handle handle = gbm_bo_get_handle_for_plane(bo, i);
|
||||||
|
if (handle.s32 < 0) {
|
||||||
|
wlr_log(WLR_ERROR, "gbm_bo_get_handle_for_plane failed");
|
||||||
|
goto error_fd;
|
||||||
|
}
|
||||||
|
|
||||||
|
int drm_fd = gbm_device_get_fd(gbm_bo_get_device(bo));
|
||||||
|
int ret = drmPrimeHandleToFD(drm_fd, handle.s32,
|
||||||
|
DRM_CLOEXEC, &attribs.fd[i]);
|
||||||
|
if (ret < 0 || attribs.fd[i] < 0) {
|
||||||
|
wlr_log_errno(WLR_ERROR, "drmPrimeHandleToFD failed");
|
||||||
|
goto error_fd;
|
||||||
|
}
|
||||||
|
|
||||||
|
attribs.offset[i] = gbm_bo_get_offset(bo, i);
|
||||||
|
attribs.stride[i] = gbm_bo_get_stride_for_plane(bo, i);
|
||||||
|
}
|
||||||
|
|
||||||
|
memcpy(&buffer->dmabuf, &attribs, sizeof(attribs));
|
||||||
|
return true;
|
||||||
|
|
||||||
|
error_fd:
|
||||||
|
for (int j = 0; j < i; ++j) {
|
||||||
|
close(attribs.fd[j]);
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
static bool buffer_get_dmabuf(struct wlr_buffer *wlr_buffer,
|
||||||
|
struct wlr_dmabuf_attributes *attribs) {
|
||||||
|
struct wlr_gbm_buffer *buffer =
|
||||||
|
get_gbm_buffer_from_buffer(wlr_buffer);
|
||||||
|
|
||||||
|
memset(attribs, 0, sizeof(*attribs));
|
||||||
|
|
||||||
|
// Only export the buffer once
|
||||||
|
if (buffer->dmabuf.n_planes == 0) {
|
||||||
|
if (!buffer_create_dmabuf(buffer)) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
memcpy(attribs, &buffer->dmabuf, sizeof(buffer->dmabuf));
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
static const struct wlr_buffer_impl buffer_impl = {
|
||||||
|
.destroy = buffer_destroy,
|
||||||
|
.get_dmabuf = buffer_get_dmabuf,
|
||||||
|
};
|
||||||
|
|
||||||
|
static const struct wlr_allocator_interface allocator_impl;
|
||||||
|
|
||||||
|
static struct wlr_gbm_allocator *get_gbm_alloc_from_alloc(
|
||||||
|
struct wlr_allocator *alloc) {
|
||||||
|
assert(alloc->impl == &allocator_impl);
|
||||||
|
return (struct wlr_gbm_allocator *)alloc;
|
||||||
|
}
|
||||||
|
|
||||||
|
struct wlr_gbm_allocator *wlr_gbm_allocator_create(int fd) {
|
||||||
|
struct wlr_gbm_allocator *alloc = calloc(1, sizeof(*alloc));
|
||||||
|
if (alloc == NULL) {
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
wlr_allocator_init(&alloc->base, &allocator_impl);
|
||||||
|
|
||||||
|
alloc->fd = fd;
|
||||||
|
|
||||||
|
alloc->gbm_device = gbm_create_device(fd);
|
||||||
|
if (alloc->gbm_device == NULL) {
|
||||||
|
wlr_log(WLR_ERROR, "gbm_create_device failed");
|
||||||
|
free(alloc);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
return alloc;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void allocator_destroy(struct wlr_allocator *wlr_alloc) {
|
||||||
|
struct wlr_gbm_allocator *alloc = get_gbm_alloc_from_alloc(wlr_alloc);
|
||||||
|
gbm_device_destroy(alloc->gbm_device);
|
||||||
|
close(alloc->fd);
|
||||||
|
free(alloc);
|
||||||
|
}
|
||||||
|
|
||||||
|
static struct wlr_buffer *allocator_create_buffer(
|
||||||
|
struct wlr_allocator *wlr_alloc, int width, int height,
|
||||||
|
const struct wlr_drm_format *format) {
|
||||||
|
struct wlr_gbm_allocator *alloc = get_gbm_alloc_from_alloc(wlr_alloc);
|
||||||
|
struct wlr_gbm_buffer *buffer =
|
||||||
|
create_buffer(alloc->gbm_device, width, height, format);
|
||||||
|
if (buffer == NULL) {
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
return &buffer->base;
|
||||||
|
}
|
||||||
|
|
||||||
|
static const struct wlr_allocator_interface allocator_impl = {
|
||||||
|
.destroy = allocator_destroy,
|
||||||
|
.create_buffer = allocator_create_buffer,
|
||||||
|
};
|
Loading…
Reference in new issue