blob: 2be0a7e30cc0592a1b8a006067129aee48854a04 [file] [log] [blame]
/*
* Copyright © 2011 Benjamin Franzke
* Copyright © 2010, 2013 Intel Corporation
*
* Permission is hereby granted, free of charge, to any person obtaining a
* copy of this software and associated documentation files (the "Software"),
* to deal in the Software without restriction, including without limitation
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
* and/or sell copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice (including the next
* paragraph) shall be included in all copies or substantial portions of the
* Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
* DEALINGS IN THE SOFTWARE.
*/
#include "config.h"
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <stdbool.h>
#include <assert.h>
#include <unistd.h>
#include <sys/mman.h>
#include <signal.h>
#include <time.h>
#include <sys/poll.h>
#include <float.h>
#include <math.h>
#include <wayland-client.h>
#include "shared/os-compatibility.h"
#include "shared/xalloc.h"
#include "shared/zalloc.h"
struct device {
enum { KEYBOARD, POINTER } type;
int start_time;
int end_time;
struct wl_list link;
union {
struct wl_keyboard *keyboard;
struct wl_pointer *pointer;
} p;
};
struct display {
struct wl_display *display;
struct wl_registry *registry;
struct wl_compositor *compositor;
struct wl_shell *shell;
struct wl_seat *seat;
struct wl_shm *shm;
uint32_t formats;
struct wl_list devices;
};
struct window {
struct display *display;
int width, height;
struct wl_surface *surface;
struct wl_shell_surface *shell_surface;
};
static void
buffer_release(void *data, struct wl_buffer *buffer)
{
wl_buffer_destroy(buffer);
}
static const struct wl_buffer_listener buffer_listener = {
buffer_release
};
static int
attach_buffer(struct window *window, int width, int height)
{
struct wl_shm_pool *pool;
struct wl_buffer *buffer;
int fd, size, stride;
stride = width * 4;
size = stride * height;
fd = os_create_anonymous_file(size);
if (fd < 0) {
fprintf(stderr, "creating a buffer file for %d B failed: %m\n",
size);
return -1;
}
pool = wl_shm_create_pool(window->display->shm, fd, size);
buffer = wl_shm_pool_create_buffer(pool, 0,
width, height,
stride,
WL_SHM_FORMAT_XRGB8888);
wl_surface_attach(window->surface, buffer, 0, 0);
wl_buffer_add_listener(buffer, &buffer_listener, buffer);
wl_shm_pool_destroy(pool);
close(fd);
return 0;
}
static void
handle_ping(void *data, struct wl_shell_surface *shell_surface,
uint32_t serial)
{
wl_shell_surface_pong(shell_surface, serial);
}
static void
handle_configure(void *data, struct wl_shell_surface *shell_surface,
uint32_t edges, int32_t width, int32_t height)
{
}
static void
handle_popup_done(void *data, struct wl_shell_surface *shell_surface)
{
}
static const struct wl_shell_surface_listener shell_surface_listener = {
handle_ping,
handle_configure,
handle_popup_done
};
static struct window *
create_window(struct display *display, int width, int height)
{
struct window *window;
window = xzalloc(sizeof *window);
window->display = display;
window->width = width;
window->height = height;
window->surface = wl_compositor_create_surface(display->compositor);
window->shell_surface = wl_shell_get_shell_surface(display->shell,
window->surface);
if (window->shell_surface)
wl_shell_surface_add_listener(window->shell_surface,
&shell_surface_listener, window);
wl_shell_surface_set_title(window->shell_surface, "simple-shm");
wl_shell_surface_set_toplevel(window->shell_surface);
wl_surface_damage(window->surface, 0, 0, width, height);
attach_buffer(window, width, height);
wl_surface_commit(window->surface);
return window;
}
static void
destroy_window(struct window *window)
{
wl_shell_surface_destroy(window->shell_surface);
wl_surface_destroy(window->surface);
free(window);
}
static void
shm_format(void *data, struct wl_shm *wl_shm, uint32_t format)
{
struct display *d = data;
d->formats |= (1 << format);
}
struct wl_shm_listener shm_listener = {
shm_format
};
static void
registry_handle_global(void *data, struct wl_registry *registry,
uint32_t id, const char *interface, uint32_t version)
{
struct display *d = data;
if (strcmp(interface, "wl_compositor") == 0) {
d->compositor =
wl_registry_bind(registry,
id, &wl_compositor_interface, 1);
} else if (strcmp(interface, "wl_shell") == 0) {
d->shell = wl_registry_bind(registry,
id, &wl_shell_interface, 1);
} else if (strcmp(interface, "wl_shm") == 0) {
d->shm = wl_registry_bind(registry,
id, &wl_shm_interface, 1);
wl_shm_add_listener(d->shm, &shm_listener, d);
} else if (strcmp(interface, "wl_seat") == 0 &&
d->seat == NULL) {
d->seat = wl_registry_bind(registry,
id, &wl_seat_interface, 3);
}
}
static void
registry_handle_global_remove(void *data, struct wl_registry *registry,
uint32_t name)
{
}
static const struct wl_registry_listener registry_listener = {
registry_handle_global,
registry_handle_global_remove
};
static struct display *
create_display(void)
{
struct display *display;
display = xzalloc(sizeof *display);
display->display = wl_display_connect(NULL);
assert(display->display);
display->formats = 0;
display->registry = wl_display_get_registry(display->display);
wl_registry_add_listener(display->registry,
&registry_listener, display);
wl_display_roundtrip(display->display);
if (display->shm == NULL) {
fprintf(stderr, "No wl_shm global\n");
exit(1);
}
wl_display_roundtrip(display->display);
if (!(display->formats & (1 << WL_SHM_FORMAT_XRGB8888))) {
fprintf(stderr, "WL_SHM_FORMAT_XRGB32 not available\n");
exit(1);
}
wl_list_init(&display->devices);
return display;
}
static void
pointer_handle_enter(void *data, struct wl_pointer *pointer,
uint32_t serial, struct wl_surface *surface,
wl_fixed_t sx_w, wl_fixed_t sy_w)
{
}
static void
pointer_handle_leave(void *data, struct wl_pointer *pointer,
uint32_t serial, struct wl_surface *surface)
{
}
static void
pointer_handle_motion(void *data, struct wl_pointer *pointer,
uint32_t time, wl_fixed_t sx_w, wl_fixed_t sy_w)
{
}
static void
pointer_handle_button(void *data, struct wl_pointer *pointer, uint32_t serial,
uint32_t time, uint32_t button, uint32_t state_w)
{
}
static void
pointer_handle_axis(void *data, struct wl_pointer *pointer,
uint32_t time, uint32_t axis, wl_fixed_t value)
{
}
static const struct wl_pointer_listener pointer_listener = {
pointer_handle_enter,
pointer_handle_leave,
pointer_handle_motion,
pointer_handle_button,
pointer_handle_axis,
};
static void
keyboard_handle_keymap(void *data, struct wl_keyboard *keyboard,
uint32_t format, int fd, uint32_t size)
{
}
static void
keyboard_handle_enter(void *data, struct wl_keyboard *keyboard,
uint32_t serial, struct wl_surface *surface,
struct wl_array *keys)
{
}
static void
keyboard_handle_leave(void *data, struct wl_keyboard *keyboard,
uint32_t serial, struct wl_surface *surface)
{
}
static void
keyboard_handle_key(void *data, struct wl_keyboard *keyboard,
uint32_t serial, uint32_t time, uint32_t key,
uint32_t state_w)
{
}
static void
keyboard_handle_modifiers(void *data, struct wl_keyboard *keyboard,
uint32_t serial, uint32_t mods_depressed,
uint32_t mods_latched, uint32_t mods_locked,
uint32_t group)
{
}
static const struct wl_keyboard_listener keyboard_listener = {
keyboard_handle_keymap,
keyboard_handle_enter,
keyboard_handle_leave,
keyboard_handle_key,
keyboard_handle_modifiers,
};
static void
start_device(struct display *display, struct device *device)
{
if (display->seat == NULL)
return;
switch (device->type) {
case KEYBOARD:
if (device->p.keyboard == NULL) {
device->p.keyboard =
wl_seat_get_keyboard(display->seat);
wl_keyboard_add_listener(device->p.keyboard,
&keyboard_listener,
NULL);
}
break;
case POINTER:
if (device->p.pointer == NULL) {
device->p.pointer =
wl_seat_get_pointer(display->seat);
wl_pointer_add_listener(device->p.pointer,
&pointer_listener,
NULL);
}
break;
}
}
static void
destroy_device(struct device *device)
{
switch (device->type) {
case KEYBOARD:
if (device->p.keyboard)
wl_keyboard_release(device->p.keyboard);
break;
case POINTER:
if (device->p.pointer)
wl_pointer_release(device->p.pointer);
break;
}
wl_list_remove(&device->link);
free(device);
}
static void
destroy_devices(struct display *display)
{
struct device *device, *tmp;
wl_list_for_each_safe(device, tmp, &display->devices, link)
destroy_device(device);
}
static void
destroy_display(struct display *display)
{
destroy_devices(display);
if (display->shm)
wl_shm_destroy(display->shm);
if (display->shell)
wl_shell_destroy(display->shell);
if (display->seat)
wl_seat_destroy(display->seat);
if (display->compositor)
wl_compositor_destroy(display->compositor);
wl_registry_destroy(display->registry);
wl_display_flush(display->display);
wl_display_disconnect(display->display);
free(display);
}
static int running = 1;
static void
signal_int(int signum)
{
running = 0;
}
static int
create_device(struct display *display, const char *time_desc, int type)
{
int start_time;
int end_time = -1;
char *tail;
struct device *device;
if (time_desc == NULL) {
fprintf(stderr, "missing time description\n");
return -1;
}
errno = 0;
start_time = strtoul(time_desc, &tail, 10);
if (errno || tail == time_desc)
goto error;
if (*tail == ':') {
time_desc = tail + 1;
end_time = strtoul(time_desc, &tail, 10);
if (errno || tail == time_desc || *tail != '\0')
goto error;
} else if (*tail != '\0') {
goto error;
}
device = xzalloc(sizeof *device);
device->type = type;
device->start_time = start_time;
device->end_time = end_time;
wl_list_insert(&display->devices, &device->link);
return 0;
error:
fprintf(stderr, "invalid time description\n");
return -1;
}
static struct timespec begin_time;
static void
reset_timer(void)
{
clock_gettime(CLOCK_MONOTONIC, &begin_time);
}
static double
read_timer(void)
{
struct timespec t;
clock_gettime(CLOCK_MONOTONIC, &t);
return (double)(t.tv_sec - begin_time.tv_sec) +
1e-9 * (t.tv_nsec - begin_time.tv_nsec);
}
static void
main_loop(struct display *display)
{
reset_timer();
while (running) {
struct device *device, *tmp;
struct pollfd fds[1];
double sleep_time = DBL_MAX;
double now;
if (wl_display_dispatch_pending(display->display) == -1)
break;
if (wl_display_flush(display->display) == -1)
break;
now = read_timer();
wl_list_for_each(device, &display->devices, link) {
double next_time = device->start_time - now;
if (next_time < 0.0) {
sleep_time = 0.0;
break;
} else if (next_time < sleep_time) {
sleep_time = next_time;
}
next_time = device->end_time - now;
if (next_time < 0.0) {
sleep_time = 0.0;
break;
} else if (next_time < sleep_time) {
sleep_time = next_time;
}
}
fds[0].fd = wl_display_get_fd(display->display);
fds[0].events = POLLIN;
fds[0].revents = 0;
poll(fds,
sizeof fds / sizeof fds[0],
sleep_time == DBL_MAX ? -1 : ceil(sleep_time * 1000.0));
if (fds[0].revents &&
wl_display_dispatch(display->display) == -1)
break;
now = read_timer();
wl_list_for_each_safe(device, tmp, &display->devices, link) {
if (device->start_time <= now)
start_device(display, device);
if (device->end_time >= 0 && device->end_time <= now)
destroy_device(device);
}
}
}
int
main(int argc, char **argv)
{
struct sigaction sigint;
struct display *display;
struct window *window;
int i;
display = create_display();
window = create_window(display, 250, 250);
for (i = 1; i < argc; i++) {
if (!strncmp(argv[i], "-p", 2)) {
char *arg;
if (argv[i][2]) {
arg = argv[i] + 2;
} else {
arg = argv[i + 1];
i++;
}
if (create_device(display, arg, POINTER) == -1)
return 1;
} else if (!strncmp(argv[i], "-k", 2)) {
char *arg;
if (argv[i][2]) {
arg = argv[i] + 2;
} else {
arg = argv[i + 1];
i++;
}
if (create_device(display, arg, KEYBOARD) == -1)
return 1;
} else {
fprintf(stderr, "unknown argument %s\n", argv[i]);
return 1;
}
}
sigint.sa_handler = signal_int;
sigemptyset(&sigint.sa_mask);
sigint.sa_flags = SA_RESETHAND;
sigaction(SIGINT, &sigint, NULL);
main_loop(display);
fprintf(stderr, "multi-resource exiting\n");
destroy_window(window);
destroy_display(display);
return 0;
}