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                 free(clr);
 114 }
 115
 116 void
 117 drw_setfont(Drw *drw, Fnt *font) {
 118         if(drw)
 119                 drw->font = font;
 120 }
 121
 122 void
 123 drw_setscheme(Drw *drw, ClrScheme *scheme) {
 124         if(drw && scheme)
 125                 drw->scheme = scheme;
 126 }
 127
 128 void
 129 drw_rect(Drw *drw, int x, int y, unsigned int w, unsigned int h, int filled, int empty, int invert) {
 130         int dx;
 131
 132         if(!drw || !drw->font || !drw->scheme)
 133                 return;
 134         XSetForeground(drw->dpy, drw->gc, invert ? drw->scheme->bg->rgb : drw->scheme->fg->rgb);
 135         dx = (drw->font->ascent + drw->font->descent + 2) / 4;
 136         if(filled)
 137                 XFillRectangle(drw->dpy, drw->drawable, drw->gc, x+1, y+1, dx+1, dx+1);
 138         else if(empty)
 139                 XDrawRectangle(drw->dpy, drw->drawable, drw->gc, x+1, y+1, dx, dx);
 140 }
 141
 142 void
 143 drw_text(Drw *drw, int x, int y, unsigned int w, unsigned int h, const char *text, int invert) {
 144         _drw_text(drw, x, y, w, h, text, invert, 1);
 145 }
 146
 147 void
 148 drw_text_nopad(Drw *drw, int x, int y, unsigned int w, unsigned int h, const char *text, int invert) {
 149         _drw_text(drw, x, y, w, h, text, invert, 0);
 150 }
 151
 152 void
 153 _drw_text(Drw *drw, int x, int y, unsigned int w, unsigned int h, const char *text, int invert, int pad) {
 154         char buf[256];
 155         int i, tx, ty, th, len, olen;
 156         Extnts tex;
 157
 158         if(!drw || !drw->scheme)
 159                 return;
 160         XSetForeground(drw->dpy, drw->gc, invert ? drw->scheme->fg->rgb : drw->scheme->bg->rgb);
 161         XFillRectangle(drw->dpy, drw->drawable, drw->gc, x, y, w, h);
 162         if(!text || !drw->font)
 163                 return;
 164         olen = strlen(text);
 165         drw_font_getexts(drw->font, text, olen, &tex);
 166         th = drw->font->ascent + drw->font->descent;
 167         ty = y + (h / 2) - (th / 2) + drw->font->ascent;
 168         tx = x + (h / 2) * pad;
 169         /* shorten text if necessary */
 170         for(len = MIN(olen, sizeof buf); len && (tex.w > w - tex.h * pad || w < tex.h * pad); len--)
 171                 drw_font_getexts(drw->font, text, len, &tex);
 172         if(!len)
 173                 return;
 174         memcpy(buf, text, len);
 175         if(len < olen)
 176                 for(i = len; i && i > len - 3; buf[--i] = '.');
 177         XSetForeground(drw->dpy, drw->gc, invert ? drw->scheme->bg->rgb : drw->scheme->fg->rgb);
 178         if(drw->font->set)
 179                 XmbDrawString(drw->dpy, drw->drawable, drw->font->set, drw->gc, tx, ty, buf, len);
 180         else
 181                 XDrawString(drw->dpy, drw->drawable, drw->gc, tx, ty, buf, len);
 182 }
 183
 184 void
 185 drw_map(Drw *drw, Window win, int x, int y, unsigned int w, unsigned int h) {
 186         if(!drw)
 187                 return;
 188         XCopyArea(drw->dpy, drw->drawable, win, drw->gc, x, y, w, h, x, y);
 189         XSync(drw->dpy, False);
 190 }
 191
 192
 193 void
 194 drw_font_getexts(Fnt *font, const char *text, unsigned int len, Extnts *tex) {
 195         XRectangle r;
 196
 197         if(!font || !text)
 198                 return;
 199         if(font->set) {
 200                 XmbTextExtents(font->set, text, len, NULL, &r);
 201                 tex->w = r.width;
 202                 tex->h = r.height;
 203         }
 204         else {
 205                 tex->h = font->ascent + font->descent;
 206                 tex->w = XTextWidth(font->xfont, text, len);
 207         }
 208 }
 209
 210 unsigned int
 211 drw_font_getexts_width(Fnt *font, const char *text, unsigned int len) {
 212         Extnts tex;
 213
 214         if(!font)
 215                 return -1;
 216         drw_font_getexts(font, text, len, &tex);
 217         return tex.w;
 218 }
 219
 220 Cur *
 221 drw_cur_create(Drw *drw, int shape) {
 222         Cur *cur = (Cur *)calloc(1, sizeof(Cur));
 223
 224         if(!drw || !cur)
 225                 return NULL;
 226         cur->cursor = XCreateFontCursor(drw->dpy, shape);
 227         return cur;
 228 }
 229
 230 void
 231 drw_cur_free(Drw *drw, Cur *cursor) {
 232         if(!drw || !cursor)
 233                 return;
 234         XFreeCursor(drw->dpy, cursor->cursor);
 235         free(cursor);
 236 }