wayland-system-compositor.c

/*
 * Copyright © 2008 Kristian Høgsberg
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 */

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <stdint.h>
#include <stdarg.h>
#include <termios.h>
#include <sys/ioctl.h>
#include <fcntl.h>
#include <unistd.h>
#include <cairo.h>
#include <gdk-pixbuf/gdk-pixbuf.h>
#include <math.h>
#include <linux/input.h>
#include <linux/vt.h>
#include <xf86drmMode.h>
#include <time.h>

#define LIBUDEV_I_KNOW_THE_API_IS_SUBJECT_TO_CHANGE
#include <libudev.h>

#include <GL/gl.h>
#include <eagle.h>

#include "wayland.h"
#include "wayland-protocol.h"
#include "cairo-util.h"
#include "wayland-system-compositor.h"

#define ARRAY_LENGTH(a) (sizeof (a) / sizeof (a)[0])

struct wlsc_matrix {
	GLdouble d[16];
};

struct wl_visual {
	struct wl_object base;
};

struct wlsc_surface;

struct wlsc_listener {
	struct wl_list link;
	void (*func)(struct wlsc_listener *listener,
		     struct wlsc_surface *surface);
};

struct wlsc_output {
	struct wl_object base;
	struct wl_list link;
	struct wlsc_compositor *compositor;
	struct wlsc_surface *background;
	EGLSurface surface;
	int32_t x, y, width, height;

	drmModeModeInfo *mode;
	uint32_t fb_id;
	uint32_t crtc_id;
	uint32_t connector_id;
};

struct wlsc_input_device {
	struct wl_object base;
	int32_t x, y;
	struct wlsc_compositor *ec;
	struct wlsc_surface *sprite;
	struct wl_list link;

	int grab;
	struct wlsc_surface *grab_surface;
	struct wlsc_surface *pointer_focus;
	struct wlsc_surface *keyboard_focus;
	struct wl_array keys;

	struct wlsc_listener listener;
};

struct wlsc_compositor {
	struct wl_compositor base;
	struct wl_visual argb_visual, premultiplied_argb_visual, rgb_visual;

	EGLDisplay display;
	EGLContext context;
	EGLConfig config;
	struct wl_display *wl_display;

	struct wl_list output_list;
	struct wl_list input_device_list;
	struct wl_list surface_list;

	struct wl_list surface_destroy_listener_list;

	struct wl_event_source *term_signal_source;

        /* tty handling state */
	int tty_fd;
	uint32_t vt_active : 1;

	struct termios terminal_attributes;
	struct wl_event_source *tty_input_source;
	struct wl_event_source *enter_vt_source;
	struct wl_event_source *leave_vt_source;

	struct udev *udev;

	/* Repaint state. */
	struct wl_event_source *timer_source;
	int repaint_needed;
	int repaint_on_timeout;
	struct timespec previous_swap;
	uint32_t current_frame;

	uint32_t meta_state;
	struct wl_list animate_list;
	struct wlsc_surface *primary;
};

#define META_DOWN 256

struct wlsc_animate {
	struct wl_list link;
	void (*animate)(struct wlsc_animate *animate, 
			struct wlsc_compositor *compositor,
			uint32_t frame, uint32_t msecs);
};

struct wlsc_vector {
	GLdouble x, y, z;
};

struct wlsc_surface {
	struct wl_surface base;
	struct wlsc_compositor *compositor;
	struct wl_visual *visual;
	GLuint texture;
	struct wl_map map;
	EGLSurface surface;
	int width, height;
	struct wl_list link;
	struct wlsc_matrix matrix;

	struct wlsc_vector target, current, previous;
	GLdouble target_angle, current_angle, previous_angle;
	struct wlsc_animate animate;
};

static const char *option_background = "background.jpg";

static const GOptionEntry option_entries[] = {
	{ "background", 'b', 0, G_OPTION_ARG_STRING,
	  &option_background, "Background image" },
	{ NULL }
};

struct screenshooter {
	struct wl_object base;
	struct wlsc_compositor *ec;
};

struct screenshooter_interface {
	void (*shoot)(struct wl_client *client, struct screenshooter *shooter);
};

static void
screenshooter_shoot(struct wl_client *client, struct screenshooter *shooter)
{
	struct wlsc_compositor *ec = shooter->ec;
	struct wlsc_output *output;
	char buffer[256];
	GdkPixbuf *pixbuf, *normal;
	GError *error = NULL;
	unsigned char *data;
	int i, j;

	i = 0;
	output = container_of(ec->output_list.next, struct wlsc_output, link);
	while (&output->link != &ec->output_list) {
		snprintf(buffer, sizeof buffer, "wayland-screenshot-%d.png", i++);
		data = malloc(output->width * output->height * 4);
		if (data == NULL) {
			fprintf(stderr, "couldn't allocate image buffer\n");
			continue;
		}

		glReadBuffer(GL_FRONT);
		glPixelStorei(GL_PACK_ALIGNMENT, 1);
		glReadPixels(0, 0, output->width, output->height,
			     GL_RGBA, GL_UNSIGNED_BYTE, data);

		/* FIXME: We should just use a RGB visual for the frontbuffer. */
		for (j = 3; j < output->width * output->height * 4; j += 4)
			data[j] = 0xff;

		pixbuf = gdk_pixbuf_new_from_data(data, GDK_COLORSPACE_RGB, TRUE,
						  8, output->width, output->height, output->width * 4,
						  NULL, NULL);
		normal = gdk_pixbuf_flip(pixbuf, FALSE);
		gdk_pixbuf_save(normal, buffer, "png", &error, NULL);
		gdk_pixbuf_unref(normal);
		gdk_pixbuf_unref(pixbuf);
		free(data);

		output = container_of(output->link.next,
				      struct wlsc_output, link);
	}
}

static const struct wl_message screenshooter_methods[] = {
	{ "shoot", "", NULL }
};

static const struct wl_interface screenshooter_interface = {
	"screenshooter", 1,
	ARRAY_LENGTH(screenshooter_methods),
	screenshooter_methods,
};

struct screenshooter_interface screenshooter_implementation = {
	screenshooter_shoot
};

