#include #include #include #include #include #include #include #include #include #include "tablet-unstable-v2-protocol.h" static const struct wlr_tablet_pad_v2_grab_interface default_pad_grab_interface; struct tablet_pad_auxiliary_user_data { struct wlr_tablet_pad_client_v2 *pad; size_t index; }; static void handle_tablet_pad_v2_destroy(struct wl_client *client, struct wl_resource *resource) { wl_resource_destroy(resource); } static void destroy_tablet_pad_ring_v2(struct wl_resource *resource) { struct tablet_pad_auxiliary_user_data *aux = wl_resource_get_user_data(resource); if (!aux) { return; } aux->pad->rings[aux->index] = NULL; free(aux); wl_resource_set_user_data(resource, NULL); } static void handle_tablet_pad_ring_v2_set_feedback(struct wl_client *client, struct wl_resource *resource, const char *description, uint32_t serial) { struct tablet_pad_auxiliary_user_data *aux = wl_resource_get_user_data(resource); if (!aux) { return; } struct wlr_tablet_v2_event_feedback evt = { .serial = serial, .description = description, .index = aux->index }; wl_signal_emit_mutable(&aux->pad->pad->events.ring_feedback, &evt); } static void handle_tablet_pad_ring_v2_destroy(struct wl_client *client, struct wl_resource *resource) { wl_resource_destroy(resource); } static const struct zwp_tablet_pad_ring_v2_interface tablet_pad_ring_impl = { .set_feedback = handle_tablet_pad_ring_v2_set_feedback, .destroy = handle_tablet_pad_ring_v2_destroy, }; static void destroy_tablet_pad_strip_v2(struct wl_resource *resource) { struct tablet_pad_auxiliary_user_data *aux = wl_resource_get_user_data(resource); if (!aux) { return; } aux->pad->strips[aux->index] = NULL; free(aux); wl_resource_set_user_data(resource, NULL); } static void handle_tablet_pad_strip_v2_set_feedback(struct wl_client *client, struct wl_resource *resource, const char *description, uint32_t serial) { struct tablet_pad_auxiliary_user_data *aux = wl_resource_get_user_data(resource); if (!aux) { return; } struct wlr_tablet_v2_event_feedback evt = { .serial = serial, .description = description, .index = aux->index }; wl_signal_emit_mutable(&aux->pad->pad->events.strip_feedback, &evt); } static void handle_tablet_pad_strip_v2_destroy(struct wl_client *client, struct wl_resource *resource) { wl_resource_destroy(resource); } static const struct zwp_tablet_pad_strip_v2_interface tablet_pad_strip_impl = { .set_feedback = handle_tablet_pad_strip_v2_set_feedback, .destroy = handle_tablet_pad_strip_v2_destroy, }; static void handle_tablet_pad_v2_set_feedback( struct wl_client *client, struct wl_resource *resource, uint32_t button, const char *description, uint32_t serial) { struct wlr_tablet_pad_client_v2 *pad = tablet_pad_client_from_resource(resource); if (!pad) { return; } struct wlr_tablet_v2_event_feedback evt = { .serial = serial, .index = button, .description = description, }; wl_signal_emit_mutable(&pad->pad->events.button_feedback, &evt); } static const struct zwp_tablet_pad_v2_interface tablet_pad_impl = { .set_feedback = handle_tablet_pad_v2_set_feedback, .destroy = handle_tablet_pad_v2_destroy, }; static void destroy_tablet_pad_group_v2(struct wl_resource *resource) { struct tablet_pad_auxiliary_user_data *aux = wl_resource_get_user_data(resource); if (!aux) { return; } aux->pad->groups[aux->index] = NULL; free(aux); wl_resource_set_user_data(resource, NULL); } void destroy_tablet_pad_v2(struct wl_resource *resource) { struct wlr_tablet_pad_client_v2 *pad = tablet_pad_client_from_resource(resource); if (!pad) { return; } wl_list_remove(&pad->seat_link); wl_list_remove(&pad->pad_link); /* This isn't optimal, if the client destroys the resources in another * order, it will be disconnected. * But this makes things *way* easier for us, and (untested) I doubt * clients will destroy it in another order. */ for (size_t i = 0; i < pad->group_count; ++i) { if (pad->groups[i]) { destroy_tablet_pad_group_v2(pad->groups[i]); } } free(pad->groups); for (size_t i = 0; i < pad->ring_count; ++i) { if (pad->rings[i]) { destroy_tablet_pad_ring_v2(pad->rings[i]); } } free(pad->rings); for (size_t i = 0; i < pad->strip_count; ++i) { if (pad->strips[i]) { destroy_tablet_pad_strip_v2(pad->strips[i]); } } free(pad->strips); if (pad->pad->current_client == pad) { pad->pad->current_client = NULL; } free(pad); wl_resource_set_user_data(resource, NULL); } static void handle_tablet_pad_group_v2_destroy(struct wl_client *client, struct wl_resource *resource) { wl_resource_destroy(resource); } static const struct zwp_tablet_pad_group_v2_interface tablet_pad_group_impl = { .destroy = handle_tablet_pad_group_v2_destroy, }; static void add_tablet_pad_group(struct wlr_tablet_v2_tablet_pad *pad, struct wlr_tablet_pad_client_v2 *client, struct wlr_tablet_pad_group *group, size_t index) { uint32_t version = wl_resource_get_version(client->resource); client->groups[index] = wl_resource_create(client->client, &zwp_tablet_pad_group_v2_interface, version, 0); if (!client->groups[index]) { wl_client_post_no_memory(client->client); return; } struct tablet_pad_auxiliary_user_data *user_data = calloc(1, sizeof(*user_data)); if (!user_data) { wl_client_post_no_memory(client->client); return; } user_data->pad = client; user_data->index = index; wl_resource_set_implementation(client->groups[index], &tablet_pad_group_impl, user_data, destroy_tablet_pad_group_v2); zwp_tablet_pad_v2_send_group(client->resource, client->groups[index]); zwp_tablet_pad_group_v2_send_modes(client->groups[index], group->mode_count); struct wl_array button_array; wl_array_init(&button_array); wl_array_add(&button_array, group->button_count * sizeof(int)); memcpy(button_array.data, group->buttons, group->button_count * sizeof(int)); zwp_tablet_pad_group_v2_send_buttons(client->groups[index], &button_array); wl_array_release(&button_array); client->strip_count = group->strip_count; for (size_t i = 0; i < group->strip_count; ++i) { size_t strip = group->strips[i]; struct tablet_pad_auxiliary_user_data *user_data = calloc(1, sizeof(*user_data)); if (!user_data) { wl_client_post_no_memory(client->client); return; } user_data->pad = client; user_data->index = strip; client->strips[strip] = wl_resource_create(client->client, &zwp_tablet_pad_strip_v2_interface, version, 0); if (!client->strips[strip]) { free(user_data); wl_client_post_no_memory(client->client); return; } wl_resource_set_implementation(client->strips[strip], &tablet_pad_strip_impl, user_data, destroy_tablet_pad_strip_v2); zwp_tablet_pad_group_v2_send_strip(client->groups[index], client->strips[strip]); } client->ring_count = group->ring_count; for (size_t i = 0; i < group->ring_count; ++i) { size_t ring = group->rings[i]; struct tablet_pad_auxiliary_user_data *user_data = calloc(1, sizeof(*user_data)); if (!user_data) { wl_client_post_no_memory(client->client); return; } user_data->pad = client; user_data->index = ring; client->rings[ring] = wl_resource_create(client->client, &zwp_tablet_pad_ring_v2_interface, version, 0); if (!client->rings[ring]) { free(user_data); wl_client_post_no_memory(client->client); return; } wl_resource_set_implementation(client->rings[ring], &tablet_pad_ring_impl, user_data, destroy_tablet_pad_ring_v2); zwp_tablet_pad_group_v2_send_ring(client->groups[index], client->rings[ring]); } zwp_tablet_pad_group_v2_send_done(client->groups[index]); } void add_tablet_pad_client(struct wlr_tablet_seat_client_v2 *seat, struct wlr_tablet_v2_tablet_pad *pad) { struct wlr_tablet_pad_client_v2 *client = calloc(1, sizeof(*client)); if (!client) { wl_client_post_no_memory(seat->wl_client); return; } client->pad = pad; client->seat = seat; client->groups = calloc(wl_list_length(&pad->wlr_pad->groups), sizeof(*client->groups)); if (!client->groups) { wl_client_post_no_memory(seat->wl_client); free(client); return; } client->rings = calloc(pad->wlr_pad->ring_count, sizeof(*client->rings)); if (!client->rings) { wl_client_post_no_memory(seat->wl_client); free(client->groups); free(client); return; } client->strips = calloc(pad->wlr_pad->strip_count, sizeof(*client->strips)); if (!client->strips) { wl_client_post_no_memory(seat->wl_client); free(client->groups); free(client->rings); free(client); return; } uint32_t version = wl_resource_get_version(seat->resource); client->resource = wl_resource_create(seat->wl_client, &zwp_tablet_pad_v2_interface, version, 0); if (!client->resource) { wl_client_post_no_memory(seat->wl_client); free(client->groups); free(client->rings); free(client->strips); free(client); return; } wl_resource_set_implementation(client->resource, &tablet_pad_impl, client, destroy_tablet_pad_v2); zwp_tablet_seat_v2_send_pad_added(seat->resource, client->resource); client->client = seat->wl_client; // Send the expected events if (pad->wlr_pad->button_count) { zwp_tablet_pad_v2_send_buttons(client->resource, pad->wlr_pad->button_count); } const char **path_ptr; wl_array_for_each(path_ptr, &pad->wlr_pad->paths) { zwp_tablet_pad_v2_send_path(client->resource, *path_ptr); } size_t i = 0; struct wlr_tablet_pad_group *group; client->group_count = pad->group_count; wl_list_for_each(group, &pad->wlr_pad->groups, link) { add_tablet_pad_group(pad, client, group, i++); } zwp_tablet_pad_v2_send_done(client->resource); wl_list_insert(&seat->pads, &client->seat_link); wl_list_insert(&pad->clients, &client->pad_link); } static void handle_wlr_tablet_pad_destroy(struct wl_listener *listener, void *data) { struct wlr_tablet_v2_tablet_pad *pad = wl_container_of(listener, pad, pad_destroy); struct wlr_tablet_pad_client_v2 *client; struct wlr_tablet_pad_client_v2 *tmp_client; wl_list_for_each_safe(client, tmp_client, &pad->clients, pad_link) { zwp_tablet_pad_v2_send_removed(client->resource); destroy_tablet_pad_v2(client->resource); } wl_list_remove(&pad->clients); wl_list_remove(&pad->link); wl_list_remove(&pad->pad_destroy.link); wl_list_remove(&pad->events.button_feedback.listener_list); wl_list_remove(&pad->events.strip_feedback.listener_list); wl_list_remove(&pad->events.ring_feedback.listener_list); free(pad); } struct wlr_tablet_v2_tablet_pad *wlr_tablet_pad_create( struct wlr_tablet_manager_v2 *manager, struct wlr_seat *wlr_seat, struct wlr_input_device *wlr_device) { assert(wlr_device->type == WLR_INPUT_DEVICE_TABLET_PAD); struct wlr_tablet_seat_v2 *seat = get_or_create_tablet_seat(manager, wlr_seat); if (!seat) { return NULL; } struct wlr_tablet_pad *wlr_pad = wlr_tablet_pad_from_input_device(wlr_device); struct wlr_tablet_v2_tablet_pad *pad = calloc(1, sizeof(*pad)); if (!pad) { return NULL; } pad->default_grab.interface = &default_pad_grab_interface; pad->default_grab.pad = pad; pad->grab = &pad->default_grab; pad->group_count = wl_list_length(&wlr_pad->groups); pad->groups = calloc(pad->group_count, sizeof(uint32_t)); if (!pad->groups) { free(pad); return NULL; } pad->wlr_pad = wlr_pad; wl_list_init(&pad->clients); pad->pad_destroy.notify = handle_wlr_tablet_pad_destroy; wl_signal_add(&wlr_device->events.destroy, &pad->pad_destroy); wl_list_insert(&seat->pads, &pad->link); // We need to create a tablet client for all clients on the seat struct wlr_tablet_seat_client_v2 *pos; wl_list_for_each(pos, &seat->clients, seat_link) { // Tell the clients about the new tool add_tablet_pad_client(pos, pad); } wl_signal_init(&pad->events.button_feedback); wl_signal_init(&pad->events.strip_feedback); wl_signal_init(&pad->events.ring_feedback); return pad; } struct wlr_tablet_pad_client_v2 *tablet_pad_client_from_resource(struct wl_resource *resource) { assert(wl_resource_instance_of(resource, &zwp_tablet_pad_v2_interface, &tablet_pad_impl)); return wl_resource_get_user_data(resource); } /* Actual protocol foo */ uint32_t wlr_send_tablet_v2_tablet_pad_enter( struct wlr_tablet_v2_tablet_pad *pad, struct wlr_tablet_v2_tablet *tablet, struct wlr_surface *surface) { struct wl_client *client = wl_resource_get_client(surface->resource); struct wlr_tablet_client_v2 *tablet_tmp; struct wlr_tablet_client_v2 *tablet_client = NULL; wl_list_for_each(tablet_tmp, &tablet->clients, tablet_link) { if (tablet_tmp->client == client) { tablet_client = tablet_tmp; break; } } // Couldn't find the client binding for the surface's client. Either // the client didn't bind tablet_v2 at all, or not for the relevant // seat if (!tablet_client) { return 0; } struct wlr_tablet_pad_client_v2 *pad_tmp = NULL; struct wlr_tablet_pad_client_v2 *pad_client = NULL; wl_list_for_each(pad_tmp, &pad->clients, pad_link) { if (pad_tmp->client == client) { pad_client = pad_tmp; break; } } // Couldn't find the client binding for the surface's client. Either // the client didn't bind tablet_v2 at all, or not for the relevant // seat if (!pad_client) { return 0; } pad->current_client = pad_client; uint32_t serial = wlr_seat_client_next_serial( pad_client->seat->seat_client); zwp_tablet_pad_v2_send_enter(pad_client->resource, serial, tablet_client->resource, surface->resource); struct timespec now; clock_gettime(CLOCK_MONOTONIC, &now); uint32_t time = now.tv_nsec / 1000; for (size_t i = 0; i < pad->group_count; ++i) { if (pad_client->groups[i]) { zwp_tablet_pad_group_v2_send_mode_switch( pad_client->groups[i], time, serial, pad->groups[i]); } } return serial; } void wlr_send_tablet_v2_tablet_pad_button( struct wlr_tablet_v2_tablet_pad *pad, size_t button, uint32_t time, enum zwp_tablet_pad_v2_button_state state) { if (pad->current_client) { zwp_tablet_pad_v2_send_button(pad->current_client->resource, time, button, state); } } void wlr_send_tablet_v2_tablet_pad_strip(struct wlr_tablet_v2_tablet_pad *pad, uint32_t strip, double position, bool finger, uint32_t time) { if (!pad->current_client || !pad->current_client->strips || !pad->current_client->strips[strip]) { return; } struct wl_resource *resource = pad->current_client->strips[strip]; if (finger) { zwp_tablet_pad_strip_v2_send_source(resource, ZWP_TABLET_PAD_STRIP_V2_SOURCE_FINGER); } if (position < 0) { zwp_tablet_pad_strip_v2_send_stop(resource); } else { zwp_tablet_pad_strip_v2_send_position(resource, position * 65535); } zwp_tablet_pad_strip_v2_send_frame(resource, time); } void wlr_send_tablet_v2_tablet_pad_ring(struct wlr_tablet_v2_tablet_pad *pad, uint32_t ring, double position, bool finger, uint32_t time) { if (!pad->current_client || !pad->current_client->rings || !pad->current_client->rings[ring]) { return; } struct wl_resource *resource = pad->current_client->rings[ring]; if (finger) { zwp_tablet_pad_ring_v2_send_source(resource, ZWP_TABLET_PAD_RING_V2_SOURCE_FINGER); } if (position < 0) { zwp_tablet_pad_ring_v2_send_stop(resource); } else { zwp_tablet_pad_ring_v2_send_angle(resource, wl_fixed_from_double(position)); } zwp_tablet_pad_ring_v2_send_frame(resource, time); } uint32_t wlr_send_tablet_v2_tablet_pad_leave(struct wlr_tablet_v2_tablet_pad *pad, struct wlr_surface *surface) { struct wl_client *client = wl_resource_get_client(surface->resource); if (!pad->current_client || client != pad->current_client->client) { return 0; } uint32_t serial = wlr_seat_client_next_serial( pad->current_client->seat->seat_client); zwp_tablet_pad_v2_send_leave(pad->current_client->resource, serial, surface->resource); return serial; } uint32_t wlr_send_tablet_v2_tablet_pad_mode(struct wlr_tablet_v2_tablet_pad *pad, size_t group, uint32_t mode, uint32_t time) { if (!pad->current_client || !pad->current_client->groups || !pad->current_client->groups[group] ) { return 0; } if (pad->groups[group] == mode) { return 0; } pad->groups[group] = mode; uint32_t serial = wlr_seat_client_next_serial( pad->current_client->seat->seat_client); zwp_tablet_pad_group_v2_send_mode_switch( pad->current_client->groups[group], time, serial, mode); return serial; } bool wlr_surface_accepts_tablet_v2(struct wlr_tablet_v2_tablet *tablet, struct wlr_surface *surface) { struct wl_client *client = wl_resource_get_client(surface->resource); if (tablet->current_client && tablet->current_client->client == client) { return true; } struct wlr_tablet_client_v2 *tablet_tmp; wl_list_for_each(tablet_tmp, &tablet->clients, tablet_link) { if (tablet_tmp->client == client) { return true; } } return false; } uint32_t wlr_tablet_v2_tablet_pad_notify_enter( struct wlr_tablet_v2_tablet_pad *pad, struct wlr_tablet_v2_tablet *tablet, struct wlr_surface *surface) { if (pad->grab && pad->grab->interface->enter) { return pad->grab->interface->enter(pad->grab, tablet, surface); } return 0; } void wlr_tablet_v2_tablet_pad_notify_button( struct wlr_tablet_v2_tablet_pad *pad, size_t button, uint32_t time, enum zwp_tablet_pad_v2_button_state state) { if (pad->grab && pad->grab->interface->button) { pad->grab->interface->button(pad->grab, button, time, state); } } void wlr_tablet_v2_tablet_pad_notify_strip( struct wlr_tablet_v2_tablet_pad *pad, uint32_t strip, double position, bool finger, uint32_t time) { if (pad->grab && pad->grab->interface->strip) { pad->grab->interface->strip(pad->grab, strip, position, finger, time); } } void wlr_tablet_v2_tablet_pad_notify_ring( struct wlr_tablet_v2_tablet_pad *pad, uint32_t ring, double position, bool finger, uint32_t time) { if (pad->grab && pad->grab->interface->ring) { pad->grab->interface->ring(pad->grab, ring, position, finger, time); } } uint32_t wlr_tablet_v2_tablet_pad_notify_leave( struct wlr_tablet_v2_tablet_pad *pad, struct wlr_surface *surface) { if (pad->grab && pad->grab->interface->leave) { return pad->grab->interface->leave(pad->grab, surface); } return 0; } uint32_t wlr_tablet_v2_tablet_pad_notify_mode( struct wlr_tablet_v2_tablet_pad *pad, size_t group, uint32_t mode, uint32_t time) { if (pad->grab && pad->grab->interface->mode) { return pad->grab->interface->mode(pad->grab, group, mode, time); } return 0; } void wlr_tablet_v2_start_grab(struct wlr_tablet_v2_tablet_pad *pad, struct wlr_tablet_pad_v2_grab *grab) { if (grab != &pad->default_grab) { struct wlr_tablet_pad_v2_grab *prev = pad->grab; grab->pad = pad; pad->grab = grab; if (prev && prev->interface->cancel) { prev->interface->cancel(prev); } } } void wlr_tablet_v2_end_grab(struct wlr_tablet_v2_tablet_pad *pad) { struct wlr_tablet_pad_v2_grab *grab = pad->grab; if (grab && grab != &pad->default_grab) { pad->grab = &pad->default_grab; if (grab->interface->cancel) { grab->interface->cancel(grab); } } } static uint32_t default_pad_enter( struct wlr_tablet_pad_v2_grab *grab, struct wlr_tablet_v2_tablet *tablet, struct wlr_surface *surface) { return wlr_send_tablet_v2_tablet_pad_enter(grab->pad, tablet, surface); } static void default_pad_button(struct wlr_tablet_pad_v2_grab *grab,size_t button, uint32_t time, enum zwp_tablet_pad_v2_button_state state) { wlr_send_tablet_v2_tablet_pad_button(grab->pad, button, time, state); } static void default_pad_strip(struct wlr_tablet_pad_v2_grab *grab, uint32_t strip, double position, bool finger, uint32_t time) { wlr_send_tablet_v2_tablet_pad_strip(grab->pad, strip, position, finger, time); } static void default_pad_ring(struct wlr_tablet_pad_v2_grab *grab, uint32_t ring, double position, bool finger, uint32_t time) { wlr_send_tablet_v2_tablet_pad_ring(grab->pad, ring, position, finger, time); } static uint32_t default_pad_leave(struct wlr_tablet_pad_v2_grab *grab, struct wlr_surface *surface) { return wlr_send_tablet_v2_tablet_pad_leave(grab->pad, surface); } static uint32_t default_pad_mode(struct wlr_tablet_pad_v2_grab *grab, size_t group, uint32_t mode, uint32_t time) { return wlr_send_tablet_v2_tablet_pad_mode(grab->pad, group, mode, time); } static void default_pad_cancel(struct wlr_tablet_pad_v2_grab *grab) { // Do nothing, the default cancel can be ignored. } static const struct wlr_tablet_pad_v2_grab_interface default_pad_grab_interface = { .enter = default_pad_enter, .button = default_pad_button, .strip = default_pad_strip, .ring = default_pad_ring, .leave = default_pad_leave, .mode = default_pad_mode, .cancel = default_pad_cancel, };