Xinqi Bao's Git

projects / dmenu.git / blobdiff
summary | log | commit | diff | tree
raw | inline | side by side
paring
[dmenu.git] / draw.c
diff --git a/draw.c b/draw.c
index 41022f0..80a5074 100644 (file)
--- a/draw.c
+++ b/draw.c
@@ -1,143 +1,191 @@
 /* See LICENSE file for copyright and license details. */
 /* See LICENSE file for copyright and license details. */
+#include <locale.h>
+#include <stdarg.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <X11/Xlib.h>
+#include "draw.h"
 
 
-/* enums */
-enum { ColFG, ColBG, ColLast };
-
-/* typedefs */
-typedef struct {
-       int x, y, w, h;
-       unsigned long norm[ColLast];
-       unsigned long sel[ColLast];
-       Drawable drawable;
-       GC gc;
-       struct {
-               XFontStruct *xfont;
-               XFontSet set;
-               int ascent;
-               int descent;
-               int height;
-       } font;
-} DC; /* draw context */
-
-/* forward declarations */
-static void dccleanup(void);
-static void dcsetup(void);
-static void drawtext(const char *text, unsigned long col[ColLast]);
-static unsigned long getcolor(const char *colstr);
-static void initfont(const char *fontstr);
-static int textnw(const char *text, unsigned int len);
-static int textw(const char *text);
-
-static DC dc;
+#define MAX(a, b)   ((a) > (b) ? (a) : (b))
+#define MIN(a, b)   ((a) < (b) ? (a) : (b))
+#define DEFFONT     "fixed"
 
 
-void
-dccleanup(void) {
-       if(dc.font.set)
-               XFreeFontSet(dpy, dc.font.set);
-       else
-               XFreeFont(dpy, dc.font.xfont);
-       XFreePixmap(dpy, dc.drawable);
-       XFreeGC(dpy, dc.gc);
-}
+static Bool loadfont(DC *dc, const char *fontstr);
 
 void
 
 void
-dcsetup(void) {
-       /* style */
-       dc.norm[ColBG] = getcolor(normbgcolor);
-       dc.norm[ColFG] = getcolor(normfgcolor);
-       dc.sel[ColBG] = getcolor(selbgcolor);
-       dc.sel[ColFG] = getcolor(selfgcolor);
-       initfont(font);
-
-       /* pixmap */
-       dc.drawable = XCreatePixmap(dpy, parent, mw, mh, DefaultDepth(dpy, screen));
-       dc.gc = XCreateGC(dpy, parent, 0, NULL);
-       XSetLineAttributes(dpy, dc.gc, 1, LineSolid, CapButt, JoinMiter);
-       if(!dc.font.set)
-               XSetFont(dpy, dc.gc, dc.font.xfont->fid);
+drawrect(DC *dc, int x, int y, unsigned int w, unsigned int h, Bool fill, unsigned long color) {
+       XRectangle r = { dc->x + x, dc->y + y, w, h };
+
+       if(!fill) {
+               r.width -= 1;
+               r.height -= 1;
+       }
+       XSetForeground(dc->dpy, dc->gc, color);
+       (fill ? XFillRectangles : XDrawRectangles)(dc->dpy, dc->canvas, dc->gc, &r, 1);
 }
 
 }
 
+
 void
 void
