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         int recreate_pixmap;
  29         unsigned texture;
  30         unsigned mask_texture;
  31         int use_mask;
  32         int recreate_mask;
  33 } CompositedWindow;
  34
  35 enum
  36 {
  37         CYLINDRICAL = 1,
  38         SPHERICAL
  39 };
  40
  41 typedef struct CompositedMonitor
  42 {
  43         char *name;
  44         int enabled;
  45         int x;
  46         int y;
  47         unsigned width;
  48         unsigned height;
  49         float keystone_vertical;
  50         float curvature_type;
  51         float curvature_depth;
  52         float vertical_center;
  53         float perspective;
  54         unsigned vertex_buffer;
  55         unsigned index_buffer;
  56         unsigned vertex_array;
  57         unsigned tessellation;
  58         unsigned nelements;
  59         short *geometry_data;
  60         unsigned geometry_data_size;
  61 } CompositedMonitor;
  62
  63 typedef struct CompositedScreen
  64 {
  65         int number;
  66         Window root;
  67         unsigned width;
  68         unsigned height;
  69         Window overlay;
  70         Window render_window;
  71         GLXFBConfig fbconfig;
  72         GLXWindow glx_window;
  73         GLXContext glx_context;
  74         unsigned shaders[3];
  75         unsigned program;
  76         int geometry_loc;
  77         unsigned masked_program;
  78         int masked_geometry_loc;
  79         unsigned window_vertex_buffer;
  80         unsigned window_vertex_array;
  81         unsigned framebuffer;
  82         unsigned fb_texture;
  83         Pixmap root_pixmap;
  84         GLXPixmap root_glx_pixmap;
  85         unsigned root_texture;
  86         CompositedWindow *windows;
  87         unsigned nwindows;
  88         unsigned windows_capacity;
  89         CompositedMonitor *monitors;
  90         unsigned nmonitors;
  91         int dirty;
  92 } CompositedScreen;
  93
  94 typedef struct Compositor
  95 {
  96         Display *display;
  97         CompositedScreen *screens;
  98         unsigned nscreens;
  99         int damage_event;
 100         int shape_event;
 101         int randr_event;
 102         PFNGLXCREATECONTEXTATTRIBSARBPROC glXCreateContextAttribsARB;
 103         PFNGLXBINDTEXIMAGEEXTPROC glXBindTexImageEXT;
 104         PFNGLXRELEASETEXIMAGEEXTPROC glXReleaseTexImageEXT;
 105         Atom root_pmap_atom;
 106         Atom correction_atom;
 107         Atom monitors_atom;
 108         Atom geometry_data_atom;
 109         Window selection_window;
 110         int dirty;
 111 } Compositor;
 112
 113 static const char *vshader_src =
 114         "#version 130\n"
 115         "uniform vec4 geometry;\n"
 116         "in vec2 vertex;\n"
 117         "in vec2 texture_coord;\n"
 118         "out vec2 texcoord;\n"
 119         "void main()\n"
 120         "{\n"
 121         "  gl_Position = vec4((geometry.xy+vertex*geometry.zw)*2.0-1.0, 0.0, 1.0);\n"
 122         "  texcoord = texture_coord;\n"
 123         "}\n";
 124
 125 static const char *fshader_src =
 126         "#version 130\n"
 127         "uniform sampler2D image;\n"
 128         "in vec2 texcoord;\n"
 129         "out vec4 frag_color;\n"
 130         "void main()\n"
 131         "{\n"
 132         "  frag_color = texture(image, texcoord);\n"
 133         "}\n";
 134
 135 static const char *masked_fshader_src =
 136         "#version 130\n"
 137         "uniform sampler2D image;\n"
 138         "uniform sampler2D mask;\n"
 139         "in vec2 texcoord;\n"
 140         "out vec4 frag_color;\n"
 141         "void main()\n"
 142         "{\n"
 143         "  if(texture(mask, texcoord).r==0.0)\n"
 144         "    discard;\n"
 145         "  frag_color = texture(image, texcoord);\n"
 146         "}\n";
 147
 148 static const float window_vertices[] =
 149 {
 150         /* vertex    texcoord */
 151         0.0f, 1.0f,  0.0f, 0.0f,
 152         0.0f, 0.0f,  0.0f, 1.0f,
 153         1.0f, 1.0f,  1.0f, 0.0f,
 154         1.0f, 0.0f,  1.0f, 1.0f
 155 };
 156
 157 int terminate_requested = 0;
 158
 159 int x_error_handler(Display *display, XErrorEvent *event)
 160 {
 161         printf("Ignoring X error %d on resource %lx\n", event->error_code, event->resourceid);
 162         (void)display;
 163         return 0;
 164 }
 165
 166 int with_error(const char *message)
 167 {
 168         fprintf(stderr, "%s\n", message);
 169         return 0;
 170 }
 171
 172 int initialize_gl(Compositor *compositor, CompositedScreen *screen)
 173 {
 174         int attribs[9];
 175         unsigned i;
 176         GLXFBConfig *configs;
 177         int nconfigs;
 178         XVisualInfo *vi;
 179         XSetWindowAttributes win_attr;
 180
 181         i = 0;
 182         attribs[i++] = GLX_DRAWABLE_TYPE;
 183         attribs[i++] = GLX_WINDOW_BIT;
 184         attribs[i++] = GLX_RENDER_TYPE;
 185         attribs[i++] = GLX_RGBA_BIT;
 186         attribs[i++] = GLX_DOUBLEBUFFER;
 187         attribs[i++] = True;
 188         attribs[i++] = GLX_BIND_TO_TEXTURE_RGBA_EXT;
 189         attribs[i++] = True;
 190         attribs[i] = None;
 191
 192         configs = glXChooseFBConfig(compositor->display, screen->number, attribs, &nconfigs);
 193         if(!configs || !nconfigs)
 194                 return with_error("Could not find a suitable FBConfig");
 195         screen->fbconfig = configs[0];
 196         XFree(configs);
 197
 198         vi = glXGetVisualFromFBConfig(compositor->display, screen->fbconfig);
 199         win_attr.colormap = XCreateColormap(compositor->display, screen->root, vi->visual, AllocNone);
 200         screen->render_window = XCreateWindow(compositor->display, screen->overlay, 0, 0, screen->width, screen->height, 0, vi->depth, InputOutput, vi->visual, CWColormap, &win_attr);
 201         XMapWindow(compositor->display, screen->render_window);
 202
 203         screen->glx_window = glXCreateWindow(compositor->display, screen->fbconfig, screen->render_window, NULL);
 204
 205         i = 0;
 206         attribs[i++] = GLX_CONTEXT_FLAGS_ARB;
 207         attribs[i++] = GLX_CONTEXT_FORWARD_COMPATIBLE_BIT_ARB;
 208         attribs[i++] = GLX_CONTEXT_MAJOR_VERSION_ARB;
 209         attribs[i++] = 3;
 210         attribs[i++] = GLX_CONTEXT_MINOR_VERSION_ARB;
 211         attribs[i++] = 1;
 212         attribs[i] = None;
 213
 214         screen->glx_context = compositor->glXCreateContextAttribsARB(compositor->display, screen->fbconfig, NULL, True, attribs);
 215
 216         XFree(vi);
 217
 218         return 1;
 219 }
 220
 221 void use_gl(Compositor *compositor, CompositedScreen *screen)
 222 {
 223         glXMakeContextCurrent(compositor->display, screen->glx_window, screen->glx_window, screen->glx_context);
 224 }
 225
 226 unsigned compile_shader(GLenum type, const char *source)
 227 {
 228         unsigned shader;
 229         int status;
 230         char info_log[1024];
 231         GLsizei length;
 232
 233         shader = glCreateShader(type);
 234         glShaderSource(shader, 1, &source, NULL);
 235         glCompileShader(shader);
 236         glGetShaderiv(shader, GL_COMPILE_STATUS, &status);
 237         glGetShaderInfoLog(shader, sizeof(info_log), &length, info_log);
 238         if(!status)
 239         {
 240                 fprintf(stderr, "Shader compilation failed:\n%s\n", info_log);
 241                 glDeleteShader(shader);
 242                 return 0;
 243         }
 244         else if(length)
 245                 printf("Shader info log:\n%s\n", info_log);
 246
 247         return shader;
 248 }
 249
 250 unsigned link_program(unsigned vshader, unsigned fshader)
 251 {
 252         unsigned program;
 253         int status;
 254         char info_log[1024];
 255         GLsizei length;
 256
 257         program = glCreateProgram();
 258         glAttachShader(program, vshader);
 259         glAttachShader(program, fshader);
 260         glBindAttribLocation(program, 0, "vertex");
 261         glBindAttribLocation(program, 1, "texture_coord");
 262         glBindFragDataLocation(program, 0, "frag_color");
 263         glLinkProgram(program);
 264
 265         glGetProgramiv(program, GL_LINK_STATUS, &status);
 266         glGetProgramInfoLog(program, sizeof(info_log), &length, info_log);
 267         if(!status)
 268         {
 269                 fprintf(stderr, "Program link failed:\n%s\n", info_log);
 270                 glDeleteProgram(program);
 271                 return 0;
 272         }
 273         else if(length)
 274                 printf("Program info log:\n%s\n", info_log);
 275
 276         return program;
 277 }
 278
 279 unsigned create_2d_texture()
 280 {
 281         unsigned texture;
 282         glGenTextures(1, &texture);
 283         glBindTexture(GL_TEXTURE_2D, texture);
 284         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
 285         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
 286         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
 287         return texture;
 288 }
 289
 290 int create_gl_resources(CompositedScreen *screen)
 291 {
 292         unsigned stride;
 293         int loc;
 294
 295         screen->shaders[0] = compile_shader(GL_VERTEX_SHADER, vshader_src);
 296         screen->shaders[1] = compile_shader(GL_FRAGMENT_SHADER, fshader_src);
 297         screen->shaders[2] = compile_shader(GL_FRAGMENT_SHADER, masked_fshader_src);
 298         if(!screen->shaders[0] || !screen->shaders[1] || !screen->shaders[2])
 299                 return 0;
 300
 301         screen->program = link_program(screen->shaders[0], screen->shaders[1]);
 302         if(!screen->program)
 303                 return 0;
 304
 305         screen->masked_program = link_program(screen->shaders[0], screen->shaders[2]);
 306         if(!screen->masked_program)
 307                 return 0;
 308
 309         screen->geometry_loc = glGetUniformLocation(screen->program, "geometry");
 310         screen->masked_geometry_loc = glGetUniformLocation(screen->masked_program, "geometry");
 311
 312         loc = glGetUniformLocation(screen->masked_program, "mask");
 313         if(loc>=0)
 314         {
 315                 glUseProgram(screen->masked_program);
 316                 glUniform1i(loc, 1);
 317         }
 318
 319         glGenBuffers(1, &screen->window_vertex_buffer);
 320         glBindBuffer(GL_ARRAY_BUFFER, screen->window_vertex_buffer);
 321         glBufferData(GL_ARRAY_BUFFER, sizeof(window_vertices), window_vertices, GL_STATIC_DRAW);
 322
 323         stride = 4*sizeof(float);
 324         glGenVertexArrays(1, &screen->window_vertex_array);
 325         glBindVertexArray(screen->window_vertex_array);
 326         glVertexAttribPointer(0, 2, GL_FLOAT, GL_FALSE, stride, NULL);
 327         glEnableVertexAttribArray(0);
 328         glVertexAttribPointer(1, 2, GL_FLOAT, GL_FALSE, stride, (void *)(2*sizeof(float)));
 329         glEnableVertexAttribArray(1);
 330         glBindVertexArray(0);
 331
 332         glBindBuffer(GL_ARRAY_BUFFER, 0);
 333
 334         screen->fb_texture = create_2d_texture();
 335         glTexImage2D(GL_TEXTURE_2D, 0, GL_RGB8, screen->width, screen->height, 0, GL_RGB, GL_UNSIGNED_BYTE, NULL);
 336         glBindTexture(GL_TEXTURE_2D, 0);
 337
 338         glGenFramebuffers(1, &screen->framebuffer);
 339         glBindFramebuffer(GL_DRAW_FRAMEBUFFER, screen->framebuffer);
 340         glFramebufferTexture2D(GL_DRAW_FRAMEBUFFER, GL_COLOR_ATTACHMENT0, GL_TEXTURE_2D, screen->fb_texture, 0);
 341         glDepthMask(GL_FALSE);
 342         glBindFramebuffer(GL_DRAW_FRAMEBUFFER, 0);
 343
 344         screen->root_texture = create_2d_texture();
 345
 346         return 1;
 347 }
 348
 349 CompositedWindow *find_window(CompositedScreen *screen, Window w)
 350 {
 351         unsigned i;
 352
 353         for(i=0; i<screen->nwindows; ++i)
 354                 if(screen->windows[i].window==w)
 355                         return &screen->windows[i];
 356
 357         return NULL;
 358 }
 359
 360 CompositedWindow *find_window_global(Compositor *compositor, Window w, CompositedScreen **screen)
 361 {
 362         unsigned i, j;
 363
 364         for(i=0; i<compositor->nscreens; ++i)
 365                 for(j=0; j<compositor->screens[i].nwindows; ++j)
 366                         if(compositor->screens[i].windows[j].window==w)
 367                         {
 368                                 if(screen)
 369                                         *screen = &compositor->screens[i];
 370                                 return &compositor->screens[i].windows[j];
 371                         }
 372
 373         if(screen)
 374                 *screen = NULL;
 375         return NULL;
 376 }
 377
 378 GLXPixmap pixmap_to_glx_pixmap(Compositor *compositor, CompositedScreen *screen, Pixmap pixmap)
 379 {
 380         int attribs[5];
 381         unsigned i;
 382
 383         i = 0;
 384         attribs[i++] = GLX_TEXTURE_TARGET_EXT;
 385         attribs[i++] = GLX_TEXTURE_2D_EXT;
 386         attribs[i++] = GLX_TEXTURE_FORMAT_EXT;
 387         attribs[i++] = GLX_TEXTURE_FORMAT_RGBA_EXT;
 388         attribs[i++] = None;
 389
 390         return glXCreatePixmap(compositor->display, screen->fbconfig, pixmap, attribs);
 391 }
 392
 393 void create_window_pixmap(Compositor *compositor, CompositedScreen *screen, CompositedWindow *window)
 394 {
 395         if(window->pixmap)
 396         {
 397                 glXDestroyPixmap(compositor->display, window->glx_pixmap);
 398                 XFreePixmap(compositor->display, window->pixmap);
 399         }
 400
 401         window->pixmap = XCompositeNameWindowPixmap(compositor->display, window->window);
 402         window->glx_pixmap = pixmap_to_glx_pixmap(compositor, screen, window->pixmap);
 403         window->recreate_pixmap = 0;
 404 }
 405
 406 void update_window_mask(Compositor *compositor, CompositedWindow *window)
 407 {
 408         Bool bounding_shaped;
 409         Bool clip_shaped;
 410         int xi, yi;
 411         XRectangle *rects;
 412         int rect_count;
 413         int rect_order;
 414         unsigned width;
 415         unsigned height;
 416         unsigned char *data;
 417         int i;
 418         unsigned y;
 419
 420         XShapeQueryExtents(compositor->display, window->window, &bounding_shaped, &xi, &yi, &width, &height, &clip_shaped, &xi, &yi, &width, &height);
 421         window->use_mask = bounding_shaped;
 422         if(!window->use_mask)
 423                 return;
 424
 425         rects = XShapeGetRectangles(compositor->display, window->window, ShapeBounding, &rect_count, &rect_order);
 426         if(!rects)
 427                 return;
 428
 429         width = window->width+2*window->border;
 430         height = window->height+2*window->border;
 431         data = (unsigned char *)malloc(width*height);
 432         memset(data, 0, width*height);
 433         for(i=0; i<rect_count; ++i)
 434         {
 435                 rects[i].x += window->border;
 436                 rects[i].y += window->border;
 437                 if(rects[i].x>=(int)width || rects[i].y>=(int)height)
 438                         continue;
 439
 440                 if(rects[i].x<0)
 441                 {
 442                         if(-rects[i].x>rects[i].width)
 443                                 continue;
 444                         rects[i].width += rects[i].x;
 445                         rects[i].x = 0;
 446                 }
 447
 448                 if(rects[i].y<0)
 449                 {
 450                         if(-rects[i].y>rects[i].height)
 451                                 continue;
 452                         rects[i].height += rects[i].y;
 453                         rects[i].y = 0;
 454                 }
 455
 456                 if(rects[i].x+rects[i].width>(int)width)
 457                         rects[i].width = width-rects[i].x;
 458                 if(rects[i].y+rects[i].height>(int)height)
 459                         rects[i].height = height-rects[i].y;
 460
 461                 for(y=0; y<rects[i].height; ++y)
 462                 {
 463                         unsigned char *row = data+(rects[i].y+y)*width+rects[i].x;
 464                         memset(row, 255, rects[i].width);
 465                 }
 466         }
 467
 468         XFree(rects);
 469
 470         glBindTexture(GL_TEXTURE_2D, window->mask_texture);
 471         glPixelStorei(GL_UNPACK_ALIGNMENT, 1);
 472         glTexImage2D(GL_TEXTURE_2D, 0, GL_R8, width, height, 0, GL_RED, GL_UNSIGNED_BYTE, data);
 473
 474         free(data);
 475
 476         window->recreate_mask = 0;
 477 }
 478
 479 CompositedWindow *add_window(Compositor *compositor, CompositedScreen *screen, Window w)
 480 {
 481         CompositedWindow *window;
 482         XWindowAttributes win_attr;
 483
 484         if(w==screen->root || w==screen->overlay)
 485                 return NULL;
 486
 487         if(!XGetWindowAttributes(compositor->display, w, &win_attr))
 488         {
 489                 printf("XGetWindowAttributes failed; probably the window was already destroyed\n");
 490                 return NULL;
 491         }
 492         if(win_attr.class==InputOnly)
 493                 return NULL;
 494
 495         if(find_window(screen, w))
 496                 return NULL;
 497
 498         if(screen->nwindows==screen->windows_capacity)
 499         {
 500                 screen->windows = (CompositedWindow *)realloc(screen->windows, (screen->windows_capacity+1)*sizeof(CompositedWindow));
 501                 ++screen->windows_capacity;
 502         }
 503
 504         window = &screen->windows[screen->nwindows++];
 505
 506         window->window = w;
 507         window->x = win_attr.x;
 508         window->y = win_attr.y;
 509         window->width = win_attr.width;
 510         window->height = win_attr.height;
 511         window->border = win_attr.border_width;
 512         window->map_state = win_attr.map_state;
 513
 514         window->damage = XDamageCreate(compositor->display, window->window, XDamageReportNonEmpty);
 515         window->pixmap = None;
 516         window->glx_pixmap = None;
 517
 518         XCompositeRedirectWindow(compositor->display, window->window, CompositeRedirectManual);
 519         window->recreate_pixmap = (window->map_state==IsViewable);
 520
 521         window->texture = create_2d_texture();
 522         window->mask_texture = create_2d_texture();
 523         window->use_mask = 0;
 524         window->recreate_mask = (window->map_state==IsViewable);
 525
 526         XShapeSelectInput(compositor->display, window->window, ShapeNotifyMask);
 527
 528         return window;
 529 }
 530
 531 void remove_window(Compositor *compositor, CompositedScreen *screen, CompositedWindow *window, int destroyed)
 532 {
 533         unsigned i;
 534
 535         glDeleteTextures(1, &window->texture);
 536         if(!destroyed)
 537         {
 538                 XDamageDestroy(compositor->display, window->damage);
 539                 if(window->pixmap)
 540                 {
 541                         glXDestroyPixmap(compositor->display, window->glx_pixmap);
 542                         XFreePixmap(compositor->display, window->pixmap);
 543                 }
 544                 XCompositeUnredirectWindow(compositor->display, window->window, CompositeRedirectManual);
 545         }
 546
 547         --screen->nwindows;
 548         for(i=window-screen->windows; i<screen->nwindows; ++i)
 549                 screen->windows[i] = screen->windows[i+1];
 550 }
 551
 552 CompositedWindow *reorder_window(CompositedScreen *screen, CompositedWindow *window, Window above)
 553 {
 554         unsigned i, j;
 555         CompositedWindow hold;
 556
 557         i = window-screen->windows;
 558         if(above)
 559         {
 560                 for(j=0; j<screen->nwindows; ++j)
 561                         if(screen->windows[j].window==above)
 562                                 break;
 563
 564                 if(j>=screen->nwindows || i==j+1)
 565                         return window;
 566
 567                 if(j<i)
 568                         ++j;
 569         }
 570         else
 571                 j = 0;
 572
 573         hold = *window;
 574         if(i<j)
 575         {
 576                 for(; i<j; ++i)
 577                         screen->windows[i] = screen->windows[i+1];
 578         }
 579         else
 580         {
 581                 for(; i>j; --i)
 582                         screen->windows[i] = screen->windows[i-1];
 583         }
 584         screen->windows[j] = hold;
 585
 586         return &screen->windows[j];
 587 }
 588
 589 CompositedScreen *find_screen_by_root(Compositor *compositor, Window root)
 590 {
 591         unsigned i;
 592
 593         for(i=0; i<compositor->nscreens; ++i)
 594                 if(compositor->screens[i].root==root)
 595                         return &compositor->screens[i];
 596
 597         return NULL;
 598 }
 599
 600 void create_geometry_correction(CompositedMonitor *monitor)
 601 {
 602         unsigned buffers[2];
 603         unsigned stride;
 604
 605         glGenBuffers(2, buffers);
 606         monitor->vertex_buffer = buffers[0];
 607         monitor->index_buffer = buffers[1];
 608
 609         monitor->tessellation = 50;
 610         monitor->geometry_data_size = (monitor->tessellation+1)*(monitor->tessellation+1)*3+16;
 611         monitor->geometry_data = (short *)malloc(monitor->geometry_data_size*sizeof(short));
 612         stride = 4*sizeof(float);
 613
 614         glGenVertexArrays(1, &monitor->vertex_array);
 615         glBindVertexArray(monitor->vertex_array);
 616         glBindBuffer(GL_ARRAY_BUFFER, monitor->vertex_buffer);
 617         glVertexAttribPointer(0, 2, GL_FLOAT, GL_FALSE, stride, NULL);
 618         glEnableVertexAttribArray(0);
 619         glVertexAttribPointer(1, 2, GL_FLOAT, GL_FALSE, stride, (void *)(2*sizeof(float)));
 620         glEnableVertexAttribArray(1);
 621         glBindBuffer(GL_ARRAY_BUFFER, 0);
 622         glBindBuffer(GL_ELEMENT_ARRAY_BUFFER, monitor->index_buffer);
 623         glBindVertexArray(0);
 624 }
 625
 626 void free_geometry_correction(CompositedMonitor *monitor)
 627 {
 628         unsigned buffers[2];
 629
 630         buffers[0] = monitor->vertex_buffer;
 631         buffers[1] = monitor->index_buffer;
 632         glDeleteBuffers(2, buffers);
 633         monitor->vertex_buffer = 0;
 634         monitor->index_buffer = 0;
 635
 636         glDeleteVertexArrays(1, &monitor->vertex_array);
 637
 638         free(monitor->geometry_data);
 639         monitor->geometry_data = 0;
 640 }
 641
 642 void update_monitor_vertices(CompositedScreen *screen, CompositedMonitor *monitor)
 643 {
 644         unsigned t;
 645         unsigned data_size;
 646         float *vertex_data;
 647         unsigned short *index_data;
 648         unsigned x, y;
 649         unsigned i;
 650         float aspect;
 651         float curve_radius;
 652         float curve_arc;
 653         float sin_ksv;
 654         float cos_ksv;
 655         float distance;
 656         float eye[3];
 657         float look[3];
 658
 659         t = monitor->tessellation;
 660
 661         data_size = (t+1)*(t+1)*4*sizeof(float);
 662         vertex_data = (float *)malloc(data_size);
 663
 664         aspect = (float)monitor->width/monitor->height;
 665
 666         if(monitor->curvature_depth)
 667         {
 668                 curve_radius = (monitor->curvature_depth*monitor->curvature_depth+0.25f)/(2.0f*monitor->curvature_depth);
 669                 curve_arc = 2.0f*asin(0.5f/curve_radius);
 670         }
 671
 672         sin_ksv = monitor->keystone_vertical/sqrt(1.0f+monitor->keystone_vertical*monitor->keystone_vertical);
 673         cos_ksv = sqrt(1.0f-sin_ksv*sin_ksv);
 674         distance = monitor->perspective+sin_ksv*((sin_ksv>0)-monitor->vertical_center)/aspect;
 675         if(monitor->curvature_depth<0)
 676                 distance += -monitor->curvature_depth;
 677
 678         eye[0] = 0.0f;
 679         eye[1] = (monitor->vertical_center-0.5f)/aspect+sin_ksv*distance;
 680         eye[2] = cos_ksv*distance;
 681
 682         look[0] = 0.0f;
 683         look[1] = -sin_ksv;
 684         look[2] = -cos_ksv;
 685
 686         for(y=0; y<=t; ++y)
 687                 for(x=0; x<=t; ++x)
 688                 {
 689                         float *v;
 690
 691                         v = vertex_data+(y*(t+1)+x)*3;
 692                         v[0] = (float)x/t-0.5f;
 693                         v[1] = ((float)y/t-0.5f)/aspect;
 694                         v[2] = 0;
 695                         if(monitor->curvature_depth)
 696                         {
 697                                 if(monitor->curvature_type==CYLINDRICAL)
 698                                 {
 699                                         v[2] = (1.0f-cos(v[0]*curve_arc))*curve_radius-monitor->curvature_depth;
 700                                         v[0] = sin(v[0]*curve_arc)*curve_radius;
 701                                 }
 702                                 else if(monitor->curvature_type==SPHERICAL)
 703                                 {
 704                                         float r;
 705
 706                                         v[0] = tan(v[0]*curve_arc)*curve_radius;
 707                                         v[1] = tan(v[1]*curve_arc)*curve_radius;
 708                                         r = sqrt(v[0]*v[0]+v[1]*v[1]+curve_radius*curve_radius)/fabs(curve_radius);
 709                                         v[0] /= r;
 710                                         v[1] /= r;
 711                                         v[2] = curve_radius-curve_radius/r-monitor->curvature_depth;
 712                                 }
 713                         }
 714                 }
 715
 716         monitor->geometry_data[0] = t;
 717         for(i=0; i<3; ++i)
 718                 monitor->geometry_data[1+i] = eye[i]*4096;
 719         for(y=0; y<2; ++y)
 720                 for(x=0; x<2; ++x)
 721                 {
 722                         i = 1+(1+y*2+x)*3;
 723                         monitor->geometry_data[i] = ((x-0.5f)*distance/monitor->perspective)*4096;
 724                         monitor->geometry_data[i+1] = (eye[1]+look[1]*distance-(y-monitor->vertical_center)*look[2]*distance/monitor->perspective/aspect)*4096;
 725                         monitor->geometry_data[i+2] = (eye[2]+look[2]*distance+(y-monitor->vertical_center)*look[1]*distance/monitor->perspective/aspect)*4096;
 726                 }
 727         for(i=0; i<(t+1)*(t+1)*3; ++i)
 728                 monitor->geometry_data[16+i] = vertex_data[i]*4096;
 729
 730         for(y=t; y<=t; --y)
 731                 for(x=t; x<=t; --x)
 732                 {
 733                         float *v;
 734                         float px, py, pz;
 735                         float scale;
 736
 737                         v = vertex_data+(y*(t+1)+x)*3;
 738                         px = v[0]-eye[0];
 739                         py = (v[0]-eye[0])*look[0] - (v[1]-eye[1])*look[2] + (v[2]-eye[2])*look[1];
 740                         pz = (v[0]-eye[0])*look[0] + (v[1]-eye[1])*look[1] + (v[2]-eye[2])*look[2];
 741                         scale = monitor->perspective/pz;
 742
 743                         v = vertex_data+(y*(t+1)+x)*4;
 744                         v[0] = px*scale+0.5f;
 745                         v[1] = py*aspect*scale+monitor->vertical_center;
 746                         v[2] = (monitor->x+(float)x*monitor->width/t)/screen->width;
 747                         v[3] = 1.0f-(monitor->y+(1.0f-(float)y/t)*monitor->height)/screen->height;
 748                 }
 749
 750         glBindBuffer(GL_ARRAY_BUFFER, monitor->vertex_buffer);
 751         glBufferData(GL_ARRAY_BUFFER, data_size, vertex_data, GL_STATIC_DRAW);
 752         glBindBuffer(GL_ARRAY_BUFFER, 0);
 753         free(vertex_data);
 754
 755         monitor->nelements = t*((t+1)*2+1)-1;
 756         data_size = monitor->nelements*sizeof(unsigned short);
 757         index_data = (unsigned short *)malloc(data_size);
 758         i = 0;
 759         for(y=0; y<t; ++y)
 760         {
 761                 if(y>0)
 762                         index_data[i++] = 0xFFFF;
 763                 for(x=0; x<=t; ++x)
 764                 {
 765                         index_data[i++] = (y+1)*(t+1)+x;
 766                         index_data[i++] = y*(t+1)+x;
 767                 }
 768         }
 769         glBindBuffer(GL_ELEMENT_ARRAY_BUFFER, monitor->index_buffer);
 770         glBufferData(GL_ELEMENT_ARRAY_BUFFER, data_size, index_data, GL_STATIC_DRAW);
 771         glBindBuffer(GL_ELEMENT_ARRAY_BUFFER, 0);
 772         free(index_data);
 773 }
 774
 775 int initialize_monitor(Compositor *compositor, CompositedScreen *screen, XRRScreenResources *xrr_res, unsigned index)
 776 {
 777         CompositedMonitor *monitor;
 778         XRROutputInfo *output;
 779         int namelen;
 780         XRRCrtcInfo *crtc;
 781
 782         monitor = &screen->monitors[index];
 783
 784         output = XRRGetOutputInfo(compositor->display, xrr_res, xrr_res->outputs[index]);
 785         namelen = strlen(output->name);
 786         monitor->name = (char *)malloc(namelen+1);
 787         strcpy(monitor->name, output->name);
 788         monitor->enabled = !!output->crtc;
 789         if(!monitor->enabled)
 790         {
 791                 XRRFreeOutputInfo(output);
 792                 monitor->vertex_buffer = 0;
 793                 monitor->index_buffer = 0;
 794                 monitor->vertex_array = 0;
 795                 monitor->geometry_data = NULL;
 796                 return 1;
 797         }
 798
 799         crtc = XRRGetCrtcInfo(compositor->display, xrr_res, output->crtc);
 800         monitor->x = crtc->x;
 801         monitor->y = crtc->y;
 802         monitor->width = crtc->width;
 803         monitor->height = crtc->height;
 804         XRRFreeCrtcInfo(crtc);
 805         XRRFreeOutputInfo(output);
 806
 807         monitor->keystone_vertical = 0.0f;
 808         monitor->curvature_type = CYLINDRICAL;
 809         monitor->curvature_depth = 0.0f;
 810         monitor->vertical_center = 0.5f;
 811         monitor->perspective = 1.0f;
 812
 813         create_geometry_correction(monitor);
 814         update_monitor_vertices(screen, monitor);
 815
 816         return 1;
 817 }
 818
 819 CompositedMonitor *find_monitor_by_name(CompositedScreen *screen, char *name)
 820 {
 821         unsigned i;
 822
 823         for(i=0; i<screen->nmonitors; ++i)
 824                 if(!strcmp(screen->monitors[i].name, name))
 825                         return &screen->monitors[i];
 826
 827         return NULL;
 828 }
 829
 830 CompositedMonitor *find_monitor_by_name_global(Compositor *compositor, char *name, CompositedScreen **screen)
 831 {
 832         unsigned i, j;
 833
 834         for(i=0; i<compositor->nscreens; ++i)
 835                 for(j=0; j<compositor->screens[i].nmonitors; ++j)
 836                         if(!strcmp(compositor->screens[i].monitors[j].name, name))
 837                         {
 838                                 if(screen)
 839                                         *screen = &compositor->screens[i];
 840                                 return &compositor->screens[i].monitors[j];
 841                         }
 842
 843         if(screen)
 844                 *screen = NULL;
 845         return NULL;
 846 }
 847
 848 void update_monitors(Compositor *compositor, CompositedScreen *screen)
 849 {
 850         XRRScreenResources *xrr_res;
 851         int i;
 852
 853         xrr_res = XRRGetScreenResources(compositor->display, screen->root);
 854         for(i=0; i<xrr_res->noutput; ++i)
 855         {
 856                 XRROutputInfo *output;
 857                 CompositedMonitor *monitor;
 858                 int was_enabled;
 859                 XRRCrtcInfo *crtc;
 860
 861                 output = XRRGetOutputInfo(compositor->display, xrr_res, xrr_res->outputs[i]);
 862                 monitor = find_monitor_by_name(screen, output->name);
 863                 if(!monitor)
 864                         continue;
 865
 866                 was_enabled = monitor->enabled;
 867                 monitor->enabled = !!output->crtc;
 868                 if(!monitor->enabled)
 869                 {
 870                         XRRFreeOutputInfo(output);
 871                         if(was_enabled)
 872                                 free_geometry_correction(monitor);
 873                         continue;
 874                 }
 875
 876                 crtc = XRRGetCrtcInfo(compositor->display, xrr_res, output->crtc);
 877                 monitor->x = crtc->x;
 878                 monitor->y = crtc->y;
 879                 monitor->width = crtc->width;
 880                 monitor->height = crtc->height;
 881                 XRRFreeCrtcInfo(crtc);
 882                 XRRFreeOutputInfo(output);
 883
 884                 if(!was_enabled)
 885                         create_geometry_correction(monitor);
 886
 887                 update_monitor_vertices(screen, monitor);
 888         }
 889         XRRFreeScreenResources(xrr_res);
 890 }
 891
 892 void update_geometry_correction(Compositor *compositor, CompositedScreen *screen)
 893 {
 894         Atom prop_type;
 895         int prop_format;
 896         unsigned long overflow;
 897         unsigned long names_length;
 898         char *names;
 899         unsigned long values_length;
 900         short *values;
 901         char *name_ptr;
 902         unsigned i;
 903
 904         XGetWindowProperty(compositor->display, screen->root, compositor->monitors_atom, 0, 64, False, XA_STRING,
 905                 &prop_type, &prop_format, &names_length, &overflow, (unsigned char **)&names);
 906         if(prop_type!=XA_STRING || prop_format!=8)
 907                 return;
 908
 909         XGetWindowProperty(compositor->display, screen->root, compositor->correction_atom, 0, 64, False, XA_INTEGER,
 910                 &prop_type, &prop_format, &values_length, &overflow, (unsigned char **)&values);
 911         if(prop_type!=XA_INTEGER || prop_format!=16)
 912         {
 913                 XFree(names);
 914                 return;
 915         }
 916
 917         use_gl(compositor, screen);
 918
 919         name_ptr = names;
 920         for(i=0; i*5+4<values_length; ++i)
 921         {
 922                 CompositedMonitor *monitor;
 923
 924                 if(name_ptr>=names+names_length)
 925                         break;
 926
 927                 monitor = find_monitor_by_name(screen, name_ptr);
 928                 if(monitor)
 929                 {
 930                         monitor->keystone_vertical = values[i*5]/4096.0f;
 931                         monitor->curvature_type = values[i*5+1];
 932                         monitor->curvature_depth = values[i*5+2]/4096.0f;
 933                         monitor->vertical_center = values[i*5+3]/4096.0f;
 934                         monitor->perspective = values[i*5+4]/4096.0f;
 935
 936                         if(monitor->enabled)
 937                                 update_monitor_vertices(screen, monitor);
 938                 }
 939
 940                 name_ptr += strlen(name_ptr)+1;
 941         }
 942
 943         XFree(names);
 944         XFree(values);
 945 }
 946
 947 void update_root_pixmap(Compositor *compositor, CompositedScreen *screen)
 948 {
 949         Atom prop_type;
 950         int prop_format;
 951         unsigned long overflow;
 952         unsigned long length;
 953         long *pixmap;
 954         Window root;
 955         int x, y;
 956         unsigned width;
 957         unsigned height;
 958         unsigned border;
 959         unsigned depth;
 960
 961         use_gl(compositor, screen);
 962
 963         if(screen->root_glx_pixmap)
 964         {
 965                 glXDestroyPixmap(compositor->display, screen->root_glx_pixmap);
 966                 screen->root_glx_pixmap = None;
 967         }
 968
 969         XGetWindowProperty(compositor->display, screen->root, compositor->root_pmap_atom, 0, 1, False, XA_PIXMAP,
 970                 &prop_type, &prop_format, &length, &overflow, (unsigned char **)&pixmap);
 971         if(prop_type!=XA_PIXMAP || prop_format!=32)
 972         {
 973                 screen->root_pixmap = None;
 974                 return;
 975         }
 976
 977         screen->root_pixmap = pixmap[0];
 978         if(XGetGeometry(compositor->display, screen->root_pixmap, &root, &x, &y, &width, &height, &border, &depth))
 979                 screen->root_glx_pixmap = pixmap_to_glx_pixmap(compositor, screen, screen->root_pixmap);
 980         else
 981                 screen->root_pixmap = None;
 982
 983         XFree(pixmap);
 984 }
 985
 986 int initialize_screen(Compositor *compositor, unsigned number)
 987 {
 988         CompositedScreen *screen;
 989         const char *extensions;
 990         Window dummy_root;
 991         int x, y;
 992         unsigned border;
 993         unsigned depth;
 994         XRRScreenResources *xrr_res;
 995         Window dummy_parent;
 996         Window *children;
 997         unsigned nchildren;
 998         unsigned i;
 999
1000         screen = &compositor->screens[number];
1001         screen->number = number;
1002
1003         extensions = glXQueryExtensionsString(compositor->display, screen->number);
1004         if(!strstr(extensions, "GLX_ARB_create_context"))
1005                 return with_error("GLX_ARB_create_context is required");
1006         if(!strstr(extensions, "GLX_EXT_texture_from_pixmap"))
1007                 return with_error("GLX_EXT_texture_from_pixmap is required");
1008
1009         screen->root = RootWindow(compositor->display, screen->number);
1010         XSelectInput(compositor->display, screen->root, SubstructureNotifyMask|PropertyChangeMask);
1011         screen->overlay = XCompositeGetOverlayWindow(compositor->display, screen->root);
1012         XSelectInput(compositor->display, screen->overlay, StructureNotifyMask);
1013         XGetGeometry(compositor->display, screen->overlay, &dummy_root, &x, &y, &screen->width, &screen->height, &border, &depth);
1014         XShapeCombineRectangles(compositor->display, screen->overlay, ShapeInput, 0, 0, NULL, 0, ShapeSet, Unsorted);
1015
1016         if(!initialize_gl(compositor, screen))
1017                 return 0;
1018
1019         use_gl(compositor, screen);
1020
1021         if(!create_gl_resources(screen))
1022                 return 0;
1023
1024         xrr_res = XRRGetScreenResources(compositor->display, screen->root);
1025         screen->nmonitors = xrr_res->noutput;
1026         screen->monitors = (CompositedMonitor *)malloc(screen->nmonitors*sizeof(CompositedMonitor));
1027         for(i=0; i<screen->nmonitors; ++i)
1028                 if(!initialize_monitor(compositor, screen, xrr_res, i))
1029                         return 0;
1030         XRRFreeScreenResources(xrr_res);
1031         XRRSelectInput(compositor->display, screen->root, RRScreenChangeNotifyMask);
1032
1033         screen->root_pixmap = None;
1034         screen->root_glx_pixmap = None;
1035
1036         update_geometry_correction(compositor, screen);
1037         update_root_pixmap(compositor, screen);
1038
1039         XQueryTree(compositor->display, screen->root, &dummy_root, &dummy_parent, &children, &nchildren);
1040
1041         screen->windows = (CompositedWindow *)malloc(nchildren*sizeof(CompositedWindow));
1042         screen->nwindows = 0;
1043         screen->windows_capacity = nchildren;
1044
1045         for(i=0; i<nchildren; ++i)
1046                 add_window(compositor, screen, children[i]);
1047
1048         XFree(children);
1049
1050         screen->dirty = 1;
1051
1052         return 1;
1053 }
1054
1055 int initialize_compositor(Compositor *compositor)
1056 {
1057         int event_base;
1058         int error_base;
1059         int major_ver;
1060         int minor_ver;
1061         unsigned i;
1062
1063         compositor->display = XOpenDisplay(NULL);
1064         if(!compositor->display)
1065                 return with_error("Could not open X display");
1066
1067         XSetErrorHandler(&x_error_handler);
1068
1069         if(!XCompositeQueryExtension(compositor->display, &event_base, &error_base))
1070                 return with_error("XComposite is required but was not found");
1071         else if(!XCompositeQueryVersion(compositor->display, &major_ver, &minor_ver))
1072                 return with_error("Cannot determine XComposite version");
1073         else if(major_ver==0 && minor_ver<3)
1074                 return with_error("XComposite 0.3 or later is required");
1075
1076         if(!glXQueryExtension(compositor->display, &event_base, &error_base))
1077                 return with_error("GLX is required but was not found");
1078         else if(!glXQueryVersion(compositor->display, &major_ver, &minor_ver))
1079                 return with_error("Cannot determine GLX version");
1080         else if(major_ver<1 || (major_ver==1 && minor_ver<4))
1081                 return with_error("GLX 1.4 or later is required");
1082
1083         if(!XDamageQueryExtension(compositor->display, &compositor->damage_event, &error_base))
1084                 return with_error("XDamage is required but was not found");
1085         else if(!XDamageQueryVersion(compositor->display, &major_ver, &minor_ver))
1086                 return with_error("Cannot determine XDamage version");
1087         else if(major_ver<1)
1088                 return with_error("XDamage 1.0 or later is required");
1089
1090         if(!XShapeQueryExtension(compositor->display, &compositor->shape_event, &error_base))
1091                 return with_error("XShape is required but was not found");
1092         else if(!XShapeQueryVersion(compositor->display, &major_ver, &minor_ver))
1093                 return with_error("Cannot determine XShape version");
1094         else if(major_ver<1 || (major_ver==1 && minor_ver<1))
1095                 return with_error("XShape 1.1 or later is required");
1096
1097         if(!XRRQueryExtension(compositor->display, &compositor->randr_event, &error_base))
1098                 return with_error("XRandR is required but was not found");
1099         else if(!XRRQueryVersion(compositor->display, &major_ver, &minor_ver))
1100                 return with_error("Cannot determine XRandR version");
1101         else if(major_ver<1 || (major_ver==1 && minor_ver<2))
1102                 return with_error("XRandR 1.2 or later is required");
1103
1104         compositor->glXCreateContextAttribsARB = (PFNGLXCREATECONTEXTATTRIBSARBPROC)glXGetProcAddress((unsigned char *)"glXCreateContextAttribsARB");
1105         compositor->glXBindTexImageEXT = (PFNGLXBINDTEXIMAGEEXTPROC)glXGetProcAddress((unsigned char *)"glXBindTexImageEXT");
1106         compositor->glXReleaseTexImageEXT = (PFNGLXRELEASETEXIMAGEEXTPROC)glXGetProcAddress((unsigned char *)"glXReleaseTexImageEXT");
1107
1108         compositor->root_pmap_atom = XInternAtom(compositor->display, "_XROOTPMAP_ID", False);
1109         compositor->correction_atom = XInternAtom(compositor->display, "_MSP_GEOMETRY_CORRECTION", False);
1110         compositor->monitors_atom = XInternAtom(compositor->display, "_MSP_GEOMETRY_CORRECTION_MONITORS", False);
1111         compositor->geometry_data_atom = XInternAtom(compositor->display, "_MSP_GEOMETRY_CORRECTION_DATA", False);
1112
1113         compositor->nscreens = ScreenCount(compositor->display);
1114         compositor->screens = (CompositedScreen *)malloc(compositor->nscreens*sizeof(CompositedScreen));
1115         for(i=0; i<compositor->nscreens; ++i)
1116                 if(!initialize_screen(compositor, i))
1117                         return 0;
1118
1119         compositor->selection_window = XCreateWindow(compositor->display, compositor->screens[0].root, 0, 0, 1, 1, 0, CopyFromParent, InputOutput, CopyFromParent, 0, NULL);
1120         XSetSelectionOwner(compositor->display, compositor->geometry_data_atom, compositor->selection_window, CurrentTime);
1121
1122         compositor->dirty = 1;
1123
1124         return 1;
1125 }
1126
1127 void shutdown_screen(Compositor *compositor, CompositedScreen *screen)
1128 {
1129         unsigned i;
1130
1131         use_gl(compositor, screen);
1132
1133         for(i=0; i<screen->nwindows; ++i)
1134         {
1135                 glDeleteTextures(1, &screen->windows[i].texture);
1136                 glDeleteTextures(1, &screen->windows[i].mask_texture);
1137                 if(screen->windows[i].pixmap)
1138                 {
1139                         glXDestroyPixmap(compositor->display, screen->windows[i].glx_pixmap);
1140                         XFreePixmap(compositor->display, screen->windows[i].pixmap);
1141                         XDamageDestroy(compositor->display, screen->windows[i].damage);
1142                 }
1143         }
1144
1145         for(i=0; i<screen->nmonitors; ++i)
1146         {
1147                 free(screen->monitors[i].name);
1148                 if(screen->monitors[i].enabled)
1149                         free_geometry_correction(&screen->monitors[i]);
1150         }
1151
1152         glDeleteTextures(1, &screen->root_texture);
1153         glXDestroyPixmap(compositor->display, screen->root_glx_pixmap);
1154
1155         glDeleteBuffers(1, &screen->window_vertex_buffer);
1156         glDeleteVertexArrays(1, &screen->window_vertex_array);
1157         glDeleteFramebuffers(1, &screen->framebuffer);
1158         glDeleteTextures(1, &screen->fb_texture);
1159         glDeleteProgram(screen->program);
1160         glDeleteProgram(screen->masked_program);
1161         for(i=0; i<3; ++i)
1162                 glDeleteShader(screen->shaders[i]);
1163
1164         glXMakeContextCurrent(compositor->display, None, None, NULL);
1165         glXDestroyContext(compositor->display, screen->glx_context);
1166         glXDestroyWindow(compositor->display, screen->glx_window);
1167         XDestroyWindow(compositor->display, screen->render_window);
1168
1169         XCompositeReleaseOverlayWindow(compositor->display, screen->overlay);
1170
1171         free(screen->windows);
1172         free(screen->monitors);
1173 }
1174
1175 void shutdown_compositor(Compositor *compositor)
1176 {
1177         unsigned i;
1178
1179         for(i=0; i<compositor->nscreens; ++i)
1180                 shutdown_screen(compositor, &compositor->screens[i]);
1181         free(compositor->screens);
1182
1183         XDestroyWindow(compositor->display, compositor->selection_window);
1184
1185         XCloseDisplay(compositor->display);
1186 }
1187
1188 void mark_dirty(Compositor *compositor, CompositedScreen *screen)
1189 {
1190         compositor->dirty = 1;
1191         screen->dirty = 1;
1192 }
1193
1194 void process_create_window_event(Compositor *compositor, XCreateWindowEvent *event)
1195 {
1196         CompositedScreen *screen;
1197
1198         if((screen = find_screen_by_root(compositor, event->parent)))
1199                 add_window(compositor, screen, event->window);
1200 }
1201
1202 void process_destroy_window_event(Compositor *compositor, XDestroyWindowEvent *event)
1203 {
1204         CompositedScreen *screen;
1205         CompositedWindow *window;
1206
1207         if((screen = find_screen_by_root(compositor, event->event)))
1208                 if((window = find_window(screen, event->window)))
1209                 {
1210                         use_gl(compositor, screen);
1211                         remove_window(compositor, screen, window, 1);
1212                 }
1213 }
1214
1215 void process_map_event(Compositor *compositor, XMapEvent *event)
1216 {
1217         CompositedScreen *screen;
1218         CompositedWindow *window;
1219
1220         screen = find_screen_by_root(compositor, event->event);
1221         if(!screen)
1222                 return;
1223
1224         window = find_window(screen, event->window);
1225         if(!window)
1226                 return;
1227
1228         window->map_state = IsViewable;
1229         window->recreate_pixmap = 1;
1230         window->recreate_mask = 1;
1231
1232         mark_dirty(compositor, screen);
1233 }
1234
1235 void process_unmap_event(Compositor *compositor, XUnmapEvent *event)
1236 {
1237         CompositedScreen *screen;
1238         CompositedWindow *window;
1239
1240         screen = find_screen_by_root(compositor, event->event);
1241         if(!screen)
1242                 return;
1243
1244         window = find_window(screen, event->window);
1245         if(window)
1246                 window->map_state = IsUnviewable;
1247
1248         mark_dirty(compositor, screen);
1249 }
1250
1251 void process_reparent_event(Compositor *compositor, XReparentEvent *event)
1252 {
1253         CompositedScreen *screen;
1254         CompositedWindow *window;
1255
1256         screen = find_screen_by_root(compositor, event->event);
1257         if(!screen)
1258                 return;
1259
1260         if(event->parent==screen->root)
1261                 window = add_window(compositor, screen, event->window);
1262         else
1263         {
1264                 window = find_window(screen, event->window);
1265                 if(!window)
1266                         return;
1267
1268                 remove_window(compositor, screen, window, 0);
1269         }
1270
1271         if(window && window->map_state==IsViewable)
1272                 mark_dirty(compositor, screen);
1273 }
1274
1275 void process_configure_event(Compositor *compositor, XConfigureEvent *event)
1276 {
1277         CompositedScreen *screen;
1278         CompositedWindow *window;
1279
1280         screen = find_screen_by_root(compositor, event->event);
1281         if(!screen)
1282                 return;
1283
1284         if(event->window==screen->overlay)
1285         {
1286                 screen->width = event->width;
1287                 screen->height = event->height;
1288
1289                 XResizeWindow(compositor->display, screen->render_window, screen->width, screen->height);
1290                 use_gl(compositor, screen);
1291                 glBindTexture(GL_TEXTURE_2D, screen->fb_texture);
1292                 glTexImage2D(GL_TEXTURE_2D, 0, GL_RGB8, screen->width, screen->height, 0, GL_RGB, GL_UNSIGNED_BYTE, NULL);
1293                 glBindTexture(GL_TEXTURE_2D, 0);
1294                 return;
1295         }
1296
1297         window = find_window(screen, event->window);
1298         if(!window)
1299                 return;
1300
1301         window->x = event->x;
1302         window->y = event->y;
1303         if((unsigned)event->width!=window->width || (unsigned)event->height!=window->height || (unsigned)event->border_width!=window->border)
1304         {
1305                 window->width = event->width;
1306                 window->height = event->height;
1307                 window->border = event->border_width;
1308                 window->recreate_pixmap = 1;
1309         }
1310         reorder_window(screen, window, event->above);
1311
1312         if(window->map_state==IsViewable)
1313                 mark_dirty(compositor, screen);
1314 }
1315
1316 void process_property_event(Compositor *compositor, XPropertyEvent *event)
1317 {
1318         CompositedScreen *screen;
1319
1320         screen = find_screen_by_root(compositor, event->window);
1321         if(!screen)
1322                 return;
1323
1324         if(event->atom==compositor->correction_atom)
1325                 update_geometry_correction(compositor, screen);
1326         else if(event->atom==compositor->root_pmap_atom)
1327                 update_root_pixmap(compositor, screen);
1328 }
1329
1330 void process_selection_request_event(Compositor *compositor, XSelectionRequestEvent *event)
1331 {
1332         Atom prop_type;
1333         int prop_format;
1334         unsigned long overflow;
1335         unsigned long names_length;
1336         XSelectionEvent notify;
1337
1338         if(event->selection==compositor->geometry_data_atom)
1339         {
1340                 char *monitor_name;
1341                 CompositedMonitor *monitor;
1342
1343                 XGetWindowProperty(compositor->display, event->requestor, event->property, 0, 64, False, XA_STRING,
1344                         &prop_type, &prop_format, &names_length, &overflow, (unsigned char **)&monitor_name);
1345                 if(prop_type!=XA_STRING || prop_format!=8)
1346                         return;
1347
1348                 monitor = find_monitor_by_name_global(compositor, monitor_name, NULL);
1349                 if(monitor && monitor->enabled)
1350                         XChangeProperty(compositor->display, event->requestor, event->property, XA_INTEGER, 16, PropModeReplace, (unsigned char *)monitor->geometry_data, monitor->geometry_data_size);
1351
1352                 notify.type = SelectionNotify;
1353                 notify.requestor = event->requestor;
1354                 notify.selection = event->selection;
1355                 notify.target = event->target;
1356                 notify.property = (monitor ? event->property : None);
1357                 notify.time = event->time;
1358                 XSendEvent(compositor->display, event->requestor, False, 0, (XEvent *)&notify);
1359         }
1360 }
1361
1362 void process_damage_event(Compositor *compositor, XDamageNotifyEvent *event)
1363 {
1364         CompositedScreen *screen;
1365         CompositedWindow *window;
1366
1367         window = find_window_global(compositor, event->drawable, &screen);
1368         if(window && window->map_state==IsViewable)
1369                 mark_dirty(compositor, screen);
1370 }
1371
1372 void process_shape_event(Compositor *compositor, XShapeEvent *event)
1373 {
1374         CompositedScreen *screen;
1375         CompositedWindow *window;
1376
1377         if(event->kind!=ShapeBounding)
1378                 return;
1379
1380         window = find_window_global(compositor, event->window, &screen);
1381         if(window && window->map_state==IsViewable)
1382         {
1383                 window->recreate_mask = 1;
1384                 mark_dirty(compositor, screen);
1385         }
1386 }
1387
1388 void process_randr_event(Compositor *compositor, XRRScreenChangeNotifyEvent *event)
1389 {
1390         CompositedScreen *screen;
1391
1392         if((screen = find_screen_by_root(compositor, event->root)))
1393                 update_monitors(compositor, screen);
1394 }
1395
1396 void process_events(Compositor *compositor)
1397 {
1398         int pending;
1399         XEvent event;
1400
1401         pending = 0;
1402         while((pending || !compositor->dirty) && !terminate_requested)
1403         {
1404                 if(!pending)
1405                         pending = XPending(compositor->display);
1406
1407                 XNextEvent(compositor->display, &event);
1408                 if(pending)
1409                         --pending;
1410
1411                 switch(event.type)
1412                 {
1413                 case CreateNotify:
1414                         process_create_window_event(compositor, &event.xcreatewindow);
1415                         break;
1416                 case DestroyNotify:
1417                         process_destroy_window_event(compositor, &event.xdestroywindow);
1418                         break;
1419                 case MapNotify:
1420                         process_map_event(compositor, &event.xmap);
1421                         break;
1422                 case UnmapNotify:
1423                         process_unmap_event(compositor, &event.xunmap);
1424                         break;
1425                 case ReparentNotify:
1426                         process_reparent_event(compositor, &event.xreparent);
1427                         break;
1428                 case ConfigureNotify:
1429                         process_configure_event(compositor, &event.xconfigure);
1430                         break;
1431                 case PropertyNotify:
1432                         process_property_event(compositor, &event.xproperty);
1433                         break;
1434                 case SelectionRequest:
1435                         process_selection_request_event(compositor, &event.xselectionrequest);
1436                         break;
1437                 default:
1438                         if(event.type==compositor->damage_event+XDamageNotify)
1439                                 process_damage_event(compositor, (XDamageNotifyEvent *)&event);
1440                         else if(event.type==compositor->shape_event+ShapeNotify)
1441                                 process_shape_event(compositor, (XShapeEvent *)&event);
1442                         else if(event.type==compositor->randr_event+RRScreenChangeNotify)
1443                                 process_randr_event(compositor, (XRRScreenChangeNotifyEvent *)&event);
1444                 }
1445         }
1446 }
1447
1448 void refresh_screen(Compositor *compositor, CompositedScreen *screen)
1449 {
1450         unsigned i;
1451         int use_mask;
1452
1453         for(i=0; i<screen->nwindows; ++i)
1454                 if(screen->windows[i].map_state==IsViewable)
1455                         XDamageSubtract(compositor->display, screen->windows[i].damage, None, None);
1456         glXWaitX();
1457
1458         use_gl(compositor, screen);
1459
1460         glViewport(0, 0, screen->width, screen->height);
1461         glClearColor(0.0f, 0.0f, 0.0f, 0.0f);
1462         glBindFramebuffer(GL_DRAW_FRAMEBUFFER, screen->framebuffer);
1463
1464         use_mask = -1;
1465         glBindVertexArray(screen->window_vertex_array);
1466
1467         if(screen->root_pixmap)
1468         {
1469                 use_mask = 0;
1470                 glUseProgram(screen->program);
1471                 glBindTexture(GL_TEXTURE_2D, screen->root_texture);
1472                 compositor->glXBindTexImageEXT(compositor->display, screen->root_glx_pixmap, GLX_FRONT_LEFT_EXT, NULL);
1473                 glUniform4f(screen->geometry_loc, 0.0f, 0.0f, 1.0f, 1.0f);
1474                 glDrawArrays(GL_TRIANGLE_STRIP, 0, 4);
1475                 compositor->glXReleaseTexImageEXT(compositor->display, screen->root_glx_pixmap, GLX_FRONT_LEFT_EXT);
1476         }
1477         else
1478                 glClear(GL_COLOR_BUFFER_BIT);
1479
1480         for(i=0; i<screen->nwindows; ++i)
1481         {
1482                 CompositedWindow *window;
1483
1484                 window = &screen->windows[i];
1485                 if(window->map_state!=IsViewable)
1486                         continue;
1487
1488                 if(window->use_mask!=use_mask)
1489                 {
1490                         use_mask = window->use_mask;
1491                         glUseProgram(use_mask ? screen->masked_program : screen->program);
1492                 }
1493
1494                 if(window->use_mask)
1495                 {
1496                         glActiveTexture(GL_TEXTURE1);
1497                         glBindTexture(GL_TEXTURE_2D, window->mask_texture);
1498                         glActiveTexture(GL_TEXTURE0);
1499                 }
1500
1501                 if(window->recreate_pixmap)
1502                         create_window_pixmap(compositor, screen, window);
1503                 if(window->recreate_mask)
1504                         update_window_mask(compositor, window);
1505
1506                 glBindTexture(GL_TEXTURE_2D, window->texture);
1507                 compositor->glXBindTexImageEXT(compositor->display, window->glx_pixmap, GLX_FRONT_LEFT_EXT, NULL);
1508                 glUniform4f((use_mask ? screen->masked_geometry_loc : screen->geometry_loc),
1509                         (float)window->x/screen->width, 1.0f-(float)(window->y+window->height)/screen->height,
1510                         (float)(window->width+2*window->border)/screen->width, (float)(window->height+2*window->border)/screen->height);
1511                 glDrawArrays(GL_TRIANGLE_STRIP, 0, 4);
1512                 compositor->glXReleaseTexImageEXT(compositor->display, window->glx_pixmap, GLX_FRONT_LEFT_EXT);
1513         }
1514
1515         glBindFramebuffer(GL_DRAW_FRAMEBUFFER, 0);
1516         glClear(GL_COLOR_BUFFER_BIT);
1517         glBindTexture(GL_TEXTURE_2D, screen->fb_texture);
1518         glEnable(GL_PRIMITIVE_RESTART);
1519         glPrimitiveRestartIndex(0xFFFF);
1520         if(use_mask)
1521                 glUseProgram(screen->program);
1522
1523         for(i=0; i<screen->nmonitors; ++i)
1524         {
1525                 CompositedMonitor *monitor = &screen->monitors[i];
1526                 if(!monitor->enabled)
1527                         continue;
1528
1529                 glUniform4f(screen->geometry_loc,
1530                         (float)monitor->x/screen->width, 1.0f-(float)(monitor->y+monitor->height)/screen->height,
1531                         (float)monitor->width/screen->width, (float)monitor->height/screen->height);
1532
1533                 glBindVertexArray(monitor->vertex_array);
1534                 glDrawElements(GL_TRIANGLE_STRIP, monitor->nelements, GL_UNSIGNED_SHORT, NULL);
1535         }
1536
1537         glBindVertexArray(0);
1538
1539         glXSwapBuffers(compositor->display, screen->glx_window);
1540
1541         screen->dirty = 0;
1542 }
1543
1544 void refresh_all_screens(Compositor *compositor)
1545 {
1546         unsigned i;
1547
1548         for(i=0; i<compositor->nscreens; ++i)
1549         {
1550                 CompositedScreen *screen = &compositor->screens[i];
1551                 if(screen->dirty)
1552                         refresh_screen(compositor, screen);
1553         }
1554
1555         compositor->dirty = 0;
1556 }
1557
1558 void sighandler(int sig)
1559 {
1560         terminate_requested = 1;
1561         (void)sig;
1562 }
1563
1564 int main()
1565 {
1566         Compositor compositor;
1567
1568         signal(SIGINT, &sighandler);
1569         signal(SIGTERM, &sighandler);
1570
1571         if(!initialize_compositor(&compositor))
1572                 return 1;
1573
1574         while(!terminate_requested)
1575         {
1576                 process_events(&compositor);
1577                 if(compositor.dirty && !terminate_requested)
1578                         refresh_all_screens(&compositor);
1579         }
1580
1581         shutdown_compositor(&compositor);
1582
1583         return 0;
1584 }