drw.c

   1 /* See LICENSE file for copyright and license details. */
   2 #include <stdio.h>
   3 #include <stdlib.h>
   4 #include <string.h>
   5 #include <X11/Xlib.h>
   6
   7 #include "drw.h"
   8 #include "util.h"
   9
  10 Drw *
  11 drw_create(Display *dpy, int screen, Window root, unsigned int w, unsigned int h) {
  12         Drw *drw = (Drw *)calloc(1, sizeof(Drw));
  13         if(!drw)
  14                 return NULL;
  15         drw->dpy = dpy;
  16         drw->screen = screen;
  17         drw->root = root;
  18         drw->w = w;
  19         drw->h = h;
  20         drw->drawable = XCreatePixmap(dpy, root, w, h, DefaultDepth(dpy, screen));
  21         drw->gc = XCreateGC(dpy, root, 0, NULL);
  22         XSetLineAttributes(dpy, drw->gc, 1, LineSolid, CapButt, JoinMiter);
  23         return drw;
  24 }
  25
  26 void
  27 drw_resize(Drw *drw, unsigned int w, unsigned int h) {
  28         if(!drw)
  29                 return;
  30         drw->w = w;
  31         drw->h = h;
  32         if(drw->drawable != 0)
  33                 XFreePixmap(drw->dpy, drw->drawable);
  34         drw->drawable = XCreatePixmap(drw->dpy, drw->root, w, h, DefaultDepth(drw->dpy, drw->screen));
  35 }
  36
  37 void
  38 drw_free(Drw *drw) {
  39         XFreePixmap(drw->dpy, drw->drawable);
  40         XFreeGC(drw->dpy, drw->gc);
  41         free(drw);
  42 }
  43
  44 Fnt *
  45 drw_font_create(Display *dpy, const char *fontname) {
  46         Fnt *font;
  47         char *def, **missing;
  48         int n;
  49
  50         font = (Fnt *)calloc(1, sizeof(Fnt));
  51         if(!font)
  52                 return NULL;
  53         font->set = XCreateFontSet(dpy, fontname, &missing, &n, &def);
  54         if(missing) {
  55                 while(n--)
  56                         fprintf(stderr, "drw: missing fontset: %s\n", missing[n]);
  57                 XFreeStringList(missing);
  58         }
  59         if(font->set) {
  60                 XFontStruct **xfonts;
  61                 char **font_names;
  62                 XExtentsOfFontSet(font->set);
  63                 n = XFontsOfFontSet(font->set, &xfonts, &font_names);
  64                 while(n--) {
  65                         font->ascent = MAX(font->ascent, (*xfonts)->ascent);
  66                         font->descent = MAX(font->descent,(*xfonts)->descent);
  67                         xfonts++;
  68                 }
  69         }
  70         else {
  71                 if(!(font->xfont = XLoadQueryFont(dpy, fontname))
  72                 && !(font->xfont = XLoadQueryFont(dpy, "fixed")))
  73                         die("error, cannot load font: '%s'\n", fontname);
  74                 font->ascent = font->xfont->ascent;
  75                 font->descent = font->xfont->descent;
  76         }
  77         font->h = font->ascent + font->descent;
  78         return font;
  79 }
  80
  81 void
  82 drw_font_free(Display *dpy, Fnt *font) {
  83         if(!font)
  84                 return;
  85         if(font->set)
  86                 XFreeFontSet(dpy, font->set);
  87         else
  88                 XFreeFont(dpy, font->xfont);
  89         free(font);
  90 }
  91
  92 Clr *
  93 drw_clr_create(Drw *drw, const char *clrname) {
  94         Clr *clr;
  95         Colormap cmap;
  96         XColor color;
  97
  98         if(!drw)
  99                 return NULL;
 100         clr = (Clr *)calloc(1, sizeof(Clr));
 101         if(!clr)
 102                 return NULL;
 103         cmap = DefaultColormap(drw->dpy, drw->screen);
 104         if(!XAllocNamedColor(drw->dpy, cmap, clrname, &color, &color))
 105                 die("error, cannot allocate color '%s'\n", clrname);
 106         clr->rgb = color.pixel;
 107         return clr;
 108 }
 109
 110 void
 111 drw_clr_free(Clr *clr) {
 112         if(!clr)
 113                 return;
 114         free(clr);
 115 }
 116
 117 void
 118 drw_setfont(Drw *drw, Fnt *font) {
 119         if(drw)
 120                 drw->font = font;
 121 }
 122
 123 void
 124 drw_settheme(Drw *drw, Theme *theme) {
 125         if(!drw || !theme)
 126                 return;
 127         drw->theme = theme;
 128 }
 129
 130 void
 131 drw_rect(Drw *drw, int x, int y, unsigned int w, unsigned int h, int filled, int empty, int invert) {
 132         int dx;
 133
 134         if(!drw || !drw->font || !drw->theme)
 135                 return;
 136         XSetForeground(drw->dpy, drw->gc, invert ? drw->theme->bg->rgb : drw->theme->fg->rgb);
 137         dx = (drw->font->ascent + drw->font->descent + 2) / 4;
 138         if(filled)
 139                 XFillRectangle(drw->dpy, drw->drawable, drw->gc, x+1, y+1, dx+1, dx+1);
 140         else if(empty)
 141                 XDrawRectangle(drw->dpy, drw->drawable, drw->gc, x+1, y+1, dx, dx);
 142 }
 143
 144 void
 145 drw_text(Drw *drw, int x, int y, unsigned int w, unsigned int h, const char *text, int invert) {
 146         char buf[256];
 147         int i, tx, ty, th, len, olen;
 148         Extnts tex;
 149
 150         if(!drw || !drw->theme)
 151                 return;
 152         XSetForeground(drw->dpy, drw->gc, invert ? drw->theme->fg->rgb : drw->theme->bg->rgb);
 153         XFillRectangle(drw->dpy, drw->drawable, drw->gc, x, y, w, h);
 154         if(!text || !drw->font)
 155                 return;
 156         olen = strlen(text);
 157         drw_font_getexts(drw->font, text, olen, &tex);
 158         th = drw->font->ascent + drw->font->descent;
 159         ty = y + (h / 2) - (th / 2) + drw->font->ascent;
 160         tx = x + (h / 2);
 161         /* shorten text if necessary */
 162         for(len = MIN(olen, sizeof buf); len && tex.w > w - tex.h; len--)
 163                 drw_font_getexts(drw->font, text, len, &tex);
 164         if(!len)
 165                 return;
 166         memcpy(buf, text, len);
 167         if(len < olen)
 168                 for(i = len; i && i > len - 3; buf[--i] = '.');
 169         XSetForeground(drw->dpy, drw->gc, invert ? drw->theme->bg->rgb : drw->theme->fg->rgb);
 170         if(drw->font->set)
 171                 XmbDrawString(drw->dpy, drw->drawable, drw->font->set, drw->gc, tx, ty, buf, len);
 172         else
 173                 XDrawString(drw->dpy, drw->drawable, drw->gc, tx, ty, buf, len);
 174 }
 175
 176 void
 177 drw_map(Drw *drw, Window win, int x, int y, unsigned int w, unsigned int h) {
 178         if(!drw)
 179                 return;
 180         XCopyArea(drw->dpy, drw->drawable, win, drw->gc, x, y, w, h, x, y);
 181         XSync(drw->dpy, False);
 182 }
 183
 184
 185 void
 186 drw_font_getexts(Fnt *font, const char *text, unsigned int len, Extnts *tex) {
 187         XRectangle r;
 188
 189         if(!font || !text)
 190                 return;
 191         if(font->set) {
 192                 XmbTextExtents(font->set, text, len, NULL, &r);
 193                 tex->w = r.width;
 194                 tex->h = r.height;
 195         }
 196         else {
 197                 tex->h = font->ascent + font->descent;
 198                 tex->w = XTextWidth(font->xfont, text, len);
 199         }
 200 }
 201
 202 unsigned int
 203 drw_font_getexts_width(Fnt *font, const char *text, unsigned int len) {
 204         Extnts tex;
 205
 206         if(!font)
 207                 return -1;
 208         drw_font_getexts(font, text, len, &tex);
 209         return tex.w;
 210 }
 211
 212 Cur *
 213 drw_cur_create(Drw *drw, int shape) {
 214         Cur *cur = (Cur *)calloc(1, sizeof(Cur));
 215
 216         if(!drw || !cur)
 217                 return NULL;
 218         cur->cursor = XCreateFontCursor(drw->dpy, shape);
 219         return cur;
 220 }
 221
 222 void
 223 drw_cur_free(Drw *drw, Cur *cursor) {
 224         if(!drw || !cursor)
 225                 return;
 226         XFreeCursor(drw->dpy, cursor->cursor);
 227         free(cursor);
 228 }