-drawtext(const char *text, unsigned long col[ColLast]) {
+drawtext(DC *dc, const char *text, unsigned long col[ColLast]) {
        char buf[256];
        char buf[256];
-       int i, x, y, h, len, olen;
-       XRectangle r = { dc.x, dc.y, dc.w, dc.h };
-
-       XSetForeground(dpy, dc.gc, col[ColBG]);
-       XFillRectangles(dpy, dc.drawable, dc.gc, &r, 1);
-       if(!text)
-               return;
-       olen = strlen(text);
-       h = dc.font.height;
-       y = dc.y + ((h+2) / 2) - (h / 2) + dc.font.ascent;
-       x = dc.x + (h / 2);
+       size_t n, mn;
+
        /* shorten text if necessary */
        /* shorten text if necessary */
-       for(len = MIN(olen, sizeof buf); len && textnw(text, len) > dc.w - h; len--);
-       if(!len)
-               return;
-       memcpy(buf, text, len);
-       if(len < olen)
-               for(i = len; i && i > len - 3; buf[--i] = '.');
-       XSetForeground(dpy, dc.gc, col[ColFG]);
-       if(dc.font.set)
-               XmbDrawString(dpy, dc.drawable, dc.font.set, dc.gc, x, y, buf, len);
-       else
-               XDrawString(dpy, dc.drawable, dc.gc, x, y, buf, len);
+       n = strlen(text);
+       for(mn = MIN(n, sizeof buf); textnw(dc, text, mn) > dc->w - dc->font.height/2; mn--)
+               if(mn == 0)
+                       return;
+       memcpy(buf, text, mn);
+       if(mn < n)
+               for(n = MAX(mn-3, 0); n < mn; buf[n++] = '.');
+
+       drawrect(dc, 0, 0, dc->w, dc->h, True, BG(dc, col));
+       drawtextn(dc, buf, mn, col);
+}
+
+void
+drawtextn(DC *dc, const char *text, size_t n, unsigned long col[ColLast]) {
+       int x, y;
+
+       x = dc->x + dc->font.height/2;
+       y = dc->y + dc->font.ascent+1;
+
+       XSetForeground(dc->dpy, dc->gc, FG(dc, col));
+       if(dc->font.set)
+               XmbDrawString(dc->dpy, dc->canvas, dc->font.set, dc->gc, x, y, text, n);
+       else {
+               XSetFont(dc->dpy, dc->gc, dc->font.xfont->fid);
+               XDrawString(dc->dpy, dc->canvas, dc->gc, x, y, text, n);
+       }
+}
+
+void
+eprintf(const char *fmt, ...) {
+       va_list ap;
+
+       fprintf(stderr, "%s: ", progname);
+       va_start(ap, fmt);
+       vfprintf(stderr, fmt, ap);
+       va_end(ap);
+       exit(EXIT_FAILURE);
+}
+
+void
+freedc(DC *dc) {
+       if(dc->font.set)
+               XFreeFontSet(dc->dpy, dc->font.set);
+       if(dc->font.xfont)
+               XFreeFont(dc->dpy, dc->font.xfont);
+       if(dc->canvas)
+               XFreePixmap(dc->dpy, dc->canvas);
+       XFreeGC(dc->dpy, dc->gc);
+       XCloseDisplay(dc->dpy);
+       free(dc);
 }
 
 unsigned long
 }
 
 unsigned long
-getcolor(const char *colstr) {
-       Colormap cmap = DefaultColormap(dpy, screen);
+getcolor(DC *dc, const char *colstr) {
+       Colormap cmap = DefaultColormap(dc->dpy, DefaultScreen(dc->dpy));
        XColor color;
 
        XColor color;
 
-       if(!XAllocNamedColor(dpy, cmap, colstr, &color, &color))
-               eprint("drawtext: cannot allocate color '%s'\n", colstr);
+       if(!XAllocNamedColor(dc->dpy, cmap, colstr, &color, &color))
+               eprintf("cannot allocate color '%s'\n", colstr);
        return color.pixel;
 }
 
        return color.pixel;
 }
 
+DC *
+initdc(void) {
+       DC *dc;
+
+       if(!setlocale(LC_CTYPE, "") || !XSupportsLocale())
+               weprintf("no locale support\n");
+       if(!(dc = calloc(1, sizeof *dc)))
+               eprintf("cannot malloc %u bytes\n", sizeof *dc);
+       if(!(dc->dpy = XOpenDisplay(NULL)))
+               eprintf("cannot open display\n");
+
+       dc->gc = XCreateGC(dc->dpy, DefaultRootWindow(dc->dpy), 0, NULL);
+       XSetLineAttributes(dc->dpy, dc->gc, 1, LineSolid, CapButt, JoinMiter);
+       return dc;
+}
+
 void
 void
