compositor/compositor.c

   1 /*
   2  * Copyright © 2008 Kristian Høgsberg
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software Foundation,
  16  * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  17  */
  18
  19 #define _GNU_SOURCE
  20
  21 #include "config.h"
  22
  23 #include <stdio.h>
  24 #include <string.h>
  25 #include <stdlib.h>
  26 #include <stdint.h>
  27 #include <stdarg.h>
  28 #include <sys/ioctl.h>
  29 #include <fcntl.h>
  30 #include <unistd.h>
  31 #include <gdk-pixbuf/gdk-pixbuf.h>
  32 #include <math.h>
  33 #include <linux/input.h>
  34
  35 #include "wayland-server.h"
  36 #include "compositor.h"
  37
  38 /* The plan here is to generate a random anonymous socket name and
  39  * advertise that through a service on the session dbus.
  40  */
  41 static const char *option_socket_name = NULL;
  42 static const char *option_background = "background.jpg";
  43 static const char *option_geometry = "1024x640";
  44 static int option_connector = 0;
  45
  46 static const GOptionEntry option_entries[] = {
  47         { "background", 'b', 0, G_OPTION_ARG_STRING,
  48           &option_background, "Background image" },
  49         { "connector", 'c', 0, G_OPTION_ARG_INT,
  50           &option_connector, "KMS connector" },
  51         { "geometry", 'g', 0, G_OPTION_ARG_STRING,
  52           &option_geometry, "Geometry" },
  53         { "socket", 's', 0, G_OPTION_ARG_STRING,
  54           &option_socket_name, "Socket Name" },
  55         { NULL }
  56 };
  57
  58 static void
  59 wlsc_matrix_init(struct wlsc_matrix *matrix)
  60 {
  61         static const struct wlsc_matrix identity = {
  62                 { 1, 0, 0, 0,  0, 1, 0, 0,  0, 0, 1, 0,  0, 0, 0, 1 }
  63         };
  64
  65         memcpy(matrix, &identity, sizeof identity);
  66 }
  67
  68 static void
  69 wlsc_matrix_multiply(struct wlsc_matrix *m, const struct wlsc_matrix *n)
  70 {
  71         struct wlsc_matrix tmp;
  72         const GLfloat *row, *column;
  73         div_t d;
  74         int i, j;
  75
  76         for (i = 0; i < 16; i++) {
  77                 tmp.d[i] = 0;
  78                 d = div(i, 4);
  79                 row = m->d + d.quot * 4;
  80                 column = n->d + d.rem;
  81                 for (j = 0; j < 4; j++)
  82                         tmp.d[i] += row[j] * column[j * 4];
  83         }
  84         memcpy(m, &tmp, sizeof tmp);
  85 }
  86
  87 static void
  88 wlsc_matrix_translate(struct wlsc_matrix *matrix, GLfloat x, GLfloat y, GLfloat z)
  89 {
  90         struct wlsc_matrix translate = {
  91                 { 1, 0, 0, 0,  0, 1, 0, 0,  0, 0, 1, 0,  x, y, z, 1 }
  92         };
  93
  94         wlsc_matrix_multiply(matrix, &translate);
  95 }
  96
  97 static void
  98 wlsc_matrix_scale(struct wlsc_matrix *matrix, GLfloat x, GLfloat y, GLfloat z)
  99 {
 100         struct wlsc_matrix scale = {
 101                 { x, 0, 0, 0,  0, y, 0, 0,  0, 0, z, 0,  0, 0, 0, 1 }
 102         };
 103
 104         wlsc_matrix_multiply(matrix, &scale);
 105 }
 106
 107 static void
 108 wlsc_matrix_transform(struct wlsc_matrix *matrix, struct wlsc_vector *v)
 109 {
 110         int i, j;
 111         struct wlsc_vector t;
 112
 113         for (i = 0; i < 4; i++) {
 114                 t.f[i] = 0;
 115                 for (j = 0; j < 4; j++)
 116                         t.f[i] += v->f[j] * matrix->d[i + j * 4];
 117         }
 118
 119         *v = t;
 120 }
 121
 122 static struct wlsc_surface *
 123 wlsc_surface_create(struct wlsc_compositor *compositor,
 124                     int32_t x, int32_t y, int32_t width, int32_t height)
 125 {
 126         struct wlsc_surface *surface;
 127
 128         surface = malloc(sizeof *surface);
 129         if (surface == NULL)
 130                 return NULL;
 131
 132         wl_list_init(&surface->surface.destroy_listener_list);
 133         wl_list_init(&surface->link);
 134         surface->map_type = WLSC_SURFACE_MAP_UNMAPPED;
 135
 136         glGenTextures(1, &surface->texture);
 137         glBindTexture(GL_TEXTURE_2D, surface->texture);
 138         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
 139         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
 140         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
 141         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
 142
 143         surface->compositor = compositor;
 144         surface->visual = NULL;
 145         surface->x = x;
 146         surface->y = y;
 147         surface->width = width;
 148         surface->height = height;
 149         wlsc_matrix_init(&surface->matrix);
 150         wlsc_matrix_scale(&surface->matrix, width, height, 1);
 151         wlsc_matrix_translate(&surface->matrix, x, y, 0);
 152
 153         wlsc_matrix_init(&surface->matrix_inv);
 154         wlsc_matrix_translate(&surface->matrix_inv, -x, -y, 0);
 155         wlsc_matrix_scale(&surface->matrix_inv, 1.0 / width, 1.0 / height, 1);
 156
 157         return surface;
 158 }
 159
 160 uint32_t
 161 get_time(void)
 162 {
 163         struct timeval tv;
 164
 165         gettimeofday(&tv, NULL);
 166
 167         return tv.tv_sec * 1000 + tv.tv_usec / 1000;
 168 }
 169
 170 static void
 171 destroy_surface(struct wl_resource *resource, struct wl_client *client)
 172 {
 173         struct wlsc_surface *surface =
 174                 container_of(resource, struct wlsc_surface, surface.resource);
 175         struct wlsc_compositor *compositor = surface->compositor;
 176         struct wl_listener *l, *next;
 177         uint32_t time;
 178
 179         wl_list_remove(&surface->link);
 180         glDeleteTextures(1, &surface->texture);
 181
 182         time = get_time();
 183         wl_list_for_each_safe(l, next,
 184                               &surface->surface.destroy_listener_list, link)
 185                 l->func(l, &surface->surface, time);
 186
 187         free(surface);
 188
 189         wlsc_compositor_schedule_repaint(compositor);
 190 }
 191
 192 uint32_t *
 193 wlsc_load_image(const char *filename, int width, int height)
 194 {
 195         GdkPixbuf *pixbuf;
 196         GError *error = NULL;
 197         int stride, i, n_channels;
 198         unsigned char *pixels, *end, *argb_pixels, *s, *d;
 199
 200         pixbuf = gdk_pixbuf_new_from_file_at_scale(filename,
 201                                                    width, height,
 202                                                    FALSE, &error);
 203         if (error != NULL) {
 204                 fprintf(stderr, "failed to load image: %s\n", error->message);
 205                 g_error_free(error);
 206                 return NULL;
 207         }
 208
 209         stride = gdk_pixbuf_get_rowstride(pixbuf);
 210         pixels = gdk_pixbuf_get_pixels(pixbuf);
 211         n_channels = gdk_pixbuf_get_n_channels(pixbuf);
 212
 213         argb_pixels = malloc (height * width * 4);
 214         if (argb_pixels == NULL) {
 215                 g_object_unref(pixbuf);
 216                 return NULL;
 217         }
 218
 219         if (n_channels == 4) {
 220                 for (i = 0; i < height; i++) {
 221                         s = pixels + i * stride;
 222                         end = s + width * 4;
 223                         d = argb_pixels + i * width * 4;
 224                         while (s < end) {
 225                                 unsigned int t;
 226
 227 #define MULT(_d,c,a,t) \
 228         do { t = c * a + 0x7f; _d = ((t >> 8) + t) >> 8; } while (0)
 229
 230                                 MULT(d[0], s[2], s[3], t);
 231                                 MULT(d[1], s[1], s[3], t);
 232                                 MULT(d[2], s[0], s[3], t);
 233                                 d[3] = s[3];
 234                                 s += 4;
 235                                 d += 4;
 236                         }
 237                 }
 238         } else if (n_channels == 3) {
 239                 for (i = 0; i < height; i++) {
 240                         s = pixels + i * stride;
 241                         end = s + width * 3;
 242                         d = argb_pixels + i * width * 4;
 243                         while (s < end) {
 244                                 d[0] = s[2];
 245                                 d[1] = s[1];
 246                                 d[2] = s[0];
 247                                 d[3] = 0xff;
 248                                 s += 3;
 249                                 d += 4;
 250                         }
 251                 }
 252         }
 253
 254         g_object_unref(pixbuf);
 255
 256         return (uint32_t *) argb_pixels;
 257 }
 258
 259 static struct wl_buffer *
 260 create_buffer_from_png(struct wlsc_compositor *ec,
 261                        const char *filename, int width, int height)
 262 {
 263         uint32_t *pixels;
 264         struct wl_buffer *buffer;
 265
 266         pixels = wlsc_load_image(filename, width, height);
 267         if(pixels == NULL)
 268             return NULL;
 269
 270         buffer = ec->create_buffer(ec, width, height,
 271                                    &ec->compositor.premultiplied_argb_visual,
 272                                    pixels);
 273
 274         free(pixels);
 275
 276         return buffer;
 277 }
 278
 279 static const struct {
 280         const char *filename;
 281         int hotspot_x, hotspot_y;
 282 } pointer_images[] = {
 283         { DATADIR "/wayland/bottom_left_corner.png",     6, 30 },
 284         { DATADIR "/wayland/bottom_right_corner.png",   28, 28 },
 285         { DATADIR "/wayland/bottom_side.png",           16, 20 },
 286         { DATADIR "/wayland/grabbing.png",              20, 17 },
 287         { DATADIR "/wayland/left_ptr.png",              10,  5 },
 288         { DATADIR "/wayland/left_side.png",             10, 20 },
 289         { DATADIR "/wayland/right_side.png",            30, 19 },
 290         { DATADIR "/wayland/top_left_corner.png",        8,  8 },
 291         { DATADIR "/wayland/top_right_corner.png",      26,  8 },
 292         { DATADIR "/wayland/top_side.png",              18,  8 },
 293         { DATADIR "/wayland/xterm.png",                 15, 15 }
 294 };
 295
 296 static void
 297 create_pointer_images(struct wlsc_compositor *ec)
 298 {
 299         int i, count;
 300         const int width = 32, height = 32;
 301
 302         count = ARRAY_LENGTH(pointer_images);
 303         ec->pointer_buffers = malloc(count * sizeof *ec->pointer_buffers);
 304         for (i = 0; i < count; i++) {
 305                 ec->pointer_buffers[i] =
 306                         create_buffer_from_png(ec,
 307                                                pointer_images[i].filename,
 308                                                width, height);
 309         }
 310 }
 311
 312 static struct wlsc_surface *
 313 background_create(struct wlsc_output *output, const char *filename)
 314 {
 315         struct wlsc_surface *background;
 316         struct wl_buffer *buffer;
 317
 318         background = wlsc_surface_create(output->compositor,
 319                                          output->x, output->y,
 320                                          output->width, output->height);
 321         if (background == NULL)
 322                 return NULL;
 323
 324         buffer = create_buffer_from_png(output->compositor,
 325                                         filename,
 326                                         output->width, output->height);
 327         if (buffer == NULL) {
 328                 free(background);
 329                 return NULL;
 330         }
 331         buffer->attach(buffer, &background->surface);
 332
 333         return background;
 334 }
 335
 336 static void
 337 wlsc_surface_draw(struct wlsc_surface *es, struct wlsc_output *output)
 338 {
 339         struct wlsc_compositor *ec = es->compositor;
 340         struct wlsc_matrix tmp;
 341
 342         tmp = es->matrix;
 343         wlsc_matrix_multiply(&tmp, &output->matrix);
 344         glUniformMatrix4fv(ec->proj_uniform, 1, GL_FALSE, tmp.d);
 345         glUniform1i(ec->tex_uniform, 0);
 346
 347         if (es->visual == &ec->compositor.argb_visual) {
 348                 glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
 349                 glEnable(GL_BLEND);
 350         } else if (es->visual == &ec->compositor.premultiplied_argb_visual) {
 351                 glBlendFunc(GL_ONE, GL_ONE_MINUS_SRC_ALPHA);
 352                 glEnable(GL_BLEND);
 353         } else {
 354                 glDisable(GL_BLEND);
 355         }
 356
 357         glBindTexture(GL_TEXTURE_2D, es->texture);
 358         glBindBuffer(GL_ARRAY_BUFFER, ec->vbo);
 359         glVertexAttribPointer(0, 3, GL_FLOAT, GL_FALSE,
 360                               5 * sizeof(GLfloat), NULL);
 361         glVertexAttribPointer(1, 2, GL_FLOAT, GL_FALSE,
 362                               5 * sizeof(GLfloat), (GLfloat *) 0 + 3);
 363         glEnableVertexAttribArray(0);
 364         glEnableVertexAttribArray(1);
 365         glDrawArrays(GL_TRIANGLE_STRIP, 0, 4);
 366 }
 367
 368 static void
 369 wlsc_surface_raise(struct wlsc_surface *surface)
 370 {
 371         struct wlsc_compositor *compositor = surface->compositor;
 372
 373         wl_list_remove(&surface->link);
 374         wl_list_insert(&compositor->surface_list, &surface->link);
 375 }
 376
 377 void
 378 wlsc_surface_update_matrix(struct wlsc_surface *es)
 379 {
 380         wlsc_matrix_init(&es->matrix);
 381         wlsc_matrix_scale(&es->matrix, es->width, es->height, 1);
 382         wlsc_matrix_translate(&es->matrix, es->x, es->y, 0);
 383
 384         wlsc_matrix_init(&es->matrix_inv);
 385         wlsc_matrix_translate(&es->matrix_inv, -es->x, -es->y, 0);
 386         wlsc_matrix_scale(&es->matrix_inv,
 387                           1.0 / es->width, 1.0 / es->height, 1);
 388 }
 389
 390 void
 391 wlsc_compositor_finish_frame(struct wlsc_compositor *compositor, int msecs)
 392 {
 393         wl_display_post_frame(compositor->wl_display, msecs);
 394         wl_event_source_timer_update(compositor->timer_source, 5);
 395 }
 396
 397 static void
 398 wlsc_output_repaint(struct wlsc_output *output)
 399 {
 400         struct wlsc_compositor *ec = output->compositor;
 401         struct wlsc_surface *es;
 402         struct wlsc_input_device *eid;
 403
 404         glViewport(0, 0, output->width, output->height);
 405
 406         es = container_of(ec->surface_list.next, struct wlsc_surface, link);
 407         if (es->map_type == WLSC_SURFACE_MAP_FULLSCREEN &&
 408             es->fullscreen_output == output) {
 409                 if (es->width < output->width || es->height < output->height)
 410                         glClear(GL_COLOR_BUFFER_BIT);
 411                 wlsc_surface_draw(es, output);
 412         } else {
 413                 if (output->background)
 414                         wlsc_surface_draw(output->background, output);
 415                 else
 416                         glClear(GL_COLOR_BUFFER_BIT);
 417
 418                 wl_list_for_each_reverse(es, &ec->surface_list, link)
 419                         wlsc_surface_draw(es, output);
 420         }
 421
 422         if (ec->focus)
 423                 wl_list_for_each(eid, &ec->input_device_list, link)
 424                         wlsc_surface_draw(eid->sprite, output);
 425 }
 426
 427 static void
 428 repaint(void *data)
 429 {
 430         struct wlsc_compositor *ec = data;
 431         struct wlsc_output *output;
 432
 433         if (!ec->repaint_needed) {
 434                 ec->repaint_on_timeout = 0;
 435                 return;
 436         }
 437
 438         wl_list_for_each(output, &ec->output_list, link)
 439                 wlsc_output_repaint(output);
 440
 441         ec->present(ec);
 442
 443         ec->repaint_needed = 0;
 444 }
 445
 446 void
 447 wlsc_compositor_schedule_repaint(struct wlsc_compositor *compositor)
 448 {
 449         compositor->repaint_needed = 1;
 450         if (compositor->repaint_on_timeout)
 451                 return;
 452
 453         wl_event_source_timer_update(compositor->timer_source, 1);
 454         compositor->repaint_on_timeout = 1;
 455 }
 456
 457 static void
 458 surface_destroy(struct wl_client *client,
 459                 struct wl_surface *surface)
 460 {
 461         wl_resource_destroy(&surface->resource, client);
 462 }
 463
 464 static void
 465 surface_attach(struct wl_client *client,
 466                struct wl_surface *surface, struct wl_buffer *buffer,
 467                int32_t x, int32_t y)
 468 {
 469         struct wlsc_surface *es = (struct wlsc_surface *) surface;
 470
 471         buffer->attach(buffer, surface);
 472         es->buffer = buffer;
 473         es->x += x;
 474         es->y += y;
 475         es->width = buffer->width;
 476         es->height = buffer->height;
 477         wlsc_surface_update_matrix(es);
 478 }
 479
 480 static void
 481 surface_map_toplevel(struct wl_client *client,
 482                      struct wl_surface *surface)
 483 {
 484         struct wlsc_surface *es = (struct wlsc_surface *) surface;
 485
 486         switch (es->map_type) {
 487         case WLSC_SURFACE_MAP_UNMAPPED:
 488                 es->x = 10 + random() % 400;
 489                 es->y = 10 + random() % 400;
 490                 wlsc_surface_update_matrix(es);
 491                 wl_list_insert(&es->compositor->surface_list, &es->link);
 492                 break;
 493         case WLSC_SURFACE_MAP_TOPLEVEL:
 494                 return;
 495         case WLSC_SURFACE_MAP_FULLSCREEN:
 496                 es->fullscreen_output = NULL;
 497                 es->x = es->saved_x;
 498                 es->y = es->saved_y;
 499                 wlsc_surface_update_matrix(es);
 500                 break;
 501         default:
 502                 break;
 503         }
 504
 505         wlsc_compositor_schedule_repaint(es->compositor);
 506         es->map_type = WLSC_SURFACE_MAP_TOPLEVEL;
 507 }
 508
 509 static void
 510 surface_map_transient(struct wl_client *client,
 511                       struct wl_surface *surface, struct wl_surface *parent,
 512                       int x, int y, uint32_t flags)
 513 {
 514         struct wlsc_surface *es = (struct wlsc_surface *) surface;
 515         struct wlsc_surface *pes = (struct wlsc_surface *) parent;
 516
 517         switch (es->map_type) {
 518         case WLSC_SURFACE_MAP_UNMAPPED:
 519                 wl_list_insert(&es->compositor->surface_list, &es->link);
 520                 break;
 521         case WLSC_SURFACE_MAP_FULLSCREEN:
 522                 es->fullscreen_output = NULL;
 523                 break;
 524         default:
 525                 break;
 526         }
 527
 528         es->x = pes->x + x;
 529         es->y = pes->y + y;
 530
 531         wlsc_surface_update_matrix(es);
 532         wlsc_compositor_schedule_repaint(es->compositor);
 533         es->map_type = WLSC_SURFACE_MAP_TRANSIENT;
 534 }
 535
 536 static void
 537 surface_map_fullscreen(struct wl_client *client, struct wl_surface *surface)
 538 {
 539         struct wlsc_surface *es = (struct wlsc_surface *) surface;
 540         struct wlsc_output *output;
 541
 542         switch (es->map_type) {
 543         case WLSC_SURFACE_MAP_UNMAPPED:
 544                 es->x = 10 + random() % 400;
 545                 es->y = 10 + random() % 400;
 546                 wl_list_insert(&es->compositor->surface_list, &es->link);
 547                 break;
 548         case WLSC_SURFACE_MAP_FULLSCREEN:
 549                 return;
 550         default:
 551                 break;
 552         }
 553
 554         /* FIXME: Fullscreen on first output */
 555         /* FIXME: Handle output going away */
 556         output = container_of(es->compositor->output_list.next,
 557                               struct wlsc_output, link);
 558
 559         es->saved_x = es->x;
 560         es->saved_y = es->y;
 561         es->x = (output->width - es->width) / 2;
 562         es->y = (output->height - es->height) / 2;
 563         es->fullscreen_output = output;
 564         wlsc_surface_update_matrix(es);
 565         wlsc_compositor_schedule_repaint(es->compositor);
 566         es->map_type = WLSC_SURFACE_MAP_FULLSCREEN;
 567 }
 568
 569 static void
 570 surface_damage(struct wl_client *client,
 571                struct wl_surface *surface,
 572                int32_t x, int32_t y, int32_t width, int32_t height)
 573 {
 574         struct wlsc_surface *es = (struct wlsc_surface *) surface;
 575
 576         es->buffer->damage(es->buffer, surface, x, y, width, height);
 577         wlsc_compositor_schedule_repaint(es->compositor);
 578 }
 579
 580 const static struct wl_surface_interface surface_interface = {
 581         surface_destroy,
 582         surface_attach,
 583         surface_map_toplevel,
 584         surface_map_transient,
 585         surface_map_fullscreen,
 586         surface_damage
 587 };
 588
 589 static void
 590 wlsc_input_device_attach(struct wlsc_input_device *device,
 591                          struct wl_buffer *buffer, int x, int y)
 592 {
 593         struct wlsc_compositor *ec =
 594                 (struct wlsc_compositor *) device->input_device.compositor;
 595
 596         buffer->attach(buffer, &device->sprite->surface);
 597         device->hotspot_x = x;
 598         device->hotspot_y = y;
 599
 600         device->sprite->x = device->input_device.x - device->hotspot_x;
 601         device->sprite->y = device->input_device.y - device->hotspot_y;
 602         device->sprite->width = buffer->width;
 603         device->sprite->height = buffer->height;
 604         wlsc_surface_update_matrix(device->sprite);
 605
 606         wlsc_compositor_schedule_repaint(ec);
 607 }
 608
 609
 610 void
 611 wlsc_input_device_set_pointer_image(struct wlsc_input_device *device,
 612                                     enum wlsc_pointer_type type)
 613 {
 614         struct wlsc_compositor *compositor =
 615                 (struct wlsc_compositor *) device->input_device.compositor;
 616
 617         wlsc_input_device_attach(device,
 618                                  compositor->pointer_buffers[type],
 619                                  pointer_images[type].hotspot_x,
 620                                  pointer_images[type].hotspot_y);
 621 }
 622
 623 static void
 624 compositor_create_surface(struct wl_client *client,
 625                           struct wl_compositor *compositor, uint32_t id)
 626 {
 627         struct wlsc_compositor *ec = (struct wlsc_compositor *) compositor;
 628         struct wlsc_surface *surface;
 629
 630         surface = wlsc_surface_create(ec, 0, 0, 0, 0);
 631         if (surface == NULL) {
 632                 wl_client_post_no_memory(client);
 633                 return;
 634         }
 635
 636         surface->surface.resource.destroy = destroy_surface;
 637
 638         surface->surface.resource.object.id = id;
 639         surface->surface.resource.object.interface = &wl_surface_interface;
 640         surface->surface.resource.object.implementation =
 641                 (void (**)(void)) &surface_interface;
 642         surface->surface.client = client;
 643
 644         wl_client_add_resource(client, &surface->surface.resource);
 645 }
 646
 647 const static struct wl_compositor_interface compositor_interface = {
 648         compositor_create_surface,
 649 };
 650
 651 static void
 652 wlsc_surface_transform(struct wlsc_surface *surface,
 653                        int32_t x, int32_t y, int32_t *sx, int32_t *sy)
 654 {
 655         struct wlsc_vector v = { { x, y, 0, 1 } };
 656
 657         wlsc_matrix_transform(&surface->matrix_inv, &v);
 658         *sx = v.f[0] * surface->width;
 659         *sy = v.f[1] * surface->height;
 660 }
 661
 662 struct wlsc_surface *
 663 pick_surface(struct wl_input_device *device, int32_t *sx, int32_t *sy)
 664 {
 665         struct wlsc_compositor *ec =
 666                 (struct wlsc_compositor *) device->compositor;
 667         struct wlsc_surface *es;
 668
 669         wl_list_for_each(es, &ec->surface_list, link) {
 670                 wlsc_surface_transform(es, device->x, device->y, sx, sy);
 671                 if (0 <= *sx && *sx < es->width &&
 672                     0 <= *sy && *sy < es->height)
 673                         return es;
 674         }
 675
 676         return NULL;
 677 }
 678
 679
 680 static void
 681 motion_grab_motion(struct wl_grab *grab,
 682                    uint32_t time, int32_t x, int32_t y)
 683 {
 684         struct wlsc_input_device *device =
 685                 (struct wlsc_input_device *) grab->input_device;
 686         struct wlsc_surface *es =
 687                 (struct wlsc_surface *) device->input_device.pointer_focus;
 688         int32_t sx, sy;
 689
 690         wlsc_surface_transform(es, x, y, &sx, &sy);
 691         wl_client_post_event(es->surface.client,
 692                              &device->input_device.object,
 693                              WL_INPUT_DEVICE_MOTION,
 694                              time, x, y, sx, sy);
 695 }
 696
 697 static void
 698 motion_grab_button(struct wl_grab *grab,
 699                    uint32_t time, int32_t button, int32_t state)
 700 {
 701         wl_client_post_event(grab->input_device->pointer_focus->client,
 702                              &grab->input_device->object,
 703                              WL_INPUT_DEVICE_BUTTON,
 704                              time, button, state);
 705 }
 706
 707 static void
 708 motion_grab_end(struct wl_grab *grab, uint32_t time)
 709 {
 710 }
 711
 712 static const struct wl_grab_interface motion_grab_interface = {
 713         motion_grab_motion,
 714         motion_grab_button,
 715         motion_grab_end
 716 };
 717
 718 void
 719 notify_motion(struct wl_input_device *device, uint32_t time, int x, int y)
 720 {
 721         struct wlsc_surface *es;
 722         struct wlsc_compositor *ec =
 723                 (struct wlsc_compositor *) device->compositor;
 724         struct wlsc_output *output;
 725         const struct wl_grab_interface *interface;
 726         struct wlsc_input_device *wd = (struct wlsc_input_device *) device;
 727         int32_t sx, sy;
 728
 729         /* FIXME: We need some multi head love here. */
 730         output = container_of(ec->output_list.next, struct wlsc_output, link);
 731         if (x < output->x)
 732                 x = 0;
 733         if (y < output->y)
 734                 y = 0;
 735         if (x >= output->x + output->width)
 736                 x = output->x + output->width - 1;
 737         if (y >= output->y + output->height)
 738                 y = output->y + output->height - 1;
 739
 740         device->x = x;
 741         device->y = y;
 742
 743         if (device->grab) {
 744                 interface = device->grab->interface;
 745                 interface->motion(device->grab, time, x, y);
 746         } else {
 747                 es = pick_surface(device, &sx, &sy);
 748                 wl_input_device_set_pointer_focus(device,
 749                                                   &es->surface,
 750                                                   time, x, y, sx, sy);
 751                 if (es)
 752                         wl_client_post_event(es->surface.client,
 753                                              &device->object,
 754                                              WL_INPUT_DEVICE_MOTION,
 755                                              time, x, y, sx, sy);
 756         }
 757
 758         wd->sprite->x = device->x - wd->hotspot_x;
 759         wd->sprite->y = device->y - wd->hotspot_y;
 760         wlsc_surface_update_matrix(wd->sprite);
 761
 762         wlsc_compositor_schedule_repaint(ec);
 763 }
 764
 765 void
 766 notify_button(struct wl_input_device *device,
 767               uint32_t time, int32_t button, int32_t state)
 768 {
 769         struct wlsc_input_device *wd = (struct wlsc_input_device *) device;
 770         struct wlsc_surface *surface;
 771         struct wlsc_compositor *compositor =
 772                 (struct wlsc_compositor *) device->compositor;
 773
 774         surface = (struct wlsc_surface *) device->pointer_focus;
 775
 776         if (state && surface && device->grab == NULL) {
 777                 wlsc_surface_raise(surface);
 778
 779                 if (wd->selection)
 780                         wlsc_selection_set_focus(wd->selection,
 781                                                  &surface->surface, time);
 782
 783                 wl_input_device_start_grab(device,
 784                                            &device->motion_grab,
 785                                            button, time);
 786                 wl_input_device_set_keyboard_focus(device,
 787                                                    &surface->surface,
 788                                                    time);
 789         }
 790
 791         if (state && surface && button == BTN_LEFT &&
 792             (wd->modifier_state & MODIFIER_SUPER))
 793                 shell_move(NULL,
 794                            (struct wl_shell *) &compositor->shell,
 795                            &surface->surface, device, time);
 796         else if (state && surface && button == BTN_MIDDLE &&
 797                  (wd->modifier_state & MODIFIER_SUPER))
 798                 shell_resize(NULL,
 799                              (struct wl_shell *) &compositor->shell,
 800                              &surface->surface, device, time,
 801                              WL_SHELL_RESIZE_BOTTOM_RIGHT);
 802
 803         if (device->grab)
 804                 device->grab->interface->button(device->grab, time,
 805                                                 button, state);
 806
 807         if (!state && device->grab && device->grab_button == button)
 808                 wl_input_device_end_grab(device, time);
 809 }
 810
 811 void
 812 notify_key(struct wl_input_device *device,
 813            uint32_t time, uint32_t key, uint32_t state)
 814 {
 815         struct wlsc_input_device *wd = (struct wlsc_input_device *) device;
 816         struct wlsc_compositor *compositor =
 817                 (struct wlsc_compositor *) device->compositor;
 818         uint32_t *k, *end;
 819         uint32_t modifier;
 820
 821         switch (key | wd->modifier_state) {
 822         case KEY_BACKSPACE | MODIFIER_CTRL | MODIFIER_ALT:
 823                 wl_display_terminate(compositor->wl_display);
 824                 return;
 825         }
 826
 827         switch (key) {
 828         case KEY_LEFTCTRL:
 829         case KEY_RIGHTCTRL:
 830                 modifier = MODIFIER_CTRL;
 831                 break;
 832
 833         case KEY_LEFTALT:
 834         case KEY_RIGHTALT:
 835                 modifier = MODIFIER_ALT;
 836                 break;
 837
 838         case KEY_LEFTMETA:
 839         case KEY_RIGHTMETA:
 840                 modifier = MODIFIER_SUPER;
 841                 break;
 842
 843         default:
 844                 modifier = 0;
 845                 break;
 846         }
 847
 848         if (state)
 849                 wd->modifier_state |= modifier;
 850         else
 851                 wd->modifier_state &= ~modifier;
 852
 853         end = device->keys.data + device->keys.size;
 854         for (k = device->keys.data; k < end; k++) {
 855                 if (*k == key)
 856                         *k = *--end;
 857         }
 858         device->keys.size = (void *) end - device->keys.data;
 859         if (state) {
 860                 k = wl_array_add(&device->keys, sizeof *k);
 861                 *k = key;
 862         }
 863
 864         if (device->keyboard_focus != NULL)
 865                 wl_client_post_event(device->keyboard_focus->client,
 866                                      &device->object,
 867                                      WL_INPUT_DEVICE_KEY, time, key, state);
 868 }
 869
 870 void
 871 notify_pointer_focus(struct wl_input_device *device,
 872                      uint32_t time, struct wlsc_output *output,
 873                      int32_t x, int32_t y)
 874 {
 875         struct wlsc_input_device *wd = (struct wlsc_input_device *) device;
 876         struct wlsc_compositor *compositor =
 877                 (struct wlsc_compositor *) device->compositor;
 878         struct wlsc_surface *es;
 879         int32_t sx, sy;
 880
 881         if (output) {
 882                 device->x = x;
 883                 device->y = y;
 884                 es = pick_surface(device, &sx, &sy);
 885                 wl_input_device_set_pointer_focus(device,
 886                                                   &es->surface,
 887                                                   time, x, y, sx, sy);
 888
 889                 compositor->focus = 1;
 890
 891                 wd->sprite->x = device->x - wd->hotspot_x;
 892                 wd->sprite->y = device->y - wd->hotspot_y;
 893                 wlsc_surface_update_matrix(wd->sprite);
 894         } else {
 895                 wl_input_device_set_pointer_focus(device, NULL,
 896                                                   time, 0, 0, 0, 0);
 897                 compositor->focus = 0;
 898         }
 899
 900         wlsc_compositor_schedule_repaint(compositor);
 901 }
 902
 903 void
 904 notify_keyboard_focus(struct wl_input_device *device_base,
 905                       uint32_t time, struct wlsc_output *output,
 906                       struct wl_array *keys)
 907 {
 908         struct wlsc_input_device *device =
 909                 (struct wlsc_input_device *) device_base;
 910
 911         if (output) {
 912                 wl_array_copy(&device->input_device.keys, keys);
 913                 wl_input_device_set_keyboard_focus(&device->input_device,
 914                                                    device->saved_keyboard_focus,
 915                                                    time);
 916         } else {
 917                 device->saved_keyboard_focus =
 918                         device->input_device.keyboard_focus;
 919                 wl_input_device_set_keyboard_focus(&device->input_device,
 920                                                    NULL, time);
 921         }
 922 }
 923
 924
 925 static void
 926 input_device_attach(struct wl_client *client,
 927                     struct wl_input_device *device_base,
 928                     uint32_t time,
 929                     struct wl_buffer *buffer, int32_t x, int32_t y)
 930 {
 931         struct wlsc_input_device *device =
 932                 (struct wlsc_input_device *) device_base;
 933
 934         if (time < device->input_device.pointer_focus_time)
 935                 return;
 936         if (device->input_device.pointer_focus == NULL)
 937                 return;
 938         if (device->input_device.pointer_focus->client != client)
 939                 return;
 940
 941         if (buffer == NULL) {
 942                 wlsc_input_device_set_pointer_image(device,
 943                                                     WLSC_POINTER_LEFT_PTR);
 944                 return;
 945         }
 946
 947         wlsc_input_device_attach(device, buffer, x, y);
 948 }
 949
 950 const static struct wl_input_device_interface input_device_interface = {
 951         input_device_attach,
 952 };
 953
 954 void
 955 wlsc_input_device_init(struct wlsc_input_device *device,
 956                        struct wlsc_compositor *ec)
 957 {
 958         wl_input_device_init(&device->input_device, &ec->compositor);
 959
 960         device->input_device.object.interface = &wl_input_device_interface;
 961         device->input_device.object.implementation =
 962                 (void (**)(void)) &input_device_interface;
 963         wl_display_add_object(ec->wl_display, &device->input_device.object);
 964         wl_display_add_global(ec->wl_display, &device->input_device.object, NULL);
 965
 966         device->sprite = wlsc_surface_create(ec,
 967                                              device->input_device.x,
 968                                              device->input_device.y, 32, 32);
 969         device->hotspot_x = 16;
 970         device->hotspot_y = 16;
 971
 972         device->input_device.motion_grab.interface = &motion_grab_interface;
 973
 974         wl_list_insert(ec->input_device_list.prev, &device->link);
 975
 976         wlsc_input_device_set_pointer_image(device, WLSC_POINTER_LEFT_PTR);
 977 }
 978
 979 static void
 980 wlsc_output_post_geometry(struct wl_client *client, struct wl_object *global)
 981 {
 982         struct wlsc_output *output =
 983                 container_of(global, struct wlsc_output, object);
 984
 985         wl_client_post_event(client, global,
 986                              WL_OUTPUT_GEOMETRY,
 987                              output->x, output->y,
 988                              output->width, output->height);
 989 }
 990
 991 static const char vertex_shader[] =
 992         "uniform mat4 proj;\n"
 993         "attribute vec4 position;\n"
 994         "attribute vec2 texcoord;\n"
 995         "varying vec2 v_texcoord;\n"
 996         "void main()\n"
 997         "{\n"
 998         "   gl_Position = proj * position;\n"
 999         "   v_texcoord = texcoord;\n"
