mirror of
				https://gitlab.freedesktop.org/wlroots/wlroots.git
				synced 2025-11-03 09:01:40 -05:00 
			
		
		
		
	
		
			
				
	
	
		
			295 lines
		
	
	
	
		
			7.9 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			295 lines
		
	
	
	
		
			7.9 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
#include <assert.h>
 | 
						|
#include <GLES2/gl2.h>
 | 
						|
#include <GLES2/gl2ext.h>
 | 
						|
#include <stdint.h>
 | 
						|
#include <stdlib.h>
 | 
						|
#include <wayland-server-protocol.h>
 | 
						|
#include <wayland-util.h>
 | 
						|
#include <wlr/render/wlr_texture.h>
 | 
						|
#include <wlr/render/egl.h>
 | 
						|
#include <wlr/render/interface.h>
 | 
						|
#include <wlr/types/wlr_matrix.h>
 | 
						|
#include <wlr/util/log.h>
 | 
						|
#include "glapi.h"
 | 
						|
#include "render/gles2.h"
 | 
						|
#include "util/signal.h"
 | 
						|
 | 
						|
static const struct wlr_texture_impl texture_impl;
 | 
						|
 | 
						|
static struct wlr_gles2_texture *gles2_get_texture(
 | 
						|
		struct wlr_texture *wlr_texture) {
 | 
						|
	assert(wlr_texture->impl == &texture_impl);
 | 
						|
	return (struct wlr_gles2_texture *)wlr_texture;
 | 
						|
}
 | 
						|
 | 
						|
struct wlr_gles2_texture *get_gles2_texture_in_context(
 | 
						|
		struct wlr_texture *wlr_texture) {
 | 
						|
	struct wlr_gles2_texture *texture = gles2_get_texture(wlr_texture);
 | 
						|
	assert(wlr_egl_is_current(texture->egl));
 | 
						|
	return texture;
 | 
						|
}
 | 
						|
 | 
						|
static void gles2_texture_get_size(struct wlr_texture *wlr_texture, int *width,
 | 
						|
		int *height) {
 | 
						|
	struct wlr_gles2_texture *texture = gles2_get_texture(wlr_texture);
 | 
						|
	*width = texture->width;
 | 
						|
	*height = texture->height;
 | 
						|
}
 | 
						|
 | 
						|
static bool gles2_texture_is_opaque(struct wlr_texture *wlr_texture) {
 | 
						|
	struct wlr_gles2_texture *texture = gles2_get_texture(wlr_texture);
 | 
						|
	return !texture->has_alpha;
 | 
						|
}
 | 
						|
 | 
						|
static bool gles2_texture_write_pixels(struct wlr_texture *wlr_texture,
 | 
						|
		enum wl_shm_format wl_fmt, uint32_t stride, uint32_t width,
 | 
						|
		uint32_t height, uint32_t src_x, uint32_t src_y, uint32_t dst_x,
 | 
						|
		uint32_t dst_y, const void *data) {
 | 
						|
	struct wlr_gles2_texture *texture =
 | 
						|
		get_gles2_texture_in_context(wlr_texture);
 | 
						|
 | 
						|
	if (texture->type != WLR_GLES2_TEXTURE_GLTEX) {
 | 
						|
		wlr_log(WLR_ERROR, "Cannot write pixels to immutable texture");
 | 
						|
		return false;
 | 
						|
	}
 | 
						|
 | 
						|
	const struct wlr_gles2_pixel_format *fmt = get_gles2_format_from_wl(wl_fmt);
 | 
						|
	if (fmt == NULL) {
 | 
						|
		wlr_log(WLR_ERROR, "Unsupported pixel format %"PRIu32, wl_fmt);
 | 
						|
		return false;
 | 
						|
	}
 | 
						|
 | 
						|
	// TODO: what if the unpack subimage extension isn't supported?
 | 
						|
	PUSH_GLES2_DEBUG;
 | 
						|
 | 
						|
	glBindTexture(GL_TEXTURE_2D, texture->gl_tex);
 | 
						|
 | 
						|
	glPixelStorei(GL_UNPACK_ROW_LENGTH_EXT, stride / (fmt->bpp / 8));
 | 
						|
	glPixelStorei(GL_UNPACK_SKIP_PIXELS_EXT, src_x);
 | 
						|
	glPixelStorei(GL_UNPACK_SKIP_ROWS_EXT, src_y);
 | 
						|
 | 
						|
	glTexSubImage2D(GL_TEXTURE_2D, 0, dst_x, dst_y, width, height,
 | 
						|
		fmt->gl_format, fmt->gl_type, data);
 | 
						|
 | 
						|
	glPixelStorei(GL_UNPACK_ROW_LENGTH_EXT, 0);
 | 
						|
	glPixelStorei(GL_UNPACK_SKIP_PIXELS_EXT, 0);
 | 
						|
	glPixelStorei(GL_UNPACK_SKIP_ROWS_EXT, 0);
 | 
						|
 | 
						|
	POP_GLES2_DEBUG;
 | 
						|
	return true;
 | 
						|
}
 | 
						|
 | 
						|
