commit
49e03e111a
@ -0,0 +1,45 @@
|
||||
/*
|
||||
* This an unstable interface of wlroots. No guarantees are made regarding the
|
||||
* future consistency of this API.
|
||||
*/
|
||||
#ifndef WLR_USE_UNSTABLE
|
||||
#error "Add -DWLR_USE_UNSTABLE to enable unstable wlroots features"
|
||||
#endif
|
||||
|
||||
#ifndef WLR_INTERFACES_WLR_EXT_IMAGE_CAPTURE_SOURCE_V1_H
|
||||
#define WLR_INTERFACES_WLR_EXT_IMAGE_CAPTURE_SOURCE_V1_H
|
||||
|
||||
#include <wayland-server-core.h>
|
||||
#include <wlr/types/wlr_ext_image_capture_source_v1.h>
|
||||
|
||||
struct wlr_ext_image_copy_capture_frame_v1;
|
||||
struct wlr_swapchain;
|
||||
struct wlr_renderer;
|
||||
struct wlr_seat;
|
||||
|
||||
struct wlr_ext_image_capture_source_v1_interface {
|
||||
// TODO: drop with_cursors flag
|
||||
void (*start)(struct wlr_ext_image_capture_source_v1 *source, bool with_cursors);
|
||||
void (*stop)(struct wlr_ext_image_capture_source_v1 *source);
|
||||
void (*schedule_frame)(struct wlr_ext_image_capture_source_v1 *source);
|
||||
void (*copy_frame)(struct wlr_ext_image_capture_source_v1 *source,
|
||||
struct wlr_ext_image_copy_capture_frame_v1 *dst_frame,
|
||||
struct wlr_ext_image_capture_source_v1_frame_event *frame_event);
|
||||
struct wlr_ext_image_capture_source_v1_cursor *(*get_pointer_cursor)(
|
||||
struct wlr_ext_image_capture_source_v1 *source, struct wlr_seat *seat);
|
||||
};
|
||||
|
||||
void wlr_ext_image_capture_source_v1_init(struct wlr_ext_image_capture_source_v1 *source,
|
||||
const struct wlr_ext_image_capture_source_v1_interface *impl);
|
||||
void wlr_ext_image_capture_source_v1_finish(struct wlr_ext_image_capture_source_v1 *source);
|
||||
bool wlr_ext_image_capture_source_v1_create_resource(struct wlr_ext_image_capture_source_v1 *source,
|
||||
struct wl_client *client, uint32_t new_id);
|
||||
void wlr_ext_image_capture_source_v1_set_constraints_from_swapchain(
|
||||
struct wlr_ext_image_capture_source_v1 *source,
|
||||
struct wlr_swapchain *swapchain, struct wlr_renderer *renderer);
|
||||
|
||||
void wlr_ext_image_capture_source_v1_cursor_init(struct wlr_ext_image_capture_source_v1_cursor *source_cursor,
|
||||
const struct wlr_ext_image_capture_source_v1_interface *impl);
|
||||
void wlr_ext_image_capture_source_v1_cursor_finish(struct wlr_ext_image_capture_source_v1_cursor *source_cursor);
|
||||
|
||||
#endif
|
@ -0,0 +1,94 @@
|
||||
/*
|
||||
* This an unstable interface of wlroots. No guarantees are made regarding the
|
||||
* future consistency of this API.
|
||||
*/
|
||||
#ifndef WLR_USE_UNSTABLE
|
||||
#error "Add -DWLR_USE_UNSTABLE to enable unstable wlroots features"
|
||||
#endif
|
||||
|
||||
#ifndef WLR_TYPES_WLR_EXT_IMAGE_CAPTURE_SOURCE_V1_H
|
||||
#define WLR_TYPES_WLR_EXT_IMAGE_CAPTURE_SOURCE_V1_H
|
||||
|
||||
#include <pixman.h>
|
||||
#include <wayland-server-core.h>
|
||||
#include <wlr/render/drm_format_set.h>
|
||||
|
||||
/**
|
||||
* A screen capture source.
|
||||
*
|
||||
* When the size, device or formats change, the constraints_update event is
|
||||
* emitted.
|
||||
*
|
||||
* The device and formats advertised are suitable for copying into a
|
||||
* struct wlr_buffer.
|
||||
*/
|
||||
struct wlr_ext_image_capture_source_v1 {
|
||||
const struct wlr_ext_image_capture_source_v1_interface *impl;
|
||||
struct wl_list resources; // wl_resource_get_link()
|
||||
|
||||
uint32_t width, height;
|
||||
|
||||
uint32_t *shm_formats;
|
||||
size_t shm_formats_len;
|
||||
|
||||
dev_t dmabuf_device;
|
||||
struct wlr_drm_format_set dmabuf_formats;
|
||||
|
||||
struct {
|
||||
struct wl_signal constraints_update;
|
||||
struct wl_signal frame; // struct wlr_ext_image_capture_source_v1_frame_event
|
||||
struct wl_signal destroy;
|
||||
} events;
|
||||
};
|
||||
|
||||
/**
|
||||
* Event indicating that the source has produced a new frame.
|
||||
*/
|
||||
struct wlr_ext_image_capture_source_v1_frame_event {
|
||||
const pixman_region32_t *damage;
|
||||
};
|
||||
|
||||
/**
|
||||
* A cursor capture source.
|
||||
*
|
||||
* Provides additional cursor-specific functionality on top of
|
||||
* struct wlr_ext_image_capture_source_v1.
|
||||
*/
|
||||
struct wlr_ext_image_capture_source_v1_cursor {
|
||||
struct wlr_ext_image_capture_source_v1 base;
|
||||
|
||||
bool entered;
|
||||
int32_t x, y;
|
||||
struct {
|
||||
int32_t x, y;
|
||||
} hotspot;
|
||||
|
||||
struct {
|
||||
struct wl_signal update;
|
||||
} events;
|
||||
};
|
||||
|
||||
/**
|
||||
* Interface exposing one screen capture source per output.
|
||||
*/
|
||||
struct wlr_ext_output_image_capture_source_manager_v1 {
|
||||
struct wl_global *global;
|
||||
|
||||
// private state
|
||||
|
||||
struct wl_listener display_destroy;
|
||||
};
|
||||
|
||||
/**
|
||||
* Obtain a struct wlr_ext_image_capture_source_v1 from an ext_image_capture_source_v1
|
||||
* resource.
|
||||
*
|
||||
* Asserts that the resource has the correct type. Returns NULL if the resource
|
||||
* is inert.
|
||||
*/
|
||||
struct wlr_ext_image_capture_source_v1 *wlr_ext_image_capture_source_v1_from_resource(struct wl_resource *resource);
|
||||
|
||||
struct wlr_ext_output_image_capture_source_manager_v1 *wlr_ext_output_image_capture_source_manager_v1_create(
|
||||
struct wl_display *display, uint32_t version);
|
||||
|
||||
#endif
|
@ -0,0 +1,49 @@
|
||||
/*
|
||||
* This an unstable interface of wlroots. No guarantees are made regarding the
|
||||
* future consistency of this API.
|
||||
*/
|
||||
#ifndef WLR_USE_UNSTABLE
|
||||
#error "Add -DWLR_USE_UNSTABLE to enable unstable wlroots features"
|
||||
#endif
|
||||
|
||||
#ifndef WLR_TYPES_WLR_EXT_IMAGE_COPY_CAPTURE_V1_H
|
||||
#define WLR_TYPES_WLR_EXT_IMAGE_COPY_CAPTURE_V1_H
|
||||
|
||||
#include <wayland-server.h>
|
||||
#include <time.h>
|
||||
#include "ext-image-copy-capture-v1-protocol.h"
|
||||
|
||||
struct wlr_ext_image_copy_capture_manager_v1 {
|
||||
struct wl_global *global;
|
||||
|
||||
// private state
|
||||
|
||||
struct wl_listener display_destroy;
|
||||
};
|
||||
|
||||
struct wlr_ext_image_copy_capture_frame_v1 {
|
||||
struct wl_resource *resource;
|
||||
bool capturing;
|
||||
struct wlr_buffer *buffer;
|
||||
pixman_region32_t buffer_damage;
|
||||
|
||||
struct {
|
||||
struct wl_signal destroy;
|
||||
} events;
|
||||
|
||||
// private state
|
||||
|
||||
struct wlr_ext_image_copy_capture_session_v1 *session;
|
||||
};
|
||||
|
||||
struct wlr_ext_image_copy_capture_manager_v1 *wlr_ext_image_copy_capture_manager_v1_create(
|
||||
struct wl_display *display, uint32_t version);
|
||||
|
||||
void wlr_ext_image_copy_capture_frame_v1_ready(struct wlr_ext_image_copy_capture_frame_v1 *frame,
|
||||
enum wl_output_transform transform, const struct timespec *presentation_time);
|
||||
bool wlr_ext_image_copy_capture_frame_v1_copy_buffer(struct wlr_ext_image_copy_capture_frame_v1 *frame,
|
||||
struct wlr_buffer *src, struct wlr_renderer *renderer);
|
||||
void wlr_ext_image_copy_capture_frame_v1_fail(struct wlr_ext_image_copy_capture_frame_v1 *frame,
|
||||
enum ext_image_copy_capture_frame_v1_failure_reason reason);
|
||||
|
||||
#endif
|
@ -0,0 +1,141 @@
|
||||
#include <assert.h>
|
||||
#include <drm_fourcc.h>
|
||||
#include <stdlib.h>
|
||||
#include <sys/stat.h>
|
||||
#include <wlr/interfaces/wlr_ext_image_capture_source_v1.h>
|
||||
#include <wlr/render/allocator.h>
|
||||
#include <wlr/render/swapchain.h>
|
||||
#include <wlr/render/wlr_renderer.h>
|
||||
#include <wlr/types/wlr_buffer.h>
|
||||
#include <wlr/types/wlr_ext_image_capture_source_v1.h>
|
||||
#include "ext-image-capture-source-v1-protocol.h"
|
||||
|
||||
static void source_handle_destroy(struct wl_client *client,
|
||||
struct wl_resource *source_resource) {
|
||||
wl_resource_destroy(source_resource);
|
||||
}
|
||||
|
||||
static const struct ext_image_capture_source_v1_interface source_impl = {
|
||||
.destroy = source_handle_destroy,
|
||||
};
|
||||
|
||||
struct wlr_ext_image_capture_source_v1 *wlr_ext_image_capture_source_v1_from_resource(struct wl_resource *resource) {
|
||||
assert(wl_resource_instance_of(resource,
|
||||
&ext_image_capture_source_v1_interface, &source_impl));
|
||||
return wl_resource_get_user_data(resource);
|
||||
}
|
||||
|
||||
static void source_handle_resource_destroy(struct wl_resource *resource) {
|
||||
wl_list_remove(wl_resource_get_link(resource));
|
||||
}
|
||||
|
||||
void wlr_ext_image_capture_source_v1_init(struct wlr_ext_image_capture_source_v1 *source,
|
||||
const struct wlr_ext_image_capture_source_v1_interface *impl) {
|
||||
*source = (struct wlr_ext_image_capture_source_v1){
|
||||
.impl = impl,
|
||||
};
|
||||
wl_list_init(&source->resources);
|
||||
wl_signal_init(&source->events.destroy);
|
||||
wl_signal_init(&source->events.constraints_update);
|
||||
wl_signal_init(&source->events.frame);
|
||||
}
|
||||
|
||||
void wlr_ext_image_capture_source_v1_finish(struct wlr_ext_image_capture_source_v1 *source) {
|
||||
wl_signal_emit_mutable(&source->events.destroy, NULL);
|
||||
|
||||
struct wl_resource *resource, *resource_tmp;
|
||||
wl_resource_for_each_safe(resource, resource_tmp, &source->resources) {
|
||||
wl_resource_set_user_data(resource, NULL);
|
||||
wl_list_remove(wl_resource_get_link(resource));
|
||||
wl_list_init(wl_resource_get_link(resource));
|
||||
}
|
||||
|
||||
free(source->shm_formats);
|
||||
wlr_drm_format_set_finish(&source->dmabuf_formats);
|
||||
}
|
||||
|
||||
bool wlr_ext_image_capture_source_v1_create_resource(struct wlr_ext_image_capture_source_v1 *source,
|
||||
struct wl_client *client, uint32_t new_id) {
|
||||
struct wl_resource *resource = wl_resource_create(client,
|
||||
&ext_image_capture_source_v1_interface, 1, new_id);
|
||||
if (resource == NULL) {
|
||||
wl_client_post_no_memory(client);
|
||||
return false;
|
||||
}
|
||||
wl_resource_set_implementation(resource, &source_impl, source,
|
||||
source_handle_resource_destroy);
|
||||
if (source != NULL) {
|
||||
wl_list_insert(&source->resources, wl_resource_get_link(resource));
|
||||
} else {
|
||||
wl_list_init(wl_resource_get_link(resource));
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
static uint32_t get_swapchain_shm_format(struct wlr_swapchain *swapchain,
|
||||
struct wlr_renderer *renderer) {
|
||||
struct wlr_buffer *buffer = wlr_swapchain_acquire(swapchain);
|
||||
if (buffer == NULL) {
|
||||
return DRM_FORMAT_INVALID;
|
||||
}
|
||||
|
||||
struct wlr_texture *texture = wlr_texture_from_buffer(renderer, buffer);
|
||||
wlr_buffer_unlock(buffer);
|
||||
if (texture == NULL) {
|
||||
return DRM_FORMAT_INVALID;
|
||||
}
|
||||
|
||||
uint32_t format = wlr_texture_preferred_read_format(texture);
|
||||
wlr_texture_destroy(texture);
|
||||
|
||||
return format;
|
||||
}
|
||||
|
||||
void wlr_ext_image_capture_source_v1_set_constraints_from_swapchain(struct wlr_ext_image_capture_source_v1 *source,
|
||||
struct wlr_swapchain *swapchain, struct wlr_renderer *renderer) {
|
||||
source->width = swapchain->width;
|
||||
source->height = swapchain->height;
|
||||
|
||||
uint32_t shm_format = get_swapchain_shm_format(swapchain, renderer);
|
||||
if (shm_format != DRM_FORMAT_INVALID) {
|
||||
uint32_t *shm_formats = calloc(1, sizeof(shm_formats[0]));
|
||||
if (shm_formats == NULL) {
|
||||
return;
|
||||
}
|
||||
shm_formats[0] = shm_format;
|
||||
|
||||
source->shm_formats_len = 1;
|
||||
free(source->shm_formats);
|
||||
source->shm_formats = shm_formats;
|
||||
}
|
||||
|
||||
int drm_fd = wlr_renderer_get_drm_fd(renderer);
|
||||
if (swapchain->allocator != NULL &&
|
||||
(swapchain->allocator->buffer_caps & WLR_BUFFER_CAP_DMABUF) &&
|
||||
drm_fd >= 0) {
|
||||
struct stat dev_stat;
|
||||
if (fstat(drm_fd, &dev_stat) != 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
source->dmabuf_device = dev_stat.st_rdev;
|
||||
source->dmabuf_formats.len = 0;
|
||||
for (size_t i = 0; i < swapchain->format.len; i++) {
|
||||
wlr_drm_format_set_add(&source->dmabuf_formats,
|
||||
swapchain->format.format, swapchain->format.modifiers[i]);
|
||||
}
|
||||
}
|
||||
|
||||
wl_signal_emit_mutable(&source->events.constraints_update, NULL);
|
||||
}
|
||||
|
||||
void wlr_ext_image_capture_source_v1_cursor_init(struct wlr_ext_image_capture_source_v1_cursor *source_cursor,
|
||||
const struct wlr_ext_image_capture_source_v1_interface *impl) {
|
||||
*source_cursor = (struct wlr_ext_image_capture_source_v1_cursor){0};
|
||||
wlr_ext_image_capture_source_v1_init(&source_cursor->base, impl);
|
||||
wl_signal_init(&source_cursor->events.update);
|
||||
}
|
||||
|
||||
void wlr_ext_image_capture_source_v1_cursor_finish(struct wlr_ext_image_capture_source_v1_cursor *source_cursor) {
|
||||
wlr_ext_image_capture_source_v1_finish(&source_cursor->base);
|
||||
}
|
@ -0,0 +1,383 @@
|
||||
#include <assert.h>
|
||||
#include <stdlib.h>
|
||||
#include <wlr/interfaces/wlr_ext_image_capture_source_v1.h>
|
||||
#include <wlr/interfaces/wlr_output.h>
|
||||
#include <wlr/render/swapchain.h>
|
||||
#include <wlr/types/wlr_ext_image_capture_source_v1.h>
|
||||
#include <wlr/types/wlr_ext_image_copy_capture_v1.h>
|
||||
#include <wlr/types/wlr_output.h>
|
||||
#include <wlr/util/addon.h>
|
||||
#include "render/wlr_renderer.h"
|
||||
#include "ext-image-capture-source-v1-protocol.h"
|
||||
|
||||
#define OUTPUT_IMAGE_SOURCE_MANAGER_V1_VERSION 1
|
||||
|
||||
struct output_cursor_source {
|
||||
struct wlr_ext_image_capture_source_v1_cursor base;
|
||||
|
||||
struct wlr_output *output;
|
||||
struct wlr_buffer *prev_buffer;
|
||||
bool initialized;
|
||||
|
||||
struct wl_listener output_commit;
|
||||
struct wl_listener prev_buffer_release;
|
||||
};
|
||||
|
||||
struct wlr_ext_output_image_capture_source_v1 {
|
||||
struct wlr_ext_image_capture_source_v1 base;
|
||||
struct wlr_addon addon;
|
||||
|
||||
struct wlr_output *output;
|
||||
|
||||
struct wl_listener output_commit;
|
||||
|
||||
struct output_cursor_source cursor;
|
||||
|
||||
size_t num_started;
|
||||
bool software_cursors_locked;
|
||||
};
|
||||
|
||||
struct wlr_ext_output_image_capture_source_v1_frame_event {
|
||||
struct wlr_ext_image_capture_source_v1_frame_event base;
|
||||
struct wlr_buffer *buffer;
|
||||
struct timespec *when;
|
||||
};
|
||||
|
||||
static void output_source_start(struct wlr_ext_image_capture_source_v1 *base,
|
||||
bool with_cursors) {
|
||||
struct wlr_ext_output_image_capture_source_v1 *source = wl_container_of(base, source, base);
|
||||
source->num_started++;
|
||||
if (source->num_started > 1) {
|
||||
return;
|
||||
}
|
||||
wlr_output_lock_attach_render(source->output, true);
|
||||
if (with_cursors) {
|
||||
wlr_output_lock_software_cursors(source->output, true);
|
||||
}
|
||||
source->software_cursors_locked = with_cursors;
|
||||
}
|
||||
|
||||
static void output_source_stop(struct wlr_ext_image_capture_source_v1 *base) {
|
||||
struct wlr_ext_output_image_capture_source_v1 *source = wl_container_of(base, source, base);
|
||||
assert(source->num_started > 0);
|
||||
if (source->num_started > 0) {
|
||||
return;
|
||||
}
|
||||
wlr_output_lock_attach_render(source->output, false);
|
||||
if (source->software_cursors_locked) {
|
||||
wlr_output_lock_software_cursors(source->output, false);
|
||||
}
|
||||
}
|
||||
|
||||
static void output_source_schedule_frame(struct wlr_ext_image_capture_source_v1 *base) {
|
||||
struct wlr_ext_output_image_capture_source_v1 *source = wl_container_of(base, source, base);
|
||||
wlr_output_update_needs_frame(source->output);
|
||||
}
|
||||
|
||||
static void output_source_copy_frame(struct wlr_ext_image_capture_source_v1 *base,
|
||||
struct wlr_ext_image_copy_capture_frame_v1 *frame,
|
||||
struct wlr_ext_image_capture_source_v1_frame_event *base_event) {
|
||||
struct wlr_ext_output_image_capture_source_v1 *source = wl_container_of(base, source, base);
|
||||
struct wlr_ext_output_image_capture_source_v1_frame_event *event =
|
||||
wl_container_of(base_event, event, base);
|
||||
|
||||
if (wlr_ext_image_copy_capture_frame_v1_copy_buffer(frame,
|
||||
event->buffer, source->output->renderer)) {
|
||||
wlr_ext_image_copy_capture_frame_v1_ready(frame,
|
||||
source->output->transform, event->when);
|
||||
}
|
||||
}
|
||||
|
||||
static struct wlr_ext_image_capture_source_v1_cursor *output_source_get_pointer_cursor(
|
||||
struct wlr_ext_image_capture_source_v1 *base, struct wlr_seat *seat) {
|
||||
// TODO: handle seat
|
||||
struct wlr_ext_output_image_capture_source_v1 *source = wl_container_of(base, source, base);
|
||||
return &source->cursor.base;
|
||||
}
|
||||
|
||||
static const struct wlr_ext_image_capture_source_v1_interface output_source_impl = {
|
||||
.start = output_source_start,
|
||||
.stop = output_source_stop,
|
||||
.schedule_frame = output_source_schedule_frame,
|
||||
.copy_frame = output_source_copy_frame,
|
||||
.get_pointer_cursor = output_source_get_pointer_cursor,
|
||||
};
|
||||
|
||||
static void source_update_buffer_constraints(struct wlr_ext_output_image_capture_source_v1 *source) {
|
||||
struct wlr_output *output = source->output;
|
||||
|
||||
if (!wlr_output_configure_primary_swapchain(output, NULL, &output->swapchain)) {
|
||||
return;
|
||||
}
|
||||
|
||||
wlr_ext_image_capture_source_v1_set_constraints_from_swapchain(&source->base,
|
||||
output->swapchain, output->renderer);
|
||||
}
|
||||
|
||||
static void source_handle_output_commit(struct wl_listener *listener,
|
||||
void *data) {
|
||||
struct wlr_ext_output_image_capture_source_v1 *source = wl_container_of(listener, source, output_commit);
|
||||
struct wlr_output_event_commit *event = data;
|
||||
|
||||
if (event->state->committed & (WLR_OUTPUT_STATE_MODE | WLR_OUTPUT_STATE_RENDER_FORMAT)) {
|
||||
source_update_buffer_constraints(source);
|
||||
}
|
||||
|
||||
if (event->state->committed & WLR_OUTPUT_STATE_BUFFER) {
|
||||
struct wlr_buffer *buffer = event->state->buffer;
|
||||
|
||||
pixman_region32_t full_damage;
|
||||
pixman_region32_init_rect(&full_damage, 0, 0, buffer->width, buffer->height);
|
||||
|
||||
const pixman_region32_t *damage;
|
||||
if (event->state->committed & WLR_OUTPUT_STATE_DAMAGE) {
|
||||
damage = &event->state->damage;
|
||||
} else {
|
||||
damage = &full_damage;
|
||||
}
|
||||
|
||||
struct wlr_ext_output_image_capture_source_v1_frame_event frame_event = {
|
||||
.base = {
|
||||
.damage = damage,
|
||||
},
|
||||
.buffer = buffer,
|
||||
.when = event->when, // TODO: predict next presentation time instead
|
||||
};
|
||||
wl_signal_emit_mutable(&source->base.events.frame, &frame_event);
|
||||
|
||||
pixman_region32_fini(&full_damage);
|
||||
}
|
||||
}
|
||||
|
||||
static void output_cursor_source_init(struct output_cursor_source *cursor_source,
|
||||
struct wlr_output *output);
|
||||
static void output_cursor_source_finish(struct output_cursor_source *cursor_source);
|
||||
|
||||
static void output_addon_destroy(struct wlr_addon *addon) {
|
||||
struct wlr_ext_output_image_capture_source_v1 *source = wl_container_of(addon, source, addon);
|
||||
wlr_ext_image_capture_source_v1_finish(&source->base);
|
||||
output_cursor_source_finish(&source->cursor);
|
||||
wl_list_remove(&source->output_commit.link);
|
||||
wlr_addon_finish(&source->addon);
|
||||
free(source);
|
||||
}
|
||||
|
||||
static const struct wlr_addon_interface output_addon_impl = {
|
||||
.name = "wlr_ext_output_image_capture_source_v1",
|
||||
.destroy = output_addon_destroy,
|
||||
};
|
||||
|
||||
static void output_manager_handle_create_source(struct wl_client *client,
|
||||
struct wl_resource *manager_resource, uint32_t new_id,
|
||||
struct wl_resource *output_resource) {
|
||||
struct wlr_output *output = wlr_output_from_resource(output_resource);
|
||||
if (output == NULL) {
|
||||
wlr_ext_image_capture_source_v1_create_resource(NULL, client, new_id);
|
||||
return;
|
||||
}
|
||||
|
||||
struct wlr_ext_output_image_capture_source_v1 *source;
|
||||
struct wlr_addon *addon = wlr_addon_find(&output->addons, NULL, &output_addon_impl);
|
||||
if (addon != NULL) {
|
||||
source = wl_container_of(addon, source, addon);
|
||||
} else {
|
||||
source = calloc(1, sizeof(*source));
|
||||
if (source == NULL) {
|
||||
wl_resource_post_no_memory(manager_resource);
|
||||
return;
|
||||
}
|
||||
|
||||
wlr_ext_image_capture_source_v1_init(&source->base, &output_source_impl);
|
||||
wlr_addon_init(&source->addon, &output->addons, NULL, &output_addon_impl);
|
||||
source->output = output;
|
||||
|
||||
source->output_commit.notify = source_handle_output_commit;
|
||||
wl_signal_add(&output->events.commit, &source->output_commit);
|
||||
|
||||
source_update_buffer_constraints(source);
|
||||
|
||||
output_cursor_source_init(&source->cursor, output);
|
||||
}
|
||||
|
||||
if (!wlr_ext_image_capture_source_v1_create_resource(&source->base, client, new_id)) {
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
static void output_manager_handle_destroy(struct wl_client *client,
|
||||
struct wl_resource *manager_resource) {
|
||||
wl_resource_destroy(manager_resource);
|
||||
}
|
||||
|
||||
static const struct ext_output_image_capture_source_manager_v1_interface output_manager_impl = {
|
||||
.create_source = output_manager_handle_create_source,
|
||||
.destroy = output_manager_handle_destroy,
|
||||
};
|
||||
|
||||
static void output_manager_bind(struct wl_client *client, void *data,
|
||||
uint32_t version, uint32_t id) {
|
||||
struct wlr_ext_output_image_capture_source_manager_v1 *manager = data;
|
||||
|
||||
struct wl_resource *resource = wl_resource_create(client,
|
||||
&ext_output_image_capture_source_manager_v1_interface, version, id);
|
||||
if (!resource) {
|
||||
wl_client_post_no_memory(client);
|
||||
return;
|
||||
}
|
||||
wl_resource_set_implementation(resource, &output_manager_impl, manager, NULL);
|
||||
}
|
||||
|
||||
static void output_manager_handle_display_destroy(struct wl_listener *listener, void *data) {
|
||||
struct wlr_ext_output_image_capture_source_manager_v1 *manager =
|
||||
wl_container_of(listener, manager, display_destroy);
|
||||
wl_list_remove(&manager->display_destroy.link);
|
||||
wl_global_destroy(manager->global);
|
||||
free(manager);
|
||||
}
|
||||
|
||||
struct wlr_ext_output_image_capture_source_manager_v1 *wlr_ext_output_image_capture_source_manager_v1_create(
|
||||
struct wl_display *display, uint32_t version) {
|
||||
assert(version <= OUTPUT_IMAGE_SOURCE_MANAGER_V1_VERSION);
|
||||
|
||||
struct wlr_ext_output_image_capture_source_manager_v1 *manager = calloc(1, sizeof(*manager));
|
||||
if (manager == NULL) {
|
||||
return NULL;
|
||||
}
|
||||
|
||||
manager->global = wl_global_create(display,
|
||||
&ext_output_image_capture_source_manager_v1_interface, version, manager, output_manager_bind);
|
||||
if (manager->global == NULL) {
|
||||
free(manager);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
manager->display_destroy.notify = output_manager_handle_display_destroy;
|
||||
wl_display_add_destroy_listener(display, &manager->display_destroy);
|
||||
|
||||
return manager;
|
||||
}
|
||||
|
||||
static void output_cursor_source_schedule_frame(struct wlr_ext_image_capture_source_v1 *base) {
|
||||
struct output_cursor_source *cursor_source = wl_container_of(base, cursor_source, base);
|
||||
wlr_output_update_needs_frame(cursor_source->output);
|
||||
}
|
||||
|
||||
static void output_cursor_source_copy_frame(struct wlr_ext_image_capture_source_v1 *base,
|
||||
struct wlr_ext_image_copy_capture_frame_v1 *frame,
|
||||
struct wlr_ext_image_capture_source_v1_frame_event *base_event) {
|
||||
struct output_cursor_source *cursor_source = wl_container_of(base, cursor_source, base);
|
||||
|
||||
struct wlr_buffer *src_buffer = cursor_source->output->cursor_front_buffer;
|
||||
if (src_buffer == NULL) {
|
||||
wlr_ext_image_copy_capture_frame_v1_fail(frame, EXT_IMAGE_COPY_CAPTURE_FRAME_V1_FAILURE_REASON_STOPPED);
|
||||
return;
|
||||
}
|
||||
|
||||
if (!wlr_ext_image_copy_capture_frame_v1_copy_buffer(frame,
|
||||
src_buffer, cursor_source->output->renderer)) {
|
||||
return;
|
||||
}
|
||||
|
||||
struct timespec now;
|
||||
clock_gettime(CLOCK_MONOTONIC, &now);
|
||||
|
||||
wlr_ext_image_copy_capture_frame_v1_ready(frame, WL_OUTPUT_TRANSFORM_NORMAL, &now);
|
||||
}
|
||||
|
||||
static const struct wlr_ext_image_capture_source_v1_interface output_cursor_source_impl = {
|
||||
.schedule_frame = output_cursor_source_schedule_frame,
|
||||
.copy_frame = output_cursor_source_copy_frame,
|
||||
};
|
||||
|
||||
static void output_cursor_source_update(struct output_cursor_source *cursor_source) {
|
||||
struct wlr_output *output = cursor_source->output;
|
||||
|
||||
if (output->cursor_swapchain != NULL && !cursor_source->initialized) {
|
||||
wlr_ext_image_capture_source_v1_set_constraints_from_swapchain(&cursor_source->base.base,
|
||||
output->cursor_swapchain, output->renderer);
|
||||
cursor_source->initialized = true;
|
||||
}
|
||||
|
||||
struct wlr_output_cursor *output_cursor = output->hardware_cursor;
|
||||
if (output_cursor == NULL || !output_cursor->visible) {
|
||||
cursor_source->base.entered = false;
|
||||
wl_signal_emit_mutable(&cursor_source->base.events.update, NULL);
|
||||
return;
|
||||
}
|
||||
|
||||
if (output->cursor_swapchain != NULL &&
|
||||
((int)cursor_source->base.base.width != output->cursor_swapchain->width ||
|
||||
(int)cursor_source->base.base.height != output->cursor_swapchain->height)) {
|
||||
cursor_source->base.base.width = output->cursor_swapchain->width;
|
||||
cursor_source->base.base.height = output->cursor_swapchain->height;
|
||||
wl_signal_emit_mutable(&cursor_source->base.base.events.constraints_update, NULL);
|
||||
}
|
||||
|
||||
cursor_source->base.entered = true;
|
||||
cursor_source->base.x = round(output_cursor->x);
|
||||
cursor_source->base.y = round(output_cursor->y);
|
||||
cursor_source->base.hotspot.x = output_cursor->hotspot_x;
|
||||
cursor_source->base.hotspot.y = output_cursor->hotspot_y;
|
||||
wl_signal_emit_mutable(&cursor_source->base.events.update, NULL);
|
||||
}
|
||||
|
||||
static void output_cursor_source_handle_prev_buffer_release(struct wl_listener *listener,
|
||||
void *data) {
|
||||
struct output_cursor_source *cursor_source = wl_container_of(listener, cursor_source, prev_buffer_release);
|
||||
wl_list_remove(&cursor_source->prev_buffer_release.link);
|
||||
wl_list_init(&cursor_source->prev_buffer_release.link);
|
||||
cursor_source->prev_buffer = NULL;
|
||||
}
|
||||
|
||||
static void output_cursor_source_handle_output_commit(struct wl_listener *listener,
|
||||
void *data) {
|
||||
struct output_cursor_source *cursor_source = wl_container_of(listener, cursor_source, output_commit);
|
||||
struct wlr_output_event_commit *event = data;
|
||||
|
||||
output_cursor_source_update(cursor_source);
|
||||
|
||||
struct wlr_buffer *buffer = cursor_source->output->cursor_front_buffer;
|
||||
if (buffer != NULL && buffer != cursor_source->prev_buffer) {
|
||||
pixman_region32_t full_damage;
|
||||
pixman_region32_init_rect(&full_damage, 0, 0, buffer->width, buffer->height);
|
||||
|
||||
struct wlr_ext_output_image_capture_source_v1_frame_event frame_event = {
|
||||
.base = {
|
||||
.damage = &full_damage,
|
||||
},
|
||||
.buffer = buffer,
|
||||
.when = event->when, // TODO: predict next presentation time instead
|
||||
};
|
||||
wl_signal_emit_mutable(&cursor_source->base.base.events.frame, &frame_event);
|
||||
|
||||
pixman_region32_fini(&full_damage);
|
||||
|
||||
assert(buffer->n_locks > 0);
|
||||
cursor_source->prev_buffer = buffer;
|
||||
wl_list_remove(&cursor_source->prev_buffer_release.link);
|
||||
cursor_source->prev_buffer_release.notify = output_cursor_source_handle_prev_buffer_release;
|
||||
wl_signal_add(&buffer->events.release, &cursor_source->prev_buffer_release);
|
||||
}
|
||||
}
|
||||
|
||||
static void output_cursor_source_init(struct output_cursor_source *cursor_source,
|
||||
struct wlr_output *output) {
|
||||
wlr_ext_image_capture_source_v1_cursor_init(&cursor_source->base, &output_cursor_source_impl);
|
||||
|
||||
// Caller is responsible for destroying the output cursor source when the
|
||||
// output is destroyed
|
||||
cursor_source->output = output;
|
||||
|
||||
cursor_source->output_commit.notify = output_cursor_source_handle_output_commit;
|
||||
wl_signal_add(&output->events.commit, &cursor_source->output_commit);
|
||||
|
||||
wl_list_init(&cursor_source->prev_buffer_release.link);
|
||||
|
||||
output_cursor_source_update(cursor_source);
|
||||
}
|
||||
|
||||
static void output_cursor_source_finish(struct output_cursor_source *cursor_source) {
|
||||
wlr_ext_image_capture_source_v1_cursor_finish(&cursor_source->base);
|
||||
wl_list_remove(&cursor_source->output_commit.link);
|
||||
wl_list_remove(&cursor_source->prev_buffer_release.link);
|
||||
}
|
@ -0,0 +1,685 @@
|
||||
#include <assert.h>
|
||||
#include <pixman.h>
|
||||
#include <stdlib.h>
|
||||
#include <wlr/interfaces/wlr_ext_image_capture_source_v1.h>
|
||||
#include <wlr/types/wlr_buffer.h>
|
||||
#include <wlr/types/wlr_ext_image_copy_capture_v1.h>
|
||||
#include <wlr/types/wlr_seat.h>
|
||||
#include <wlr/render/wlr_renderer.h>
|
||||
|
||||
#define IMAGE_COPY_CAPTURE_MANAGER_V1_VERSION 1
|
||||
|
||||
struct wlr_ext_image_copy_capture_session_v1 {
|
||||
struct wl_resource *resource;
|
||||
struct wlr_ext_image_capture_source_v1 *source;
|
||||
struct wlr_ext_image_copy_capture_frame_v1 *frame;
|
||||
|
||||
struct wl_listener source_destroy;
|
||||
struct wl_listener source_constraints_update;
|
||||
struct wl_listener source_frame;
|
||||
|
||||
pixman_region32_t damage;
|
||||
};
|
||||
|
||||
struct wlr_ext_image_copy_capture_cursor_session_v1 {
|
||||
struct wl_resource *resource;
|
||||
struct wlr_ext_image_capture_source_v1_cursor *source;
|
||||
bool capture_session_created;
|
||||
|
||||
struct {
|
||||
bool entered;
|
||||
int32_t x, y;
|
||||
struct {
|
||||
int32_t x, y;
|
||||
} hotspot;
|
||||
} prev;
|
||||
|
||||
struct wl_listener source_destroy;
|
||||
struct wl_listener source_update;
|
||||
};
|
||||
|
||||
static const struct ext_image_copy_capture_frame_v1_interface frame_impl;
|
||||
static const struct ext_image_copy_capture_session_v1_interface session_impl;
|
||||
static const struct ext_image_copy_capture_cursor_session_v1_interface cursor_session_impl;
|
||||
|
||||
static struct wlr_ext_image_copy_capture_frame_v1 *frame_from_resource(
|
||||
struct wl_resource *resource) {
|
||||
assert(wl_resource_instance_of(resource,
|
||||
&ext_image_copy_capture_frame_v1_interface, &frame_impl));
|
||||
return wl_resource_get_user_data(resource);
|
||||
}
|
||||
|
||||
static struct wlr_ext_image_copy_capture_session_v1 *session_from_resource(
|
||||
struct wl_resource *resource) {
|
||||
assert(wl_resource_instance_of(resource,
|
||||
&ext_image_copy_capture_session_v1_interface, &session_impl));
|
||||
return wl_resource_get_user_data(resource);
|
||||
}
|
||||
|
||||
static struct wlr_ext_image_copy_capture_cursor_session_v1 *cursor_session_from_resource(
|
||||
struct wl_resource *resource) {
|
||||
assert(wl_resource_instance_of(resource,
|
||||
&ext_image_copy_capture_cursor_session_v1_interface, &cursor_session_impl));
|
||||
return wl_resource_get_user_data(resource);
|
||||
}
|
||||
|
||||
static void frame_destroy(struct wlr_ext_image_copy_capture_frame_v1 *frame) {
|
||||
if (frame == NULL) {
|
||||
return;
|
||||
}
|
||||
wl_signal_emit_mutable(&frame->events.destroy, NULL);
|
||||
wl_resource_set_user_data(frame->resource, NULL);
|
||||
wlr_buffer_unlock(frame->buffer);
|
||||
pixman_region32_fini(&frame->buffer_damage);
|
||||
if (frame->session->frame == frame) {
|
||||
frame->session->frame = NULL;
|
||||
}
|
||||
free(frame);
|
||||
}
|
||||
|
||||
static void frame_handle_resource_destroy(struct wl_resource *resource) {
|
||||
frame_destroy(wl_resource_get_user_data(resource));
|
||||
}
|
||||
|
||||
void wlr_ext_image_copy_capture_frame_v1_ready(struct wlr_ext_image_copy_capture_frame_v1 *frame,
|
||||
enum wl_output_transform transform,
|
||||
const struct timespec *presentation_time) {
|
||||
assert(frame->capturing);
|
||||
|
||||
int rects_len = 0;
|
||||
const pixman_box32_t *rects =
|
||||
pixman_region32_rectangles(&frame->session->damage, &rects_len);
|
||||
for (int i = 0; i < rects_len; i++) {
|
||||
const pixman_box32_t *rect = &rects[i];
|
||||
ext_image_copy_capture_frame_v1_send_damage(frame->resource,
|
||||
rect->x1, rect->y1, rect->x2 - rect->x1, rect->y2 - rect->y1);
|
||||
}
|
||||
|
||||
ext_image_copy_capture_frame_v1_send_transform(frame->resource, transform);
|
||||
ext_image_copy_capture_frame_v1_send_presentation_time(frame->resource,
|
||||
presentation_time->tv_sec >> 32, presentation_time->tv_sec,
|
||||
presentation_time->tv_nsec);
|
||||
ext_image_copy_capture_frame_v1_send_ready(frame->resource);
|
||||
frame_destroy(frame);
|
||||
}
|
||||
|
||||
static bool copy_dmabuf(struct wlr_buffer *dst,
|
||||
struct wlr_buffer *src, struct wlr_renderer *renderer,
|
||||
const pixman_region32_t *clip) {
|
||||
struct wlr_texture *texture = wlr_texture_from_buffer(renderer, src);
|
||||
if (texture == NULL) {
|
||||
return false;
|
||||
}
|
||||
|
||||
bool ok = false;
|
||||
struct wlr_render_pass *pass = wlr_renderer_begin_buffer_pass(renderer, dst, NULL);
|
||||
if (!pass) {
|
||||
goto out;
|
||||
}
|
||||
|
||||
wlr_render_pass_add_texture(pass, &(struct wlr_render_texture_options) {
|
||||
.texture = texture,
|
||||
.clip = clip,
|
||||
.blend_mode = WLR_RENDER_BLEND_MODE_NONE,
|
||||
});
|
||||
|
||||
ok = wlr_render_pass_submit(pass);
|
||||
|
||||
out:
|
||||
wlr_texture_destroy(texture);
|
||||
return ok;
|
||||
}
|
||||
|
||||
static bool copy_shm(void *data, uint32_t format, size_t stride,
|
||||
struct wlr_buffer *src, struct wlr_renderer *renderer) {
|
||||
// TODO: bypass renderer if source buffer supports data ptr access
|
||||
struct wlr_texture *texture = wlr_texture_from_buffer(renderer, src);
|
||||
if (!texture) {
|
||||
return false;
|
||||
}
|
||||
|
||||
// TODO: only copy damaged region
|
||||
bool ok = wlr_texture_read_pixels(texture, &(struct wlr_texture_read_pixels_options){
|
||||
.data = data,
|
||||
.format = format,
|
||||
.stride = stride,
|
||||
});
|
||||
|
||||
wlr_texture_destroy(texture);
|
||||
|
||||
return ok;
|
||||
}
|
||||
|
||||
bool wlr_ext_image_copy_capture_frame_v1_copy_buffer(struct wlr_ext_image_copy_capture_frame_v1 *frame,
|
||||
struct wlr_buffer *src, struct wlr_renderer *renderer) {
|
||||
struct wlr_buffer *dst = frame->buffer;
|
||||
|
||||
if (src->width != dst->width || src->height != dst->height) {
|
||||
wlr_ext_image_copy_capture_frame_v1_fail(frame,
|
||||
EXT_IMAGE_COPY_CAPTURE_FRAME_V1_FAILURE_REASON_BUFFER_CONSTRAINTS);
|
||||
return false;
|
||||
}
|
||||
|
||||
bool ok = false;
|
||||
enum ext_image_copy_capture_frame_v1_failure_reason failure_reason =
|
||||
EXT_IMAGE_COPY_CAPTURE_FRAME_V1_FAILURE_REASON_UNKNOWN;
|
||||
struct wlr_dmabuf_attributes dmabuf;
|
||||
void *data;
|
||||
uint32_t format;
|
||||
size_t stride;
|
||||
if (wlr_buffer_get_dmabuf(dst, &dmabuf)) {
|
||||
if (frame->session->source->dmabuf_formats.len == 0) {
|
||||
ok = false;
|
||||
failure_reason = EXT_IMAGE_COPY_CAPTURE_FRAME_V1_FAILURE_REASON_BUFFER_CONSTRAINTS;
|
||||
} else {
|
||||
ok = copy_dmabuf(dst, src, renderer, &frame->buffer_damage);
|
||||
}
|
||||
} else if (wlr_buffer_begin_data_ptr_access(dst,
|
||||
WLR_BUFFER_DATA_PTR_ACCESS_WRITE, &data, &format, &stride)) {
|
||||
if (frame->session->source->shm_formats_len == 0) {
|
||||
ok = false;
|
||||
failure_reason = EXT_IMAGE_COPY_CAPTURE_FRAME_V1_FAILURE_REASON_BUFFER_CONSTRAINTS;
|
||||
} else {
|
||||
ok = copy_shm(data, format, stride, src, renderer);
|
||||
}
|
||||
wlr_buffer_end_data_ptr_access(dst);
|
||||
}
|
||||
if (!ok) {
|
||||
wlr_ext_image_copy_capture_frame_v1_fail(frame, failure_reason);
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void wlr_ext_image_copy_capture_frame_v1_fail(struct wlr_ext_image_copy_capture_frame_v1 *frame,
|
||||
enum ext_image_copy_capture_frame_v1_failure_reason reason) {
|
||||
ext_image_copy_capture_frame_v1_send_failed(frame->resource, reason);
|
||||
frame_destroy(frame);
|
||||
}
|
||||
|
||||
static void frame_handle_destroy(struct wl_client *client,
|
||||
struct wl_resource *frame_resource) {
|
||||
wl_resource_destroy(frame_resource);
|
||||
}
|
||||
|
||||
static void frame_handle_attach_buffer(struct wl_client *client,
|
||||
struct wl_resource *frame_resource, struct wl_resource *buffer_resource) {
|
||||
struct wlr_ext_image_copy_capture_frame_v1 *frame = frame_from_resource(frame_resource);
|
||||
if (frame == NULL) {
|
||||
return;
|
||||
}
|
||||
|
||||
if (frame->capturing) {
|
||||
wl_resource_post_error(frame->resource,
|
||||
EXT_IMAGE_COPY_CAPTURE_FRAME_V1_ERROR_ALREADY_CAPTURED,
|
||||
"attach_buffer sent after capture");
|
||||
return;
|
||||
}
|
||||
|
||||
struct wlr_buffer *buffer = wlr_buffer_try_from_resource(buffer_resource);
|
||||
if (buffer == NULL) {
|
||||
wl_resource_post_no_memory(frame_resource);
|
||||
return;
|
||||
}
|
||||
|
||||
wlr_buffer_unlock(frame->buffer);
|
||||
frame->buffer = buffer;
|
||||
}
|
||||
|
||||
static void frame_handle_damage_buffer(struct wl_client *client,
|
||||
struct wl_resource *frame_resource, int32_t x, int32_t y,
|
||||
int32_t width, int32_t height) {
|
||||
struct wlr_ext_image_copy_capture_frame_v1 *frame = frame_from_resource(frame_resource);
|
||||
if (frame == NULL) {
|
||||
return;
|
||||
}
|
||||
|
||||
if (frame->capturing) {
|
||||
wl_resource_post_error(frame->resource,
|
||||
EXT_IMAGE_COPY_CAPTURE_FRAME_V1_ERROR_ALREADY_CAPTURED,
|
||||
"damage_buffer sent after capture");
|
||||
return;
|
||||
}
|
||||
|
||||
if (x < 0 || y < 0 || width <= 0 || height <= 0) {
|
||||
wl_resource_post_error(frame->resource,
|
||||
EXT_IMAGE_COPY_CAPTURE_FRAME_V1_ERROR_INVALID_BUFFER_DAMAGE,
|
||||
"Invalid buffer damage coordinates");
|
||||
return;
|
||||
}
|
||||
|
||||
pixman_region32_union_rect(&frame->buffer_damage, &frame->buffer_damage,
|
||||
x, y, width, height);
|
||||
}
|
||||
|
||||
static void frame_handle_capture(struct wl_client *client,
|
||||
struct wl_resource *frame_resource) {
|
||||
struct wlr_ext_image_copy_capture_frame_v1 *frame = frame_from_resource(frame_resource);
|
||||
if (frame == NULL) {
|
||||
return;
|
||||
}
|
||||
|
||||
if (frame->capturing) {
|
||||
wl_resource_post_error(frame->resource,
|
||||
EXT_IMAGE_COPY_CAPTURE_FRAME_V1_ERROR_ALREADY_CAPTURED,
|
||||
"capture sent twice");
|
||||
return;
|
||||
}
|
||||
|
||||
if (frame->buffer == NULL) {
|
||||
wl_resource_post_error(frame->resource,
|
||||
EXT_IMAGE_COPY_CAPTURE_FRAME_V1_ERROR_NO_BUFFER,
|
||||
"No buffer attached");
|
||||
return;
|
||||
}
|
||||
|
||||
frame->capturing = true;
|
||||
|
||||
bool need_frame = pixman_region32_not_empty(&frame->session->damage);
|
||||
struct wlr_ext_image_capture_source_v1 *source = frame->session->source;
|
||||
if (need_frame && source->impl->schedule_frame) {
|
||||
source->impl->schedule_frame(source);
|
||||
}
|
||||
}
|
||||
|
||||
static const struct ext_image_copy_capture_frame_v1_interface frame_impl = {
|
||||
.destroy = frame_handle_destroy,
|
||||
.attach_buffer = frame_handle_attach_buffer,
|
||||
.damage_buffer = frame_handle_damage_buffer,
|
||||
.capture = frame_handle_capture,
|
||||
};
|
||||
|
||||
static void session_handle_destroy(struct wl_client *client,
|
||||
struct wl_resource *session_resource) {
|
||||
wl_resource_destroy(session_resource);
|
||||
}
|
||||
|
||||
static void session_handle_create_frame(struct wl_client *client,
|
||||
struct wl_resource *session_resource, uint32_t new_id) {
|
||||
struct wlr_ext_image_copy_capture_session_v1 *session = session_from_resource(session_resource);
|
||||
|
||||
if (session->frame != NULL) {
|
||||
wl_resource_post_error(session_resource,
|
||||
EXT_IMAGE_COPY_CAPTURE_SESSION_V1_ERROR_DUPLICATE_FRAME,
|
||||
"session already has a frame object");
|
||||
return;
|
||||
}
|
||||
|
||||
uint32_t version = wl_resource_get_version(session_resource);
|
||||
struct wl_resource *frame_resource = wl_resource_create(client,
|
||||
&ext_image_copy_capture_frame_v1_interface, version, new_id);
|
||||
if (frame_resource == NULL) {
|
||||
wl_resource_post_no_memory(frame_resource);
|
||||
return;
|
||||
}
|
||||
wl_resource_set_implementation(frame_resource, &frame_impl, NULL,
|
||||
frame_handle_resource_destroy);
|
||||
|
||||
if (session == NULL) {
|
||||
ext_image_copy_capture_frame_v1_send_failed(frame_resource,
|
||||
EXT_IMAGE_COPY_CAPTURE_FRAME_V1_FAILURE_REASON_STOPPED);
|
||||
return;
|
||||
}
|
||||
|
||||
struct wlr_ext_image_copy_capture_frame_v1 *frame = calloc(1, sizeof(*frame));
|
||||
if (frame == NULL) {
|
||||
wl_resource_post_no_memory(session_resource);
|
||||
return;
|
||||
}
|
||||
|
||||
frame->resource = frame_resource;
|
||||
frame->session = session;
|
||||
pixman_region32_init(&frame->buffer_damage);
|
||||
wl_signal_init(&frame->events.destroy);
|
||||
|
||||
wl_resource_set_user_data(frame_resource, frame);
|
||||
|
||||
session->frame = frame;
|
||||
}
|
||||
|
||||
static const struct ext_image_copy_capture_session_v1_interface session_impl = {
|
||||
.destroy = session_handle_destroy,
|
||||
.create_frame = session_handle_create_frame,
|
||||
};
|
||||
|
||||
static void session_send_constraints(struct wlr_ext_image_copy_capture_session_v1 *session) {
|
||||
struct wlr_ext_image_capture_source_v1 *source = session->source;
|
||||
|
||||
ext_image_copy_capture_session_v1_send_buffer_size(session->resource,
|
||||
source->width, source->height);
|
||||
|
||||
for (size_t i = 0; i < source->shm_formats_len; i++) {
|
||||
ext_image_copy_capture_session_v1_send_shm_format(session->resource,
|
||||
source->shm_formats[i]);
|
||||
}
|
||||
|
||||
if (source->dmabuf_formats.len > 0) {
|
||||
struct wl_array dev_id_array = {
|
||||
.data = &source->dmabuf_device,
|
||||
.size = sizeof(source->dmabuf_device),
|
||||
};
|
||||
ext_image_copy_capture_session_v1_send_dmabuf_device(session->resource,
|
||||
&dev_id_array);
|
||||
}
|
||||
for (size_t i = 0; i < source->dmabuf_formats.len; i++) {
|
||||
struct wlr_drm_format *fmt = &source->dmabuf_formats.formats[i];
|
||||
struct wl_array modifiers_array = {
|
||||
.data = fmt->modifiers,
|
||||
.size = fmt->len * sizeof(fmt->modifiers[0]),
|
||||
};
|
||||
ext_image_copy_capture_session_v1_send_dmabuf_format(session->resource,
|
||||
fmt->format, &modifiers_array);
|
||||
}
|
||||
|
||||
ext_image_copy_capture_session_v1_send_done(session->resource);
|
||||
}
|
||||
|
||||
static void session_destroy(struct wlr_ext_image_copy_capture_session_v1 *session) {
|
||||
if (session == NULL) {
|
||||
return;
|
||||
}
|
||||
|
||||
if (session->frame != NULL) {
|
||||
wlr_ext_image_copy_capture_frame_v1_fail(session->frame,
|
||||
EXT_IMAGE_COPY_CAPTURE_FRAME_V1_FAILURE_REASON_STOPPED);
|
||||
}
|
||||
|
||||
if (session->source->impl->stop) {
|
||||
session->source->impl->stop(session->source);
|
||||
}
|
||||
|
||||
ext_image_copy_capture_session_v1_send_stopped(session->resource);
|
||||
wl_resource_set_user_data(session->resource, NULL);
|
||||
|
||||
pixman_region32_fini(&session->damage);
|
||||
wl_list_remove(&session->source_destroy.link);
|
||||
wl_list_remove(&session->source_constraints_update.link);
|
||||
wl_list_remove(&session->source_frame.link);
|
||||
free(session);
|
||||
}
|
||||
|
||||
static void session_handle_source_destroy(struct wl_listener *listener, void *data) {
|
||||
struct wlr_ext_image_copy_capture_session_v1 *session = wl_container_of(listener, session, source_destroy);
|
||||
session_destroy(session);
|
||||
}
|
||||
|
||||
static void session_handle_source_constraints_update(struct wl_listener *listener,
|
||||
void *data) {
|
||||
struct wlr_ext_image_copy_capture_session_v1 *session =
|
||||
wl_container_of(listener, session, source_constraints_update);
|
||||
session_send_constraints(session);
|
||||
}
|
||||
|
||||
static void session_handle_source_frame(struct wl_listener *listener, void *data) {
|
||||
struct wlr_ext_image_copy_capture_session_v1 *session = wl_container_of(listener, session, source_frame);
|
||||
struct wlr_ext_image_capture_source_v1_frame_event *event = data;
|
||||
|
||||
pixman_region32_union(&session->damage, &session->damage, event->damage);
|
||||
|
||||
struct wlr_ext_image_copy_capture_frame_v1 *frame = session->frame;
|
||||
if (frame != NULL && frame->capturing &&
|
||||
pixman_region32_not_empty(&session->damage)) {
|
||||
pixman_region32_union(&frame->buffer_damage,
|
||||
&frame->buffer_damage, &session->damage);
|
||||
|
||||
struct wlr_ext_image_capture_source_v1 *source = frame->session->source;
|
||||
source->impl->copy_frame(source, frame, event);
|
||||
|
||||
// TODO: don't clear on copy error
|
||||
pixman_region32_clear(&session->damage);
|
||||
}
|
||||
}
|
||||
|
||||
static void session_handle_resource_destroy(struct wl_resource *resource) {
|
||||
session_destroy(session_from_resource(resource));
|
||||
}
|
||||
|
||||
static void session_create(struct wl_resource *parent_resource, uint32_t new_id,
|
||||
struct wlr_ext_image_capture_source_v1 *source, uint32_t options) {
|
||||
struct wl_client *client = wl_resource_get_client(parent_resource);
|
||||
uint32_t version = wl_resource_get_version(parent_resource);
|
||||
struct wl_resource *session_resource = wl_resource_create(client,
|
||||
&ext_image_copy_capture_session_v1_interface, version, new_id);
|
||||
if (session_resource == NULL) {
|
||||
wl_resource_post_no_memory(parent_resource);
|
||||
return;
|
||||
}
|
||||
wl_resource_set_implementation(session_resource, &session_impl, NULL,
|
||||
session_handle_resource_destroy);
|
||||
|
||||
if (source == NULL) {
|
||||
ext_image_copy_capture_session_v1_send_stopped(session_resource);
|
||||
return;
|
||||
}
|
||||
|
||||
struct wlr_ext_image_copy_capture_session_v1 *session = calloc(1, sizeof(*session));
|
||||
if (session == NULL) {
|
||||
wl_resource_post_no_memory(parent_resource);
|
||||
return;
|
||||
}
|
||||
|
||||
if (source->impl->start) {
|
||||
source->impl->start(source, options & EXT_IMAGE_COPY_CAPTURE_MANAGER_V1_OPTIONS_PAINT_CURSORS);
|
||||
}
|
||||
|
||||
session->resource = session_resource;
|
||||
session->source = source;
|
||||
pixman_region32_init_rect(&session->damage, 0, 0, source->width,
|
||||
source->height);
|
||||
|
||||
session->source_destroy.notify = session_handle_source_destroy;
|
||||
wl_signal_add(&source->events.destroy, &session->source_destroy);
|
||||
|
||||
session->source_constraints_update.notify = session_handle_source_constraints_update;
|
||||
wl_signal_add(&source->events.constraints_update, &session->source_constraints_update);
|
||||
|
||||
session->source_frame.notify = session_handle_source_frame;
|
||||
wl_signal_add(&source->events.frame, &session->source_frame);
|
||||
|
||||
wl_resource_set_user_data(session_resource, session);
|
||||
session_send_constraints(session);
|
||||
}
|
||||
|
||||
static void cursor_session_destroy(struct wlr_ext_image_copy_capture_cursor_session_v1 *cursor_session) {
|
||||
if (cursor_session == NULL) {
|
||||
return;
|
||||
}
|
||||
if (cursor_session->source->entered) {
|
||||
ext_image_copy_capture_cursor_session_v1_send_leave(cursor_session->resource);
|
||||
}
|
||||
wl_resource_set_user_data(cursor_session->resource, NULL);
|
||||
wl_list_remove(&cursor_session->source_destroy.link);
|
||||
wl_list_remove(&cursor_session->source_update.link);
|
||||
free(cursor_session);
|
||||
}
|
||||
|
||||
static void cursor_session_handle_destroy(struct wl_client *client,
|
||||
struct wl_resource *cursor_session_resource) {
|
||||
wl_resource_destroy(cursor_session_resource);
|
||||
}
|
||||
|
||||
static void cursor_session_handle_get_capture_session(struct wl_client *client,
|
||||
struct wl_resource *cursor_session_resource, uint32_t new_id) {
|
||||
struct wlr_ext_image_copy_capture_cursor_session_v1 *cursor_session =
|
||||
cursor_session_from_resource(cursor_session_resource);
|
||||
|
||||
if (cursor_session != NULL && cursor_session->capture_session_created) {
|
||||
wl_resource_post_error(cursor_session_resource,
|
||||
EXT_IMAGE_COPY_CAPTURE_CURSOR_SESSION_V1_ERROR_DUPLICATE_SESSION,
|
||||
"get_capture_session sent twice");
|
||||
return;
|
||||
}
|
||||
|
||||
struct wlr_ext_image_capture_source_v1 *source = NULL;
|
||||
if (cursor_session != NULL) {
|
||||
source = &cursor_session->source->base;
|
||||
}
|
||||
|
||||
session_create(cursor_session_resource, new_id, source, 0);
|
||||
}
|
||||
|
||||
static const struct ext_image_copy_capture_cursor_session_v1_interface cursor_session_impl = {
|
||||
.destroy = cursor_session_handle_destroy,
|
||||
.get_capture_session = cursor_session_handle_get_capture_session,
|
||||
};
|
||||
|
||||
static void cursor_session_handle_resource_destroy(struct wl_resource *resource) {
|
||||
struct wlr_ext_image_copy_capture_cursor_session_v1 *cursor_session =
|
||||
cursor_session_from_resource(resource);
|
||||
cursor_session_destroy(cursor_session);
|
||||
}
|
||||
|
||||
static void cursor_session_handle_source_destroy(struct wl_listener *listener,
|
||||
void *data) {
|
||||
struct wlr_ext_image_copy_capture_cursor_session_v1 *cursor_session =
|
||||
wl_container_of(listener, cursor_session, source_destroy);
|
||||
cursor_session_destroy(cursor_session);
|
||||
}
|
||||
|
||||
static void cursor_session_update(
|
||||
struct wlr_ext_image_copy_capture_cursor_session_v1 *cursor_session) {
|
||||
struct wlr_ext_image_capture_source_v1_cursor *cursor_source = cursor_session->source;
|
||||
|
||||
if (cursor_source->entered && !cursor_session->prev.entered) {
|
||||
ext_image_copy_capture_cursor_session_v1_send_enter(cursor_session->resource);
|
||||
}
|
||||
if (!cursor_source->entered && cursor_session->prev.entered) {
|
||||
ext_image_copy_capture_cursor_session_v1_send_leave(cursor_session->resource);
|
||||
}
|
||||
|
||||
if (cursor_source->x != cursor_session->prev.x ||
|
||||
cursor_source->y != cursor_session->prev.y) {
|
||||
ext_image_copy_capture_cursor_session_v1_send_position(cursor_session->resource,
|
||||
cursor_source->x, cursor_source->y);
|
||||
}
|
||||
|
||||
if (cursor_source->hotspot.x != cursor_session->prev.hotspot.x ||
|
||||
cursor_source->hotspot.y != cursor_session->prev.hotspot.y) {
|
||||
ext_image_copy_capture_cursor_session_v1_send_hotspot(cursor_session->resource,
|
||||
cursor_source->hotspot.x, cursor_source->hotspot.y);
|
||||
}
|
||||
|
||||
cursor_session->prev.entered = cursor_source->entered;
|
||||
cursor_session->prev.x = cursor_source->x;
|
||||
cursor_session->prev.y = cursor_source->y;
|
||||
cursor_session->prev.hotspot.y = cursor_source->hotspot.y;
|
||||
cursor_session->prev.hotspot.y = cursor_source->hotspot.y;
|
||||
}
|
||||
|
||||
static void cursor_session_handle_source_update(struct wl_listener *listener,
|
||||
void *data) {
|
||||
struct wlr_ext_image_copy_capture_cursor_session_v1 *cursor_session =
|
||||
wl_container_of(listener, cursor_session, source_update);
|
||||
cursor_session_update(cursor_session);
|
||||
}
|
||||
|
||||
static void manager_handle_create_session(struct wl_client *client,
|
||||
struct wl_resource *manager_resource, uint32_t new_id,
|
||||
struct wl_resource *source_resource, uint32_t options) {
|
||||
struct wlr_ext_image_capture_source_v1 *source =
|
||||
wlr_ext_image_capture_source_v1_from_resource(source_resource);
|
||||
session_create(manager_resource, new_id, source, options);
|
||||
}
|
||||
|
||||
static void manager_handle_create_pointer_cursor_session(struct wl_client *client,
|
||||
struct wl_resource *manager_resource, uint32_t new_id,
|
||||
struct wl_resource *source_resource, struct wl_resource *pointer_resource) {
|
||||
struct wlr_ext_image_capture_source_v1 *source = wlr_ext_image_capture_source_v1_from_resource(source_resource);
|
||||
struct wlr_seat_client *seat_client = wlr_seat_client_from_pointer_resource(pointer_resource);
|
||||
|
||||
struct wlr_seat *seat = NULL;
|
||||
if (seat_client != NULL) {
|
||||
seat = seat_client->seat;
|
||||
}
|
||||
|
||||
struct wlr_ext_image_capture_source_v1_cursor *source_cursor = NULL;
|
||||
if (source != NULL && seat != NULL && source->impl->get_pointer_cursor) {
|
||||
source_cursor = source->impl->get_pointer_cursor(source, seat);
|
||||
}
|
||||
|
||||
uint32_t version = wl_resource_get_version(manager_resource);
|
||||
struct wl_resource *cursor_session_resource = wl_resource_create(client,
|
||||
&ext_image_copy_capture_cursor_session_v1_interface, version, new_id);
|
||||
if (cursor_session_resource == NULL) {
|
||||
wl_resource_post_no_memory(manager_resource);
|
||||
return;
|
||||
}
|
||||
wl_resource_set_implementation(cursor_session_resource, &cursor_session_impl, NULL,
|
||||
cursor_session_handle_resource_destroy);
|
||||
|
||||
if (source_cursor == NULL) {
|
||||
return; // leave inert
|
||||
}
|
||||
|
||||
struct wlr_ext_image_copy_capture_cursor_session_v1 *cursor_session = calloc(1, sizeof(*cursor_session));
|
||||
if (cursor_session == NULL) {
|
||||
wl_resource_post_no_memory(manager_resource);
|
||||
return;
|
||||
}
|
||||
|
||||
cursor_session->resource = cursor_session_resource;
|
||||
cursor_session->source = source_cursor;
|
||||
|
||||
cursor_session->source_destroy.notify = cursor_session_handle_source_destroy;
|
||||
wl_signal_add(&source_cursor->base.events.destroy, &cursor_session->source_destroy);
|
||||
|
||||
cursor_session->source_update.notify = cursor_session_handle_source_update;
|
||||
wl_signal_add(&source_cursor->events.update, &cursor_session->source_update);
|
||||
|
||||
wl_resource_set_user_data(cursor_session_resource, cursor_session);
|
||||
|
||||
cursor_session_update(cursor_session);
|
||||
}
|
||||
|
||||
static void manager_handle_destroy(struct wl_client *client,
|
||||
struct wl_resource *manager_resource) {
|
||||
wl_resource_destroy(manager_resource);
|
||||
}
|
||||
|
||||
static const struct ext_image_copy_capture_manager_v1_interface manager_impl = {
|
||||
.create_session = manager_handle_create_session,
|
||||
.create_pointer_cursor_session = manager_handle_create_pointer_cursor_session,
|
||||
.destroy = manager_handle_destroy,
|
||||
};
|
||||
|
||||
static void manager_bind(struct wl_client *client, void *data,
|
||||
uint32_t version, uint32_t id) {
|
||||
struct wl_resource *resource = wl_resource_create(client,
|
||||
&ext_image_copy_capture_manager_v1_interface, version, id);
|
||||
if (!resource) {
|
||||
wl_client_post_no_memory(client);
|
||||
return;
|
||||
}
|
||||
wl_resource_set_implementation(resource, &manager_impl, NULL, NULL);
|
||||
}
|
||||
|
||||
static void manager_handle_display_destroy(struct wl_listener *listener, void *data) {
|
||||
struct wlr_ext_image_copy_capture_manager_v1 *manager =
|
||||
wl_container_of(listener, manager, display_destroy);
|
||||
wl_list_remove(&manager->display_destroy.link);
|
||||
wl_global_destroy(manager->global);
|
||||
free(manager);
|
||||
}
|
||||
|
||||
struct wlr_ext_image_copy_capture_manager_v1 *wlr_ext_image_copy_capture_manager_v1_create(
|
||||
struct wl_display *display, uint32_t version) {
|
||||
assert(version <= IMAGE_COPY_CAPTURE_MANAGER_V1_VERSION);
|
||||
|
||||
struct wlr_ext_image_copy_capture_manager_v1 *manager = calloc(1, sizeof(*manager));
|
||||
if (manager == NULL) {
|
||||
return NULL;
|
||||
}
|
||||
|
||||
manager->global = wl_global_create(display,
|
||||
&ext_image_copy_capture_manager_v1_interface, version, manager, manager_bind);
|
||||
if (manager->global == NULL) {
|
||||
free(manager);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
manager->display_destroy.notify = manager_handle_display_destroy;
|
||||
wl_display_add_destroy_listener(display, &manager->display_destroy);
|
||||
|
||||
return manager;
|
||||
}
|
Loading…
Reference in new issue