1000         "}\n";
1001
1002 static const char fragment_shader[] =
1003         "precision mediump float;\n"
1004         "varying vec2 v_texcoord;\n"
1005         "uniform sampler2D tex;\n"
1006         "void main()\n"
1007         "{\n"
1008         "   gl_FragColor = texture2D(tex, v_texcoord)\n;"
1009         "}\n";
1010
1011 static int
1012 init_shaders(struct wlsc_compositor *ec)
1013 {
1014         GLuint v, f, program;
1015         const char *p;
1016         char msg[512];
1017         GLfloat vertices[4 * 5];
1018         GLint status;
1019
1020         p = vertex_shader;
1021         v = glCreateShader(GL_VERTEX_SHADER);
1022         glShaderSource(v, 1, &p, NULL);
1023         glCompileShader(v);
1024         glGetShaderiv(v, GL_COMPILE_STATUS, &status);
1025         if (!status) {
1026                 glGetShaderInfoLog(v, sizeof msg, NULL, msg);
1027                 fprintf(stderr, "vertex shader info: %s\n", msg);
1028                 return -1;
1029         }
1030
1031         p = fragment_shader;
1032         f = glCreateShader(GL_FRAGMENT_SHADER);
1033         glShaderSource(f, 1, &p, NULL);
1034         glCompileShader(f);
1035         glGetShaderiv(f, GL_COMPILE_STATUS, &status);
1036         if (!status) {
1037                 glGetShaderInfoLog(f, sizeof msg, NULL, msg);
1038                 fprintf(stderr, "fragment shader info: %s\n", msg);
1039                 return -1;
1040         }
1041
1042         program = glCreateProgram();
1043         glAttachShader(program, v);
1044         glAttachShader(program, f);
1045         glBindAttribLocation(program, 0, "position");
1046         glBindAttribLocation(program, 1, "texcoord");
1047
1048         glLinkProgram(program);
1049         glGetProgramiv(program, GL_LINK_STATUS, &status);
1050         if (!status) {
1051                 glGetProgramInfoLog(program, sizeof msg, NULL, msg);
1052                 fprintf(stderr, "link info: %s\n", msg);
1053                 return -1;
1054         }
1055
1056         glUseProgram(program);
1057         ec->proj_uniform = glGetUniformLocation(program, "proj");
1058         ec->tex_uniform = glGetUniformLocation(program, "tex");
1059
1060         vertices[ 0] = 0.0;
1061         vertices[ 1] = 0.0;
1062         vertices[ 2] = 0.0;
1063         vertices[ 3] = 0.0;
1064         vertices[ 4] = 0.0;
1065
1066         vertices[ 5] = 0.0;
1067         vertices[ 6] = 1.0;
1068         vertices[ 7] = 0.0;
1069         vertices[ 8] = 0.0;
1070         vertices[ 9] = 1.0;
1071
1072         vertices[10] = 1.0;
1073         vertices[11] = 0.0;
1074         vertices[12] = 0.0;
1075         vertices[13] = 1.0;
1076         vertices[14] = 0.0;
1077
1078         vertices[15] = 1.0;
1079         vertices[16] = 1.0;
1080         vertices[17] = 0.0;
1081         vertices[18] = 1.0;
1082         vertices[19] = 1.0;
1083
1084         glGenBuffers(1, &ec->vbo);
1085         glBindBuffer(GL_ARRAY_BUFFER, ec->vbo);
1086         glBufferData(GL_ARRAY_BUFFER, sizeof vertices, vertices, GL_STATIC_DRAW);
1087
1088         return 0;
1089 }
1090
1091 void
1092 wlsc_output_init(struct wlsc_output *output, struct wlsc_compositor *c,
1093                  int x, int y, int width, int height)
1094 {
1095         output->compositor = c;
1096         output->x = x;
1097         output->y = y;
1098         output->width = width;
1099         output->height = height;
1100
1101         output->background =
1102                 background_create(output, option_background);
1103
1104         wlsc_matrix_init(&output->matrix);
1105         wlsc_matrix_translate(&output->matrix,
1106                               -output->x - output->width / 2.0,
1107                               -output->y - output->height / 2.0, 0);
1108         wlsc_matrix_scale(&output->matrix,
1109                           2.0 / output->width, 2.0 / output->height, 1);
1110
1111         output->object.interface = &wl_output_interface;
1112         wl_display_add_object(c->wl_display, &output->object);
1113         wl_display_add_global(c->wl_display, &output->object,
1114                               wlsc_output_post_geometry);
1115 }
1116
1117 int
1118 wlsc_compositor_init(struct wlsc_compositor *ec, struct wl_display *display)
1119 {
1120         struct wl_event_loop *loop;
1121         const char *extensions;
1122
1123         ec->wl_display = display;
1124
1125         wl_compositor_init(&ec->compositor, &compositor_interface, display);
1126
1127         wlsc_shm_init(ec);
1128
1129         wlsc_shell_init(ec);
1130
1131         wl_list_init(&ec->surface_list);
1132         wl_list_init(&ec->input_device_list);
1133         wl_list_init(&ec->output_list);
1134
1135         create_pointer_images(ec);
1136
1137         screenshooter_create(ec);
1138
1139         extensions = (const char *) glGetString(GL_EXTENSIONS);
1140         if (!strstr(extensions, "GL_EXT_texture_format_BGRA8888")) {
1141                 fprintf(stderr,
1142                         "GL_EXT_texture_format_BGRA8888 not available\n");
1143                 return -1;
1144         }
1145
1146         glGenFramebuffers(1, &ec->fbo);
1147         glBindFramebuffer(GL_FRAMEBUFFER, ec->fbo);
1148         glActiveTexture(GL_TEXTURE0);
1149         if (init_shaders(ec) < 0)
1150                 return -1;
1151
1152         loop = wl_display_get_event_loop(ec->wl_display);
1153         ec->timer_source = wl_event_loop_add_timer(loop, repaint, ec);
1154         wlsc_compositor_schedule_repaint(ec);
1155
1156         return 0;
1157 }
1158
1159 static void on_term_signal(int signal_number, void *data)
1160 {
1161         struct wlsc_compositor *ec = data;
1162
1163         wl_display_terminate(ec->wl_display);
1164 }
1165
1166 int main(int argc, char *argv[])
1167 {
1168         struct wl_display *display;
1169         struct wlsc_compositor *ec;
1170         struct wl_event_loop *loop;
1171         GError *error = NULL;
1172         GOptionContext *context;
1173         int width, height;
1174
1175         g_type_init(); /* GdkPixbuf needs this, it seems. */
1176
1177         context = g_option_context_new(NULL);
1178         g_option_context_add_main_entries(context, option_entries, "Wayland");
1179         if (!g_option_context_parse(context, &argc, &argv, &error)) {
1180                 fprintf(stderr, "option parsing failed: %s\n", error->message);
1181                 exit(EXIT_FAILURE);
1182         }
1183         if (sscanf(option_geometry, "%dx%d", &width, &height) != 2) {
1184                 fprintf(stderr, "invalid geometry option: %s \n",
1185                         option_geometry);
1186                 exit(EXIT_FAILURE);
1187         }
1188
1189         display = wl_display_create();
1190
1191         ec = NULL;
1192
1193 #if BUILD_WAYLAND_COMPOSITOR
1194         if (getenv("WAYLAND_DISPLAY"))
1195                 ec = wayland_compositor_create(display, width, height);
1196 #endif
1197
1198 #if BUILD_X11_COMPOSITOR
1199         if (ec == NULL && getenv("DISPLAY"))
1200                 ec = x11_compositor_create(display, width, height);
1201 #endif
1202
1203 #if BUILD_DRM_COMPOSITOR
1204         if (ec == NULL)
1205                 ec = drm_compositor_create(display, option_connector);
1206 #endif
1207
1208         if (ec == NULL) {
1209                 fprintf(stderr, "failed to create compositor\n");
1210                 exit(EXIT_FAILURE);
1211         }
1212
1213         if (wl_display_add_socket(display, option_socket_name)) {
1214                 fprintf(stderr, "failed to add socket: %m\n");
1215                 exit(EXIT_FAILURE);
1216         }
1217
1218         loop = wl_display_get_event_loop(ec->wl_display);
1219         wl_event_loop_add_signal(loop, SIGTERM, on_term_signal, ec);
1220         wl_event_loop_add_signal(loop, SIGINT, on_term_signal, ec);
1221
1222         wl_display_run(display);
1223
1224         wl_display_destroy(display);
1225
1226         ec->destroy(ec);
1227
1228         return 0;
1229 }