#include #include #include #include #include #include #include #include #include #include #include #include #include #include #include // Here bet dragons #define max(a,b) ((a) > (b) ? (a) : (b)) #define min(a,b) ((a) < (b) ? (a) : (b)) #define indexof(c,s) (strchr((s),(c))-(s)) typedef struct font_t { xcb_font_t ptr; xcb_charinfo_t *width_lut; XftFont *xft_ft; int ascent; int descent, height; uint16_t char_max; uint16_t char_min; } font_t; typedef struct monitor_t { int x, width; xcb_window_t window; xcb_pixmap_t pixmap; struct monitor_t *prev, *next; } monitor_t; typedef struct area_t { int begin, end, align, button; xcb_window_t window; char *cmd; } area_t; #define N 20 typedef struct area_stack_t { int pos; area_t slot[N]; } area_stack_t; enum { ATTR_OVERL = (1<<0), ATTR_UNDERL = (1<<1), }; enum { ALIGN_L = 0, ALIGN_C, ALIGN_R }; enum { GC_DRAW = 0, GC_CLEAR, GC_ATTR, GC_MAX }; #define MAX_FONT_COUNT 5 /* 0 <= FONT_CACHE_SIZE <= 65536 */ #define FONT_CACHE_SIZE 256 static Display *dpy; static xcb_connection_t *c; static xcb_screen_t *scr; static int scr_nbr = 0; static xcb_gcontext_t gc[GC_MAX]; static xcb_visualid_t visual; static Visual *visual_ptr; static xcb_colormap_t colormap; static monitor_t *monhead, *montail; static font_t *font_list[MAX_FONT_COUNT]; static char *font_names[MAX_FONT_COUNT]; static font_t *font_cache[FONT_CACHE_SIZE]; static uint32_t attrs = 0; static bool dock = false; static bool topbar = true; static int bw = -1, bh = -1, bx = 0, by = 0; static int bu = 1; /* Underline height */ static uint32_t fgc, bgc, ugc; static uint32_t dfgc, dbgc; static area_stack_t astack; static XftColor sel_fg; static XftDraw *xft_draw; #define MAX_WIDTHS (1 << 16) static wchar_t xft_char[MAX_WIDTHS]; static char xft_width[MAX_WIDTHS]; void update_gc (void) { xcb_change_gc(c, gc[GC_DRAW], XCB_GC_BACKGROUND | XCB_GC_FOREGROUND, (const uint32_t []) { fgc, bgc }); xcb_change_gc(c, gc[GC_CLEAR], XCB_GC_FOREGROUND, (const uint32_t []) { bgc }); xcb_change_gc(c, gc[GC_ATTR], XCB_GC_FOREGROUND, (const uint32_t []) { ugc }); XftColorFree(dpy, visual_ptr, colormap , &sel_fg); char color[8] = "#ffffff"; snprintf(color, sizeof(color), "#%06X", fgc); if (!XftColorAllocName (dpy, visual_ptr, colormap, color, &sel_fg)) { fprintf(stderr, "Couldn't allocate xft font color '%s'\n", color); } } void fill_rect (xcb_drawable_t d, xcb_gcontext_t gc, int x, int y, int width, int height) { xcb_poly_fill_rectangle(c, d, gc, 1, (const xcb_rectangle_t []) { { x, y, width, height } }); } int xft_char_width_slot (wchar_t ch) { int slot = ch % MAX_WIDTHS; while (xft_char[slot] != 0 && xft_char[slot] != ch) { slot = (slot + 1) % MAX_WIDTHS; } return slot; } int xft_char_width (wchar_t ch, font_t *cur_font) { int slot = xft_char_width_slot(ch); if (!xft_char[slot]) { XGlyphInfo gi; FT_UInt glyph = XftCharIndex (dpy, cur_font->xft_ft, ch); XftFontLoadGlyphs (dpy, cur_font->xft_ft, FcFalse, &glyph, 1); XftGlyphExtents (dpy, cur_font->xft_ft, &glyph, 1, &gi); XftFontUnloadGlyphs (dpy, cur_font->xft_ft, &glyph, 1); xft_char[slot] = ch; xft_width[slot] = gi.xOff; return gi.xOff; } else if (xft_char[slot] == ch) return xft_width[slot]; else return 0; } int draw_char (monitor_t *mon, font_t *cur_font, int x, int align, wchar_t ch) { int ch_width; if (cur_font->xft_ft) { ch_width = xft_char_width(ch, cur_font); } else { ch_width = cur_font->width_lut[ch - cur_font->char_min].character_width; } switch (align) { case ALIGN_C: xcb_copy_area(c, mon->pixmap, mon->pixmap, gc[GC_DRAW], mon->width / 2 - x / 2, 0, mon->width / 2 - (x + ch_width) / 2, 0, x, bh); x = mon->width / 2 - (x + ch_width) / 2 + x; break; case ALIGN_R: xcb_copy_area(c, mon->pixmap, mon->pixmap, gc[GC_DRAW], mon->width - x, 0, mon->width - x - ch_width, 0, x, bh); x = mon->width - ch_width; break; } /* Draw the background first */ fill_rect(mon->pixmap, gc[GC_CLEAR], x, by, ch_width, bh); int y = bh / 2 + cur_font->height / 2- cur_font->descent; if (cur_font->xft_ft) { XftDrawString32 (xft_draw, &sel_fg, cur_font->xft_ft, x,y, &ch, 1); } else { char c_ = ch > 127 ? ' ' : ch; /* xcb accepts string in UCS-2 BE, so swap */ ch = (ch >> 8) | (ch << 8); /* String baseline coordinates */ xcb_image_text_8(c, 1, mon->pixmap, gc[GC_DRAW], x,y, &c_); } /* We can render both at the same time */ if (attrs & ATTR_OVERL) fill_rect(mon->pixmap, gc[GC_ATTR], x, 0, ch_width, bu); if (attrs & ATTR_UNDERL) fill_rect(mon->pixmap, gc[GC_ATTR], x, bh - bu, ch_width, bu); return ch_width; } int utf8decode(char *s, wchar_t *u) { unsigned char c; int i, n, rtn; rtn = 1; c = *s; if(~c & 0x80) { /* 0xxxxxxx */ *u = c; return rtn; } else if((c & 0xE0) == 0xC0) { /* 110xxxxx */ *u = c & 0x1F; n = 1; } else if((c & 0xF0) == 0xE0) { /* 1110xxxx */ *u = c & 0x0F; n = 2; } else if((c & 0xF8) == 0xF0) { /* 11110xxx */ *u = c & 0x07; n = 3; } else { goto invalid; } for(i = n, ++s; i > 0; --i, ++rtn, ++s) { c = *s; if((c & 0xC0) != 0x80) /* 10xxxxxx */ goto invalid; *u <<= 6; *u |= c & 0x3F; } if((n == 1 && *u < 0x80) || (n == 2 && *u < 0x800) || (n == 3 && *u < 0x10000) || (*u >= 0xD800 && *u <= 0xDFFF)) { goto invalid; } return rtn; invalid: *u = 0xFFFD; return rtn; } uint32_t parse_color (const char *str, char **end, const uint32_t def) { xcb_alloc_named_color_reply_t *nc_reply = 0; int str_len; uint32_t ret; if (!str) return def; /* Reset */ if (str[0] == '-') { if (end) *end = (char *)str + 1; return def; } /* Hex representation */ if (str[0] == '#') { errno = 0; uint32_t tmp = strtoul(str + 1, end, 16); /* Some error checking is definitely good */ if (errno) tmp = def; /* Xorg uses colors with premultiplied alpha */ unsigned int a = (tmp&0xff000000) >> 24; unsigned int r = (tmp&0x00ff0000) >> 16; unsigned int g = (tmp&0x0000ff00) >> 8; unsigned int b = (tmp&0x000000ff); if (a) { r = (r * 255) / a; g = (g * 255) / a; b = (b * 255) / a; /* Clamp on overflow */ if (r > 255) r = 255; if (g > 255) g = 255; if (b > 255) b = 255; } else { } return a << 24 | r << 16 | g << 8 | b; } /* Actual color name, resolve it */ str_len = 0; while (isalpha(str[str_len])) str_len++; nc_reply = xcb_alloc_named_color_reply(c, xcb_alloc_named_color(c, colormap, str_len, str), NULL); if (!nc_reply) fprintf(stderr, "Could not alloc color \"%.*s\"\n", str_len, str); ret = (nc_reply) ? nc_reply->pixel : def; free(nc_reply); if (end) *end = (char *)str + str_len; return ret; } void set_attribute (const char modifier, const char attribute) { int pos = indexof(attribute, "ou"); if (pos < 0) { fprintf(stderr, "Invalid attribute \"%c\" found\n", attribute); return; } switch (modifier) { case '+': attrs |= (1<= astack.slot[i].begin && x < astack.slot[i].end) return &astack.slot[i]; return NULL; } void area_shift (xcb_window_t win, const int align, int delta) { if (align == ALIGN_L) return; if (align == ALIGN_C) delta /= 2; for (int i = 0; i < astack.pos; i++) { if (astack.slot[i].window == win && astack.slot[i].align == align) { astack.slot[i].begin -= delta; astack.slot[i].end -= delta; } } } bool area_add (char *str, const char *optend, char **end, monitor_t *mon, const int x, const int align, const int button) { char *p = str; area_t *a = &astack.slot[astack.pos]; if (astack.pos == N) { fprintf(stderr, "astack overflow!\n"); return false; } /* A wild close area tag appeared! */ if (*p != ':') { *end = p; /* Basic safety checks */ if (!a->cmd || a->align != align || a->window != mon->window) return false; const int size = x - a->begin; switch (align) { case ALIGN_L: a->end = x; break; case ALIGN_C: a->begin = mon->width / 2 - size / 2 + a->begin / 2; a->end = a->begin + size; break; case ALIGN_R: /* The newest is the rightmost one */ a->begin = mon->width - size; a->end = mon->width; break; } astack.pos++; return true; } char *trail = strchr(++p, ':'); /* Find the trailing : and make sure it's whitin the formatting block, also reject empty commands */ if (!trail || p == trail || trail > optend) { *end = p; return false; } *trail = '\0'; /* This is a pointer to the string buffer allocated in the main */ a->cmd = p; a->align = align; a->begin = x; a->window = mon->window; a->button = button; *end = trail + 1; return true; } /* returns NULL if character cannot be printed */ font_t * select_drawable_font (const wchar_t c) { /* if the end is reached without finding an apropriate font, return NULL. * If the font can draw the character, return it. */ for (int i = 0; font_list[i] != NULL; i++) { font_t *font = font_list[i]; if ((c > font->char_min && c < font->char_max && font->width_lut[c - font->char_min].character_width != 0)|| (font->xft_ft && XftCharExists(dpy, font->xft_ft, c))) return font; } return NULL; } /* returns NULL if character cannot be printed */ font_t * select_drawable_font_cache (wchar_t c) { if (c < FONT_CACHE_SIZE) return font_cache[c]; else return select_drawable_font(c); } void parse (char *text) { font_t *cur_font; monitor_t *cur_mon; int pos_x, align, button; char *p = text, *end; uint32_t tmp; pos_x = 0; align = ALIGN_L; cur_mon = monhead; memset(&astack, 0, sizeof(area_stack_t)); for (monitor_t *m = monhead; m != NULL; m = m->next) fill_rect(m->pixmap, gc[GC_CLEAR], 0, 0, m->width, bh); /* Create xft drawable */ if (!(xft_draw = XftDrawCreate (dpy, cur_mon->pixmap, visual_ptr , colormap ))) { fprintf(stderr, "Couldn't create xft drawable\n"); } for (;;) { if (*p == '\0' || *p == '\n') return; if (*p == '%' && p++ && *p == '{' && (end = strchr(p++, '}'))) { while (p < end) { while (isspace(*p)) p++; switch (*p++) { case '+': set_attribute('+', *p++); break; case '-': set_attribute('-', *p++); break; case '!': set_attribute('!', *p++); break; case 'R': tmp = fgc; fgc = bgc; bgc = tmp; update_gc(); break; case 'l': pos_x = 0; align = ALIGN_L; break; case 'c': pos_x = 0; align = ALIGN_C; break; case 'r': pos_x = 0; align = ALIGN_R; break; case 'A': button = XCB_BUTTON_INDEX_1; /* The range is 1-5 */ if (isdigit(*p) && (*p > '0' && *p < '6')) button = *p++ - '0'; area_add(p, end, &p, cur_mon, pos_x, align, button); break; case 'B': bgc = parse_color(p, &p, dbgc); update_gc(); break; case 'F': fgc = parse_color(p, &p, dfgc); update_gc(); break; case 'U': ugc = parse_color(p, &p, dbgc); update_gc(); break; case 'S': if (*p == '+' && cur_mon->next) { cur_mon = cur_mon->next; } else if (*p == '-' && cur_mon->prev) { cur_mon = cur_mon->prev; } else if (*p == 'f') { cur_mon = monhead; } else if (*p == 'l') { cur_mon = montail ? montail : monhead; } else if (isdigit(*p)) { cur_mon = monhead; for (int i = 0; i != *p-'0' && cur_mon->next; i++) cur_mon = cur_mon->next; } else { p++; continue; } p++; pos_x = 0; break; /* In case of error keep parsing after the closing } */ default: p = end; } } /* Eat the trailing } */ p++; } else { /* utf-8 -> ucs-2 */ wchar_t t; p += utf8decode(p, &t); cur_font = select_drawable_font(t); if (!cur_font) continue; if(cur_font->ptr) xcb_change_gc(c, gc[GC_DRAW] , XCB_GC_FONT, (const uint32_t []) { cur_font->ptr }); int w = draw_char(cur_mon, cur_font, pos_x, align, t); pos_x += w; area_shift(cur_mon->window, align, w); } } XftDrawDestroy (xft_draw); } font_t * font_load (const char *str) { xcb_query_font_cookie_t queryreq; xcb_query_font_reply_t *font_info; xcb_void_cookie_t cookie; xcb_font_t font; font = xcb_generate_id(c); font_t *ret = calloc(1, sizeof(font_t)); if (!ret) return NULL; cookie = xcb_open_font_checked(c, font, strlen(str), str); if (!xcb_request_check (c, cookie)) { queryreq = xcb_query_font(c, font); font_info = xcb_query_font_reply(c, queryreq, NULL); ret->ptr = font; ret->descent = font_info->font_descent; ret->height = font_info->font_ascent + font_info->font_descent; ret->char_max = font_info->max_byte1 << 8 | font_info->max_char_or_byte2; ret->char_min = font_info->min_byte1 << 8 | font_info->min_char_or_byte2; ret->width_lut = xcb_query_font_char_infos(font_info); } else if ((ret->xft_ft = XftFontOpenName (dpy, scr_nbr, str))) { ret->ascent = ret->xft_ft->ascent; ret->descent = ret->xft_ft->descent; ret->height = ret->ascent + ret->descent; } else { fprintf(stderr, "Could not load font %s\n", str); return NULL; } return ret; } enum { NET_WM_WINDOW_TYPE, NET_WM_WINDOW_TYPE_DOCK, NET_WM_DESKTOP, NET_WM_STRUT_PARTIAL, NET_WM_STRUT, NET_WM_STATE, NET_WM_STATE_STICKY, NET_WM_STATE_ABOVE, }; void set_ewmh_atoms (void) { const char *atom_names[] = { "_NET_WM_WINDOW_TYPE", "_NET_WM_WINDOW_TYPE_DOCK", "_NET_WM_DESKTOP", "_NET_WM_STRUT_PARTIAL", "_NET_WM_STRUT", "_NET_WM_STATE", /* Leave those at the end since are batch-set */ "_NET_WM_STATE_STICKY", "_NET_WM_STATE_ABOVE", }; const int atoms = sizeof(atom_names)/sizeof(char *); xcb_intern_atom_cookie_t atom_cookie[atoms]; xcb_atom_t atom_list[atoms]; xcb_intern_atom_reply_t *atom_reply; /* As suggested fetch all the cookies first (yum!) and then retrieve the * atoms to exploit the async'ness */ for (int i = 0; i < atoms; i++) atom_cookie[i] = xcb_intern_atom(c, 0, strlen(atom_names[i]), atom_names[i]); for (int i = 0; i < atoms; i++) { atom_reply = xcb_intern_atom_reply(c, atom_cookie[i], NULL); if (!atom_reply) return; atom_list[i] = atom_reply->atom; free(atom_reply); } /* Prepare the strut array */ for (monitor_t *mon = monhead; mon; mon = mon->next) { int strut[12] = {0}; if (topbar) { strut[2] = bh; strut[8] = mon->x; strut[9] = mon->x + mon->width; } else { strut[3] = bh; strut[10] = mon->x; strut[11] = mon->x + mon->width; } xcb_change_property(c, XCB_PROP_MODE_REPLACE, mon->window, atom_list[NET_WM_WINDOW_TYPE], XCB_ATOM_ATOM, 32, 1, &atom_list[NET_WM_WINDOW_TYPE_DOCK]); xcb_change_property(c, XCB_PROP_MODE_APPEND, mon->window, atom_list[NET_WM_STATE], XCB_ATOM_ATOM, 32, 2, &atom_list[NET_WM_STATE_STICKY]); xcb_change_property(c, XCB_PROP_MODE_REPLACE, mon->window, atom_list[NET_WM_DESKTOP], XCB_ATOM_CARDINAL, 32, 1, (const uint32_t []) { -1 } ); xcb_change_property(c, XCB_PROP_MODE_REPLACE, mon->window, atom_list[NET_WM_STRUT_PARTIAL], XCB_ATOM_CARDINAL, 32, 12, strut); xcb_change_property(c, XCB_PROP_MODE_REPLACE, mon->window, atom_list[NET_WM_STRUT], XCB_ATOM_CARDINAL, 32, 4, strut); } } monitor_t * monitor_new (int x, int y, int width, int height) { monitor_t *ret; ret = calloc(1, sizeof(monitor_t)); if (!ret) { fprintf(stderr, "Failed to allocate new monitor\n"); exit(EXIT_FAILURE); } ret->x = x; ret->width = width; ret->next = ret->prev = NULL; int win_y = (topbar ? by : height - bh - by) + y; ret->window = xcb_generate_id(c); xcb_create_window(c, XCB_COPY_FROM_PARENT, ret->window, scr->root, x, win_y, width, bh, 0, XCB_WINDOW_CLASS_INPUT_OUTPUT, scr->root_visual, XCB_CW_BACK_PIXEL | XCB_CW_EVENT_MASK , (const uint32_t []) { bgc, XCB_EVENT_MASK_EXPOSURE | XCB_EVENT_MASK_BUTTON_PRESS }); ret->pixmap = xcb_generate_id(c); xcb_create_pixmap(c, scr->root_depth, ret->pixmap, ret->window, width, bh); return ret; } void monitor_add (monitor_t *mon) { if (!monhead) { monhead = mon; } else if (!montail) { montail = mon; monhead->next = mon; mon->prev = monhead; } else { mon->prev = montail; montail->next = mon; montail = montail->next; } } int rect_sort_cb (const void *p1, const void *p2) { const xcb_rectangle_t *r1 = (xcb_rectangle_t *)p1; const xcb_rectangle_t *r2 = (xcb_rectangle_t *)p2; if (r1->x < r2->x || r1->y < r2->y) return -1; if (r1->x > r2->x || r1->y > r2->y) return 1; return 0; } void monitor_create_chain (xcb_rectangle_t *rects, const int num) { int i; int width = 0, height = 0; int left = bx; /* Sort before use */ qsort(rects, num, sizeof(xcb_rectangle_t), rect_sort_cb); for (i = 0; i < num; i++) { int h = rects[i].y + rects[i].height; /* Accumulated width of all monitors */ width += rects[i].width; /* Get height of screen from y_offset + height of lowest monitor */ if (h >= height) height = h; } if (bw < 0) bw = width - bx; if (bh < 0 || bh > height) bh = font_list[0]->height + bu + 2; /* Check the geometry */ if (bx + bw > width || by + bh > height) { fprintf(stderr, "The geometry specified doesn't fit the screen!\n"); exit(EXIT_FAILURE); } /* Left is a positive number or zero therefore monitors with zero width are excluded */ width = bw; for (i = 0; i < num; i++) { if (rects[i].y + rects[i].height < by) continue; if (rects[i].width > left) { monitor_t *mon = monitor_new( rects[i].x + left, rects[i].y, min(width, rects[i].width - left), rects[i].height); monitor_add(mon); width -= rects[i].width - left; /* No need to check for other monitors */ if (width <= 0) break; } left -= rects[i].width; if (left < 0) left = 0; } } void get_randr_monitors (void) { xcb_randr_get_screen_resources_current_reply_t *rres_reply; xcb_randr_output_t *outputs; int i, j, num, valid = 0; rres_reply = xcb_randr_get_screen_resources_current_reply(c, xcb_randr_get_screen_resources_current(c, scr->root), NULL); if (!rres_reply) { fprintf(stderr, "Failed to get current randr screen resources\n"); return; } num = xcb_randr_get_screen_resources_current_outputs_length(rres_reply); outputs = xcb_randr_get_screen_resources_current_outputs(rres_reply); /* There should be at least one output */ if (num < 1) { free(rres_reply); return; } xcb_rectangle_t rects[num]; /* Get all outputs */ for (i = 0; i < num; i++) { xcb_randr_get_output_info_reply_t *oi_reply; xcb_randr_get_crtc_info_reply_t *ci_reply; oi_reply = xcb_randr_get_output_info_reply(c, xcb_randr_get_output_info(c, outputs[i], XCB_CURRENT_TIME), NULL); /* Output disconnected or not attached to any CRTC ? */ if (!oi_reply || oi_reply->crtc == XCB_NONE || oi_reply->connection != XCB_RANDR_CONNECTION_CONNECTED) { rects[i].width = 0; continue; } ci_reply = xcb_randr_get_crtc_info_reply(c, xcb_randr_get_crtc_info(c, oi_reply->crtc, XCB_CURRENT_TIME), NULL); free(oi_reply); if (!ci_reply) { fprintf(stderr, "Failed to get RandR ctrc info\n"); free(rres_reply); return; } /* There's no need to handle rotated screens here (see #69) */ rects[i] = (xcb_rectangle_t) { ci_reply->x, ci_reply->y, ci_reply->width, ci_reply->height }; free(ci_reply); valid++; } free(rres_reply); /* Check for clones and inactive outputs */ for (i = 0; i < num; i++) { if (rects[i].width == 0) continue; for (j = 0; j < num; j++) { /* Does I countain J ? */ if (i != j && rects[j].width) { if (rects[j].x >= rects[i].x && rects[j].x + rects[j].width <= rects[i].x + rects[i].width && rects[j].y >= rects[i].y && rects[j].y + rects[j].height <= rects[i].y + rects[i].height) { rects[j].width = 0; valid--; } } } } if (valid < 1) { fprintf(stderr, "No usable RandR output found\n"); return; } xcb_rectangle_t r[valid]; for (i = j = 0; i < num && j < valid; i++) if (rects[i].width != 0) r[j++] = rects[i]; monitor_create_chain(r, valid); } void get_xinerama_monitors (void) { xcb_xinerama_query_screens_reply_t *xqs_reply; xcb_xinerama_screen_info_iterator_t iter; int screens; xqs_reply = xcb_xinerama_query_screens_reply(c, xcb_xinerama_query_screens_unchecked(c), NULL); iter = xcb_xinerama_query_screens_screen_info_iterator(xqs_reply); screens = iter.rem; xcb_rectangle_t rects[screens]; /* Fetch all the screens first */ for (int i = 0; iter.rem; i++) { rects[i].x = iter.data->x_org; rects[i].y = iter.data->y_org; rects[i].width = iter.data->width; rects[i].height = iter.data->height; xcb_xinerama_screen_info_next(&iter); } free(xqs_reply); monitor_create_chain(rects, screens); } xcb_visualid_t get_visual (void) { visual_ptr = DefaultVisual(dpy, scr_nbr); return scr->root_visual; //scr->root_visual; } void xconn (void) { if ((dpy = XOpenDisplay(0)) == NULL) { fprintf (stderr, "Couldnt open display\n"); } if ((c = XGetXCBConnection(dpy)) == NULL) { fprintf (stderr, "Couldnt connect to X\n"); exit (EXIT_FAILURE); } /* Grab infos from the first screen */ scr = xcb_setup_roots_iterator(xcb_get_setup(c)).data; /* Try to get a RGBA visual and build the colormap for that */ visual = get_visual(); colormap = DefaultColormap(dpy, scr_nbr); } void init (void) { /* Load the fonts */ for (int i = 0; font_names[i]; i++) { font_list[i] = font_load(font_names[i]); if (!font_list[i]) exit(EXIT_FAILURE); } if (!font_list[0]) font_list[0] = font_load("fixed"); if (!font_list[0]) exit(EXIT_FAILURE); for (uint16_t i = 0; i < FONT_CACHE_SIZE; i++) font_cache[i] = select_drawable_font(i); /* To make the alignment uniform, find maximum height */ int maxh = font_list[0]->height; for (int i = 1; font_list[i]; i++) maxh = max(maxh, font_list[i]->height); /* Set maximum height to all fonts */ for (int i = 0; font_list[i]; i++) font_list[i]->height = maxh; /* Generate a list of screens */ const xcb_query_extension_reply_t *qe_reply; /* Initialiaze monitor list head and tail */ monhead = montail = NULL; /* Check if RandR is present */ qe_reply = xcb_get_extension_data(c, &xcb_randr_id); if (qe_reply && qe_reply->present) { get_randr_monitors(); } else { qe_reply = xcb_get_extension_data(c, &xcb_xinerama_id); /* Check if Xinerama extension is present and active */ if (qe_reply && qe_reply->present) { xcb_xinerama_is_active_reply_t *xia_reply; xia_reply = xcb_xinerama_is_active_reply(c, xcb_xinerama_is_active(c), NULL); if (xia_reply && xia_reply->state) get_xinerama_monitors(); free(xia_reply); } } if (!monhead) { /* If I fits I sits */ if (bw < 0) bw = scr->width_in_pixels - bx; /* Adjust the height */ if (bh < 0 || bh > scr->height_in_pixels) bh = maxh + bu + 2; /* Check the geometry */ if (bx + bw > scr->width_in_pixels || by + bh > scr->height_in_pixels) { fprintf(stderr, "The geometry specified doesn't fit the screen!\n"); exit(EXIT_FAILURE); } /* If no RandR outputs or Xinerama screens, fall back to using whole screen */ monhead = monitor_new(0, 0, bw, scr->height_in_pixels); } if (!monhead) exit(EXIT_FAILURE); /* For WM that support EWMH atoms */ set_ewmh_atoms(); /* Create the gc for drawing */ gc[GC_DRAW] = xcb_generate_id(c); xcb_create_gc(c, gc[GC_DRAW], monhead->pixmap, XCB_GC_FOREGROUND | XCB_GC_BACKGROUND, (const uint32_t []) { fgc, bgc }); gc[GC_CLEAR] = xcb_generate_id(c); xcb_create_gc(c, gc[GC_CLEAR], monhead->pixmap, XCB_GC_FOREGROUND, (const uint32_t []) { bgc }); gc[GC_ATTR] = xcb_generate_id(c); xcb_create_gc(c, gc[GC_ATTR], monhead->pixmap, XCB_GC_FOREGROUND, (const uint32_t []) { ugc }); /* Make the bar visible and clear the pixmap */ for (monitor_t *mon = monhead; mon; mon = mon->next) { fill_rect(mon->pixmap, gc[GC_CLEAR], 0, 0, mon->width, bh); xcb_map_window(c, mon->window); } char color[8] = "#ffffff"; snprintf(color, sizeof(color), "#%06X", fgc); if (!XftColorAllocName (dpy, visual_ptr, colormap, color, &sel_fg)) { fprintf(stderr, "Couldn't allocate xft font color '%s'\n", color); } xcb_flush(c); } void cleanup (void) { for (int i = 0; font_list[i]; i++) { if (font_list[i]->xft_ft) { XftFontClose (dpy, font_list[i]->xft_ft); } else { xcb_close_font(c, font_list[i]->ptr); } free(font_list[i]); font_list[i] = NULL; } while (monhead) { monitor_t *next = monhead->next; xcb_destroy_window(c, monhead->window); xcb_free_pixmap(c, monhead->pixmap); free(monhead); monhead = next; } XftColorFree(dpy, visual_ptr, colormap, &sel_fg); if (gc[GC_DRAW]) xcb_free_gc(c, gc[GC_DRAW]); if (gc[GC_CLEAR]) xcb_free_gc(c, gc[GC_CLEAR]); if (gc[GC_ATTR]) xcb_free_gc(c, gc[GC_ATTR]); if (c) xcb_disconnect(c); } void sighandle (int signal) { if (signal == SIGINT || signal == SIGTERM) exit(EXIT_SUCCESS); } /* Parse an X-styled geometry string, we don't support signed offsets tho. */ bool parse_geometry_string (char *str, int *tmp) { char *p = str; int i = 0, j; if (!str || !str[0]) return false; /* The leading = is optional */ if (*p == '=') p++; while (*p) { /* A geometry string has only 4 fields */ if (i >= 4) { fprintf(stderr, "Invalid geometry specified\n"); return false; } /* Move on if we encounter a 'x' or '+' */ if (*p == 'x') { if (i > 0) /* The 'x' must precede '+' */ break; i++; p++; continue; } if (*p == '+') { if (i < 1) /* Stray '+', skip the first two fields */ i = 2; else i++; p++; continue; } /* A digit must follow */ if (!isdigit(*p)) { fprintf(stderr, "Invalid geometry specified\n"); return false; } /* Try to parse the number */ errno = 0; j = strtoul(p, &p, 10); if (errno) { fprintf(stderr, "Invalid geometry specified\n"); return false; } tmp[i] = j; } return true; } void parse_font_list (char *str) { char *tok; if (!str) return; tok = strtok(str, ","); for (int i = 0; tok; i++) { if (i >= MAX_FONT_COUNT) { fprintf(stderr, "Too many fonts; maximum %i\n", MAX_FONT_COUNT); return; } font_names[i] = tok; tok = strtok(NULL, ","); } } int main (int argc, char **argv) { struct pollfd pollin[2] = { { .fd = STDIN_FILENO, .events = POLLIN }, { .fd = -1 , .events = POLLIN }, }; xcb_generic_event_t *ev; xcb_expose_event_t *expose_ev; xcb_button_press_event_t *press_ev; char input[4096] = {0, }; bool permanent = false; int geom_v[4] = { -1, -1, 0, 0 }; /* Install the parachute! */ atexit(cleanup); signal(SIGINT, sighandle); signal(SIGTERM, sighandle); /* Connect to the Xserver and initialize scr */ xconn(); /* B/W combo */ dbgc = bgc = parse_color("black", NULL, scr->black_pixel); dfgc = fgc = parse_color("white", NULL, scr->white_pixel); ugc = fgc; char ch; while ((ch = getopt(argc, argv, "hg:bdf:a:pu:B:F:")) != -1) { switch (ch) { case 'h': printf ("usage: %s [-h | -g | -b | -d | -f | -a | -p | -u | -B | -F]\n" "\t-h Show this help\n" "\t-g Set the bar geometry {width}x{height}+{xoffset}+{yoffset}\n" "\t-b Put bar at the bottom of the screen\n" "\t-d Force docking (use this if your WM isn't EWMH compliant)\n" "\t-f Bar font list, comma separated\n" "\t-p Don't close after the data ends\n" "\t-u Set the underline/overline height in pixels\n" "\t-B Set background color in #AARRGGBB\n" "\t-F Set foreground color in #AARRGGBB\n", argv[0]); exit (EXIT_SUCCESS); case 'g': (void)parse_geometry_string(optarg, geom_v); break; case 'p': permanent = true; break; case 'b': topbar = false; break; case 'd': dock = true; break; case 'f': parse_font_list(optarg); break; case 'u': bu = strtoul(optarg, NULL, 10); break; case 'B': dbgc = bgc = parse_color(optarg, NULL, scr->black_pixel); break; case 'F': dfgc = fgc = parse_color(optarg, NULL, scr->white_pixel); break; } } /* Copy the geometry values in place */ bw = geom_v[0]; bh = geom_v[1]; bx = geom_v[2]; by = geom_v[3]; /* Do the heavy lifting */ init(); /* Get the fd to Xserver */ pollin[1].fd = xcb_get_file_descriptor(c); for (;;) { bool redraw = false; if (poll(pollin, 2, -1) > 0) { if (pollin[0].revents & POLLHUP) { /* No more data... */ if (permanent) pollin[0].fd = -1; /* ...null the fd and continue polling :D */ else break; /* ...bail out */ } if (pollin[0].revents & POLLIN) { /* New input, process it */ if (fgets(input, sizeof(input), stdin) == NULL) break; /* EOF received */ parse(input); redraw = true; } if (pollin[1].revents & POLLIN) { /* Xserver broadcasted an event */ while ((ev = xcb_poll_for_event(c))) { expose_ev = (xcb_expose_event_t *)ev; switch (ev->response_type & 0x7F) { case XCB_EXPOSE: if (expose_ev->count == 0) redraw = true; break; case XCB_BUTTON_PRESS: press_ev = (xcb_button_press_event_t *)ev; { area_t *area = area_get(press_ev->event, press_ev->event_x); /* Respond to the click */ if (area && area->button == press_ev->detail) { write(STDOUT_FILENO, area->cmd, strlen(area->cmd)); write(STDOUT_FILENO, "\n", 1); } } break; } free(ev); } } } if (redraw) { /* Copy our temporary pixmap onto the window */ for (monitor_t *mon = monhead; mon; mon = mon->next) { xcb_copy_area(c, mon->pixmap, mon->window, gc[GC_DRAW], 0, 0, 0, 0, mon->width, bh); } } xcb_flush(c); } return EXIT_SUCCESS; }