1 #include <u.h>
2 #include <libc.h>
3 #include <draw.h>
4 #include <mouse.h>
5 
6 Display	*display;
7 Font	*font;
8 Image	*screen;
9 int	_drawdebug;
10 
11 static char deffontname[] = "*default*";
12 Screen	*_screen;
13 
14 int		debuglockdisplay = 1;
15 char	*winsize;
16 
17 int		visibleclicks = 0;
18 Image	*mousebuttons;
19 Image	*mousesave;
20 Mouse	_drawmouse;
21 
22 /*
23 static void
24 drawshutdown(void)
25 {
26 	Display *d;
27 
28 	d = display;
29 	if(d){
30 		display = nil;
31 		closedisplay(d);
32 	}
33 }
34 */
35 
36 int
37 geninitdraw(char *devdir, void(*error)(Display*, char*), char *fontname, char *label, char *windir, int ref)
38 {
39 	Subfont *df;
40 	char buf[128], *p;
41 
42 	if(label == nil)
43 		label = argv0;
44 	display = _initdisplay(error, label);
45 	if(display == nil)
46 		return -1;
47 
48 	/*
49 	 * Set up default font
50 	 */
51 	df = getdefont(display);
52 	display->defaultsubfont = df;
53 	if(df == nil){
54 		fprint(2, "imageinit: can't open default subfont: %r\n");
55     Error:
56 		closedisplay(display);
57 		display = nil;
58 		return -1;
59 	}
60 	if(fontname == nil)
61 		fontname = getenv("font");
62 
63 	/*
64 	 * Build fonts with caches==depth of screen, for speed.
65 	 * If conversion were faster, we'd use 0 and save memory.
66 	 */
67 	if(fontname == nil){
68 		snprint(buf, sizeof buf, "%d %d\n0 %d\t%s\n", df->height, df->ascent,
69 			df->n-1, deffontname);
70 //BUG: Need something better for this	installsubfont("*default*", df);
71 		font = buildfont(display, buf, deffontname);
72 		if(font == nil){
73 			fprint(2, "imageinit: can't open default font: %r\n");
74 			goto Error;
75 		}
76 	}else{
77 		font = openfont(display, fontname);	/* BUG: grey fonts */
78 		if(font == nil){
79 			fprint(2, "imageinit: can't open font %s: %r\n", fontname);
80 			goto Error;
81 		}
82 	}
83 	display->defaultfont = font;
84 
85 	_screen = allocscreen(display->image, display->white, 0);
86 	display->screenimage = display->image;	/* _allocwindow wants screenimage->chan */
87 	screen = _allocwindow(nil, _screen, display->image->r, Refnone, DWhite);
88 	if(screen == nil){
89 		fprint(2, "_allocwindow: %r\n");
90 		goto Error;
91 	}
92 	display->screenimage = screen;
93 	draw(screen, screen->r, display->white, nil, ZP);
94 	flushimage(display, 1);
95 
96 	p = getenv("visibleclicks");
97 	visibleclicks = p != nil && *p == '1';
98 	if(visibleclicks) {
99 		Font *f;
100 		
101 		f = display->defaultfont;
102 		mousebuttons = allocimage(display, Rect(0,0,64,22), screen->chan, 0, DWhite);
103 		border(mousebuttons, mousebuttons->r, 1, display->black, ZP);
104 		border(mousebuttons, Rect(0, 0, 22, 22), 1, display->black, ZP);
105 		border(mousebuttons, Rect(42, 0, 64, 22), 1, display->black, ZP);
106 		string(mousebuttons, Pt(10-stringwidth(display->defaultfont, "1")/2, 11-f->height/2), display->black, ZP, display->defaultfont, "1");
107 		string(mousebuttons, Pt(21+10-stringwidth(display->defaultfont, "2")/2, 11-f->height/2), display->black, ZP, display->defaultfont, "2");
108 		string(mousebuttons, Pt(42+10-stringwidth(display->defaultfont, "3")/2, 11-f->height/2), display->black, ZP, display->defaultfont, "3");
109 		mousesave = allocimage(display, Rect(0,0,64,22), screen->chan, 0, 0);
110 	}
111 
112 	/*
113 	 * I don't see any reason to go away gracefully,
114 	 * and if some other proc exits holding the display
115 	 * lock, this atexit call never finishes.
116 	 *
117 	 * atexit(drawshutdown);
118 	 */
119 	return 1;
120 }
121 
122 int
123 initdraw(void (*error)(Display*, char*), char *fontname, char *label)
124 {
125 	return geninitdraw("/dev", error, fontname, label, "/dev", Refnone);
126 }
127 
128 extern int _freeimage1(Image*);
129 
130 static Image*
131 getimage0(Display *d, Image *image)
132 {
133 	char info[12*12+1];
134 	uchar *a;
135 	int n;
136 
137 	/*
138 	 * If there's an old screen, it has id 0.  The 'J' request below
139 	 * will try to install the new screen as id 0, so the old one 
140 	 * must be freed first.
141 	 */
142 	if(image){
143 		_freeimage1(image);
144 		memset(image, 0, sizeof(Image));
145 	}
146 
147 	a = bufimage(d, 2);
148 	a[0] = 'J';
149 	a[1] = 'I';
150 	if(flushimage(d, 0) < 0){
151 		fprint(2, "cannot read screen info: %r\n");
152 		return nil;
153 	}
154 
155 	n = _displayrddraw(d, info, sizeof info);
156 	if(n != 12*12){
157 		fprint(2, "short screen info\n");
158 		return nil;
159 	}
160 
161 	if(image == nil){
162 		image = mallocz(sizeof(Image), 1);
163 		if(image == nil){
164 			fprint(2, "cannot allocate image: %r\n");
165 			return nil;
166 		}
167 	}
168 
169 	image->display = d;
170 	image->id = 0;
171 	image->chan = strtochan(info+2*12);
172 	image->depth = chantodepth(image->chan);
173 	image->repl = atoi(info+3*12);
174 	image->r.min.x = atoi(info+4*12);
175 	image->r.min.y = atoi(info+5*12);
176 	image->r.max.x = atoi(info+6*12);
177 	image->r.max.y = atoi(info+7*12);
178 	image->clipr.min.x = atoi(info+8*12);
179 	image->clipr.min.y = atoi(info+9*12);
180 	image->clipr.max.x = atoi(info+10*12);
181 	image->clipr.max.y = atoi(info+11*12);
182 	return image;
183 }
184 
185 /*
186  * Attach, or possibly reattach, to window.
187  * If reattaching, maintain value of screen pointer.
188  */
189 int
190 getwindow(Display *d, int ref)
191 {
192 	Image *i, *oi;
193 	
194 	/* XXX check for destroyed? */
195 	
196 	/*
197 	 * Libdraw promises not to change the value of "screen",
198 	 * so we have to reuse the image structure
199 	 * memory we already have.
200 	 */
201 	oi = d->image;
202 	i = getimage0(d, oi);
203 	if(i == nil)
204 		sysfatal("getwindow failed");
205 	d->image = i;
206 	/* fprint(2, "getwindow %p -> %p\n", oi, i); */
207 
208 	freescreen(_screen);
209 	_screen = allocscreen(i, d->white, 0);
210 	_freeimage1(screen);
211 	screen = _allocwindow(screen, _screen, i->r, ref, DWhite);
212 	d->screenimage = screen;
213 	return 0;
214 }
215 
216 Display*
217 _initdisplay(void (*error)(Display*, char*), char *label)
218 {
219 	Display *disp;
220 	Image *image;
221 
222 	fmtinstall('P', Pfmt);
223 	fmtinstall('R', Rfmt);
224 
225 	disp = mallocz(sizeof(Display), 1);
226 	if(disp == nil){
227     Error1:
228 		return nil;
229 	}
230 	disp->srvfd = -1;
231 	image = nil;
232 	if(0){
233     Error2:
234 		free(image);
235 		free(disp);
236 		goto Error1;
237 	}
238 	disp->bufsize = 65500;
239 	disp->buf = malloc(disp->bufsize+5);	/* +5 for flush message */
240 	disp->bufp = disp->buf;
241 	disp->error = error;
242 	qlock(&disp->qlock);
243 
244 	if(disp->buf == nil)
245 		goto Error2;
246 	if(0){
247     Error3:
248 		free(disp->buf);
249 		goto Error2;
250 	}
251 	
252 	if(_displaymux(disp) < 0
253 	|| _displayconnect(disp) < 0
254 	|| _displayinit(disp, label, winsize) < 0)
255 		goto Error3;
256 	if(0){
257     Error4:
258 		close(disp->srvfd);
259 		goto Error3;
260 	}
261 
262 	image = getimage0(disp, nil);
263 	if(image == nil)
264 		goto Error4;
265 	
266 	disp->image = image;
267 	disp->white = allocimage(disp, Rect(0, 0, 1, 1), GREY1, 1, DWhite);
268 	disp->black = allocimage(disp, Rect(0, 0, 1, 1), GREY1, 1, DBlack);
269 	if(disp->white == nil || disp->black == nil){
270 		free(disp->white);
271 		free(disp->black);
272 		goto Error4;
273 	}
274 	
275 	disp->opaque = disp->white;
276 	disp->transparent = disp->black;
277 
278 	return disp;
279 }
280 
281 /*
282  * Call with d unlocked.
283  * Note that disp->defaultfont and defaultsubfont are not freed here.
284  */
285 void
286 closedisplay(Display *disp)
287 {
288 	int fd;
289 	char buf[128];
290 
291 	if(disp == nil)
292 		return;
293 	if(disp == display)
294 		display = nil;
295 	if(disp->oldlabel[0]){
296 		snprint(buf, sizeof buf, "%s/label", disp->windir);
297 		fd = open(buf, OWRITE);
298 		if(fd >= 0){
299 			write(fd, disp->oldlabel, strlen(disp->oldlabel));
300 			close(fd);
301 		}
302 	}
303 
304 	free(disp->devdir);
305 	free(disp->windir);
306 	if(disp->white)
307 		freeimage(disp->white);
308 	if(disp->black)
309 		freeimage(disp->black);
310 	if(disp->srvfd >= 0)
311 		close(disp->srvfd);
312 	free(disp);
313 }
314 
315 void
316 lockdisplay(Display *disp)
317 {
318 	if(debuglockdisplay){
319 		/* avoid busy looping; it's rare we collide anyway */
320 		while(!canqlock(&disp->qlock)){
321 			fprint(1, "proc %d waiting for display lock...\n", getpid());
322 			sleep(1000);
323 		}
324 	}else
325 		qlock(&disp->qlock);
326 }
327 
328 void
329 unlockdisplay(Display *disp)
330 {
331 	qunlock(&disp->qlock);
332 }
333 
334 void
335 drawerror(Display *d, char *s)
336 {
337 	char err[ERRMAX];
338 
339 	if(d->error)
340 		d->error(d, s);
341 	else{
342 		errstr(err, sizeof err);
343 		fprint(2, "draw: %s: %s\n", s, err);
344 		exits(s);
345 	}
346 }
347 
348 static
349 int
350 doflush(Display *d)
351 {
352 	int n;
353 
354 	n = d->bufp-d->buf;
355 	if(n <= 0)
356 		return 1;
357 
358 	if(_displaywrdraw(d, d->buf, n) != n){
359 		if(_drawdebug)
360 			fprint(2, "flushimage fail: d=%p: %r\n", d); /**/
361 		d->bufp = d->buf;	/* might as well; chance of continuing */
362 		return -1;
363 	}
364 	d->bufp = d->buf;
365 	return 1;
366 }
367 
368 int
369 flushimage(Display *d, int visible)
370 {
371 	if(visible == 1 && visibleclicks && mousebuttons && _drawmouse.buttons) {
372 		Rectangle r, r1;
373 		int ret;
374 
375 		r = mousebuttons->r;
376 		r = rectaddpt(r, _drawmouse.xy);
377 		r = rectaddpt(r, Pt(-Dx(mousebuttons->r)/2, -Dy(mousebuttons->r)-3));
378 		drawop(mousesave, mousesave->r, screen, nil, r.min, S);
379 		
380 		r1 = rectaddpt(Rect(0, 0, 22, 22), r.min);
381 		if(_drawmouse.buttons & 1)
382 			drawop(screen, r1, mousebuttons, nil, ZP, S);
383 		r1 = rectaddpt(r1, Pt(21, 0));
384 		if(_drawmouse.buttons & 2)
385 			drawop(screen, r1, mousebuttons, nil, Pt(21, 0), S);
386 		r1 = rectaddpt(r1, Pt(21, 0));
387 		if(_drawmouse.buttons & 4)
388 			drawop(screen, r1, mousebuttons, nil, Pt(42, 0), S);
389 		ret = flushimage(d, 2);
390 		drawop(screen, r, mousesave, nil, ZP, S);
391 		return ret;
392 	}
393 	
394 	if(visible){
395 		*d->bufp++ = 'v';	/* five bytes always reserved for this */
396 		if(d->_isnewdisplay){
397 			BPLONG(d->bufp, d->screenimage->id);
398 			d->bufp += 4;
399 		}
400 	}
401 	return doflush(d);
402 }
403 
404 uchar*
405 bufimage(Display *d, int n)
406 {
407 	uchar *p;
408 
409 	if(n<0 || d == nil || n>d->bufsize){
410 		abort();
411 		werrstr("bad count in bufimage");
412 		return 0;
413 	}
414 	if(d->bufp+n > d->buf+d->bufsize)
415 		if(doflush(d) < 0)
416 			return 0;
417 	p = d->bufp;
418 	d->bufp += n;
419 	return p;
420 }
421