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