1 #include <u.h>
2 #include <libc.h>
3 #include <draw.h>
4 #include <thread.h>
5 #include <cursor.h>
6 #include <mouse.h>
7 #include <keyboard.h>
8 #include <frame.h>
9 #include <fcall.h>
10 #include <plumb.h>
11 #include "dat.h"
12 #include "fns.h"
13 
14 static Rune Lheader[] = {
15 	'N', 'e', 'w', ' ',
16 	'C', 'u', 't', ' ',
17 	'P', 'a', 's', 't', 'e', ' ',
18 	'S', 'n', 'a', 'r', 'f', ' ',
19 	'S', 'o', 'r', 't', ' ',
20 	'Z', 'e', 'r', 'o', 'x', ' ',
21 	'D', 'e', 'l', 'c', 'o', 'l', ' ',
22 	0
23 };
24 
25 void
26 colinit(Column *c, Rectangle r)
27 {
28 	Rectangle r1;
29 	Text *t;
30 
31 	draw(screen, r, display->white, nil, ZP);
32 	c->r = r;
33 	c->w = nil;
34 	c->nw = 0;
35 	t = &c->tag;
36 	t->w = nil;
37 	t->col = c;
38 	r1 = r;
39 	r1.max.y = r1.min.y + font->height;
40 	textinit(t, fileaddtext(nil, t), r1, &reffont, tagcols);
41 	t->what = Columntag;
42 	r1.min.y = r1.max.y;
43 	r1.max.y += Border;
44 	draw(screen, r1, display->black, nil, ZP);
45 	textinsert(t, 0, Lheader, 38, TRUE);
46 	textsetselect(t, t->file->b.nc, t->file->b.nc);
47 	draw(screen, t->scrollr, colbutton, nil, colbutton->r.min);
48 	c->safe = TRUE;
49 }
50 
51 Window*
52 coladd(Column *c, Window *w, Window *clone, int y)
53 {
54 	Rectangle r, r1;
55 	Window *v;
56 	int i, t;
57 
58 	v = nil;
59 	r = c->r;
60 	r.min.y = c->tag.fr.r.max.y+Border;
61 	if(y<r.min.y && c->nw>0){	/* steal half of last window by default */
62 		v = c->w[c->nw-1];
63 		y = v->body.fr.r.min.y+Dy(v->body.fr.r)/2;
64 	}
65 	/* look for window we'll land on */
66 	for(i=0; i<c->nw; i++){
67 		v = c->w[i];
68 		if(y < v->r.max.y)
69 			break;
70 	}
71 	if(c->nw > 0){
72 		if(i < c->nw)
73 			i++;	/* new window will go after v */
74 		/*
75 		 * if v's too small, grow it first.
76 		 */
77 		if(!c->safe || v->body.fr.maxlines<=3){
78 			colgrow(c, v, 1);
79 			y = v->body.fr.r.min.y+Dy(v->body.fr.r)/2;
80 		}
81 		r = v->r;
82 		if(i == c->nw)
83 			t = c->r.max.y;
84 		else
85 			t = c->w[i]->r.min.y-Border;
86 		r.max.y = t;
87 		draw(screen, r, textcols[BACK], nil, ZP);
88 		r1 = r;
89 		y = min(y, t-(v->tag.fr.font->height+v->body.fr.font->height+Border+1));
90 		r1.max.y = min(y, v->body.fr.r.min.y+v->body.fr.nlines*v->body.fr.font->height);
91 		r1.min.y = winresize(v, r1, FALSE);
92 		r1.max.y = r1.min.y+Border;
93 		draw(screen, r1, display->black, nil, ZP);
94 		r.min.y = r1.max.y;
95 	}
96 	if(w == nil){
97 		w = emalloc(sizeof(Window));
98 		w->col = c;
99 		draw(screen, r, textcols[BACK], nil, ZP);
100 		wininit(w, clone, r);
101 	}else{
102 		w->col = c;
103 		winresize(w, r, FALSE);
104 	}
105 	w->tag.col = c;
106 	w->tag.row = c->row;
107 	w->body.col = c;
108 	w->body.row = c->row;
109 	c->w = realloc(c->w, (c->nw+1)*sizeof(Window*));
110 	memmove(c->w+i+1, c->w+i, (c->nw-i)*sizeof(Window*));
111 	c->nw++;
112 	c->w[i] = w;
113 	savemouse(w);
114 	/* near but not on the button */
115 	moveto(mousectl, addpt(w->tag.scrollr.max, Pt(3, 3)));
116 	barttext = &w->body;
117 	c->safe = TRUE;
118 	return w;
119 }
120 
121 void
122 colclose(Column *c, Window *w, int dofree)
123 {
124 	Rectangle r;
125 	int i;
126 
127 	/* w is locked */
128 	if(!c->safe)
129 		colgrow(c, w, 1);
130 	for(i=0; i<c->nw; i++)
131 		if(c->w[i] == w)
132 			goto Found;
133 	error("can't find window");
134   Found:
135 	r = w->r;
136 	w->tag.col = nil;
137 	w->body.col = nil;
138 	w->col = nil;
139 	restoremouse(w);
140 	if(dofree){
141 		windelete(w);
142 		winclose(w);
143 	}
144 	memmove(c->w+i, c->w+i+1, (c->nw-i)*sizeof(Window*));
145 	c->nw--;
146 	c->w = realloc(c->w, c->nw*sizeof(Window*));
147 	if(c->nw == 0){
148 		draw(screen, r, display->white, nil, ZP);
149 		return;
150 	}
151 	if(i == c->nw){		/* extend last window down */
152 		w = c->w[i-1];
153 		r.min.y = w->r.min.y;
154 		r.max.y = c->r.max.y;
155 	}else{			/* extend next window up */
156 		w = c->w[i];
157 		r.max.y = w->r.max.y;
158 	}
159 	draw(screen, r, textcols[BACK], nil, ZP);
160 	if(c->safe)
161 		winresize(w, r, FALSE);
162 }
163 
164 void
165 colcloseall(Column *c)
166 {
167 	int i;
168 	Window *w;
169 
170 	if(c == activecol)
171 		activecol = nil;
172 	textclose(&c->tag);
173 	for(i=0; i<c->nw; i++){
174 		w = c->w[i];
175 		winclose(w);
176 	}
177 	c->nw = 0;
178 	free(c->w);
179 	free(c);
180 	clearmouse();
181 }
182 
183 void
184 colmousebut(Column *c)
185 {
186 	moveto(mousectl, divpt(addpt(c->tag.scrollr.min, c->tag.scrollr.max), 2));
187 }
188 
189 void
190 colresize(Column *c, Rectangle r)
191 {
192 	int i;
193 	Rectangle r1, r2;
194 	Window *w;
195 
196 	clearmouse();
197 	r1 = r;
198 	r1.max.y = r1.min.y + c->tag.fr.font->height;
199 	textresize(&c->tag, r1);
200 	draw(screen, c->tag.scrollr, colbutton, nil, colbutton->r.min);
201 	r1.min.y = r1.max.y;
202 	r1.max.y += Border;
203 	draw(screen, r1, display->black, nil, ZP);
204 	r1.max.y = r.max.y;
205 	for(i=0; i<c->nw; i++){
206 		w = c->w[i];
207 		w->maxlines = 0;
208 		if(i == c->nw-1)
209 			r1.max.y = r.max.y;
210 		else
211 			r1.max.y = r1.min.y+(Dy(w->r)+Border)*Dy(r)/Dy(c->r);
212 		r2 = r1;
213 		r2.max.y = r2.min.y+Border;
214 		draw(screen, r2, display->black, nil, ZP);
215 		r1.min.y = r2.max.y;
216 		r1.min.y = winresize(w, r1, FALSE);
217 	}
218 	c->r = r;
219 }
220 
221 static
222 int
223 colcmp(const void *a, const void *b)
224 {
225 	Rune *r1, *r2;
226 	int i, nr1, nr2;
227 
228 	r1 = (*(Window**)a)->body.file->name;
229 	nr1 = (*(Window**)a)->body.file->nname;
230 	r2 = (*(Window**)b)->body.file->name;
231 	nr2 = (*(Window**)b)->body.file->nname;
232 	for(i=0; i<nr1 && i<nr2; i++){
233 		if(*r1 != *r2)
234 			return *r1-*r2;
235 		r1++;
236 		r2++;
237 	}
238 	return nr1-nr2;
239 }
240 
241 void
242 colsort(Column *c)
243 {
244 	int i, y;
245 	Rectangle r, r1, *rp;
246 	Window **wp, *w;
247 
248 	if(c->nw == 0)
249 		return;
250 	clearmouse();
251 	rp = emalloc(c->nw*sizeof(Rectangle));
252 	wp = emalloc(c->nw*sizeof(Window*));
253 	memmove(wp, c->w, c->nw*sizeof(Window*));
254 	qsort(wp, c->nw, sizeof(Window*), colcmp);
255 	for(i=0; i<c->nw; i++)
256 		rp[i] = wp[i]->r;
257 	r = c->r;
258 	r.min.y = c->tag.fr.r.max.y;
259 	draw(screen, r, textcols[BACK], nil, ZP);
260 	y = r.min.y;
261 	for(i=0; i<c->nw; i++){
262 		w = wp[i];
263 		r.min.y = y;
264 		if(i == c->nw-1)
265 			r.max.y = c->r.max.y;
266 		else
267 			r.max.y = r.min.y+Dy(w->r)+Border;
268 		r1 = r;
269 		r1.max.y = r1.min.y+Border;
270 		draw(screen, r1, display->black, nil, ZP);
271 		r.min.y = r1.max.y;
272 		y = winresize(w, r, FALSE);
273 	}
274 	free(rp);
275 	free(c->w);
276 	c->w = wp;
277 }
278 
279 void
280 colgrow(Column *c, Window *w, int but)
281 {
282 	Rectangle r, cr;
283 	int i, j, k, l, y1, y2, *nl, *ny, tot, nnl, onl, dnl, h;
284 	Window *v;
285 
286 	for(i=0; i<c->nw; i++)
287 		if(c->w[i] == w)
288 			goto Found;
289 	error("can't find window");
290 
291   Found:
292 	cr = c->r;
293 	if(but < 0){	/* make sure window fills its own space properly */
294 		r = w->r;
295 		if(i==c->nw-1 || c->safe==FALSE)
296 			r.max.y = cr.max.y;
297 		else
298 			r.max.y = c->w[i+1]->r.min.y;
299 		winresize(w, r, FALSE);
300 		return;
301 	}
302 	cr.min.y = c->w[0]->r.min.y;
303 	if(but == 3){	/* full size */
304 		if(i != 0){
305 			v = c->w[0];
306 			c->w[0] = w;
307 			c->w[i] = v;
308 		}
309 		draw(screen, cr, textcols[BACK], nil, ZP);
310 		winresize(w, cr, FALSE);
311 		for(i=1; i<c->nw; i++)
312 			c->w[i]->body.fr.maxlines = 0;
313 		c->safe = FALSE;
314 		return;
315 	}
316 	/* store old #lines for each window */
317 	onl = w->body.fr.maxlines;
318 	nl = emalloc(c->nw * sizeof(int));
319 	ny = emalloc(c->nw * sizeof(int));
320 	tot = 0;
321 	for(j=0; j<c->nw; j++){
322 		l = c->w[j]->body.fr.maxlines;
323 		nl[j] = l;
324 		tot += l;
325 	}
326 	/* approximate new #lines for this window */
327 	if(but == 2){	/* as big as can be */
328 		memset(nl, 0, c->nw * sizeof(int));
329 		goto Pack;
330 	}
331 	nnl = min(onl + max(min(5, w->maxlines), onl/2), tot);
332 	if(nnl < w->maxlines)
333 		nnl = (w->maxlines+nnl)/2;
334 	if(nnl == 0)
335 		nnl = 2;
336 	dnl = nnl - onl;
337 	/* compute new #lines for each window */
338 	for(k=1; k<c->nw; k++){
339 		/* prune from later window */
340 		j = i+k;
341 		if(j<c->nw && nl[j]){
342 			l = min(dnl, max(1, nl[j]/2));
343 			nl[j] -= l;
344 			nl[i] += l;
345 			dnl -= l;
346 		}
347 		/* prune from earlier window */
348 		j = i-k;
349 		if(j>=0 && nl[j]){
350 			l = min(dnl, max(1, nl[j]/2));
351 			nl[j] -= l;
352 			nl[i] += l;
353 			dnl -= l;
354 		}
355 	}
356     Pack:
357 	/* pack everyone above */
358 	y1 = cr.min.y;
359 	for(j=0; j<i; j++){
360 		v = c->w[j];
361 		r = v->r;
362 		r.min.y = y1;
363 		r.max.y = y1+Dy(v->tag.all);
364 		if(nl[j])
365 			r.max.y += 1 + nl[j]*v->body.fr.font->height;
366 		if(!c->safe || !eqrect(v->r, r)){
367 			draw(screen, r, textcols[BACK], nil, ZP);
368 			winresize(v, r, c->safe);
369 		}
370 		r.min.y = v->r.max.y;
371 		r.max.y += Border;
372 		draw(screen, r, display->black, nil, ZP);
373 		y1 = r.max.y;
374 	}
375 	/* scan to see new size of everyone below */
376 	y2 = c->r.max.y;
377 	for(j=c->nw-1; j>i; j--){
378 		v = c->w[j];
379 		r = v->r;
380 		r.min.y = y2-Dy(v->tag.all);
381 		if(nl[j])
382 			r.min.y -= 1 + nl[j]*v->body.fr.font->height;
383 		r.min.y -= Border;
384 		ny[j] = r.min.y;
385 		y2 = r.min.y;
386 	}
387 	/* compute new size of window */
388 	r = w->r;
389 	r.min.y = y1;
390 	r.max.y = r.min.y+Dy(w->tag.all);
391 	h = w->body.fr.font->height;
392 	if(y2-r.max.y >= 1+h+Border){
393 		r.max.y += 1;
394 		r.max.y += h*((y2-r.max.y)/h);
395 	}
396 	/* draw window */
397 	if(!c->safe || !eqrect(w->r, r)){
398 		draw(screen, r, textcols[BACK], nil, ZP);
399 		winresize(w, r, c->safe);
400 	}
401 	if(i < c->nw-1){
402 		r.min.y = r.max.y;
403 		r.max.y += Border;
404 		draw(screen, r, display->black, nil, ZP);
405 		for(j=i+1; j<c->nw; j++)
406 			ny[j] -= (y2-r.max.y);
407 	}
408 	/* pack everyone below */
409 	y1 = r.max.y;
410 	for(j=i+1; j<c->nw; j++){
411 		v = c->w[j];
412 		r = v->r;
413 		r.min.y = y1;
414 		r.max.y = y1+Dy(v->tag.all);
415 		if(nl[j])
416 			r.max.y += 1 + nl[j]*v->body.fr.font->height;
417 		if(!c->safe || !eqrect(v->r, r)){
418 			draw(screen, r, textcols[BACK], nil, ZP);
419 			winresize(v, r, c->safe);
420 		}
421 		if(j < c->nw-1){	/* no border on last window */
422 			r.min.y = v->r.max.y;
423 			r.max.y += Border;
424 			draw(screen, r, display->black, nil, ZP);
425 		}
426 		y1 = r.max.y;
427 	}
428 	r = w->r;
429 	r.min.y = y1;
430 	r.max.y = c->r.max.y;
431 	draw(screen, r, textcols[BACK], nil, ZP);
432 	free(nl);
433 	free(ny);
434 	c->safe = TRUE;
435 	winmousebut(w);
436 }
437 
438 void
439 coldragwin(Column *c, Window *w, int but)
440 {
441 	Rectangle r;
442 	int i, b;
443 	Point p, op;
444 	Window *v;
445 	Column *nc;
446 
447 	clearmouse();
448 	setcursor(mousectl, &boxcursor);
449 	b = mouse->buttons;
450 	op = mouse->xy;
451 	while(mouse->buttons == b)
452 		readmouse(mousectl);
453 	setcursor(mousectl, nil);
454 	if(mouse->buttons){
455 		while(mouse->buttons)
456 			readmouse(mousectl);
457 		return;
458 	}
459 
460 	for(i=0; i<c->nw; i++)
461 		if(c->w[i] == w)
462 			goto Found;
463 	error("can't find window");
464 
465   Found:
466 	p = mouse->xy;
467 	if(abs(p.x-op.x)<5 && abs(p.y-op.y)<5){
468 		colgrow(c, w, but);
469 		winmousebut(w);
470 		return;
471 	}
472 	/* is it a flick to the right? */
473 	if(abs(p.y-op.y)<10 && p.x>op.x+30 && rowwhichcol(c->row, p)==c)
474 		p.x += Dx(w->r);	/* yes: toss to next column */
475 	nc = rowwhichcol(c->row, p);
476 	if(nc!=nil && nc!=c){
477 		colclose(c, w, FALSE);
478 		coladd(nc, w, nil, p.y);
479 		winmousebut(w);
480 		return;
481 	}
482 	if(i==0 && c->nw==1)
483 		return;			/* can't do it */
484 	if((i>0 && p.y<c->w[i-1]->r.min.y) || (i<c->nw-1 && p.y>w->r.max.y)
485 	|| (i==0 && p.y>w->r.max.y)){
486 		/* shuffle */
487 		colclose(c, w, FALSE);
488 		coladd(c, w, nil, p.y);
489 		winmousebut(w);
490 		return;
491 	}
492 	if(i == 0)
493 		return;
494 	v = c->w[i-1];
495 	if(p.y < v->tag.all.max.y)
496 		p.y = v->tag.all.max.y;
497 	if(p.y > w->r.max.y-Dy(w->tag.all)-Border)
498 		p.y = w->r.max.y-Dy(w->tag.all)-Border;
499 	r = v->r;
500 	r.max.y = p.y;
501 	if(r.max.y > v->body.fr.r.min.y){
502 		r.max.y -= (r.max.y-v->body.fr.r.min.y)%v->body.fr.font->height;
503 		if(v->body.fr.r.min.y == v->body.fr.r.max.y)
504 			r.max.y++;
505 	}
506 	if(!eqrect(v->r, r)){
507 		draw(screen, r, textcols[BACK], nil, ZP);
508 		winresize(v, r, c->safe);
509 	}
510 	r.min.y = v->r.max.y;
511 	r.max.y = r.min.y+Border;
512 	draw(screen, r, display->black, nil, ZP);
513 	r.min.y = r.max.y;
514 	if(i == c->nw-1)
515 		r.max.y = c->r.max.y;
516 	else
517 		r.max.y = c->w[i+1]->r.min.y-Border;
518 	if(!eqrect(w->r, r)){
519 		draw(screen, r, textcols[BACK], nil, ZP);
520 		winresize(w, r, c->safe);
521 	}
522 	c->safe = TRUE;
523     	winmousebut(w);
524 }
525 
526 Text*
527 colwhich(Column *c, Point p)
528 {
529 	int i;
530 	Window *w;
531 
532 	if(!ptinrect(p, c->r))
533 		return nil;
534 	if(ptinrect(p, c->tag.all))
535 		return &c->tag;
536 	for(i=0; i<c->nw; i++){
537 		w = c->w[i];
538 		if(ptinrect(p, w->r)){
539 			if(ptinrect(p, w->tag.all))
540 				return &w->tag;
541 			return &w->body;
542 		}
543 	}
544 	return nil;
545 }
546 
547 int
548 colclean(Column *c)
549 {
550 	int i, clean;
551 
552 	clean = TRUE;
553 	for(i=0; i<c->nw; i++)
554 		clean &= winclean(c->w[i], TRUE);
555 	return clean;
556 }