static struct screenshooter *
screenshooter_create(struct wlsc_compositor *ec)
{
	struct screenshooter *shooter;

	shooter = malloc(sizeof *shooter);
	if (shooter == NULL)
		return NULL;

	shooter->base.interface = &screenshooter_interface;
	shooter->base.implementation = (void(**)(void)) &screenshooter_implementation;
	shooter->ec = ec;

	return shooter;
};
				   
static void
wlsc_matrix_init(struct wlsc_matrix *matrix)
{
	static const struct wlsc_matrix identity = {
		{ 1, 0, 0, 0,  0, 1, 0, 0,  0, 0, 1, 0,  0, 0, 0, 1 }
	};

	memcpy(matrix, &identity, sizeof identity);
}

static void
wlsc_matrix_multiply(struct wlsc_matrix *m, const struct wlsc_matrix *n)
{
	struct wlsc_matrix tmp;
	const GLdouble *row, *column;
	div_t d;
	int i, j;

	for (i = 0; i < 16; i++) {
		tmp.d[i] = 0;
		d = div(i, 4);
		row = m->d + d.quot * 4;
		column = n->d + d.rem;
		for (j = 0; j < 4; j++)
			tmp.d[i] += row[j] * column[j * 4];
	}
	memcpy(m, &tmp, sizeof tmp);
}

static void
wlsc_matrix_translate(struct wlsc_matrix *matrix, GLdouble x, GLdouble y, GLdouble z)
{
	struct wlsc_matrix translate = {
		{ 1, 0, 0, 0,  0, 1, 0, 0,  0, 0, 1, 0,  x, y, z, 1 }
	};

	wlsc_matrix_multiply(matrix, &translate);
}

static void
wlsc_matrix_scale(struct wlsc_matrix *matrix, GLdouble x, GLdouble y, GLdouble z)
{
	struct wlsc_matrix scale = {
		{ x, 0, 0, 0,  0, y, 0, 0,  0, 0, z, 0,  0, 0, 0, 1 }
	};

	wlsc_matrix_multiply(matrix, &scale);
}

static void
wlsc_matrix_rotate(struct wlsc_matrix *matrix,
		   GLdouble angle, GLdouble x, GLdouble y, GLdouble z)
{
	GLdouble c = cos(angle);
	GLdouble s = sin(angle);
	struct wlsc_matrix rotate = {
		{ x * x * (1 - c) + c,     y * x * (1 - c) + z * s, x * z * (1 - c) - y * s, 0,
		  x * y * (1 - c) - z * s, y * y * (1 - c) + c,     y * z * (1 - c) - x * s, 0, 
		  x * z * (1 - c) + y * x, y * z * (1 - c) - x * s, z * z * (1 - c) + c,     0,
		  0, 0, 0, 1 }
	};

	wlsc_matrix_multiply(matrix, &rotate);
}

static void
wlsc_surface_update_matrix(struct wlsc_surface *es)
{
	GLdouble tx, ty;

	tx = es->map.x + es->map.width / 2;
	ty = es->map.y + es->map.height / 2;

	wlsc_matrix_init(&es->matrix);
	wlsc_matrix_translate(&es->matrix, -tx, -ty, 0);
	wlsc_matrix_rotate(&es->matrix, es->current_angle, 0, 1, 0);
	wlsc_matrix_translate(&es->matrix, tx + es->current.x,
			      ty + es->current.y, es->current.z);
}

static void
wlsc_surface_init(struct wlsc_surface *surface,
		  struct wlsc_compositor *compositor, struct wl_visual *visual,
		  int32_t x, int32_t y, int32_t width, int32_t height)
{
	glGenTextures(1, &surface->texture);
	surface->compositor = compositor;
	surface->map.x = x;
	surface->map.y = y;
	surface->map.width = width;
	surface->map.height = height;
	surface->surface = EGL_NO_SURFACE;
	surface->visual = visual;
	surface->current.x = 0;
	surface->current.y = 0;
	surface->current.z = 0;
	surface->current_angle = 0;
	surface->target.x = 0;
	surface->target.y = 0;
	surface->target.z = 0;
	surface->target_angle = 0;
	surface->previous_angle = 0;

	wlsc_surface_update_matrix(surface);
}

static struct wlsc_surface *
wlsc_surface_create_from_cairo_surface(struct wlsc_compositor *ec,
				      cairo_surface_t *surface,
				      int x, int y, int width, int height)
{
	struct wlsc_surface *es;
	int stride;
	void *data;

	stride = cairo_image_surface_get_stride(surface);
	data = cairo_image_surface_get_data(surface);

	es = malloc(sizeof *es);
	if (es == NULL)
		return NULL;

	wlsc_surface_init(es, ec, &ec->premultiplied_argb_visual,
			  x, y, width, height);
	glBindTexture(GL_TEXTURE_2D, es->texture);
	glTexEnvf(GL_TEXTURE_ENV, GL_TEXTURE_ENV_MODE, GL_REPLACE);
	glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_R, GL_REPEAT);
	glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
	glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
	glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, width, height, 0,
		     GL_BGRA, GL_UNSIGNED_BYTE, data);

	return es;
}

static void
wlsc_surface_destroy(struct wlsc_surface *surface,
		     struct wlsc_compositor *compositor)
{
	struct wlsc_listener *l;

	l = container_of(compositor->surface_destroy_listener_list.next,
			 struct wlsc_listener, link);
	while (&l->link != &compositor->surface_destroy_listener_list) {
		l->func(l, surface);
		l = container_of(l->link.next, struct wlsc_listener, link);
	}

	wl_list_remove(&surface->link);
	wl_list_remove(&surface->animate.link);

	glDeleteTextures(1, &surface->texture);
	if (surface->surface != EGL_NO_SURFACE)
		eglDestroySurface(compositor->display, surface->surface);
	free(surface);
}

