source/main.c

   1 #define GL_GLEXT_PROTOTYPES
   2 #include <stdio.h>
   3 #include <stdlib.h>
   4 #include <string.h>
   5 #include <signal.h>
   6 #include <math.h>
   7 #include <X11/Xlib.h>
   8 #include <X11/Xatom.h>
   9 #include <X11/extensions/Xcomposite.h>
  10 #include <X11/extensions/Xdamage.h>
  11 #include <X11/extensions/shape.h>
  12 #include <X11/extensions/Xrandr.h>
  13 #include <GL/gl.h>
  14 #include <GL/glx.h>
  15
  16 typedef struct CompositedWindow
  17 {
  18         Window window;
  19         int x;
  20         int y;
  21         unsigned width;
  22         unsigned height;
  23         unsigned border;
  24         int map_state;
  25         Damage damage;
  26         Pixmap pixmap;
  27         GLXPixmap glx_pixmap;
  28         unsigned texture;
  29 } CompositedWindow;
  30
  31 typedef struct CompositedMonitor
  32 {
  33         char *name;
  34         int enabled;
  35         int x;
  36         int y;
  37         unsigned width;
  38         unsigned height;
  39         float keystone_vertical;
  40         float cylinder_depth;
  41         float vertical_center;
  42         float perspective;
  43         unsigned vertex_buffer;
  44         unsigned index_buffer;
  45         unsigned vertex_array;
  46         unsigned tessellation;
  47         unsigned nelements;
  48 } CompositedMonitor;
  49
  50 typedef struct CompositedScreen
  51 {
  52         int number;
  53         Window root;
  54         unsigned width;
  55         unsigned height;
  56         Window overlay;
  57         Window render_window;
  58         GLXFBConfig fbconfig;
  59         GLXWindow glx_window;
  60         GLXContext glx_context;
  61         unsigned shaders[2];
  62         unsigned program;
  63         unsigned geometry_loc;
  64         unsigned window_vertex_buffer;
  65         unsigned window_vertex_array;
  66         unsigned framebuffer;
  67         unsigned fb_texture;
  68         CompositedWindow *windows;
  69         unsigned nwindows;
  70         unsigned windows_capacity;
  71         CompositedMonitor *monitors;
  72         unsigned nmonitors;
  73         int dirty;
  74 } CompositedScreen;
  75
  76 typedef struct Compositor
  77 {
  78         Display *display;
  79         CompositedScreen *screens;
  80         unsigned nscreens;
  81         int damage_event;
  82         PFNGLXCREATECONTEXTATTRIBSARBPROC glXCreateContextAttribsARB;
  83         PFNGLXBINDTEXIMAGEEXTPROC glXBindTexImageEXT;
  84         PFNGLXRELEASETEXIMAGEEXTPROC glXReleaseTexImageEXT;
  85         Atom correction_atom;
  86         Atom monitors_atom;
  87         int dirty;
  88 } Compositor;
  89
  90 static const char *vshader =
  91         "#version 150\n"
  92         "uniform vec4 geometry;\n"
  93         "in vec2 vertex;\n"
  94         "in vec2 texture_coord;\n"
  95         "out vec2 texcoord;\n"
  96         "void main()\n"
  97         "{\n"
  98         "  gl_Position = vec4((geometry.xy+vertex*geometry.zw)*2.0-1.0, 0.0, 1.0);\n"
  99         "  texcoord = texture_coord;\n"
 100         "}\n";
 101
 102 static const char *fshader =
 103         "#version 150\n"
 104         "uniform sampler2D image;\n"
 105         "in vec2 texcoord;\n"
 106         "out vec4 frag_color;\n"
 107         "void main()\n"
 108         "{\n"
 109         "       frag_color = texture(image, texcoord);\n"
 110         "}\n";
 111
 112 static const float window_vertices[] =
 113 {
 114         /* vertex    texcoord */
 115         0.0f, 1.0f,  0.0f, 0.0f,
 116         0.0f, 0.0f,  0.0f, 1.0f,
 117         1.0f, 1.0f,  1.0f, 0.0f,
 118         1.0f, 0.0f,  1.0f, 1.0f
 119 };
 120
 121 int terminate_requested = 0;
 122
 123 int x_error_handler(Display *display, XErrorEvent *event)
 124 {
 125         printf("Ignoring X error %d on resource %lx\n", event->error_code, event->resourceid);
 126         (void)display;
 127         return 0;
 128 }
 129
 130 int with_error(const char *message)
 131 {
 132         fprintf(stderr, "%s\n", message);
 133         return 0;
 134 }
 135
 136 int initialize_gl(Compositor *compositor, CompositedScreen *screen)
 137 {
 138         int attribs[9];
 139         unsigned i;
 140         GLXFBConfig *configs;
 141         int nconfigs;
 142         XVisualInfo *vi;
 143         XSetWindowAttributes win_attr;
 144
 145         i = 0;
 146         attribs[i++] = GLX_DRAWABLE_TYPE;
 147         attribs[i++] = GLX_WINDOW_BIT;
 148         attribs[i++] = GLX_RENDER_TYPE;
 149         attribs[i++] = GLX_RGBA_BIT;
 150         attribs[i++] = GLX_DOUBLEBUFFER;
 151         attribs[i++] = True;
 152         attribs[i++] = GLX_BIND_TO_TEXTURE_RGBA_EXT;
 153         attribs[i++] = True;
 154         attribs[i] = None;
 155
 156         configs = glXChooseFBConfig(compositor->display, screen->number, attribs, &nconfigs);
 157         if(!configs || !nconfigs)
 158                 return with_error("Could not find a suitable FBConfig");
 159         screen->fbconfig = configs[0];
 160         XFree(configs);
 161
 162         vi = glXGetVisualFromFBConfig(compositor->display, screen->fbconfig);
 163         win_attr.colormap = XCreateColormap(compositor->display, screen->root, vi->visual, AllocNone);
 164         screen->render_window = XCreateWindow(compositor->display, screen->overlay, 0, 0, screen->width, screen->height, 0, vi->depth, InputOutput, vi->visual, CWColormap, &win_attr);
 165         XMapWindow(compositor->display, screen->render_window);
 166
 167         screen->glx_window = glXCreateWindow(compositor->display, screen->fbconfig, screen->render_window, NULL);
 168
 169         i = 0;
 170         attribs[i++] = GLX_CONTEXT_FLAGS_ARB;
 171         attribs[i++] = GLX_CONTEXT_FORWARD_COMPATIBLE_BIT_ARB;
 172         attribs[i++] = GLX_CONTEXT_MAJOR_VERSION_ARB;
 173         attribs[i++] = 3;
 174         attribs[i++] = GLX_CONTEXT_MINOR_VERSION_ARB;
 175         attribs[i++] = 1;
 176         attribs[i] = None;
 177
 178         screen->glx_context = compositor->glXCreateContextAttribsARB(compositor->display, screen->fbconfig, NULL, True, attribs);
 179
 180         XFree(vi);
 181
 182         return 1;
 183 }
 184
 185 void use_gl(Compositor *compositor, CompositedScreen *screen)
 186 {
 187         glXMakeContextCurrent(compositor->display, screen->glx_window, screen->glx_window, screen->glx_context);
 188 }
 189
 190 unsigned compile_shader(GLenum type, const char *source)
 191 {
 192         unsigned shader;
 193         int status;
 194         char info_log[1024];
 195         GLsizei length;
 196
 197         shader = glCreateShader(type);
 198         glShaderSource(shader, 1, &source, NULL);
 199         glCompileShader(shader);
 200         glGetShaderiv(shader, GL_COMPILE_STATUS, &status);
 201         glGetShaderInfoLog(shader, sizeof(info_log), &length, info_log);
 202         if(!status)
 203         {
 204                 fprintf(stderr, "Shader compilation failed:\n%s\n", info_log);
 205                 glDeleteShader(shader);
 206                 return 0;
 207         }
 208         else if(length)
 209                 printf("Shader info log:\n%s\n", info_log);
 210
 211         return shader;
 212 }
 213
 214 unsigned link_program(unsigned *shaders, unsigned nshaders)
 215 {
 216         unsigned program;
 217         unsigned i;
 218         int status;
 219         char info_log[1024];
 220         GLsizei length;
 221
 222         program = glCreateProgram();
 223         for(i=0; i<nshaders; ++i)
 224                 glAttachShader(program, shaders[i]);
 225         glBindAttribLocation(program, 0, "vertex");
 226         glBindAttribLocation(program, 1, "texture_coord");
 227         glBindFragDataLocation(program, 0, "frag_color");
 228         glLinkProgram(program);
 229
 230         glGetProgramiv(program, GL_LINK_STATUS, &status);
 231         glGetProgramInfoLog(program, sizeof(info_log), &length, info_log);
 232         if(!status)
 233         {
 234                 fprintf(stderr, "Program link failed:\n%s\n", info_log);
 235                 glDeleteProgram(program);
 236                 return 0;
 237         }
 238         else if(length)
 239                 printf("Program info log:\n%s\n", info_log);
 240
 241         return program;
 242 }
 243
 244 int create_gl_resources(Compositor *compositor, CompositedScreen *screen)
 245 {
 246         unsigned stride;
 247
 248         use_gl(compositor, screen);
 249
 250         screen->shaders[0] = compile_shader(GL_VERTEX_SHADER, vshader);
 251         screen->shaders[1] = compile_shader(GL_FRAGMENT_SHADER, fshader);
 252         if(!screen->shaders[0] || !screen->shaders[1])
 253                 return 0;
 254
 255         screen->program = link_program(screen->shaders, 2);
 256         if(!screen->program)
 257                 return 0;
 258
 259         screen->geometry_loc = glGetUniformLocation(screen->program, "geometry");
 260
 261         glGenBuffers(1, &screen->window_vertex_buffer);
 262         glBindBuffer(GL_ARRAY_BUFFER, screen->window_vertex_buffer);
 263         glBufferData(GL_ARRAY_BUFFER, sizeof(window_vertices), window_vertices, GL_STATIC_DRAW);
 264
 265         stride = 4*sizeof(float);
 266         glGenVertexArrays(1, &screen->window_vertex_array);
 267         glBindVertexArray(screen->window_vertex_array);
 268         glVertexAttribPointer(0, 2, GL_FLOAT, GL_FALSE, stride, NULL);
 269         glEnableVertexAttribArray(0);
 270         glVertexAttribPointer(1, 2, GL_FLOAT, GL_FALSE, stride, (void *)(2*sizeof(float)));
 271         glEnableVertexAttribArray(1);
 272         glBindVertexArray(0);
 273
 274         glBindBuffer(GL_ARRAY_BUFFER, 0);
 275
 276         glGenTextures(1, &screen->fb_texture);
 277         glBindTexture(GL_TEXTURE_2D, screen->fb_texture);
 278         glTexImage2D(GL_TEXTURE_2D, 0, GL_RGB8, screen->width, screen->height, 0, GL_RGB, GL_UNSIGNED_BYTE, NULL);
 279         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
 280         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
 281         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
 282         glBindTexture(GL_TEXTURE_2D, 0);
 283
 284         glGenFramebuffers(1, &screen->framebuffer);
 285         glBindFramebuffer(GL_DRAW_FRAMEBUFFER, screen->framebuffer);
 286         glFramebufferTexture2D(GL_DRAW_FRAMEBUFFER, GL_COLOR_ATTACHMENT0, GL_TEXTURE_2D, screen->fb_texture, 0);
 287         glDepthMask(GL_FALSE);
 288         glBindFramebuffer(GL_DRAW_FRAMEBUFFER, 0);
 289
 290         return 1;
 291 }
 292
 293 CompositedWindow *find_window(CompositedScreen *screen, Window w)
 294 {
 295         unsigned i;
 296
 297         for(i=0; i<screen->nwindows; ++i)
 298                 if(screen->windows[i].window==w)
 299                         return &screen->windows[i];
 300
 301         return NULL;
 302 }
 303
 304 void create_window_pixmap(Compositor *compositor, CompositedScreen *screen, CompositedWindow *window)
 305 {
 306         int attribs[5];
 307         unsigned i;
 308
 309         if(window->pixmap)
 310         {
 311                 glXDestroyPixmap(compositor->display, window->glx_pixmap);
 312                 XFreePixmap(compositor->display, window->pixmap);
 313         }
 314
 315         i = 0;
 316         attribs[i++] = GLX_TEXTURE_TARGET_EXT;
 317         attribs[i++] = GLX_TEXTURE_2D_EXT;
 318         attribs[i++] = GLX_TEXTURE_FORMAT_EXT;
 319         attribs[i++] = GLX_TEXTURE_FORMAT_RGBA_EXT;
 320         attribs[i++] = None;
 321
 322         window->pixmap = XCompositeNameWindowPixmap(compositor->display, window->window);
 323         window->glx_pixmap = glXCreatePixmap(compositor->display, screen->fbconfig, window->pixmap, attribs);
 324 }
 325
 326 CompositedWindow *add_window(Compositor *compositor, CompositedScreen *screen, Window w)
 327 {
 328         CompositedWindow *window;
 329         XWindowAttributes win_attr;
 330
 331         if(w==screen->root || w==screen->overlay)
 332                 return NULL;
 333
 334         if(!XGetWindowAttributes(compositor->display, w, &win_attr))
 335         {
 336                 printf("XGetWindowAttributes failed; probably the window was already destroyed\n");
 337                 return NULL;
 338         }
 339         if(win_attr.class==InputOnly)
 340                 return NULL;
 341
 342         if(find_window(screen, w))
 343                 return NULL;
 344
 345         if(screen->nwindows==screen->windows_capacity)
 346         {
 347                 screen->windows = (CompositedWindow *)realloc(screen->windows, (screen->windows_capacity+1)*sizeof(CompositedWindow));
 348                 ++screen->windows_capacity;
 349         }
 350
 351         window = &screen->windows[screen->nwindows++];
 352
 353         window->window = w;
 354         window->x = win_attr.x;
 355         window->y = win_attr.y;
 356         window->width = win_attr.width;
 357         window->height = win_attr.height;
 358         window->border = win_attr.border_width;
 359         window->map_state = win_attr.map_state;
 360
 361         window->damage = XDamageCreate(compositor->display, window->window, XDamageReportNonEmpty);
 362         window->pixmap = None;
 363         window->glx_pixmap = None;
 364
 365         XCompositeRedirectWindow(compositor->display, window->window, CompositeRedirectManual);
 366         if(window->map_state==IsViewable)
 367                 create_window_pixmap(compositor, screen, window);
 368
 369         glGenTextures(1, &window->texture);
 370         glBindTexture(GL_TEXTURE_2D, window->texture);
 371         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
 372         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
 373         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
 374
 375         return window;
 376 }
 377
 378 void remove_window(Compositor *compositor, CompositedScreen *screen, CompositedWindow *window, int destroyed)
 379 {
 380         unsigned i;
 381
 382         glDeleteTextures(1, &window->texture);
 383         if(!destroyed)
 384         {
 385                 XDamageDestroy(compositor->display, window->damage);
 386                 if(window->pixmap)
 387                 {
 388                         glXDestroyPixmap(compositor->display, window->glx_pixmap);
 389                         XFreePixmap(compositor->display, window->pixmap);
 390                 }
 391                 XCompositeUnredirectWindow(compositor->display, window->window, CompositeRedirectManual);
 392         }
 393
 394         --screen->nwindows;
 395         for(i=window-screen->windows; i<screen->nwindows; ++i)
 396                 screen->windows[i] = screen->windows[i+1];
 397 }
 398
 399 CompositedWindow *reorder_window(CompositedScreen *screen, CompositedWindow *window, Window above)
 400 {
 401         unsigned i, j;
 402         CompositedWindow hold;
 403
 404         i = window-screen->windows;
 405         if(above)
 406         {
 407                 for(j=0; j<screen->nwindows; ++j)
 408                         if(screen->windows[j].window==above)
 409                                 break;
 410
 411                 if(j>=screen->nwindows || i==j+1)
 412                         return window;
 413
 414                 if(j<i)
 415                         ++j;
 416         }
 417         else
 418                 j = 0;
 419
 420         hold = *window;
 421         if(i<j)
 422         {
 423                 for(; i<j; ++i)
 424                         screen->windows[i] = screen->windows[i+1];
 425         }
 426         else
 427         {
 428                 for(; i>j; --i)
 429                         screen->windows[i] = screen->windows[i-1];
 430         }
 431         screen->windows[j] = hold;
 432
 433         return &screen->windows[j];
 434 }
 435
 436 CompositedScreen *find_screen_by_root(Compositor *compositor, Window root)
 437 {
 438         unsigned i;
 439
 440         for(i=0; i<compositor->nscreens; ++i)
 441                 if(compositor->screens[i].root==root)
 442                         return &compositor->screens[i];
 443
 444         return NULL;
 445 }
 446
 447 CompositedScreen *find_screen_by_window(Compositor *compositor, Window w)
 448 {
 449         unsigned i, j;
 450
 451         for(i=0; i<compositor->nscreens; ++i)
 452                 for(j=0; j<compositor->screens[i].nwindows; ++j)
 453                         if(compositor->screens[i].windows[j].window==w)
 454                                 return &compositor->screens[i];
 455
 456         return NULL;
 457 }
 458
 459 void update_monitor_vertices(CompositedScreen *screen, CompositedMonitor *monitor)
 460 {
 461         unsigned t;
 462         unsigned data_size;
 463         float *vertex_data;
 464         unsigned short *index_data;
 465         unsigned x, y;
 466         unsigned i;
 467         float aspect;
 468         float cyl_w_sq;
 469         float sin_ksv;
 470         float cos_ksv;
 471
 472         t = monitor->tessellation;
 473
 474         data_size = (t+1)*(t+1)*4*sizeof(float);
 475         vertex_data = (float *)malloc(data_size);
 476         aspect = (float)monitor->width/monitor->height;
 477         cyl_w_sq = 0.25f-(0.5f-monitor->cylinder_depth)*(0.5f-monitor->cylinder_depth);
 478         sin_ksv = monitor->keystone_vertical/sqrt(1.0f+monitor->keystone_vertical*monitor->keystone_vertical);
 479         cos_ksv = sqrt(1.0f-sin_ksv*sin_ksv);
 480         for(y=0; y<=t; ++y)
 481                 for(x=0; x<=t; ++x)
 482                 {
 483                         float xf, yf, z, rz;
 484                         float scale;
 485                         float *v;
 486
 487                         xf = (float)x/t;
 488                         yf = (float)y/t;
 489
 490                         v = vertex_data+(y*(t+1)+x)*4;
 491                         v[2] = (monitor->x+xf*monitor->width)/screen->width;
 492                         v[3] = (monitor->y+yf*monitor->height)/screen->height;
 493
 494                         z = sqrt(0.25f-(xf-0.5f)*(xf-0.5f)*4.0f*cyl_w_sq)-0.5f+monitor->cylinder_depth;
 495                         if(monitor->keystone_vertical>0)
 496                         {
 497                                 rz = (1.0f-yf)/aspect*sin_ksv+z*cos_ksv;
 498                                 yf = 1.0f-((1.0f-yf)*cos_ksv-z*aspect*sin_ksv);
 499                                 z = rz;
 500                         }
 501                         else if(monitor->keystone_vertical<0)
 502                         {
 503                                 rz = z*cos_ksv-yf/aspect*sin_ksv;
 504                                 yf = yf*cos_ksv+z*aspect*sin_ksv;
 505                                 z = rz;
 506                         }
 507                         scale = monitor->perspective/(monitor->perspective+z);
 508                         v[0] = 0.5f+(xf-0.5f)*scale;
 509                         v[1] = monitor->vertical_center+(yf-monitor->vertical_center)*scale;
 510                 }
 511         glBindBuffer(GL_ARRAY_BUFFER, monitor->vertex_buffer);
 512         glBufferData(GL_ARRAY_BUFFER, data_size, vertex_data, GL_STATIC_DRAW);
 513         glBindBuffer(GL_ARRAY_BUFFER, 0);
 514         free(vertex_data);
 515
 516         monitor->nelements = t*((t+1)*2+1)-1;
 517         data_size = monitor->nelements*sizeof(unsigned short);
 518         index_data = (unsigned short *)malloc(data_size);
 519         i = 0;
 520         for(y=0; y<t; ++y)
 521         {
 522                 if(y>0)
 523                         index_data[i++] = 0xFFFF;
 524                 for(x=0; x<=t; ++x)
 525                 {
 526                         index_data[i++] = (y+1)*(t+1)+x;
 527                         index_data[i++] = y*(t+1)+x;
 528                 }
 529         }
 530         glBindBuffer(GL_ELEMENT_ARRAY_BUFFER, monitor->index_buffer);
 531         glBufferData(GL_ELEMENT_ARRAY_BUFFER, data_size, index_data, GL_STATIC_DRAW);
 532         glBindBuffer(GL_ELEMENT_ARRAY_BUFFER, 0);
 533         free(index_data);
 534 }
 535
 536 int initialize_monitor(Compositor *compositor, CompositedScreen *screen, XRRScreenResources *xrr_res, unsigned index)
 537 {
 538         CompositedMonitor *monitor;
 539         XRROutputInfo *output;
 540         int namelen;
 541         XRRCrtcInfo *crtc;
 542         unsigned buffers[2];
 543         unsigned stride;
 544
 545         monitor = &screen->monitors[index];
 546
 547         output = XRRGetOutputInfo(compositor->display, xrr_res, xrr_res->outputs[index]);
 548         namelen = strlen(output->name);
 549         monitor->name = (char *)malloc(namelen+1);
 550         strcpy(monitor->name, output->name);
 551         monitor->enabled = !!output->crtc;
 552         if(!monitor->enabled)
 553         {
 554                 XRRFreeOutputInfo(output);
 555                 return 1;
 556         }
 557
 558         crtc = XRRGetCrtcInfo(compositor->display, xrr_res, output->crtc);
 559         monitor->x = crtc->x;
 560         monitor->y = crtc->y;
 561         monitor->width = crtc->width;
 562         monitor->height = crtc->height;
 563         XRRFreeCrtcInfo(crtc);
 564         XRRFreeOutputInfo(output);
 565
 566         monitor->keystone_vertical = 0.0f;
 567         monitor->cylinder_depth = 0.0f;
 568         monitor->vertical_center = 0.5f;
 569         monitor->perspective = 1.0f;
 570
 571         glGenBuffers(2, buffers);
 572         monitor->vertex_buffer = buffers[0];
 573         monitor->index_buffer = buffers[1];
 574
 575         monitor->tessellation = 50;
 576         update_monitor_vertices(screen, monitor);
 577         stride = 4*sizeof(float);
 578
 579         glGenVertexArrays(1, &monitor->vertex_array);
 580         glBindVertexArray(monitor->vertex_array);
 581         glBindBuffer(GL_ARRAY_BUFFER, monitor->vertex_buffer);
 582         glVertexAttribPointer(0, 2, GL_FLOAT, GL_FALSE, stride, NULL);
 583         glEnableVertexAttribArray(0);
 584         glVertexAttribPointer(1, 2, GL_FLOAT, GL_FALSE, stride, (void *)(2*sizeof(float)));
 585         glEnableVertexAttribArray(1);
 586         glBindBuffer(GL_ARRAY_BUFFER, 0);
 587         glBindBuffer(GL_ELEMENT_ARRAY_BUFFER, monitor->index_buffer);
 588         glBindVertexArray(0);
 589
 590         return 1;
 591 }
 592
 593 CompositedMonitor *find_monitor_by_name(CompositedScreen *screen, char *name)
 594 {
 595         unsigned i;
 596
 597         for(i=0; i<screen->nmonitors; ++i)
 598                 if(!strcmp(screen->monitors[i].name, name))
 599                         return &screen->monitors[i];
 600
 601         return NULL;
 602 }
 603
 604 void update_geometry_correction(Compositor *compositor, CompositedScreen *screen)
 605 {
 606         Atom prop_type;
 607         int prop_format;
 608         unsigned long overflow;
 609         unsigned long names_length;
 610         char *names;
 611         unsigned long values_length;
 612         short *values;
 613         char *name_ptr;
 614         unsigned i;
 615
 616         XGetWindowProperty(compositor->display, screen->root, compositor->monitors_atom, 0, 64, False, XA_STRING,
 617                 &prop_type, &prop_format, &names_length, &overflow, (unsigned char **)&names);
 618         if(prop_type!=XA_STRING || prop_format!=8)
 619                 return;
 620
 621         XGetWindowProperty(compositor->display, screen->root, compositor->correction_atom, 0, 64, False, XA_INTEGER,
 622                 &prop_type, &prop_format, &values_length, &overflow, (unsigned char **)&values);
 623         if(prop_type!=XA_INTEGER || prop_format!=16)
 624         {
 625                 XFree(names);
 626                 return;
 627         }
 628
 629         use_gl(compositor, screen);
 630
 631         name_ptr = names;
 632         for(i=0; i*4<values_length; ++i)
 633         {
 634                 CompositedMonitor *monitor;
 635
 636                 if(name_ptr>=names+names_length)
 637                         break;
 638
 639                 monitor = find_monitor_by_name(screen, name_ptr);
 640                 if(monitor)
 641                 {
 642                         monitor->keystone_vertical = values[i*4]/10000.0f;
 643                         monitor->cylinder_depth = values[i*4+1]/10000.0f;
 644                         monitor->vertical_center = values[i*4+2]/10000.0f;
 645                         monitor->perspective = values[i*4+3]/10000.0f;
 646
 647                         update_monitor_vertices(screen, monitor);
 648                 }
 649
 650                 name_ptr += strlen(name_ptr)+1;
 651         }
 652
 653         XFree(names);
 654         XFree(values);
 655 }
 656
 657 int initialize_screen(Compositor *compositor, unsigned number)
 658 {
 659         CompositedScreen *screen;
 660         const char *extensions;
 661         Window dummy_root;
 662         int x, y;
 663         unsigned border;
 664         unsigned depth;
 665         XRRScreenResources *xrr_res;
 666         Window dummy_parent;
 667         Window *children;
 668         unsigned nchildren;
 669         unsigned i;
 670
 671         screen = &compositor->screens[number];
 672         screen->number = number;
 673
 674         extensions = glXQueryExtensionsString(compositor->display, screen->number);
 675         if(!strstr(extensions, "GLX_ARB_create_context"))
 676                 return with_error("GLX_ARB_create_context is required");
 677         if(!strstr(extensions, "GLX_EXT_texture_from_pixmap"))
 678                 return with_error("GLX_EXT_texture_from_pixmap is required");
 679
 680         screen->root = RootWindow(compositor->display, screen->number);
 681         XSelectInput(compositor->display, screen->root, SubstructureNotifyMask|PropertyChangeMask);
 682         screen->overlay = XCompositeGetOverlayWindow(compositor->display, screen->root);
 683         XGetGeometry(compositor->display, screen->overlay, &dummy_root, &x, &y, &screen->width, &screen->height, &border, &depth);
 684         XShapeCombineRectangles(compositor->display, screen->overlay, ShapeInput, 0, 0, NULL, 0, ShapeSet, Unsorted);
 685
 686         if(!initialize_gl(compositor, screen))
 687                 return 0;
 688
 689         if(!create_gl_resources(compositor, screen))
 690                 return 0;
 691
 692         xrr_res = XRRGetScreenResources(compositor->display, screen->root);
 693         screen->nmonitors = xrr_res->noutput;
 694         screen->monitors = (CompositedMonitor *)malloc(screen->nmonitors*sizeof(CompositedMonitor));
 695         for(i=0; i<screen->nmonitors; ++i)
 696                 if(!initialize_monitor(compositor, screen, xrr_res, i))
 697                         return 0;
 698         XRRFreeScreenResources(xrr_res);
 699
 700         update_geometry_correction(compositor, screen);
 701
 702         XQueryTree(compositor->display, screen->root, &dummy_root, &dummy_parent, &children, &nchildren);
 703
 704         screen->windows = (CompositedWindow *)malloc(nchildren*sizeof(CompositedWindow));
 705         screen->nwindows = 0;
 706         screen->windows_capacity = nchildren;
 707
 708         for(i=0; i<nchildren; ++i)
 709                 add_window(compositor, screen, children[i]);
 710
 711         XFree(children);
 712
 713         screen->dirty = 1;
 714
 715         return 1;
 716 }
 717
 718 int initialize_compositor(Compositor *compositor)
 719 {
 720         int event_base;
 721         int error_base;
 722         int major_ver;
 723         int minor_ver;
 724         unsigned i;
 725
 726         compositor->display = XOpenDisplay(NULL);
 727         if(!compositor->display)
 728                 return with_error("Could not open X display");
 729
 730         XSetErrorHandler(&x_error_handler);
 731
 732         if(!XCompositeQueryExtension(compositor->display, &event_base, &error_base))
 733                 return with_error("XComposite is required but was not found");
 734         else if(!XCompositeQueryVersion(compositor->display, &major_ver, &minor_ver))
 735                 return with_error("Cannot determine XComposite version");
 736         else if(major_ver==0 && minor_ver<3)
 737                 return with_error("XComposite 0.3 or later is required");
 738
 739         if(!glXQueryExtension(compositor->display, &event_base, &error_base))
 740                 return with_error("GLX is required but was not found");
 741         else if(!glXQueryVersion(compositor->display, &major_ver, &minor_ver))
 742                 return with_error("Cannot determine GLX version");
 743         else if(major_ver<1 || (major_ver==1 && minor_ver<4))
 744                 return with_error("GLX 1.4 or later is required");
 745
 746         if(!XDamageQueryExtension(compositor->display, &compositor->damage_event, &error_base))
 747                 return with_error("XDamage is required but was not found");
 748         else if(!XDamageQueryVersion(compositor->display, &major_ver, &minor_ver))
 749                 return with_error("Cannot determine XDamage version");
 750         else if(major_ver<1)
 751                 return with_error("XDamage 1.0 or later is required");
 752
 753         if(!XShapeQueryExtension(compositor->display, &event_base, &error_base))
 754                 return with_error("XShape is required but was not found");
 755         else if(!XShapeQueryVersion(compositor->display, &major_ver, &minor_ver))
 756                 return with_error("Cannot determine XShape version");
 757         else if(major_ver<1 || (major_ver==1 && minor_ver<1))
 758                 return with_error("XShape 1.1 or later is required");
 759
 760         if(!XRRQueryExtension(compositor->display, &event_base, &error_base))
 761                 return with_error("XRandR is required but was not found");
 762         else if(!XRRQueryVersion(compositor->display, &major_ver, &minor_ver))
 763                 return with_error("Cannot determine XRandR version");
 764         else if(major_ver<1 || (major_ver==1 && minor_ver<2))
 765                 return with_error("XRandR 1.2 or later is required");
 766
 767         compositor->glXCreateContextAttribsARB = (PFNGLXCREATECONTEXTATTRIBSARBPROC)glXGetProcAddress((unsigned char *)"glXCreateContextAttribsARB");
 768         compositor->glXBindTexImageEXT = (PFNGLXBINDTEXIMAGEEXTPROC)glXGetProcAddress((unsigned char *)"glXBindTexImageEXT");
 769         compositor->glXReleaseTexImageEXT = (PFNGLXRELEASETEXIMAGEEXTPROC)glXGetProcAddress((unsigned char *)"glXReleaseTexImageEXT");
 770
 771         compositor->correction_atom = XInternAtom(compositor->display, "GEOMETRY_CORRECTION", False);
 772         compositor->monitors_atom = XInternAtom(compositor->display, "GEOMETRY_CORRECTION_MONITORS", False);
 773
 774         compositor->nscreens = ScreenCount(compositor->display);
 775         compositor->screens = (CompositedScreen *)malloc(compositor->nscreens*sizeof(CompositedScreen));
 776         for(i=0; i<compositor->nscreens; ++i)
 777                 if(!initialize_screen(compositor, i))
 778                         return 0;
 779
 780         compositor->dirty = 1;
 781
 782         return 1;
 783 }
 784
 785 void shutdown_screen(Compositor *compositor, CompositedScreen *screen)
 786 {
 787         unsigned i;
 788
 789         use_gl(compositor, screen);
 790
 791         for(i=0; i<screen->nwindows; ++i)
 792         {
 793                 glDeleteTextures(1, &screen->windows[i].texture);
 794                 if(screen->windows[i].pixmap)
 795                 {
 796                         glXDestroyPixmap(compositor->display, screen->windows[i].glx_pixmap);
 797                         XFreePixmap(compositor->display, screen->windows[i].pixmap);
 798                         XDamageDestroy(compositor->display, screen->windows[i].damage);
 799                 }
 800         }
 801
 802         for(i=0; i<screen->nmonitors; ++i)
 803         {
 804                 free(screen->monitors[i].name);
 805                 if(screen->monitors[i].enabled)
 806                 {
 807                         glDeleteBuffers(1, &screen->monitors[i].vertex_buffer);
 808                         glDeleteBuffers(1, &screen->monitors[i].index_buffer);
 809                         glDeleteVertexArrays(1, &screen->monitors[i].vertex_array);
 810                 }
 811         }
 812
 813         glDeleteBuffers(1, &screen->window_vertex_buffer);
 814         glDeleteVertexArrays(1, &screen->window_vertex_array);
 815         glDeleteFramebuffers(1, &screen->framebuffer);
 816         glDeleteTextures(1, &screen->fb_texture);
 817         glDeleteProgram(screen->program);
 818         glDeleteShader(screen->shaders[0]);
 819         glDeleteShader(screen->shaders[1]);
 820
 821         glXMakeContextCurrent(compositor->display, 0, 0, NULL);
 822         glXDestroyContext(compositor->display, screen->glx_context);
 823         glXDestroyWindow(compositor->display, screen->glx_window);
 824         XDestroyWindow(compositor->display, screen->render_window);
 825
 826         XCompositeReleaseOverlayWindow(compositor->display, screen->overlay);
 827
 828         free(screen->windows);
 829         free(screen->monitors);
 830 }
 831
 832 void shutdown_compositor(Compositor *compositor)
 833 {
 834         unsigned i;
 835
 836         for(i=0; i<compositor->nscreens; ++i)
 837                 shutdown_screen(compositor, &compositor->screens[i]);
 838         free(compositor->screens);
 839
 840         XCloseDisplay(compositor->display);
 841 }
 842
 843 void mark_dirty(Compositor *compositor, CompositedScreen *screen)
 844 {
 845         compositor->dirty = 1;
 846         screen->dirty = 1;
 847 }
 848
 849 void process_create_window_event(Compositor *compositor, XCreateWindowEvent *event)
 850 {
 851         CompositedScreen *screen;
 852
 853         if((screen = find_screen_by_root(compositor, event->parent)))
 854                 add_window(compositor, screen, event->window);
 855 }
 856
 857 void process_destroy_window_event(Compositor *compositor, XDestroyWindowEvent *event)
 858 {
 859         CompositedScreen *screen;
 860         CompositedWindow *window;
 861
 862         if((screen = find_screen_by_root(compositor, event->event)))
 863                 if((window = find_window(screen, event->window)))
 864                         remove_window(compositor, screen, window, 1);
 865 }
 866
 867 void process_map_event(Compositor *compositor, XMapEvent *event)
 868 {
 869         CompositedScreen *screen;
 870         CompositedWindow *window;
 871
 872         screen = find_screen_by_root(compositor, event->event);
 873         if(!screen)
 874                 return;
 875
 876         window = find_window(screen, event->window);
 877         if(!window)
 878                 return;
 879
 880         window->map_state = IsViewable;
 881         create_window_pixmap(compositor, screen, window);
 882
 883         mark_dirty(compositor, screen);
 884 }
 885
 886 void process_unmap_event(Compositor *compositor, XUnmapEvent *event)
 887 {
 888         CompositedScreen *screen;
 889         CompositedWindow *window;
 890
 891         screen = find_screen_by_root(compositor, event->event);
 892         if(!screen)
 893                 return;
 894
 895         window = find_window(screen, event->window);
 896         if(window)
 897                 window->map_state = IsUnviewable;
 898
 899         mark_dirty(compositor, screen);
 900 }
 901
 902 void process_reparent_event(Compositor *compositor, XReparentEvent *event)
 903 {
 904         CompositedScreen *screen;
 905         CompositedWindow *window;
 906
 907         screen = find_screen_by_root(compositor, event->event);
 908         if(!screen)
 909                 return;
 910
 911         if(event->parent==screen->root)
 912                 window = add_window(compositor, screen, event->window);
 913         else
 914         {
 915                 window = find_window(screen, event->window);
 916                 if(!window)
 917                         return;
 918
 919                 remove_window(compositor, screen, window, 0);
 920         }
 921
 922         if(window && window->map_state==IsViewable)
 923                 mark_dirty(compositor, screen);
 924 }
 925
 926 void process_configure_event(Compositor *compositor, XConfigureEvent *event)
 927 {
 928         CompositedScreen *screen;
 929         CompositedWindow *window;
 930
 931         screen = find_screen_by_root(compositor, event->event);
 932         if(!screen)
 933                 return;
 934
 935         window = find_window(screen, event->window);
 936         if(!window)
 937                 return;
 938
 939         window->x = event->x;
 940         window->y = event->y;
 941         if((unsigned)event->width!=window->width || (unsigned)event->height!=window->height || (unsigned)event->border_width!=window->border)
 942         {
 943                 window->width = event->width;
 944                 window->height = event->height;
 945                 window->border = event->border_width;
 946                 create_window_pixmap(compositor, screen, window);
 947         }
 948         reorder_window(screen, window, event->above);
 949
 950         if(window->map_state==IsViewable)
 951                 mark_dirty(compositor, screen);
 952 }
 953
 954 void process_property_event(Compositor *compositor, XPropertyEvent *event)
 955 {
 956         CompositedScreen *screen;
 957
 958         if(event->atom!=compositor->correction_atom)
 959                 return;
 960
 961         screen = find_screen_by_root(compositor, event->window);
 962         if(!screen)
 963                 return;
 964
 965         update_geometry_correction(compositor, screen);
 966 }
 967
 968 void process_damage_event(Compositor *compositor, XDamageNotifyEvent *event)
 969 {
 970         CompositedScreen *screen;
 971         CompositedWindow *window;
 972
 973         screen = find_screen_by_window(compositor, event->drawable);
 974         if(!screen)
 975                 return;
 976
 977         window = find_window(screen, event->drawable);
 978         if(window->map_state==IsViewable)
 979                 mark_dirty(compositor, screen);
 980 }
 981
 982 int process_event(Compositor *compositor)
 983 {
 984         XEvent event;
 985         if(compositor->dirty)
 986         {
 987                 if(!XCheckMaskEvent(compositor->display, -1, &event))
 988                         return 0;
 989         }
 990         else
 991                 XNextEvent(compositor->display, &event);
 992
 993         switch(event.type)
 994         {
 995         case CreateNotify:
 996                 process_create_window_event(compositor, &event.xcreatewindow);
 997                 break;
 998         case DestroyNotify:
 999                 process_destroy_window_event(compositor, &event.xdestroywindow);
1000                 break;
1001         case MapNotify:
1002                 process_map_event(compositor, &event.xmap);
1003                 break;
1004         case UnmapNotify:
1005                 process_unmap_event(compositor, &event.xunmap);
1006                 break;
1007         case ReparentNotify:
1008                 process_reparent_event(compositor, &event.xreparent);
1009                 break;
1010         case ConfigureNotify:
1011                 process_configure_event(compositor, &event.xconfigure);
1012                 break;
1013         case PropertyNotify:
1014                 process_property_event(compositor, &event.xproperty);
1015                 break;
1016         default:
1017                 if(event.type==compositor->damage_event+XDamageNotify)
1018                         process_damage_event(compositor, (XDamageNotifyEvent *)&event);
1019                 else
1020                         printf("Event %d\n", event.type);
1021         }
1022
1023         return 1;
1024 }
1025
1026 void refresh_screen(Compositor *compositor, CompositedScreen *screen)
1027 {
1028         unsigned i;
1029
1030         use_gl(compositor, screen);
1031
1032         glBindFramebuffer(GL_DRAW_FRAMEBUFFER, screen->framebuffer);
1033
1034         glClearColor(0.5f, 0.5f, 0.5f, 0.0f);
1035         glClear(GL_COLOR_BUFFER_BIT);
1036
1037         glUseProgram(screen->program);
1038         glBindVertexArray(screen->window_vertex_array);
1039         for(i=0; i<screen->nwindows; ++i)
1040         {
1041                 CompositedWindow *window = &screen->windows[i];
1042                 if(window->map_state!=IsViewable)
1043                         continue;
1044
1045                 glBindTexture(GL_TEXTURE_2D, window->texture);
1046                 compositor->glXBindTexImageEXT(compositor->display, window->glx_pixmap, GLX_FRONT_LEFT_EXT, NULL);
1047                 glUniform4f(screen->geometry_loc,
1048                         (float)window->x/screen->width, ((float)screen->height-window->y-window->height)/screen->height,
1049                         (float)(window->width+2*window->border)/screen->width, (float)(window->height+2*window->border)/screen->height);
1050                 glDrawArrays(GL_TRIANGLE_STRIP, 0, 4);
1051                 compositor->glXReleaseTexImageEXT(compositor->display, window->glx_pixmap, GLX_FRONT_LEFT_EXT);
1052                 XDamageSubtract(compositor->display, window->damage, None, None);
1053         }
1054
1055         glBindFramebuffer(GL_DRAW_FRAMEBUFFER, 0);
1056         glClearColor(0.5f, 0.0f, 0.5f, 0.0f);
1057         glClear(GL_COLOR_BUFFER_BIT);
1058         glBindTexture(GL_TEXTURE_2D, screen->fb_texture);
1059         glEnable(GL_PRIMITIVE_RESTART);
1060         glPrimitiveRestartIndex(0xFFFF);
1061
1062         for(i=0; i<screen->nmonitors; ++i)
1063         {
1064                 CompositedMonitor *monitor = &screen->monitors[i];
1065                 if(!monitor->enabled)
1066                         continue;
1067
1068                 glUniform4f(screen->geometry_loc,
1069                         (float)monitor->x/screen->width, (float)monitor->y/screen->height,
1070                         (float)monitor->width/screen->width, (float)monitor->height/screen->height);
1071
1072                 glBindVertexArray(monitor->vertex_array);
1073                 glDrawElements(GL_TRIANGLE_STRIP, monitor->nelements, GL_UNSIGNED_SHORT, NULL);
1074         }
1075
1076         glBindVertexArray(0);
1077
1078         glXSwapBuffers(compositor->display, screen->glx_window);
1079
1080         screen->dirty = 0;
1081 }
1082
1083 void refresh_all_screens(Compositor *compositor)
1084 {
1085         unsigned i;
1086
1087         for(i=0; i<compositor->nscreens; ++i)
1088         {
1089                 CompositedScreen *screen = &compositor->screens[i];
1090                 if(screen->dirty)
1091                         refresh_screen(compositor, screen);
1092         }
1093
1094         compositor->dirty = 0;
1095 }
1096
1097 void sighandler(int sig)
1098 {
1099         terminate_requested = 1;
1100         (void)sig;
1101 }
1102
1103 int main()
1104 {
1105         Compositor compositor;
1106
1107         signal(SIGINT, &sighandler);
1108         signal(SIGTERM, &sighandler);
1109
1110         if(!initialize_compositor(&compositor))
1111                 return 1;
1112
1113         while(!terminate_requested)
1114         {
1115                 if(!process_event(&compositor))
1116                         refresh_all_screens(&compositor);
1117         }
1118
1119         shutdown_compositor(&compositor);
1120
1121         return 0;
1122 }