static bool gles2_texture_to_dmabuf(struct wlr_texture *wlr_texture,
 | 
						|
		struct wlr_dmabuf_attributes *attribs) {
 | 
						|
	struct wlr_gles2_texture *texture = gles2_get_texture(wlr_texture);
 | 
						|
 | 
						|
	if (!texture->image) {
 | 
						|
		assert(texture->type == WLR_GLES2_TEXTURE_GLTEX);
 | 
						|
 | 
						|
		if (!eglCreateImageKHR) {
 | 
						|
			return false;
 | 
						|
		}
 | 
						|
 | 
						|
		texture->image = eglCreateImageKHR(texture->egl->display,
 | 
						|
			texture->egl->context, EGL_GL_TEXTURE_2D_KHR,
 | 
						|
			(EGLClientBuffer)(uintptr_t)texture->gl_tex, NULL);
 | 
						|
		if (texture->image == EGL_NO_IMAGE_KHR) {
 | 
						|
			return false;
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	uint32_t flags = 0;
 | 
						|
	if (texture->inverted_y) {
 | 
						|
		flags |= WLR_DMABUF_ATTRIBUTES_FLAGS_Y_INVERT;
 | 
						|
	}
 | 
						|
 | 
						|
	return wlr_egl_export_image_to_dmabuf(texture->egl, texture->image,
 | 
						|
		texture->width, texture->height, flags, attribs);
 | 
						|
}
 | 
						|
 | 
						|
static void gles2_texture_destroy(struct wlr_texture *wlr_texture) {
 | 
						|
	if (wlr_texture == NULL) {
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	struct wlr_gles2_texture *texture = gles2_get_texture(wlr_texture);
 | 
						|
 | 
						|
	wlr_egl_make_current(texture->egl, EGL_NO_SURFACE, NULL);
 | 
						|
 | 
						|
	PUSH_GLES2_DEBUG;
 | 
						|
 | 
						|
	if (texture->image_tex) {
 | 
						|
		glDeleteTextures(1, &texture->image_tex);
 | 
						|
	}
 | 
						|
	wlr_egl_destroy_image(texture->egl, texture->image);
 | 
						|
 | 
						|
	if (texture->type == WLR_GLES2_TEXTURE_GLTEX) {
 | 
						|
		glDeleteTextures(1, &texture->gl_tex);
 | 
						|
	}
 | 
						|
 | 
						|
	POP_GLES2_DEBUG;
 | 
						|
 | 
						|
	free(texture);
 | 
						|
}
 | 
						|
 | 
						|
static const struct wlr_texture_impl texture_impl = {
 | 
						|
	.get_size = gles2_texture_get_size,
 | 
						|
	.is_opaque = gles2_texture_is_opaque,
 | 
						|
	.write_pixels = gles2_texture_write_pixels,
 | 
						|
	.to_dmabuf = gles2_texture_to_dmabuf,
 | 
						|
	.destroy = gles2_texture_destroy,
 | 
						|
};
 | 
						|
 | 
						|
struct wlr_texture *wlr_gles2_texture_from_pixels(struct wlr_egl *egl,
 | 
						|
		enum wl_shm_format wl_fmt, uint32_t stride, uint32_t width,
 | 
						|
		uint32_t height, const void *data) {
 | 
						|
	assert(wlr_egl_is_current(egl));
 | 
						|
 | 
						|
	const struct wlr_gles2_pixel_format *fmt = get_gles2_format_from_wl(wl_fmt);
 | 
						|
	if (fmt == NULL) {
 | 
						|
		wlr_log(WLR_ERROR, "Unsupported pixel format %"PRIu32, wl_fmt);
 | 
						|
		return NULL;
 | 
						|
	}
 | 
						|
 | 
						|
	struct wlr_gles2_texture *texture =
 | 
						|
		calloc(1, sizeof(struct wlr_gles2_texture));
 | 
						|
	if (texture == NULL) {
 | 
						|
		wlr_log(WLR_ERROR, "Allocation failed");
 | 
						|
		return NULL;
 | 
						|
	}
 | 
						|
	wlr_texture_init(&texture->wlr_texture, &texture_impl);
 | 
						|
	texture->egl = egl;
 | 
						|
	texture->width = width;
 | 
						|
	texture->height = height;
 | 
						|
	texture->type = WLR_GLES2_TEXTURE_GLTEX;
 | 
						|
	texture->has_alpha = fmt->has_alpha;
 | 
						|
 | 
						|
	PUSH_GLES2_DEBUG;
 | 
						|
 | 
						|
	glGenTextures(1, &texture->gl_tex);
 | 
						|
	glBindTexture(GL_TEXTURE_2D, texture->gl_tex);
 | 
						|
 | 
						|
	glPixelStorei(GL_UNPACK_ROW_LENGTH_EXT, stride / (fmt->bpp / 8));
 | 
						|
	glTexImage2D(GL_TEXTURE_2D, 0, fmt->gl_format, width, height, 0,
 | 
						|
		fmt->gl_format, fmt->gl_type, data);
 | 
						|
	glPixelStorei(GL_UNPACK_ROW_LENGTH_EXT, 0);
 | 
						|
 | 
						|
	POP_GLES2_DEBUG;
 | 
						|
	return &texture->wlr_texture;
 | 
						|
}
 | 
						|
 | 
						|
struct wlr_texture *wlr_gles2_texture_from_wl_drm(struct wlr_egl *egl,
 | 
						|
		struct wl_resource *data) {
 | 
						|
	assert(wlr_egl_is_current(egl));
 | 
						|
 | 
						|
	if (!glEGLImageTargetTexture2DOES) {
 | 
						|
		return NULL;
 | 
						|
	}
 | 
						|
 | 
						|
	struct wlr_gles2_texture *texture =
 | 
						|
		calloc(1, sizeof(struct wlr_gles2_texture));
 | 
						|
	if (texture == NULL) {
 | 
						|
		wlr_log(WLR_ERROR, "Allocation failed");
 | 
						|
		return NULL;
 | 
						|
	}
 | 
						|
	wlr_texture_init(&texture->wlr_texture, &texture_impl);
 | 
						|
	texture->egl = egl;
 | 
						|
	texture->wl_drm = data;
 | 
						|
 | 
						|
	EGLint fmt;
 | 
						|
	texture->image = wlr_egl_create_image_from_wl_drm(egl, data, &fmt,
 | 
						|
		&texture->width, &texture->height, &texture->inverted_y);
 | 
						|
	if (texture->image == NULL) {
 | 
						|
		free(texture);
 | 
						|
		return NULL;
 | 
						|
	}
 | 
						|
 | 
						|
	GLenum target;
 | 
						|
	switch (fmt) {
 | 
						|
	case EGL_TEXTURE_RGB:
 | 
						|
	case EGL_TEXTURE_RGBA:
 | 
						|
		target = GL_TEXTURE_2D;
 | 
						|
		texture->type = WLR_GLES2_TEXTURE_WL_DRM_GL;
 | 
						|
		texture->has_alpha = fmt == EGL_TEXTURE_RGBA;
 | 
						|
		break;
 | 
						|
	case EGL_TEXTURE_EXTERNAL_WL:
 | 
						|
		target = GL_TEXTURE_EXTERNAL_OES;
 | 
						|
		texture->type = WLR_GLES2_TEXTURE_WL_DRM_EXT;
 | 
						|
		texture->has_alpha = true;
 | 
						|
		break;
 | 
						|
	default:
 | 
						|
		wlr_log(WLR_ERROR, "Invalid or unsupported EGL buffer format");
 | 
						|
		free(texture);
 | 
						|
		return NULL;
 | 
						|
	}
 | 
						|
 | 
						|
	PUSH_GLES2_DEBUG;
 | 
						|
 | 
						|
	glGenTextures(1, &texture->image_tex);
 | 
						|
	glBindTexture(target, texture->image_tex);
 | 
						|
	glEGLImageTargetTexture2DOES(target, texture->image);
 | 
						|
 | 
						|
	POP_GLES2_DEBUG;
 | 
						|
	return &texture->wlr_texture;
 | 
						|
}
 | 
						|
 | 
						|
#ifndef DRM_FORMAT_BIG_ENDIAN
 | 
						|
#define DRM_FORMAT_BIG_ENDIAN 0x80000000
 | 
						|
#endif
 | 
						|
 | 
						|
struct wlr_texture *wlr_gles2_texture_from_dmabuf(struct wlr_egl *egl,
 | 
						|
		struct wlr_dmabuf_attributes *attribs) {
 | 
						|
	assert(wlr_egl_is_current(egl));
 | 
						|
 | 
						|
	if (!glEGLImageTargetTexture2DOES) {
 | 
						|
		return NULL;
 | 
						|
	}
 | 
						|
 | 
						|
	if (!egl->exts.image_dmabuf_import_ext) {
 | 
						|
		wlr_log(WLR_ERROR, "Cannot create DMA-BUF texture: EGL extension "
 | 
						|
			"unavailable");
 | 
						|
		return NULL;
 | 
						|
	}
 | 
						|
 | 
						|
	switch (attribs->format & ~DRM_FORMAT_BIG_ENDIAN) {
 | 
						|
	case WL_SHM_FORMAT_YUYV:
 | 
						|
	case WL_SHM_FORMAT_YVYU:
 | 
						|
	case WL_SHM_FORMAT_UYVY:
 | 
						|
	case WL_SHM_FORMAT_VYUY:
 | 
						|
	case WL_SHM_FORMAT_AYUV:
 | 
						|
		// TODO: YUV based formats not yet supported, require multiple images
 | 
						|
		return false;
 | 
						|
	default:
 | 
						|
		break;
 | 
						|
	}
 | 
						|
 | 
						|
	struct wlr_gles2_texture *texture =
 | 
						|
		calloc(1, sizeof(struct wlr_gles2_texture));
 | 
						|
	if (texture == NULL) {
 | 
						|
		wlr_log(WLR_ERROR, "Allocation failed");
 | 
						|
		return NULL;
 | 
						|
	}
 | 
						|
	wlr_texture_init(&texture->wlr_texture, &texture_impl);
 | 
						|
	texture->egl = egl;
 | 
						|
	texture->width = attribs->width;
 | 
						|
	texture->height = attribs->height;
 | 
						|
	texture->type = WLR_GLES2_TEXTURE_DMABUF;
 | 
						|
	texture->has_alpha = true;
 | 
						|
	texture->inverted_y =
 | 
						|
		(attribs->flags & WLR_DMABUF_ATTRIBUTES_FLAGS_Y_INVERT) != 0;
 | 
						|
 | 
						|
	texture->image = wlr_egl_create_image_from_dmabuf(egl, attribs);
 | 
						|
	if (texture->image == NULL) {
 | 
						|
		free(texture);
 | 
						|
		return NULL;
 | 
						|
	}
 | 
						|
 | 
						|
	PUSH_GLES2_DEBUG;
 | 
						|
 | 
						|
	glGenTextures(1, &texture->image_tex);
 | 
						|
	glBindTexture(GL_TEXTURE_EXTERNAL_OES, texture->image_tex);
 | 
						|
	glEGLImageTargetTexture2DOES(GL_TEXTURE_EXTERNAL_OES, texture->image);
 | 
						|
 | 
						|
	POP_GLES2_DEBUG;
 | 
						|
	return &texture->wlr_texture;
 | 
						|
}
 |