static void
pointer_path(cairo_t *cr, int x, int y)
{
	const int end = 3, tx = 4, ty = 12, dx = 5, dy = 10;
	const int width = 16, height = 16;

	cairo_move_to(cr, x, y);
	cairo_line_to(cr, x + tx, y + ty);
	cairo_line_to(cr, x + dx, y + dy);
	cairo_line_to(cr, x + width - end, y + height);
	cairo_line_to(cr, x + width, y + height - end);
	cairo_line_to(cr, x + dy, y + dx);
	cairo_line_to(cr, x + ty, y + tx);
	cairo_close_path(cr);
}

static struct wlsc_surface *
pointer_create(struct wlsc_compositor *ec, int x, int y, int width, int height)
{
	struct wlsc_surface *es;
	const int hotspot_x = 16, hotspot_y = 16;
	cairo_surface_t *surface;
	cairo_t *cr;

	surface = cairo_image_surface_create(CAIRO_FORMAT_ARGB32,
					     width, height);

	cr = cairo_create(surface);
	pointer_path(cr, hotspot_x + 5, hotspot_y + 4);
	cairo_set_line_width(cr, 2);
	cairo_set_source_rgb(cr, 0, 0, 0);
	cairo_stroke_preserve(cr);
	cairo_fill(cr);
	blur_surface(surface, width);

	pointer_path(cr, hotspot_x, hotspot_y);
	cairo_stroke_preserve(cr);
	cairo_set_source_rgb(cr, 1, 1, 1);
	cairo_fill(cr);
	cairo_destroy(cr);

	es = wlsc_surface_create_from_cairo_surface(ec,
						   surface,
						   x - hotspot_x,
						   y - hotspot_y,
						   width, height);
	
	cairo_surface_destroy(surface);

	return es;
}

static struct wlsc_surface *
background_create(struct wlsc_output *output, const char *filename)
{
	struct wlsc_surface *background;
	GdkPixbuf *pixbuf;
	GError *error = NULL;
	void *data;
	GLenum format;

	background = malloc(sizeof *background);
	if (background == NULL)
		return NULL;
	
	g_type_init();

	pixbuf = gdk_pixbuf_new_from_file_at_scale(filename,
						   output->width,
						   output->height,
						   FALSE, &error);
	if (error != NULL) {
		free(background);
		return NULL;
	}

	data = gdk_pixbuf_get_pixels(pixbuf);

	wlsc_surface_init(background, output->compositor,
			  &output->compositor->rgb_visual,
			  output->x, output->y, output->width, output->height);

	glBindTexture(GL_TEXTURE_2D, background->texture);
	glTexEnvf(GL_TEXTURE_ENV, GL_TEXTURE_ENV_MODE, GL_REPLACE);
	glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_R, GL_REPEAT);
	glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
	glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);

        if (gdk_pixbuf_get_has_alpha(pixbuf))
		format = GL_RGBA;
	else
		format = GL_RGB;

	glTexImage2D(GL_TEXTURE_2D, 0, GL_RGB,
		     output->width, output->height, 0,
		     format, GL_UNSIGNED_BYTE, data);

	return background;
}

static void
wlsc_surface_draw(struct wlsc_surface *es)
{
	struct wlsc_compositor *ec = es->compositor;
	GLint vertices[12];
	GLint tex_coords[12] = { 0, 0,  0, 1,  1, 0,  1, 1 };
	GLuint indices[4] = { 0, 1, 2, 3 };

	vertices[0] = es->map.x;
	vertices[1] = es->map.y;
	vertices[2] = 0;

	vertices[3] = es->map.x;
	vertices[4] = es->map.y + es->map.height;
	vertices[5] = 0;

	vertices[6] = es->map.x + es->map.width;
	vertices[7] = es->map.y;
	vertices[8] = 0;

	vertices[9] = es->map.x + es->map.width;
	vertices[10] = es->map.y + es->map.height;
	vertices[11] = 0;

	if (es->visual == &ec->argb_visual) {
		glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
		glEnable(GL_BLEND);
	} else if (es->visual == &ec->premultiplied_argb_visual) {
		glBlendFunc(GL_ONE, GL_ONE_MINUS_SRC_ALPHA);
		glEnable(GL_BLEND);
	} else {
		glDisable(GL_BLEND);
	}

	glPushMatrix();
	glMultMatrixd(es->matrix.d);
	glBindTexture(GL_TEXTURE_2D, es->texture);
	glEnable(GL_TEXTURE_2D);
	glEnableClientState(GL_VERTEX_ARRAY);
	glEnableClientState(GL_TEXTURE_COORD_ARRAY);
	glVertexPointer(3, GL_INT, 0, vertices);
	glTexCoordPointer(2, GL_INT, 0, tex_coords);
	glDrawElements(GL_TRIANGLE_STRIP, 4, GL_UNSIGNED_INT, indices);
	glPopMatrix();
}

static void
wlsc_vector_add(struct wlsc_vector *v1, struct wlsc_vector *v2)
{
	v1->x += v2->x;
	v1->y += v2->y;
	v1->z += v2->z;
}

static void
wlsc_vector_subtract(struct wlsc_vector *v1, struct wlsc_vector *v2)
{
	v1->x -= v2->x;
	v1->y -= v2->y;
	v1->z -= v2->z;
}

static void
wlsc_vector_scalar(struct wlsc_vector *v1, GLdouble s)
{
	v1->x *= s;
	v1->y *= s;
	v1->z *= s;
}

static void
repaint_output(struct wlsc_output *output)
{
	struct wlsc_compositor *ec = output->compositor;
	struct wlsc_surface *es;
	struct wlsc_input_device *eid;
	double s = 3000;

	if (!eglMakeCurrent(ec->display, output->surface, output->surface, ec->context)) {
		fprintf(stderr, "failed to make context current\n");
		return;
	}

	glViewport(0, 0, output->width, output->height);
	glMatrixMode(GL_PROJECTION);
	glLoadIdentity();
	glFrustum(-output->width / s, output->width / s,
		  output->height / s, -output->height / s, 1, 2 * s);
	glMatrixMode(GL_MODELVIEW);
	glLoadIdentity();
	glClearColor(0, 0, 0.2, 1);

	glTranslatef(-output->width / 2, -output->height / 2, -s / 2);

	if (output->background)
		wlsc_surface_draw(output->background);
	else
		glClear(GL_COLOR_BUFFER_BIT);

	es = container_of(ec->surface_list.next,
			  struct wlsc_surface, link);
	while (&es->link != &ec->surface_list) {
		wlsc_surface_draw(es);
		es = container_of(es->link.next,
				   struct wlsc_surface, link);
	}

	eid = container_of(ec->input_device_list.next,
			   struct wlsc_input_device, link);
	while (&eid->link != &ec->input_device_list) {
		wlsc_surface_draw(eid->sprite);

		eid = container_of(eid->link.next,
				   struct wlsc_input_device, link);
	}

	eglSwapBuffers(ec->display, output->surface);
}

