clients/dnd.c

   1 /*
   2  * Copyright © 2010 Kristian Høgsberg
   3  *
   4  * Permission to use, copy, modify, distribute, and sell this software and its
   5  * documentation for any purpose is hereby granted without fee, provided that
   6  * the above copyright notice appear in all copies and that both that copyright
   7  * notice and this permission notice appear in supporting documentation, and
   8  * that the name of the copyright holders not be used in advertising or
   9  * publicity pertaining to distribution of the software without specific,
  10  * written prior permission.  The copyright holders make no representations
  11  * about the suitability of this software for any purpose.  It is provided "as
  12  * is" without express or implied warranty.
  13  *
  14  * THE COPYRIGHT HOLDERS DISCLAIM ALL WARRANTIES WITH REGARD TO THIS SOFTWARE,
  15  * INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS, IN NO
  16  * EVENT SHALL THE COPYRIGHT HOLDERS BE LIABLE FOR ANY SPECIAL, INDIRECT OR
  17  * CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE,
  18  * DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER
  19  * TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
  20  * OF THIS SOFTWARE.
  21  */
  22
  23 #include <assert.h>
  24 #include <stdint.h>
  25 #include <stdio.h>
  26 #include <stdlib.h>
  27 #include <string.h>
  28 #include <fcntl.h>
  29 #include <unistd.h>
  30 #include <math.h>
  31 #include <sys/time.h>
  32 #include <cairo.h>
  33 #include <sys/epoll.h>
  34
  35 #include <wayland-client.h>
  36 #include <wayland-cursor.h>
  37
  38 #include "window.h"
  39 #include "../shared/cairo-util.h"
  40
  41 struct dnd_drag;
  42
  43 struct dnd {
  44         struct window *window;
  45         struct widget *widget;
  46         struct display *display;
  47         uint32_t key;
  48         struct item *items[16];
  49         int self_only;
  50         struct dnd_drag *current_drag;
  51 };
  52
  53 struct dnd_drag {
  54         cairo_surface_t *translucent;
  55         cairo_surface_t *opaque;
  56         int hotspot_x, hotspot_y;
  57         struct dnd *dnd;
  58         struct input *input;
  59         uint32_t time;
  60         struct item *item;
  61         int x_offset, y_offset;
  62         int width, height;
  63         const char *mime_type;
  64
  65         struct wl_surface *drag_surface;
  66         struct wl_data_source *data_source;
  67 };
  68
  69 struct item {
  70         cairo_surface_t *surface;
  71         int seed;
  72         int x, y;
  73 };
  74
  75 struct dnd_flower_message {
  76         int seed, x_offset, y_offset;
  77 };
  78
  79
  80 static const int item_width = 64;
  81 static const int item_height = 64;
  82 static const int item_padding = 16;
  83
  84 static struct item *
  85 item_create(struct display *display, int x, int y, int seed)
  86 {
  87         struct item *item;
  88         struct timeval tv;
  89
  90         item = malloc(sizeof *item);
  91         if (item == NULL)
  92                 return NULL;
  93
  94
  95         gettimeofday(&tv, NULL);
  96         item->seed = seed ? seed : tv.tv_usec;
  97         srandom(item->seed);
  98
  99         const int petal_count = 3 + random() % 5;
 100         const double r1 = 20 + random() % 10;
 101         const double r2 = 5 + random() % 12;
 102         const double u = (10 + random() % 90) / 100.0;
 103         const double v = (random() % 90) / 100.0;
 104
 105         cairo_t *cr;
 106         int i;
 107         double t, dt = 2 * M_PI / (petal_count * 2);
 108         double x1, y1, x2, y2, x3, y3;
 109         struct rectangle rect;
 110
 111
 112         rect.width = item_width;
 113         rect.height = item_height;
 114         item->surface =
 115                 display_create_surface(display, NULL, &rect, SURFACE_SHM);
 116
 117         item->x = x;
 118         item->y = y;
 119
 120         cr = cairo_create(item->surface);
 121         cairo_set_operator(cr, CAIRO_OPERATOR_SOURCE);
 122         cairo_set_source_rgba(cr, 0, 0, 0, 0);
 123         cairo_paint(cr);
 124
 125         cairo_set_operator(cr, CAIRO_OPERATOR_OVER);
 126         cairo_translate(cr, item_width / 2, item_height / 2);
 127         t = random();
 128         cairo_move_to(cr, cos(t) * r1, sin(t) * r1);
 129         for (i = 0; i < petal_count; i++, t += dt * 2) {
 130                 x1 = cos(t) * r1;
 131                 y1 = sin(t) * r1;
 132                 x2 = cos(t + dt) * r2;
 133                 y2 = sin(t + dt) * r2;
 134                 x3 = cos(t + 2 * dt) * r1;
 135                 y3 = sin(t + 2 * dt) * r1;
 136
 137                 cairo_curve_to(cr,
 138                                x1 - y1 * u, y1 + x1 * u,
 139                                x2 + y2 * v, y2 - x2 * v,
 140                                x2, y2);
 141
 142                 cairo_curve_to(cr,
 143                                x2 - y2 * v, y2 + x2 * v,
 144                                x3 + y3 * u, y3 - x3 * u,
 145                                x3, y3);
 146         }
 147
 148         cairo_close_path(cr);
 149
 150         cairo_set_source_rgba(cr,
 151                               0.5 + (random() % 50) / 49.0,
 152                               0.5 + (random() % 50) / 49.0,
 153                               0.5 + (random() % 50) / 49.0,
 154                               0.5 + (random() % 100) / 99.0);
 155
 156         cairo_fill_preserve(cr);
 157
 158         cairo_set_line_width(cr, 1);
 159         cairo_set_source_rgba(cr,
 160                               0.5 + (random() % 50) / 49.0,
 161                               0.5 + (random() % 50) / 49.0,
 162                               0.5 + (random() % 50) / 49.0,
 163                               0.5 + (random() % 100) / 99.0);
 164         cairo_stroke(cr);
 165
 166         cairo_destroy(cr);
 167
 168         return item;
 169 }
 170
 171 static void
 172 dnd_redraw_handler(struct widget *widget, void *data)
 173 {
 174         struct dnd *dnd = data;
 175         struct rectangle allocation;
 176         cairo_t *cr;
 177         cairo_surface_t *surface;
 178         unsigned int i;
 179
 180         surface = window_get_surface(dnd->window);
 181         cr = cairo_create(surface);
 182         widget_get_allocation(dnd->widget, &allocation);
 183         cairo_rectangle(cr, allocation.x, allocation.y,
 184                         allocation.width, allocation.height);
 185
 186         cairo_set_operator(cr, CAIRO_OPERATOR_SOURCE);
 187         cairo_set_source_rgba(cr, 0, 0, 0, 0.8);
 188         cairo_fill(cr);
 189
 190         cairo_rectangle(cr, allocation.x, allocation.y,
 191                         allocation.width, allocation.height);
 192         cairo_clip(cr);
 193         cairo_set_operator(cr, CAIRO_OPERATOR_OVER);
 194         for (i = 0; i < ARRAY_LENGTH(dnd->items); i++) {
 195                 if (!dnd->items[i])
 196                         continue;
 197                 cairo_set_source_surface(cr, dnd->items[i]->surface,
 198                                          dnd->items[i]->x + allocation.x,
 199                                          dnd->items[i]->y + allocation.y);
 200                 cairo_paint(cr);
 201         }
 202
 203         cairo_destroy(cr);
 204         cairo_surface_destroy(surface);
 205 }
 206
 207 static void
 208 keyboard_focus_handler(struct window *window,
 209                        struct input *device, void *data)
 210 {
 211         struct dnd *dnd = data;
 212
 213         window_schedule_redraw(dnd->window);
 214 }
 215
 216 static int
 217 dnd_add_item(struct dnd *dnd, struct item *item)
 218 {
 219         unsigned int i;
 220
 221         for (i = 0; i < ARRAY_LENGTH(dnd->items); i++) {
 222                 if (dnd->items[i] == 0) {
 223                         dnd->items[i] = item;
 224                         return i;
 225                 }
 226         }
 227         return -1;
 228 }
 229
 230 static struct item *
 231 dnd_get_item(struct dnd *dnd, int32_t x, int32_t y)
 232 {
 233         struct item *item;
 234         struct rectangle allocation;
 235         unsigned int i;
 236
 237         widget_get_allocation(dnd->widget, &allocation);
 238
 239         x -= allocation.x;
 240         y -= allocation.y;
 241
 242         for (i = 0; i < ARRAY_LENGTH(dnd->items); i++) {
 243                 item = dnd->items[i];
 244                 if (item &&
 245                     item->x <= x && x < item->x + item_width &&
 246                     item->y <= y && y < item->y + item_height)
 247                         return item;
 248         }
 249
 250         return NULL;
 251 }
 252
 253 static void
 254 data_source_target(void *data,
 255                    struct wl_data_source *source, const char *mime_type)
 256 {
 257         struct dnd_drag *dnd_drag = data;
 258         struct dnd *dnd = dnd_drag->dnd;
 259         cairo_surface_t *surface;
 260         struct wl_buffer *buffer;
 261
 262         dnd_drag->mime_type = mime_type;
 263         if (mime_type)
 264                 surface = dnd_drag->opaque;
 265         else
 266                 surface = dnd_drag->translucent;
 267
 268         buffer = display_get_buffer_for_surface(dnd->display, surface);
 269         wl_surface_attach(dnd_drag->drag_surface, buffer, 0, 0);
 270         wl_surface_damage(dnd_drag->drag_surface, 0, 0,
 271                           dnd_drag->width, dnd_drag->height);
 272         wl_surface_commit(dnd_drag->drag_surface);
 273 }
 274
 275 static void
 276 data_source_send(void *data, struct wl_data_source *source,
 277                  const char *mime_type, int32_t fd)
 278 {
 279         struct dnd_flower_message dnd_flower_message;
 280         struct dnd_drag *dnd_drag = data;
 281
 282         dnd_flower_message.seed = dnd_drag->item->seed;
 283         dnd_flower_message.x_offset = dnd_drag->x_offset;
 284         dnd_flower_message.y_offset = dnd_drag->y_offset;
 285
 286         if (write(fd, &dnd_flower_message, sizeof dnd_flower_message) < 0)
 287                 abort();
 288
 289         close(fd);
 290 }
 291
 292 static void
 293 data_source_cancelled(void *data, struct wl_data_source *source)
 294 {
 295         struct dnd_drag *dnd_drag = data;
 296
 297         /* The 'cancelled' event means that the source is no longer in
 298          * use by the drag (or current selection).  We need to clean
 299          * up the drag object created and the local state. */
 300
 301         wl_data_source_destroy(dnd_drag->data_source);
 302
 303         /* Destroy the item that has been dragged out */
 304         cairo_surface_destroy(dnd_drag->item->surface);
 305         free(dnd_drag->item);
 306
 307         wl_surface_destroy(dnd_drag->drag_surface);
 308
 309         cairo_surface_destroy(dnd_drag->translucent);
 310         cairo_surface_destroy(dnd_drag->opaque);
 311         free(dnd_drag);
 312 }
 313
 314 static const struct wl_data_source_listener data_source_listener = {
 315         data_source_target,
 316         data_source_send,
 317         data_source_cancelled
 318 };
 319
 320 static cairo_surface_t *
 321 create_drag_cursor(struct dnd_drag *dnd_drag,
 322                    struct item *item, int32_t x, int32_t y, double opacity)
 323 {
 324         struct dnd *dnd = dnd_drag->dnd;
 325         cairo_surface_t *surface;
 326         struct wl_cursor_image *pointer;
 327         struct rectangle rectangle;
 328         cairo_pattern_t *pattern;
 329         cairo_t *cr;
 330
 331         pointer = display_get_pointer_image(dnd->display, CURSOR_DRAGGING);
 332         if (!pointer) {
 333                 fprintf(stderr, "WARNING: grabbing cursor image not found\n");
 334                 pointer = display_get_pointer_image(dnd->display,
 335                                                     CURSOR_LEFT_PTR);
 336                 assert(pointer && "no cursor image found");
 337         }
 338
 339         rectangle.width = item_width + 2 * pointer->width;
 340         rectangle.height = item_height + 2 * pointer->height;
 341         surface = display_create_surface(dnd->display, NULL, &rectangle,
 342                                          SURFACE_SHM);
 343
 344         cr = cairo_create(surface);
 345         cairo_translate(cr, pointer->width, pointer->height);
 346
 347         cairo_set_operator(cr, CAIRO_OPERATOR_SOURCE);
 348         cairo_set_source_rgba(cr, 0, 0, 0, 0);
 349         cairo_paint(cr);
 350
 351         cairo_set_operator(cr, CAIRO_OPERATOR_OVER);
 352         cairo_set_source_surface(cr, item->surface, 0, 0);
 353         pattern = cairo_pattern_create_rgba(0, 0, 0, opacity);
 354         cairo_mask(cr, pattern);
 355         cairo_pattern_destroy(pattern);
 356
 357         /* FIXME: more cairo-gl brokeness */
 358         surface_flush_device(surface);
 359         cairo_destroy(cr);
 360
 361         dnd_drag->hotspot_x = pointer->width + x - item->x;
 362         dnd_drag->hotspot_y = pointer->height + y - item->y;
 363         dnd_drag->width = rectangle.width;
 364         dnd_drag->height = rectangle.height;
 365
 366         return surface;
 367 }
 368
 369 static void
 370 dnd_button_handler(struct widget *widget,
 371                    struct input *input, uint32_t time,
 372                    uint32_t button, enum wl_pointer_button_state state,
 373                    void *data)
 374 {
 375         struct dnd *dnd = data;
 376         int32_t x, y;
 377         struct item *item;
 378         struct rectangle allocation;
 379         struct dnd_drag *dnd_drag;
 380         struct display *display;
 381         struct wl_compositor *compositor;
 382         struct wl_buffer *buffer;
 383         unsigned int i;
 384         uint32_t serial;
 385         cairo_surface_t *icon;
 386
 387         widget_get_allocation(dnd->widget, &allocation);
 388         input_get_position(input, &x, &y);
 389         item = dnd_get_item(dnd, x, y);
 390         x -= allocation.x;
 391         y -= allocation.y;
 392
 393         if (item && state == WL_POINTER_BUTTON_STATE_PRESSED) {
 394                 dnd_drag = malloc(sizeof *dnd_drag);
 395                 dnd_drag->dnd = dnd;
 396                 dnd_drag->input = input;
 397                 dnd_drag->time = time;
 398                 dnd_drag->item = item;
 399                 dnd_drag->x_offset = x - item->x;
 400                 dnd_drag->y_offset = y - item->y;
 401
 402                 for (i = 0; i < ARRAY_LENGTH(dnd->items); i++) {
 403                         if (item == dnd->items[i]){
 404                                 dnd->items[i] = 0;
 405                                 break;
 406                         }
 407                 }
 408
 409                 display = window_get_display(dnd->window);
 410                 compositor = display_get_compositor(display);
 411                 serial = display_get_serial(display);
 412                 dnd_drag->drag_surface =
 413                         wl_compositor_create_surface(compositor);
 414
 415                 input_ungrab(input);
 416
 417                 if (dnd->self_only) {
 418                         dnd_drag->data_source = NULL;
 419                 } else {
 420                         dnd_drag->data_source =
 421                                 display_create_data_source(dnd->display);
 422                         wl_data_source_add_listener(dnd_drag->data_source,
 423                                                     &data_source_listener,
 424                                                     dnd_drag);
 425                         wl_data_source_offer(dnd_drag->data_source,
 426                                              "application/x-wayland-dnd-flower");
 427                         wl_data_source_offer(dnd_drag->data_source,
 428                                              "text/plain; charset=utf-8");
 429                 }
 430
 431                 wl_data_device_start_drag(input_get_data_device(input),
 432                                           dnd_drag->data_source,
 433                                           window_get_wl_surface(dnd->window),
 434                                           dnd_drag->drag_surface,
 435                                           serial);
 436
 437                 input_set_pointer_image(input, CURSOR_DRAGGING);
 438
 439                 dnd_drag->opaque =
 440                         create_drag_cursor(dnd_drag, item, x, y, 1);
 441                 dnd_drag->translucent =
 442                         create_drag_cursor(dnd_drag, item, x, y, 0.2);
 443
 444                 if (dnd->self_only)
 445                         icon = dnd_drag->opaque;
 446                 else
 447                         icon = dnd_drag->translucent;
 448
 449                 buffer = display_get_buffer_for_surface(dnd->display, icon);
 450                 wl_surface_attach(dnd_drag->drag_surface, buffer,
 451                                   -dnd_drag->hotspot_x, -dnd_drag->hotspot_y);
 452                 wl_surface_damage(dnd_drag->drag_surface, 0, 0,
 453                                   dnd_drag->width, dnd_drag->height);
 454                 wl_surface_commit(dnd_drag->drag_surface);
 455
 456                 dnd->current_drag = dnd_drag;
 457                 window_schedule_redraw(dnd->window);
 458         }
 459 }
 460
 461 static int
 462 lookup_cursor(struct dnd *dnd, int x, int y)
 463 {
 464         struct item *item;
 465
 466         item = dnd_get_item(dnd, x, y);
 467         if (item)
 468                 return CURSOR_HAND1;
 469         else
 470                 return CURSOR_LEFT_PTR;
 471 }
 472
 473 static int
 474 dnd_enter_handler(struct widget *widget,
 475                   struct input *input, float x, float y, void *data)
 476 {
 477         struct dnd *dnd = data;
 478
 479         dnd->current_drag = NULL;
 480
 481         return lookup_cursor(dnd, x, y);
 482 }
 483
 484 static int
 485 dnd_motion_handler(struct widget *widget,
 486                    struct input *input, uint32_t time,
 487                    float x, float y, void *data)
 488 {
 489         return lookup_cursor(data, x, y);
 490 }
 491
 492 static void
 493 dnd_data_handler(struct window *window,
 494                  struct input *input,
 495                  float x, float y, const char **types, void *data)
 496 {
 497         struct dnd *dnd = data;
 498
 499         if (!types)
 500                 return;
 501
 502         if (dnd_get_item(dnd, x, y) || dnd->self_only) {
 503                 input_accept(input, NULL);
 504         } else {
 505                 input_accept(input, types[0]);
 506         }
 507 }
 508
 509 static void
 510 dnd_receive_func(void *data, size_t len, int32_t x, int32_t y, void *user_data)
 511 {
 512         struct dnd *dnd = user_data;
 513         struct dnd_flower_message *message = data;
 514         struct item *item;
 515         struct rectangle allocation;
 516
 517         if (len == 0) {
 518                 return;
 519         } else if (len != sizeof *message) {
 520                 fprintf(stderr, "odd message length %ld, expected %ld\n",
 521                         len, sizeof *message);
 522                 return;
 523         }
 524
 525         widget_get_allocation(dnd->widget, &allocation);
 526         item = item_create(dnd->display,
 527                            x - message->x_offset - allocation.x,
 528                            y - message->y_offset - allocation.y,
 529                            message->seed);
 530
 531         dnd_add_item(dnd, item);
 532         window_schedule_redraw(dnd->window);
 533 }
 534
 535 static void
 536 dnd_drop_handler(struct window *window, struct input *input,
 537                  int32_t x, int32_t y, void *data)
 538 {
 539         struct dnd *dnd = data;
 540         struct dnd_flower_message message;
 541
 542         if (dnd_get_item(dnd, x, y)) {
 543                 fprintf(stderr, "got 'drop', but no target\n");
 544                 return;
 545         }
 546
 547         if (!dnd->self_only) {
 548                 input_receive_drag_data(input,
 549                                         "application/x-wayland-dnd-flower",
 550                                         dnd_receive_func, dnd);
 551         } else if (dnd->current_drag) {
 552                 message.seed = dnd->current_drag->item->seed;
 553                 message.x_offset = dnd->current_drag->x_offset;
 554                 message.y_offset = dnd->current_drag->y_offset;
 555                 dnd_receive_func(&message, sizeof message, x, y, dnd);
 556                 dnd->current_drag = NULL;
 557         } else {
 558                 fprintf(stderr, "ignoring drop from another client\n");
 559         }
 560 }
 561
 562 static struct dnd *
 563 dnd_create(struct display *display)
 564 {
 565         struct dnd *dnd;
 566         int x, y;
 567         int32_t width, height;
 568         unsigned int i;
 569
 570         dnd = malloc(sizeof *dnd);
 571         if (dnd == NULL)
 572                 return dnd;
 573         memset(dnd, 0, sizeof *dnd);
 574
 575         dnd->window = window_create(display);
 576         dnd->widget = frame_create(dnd->window, dnd);
 577         window_set_title(dnd->window, "Wayland Drag and Drop Demo");
 578
 579         dnd->display = display;
 580         dnd->key = 100;
 581
 582         for (i = 0; i < ARRAY_LENGTH(dnd->items); i++) {
 583                 x = (i % 4) * (item_width + item_padding) + item_padding;
 584                 y = (i / 4) * (item_height + item_padding) + item_padding;
 585                 if ((i ^ (i >> 2)) & 1)
 586                         dnd->items[i] = item_create(display, x, y, 0);
 587                 else
 588                         dnd->items[i] = NULL;
 589         }
 590
 591         window_set_user_data(dnd->window, dnd);
 592         window_set_keyboard_focus_handler(dnd->window,
 593                                           keyboard_focus_handler);
 594         window_set_data_handler(dnd->window, dnd_data_handler);
 595         window_set_drop_handler(dnd->window, dnd_drop_handler);
 596
 597         widget_set_redraw_handler(dnd->widget, dnd_redraw_handler);
 598         widget_set_enter_handler(dnd->widget, dnd_enter_handler);
 599         widget_set_motion_handler(dnd->widget, dnd_motion_handler);
 600         widget_set_button_handler(dnd->widget, dnd_button_handler);
 601
 602         width = 4 * (item_width + item_padding) + item_padding;
 603         height = 4 * (item_height + item_padding) + item_padding;
 604
 605         frame_set_child_size(dnd->widget, width, height);
 606
 607         return dnd;
 608 }
 609
 610 int
 611 main(int argc, char *argv[])
 612 {
 613         struct display *d;
 614         struct dnd *dnd;
 615         int i;
 616
 617         d = display_create(argc, argv);
 618         if (d == NULL) {
 619                 fprintf(stderr, "failed to create display: %m\n");
 620                 return -1;
 621         }
 622
 623         dnd = dnd_create(d);
 624
 625         for (i = 1; i < argc; i++)
 626                 if (strcmp("--self-only", argv[i]) == 0)
 627                         dnd->self_only = 1;
 628
 629         display_run(d);
 630
 631         return 0;
 632 }