compositor/compositor-drm.c

   1 /*
   2  * Copyright © 2008-2010 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 <stdio.h>
  22 #include <stdlib.h>
  23 #include <string.h>
  24 #include <fcntl.h>
  25 #include <unistd.h>
  26
  27 #include <xf86drm.h>
  28 #include <xf86drmMode.h>
  29
  30 #define GL_GLEXT_PROTOTYPES
  31 #define EGL_EGLEXT_PROTOTYPES
  32 #include <GLES2/gl2.h>
  33 #include <GLES2/gl2ext.h>
  34 #include <EGL/egl.h>
  35 #include <EGL/eglext.h>
  36
  37 #include "compositor.h"
  38
  39 struct drm_compositor {
  40         struct wlsc_compositor base;
  41
  42         struct udev *udev;
  43         struct wl_event_source *drm_source;
  44
  45         struct udev_monitor *udev_monitor;
  46         struct wl_event_source *udev_drm_source;
  47
  48         struct {
  49                 int fd;
  50         } drm;
  51         uint32_t crtc_allocator;
  52         uint32_t connector_allocator;
  53         struct tty *tty;
  54 };
  55
  56 struct drm_output {
  57         struct wlsc_output   base;
  58
  59         drmModeModeInfo mode;
  60         uint32_t crtc_id;
  61         uint32_t connector_id;
  62         GLuint rbo[2];
  63         uint32_t fb_id[2];
  64         EGLImageKHR image[2];
  65         uint32_t current;
  66
  67         uint32_t fs_surf_fb_id;
  68         uint32_t pending_fs_surf_fb_id;
  69 };
  70
  71 static int
  72 drm_output_prepare_render(struct wlsc_output *output_base)
  73 {
  74         struct drm_output *output = (struct drm_output *) output_base;
  75
  76         glFramebufferRenderbuffer(GL_FRAMEBUFFER,
  77                                   GL_COLOR_ATTACHMENT0,
  78                                   GL_RENDERBUFFER,
  79                                   output->rbo[output->current]);
  80
  81         if (glCheckFramebufferStatus(GL_FRAMEBUFFER) != GL_FRAMEBUFFER_COMPLETE)
  82                 return -1;
  83
  84         return 0;
  85 }
  86
  87 static int
  88 drm_output_present(struct wlsc_output *output_base)
  89 {
  90         struct drm_output *output = (struct drm_output *) output_base;
  91         struct drm_compositor *c =
  92                 (struct drm_compositor *) output->base.compositor;
  93         int ret;
  94         uint32_t fb_id = 0;
  95
  96         if (drm_output_prepare_render(&output->base))
  97                 return -1;
  98         glFlush();
  99
 100         output->current ^= 1;
 101
 102         if (output->base.scanout_surface) {
 103                 EGLint handle, stride;
 104
 105                 eglExportDRMImageMESA(c->base.display,
 106                                       output->base.scanout_surface->image,
 107                                       NULL, &handle, &stride);
 108
 109                 ret = drmModeAddFB(c->drm.fd,
 110                                    output->base.width, output->base.height,
 111                                    32, 32, stride, handle,
 112                                    &output->fs_surf_fb_id);
 113                 if (ret)
 114                         return -1;
 115
 116                 printf("pageflip to fullscreen buffer: %d\n", handle);
 117
 118                 fb_id = output->fs_surf_fb_id;
 119         } else {
 120                 fb_id = output->fb_id[output->current ^ 1];
 121         }
 122
 123         drmModePageFlip(c->drm.fd, output->crtc_id,
 124                         fb_id,
 125                         DRM_MODE_PAGE_FLIP_EVENT, output);
 126
 127         return 0;
 128 }
 129
 130 static void
 131 page_flip_handler(int fd, unsigned int frame,
 132                   unsigned int sec, unsigned int usec, void *data)
 133 {
 134         struct drm_output *output = (struct drm_output *) data;
 135         struct drm_compositor *c =
 136                 (struct drm_compositor *) output->base.compositor;
 137         uint32_t msecs;
 138
 139         if (output->pending_fs_surf_fb_id) {
 140                 drmModeRmFB(c->drm.fd, output->pending_fs_surf_fb_id);
 141                 output->pending_fs_surf_fb_id = 0;
 142         }
 143
 144         if (output->fs_surf_fb_id) {
 145                 output->pending_fs_surf_fb_id = output->fs_surf_fb_id;
 146                 output->fs_surf_fb_id = 0;
 147         }
 148
 149         msecs = sec * 1000 + usec / 1000;
 150         wlsc_output_finish_frame(&output->base, msecs);
 151 }
 152
 153 static int
 154 drm_output_image_is_scanoutable(struct wlsc_output *output_base,
 155                                 EGLImageKHR image)
 156 {
 157         struct drm_output *output = (struct drm_output *) output_base;
 158         struct drm_compositor *c =
 159                 (struct drm_compositor *) output->base.compositor;
 160         EGLint handle, stride;
 161         int ret;
 162         uint32_t fb_id = 0;
 163
 164         eglExportDRMImageMESA(c->base.display, image,
 165                               NULL, &handle, &stride);
 166
 167         ret = drmModeAddFB(c->drm.fd,
 168                            output->base.width, output->base.height,
 169                            32, 32, stride, handle,
 170                            &fb_id);
 171         if (ret)
 172                 return 0;
 173
 174         /* FIXME: change interface to keep this fb_id,
 175          * to be used directly in next pageflip? */
 176         if (fb_id)
 177                 drmModeRmFB(c->drm.fd, fb_id);
 178
 179         return fb_id != 0;
 180 }
 181
 182 static void
 183 on_drm_input(int fd, uint32_t mask, void *data)
 184 {
 185         drmEventContext evctx;
 186
 187         memset(&evctx, 0, sizeof evctx);
 188         evctx.version = DRM_EVENT_CONTEXT_VERSION;
 189         evctx.page_flip_handler = page_flip_handler;
 190         drmHandleEvent(fd, &evctx);
 191 }
 192
 193 static int
 194 init_egl(struct drm_compositor *ec, struct udev_device *device)
 195 {
 196         EGLint major, minor;
 197         const char *extensions, *filename;
 198         int fd;
 199         static const EGLint context_attribs[] = {
 200                 EGL_CONTEXT_CLIENT_VERSION, 2,
 201                 EGL_NONE
 202         };
 203
 204         filename = udev_device_get_devnode(device);
 205         fd = open(filename, O_RDWR, O_CLOEXEC);
 206         if (fd < 0) {
 207                 /* Probably permissions error */
 208                 fprintf(stderr, "couldn't open %s, skipping\n",
 209                         udev_device_get_devnode(device));
 210                 return -1;
 211         }
 212
 213         ec->drm.fd = fd;
 214         ec->base.display = eglGetDRMDisplayMESA(ec->drm.fd);
 215         if (ec->base.display == NULL) {
 216                 fprintf(stderr, "failed to create display\n");
 217                 return -1;
 218         }
 219
 220         if (!eglInitialize(ec->base.display, &major, &minor)) {
 221                 fprintf(stderr, "failed to initialize display\n");
 222                 return -1;
 223         }
 224
 225         extensions = eglQueryString(ec->base.display, EGL_EXTENSIONS);
 226         if (!strstr(extensions, "EGL_KHR_surfaceless_opengl")) {
 227                 fprintf(stderr, "EGL_KHR_surfaceless_opengl not available\n");
 228                 return -1;
 229         }
 230
 231         if (!eglBindAPI(EGL_OPENGL_ES_API)) {
 232                 fprintf(stderr, "failed to bind api EGL_OPENGL_ES_API\n");
 233                 return -1;
 234         }
 235
 236         ec->base.context = eglCreateContext(ec->base.display, NULL,
 237                                             EGL_NO_CONTEXT, context_attribs);
 238         if (ec->base.context == NULL) {
 239                 fprintf(stderr, "failed to create context\n");
 240                 return -1;
 241         }
 242
 243         if (!eglMakeCurrent(ec->base.display, EGL_NO_SURFACE,
 244                             EGL_NO_SURFACE, ec->base.context)) {
 245                 fprintf(stderr, "failed to make context current\n");
 246                 return -1;
 247         }
 248
 249         return 0;
 250 }
 251
 252 static drmModeModeInfo builtin_1024x768 = {
 253         63500,                  /* clock */
 254         1024, 1072, 1176, 1328, 0,
 255         768, 771, 775, 798, 0,
 256         59920,
 257         DRM_MODE_FLAG_NHSYNC | DRM_MODE_FLAG_PVSYNC,
 258         0,
 259         "1024x768"
 260 };
 261
 262 static int
 263 create_output_for_connector(struct drm_compositor *ec,
 264                             drmModeRes *resources,
 265                             drmModeConnector *connector,
 266                             int x, int y)
 267 {
 268         struct drm_output *output;
 269         drmModeEncoder *encoder;
 270         drmModeModeInfo *mode;
 271         int i, ret;
 272         EGLint handle, stride, attribs[] = {
 273                 EGL_WIDTH,              0,
 274                 EGL_HEIGHT,             0,
 275                 EGL_DRM_BUFFER_FORMAT_MESA,     EGL_DRM_BUFFER_FORMAT_ARGB32_MESA,
 276                 EGL_DRM_BUFFER_USE_MESA,        EGL_DRM_BUFFER_USE_SCANOUT_MESA,
 277                 EGL_NONE
 278         };
 279
 280         output = malloc(sizeof *output);
 281         if (output == NULL)
 282                 return -1;
 283
 284         if (connector->count_modes > 0)
 285                 mode = &connector->modes[0];
 286         else
 287                 mode = &builtin_1024x768;
 288
 289         encoder = drmModeGetEncoder(ec->drm.fd, connector->encoders[0]);
 290         if (encoder == NULL) {
 291                 fprintf(stderr, "No encoder for connector.\n");
 292                 return -1;
 293         }
 294
 295         for (i = 0; i < resources->count_crtcs; i++) {
 296                 if (encoder->possible_crtcs & (1 << i) &&
 297                     !(ec->crtc_allocator & (1 << resources->crtcs[i])))
 298                         break;
 299         }
 300         if (i == resources->count_crtcs) {
 301                 fprintf(stderr, "No usable crtc for encoder.\n");
 302                 return -1;
 303         }
 304
 305         memset(output, 0, sizeof *output);
 306         wlsc_output_init(&output->base, &ec->base, x, y,
 307                          mode->hdisplay, mode->vdisplay, 0);
 308
 309         output->crtc_id = resources->crtcs[i];
 310         ec->crtc_allocator |= (1 << output->crtc_id);
 311
 312         output->connector_id = connector->connector_id;
 313         ec->connector_allocator |= (1 << output->connector_id);
 314         output->mode = *mode;
 315
 316         drmModeFreeEncoder(encoder);
 317
 318         glGenRenderbuffers(2, output->rbo);
 319         for (i = 0; i < 2; i++) {
 320                 glBindRenderbuffer(GL_RENDERBUFFER, output->rbo[i]);
 321
 322                 attribs[1] = output->base.width;
 323                 attribs[3] = output->base.height;
 324                 output->image[i] =
 325                         eglCreateDRMImageMESA(ec->base.display, attribs);
 326                 glEGLImageTargetRenderbufferStorageOES(GL_RENDERBUFFER,
 327                                                        output->image[i]);
 328                 eglExportDRMImageMESA(ec->base.display, output->image[i],
 329                                       NULL, &handle, &stride);
 330
 331                 ret = drmModeAddFB(ec->drm.fd,
 332                                    output->base.width, output->base.height,
 333                                    32, 32, stride, handle, &output->fb_id[i]);
 334                 if (ret) {
 335                         fprintf(stderr, "failed to add fb %d: %m\n", i);
 336                         return -1;
 337                 }
 338         }
 339
 340         output->current = 0;
 341         glFramebufferRenderbuffer(GL_FRAMEBUFFER,
 342                                   GL_COLOR_ATTACHMENT0,
 343                                   GL_RENDERBUFFER,
 344                                   output->rbo[output->current]);
 345         ret = drmModeSetCrtc(ec->drm.fd, output->crtc_id,
 346                              output->fb_id[output->current ^ 1], 0, 0,
 347                              &output->connector_id, 1, &output->mode);
 348         if (ret) {
 349                 fprintf(stderr, "failed to set mode: %m\n");
 350                 return -1;
 351         }
 352
 353         output->base.prepare_render = drm_output_prepare_render;
 354         output->base.present = drm_output_present;
 355         output->base.image_is_scanoutable = drm_output_image_is_scanoutable;
 356
 357         wl_list_insert(ec->base.output_list.prev, &output->base.link);
 358
 359         return 0;
 360 }
 361
 362 static int
 363 create_outputs(struct drm_compositor *ec, int option_connector)
 364 {
 365         drmModeConnector *connector;
 366         drmModeRes *resources;
 367         int i;
 368         int x = 0, y = 0;
 369
 370         resources = drmModeGetResources(ec->drm.fd);
 371         if (!resources) {
 372                 fprintf(stderr, "drmModeGetResources failed\n");
 373                 return -1;
 374         }
 375
 376         for (i = 0; i < resources->count_connectors; i++) {
 377                 connector = drmModeGetConnector(ec->drm.fd, resources->connectors[i]);
 378                 if (connector == NULL)
 379                         continue;
 380
 381                 if (connector->connection == DRM_MODE_CONNECTED &&
 382                     (option_connector == 0 ||
 383                      connector->connector_id == option_connector))
 384                         if (create_output_for_connector(ec, resources,
 385                                                         connector, x, y) < 0)
 386                                 return -1;
 387
 388                 x += container_of(ec->base.output_list.prev, struct wlsc_output,
 389                                   link)->width;
 390
 391                 drmModeFreeConnector(connector);
 392         }
 393
 394         if (wl_list_empty(&ec->base.output_list)) {
 395                 fprintf(stderr, "No currently active connector found.\n");
 396                 return -1;
 397         }
 398
 399         drmModeFreeResources(resources);
 400
 401         return 0;
 402 }
 403
 404 static int
 405 destroy_output(struct drm_output *output)
 406 {
 407         struct drm_compositor *ec =
 408                 (struct drm_compositor *) output->base.compositor;
 409         int i;
 410
 411         glFramebufferRenderbuffer(GL_FRAMEBUFFER,
 412                                   GL_COLOR_ATTACHMENT0,
 413                                   GL_RENDERBUFFER,
 414                                   0);
 415
 416         glBindRenderbuffer(GL_RENDERBUFFER, 0);
 417         glDeleteRenderbuffers(2, output->rbo);
 418
 419         for (i = 0; i < 2; i++) {
 420                 eglDestroyImageKHR(ec->base.display, output->image[i]);
 421                 drmModeRmFB(ec->drm.fd, output->fb_id[i]);
 422         }
 423
 424         ec->crtc_allocator &= ~(1 << output->crtc_id);
 425         ec->connector_allocator &= ~(1 << output->connector_id);
 426
 427         wlsc_output_destroy(&output->base);
 428         wl_list_remove(&output->base.link);
 429
 430         free(output);
 431
 432         return 0;
 433 }
 434
 435 static void
 436 update_outputs(struct drm_compositor *ec)
 437 {
 438         drmModeConnector *connector;
 439         drmModeRes *resources;
 440         struct drm_output *output, *next;
 441         int x = 0, y = 0;
 442         int x_offset = 0, y_offset = 0;
 443         uint32_t connected = 0, disconnects = 0;
 444         int i;
 445
 446         resources = drmModeGetResources(ec->drm.fd);
 447         if (!resources) {
 448                 fprintf(stderr, "drmModeGetResources failed\n");
 449                 return;
 450         }
 451
 452         /* collect new connects */
 453         for (i = 0; i < resources->count_connectors; i++) {
 454                 connector =
 455                         drmModeGetConnector(ec->drm.fd,
 456                                             resources->connectors[i]);
 457                 if (connector == NULL ||
 458                     connector->connection != DRM_MODE_CONNECTED)
 459                         continue;
 460
 461                 connected |= (1 << connector->connector_id);
 462
 463                 if (!(ec->connector_allocator & (1 << connector->connector_id))) {
 464                         struct wlsc_output *last_output =
 465                                 container_of(ec->base.output_list.prev,
 466                                              struct wlsc_output, link);
 467
 468                         /* XXX: not yet needed, we die with 0 outputs */
 469                         if (!wl_list_empty(&ec->base.output_list))
 470                                 x = last_output->x + last_output->width;
 471                         else
 472                                 x = 0;
 473                         y = 0;
 474                         create_output_for_connector(ec, resources,
 475                                                     connector, x, y);
 476                         printf("connector %d connected\n",
 477                                connector->connector_id);
 478
 479                 }
 480                 drmModeFreeConnector(connector);
 481         }
 482         drmModeFreeResources(resources);
 483
 484         disconnects = ec->connector_allocator & ~connected;
 485         if (disconnects) {
 486                 wl_list_for_each_safe(output, next, &ec->base.output_list,
 487                                       base.link) {
 488                         if (x_offset != 0 || y_offset != 0) {
 489                                 wlsc_output_move(&output->base,
 490                                                  output->base.x - x_offset,
 491                                                  output->base.y - y_offset);
 492                         }
 493
 494                         if (disconnects & (1 << output->connector_id)) {
 495                                 disconnects &= ~(1 << output->connector_id);
 496                                 printf("connector %d disconnected\n",
 497                                        output->connector_id);
 498                                 x_offset += output->base.width;
 499                                 destroy_output(output);
 500                         }
 501                 }
 502         }
 503
 504         /* FIXME: handle zero outputs, without terminating */
 505         if (ec->connector_allocator == 0)
 506                 wl_display_terminate(ec->base.wl_display);
 507 }
 508
 509 static int
 510 udev_event_is_hotplug(struct udev_device *device)
 511 {
 512         struct udev_list_entry *list, *hotplug_entry;
 513
 514         list = udev_device_get_properties_list_entry(device);
 515
 516         hotplug_entry = udev_list_entry_get_by_name(list, "HOTPLUG");
 517         if (hotplug_entry == NULL)
 518                 return 0;
 519
 520         return strcmp(udev_list_entry_get_value(hotplug_entry), "1") == 0;
 521 }
 522
 523 static void
 524 udev_drm_event(int fd, uint32_t mask, void *data)
 525 {
 526         struct drm_compositor *ec = data;
 527         struct udev_device *event;
 528
 529         event = udev_monitor_receive_device(ec->udev_monitor);
 530
 531         if (udev_event_is_hotplug(event))
 532                 update_outputs(ec);
 533
 534         udev_device_unref(event);
 535 }
 536
 537 static void
 538 drm_destroy(struct wlsc_compositor *ec)
 539 {
 540         struct drm_compositor *d = (struct drm_compositor *) ec;
 541
 542         tty_destroy(d->tty);
 543
 544         free(d);
 545 }
 546
 547 struct wlsc_compositor *
 548 drm_compositor_create(struct wl_display *display, int connector)
 549 {
 550         struct drm_compositor *ec;
 551         struct udev_enumerate *e;
 552         struct udev_list_entry *entry;
 553         struct udev_device *device;
 554         const char *path;
 555         struct wl_event_loop *loop;
 556
 557         ec = malloc(sizeof *ec);
 558         if (ec == NULL)
 559                 return NULL;
 560
 561         memset(ec, 0, sizeof *ec);
 562         ec->udev = udev_new();
 563         if (ec->udev == NULL) {
 564                 fprintf(stderr, "failed to initialize udev context\n");
 565                 return NULL;
 566         }
 567
 568         e = udev_enumerate_new(ec->udev);
 569         udev_enumerate_add_match_subsystem(e, "drm");
 570         udev_enumerate_add_match_property(e, "WAYLAND_SEAT", "1");
 571         udev_enumerate_scan_devices(e);
 572         device = NULL;
 573         udev_list_entry_foreach(entry, udev_enumerate_get_list_entry(e)) {
 574                 path = udev_list_entry_get_name(entry);
 575                 device = udev_device_new_from_syspath(ec->udev, path);
 576                 break;
 577         }
 578         udev_enumerate_unref(e);
 579
 580         if (device == NULL) {
 581                 fprintf(stderr, "no drm device found\n");
 582                 return NULL;
 583         }
 584
 585         ec->base.wl_display = display;
 586         if (init_egl(ec, device) < 0) {
 587                 fprintf(stderr, "failed to initialize egl\n");
 588                 return NULL;
 589         }
 590
 591         ec->base.destroy = drm_destroy;
 592         ec->base.create_buffer = wlsc_shm_buffer_create;
 593         ec->base.focus = 1;
 594
 595         glGenFramebuffers(1, &ec->base.fbo);
 596         glBindFramebuffer(GL_FRAMEBUFFER, ec->base.fbo);
 597
 598         /* Can't init base class until we have a current egl context */
 599         if (wlsc_compositor_init(&ec->base, display) < 0)
 600                 return NULL;
 601
 602         if (create_outputs(ec, connector) < 0) {
 603                 fprintf(stderr, "failed to create output for %s\n", path);
 604                 return NULL;
 605         }
 606
 607         evdev_input_add_devices(&ec->base, ec->udev);
 608
 609         loop = wl_display_get_event_loop(ec->base.wl_display);
 610         ec->drm_source =
 611                 wl_event_loop_add_fd(loop, ec->drm.fd,
 612                                      WL_EVENT_READABLE, on_drm_input, ec);
 613         ec->tty = tty_create(&ec->base);
 614
 615         ec->udev_monitor = udev_monitor_new_from_netlink(ec->udev, "udev");
 616         if (ec->udev_monitor == NULL) {
 617                 fprintf(stderr, "failed to intialize udev monitor\n");
 618                 return NULL;
 619         }
 620         udev_monitor_filter_add_match_subsystem_devtype(ec->udev_monitor,
 621                                                         "drm", NULL);
 622         ec->udev_drm_source =
 623                 wl_event_loop_add_fd(loop, udev_monitor_get_fd(ec->udev_monitor),
 624                                      WL_EVENT_READABLE, udev_drm_event, ec);
 625
 626         if (udev_monitor_enable_receiving(ec->udev_monitor) < 0) {
 627                 fprintf(stderr, "failed to enable udev-monitor receiving\n");
 628                 return NULL;
 629         }
 630
 631         return &ec->base;
 632 }