static void
repaint(void *data)
{
	struct wlsc_compositor *ec = data;
	struct wlsc_output *output;
	struct wlsc_animate *animate, *next;
	struct timespec ts;
	uint32_t msecs;

	if (!ec->repaint_needed) {
		ec->repaint_on_timeout = 0;
		return;
	}

	output = container_of(ec->output_list.next, struct wlsc_output, link);
	while (&output->link != &ec->output_list) {
		repaint_output(output);
		output = container_of(output->link.next,
				      struct wlsc_output, link);
	}

	ec->repaint_needed = 0;

	clock_gettime(CLOCK_MONOTONIC, &ts);
	msecs = ts.tv_sec * 1000 + ts.tv_nsec / (1000 * 1000);
	wl_display_post_frame(ec->wl_display, &ec->base,
			      ec->current_frame, msecs);

	wl_event_source_timer_update(ec->timer_source, 10);
	ec->repaint_on_timeout = 1;

	animate = container_of(ec->animate_list.next, struct wlsc_animate, link);
	while (&animate->link != &ec->animate_list) {
		next = container_of(animate->link.next,
				    struct wlsc_animate, link);
		animate->animate(animate, ec, ec->current_frame, msecs);
		animate = next;
	}

	ec->current_frame++;
}

static void
schedule_repaint(struct wlsc_compositor *ec)
{
	struct wl_event_loop *loop;

	ec->repaint_needed = 1;
	if (!ec->repaint_on_timeout) {
		loop = wl_display_get_event_loop(ec->wl_display);
		wl_event_loop_add_idle(loop, repaint, ec);
	}
}

static void
surface_destroy(struct wl_client *client,
		struct wl_surface *surface)
{
	struct wlsc_surface *es = (struct wlsc_surface *) surface;
	struct wlsc_compositor *ec = es->compositor;

	wlsc_surface_destroy(es, ec);

	schedule_repaint(ec);
}

static void
surface_attach(struct wl_client *client,
	       struct wl_surface *surface, uint32_t name, 
	       uint32_t width, uint32_t height, uint32_t stride,
	       struct wl_object *visual)
{
	struct wlsc_surface *es = (struct wlsc_surface *) surface;
	struct wlsc_compositor *ec = es->compositor;

	if (es->surface != EGL_NO_SURFACE)
		eglDestroySurface(ec->display, es->surface);

	es->width = width;
	es->height = height;
	es->surface = eglCreateSurfaceForName(ec->display, ec->config,
					      name, width, height, stride, NULL);
	if (visual == &ec->argb_visual.base)
		es->visual = &ec->argb_visual;
	else if (visual == &ec->premultiplied_argb_visual.base)
		es->visual = &ec->premultiplied_argb_visual;
	else if (visual == &ec->rgb_visual.base)
		es->visual = &ec->rgb_visual;
	else
		/* FIXME: Smack client with an exception event */;

	glBindTexture(GL_TEXTURE_2D, es->texture);
	glTexEnvf(GL_TEXTURE_ENV, GL_TEXTURE_ENV_MODE, GL_REPLACE);
	glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_R, GL_REPEAT);
	glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
	glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
	eglBindTexImage(ec->display, es->surface, GL_TEXTURE_2D);
}

static void
surface_map(struct wl_client *client,
	    struct wl_surface *surface,
	    int32_t x, int32_t y, int32_t width, int32_t height)
{
	struct wlsc_surface *es = (struct wlsc_surface *) surface;

	es->map.x = x;
	es->map.y = y;
	es->map.width = width;
	es->map.height = height;
	wlsc_surface_update_matrix(es);
}

static void
surface_copy(struct wl_client *client,
	     struct wl_surface *surface,
	     int32_t dst_x, int32_t dst_y,
	     uint32_t name, uint32_t stride,
	     int32_t x, int32_t y, int32_t width, int32_t height)
{
	struct wlsc_surface *es = (struct wlsc_surface *) surface;
	struct wlsc_compositor *ec = es->compositor;
	EGLSurface src;

	/* FIXME: glCopyPixels should work, but then we'll have to
	 * call eglMakeCurrent to set up the src and dest surfaces
	 * first.  This seems cheaper, but maybe there's a better way
	 * to accomplish this. */

	src = eglCreateSurfaceForName(ec->display, ec->config,
				      name, x + width, y + height, stride, NULL);

	eglCopyNativeBuffers(ec->display, es->surface, GL_FRONT_LEFT, dst_x, dst_y,
			     src, GL_FRONT_LEFT, x, y, width, height);
	eglDestroySurface(ec->display, src);
}

static void
surface_damage(struct wl_client *client,
	       struct wl_surface *surface,
	       int32_t x, int32_t y, int32_t width, int32_t height)
{
	/* FIXME: This need to take a damage region, of course. */
}

const static struct wl_surface_interface surface_interface = {
	surface_destroy,
	surface_attach,
	surface_map,
	surface_copy,
	surface_damage
};

