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
 10Drw *
 11drw_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
 26void
 27drw_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
 37void
 38drw_free(Drw *drw) {
 39	XFreePixmap(drw->dpy, drw->drawable);
 40	XFreeGC(drw->dpy, drw->gc);
 41	free(drw);
 42}
 43
 44Fnt *
 45drw_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
 81void
 82drw_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
 92Clr *
 93drw_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
110void
111drw_clr_free(Clr *clr) {
112	if(!clr)
113		return;
114	free(clr);
115}
116
117void
118drw_setfont(Drw *drw, Fnt *font) {
119	if(drw)
120		drw->font = font;
121}
122
123void
124drw_settheme(Drw *drw, Theme *theme) {
125	if(!drw || !theme) 
126		return;
127	drw->theme = theme;
128}
129
130void
131drw_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
144void
145drw_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
176void
177drw_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
185void
186drw_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
202unsigned int
203drw_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
212Cur *
213drw_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
222void
223drw_cur_free(Drw *drw, Cur *cursor) {
224	if(!drw || !cursor)
225		return;
226	XFreeCursor(drw->dpy, cursor->cursor);
227	free(cursor);
228}