aboutsummaryrefslogtreecommitdiff
path: root/render
diff options
context:
space:
mode:
Diffstat (limited to 'render')
-rw-r--r--render/gbm_allocator.c182
-rw-r--r--render/meson.build1
2 files changed, 183 insertions, 0 deletions
diff --git a/render/gbm_allocator.c b/render/gbm_allocator.c
new file mode 100644
index 00000000..da29501a
--- /dev/null
+++ b/render/gbm_allocator.c
@@ -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,
+};
diff --git a/render/meson.build b/render/meson.build
index 58cc6761..f68322fe 100644
--- a/render/meson.build
+++ b/render/meson.build
@@ -3,6 +3,7 @@ wlr_files += files(
'dmabuf.c',
'egl.c',
'drm_format_set.c',
+ 'gbm_allocator.c',
'gles2/pixel_format.c',
'gles2/renderer.c',
'gles2/shaders.c',