static void
animate_surface(struct wlsc_animate *animate, 
		struct wlsc_compositor *compositor,
		uint32_t frame, uint32_t msecs)
{
	struct wlsc_surface *s;
	double angle_force, angle;
	struct wlsc_vector force, tmp;
	double step = 0.3;
	double friction = 1;
	double spring = 0.2;

	s = container_of(animate, struct wlsc_surface, animate);

	angle = s->current_angle;
	angle_force = (s->target_angle - angle) * spring +
		(s->previous_angle - angle) * friction;

	s->current_angle = angle + (angle - s->previous_angle) + angle_force * step;
	s->previous_angle = angle;

	force = s->target;
	wlsc_vector_subtract(&force, &s->current);
	wlsc_vector_scalar(&force, spring);
	tmp = s->previous;
	wlsc_vector_subtract(&tmp, &s->current);
	wlsc_vector_scalar(&tmp, friction);
	wlsc_vector_add(&force, &tmp);

	wlsc_vector_scalar(&force, step);
	wlsc_vector_add(&force, &s->current);
	wlsc_vector_subtract(&force, &s->previous);
	s->previous = s->current;
	wlsc_vector_add(&s->current, &force);

	wlsc_surface_update_matrix(s);
	
	tmp = s->current;
	wlsc_vector_subtract(&tmp, &s->target);
	if (tmp.x * tmp.x + tmp.y * tmp.y + tmp.z * tmp.z > 0.001) {
		schedule_repaint(compositor);
	} else {
		wl_list_remove(&s->animate.link);
		wl_list_init(&s->animate.link);
	}
}

static void
compositor_create_surface(struct wl_client *client,
			  struct wl_compositor *compositor, uint32_t id)
{
	struct wlsc_compositor *ec = (struct wlsc_compositor *) compositor;
	struct wlsc_surface *es;

	es = malloc(sizeof *es);
	if (es == NULL)
		/* FIXME: Send OOM event. */
		return;

	wlsc_surface_init(es, ec, NULL, 0, 0, 0, 0);
	es->animate.animate = animate_surface;
	wl_list_init(&es->animate.link);

	wl_list_insert(ec->surface_list.prev, &es->link);
	wl_client_add_surface(client, &es->base,
			      &surface_interface, id);
}

static void
compositor_commit(struct wl_client *client,
		  struct wl_compositor *compositor, uint32_t key)
{
	struct wlsc_compositor *ec = (struct wlsc_compositor *) compositor;

	schedule_repaint(ec);
	wl_client_send_acknowledge(client, compositor, key, ec->current_frame);
}

const static struct wl_compositor_interface compositor_interface = {
	compositor_create_surface,
	compositor_commit
};

static void
wlsc_surface_transform(struct wlsc_surface *surface,
		       int32_t x, int32_t y, int32_t *sx, int32_t *sy)
{
	/* Transform to surface coordinates. */
	*sx = (x - surface->map.x) * surface->width / surface->map.width;
	*sy = (y - surface->map.y) * surface->height / surface->map.height;
}

static void
wlsc_input_device_set_keyboard_focus(struct wlsc_input_device *device,
				     struct wlsc_surface *surface)
{
	if (device->keyboard_focus == surface)
		return;

	if (device->keyboard_focus &&
	    (!surface || device->keyboard_focus->base.client != surface->base.client))
		wl_surface_post_event(&device->keyboard_focus->base,
				      &device->base,
				      WL_INPUT_KEYBOARD_FOCUS, NULL, &device->keys);

	if (surface)
		wl_surface_post_event(&surface->base,
				      &device->base,
				      WL_INPUT_KEYBOARD_FOCUS,
				      &surface->base, &device->keys);

	device->keyboard_focus = surface;
}

static void
wlsc_input_device_set_pointer_focus(struct wlsc_input_device *device,
				    struct wlsc_surface *surface)
{
	if (device->pointer_focus == surface)
		return;

	if (device->pointer_focus &&
	    (!surface || device->pointer_focus->base.client != surface->base.client))
		wl_surface_post_event(&device->pointer_focus->base,
				      &device->base,
				      WL_INPUT_POINTER_FOCUS, NULL);
	if (surface)
		wl_surface_post_event(&surface->base,
				      &device->base,
				      WL_INPUT_POINTER_FOCUS, &surface->base);

	device->pointer_focus = surface;
}

static struct wlsc_surface *
pick_surface(struct wlsc_input_device *device, int32_t *sx, int32_t *sy)
{
	struct wlsc_compositor *ec = device->ec;
	struct wlsc_surface *es;

	if (device->grab > 0) {
		wlsc_surface_transform(device->grab_surface,
				       device->x, device->y, sx, sy);
		return device->grab_surface;
	}

	es = container_of(ec->surface_list.prev,
			  struct wlsc_surface, link);
	while (&es->link != &ec->surface_list) {
		if (es->map.x <= device->x &&
		    device->x < es->map.x + es->map.width &&
		    es->map.y <= device->y &&
		    device->y < es->map.y + es->map.height) {
			wlsc_surface_transform(es, device->x, device->y, sx, sy);
			return es;
		}

		es = container_of(es->link.prev,
				  struct wlsc_surface, link);

	}

	return NULL;
}

void
notify_motion(struct wlsc_input_device *device, int x, int y)
{
	struct wlsc_surface *es;
	struct wlsc_compositor *ec = device->ec;
	struct wlsc_output *output;
	const int hotspot_x = 16, hotspot_y = 16;
	int32_t sx, sy;

	if (!ec->vt_active)
		return;

	/* FIXME: We need some multi head love here. */
	output = container_of(ec->output_list.next, struct wlsc_output, link);
	if (x < output->x)
		x = 0;
	if (y < output->y)
		y = 0;
	if (x >= output->x + output->width)
		x = output->x + output->width - 1;
	if (y >= output->y + output->height)
		y = output->y + output->height - 1;

	device->x = x;
	device->y = y;
	es = pick_surface(device, &sx, &sy);

	wlsc_input_device_set_pointer_focus(device, es);
		
	if (es)
		wl_surface_post_event(&es->base, &device->base,
				      WL_INPUT_MOTION, x, y, sx, sy);

	device->sprite->map.x = x - hotspot_x;
	device->sprite->map.y = y - hotspot_y;

	schedule_repaint(device->ec);
}

