From 03114c80e22a382c7af66f73130ae8251140c403 Mon Sep 17 00:00:00 2001 From: Andri Yngvason Date: Sat, 20 Jun 2020 14:33:40 +0000 Subject: [PATCH] Create a buffer abstraction --- include/buffer.h | 41 ++++++++++ src/buffer.c | 190 +++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 231 insertions(+) create mode 100644 include/buffer.h create mode 100644 src/buffer.c diff --git a/include/buffer.h b/include/buffer.h new file mode 100644 index 0000000..57c9dfa --- /dev/null +++ b/include/buffer.h @@ -0,0 +1,41 @@ +#pragma once + +#include "sys/queue.h" + +#include +#include +#include + +struct wl_buffer; + +struct wv_buffer { + TAILQ_ENTRY(wv_buffer) link; + struct wl_buffer* wl_buffer; + void* pixels; + size_t size; + pixman_image_t* image; + int width, height, stride; + uint32_t format; + bool y_inverted; +}; + +TAILQ_HEAD(wv_buffer_queue, wv_buffer); + +struct wv_buffer_pool { + struct wv_buffer_queue queue; + int width, height, stride; + uint32_t format; +}; + +struct wv_buffer* wv_buffer_create(int width, int height, int stride, + uint32_t fourcc); +void wv_buffer_destroy(struct wv_buffer* self); + +struct wv_buffer_pool* wv_buffer_pool_create(int width, int height, int stride, + uint32_t format); +void wv_buffer_pool_destroy(struct wv_buffer_pool* pool); +void wv_buffer_pool_resize(struct wv_buffer_pool* pool, + int width, int height, int stride, uint32_t format); +struct wv_buffer* wv_buffer_pool_acquire(struct wv_buffer_pool* pool); +void wv_buffer_pool_release(struct wv_buffer_pool* pool, + struct wv_buffer* buffer); diff --git a/src/buffer.c b/src/buffer.c new file mode 100644 index 0000000..8c6b0c9 --- /dev/null +++ b/src/buffer.c @@ -0,0 +1,190 @@ +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include "shm.h" +#include "sys/queue.h" +#include "buffer.h" + +extern struct wl_shm* wl_shm; + +static enum wl_shm_format fourcc_to_wl_shm(uint32_t in) +{ + assert(!(in & DRM_FORMAT_BIG_ENDIAN)); + + switch (in) { + case DRM_FORMAT_ARGB8888: return WL_SHM_FORMAT_ARGB8888; + case DRM_FORMAT_XRGB8888: return WL_SHM_FORMAT_XRGB8888; + } + + return in; +} + +static bool fourcc_to_pixman_fmt(pixman_format_code_t* dst, uint32_t src) +{ + assert(!(src & DRM_FORMAT_BIG_ENDIAN)); + + /* TODO: Add more, perhaps with the help of + * https://github.com/afrantzis/pixel-format-guide + */ + switch (src) { + case DRM_FORMAT_ARGB8888: *dst = PIXMAN_a8r8g8b8; break; + case DRM_FORMAT_XRGB8888: *dst = PIXMAN_x8r8g8b8; break; + case DRM_FORMAT_ABGR8888: *dst = PIXMAN_a8b8g8r8; break; + case DRM_FORMAT_XBGR8888: *dst = PIXMAN_x8b8g8r8; break; + case DRM_FORMAT_RGBA8888: *dst = PIXMAN_r8g8b8a8; break; + case DRM_FORMAT_RGBX8888: *dst = PIXMAN_r8g8b8x8; break; + case DRM_FORMAT_BGRA8888: *dst = PIXMAN_b8g8r8a8; break; + case DRM_FORMAT_BGRX8888: *dst = PIXMAN_b8g8r8x8; break; + default: return false; + } + + return true; +} + +struct wv_buffer* wv_buffer_create(int width, int height, int stride, + uint32_t fourcc) +{ + assert(wl_shm); + enum wl_shm_format wl_fmt = fourcc_to_wl_shm(fourcc); + pixman_format_code_t pixman_fmt = 0; + if (!fourcc_to_pixman_fmt(&pixman_fmt, fourcc)) + return NULL; + + struct wv_buffer* self = calloc(1, sizeof(*self)); + if (!self) + return NULL; + + self->width = width; + self->height = height; + self->stride = stride; + self->format = fourcc; + + self->size = height * stride; + int fd = shm_alloc_fd(self->size); + if (fd < 0) + goto failure; + + self->pixels = mmap(NULL, self->size, PROT_READ | PROT_WRITE, + MAP_SHARED, fd, 0); + if (!self->pixels) + goto mmap_failure; + + self->image = pixman_image_create_bits(pixman_fmt, width, height, + self->pixels, stride); + if (!self->image) + goto pixman_failure; + + struct wl_shm_pool* pool = wl_shm_create_pool(wl_shm, fd, self->size); + if (!pool) + goto pool_failure; + + self->wl_buffer = wl_shm_pool_create_buffer(pool, 0, width, height, + stride, wl_fmt); + wl_shm_pool_destroy(pool); + if (!self->wl_buffer) + goto shm_failure; + + close(fd); + return self; + +shm_failure: +pool_failure: + pixman_image_unref(self->image); +pixman_failure: + munmap(self->pixels, self->size); +mmap_failure: + close(fd); +failure: + free(self); + return NULL; +} + +void wv_buffer_destroy(struct wv_buffer* self) +{ + wl_buffer_destroy(self->wl_buffer); + pixman_image_unref(self->image); + munmap(self->pixels, self->size); + free(self); +} + +struct wv_buffer_pool* wv_buffer_pool_create(int width, int height, int stride, + uint32_t format) +{ + struct wv_buffer_pool* self = calloc(1, sizeof(*self)); + if (!self) + return NULL; + + TAILQ_INIT(&self->queue); + self->width = width; + self->height = height; + self->stride = stride; + self->format = format; + + return self; +} + +static void wv_buffer_pool_clear(struct wv_buffer_pool* pool) +{ + while (!TAILQ_EMPTY(&pool->queue)) { + struct wv_buffer* buffer = TAILQ_FIRST(&pool->queue); + TAILQ_REMOVE(&pool->queue, buffer, link); + wv_buffer_destroy(buffer); + } +} + +void wv_buffer_pool_destroy(struct wv_buffer_pool* pool) +{ + wv_buffer_pool_clear(pool); + free(pool); +} + +void wv_buffer_pool_resize(struct wv_buffer_pool* pool, + int width, int height, int stride, uint32_t format) +{ + if (pool->width != width || pool->height != height + || pool->stride != stride || pool->format != format) { + wv_buffer_pool_clear(pool); + } + + pool->width = width; + pool->height = height; + pool->stride = stride; + pool->format = format; +} + +struct wv_buffer* wv_buffer_pool_acquire(struct wv_buffer_pool* pool) +{ + struct wv_buffer* buffer = TAILQ_FIRST(&pool->queue); + if (buffer) { + assert(pool->width == buffer->width + && pool->height == buffer->height + && pool->stride == buffer->stride + && pool->format == buffer->format); + + TAILQ_REMOVE(&pool->queue, buffer, link); + return buffer; + } + + return wv_buffer_create(pool->width, pool->height, pool->stride, + pool->format); +} + +void wv_buffer_pool_release(struct wv_buffer_pool* pool, + struct wv_buffer* buffer) +{ + if (pool->width == buffer->width + && pool->height == buffer->height + && pool->stride == buffer->stride + && pool->format == buffer->format) { + TAILQ_INSERT_TAIL(&pool->queue, buffer, link); + } else { + wv_buffer_destroy(buffer); + } +}