]> git.sur5r.net Git - i3/i3lock/blob - xcb.c
handle screen resolution images and redraw the lock window
[i3/i3lock] / xcb.c
1 /*
2  * vim:ts=4:sw=4:expandtab
3  *
4  * © 2010-2011 Michael Stapelberg
5  *
6  * xcb.c: contains all functions which use XCB to talk to X11. Mostly wrappers
7  *        around the rather complicated/ugly parts of the XCB API.
8  *
9  */
10 #include <xcb/xcb.h>
11 #include <xcb/xcb_keysyms.h>
12 #include <xcb/xcb_image.h>
13 #include <xcb/dpms.h>
14 #include <stdio.h>
15 #include <stdlib.h>
16 #include <stdbool.h>
17 #include <unistd.h>
18 #include <assert.h>
19 #include <err.h>
20
21 #include "cursors.h"
22
23 #define curs_invisible_width 8
24 #define curs_invisible_height 8
25
26 static unsigned char curs_invisible_bits[] = {
27  0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 };
28
29 #define curs_windows_width 11
30 #define curs_windows_height 19
31
32 static unsigned char curs_windows_bits[] = {
33  0xfe, 0x07, 0xfc, 0x07, 0xfa, 0x07, 0xf6, 0x07, 0xee, 0x07, 0xde, 0x07,
34  0xbe, 0x07, 0x7e, 0x07, 0xfe, 0x06, 0xfe, 0x05, 0x3e, 0x00, 0xb6, 0x07,
35  0x6a, 0x07, 0x6c, 0x07, 0xde, 0x06, 0xdf, 0x06, 0xbf, 0x05, 0xbf, 0x05,
36  0x7f, 0x06 };
37
38 #define mask_windows_width 11
39 #define mask_windows_height 19
40
41 static unsigned char mask_windows_bits[] = {
42  0x01, 0x00, 0x03, 0x00, 0x07, 0x00, 0x0f, 0x00, 0x1f, 0x00, 0x3f, 0x00,
43  0x7f, 0x00, 0xff, 0x00, 0xff, 0x01, 0xff, 0x03, 0xff, 0x07, 0x7f, 0x00,
44  0xf7, 0x00, 0xf3, 0x00, 0xe1, 0x01, 0xe0, 0x01, 0xc0, 0x03, 0xc0, 0x03,
45  0x80, 0x01 };
46
47 static uint32_t get_colorpixel(char *hex) {
48     char strgroups[3][3] = {{hex[0], hex[1], '\0'},
49                             {hex[2], hex[3], '\0'},
50                             {hex[4], hex[5], '\0'}};
51     uint32_t rgb16[3] = {(strtol(strgroups[0], NULL, 16)),
52                          (strtol(strgroups[1], NULL, 16)),
53                          (strtol(strgroups[2], NULL, 16))};
54
55     return (rgb16[0] << 16) + (rgb16[1] << 8) + rgb16[2];
56 }
57
58 xcb_visualtype_t *get_root_visual_type(xcb_screen_t *screen) {
59     xcb_visualtype_t *visual_type = NULL;
60     xcb_depth_iterator_t depth_iter;
61     xcb_visualtype_iterator_t visual_iter;
62
63     for (depth_iter = xcb_screen_allowed_depths_iterator(screen);
64          depth_iter.rem;
65          xcb_depth_next(&depth_iter)) {
66
67         for (visual_iter = xcb_depth_visuals_iterator(depth_iter.data);
68              visual_iter.rem;
69              xcb_visualtype_next(&visual_iter)) {
70             if (screen->root_visual != visual_iter.data->visual_id)
71                 continue;
72
73             visual_type = visual_iter.data;
74             return visual_type;
75         }
76     }
77
78     return NULL;
79 }
80
81 xcb_pixmap_t create_bg_pixmap(xcb_connection_t *conn, xcb_screen_t *scr, u_int32_t* resolution, char *color) {
82     xcb_pixmap_t bg_pixmap = xcb_generate_id(conn);
83     xcb_create_pixmap(conn, scr->root_depth, bg_pixmap, scr->root,
84                       resolution[0], resolution[1]);
85
86     /* Generate a Graphics Context and fill the pixmap with background color
87      * (for images that are smaller than your screen) */
88     xcb_gcontext_t gc = xcb_generate_id(conn);
89     uint32_t values[] = { get_colorpixel(color) };
90     xcb_create_gc(conn, gc, bg_pixmap, XCB_GC_FOREGROUND, values);
91     xcb_rectangle_t rect = { 0, 0, resolution[0], resolution[1] };
92     xcb_poly_fill_rectangle(conn, bg_pixmap, gc, 1, &rect);
93
94     return bg_pixmap;
95 }
96
97 xcb_window_t open_fullscreen_window(xcb_connection_t *conn, xcb_screen_t *scr, char *color, xcb_pixmap_t pixmap) {
98     uint32_t mask = 0;
99     uint32_t values[3];
100     xcb_window_t win = xcb_generate_id(conn);
101
102     if (pixmap == XCB_NONE) {
103         mask |= XCB_CW_BACK_PIXEL;
104         values[0] = get_colorpixel(color);
105     } else {
106         mask |= XCB_CW_BACK_PIXMAP;
107         values[0] = pixmap;
108     }
109
110     mask |= XCB_CW_OVERRIDE_REDIRECT;
111     values[1] = 1;
112
113     mask |= XCB_CW_EVENT_MASK;
114     values[2] = XCB_EVENT_MASK_EXPOSURE |
115                 XCB_EVENT_MASK_KEY_PRESS |
116                 XCB_EVENT_MASK_KEY_RELEASE |
117                 XCB_EVENT_MASK_VISIBILITY_CHANGE |
118                 XCB_EVENT_MASK_STRUCTURE_NOTIFY;
119
120     xcb_create_window(conn,
121                       24,
122                       win, /* the window id */
123                       scr->root, /* parent == root */
124                       0, 0,
125                       scr->width_in_pixels,
126                       scr->height_in_pixels, /* dimensions */
127                       0, /* border = 0, we draw our own */
128                       XCB_WINDOW_CLASS_INPUT_OUTPUT,
129                       XCB_WINDOW_CLASS_COPY_FROM_PARENT, /* copy visual from parent */
130                       mask,
131                       values);
132
133     /* Map the window (= make it visible) */
134     xcb_map_window(conn, win);
135
136     /* Raise window (put it on top) */
137     values[0] = XCB_STACK_MODE_ABOVE;
138     xcb_configure_window(conn, win, XCB_CONFIG_WINDOW_STACK_MODE, values);
139
140     return win;
141 }
142
143 /*
144  * Returns the mask for Mode_switch (to be used for looking up keysymbols by
145  * keycode).
146  *
147  */
148 uint32_t get_mod_mask(xcb_connection_t *conn, xcb_key_symbols_t *symbols, uint32_t keycode) {
149     xcb_get_modifier_mapping_reply_t *modmap_r;
150     xcb_keycode_t *modmap, kc;
151     xcb_keycode_t *modeswitchcodes = xcb_key_symbols_get_keycode(symbols, keycode);
152     if (modeswitchcodes == NULL)
153         return 0;
154
155     modmap_r = xcb_get_modifier_mapping_reply(conn, xcb_get_modifier_mapping(conn), NULL);
156     modmap = xcb_get_modifier_mapping_keycodes(modmap_r);
157
158     for (int i = 0; i < 8; i++)
159         for (int j = 0; j < modmap_r->keycodes_per_modifier; j++) {
160             kc = modmap[i * modmap_r->keycodes_per_modifier + j];
161             for (xcb_keycode_t *ktest = modeswitchcodes; *ktest; ktest++) {
162                 if (*ktest != kc)
163                     continue;
164
165                 free(modeswitchcodes);
166                 free(modmap_r);
167                 return (1 << i);
168             }
169         }
170
171     return 0;
172 }
173
174 void dpms_turn_off_screen(xcb_connection_t *conn) {
175     xcb_dpms_enable(conn);
176     xcb_dpms_force_level(conn, XCB_DPMS_DPMS_MODE_OFF);
177     xcb_flush(conn);
178 }
179
180 /*
181  * Repeatedly tries to grab pointer and keyboard (up to 1000 times).
182  *
183  */
184 void grab_pointer_and_keyboard(xcb_connection_t *conn, xcb_screen_t *screen, xcb_cursor_t cursor) {
185     xcb_grab_pointer_cookie_t pcookie;
186     xcb_grab_pointer_reply_t *preply;
187
188     xcb_grab_keyboard_cookie_t kcookie;
189     xcb_grab_keyboard_reply_t *kreply;
190
191     int tries = 10000;
192
193     while (tries-- > 0) {
194         pcookie = xcb_grab_pointer(
195             conn,
196             false,               /* get all pointer events specified by the following mask */
197             screen->root,        /* grab the root window */
198             XCB_NONE,            /* which events to let through */
199             XCB_GRAB_MODE_ASYNC, /* pointer events should continue as normal */
200             XCB_GRAB_MODE_ASYNC, /* keyboard mode */
201             XCB_NONE,            /* confine_to = in which window should the cursor stay */
202             cursor,              /* we change the cursor to whatever the user wanted */
203             XCB_CURRENT_TIME
204         );
205
206         if ((preply = xcb_grab_pointer_reply(conn, pcookie, NULL)) &&
207             preply->status == XCB_GRAB_STATUS_SUCCESS) {
208             free(preply);
209             break;
210         }
211
212         /* Make this quite a bit slower */
213         usleep(50);
214     }
215
216     while (tries-- > 0) {
217         kcookie = xcb_grab_keyboard(
218             conn,
219             true,                /* report events */
220             screen->root,        /* grab the root window */
221             XCB_CURRENT_TIME,
222             XCB_GRAB_MODE_ASYNC, /* process events as normal, do not require sync */
223             XCB_GRAB_MODE_ASYNC
224         );
225
226         if ((kreply = xcb_grab_keyboard_reply(conn, kcookie, NULL)) &&
227             kreply->status == XCB_GRAB_STATUS_SUCCESS) {
228             free(kreply);
229             break;
230         }
231
232         /* Make this quite a bit slower */
233         usleep(50);
234     }
235
236     if (tries <= 0)
237         errx(EXIT_FAILURE, "Cannot grab pointer/keyboard");
238 }
239
240 xcb_cursor_t create_cursor(xcb_connection_t *conn, xcb_screen_t *screen, xcb_window_t win, int choice) {
241     xcb_pixmap_t bitmap;
242     xcb_pixmap_t mask;
243     xcb_cursor_t cursor;
244
245     unsigned char* curs_bits;
246     unsigned char* mask_bits;
247     int curs_w, curs_h;
248
249     switch (choice) {
250         case CURS_NONE:
251             curs_bits = curs_invisible_bits;
252             mask_bits = curs_invisible_bits;
253             curs_w = curs_invisible_width;
254             curs_h = curs_invisible_height;
255             break;
256         case CURS_WIN:
257             curs_bits = curs_windows_bits;
258             mask_bits = mask_windows_bits;
259             curs_w = curs_windows_width;
260             curs_h = curs_windows_height;
261             break;
262         case CURS_DEFAULT:
263         default:
264             return XCB_NONE; /* XCB_NONE is xcb's way of saying "don't change the cursor" */
265     }
266
267     bitmap = xcb_create_pixmap_from_bitmap_data(conn,
268                                                 win,
269                                                 curs_bits,
270                                                 curs_w,
271                                                 curs_h,
272                                                 1,
273                                                 screen->white_pixel,
274                                                 screen->black_pixel,
275                                                 NULL);
276
277     mask = xcb_create_pixmap_from_bitmap_data(conn,
278                                               win,
279                                               mask_bits,
280                                               curs_w,
281                                               curs_h,
282                                               1,
283                                               screen->white_pixel,
284                                               screen->black_pixel,
285                                               NULL);
286
287     cursor = xcb_generate_id(conn);
288
289     xcb_create_cursor(conn,
290                       cursor,
291                       bitmap,
292                       mask,
293                       65535,65535,65535,
294                       0,0,0,
295                       0,0);
296
297     xcb_free_pixmap(conn, bitmap);
298     xcb_free_pixmap(conn, mask);
299
300     return cursor;
301 }