void
notify_button(struct wlsc_input_device *device,
	      int32_t button, int32_t state)
{
	struct wlsc_surface *es;
	struct wlsc_compositor *ec = device->ec;
	int32_t sx, sy;

	if (!ec->vt_active)
		return;

	es = pick_surface(device, &sx, &sy);
	if (es) {
		wl_list_remove(&es->link);
		wl_list_insert(device->ec->surface_list.prev, &es->link);

		if (state) {
			/* FIXME: We need callbacks when the surfaces
			 * we reference here go away. */
			device->grab++;
			device->grab_surface = es;
			wlsc_input_device_set_keyboard_focus(device, es);
		} else {
			device->grab--;
		}

		/* FIXME: Swallow click on raise? */
		wl_surface_post_event(&es->base, &device->base,
				      WL_INPUT_BUTTON, button, state,
				      device->x, device->y, sx, sy);

		schedule_repaint(device->ec);
	}
}

static void on_term_signal(int signal_number, void *data);

static void
update_surface_targets(struct wlsc_compositor *compositor, int primary)
{
	struct wlsc_surface *s;
	int i;

	i = 0;
	s = container_of(compositor->surface_list.next,
			 struct wlsc_surface, link);
	while (&s->link != &compositor->surface_list) {
		if (i < primary) {
			s->target.x = -400 + 500 * (i - primary);
			s->target.y = 0;
			s->target.z = -1500;
			s->target_angle = M_PI / 4;
		} else if (i == primary) {
			s->target.x = 0;
			s->target.y = 0;
			s->target.z = -1000;
			s->target_angle = 0;
			compositor->primary = s;
		} else {
			s->target.x = 400 + 500 * (i - primary);
			s->target.y = 0;
			s->target.z = -1500;
			s->target_angle = -M_PI / 4;
		}
		wl_list_remove(&s->animate.link);
		wl_list_insert(compositor->animate_list.prev, &s->animate.link);
		s = container_of(s->link.next,
				 struct wlsc_surface, link);
		i++;
	}

	schedule_repaint(compositor);
}

void
notify_key(struct wlsc_input_device *device,
	   uint32_t key, uint32_t state)
{
	struct wlsc_compositor *ec = device->ec;
	uint32_t *k, *end;

	if (!ec->vt_active)
		return;

	switch (key | ec->meta_state) {
	case KEY_EJECTCD | META_DOWN:
		on_term_signal(SIGTERM, ec);
		return;

 	case KEY_1 | META_DOWN:
 	case KEY_2 | META_DOWN:
 	case KEY_3 | META_DOWN:
 	case KEY_4 | META_DOWN:
 	case KEY_5 | META_DOWN:
		update_surface_targets(ec, key - KEY_1);
		if (device->grab == 0)
			wlsc_input_device_set_keyboard_focus(device, ec->primary);
			device->keyboard_focus = ec->primary;
		return;

	case KEY_LEFTMETA:
	case KEY_RIGHTMETA:
	case KEY_LEFTMETA | META_DOWN:
	case KEY_RIGHTMETA | META_DOWN:
		ec->meta_state = state ? META_DOWN : 0;
		if (state == 0 && ec->primary != NULL) {
			ec->primary->target.z = 0;
			wl_list_remove(&ec->primary->animate.link);
			wl_list_insert(&ec->animate_list,
				       &ec->primary->animate.link);
			schedule_repaint(ec);
		}
		return;
	}

	end = device->keys.data + device->keys.size;
	for (k = device->keys.data; k < end; k++) {
		if (*k == key)
			*k = *--end;
	}
	device->keys.size = (void *) end - device->keys.data;
	if (state) {
		k = wl_array_add(&device->keys, sizeof *k);
		*k = key;
	}

	if (device->keyboard_focus != NULL)
		wl_surface_post_event(&device->keyboard_focus->base,
				      &device->base, 
				      WL_INPUT_KEY, key, state);
}

struct evdev_input_device *
evdev_input_device_create(struct wlsc_input_device *device,
			  struct wl_display *display, const char *path);

static void
handle_surface_destroy(struct wlsc_listener *listener,
		       struct wlsc_surface *surface)
{
	struct wlsc_input_device *device =
		container_of(listener, struct wlsc_input_device, listener);

	if (device->grab_surface == surface) {
		device->grab_surface = NULL;
		device->grab = 0;
	}
	if (device->keyboard_focus == surface)
		device->keyboard_focus = NULL;		
	if (device->pointer_focus == surface)
		device->pointer_focus = NULL;	
}

static struct wlsc_input_device *
create_input_device(struct wlsc_compositor *ec)
{
	struct wlsc_input_device *device;

	device = malloc(sizeof *device);
	if (device == NULL)
		return NULL;

	memset(device, 0, sizeof *device);
	device->base.interface = &wl_input_device_interface;
	device->base.implementation = NULL;
	wl_display_add_object(ec->wl_display, &device->base);
	wl_display_add_global(ec->wl_display, &device->base, NULL);
	device->x = 100;
	device->y = 100;
	device->ec = ec;

	device->listener.func = handle_surface_destroy;
	wl_list_insert(ec->surface_destroy_listener_list.prev,
		       &device->listener.link);
	wl_list_insert(ec->input_device_list.prev, &device->link);

	return device;
}

void
wlsc_device_get_position(struct wlsc_input_device *device, int32_t *x, int32_t *y)
{
	*x = device->x;
	*y = device->y;
}

static void
post_output_geometry(struct wl_client *client, struct wl_object *global)
{
	struct wlsc_output *output =
		container_of(global, struct wlsc_output, base);

	wl_client_post_event(client, global,
			     WL_OUTPUT_GEOMETRY,
			     output->width, output->height);
}

static int
init_egl(struct wlsc_compositor *ec, struct udev_device *device)
{
	static const EGLint config_attribs[] = {
		EGL_DEPTH_SIZE,		0,
		EGL_STENCIL_SIZE,	0,
		EGL_CONFIG_CAVEAT,	EGL_NONE,
		EGL_RED_SIZE,		8,
		EGL_NONE		
	};

	EGLint major, minor;

	ec->display = eglCreateDisplayNative(device);
	if (ec->display == NULL) {
		fprintf(stderr, "failed to create display\n");
		return -1;
	}

	if (!eglInitialize(ec->display, &major, &minor)) {
		fprintf(stderr, "failed to initialize display\n");
		return -1;
	}

	if (!eglChooseConfig(ec->display, config_attribs, &ec->config, 1, NULL))
		return -1;

	ec->context = eglCreateContext(ec->display, ec->config, NULL, NULL);
	if (ec->context == NULL) {
		fprintf(stderr, "failed to create context\n");
		return -1;
	}
	return 0;
}