-initfont(const char *fontstr) {
-       char *def, **missing = NULL;
+initfont(DC *dc, const char *fontstr) {
+       if(!loadfont(dc, fontstr ? fontstr : DEFFONT)) {
+               if(fontstr != NULL)
+                       weprintf("cannot load font '%s'\n", fontstr);
+               if(fontstr == NULL || !loadfont(dc, DEFFONT))
+                       eprintf("cannot load font '%s'\n", DEFFONT);
+       }
+       dc->font.height = dc->font.ascent + dc->font.descent;
+}
+
+Bool
+loadfont(DC *dc, const char *fontstr) {
+       char *def, **missing;
        int i, n;
 
        int i, n;
 
-       if(!fontstr || fontstr[0] == '\0')
-               eprint("drawtext: cannot load font: '%s'\n", fontstr);
-       dc.font.set = XCreateFontSet(dpy, fontstr, &missing, &n, &def);
-       if(missing)
-               XFreeStringList(missing);
-       if(dc.font.set) {
+       if(!*fontstr)
+               return False;
+       if((dc->font.set = XCreateFontSet(dc->dpy, fontstr, &missing, &n, &def))) {
+               char **names;
                XFontStruct **xfonts;
                XFontStruct **xfonts;
-               char **font_names;
-               dc.font.ascent = dc.font.descent = 0;
-               n = XFontsOfFontSet(dc.font.set, &xfonts, &font_names);
-               for(i = 0; i < n; i++) {
-                       dc.font.ascent = MAX(dc.font.ascent, (*xfonts)->ascent);
-                       dc.font.descent = MAX(dc.font.descent, (*xfonts)->descent);
-                       xfonts++;
+
+               n = XFontsOfFontSet(dc->font.set, &xfonts, &names);
+               for(i = dc->font.ascent = dc->font.descent = 0; i < n; i++) {
+                       dc->font.ascent = MAX(dc->font.ascent, xfonts[i]->ascent);
+                       dc->font.descent = MAX(dc->font.descent, xfonts[i]->descent);
                }
        }
                }
        }
-       else {
-               if(!(dc.font.xfont = XLoadQueryFont(dpy, fontstr))
-               && !(dc.font.xfont = XLoadQueryFont(dpy, "fixed")))
-                       eprint("drawtext: cannot load font: '%s'\n", fontstr);
-               dc.font.ascent = dc.font.xfont->ascent;
-               dc.font.descent = dc.font.xfont->descent;
+       else if((dc->font.xfont = XLoadQueryFont(dc->dpy, fontstr))) {
+               dc->font.ascent = dc->font.xfont->ascent;
+               dc->font.descent = dc->font.xfont->descent;
        }
        }
-       dc.font.height = dc.font.ascent + dc.font.descent;
+       if(missing)
+               XFreeStringList(missing);
+       return (dc->font.set || dc->font.xfont);
+}
+
+void
+mapdc(DC *dc, Window win, unsigned int w, unsigned int h) {
+       XCopyArea(dc->dpy, dc->canvas, win, dc->gc, 0, 0, w, h, 0, 0);
+}
+
+void
+resizedc(DC *dc, unsigned int w, unsigned int h) {
+       if(dc->canvas)
+               XFreePixmap(dc->dpy, dc->canvas);
+       dc->canvas = XCreatePixmap(dc->dpy, DefaultRootWindow(dc->dpy), w, h,
+                                  DefaultDepth(dc->dpy, DefaultScreen(dc->dpy)));
+       dc->x = dc->y = 0;
+       dc->w = w;
+       dc->h = h;
+       dc->invert = False;
 }
 
 int
 }
 
 int
-textnw(const char *text, unsigned int len) {
-       XRectangle r;
+textnw(DC *dc, const char *text, size_t len) {
+       if(dc->font.set) {
+               XRectangle r;
 
 
-       if(dc.font.set) {
-               XmbTextExtents(dc.font.set, text, len, NULL, &r);
+               XmbTextExtents(dc->font.set, text, len, NULL, &r);
                return r.width;
        }
                return r.width;
        }
-       return XTextWidth(dc.font.xfont, text, len);
+       return XTextWidth(dc->font.xfont, text, len);
 }
 
 int
 }
 
 int
-textw(const char *text) {
-       return textnw(text, strlen(text)) + dc.font.height;
+textw(DC *dc, const char *text) {
+       return textnw(dc, text, strlen(text)) + dc->font.height;
+}
+
+void
+weprintf(const char *fmt, ...) {
+       va_list ap;
+
+       fprintf(stderr, "%s: ", progname);
+       va_start(ap, fmt);
+       vfprintf(stderr, fmt, ap);
+       va_end(ap);
 }
 }