static int
create_output(struct wlsc_compositor *ec, struct udev_device *device)
{
	const static EGLint surface_attribs[] = {
		EGL_RENDER_BUFFER, EGL_BACK_BUFFER,
		EGL_NONE
	};

	drmModeConnector *connector;
	drmModeRes *resources;
	drmModeEncoder *encoder;
	drmModeModeInfo *mode;
	struct wlsc_output *output;
	uint32_t name, handle, stride;
	int i, ret, fd;

	if (ec->display == NULL && init_egl(ec, device) < 0) {
		fprintf(stderr, "failed to initialize egl\n");
		return -1;
	}

	output = malloc(sizeof *output);
	if (output == NULL)
		return -1;

	fd = eglGetDisplayFD(ec->display);
	resources = drmModeGetResources(fd);
	if (!resources) {
		fprintf(stderr, "drmModeGetResources failed\n");
		return -1;
	}

	for (i = 0; i < resources->count_connectors; i++) {
		connector = drmModeGetConnector(fd, resources->connectors[i]);
		if (connector == NULL)
			continue;

		if (connector->connection == DRM_MODE_CONNECTED &&
		    connector->count_modes > 0)
			break;

		drmModeFreeConnector(connector);
	}

	if (i == resources->count_connectors) {
		fprintf(stderr, "No currently active connector found.\n");
		return -1;
	}

	mode = &connector->modes[0];

	for (i = 0; i < resources->count_encoders; i++) {
		encoder = drmModeGetEncoder(fd, resources->encoders[i]);

		if (encoder == NULL)
			continue;

		if (encoder->encoder_id == connector->encoder_id)
			break;

		drmModeFreeEncoder(encoder);
	}

	output->compositor = ec;
	output->crtc_id = encoder->crtc_id;
	output->connector_id = connector->connector_id;
	output->mode = mode;
	output->x = 0;
	output->y = 0;
	output->width = mode->hdisplay;
	output->height = mode->vdisplay;

	output->surface = eglCreateSurfaceForName(ec->display,
						  ec->config,
						  0,
						  output->width,
						  output->height,
						  0, surface_attribs);
	if (output->surface == NULL) {
		fprintf(stderr, "failed to create surface\n");
		return -1;
	}

	eglGetNativeBuffer(output->surface,
			   GL_FRONT_LEFT, &name, &handle, &stride);
	ret = drmModeAddFB(fd, output->width, output->height,
			   32, 32, stride, handle, &output->fb_id);
	if (ret) {
		fprintf(stderr, "failed to add fb: %m\n");
		return -1;
	}

	ret = drmModeSetCrtc(fd, encoder->crtc_id, output->fb_id, 0, 0,
			     &connector->connector_id, 1, mode);
	if (ret) {
		fprintf(stderr, "failed to set mode: %m\n");
		return -1;
	}

	output->base.interface = &wl_output_interface;
	wl_display_add_object(ec->wl_display, &output->base);
	wl_display_add_global(ec->wl_display, &output->base, post_output_geometry);
	wl_list_insert(ec->output_list.prev, &output->link);

	if (!eglMakeCurrent(ec->display, output->surface, output->surface, ec->context)) {
		fprintf(stderr, "failed to make context current\n");
		return -1;
	}

	output->background = background_create(output, option_background);

	return 0;
}

static const struct wl_interface visual_interface = {
	"visual", 1,
};

static void
add_visuals(struct wlsc_compositor *ec)
{
	ec->argb_visual.base.interface = &visual_interface;
	ec->argb_visual.base.implementation = NULL;
	wl_display_add_object(ec->wl_display, &ec->argb_visual.base);
	wl_display_add_global(ec->wl_display, &ec->argb_visual.base, NULL);

	ec->premultiplied_argb_visual.base.interface = &visual_interface;
	ec->premultiplied_argb_visual.base.implementation = NULL;
	wl_display_add_object(ec->wl_display,
			      &ec->premultiplied_argb_visual.base);
	wl_display_add_global(ec->wl_display,
			      &ec->premultiplied_argb_visual.base, NULL);

	ec->rgb_visual.base.interface = &visual_interface;
	ec->rgb_visual.base.implementation = NULL;
	wl_display_add_object(ec->wl_display, &ec->rgb_visual.base);
	wl_display_add_global(ec->wl_display, &ec->rgb_visual.base, NULL);
}

static void on_enter_vt(int signal_number, void *data)
{
	struct wlsc_compositor *ec = data;
	struct wlsc_output *output;
	int ret, fd;

	ioctl(ec->tty_fd, VT_RELDISP, VT_ACKACQ);
	ec->vt_active = TRUE;

	fd = eglGetDisplayFD(ec->display);
	output = container_of(ec->output_list.next, struct wlsc_output, link);
	while (&output->link != &ec->output_list) {
		ret = drmModeSetCrtc(fd, output->crtc_id, output->fb_id, 0, 0,
				     &output->connector_id, 1, output->mode);
		if (ret)
			fprintf(stderr, "failed to set mode for connector %d: %m\n",
				output->connector_id);

		output = container_of(output->link.next,
				      struct wlsc_output, link);
	}
}

static void on_leave_vt(int signal_number, void *data)
{
	struct wlsc_compositor *ec = data;

	ioctl (ec->tty_fd, VT_RELDISP, 1);
	ec->vt_active = FALSE;
}

static void
on_tty_input(int fd, uint32_t mask, void *data)
{
	struct wlsc_compositor *ec = data;

	/* Ignore input to tty.  We get keyboard events from evdev
	 */
	tcflush(ec->tty_fd, TCIFLUSH);
}

static void on_term_signal(int signal_number, void *data)
{
	struct wlsc_compositor *ec = data;

	if (tcsetattr(ec->tty_fd, TCSANOW, &ec->terminal_attributes) < 0)
		fprintf(stderr, "could not restore terminal to canonical mode\n");

	exit(0);
}

static int setup_tty(struct wlsc_compositor *ec, struct wl_event_loop *loop)
{
	struct termios raw_attributes;
	struct vt_mode mode = { 0 };

	ec->tty_fd = open("/dev/tty0", O_RDWR | O_NOCTTY);
	if (ec->tty_fd <= 0) {
		fprintf(stderr, "failed to open active tty: %m\n");
		return -1;
	}

	if (tcgetattr(ec->tty_fd, &ec->terminal_attributes) < 0) {
		fprintf(stderr, "could not get terminal attributes: %m\n");
		return -1;
	}

	/* Ignore control characters and disable echo */
	raw_attributes = ec->terminal_attributes;
	cfmakeraw(&raw_attributes);

	/* Fix up line endings to be normal (cfmakeraw hoses them) */
	raw_attributes.c_oflag |= OPOST | OCRNL;

	if (tcsetattr(ec->tty_fd, TCSANOW, &raw_attributes) < 0)
		fprintf(stderr, "could not put terminal into raw mode: %m\n");

	ec->term_signal_source =
		wl_event_loop_add_signal(loop, SIGTERM, on_term_signal, ec);

	ec->tty_input_source =
		wl_event_loop_add_fd(loop, ec->tty_fd,
				     WL_EVENT_READABLE, on_tty_input, ec);

	ec->vt_active = TRUE;
	mode.mode = VT_PROCESS;
	mode.relsig = SIGUSR1;
	mode.acqsig = SIGUSR2;
	if (!ioctl(ec->tty_fd, VT_SETMODE, &mode) < 0) {
		fprintf(stderr, "failed to take control of vt handling\n");
	}

	ec->leave_vt_source =
		wl_event_loop_add_signal(loop, SIGUSR1, on_leave_vt, ec);
	ec->enter_vt_source =
		wl_event_loop_add_signal(loop, SIGUSR2, on_enter_vt, ec);

	return 0;
}

static int
init_libudev(struct wlsc_compositor *ec)
{
	struct udev_enumerate *e;
        struct udev_list_entry *entry;
	struct udev_device *device;
	const char *path;
	struct wlsc_input_device *input_device;

	ec->udev = udev_new();
	if (ec->udev == NULL) {
		fprintf(stderr, "failed to initialize udev context\n");
		return -1;
	}

	input_device = create_input_device(ec);

	e = udev_enumerate_new(ec->udev);
	udev_enumerate_add_match_subsystem(e, "input");
	udev_enumerate_add_match_property(e, "WAYLAND_SEAT", "1");
        udev_enumerate_scan_devices(e);
        udev_list_entry_foreach(entry, udev_enumerate_get_list_entry(e)) {
		path = udev_list_entry_get_name(entry);
		device = udev_device_new_from_syspath(ec->udev, path);
		evdev_input_device_create(input_device, ec->wl_display,
					  udev_device_get_devnode(device));
	}
        udev_enumerate_unref(e);

	e = udev_enumerate_new(ec->udev);
	udev_enumerate_add_match_subsystem(e, "drm");
	udev_enumerate_add_match_property(e, "WAYLAND_SEAT", "1");
        udev_enumerate_scan_devices(e);
        udev_list_entry_foreach(entry, udev_enumerate_get_list_entry(e)) {
		path = udev_list_entry_get_name(entry);
		device = udev_device_new_from_syspath(ec->udev, path);
		if (create_output(ec, device) < 0) {
			fprintf(stderr, "failed to create output for %s\n", path);
			return -1;
		}
	}
        udev_enumerate_unref(e);

	/* Create the pointer surface now that we have a current EGL context. */
	input_device->sprite =
		pointer_create(ec, input_device->x, input_device->y, 64, 64);

	return 0;
}

static struct wlsc_compositor *
wlsc_compositor_create(struct wl_display *display)
{
	struct wlsc_compositor *ec;
	struct screenshooter *shooter;
	struct wl_event_loop *loop;

	ec = malloc(sizeof *ec);
	if (ec == NULL)
		return NULL;

	memset(ec, 0, sizeof *ec);
	ec->wl_display = display;

	wl_display_set_compositor(display, &ec->base, &compositor_interface); 

	add_visuals(ec);

	wl_list_init(&ec->surface_list);
	wl_list_init(&ec->input_device_list);
	wl_list_init(&ec->output_list);
	wl_list_init(&ec->surface_destroy_listener_list);
	if (init_libudev(ec) < 0) {
		fprintf(stderr, "failed to initialize devices\n");
		return NULL;
	}

	shooter = screenshooter_create(ec);
	wl_display_add_object(display, &shooter->base);
	wl_display_add_global(display, &shooter->base, NULL);

	loop = wl_display_get_event_loop(ec->wl_display);

	setup_tty(ec, loop);

	ec->timer_source = wl_event_loop_add_timer(loop, repaint, ec);
	schedule_repaint(ec);

	wl_list_init(&ec->animate_list);

	return ec;
}

/* The plan here is to generate a random anonymous socket name and
 * advertise that through a service on the session dbus.
 */
static const char socket_name[] = "\0wayland";

int main(int argc, char *argv[])
{
	struct wl_display *display;
	struct wlsc_compositor *ec;
	GError *error = NULL;
	GOptionContext *context;

	context = g_option_context_new(NULL);
	g_option_context_add_main_entries(context, option_entries, "Wayland");
	if (!g_option_context_parse(context, &argc, &argv, &error)) {
		fprintf(stderr, "option parsing failed: %s\n", error->message);
		exit(EXIT_FAILURE);
	}

	display = wl_display_create();

	ec = wlsc_compositor_create(display);
	if (ec == NULL) {
		fprintf(stderr, "failed to create compositor\n");
		exit(EXIT_FAILURE);
	}
		
	if (wl_display_add_socket(display, socket_name, sizeof socket_name)) {
		fprintf(stderr, "failed to add socket: %m\n");
		exit(EXIT_FAILURE);
	}

	wl_display_run(display);

	return 0;
}