X-Git-Url: https://jasonwoof.com/gitweb/?a=blobdiff_plain;f=scrotwm.c;h=ba800275ec5a9495749559f00530783ca473085c;hb=81e0c66a1e7cd9a51ce48aa6b481863c5f674a67;hp=d1e0e72528202a0e93c44bdab998f1d5da0652e2;hpb=baebde8cfd12768a4c420b71bbc83c44ad29016d;p=spectrwm.git

diff --git a/scrotwm.c b/scrotwm.c
index d1e0e72..ba80027 100644
--- a/scrotwm.c
+++ b/scrotwm.c
@@ -101,6 +101,8 @@
 #define	SWM_D_MOVE		0x0010
 #define	SWM_D_STACK		0x0020
 #define	SWM_D_MOUSE		0x0040
+#define	SWM_D_PROP		0x0080
+
 
 u_int32_t		swm_debug = 0
 			    | SWM_D_MISC
@@ -109,6 +111,7 @@ u_int32_t		swm_debug = 0
 			    | SWM_D_FOCUS
 			    | SWM_D_MOVE
 			    | SWM_D_STACK
+			    | SWM_D_PROP
 			    ;
 #else
 #define DPRINTF(x...)
@@ -125,11 +128,18 @@ u_int32_t		swm_debug = 0
 #define WIDTH(r)	(r)->g.w	
 #define HEIGHT(r)	(r)->g.h
 
+#ifndef SWM_LIB
+#define SWM_LIB		"/usr/X11R6/lib/swmhack.so"
+#endif
+
+#define SWM_PROPLEN	(16)
+
 char			**start_argv;
 Atom			astate;
 int			(*xerrorxlib)(Display *, XErrorEvent *);
 int			other_wm;
 int			running = 1;
+int			ss_enabled = 0;
 int			xrandr_eventbase;
 int			ignore_enter = 0;
 unsigned int		numlockmask = 0;
@@ -146,6 +156,7 @@ char			*bar_argv[] = { NULL, NULL };
 int			bar_pipe[2];
 char			bar_ext[SWM_BAR_MAX];
 sig_atomic_t		bar_alarm = 0;
+int			bar_delay = 30;
 int			bar_enabled = 1;
 int			bar_extra = 1;
 int			bar_extra_running = 0;
@@ -164,6 +175,7 @@ char			*bar_fonts[] = {
 
 /* terminal + args */
 char			*spawn_term[] = { "xterm", NULL };
+char			*spawn_screenshot[] = { "screenshot.sh", NULL, NULL };
 char			*spawn_menu[] = { "dmenu_run", "-fn", NULL,
 			    "-nb", NULL, "-nf", NULL, "-sb", NULL, "-sf", NULL, NULL };
 
@@ -204,6 +216,7 @@ struct ws_win {
 	struct workspace	*ws;	/* always valid */
 	struct swm_screen	*s;	/* always valid, never changes */
 	XWindowAttributes	wa;
+	XSizeHints		sh;
 };
 TAILQ_HEAD(ws_win_list, ws_win);
 
@@ -289,6 +302,10 @@ union arg {
 #define SWM_ARG_ID_STACKINIT	(11)
 #define SWM_ARG_ID_CYCLEWS_UP	(12)
 #define SWM_ARG_ID_CYCLEWS_DOWN	(13)
+#define SWM_ARG_ID_CYCLESC_UP	(14)
+#define SWM_ARG_ID_CYCLESC_DOWN	(15)
+#define SWM_ARG_ID_SS_ALL	(0)
+#define SWM_ARG_ID_SS_WINDOW	(1)
 	char			**argv;
 };
 
@@ -411,6 +428,8 @@ conf_load(char *filename)
 				asprintf(&bar_fonts[0], "%s", val);
 			else if (!strncmp(var, "bar_action", strlen("bar_action")))
 				asprintf(&bar_argv[0], "%s", val);
+			else if (!strncmp(var, "bar_delay", strlen("bar_delay")))
+				bar_delay = atoi(val);
 			else
 				goto bad;
 			break;
@@ -440,7 +459,13 @@ conf_load(char *filename)
 
 		case 's':
 			if (!strncmp(var, "spawn_term", strlen("spawn_term")))
-				asprintf(&spawn_term[0], "%s", val); /* XXX args? */
+				asprintf(&spawn_term[0], "%s", val);
+			if (!strncmp(var, "screenshot_enabled",
+			    strlen("screenshot_enabled")))
+				ss_enabled = atoi(val);
+			if (!strncmp(var, "screenshot_app",
+			    strlen("screenshot_app")))
+				asprintf(&spawn_screenshot[0], "%s", val);
 			break;
 		default:
 			goto bad;
@@ -533,8 +558,7 @@ bar_update(void)
 		}
 	}
 	XSync(display, False);
-
-	alarm(60);
+	alarm(bar_delay);
 }
 
 void
@@ -701,6 +725,18 @@ quit(struct swm_region *r, union arg *args)
 }
 
 void
+unmap_all(void)
+{
+	struct ws_win		*win;
+	int			i, j;
+
+	for (i = 0; i < ScreenCount(display); i++)
+		for (j = 0; j < SWM_WS_MAX; j++)
+			TAILQ_FOREACH(win, &screens[i].ws[j].winlist, entry)
+				XUnmapWindow(display, win->id);
+}
+
+void
 restart(struct swm_region *r, union arg *args)
 {
 	DNPRINTF(SWM_D_MISC, "restart: %s\n", start_argv[0]);
@@ -712,6 +748,7 @@ restart(struct swm_region *r, union arg *args)
 
 	bar_extra_stop();
 	bar_extra = 1;
+	unmap_all();
 	XCloseDisplay(display);
 	execvp(start_argv[0], start_argv);
 	fprintf(stderr, "execvp failed\n");
@@ -738,8 +775,8 @@ root_to_region(Window root)
 	} else {
 		/* otherwise, choose a region based on pointer location */
 		TAILQ_FOREACH(r, &screens[i].rl, entry) {
-			if (x > X(r) && x < X(r) + WIDTH(r) &&
-			    y > Y(r) && y < Y(r) + HEIGHT(r))
+			if (x >= X(r) && x <= X(r) + WIDTH(r) &&
+			    y >= Y(r) && y <= Y(r) + HEIGHT(r))
 				break;
 		}
 
@@ -766,6 +803,8 @@ find_window(Window id)
 void
 spawn(struct swm_region *r, union arg *args)
 {
+	char			*ret;
+
 	DNPRINTF(SWM_D_MISC, "spawn: %s\n", args->argv[0]);
 	/*
 	 * The double-fork construct avoids zombie processes and keeps the code
@@ -775,6 +814,15 @@ spawn(struct swm_region *r, union arg *args)
 		if (fork() == 0) {
 			if (display)
 				close(ConnectionNumber(display));
+      			setenv("LD_PRELOAD", SWM_LIB, 1);
+			if (asprintf(&ret, "%d", r->ws->idx)) {
+				setenv("_SWM_WS", ret, 1);
+				free(ret);
+			}
+			if (asprintf(&ret, "%d", getpid())) {
+				setenv("_SWM_PID", ret, 1);
+				free(ret);
+			}
 			setsid();
 			execvp(args->argv[0], args->argv);
 			fprintf(stderr, "execvp failed\n");
@@ -872,7 +920,6 @@ switchws(struct swm_region *r, union arg *args)
 
 		TAILQ_FOREACH(win, &old_ws->winlist, entry)
 			XUnmapWindow(display, win->id);
-
 		old_ws->r = NULL;
 		old_ws->restack = 1;
 	} else {
@@ -901,7 +948,6 @@ cyclews(struct swm_region *r, union arg *args)
 	    r->s->idx, WIDTH(r), HEIGHT(r), X(r), Y(r), r->ws->idx);
 
 	a.id = r->ws->idx;
-
 	do {
 		switch (args->id) {
 		case SWM_ARG_ID_CYCLEWS_UP:
@@ -930,6 +976,33 @@ cyclews(struct swm_region *r, union arg *args)
 }
 
 void
+cyclescr(struct swm_region *r, union arg *args)
+{
+	struct swm_region	*rr;
+	int			i;
+
+	i = r->s->idx;
+	switch (args->id) {
+	case SWM_ARG_ID_CYCLESC_UP:
+		rr = TAILQ_NEXT(r, entry);
+		if (rr == NULL)
+			rr = TAILQ_FIRST(&screens[i].rl);
+		break;
+	case SWM_ARG_ID_CYCLESC_DOWN:
+		rr = TAILQ_PREV(r, swm_region_list, entry);
+		if (rr == NULL)
+			rr = TAILQ_LAST(&screens[i].rl, swm_region_list);
+		break;
+	default:
+		return;
+	};
+	unfocus_all();
+	XSetInputFocus(display, PointerRoot, RevertToPointerRoot, CurrentTime);
+	XWarpPointer(display, None, rr->s[i].root, 0, 0, 0, 0, rr->g.x,
+	    rr->g.y + bar_enabled ? bar_height : 0);
+}
+
+void
 swapwin(struct swm_region *r, union arg *args)
 {
 	struct ws_win		*target;
@@ -1014,9 +1087,10 @@ focus(struct swm_region *r, union arg *args)
 		return;
 	}
 
-	if (winfocus == winlostfocus)
+	if (winfocus == winlostfocus || winfocus == NULL)
 		return;
 
+	XMapRaised(display, winfocus->id);
 	focus_win(winfocus);
 	XSync(display, False);
 }
@@ -1032,7 +1106,6 @@ cycle_layout(struct swm_region *r, union arg *args)
 	if (ws->cur_layout->l_stack == NULL)
 		ws->cur_layout = &layouts[0];
 	ignore_enter = 1;
-
 	stack();
 }
 
@@ -1105,46 +1178,25 @@ stack_floater(struct ws_win *win, struct swm_region *r)
 	XConfigureWindow(display, win->id, mask, &wc);
 }
 
+#define SWAPXY(g)	do {				\
+	int tmp;					\
+	tmp = (g)->y; (g)->y = (g)->x; (g)->x = tmp;	\
+	tmp = (g)->h; (g)->h = (g)->w; (g)->w = tmp;	\
+} while (0)
 void
-vertical_config(struct workspace *ws, int id)
+stack_master(struct workspace *ws, struct swm_geometry *g, int rot, int flip)
 {
-	DNPRINTF(SWM_D_STACK, "vertical_resize: workspace: %d\n", ws->idx);
-
-	switch (id) {
-	case SWM_ARG_ID_STACKRESET:
-	case SWM_ARG_ID_STACKINIT:
-		ws->l_state.vertical_msize = SWM_V_SLICE / 2;
-		ws->l_state.vertical_mwin = 1;
-		break;
-	case SWM_ARG_ID_MASTERSHRINK:
-		if (ws->l_state.vertical_msize > 1)
-			ws->l_state.vertical_msize--;
-		break;
-	case SWM_ARG_ID_MASTERGROW:
-		if (ws->l_state.vertical_msize < SWM_V_SLICE - 1)
-			ws->l_state.vertical_msize++;
-		break;
-	case SWM_ARG_ID_MASTERADD:
-		ws->l_state.vertical_mwin++;
-		break;
-	case SWM_ARG_ID_MASTERDEL:
-		if (ws->l_state.vertical_mwin > 0)
-			ws->l_state.vertical_mwin--;
-		break;
-	default:
-		return;
-	}
-}
-
-void
-vertical_stack(struct workspace *ws, struct swm_geometry *g) {
 	XWindowChanges		wc;
-	struct swm_geometry	gg = *g;
+	struct swm_geometry	win_g, r_g = *g;
 	struct ws_win		*win, *winfocus;
-	int			i, j, split, colno, hrh, winno, main_width;
+	int			i, j, w_inc, h_inc, w_base, h_base;
+	int			hrh, extra, h_slice, last_h = 0;
+	int			split, colno, winno, mwin, msize, mscale;
+	int			remain, missing, v_slice;;
 	unsigned int		mask;
 
-	DNPRINTF(SWM_D_STACK, "vertical_stack: workspace: %d\n", ws->idx);
+	DNPRINTF(SWM_D_STACK, "stack_master: workspace: %d\n rot=%s flip=%s",
+	    ws->idx, rot ? "yes" : "no", flip ? "yes" : "no");
 
 	if ((winno = count_win(ws, 0)) == 0)
 		return;
@@ -1153,53 +1205,121 @@ vertical_stack(struct workspace *ws, struct swm_geometry *g) {
 		ws->focus = TAILQ_FIRST(&ws->winlist);
 	winfocus = cur_focus ? cur_focus : ws->focus;
 
-	if (ws->l_state.vertical_mwin &&
-	    winno > ws->l_state.vertical_mwin) {
-		split = ws->l_state.vertical_mwin;
+	win = TAILQ_FIRST(&ws->winlist);
+	if (rot) {
+		w_inc = win->sh.width_inc;
+		w_base = win->sh.base_width;
+		mwin = ws->l_state.horizontal_mwin;
+		mscale = ws->l_state.horizontal_msize;
+		SWAPXY(&r_g);
+	} else {
+		w_inc = win->sh.height_inc;
+		w_base = win->sh.base_height;
+		mwin = ws->l_state.vertical_mwin;
+		mscale = ws->l_state.vertical_msize;
+	}
+	win_g = r_g;
+
+	h_slice = r_g.h / SWM_H_SLICE;
+	if (mwin && winno > mwin) {
+		v_slice = r_g.w / SWM_V_SLICE;
+
+		split = mwin;
 		colno = split;
-		main_width = (g->w / SWM_V_SLICE) *
-		   ws->l_state.vertical_msize;
-		gg.w = main_width;
+		msize = v_slice * mscale;
+
+		if (w_inc > 1 && w_inc < v_slice) {
+			/* adjust for window's requested size increment */
+			remain = (win_g.w - w_base) % w_inc;
+			missing = w_inc - remain;
+
+			if (missing <= extra || j == 0) {
+				extra -= missing;
+				win_g.w += missing;
+			} else {
+				win_g.w -= remain;
+				extra += remain;
+			}
+		}
+
+		win_g.w = msize;
+		if (flip) 
+			win_g.x += r_g.w - msize;
 	} else {
 		colno = winno;
 		split = 0;
 	}
-	hrh = g->h / colno;
-	gg.h = hrh - 2;
+	hrh = r_g.h / colno;
+	extra = r_g.h - (colno * hrh);
+	win_g.h = hrh - 2;
 
 	i = j = 0;
 	TAILQ_FOREACH(win, &ws->winlist, entry) {
 		if (split && i == split) {
 			colno = winno - split;
-			hrh = (g->h / colno);
-			gg.x += main_width + 2;
-			gg.w = g->w - (main_width + 2);
-			gg.h = hrh - 2;
+			hrh = (r_g.h / colno);
+			extra = r_g.h - (colno * hrh);
+			if (flip)
+				win_g.x = r_g.x;
+			else
+				win_g.x += msize + 2;
+			win_g.w = r_g.w - (msize + 2);
 			j = 0;
 		}
-		if (j == colno - 1)
-			gg.h = (hrh + (g->h - (colno * hrh)));
+		win_g.h = hrh - 2;
+		if (rot) {
+			h_inc = win->sh.width_inc;
+			h_base = win->sh.base_width;
+		} else {
+		   	h_inc =	win->sh.height_inc;	
+			h_base = win->sh.base_height;
+		}
+		if (j == colno - 1) {
+			win_g.h = hrh + extra;
+		} else if (h_inc > 1 && h_inc < h_slice) {
+			/* adjust for window's requested size increment */
+			remain = (win_g.h - h_base) % h_inc;
+			missing = h_inc - remain;
+
+			if (missing <= extra || j == 0) {
+				extra -= missing;
+				win_g.h += missing;
+			} else {
+				win_g.h -= remain;
+				extra += remain;
+			}
+		}
 		 
 		if (j == 0)
-			gg.y = g->y;
+			win_g.y = r_g.y;
 		else
-			gg.y += hrh;
-
+			win_g.y += last_h + 2;
 
 		if (win->transient != 0 || win->floating != 0)
 			stack_floater(win, ws->r);
 		else {
 			bzero(&wc, sizeof wc);
 			wc.border_width = 1;
-			win->g.x = wc.x = gg.x;
-			win->g.y = wc.y = gg.y;
-			win->g.w = wc.width = gg.w;
-			win->g.h = wc.height = gg.h;
+			if (rot) {
+				win->g.x = wc.x = win_g.y;
+				win->g.y = wc.y = win_g.x;
+				win->g.w = wc.width = win_g.h;
+				win->g.h = wc.height = win_g.w;
+			} else {
+				win->g.x = wc.x = win_g.x;
+				win->g.y = wc.y = win_g.y;
+				win->g.w = wc.width = win_g.w;
+				win->g.h = wc.height = win_g.h;
+			}
 			mask = CWX | CWY | CWWidth | CWHeight | CWBorderWidth;
 			XConfigureWindow(display, win->id, mask, &wc);
+			/*
+			fprintf(stderr, "vertical_stack: win %d x %d y %d w %d h %d bw %d\n", win->id, win->g.x, win->g.y, win->g.w , win->g.h, wc.border_width);
+			*/
 		}
 
 		XMapRaised(display, win->id);
+		last_h = win_g.h;
 		i++;
 		j++;
 	}
@@ -1209,6 +1329,45 @@ vertical_stack(struct workspace *ws, struct swm_geometry *g) {
 }
 
 void
+vertical_config(struct workspace *ws, int id)
+{
+	DNPRINTF(SWM_D_STACK, "vertical_resize: workspace: %d\n", ws->idx);
+
+	switch (id) {
+	case SWM_ARG_ID_STACKRESET:
+	case SWM_ARG_ID_STACKINIT:
+		ws->l_state.vertical_msize = SWM_V_SLICE / 2;
+		ws->l_state.vertical_mwin = 1;
+		break;
+	case SWM_ARG_ID_MASTERSHRINK:
+		if (ws->l_state.vertical_msize > 1)
+			ws->l_state.vertical_msize--;
+		break;
+	case SWM_ARG_ID_MASTERGROW:
+		if (ws->l_state.vertical_msize < SWM_V_SLICE - 1)
+			ws->l_state.vertical_msize++;
+		break;
+	case SWM_ARG_ID_MASTERADD:
+		ws->l_state.vertical_mwin++;
+		break;
+	case SWM_ARG_ID_MASTERDEL:
+		if (ws->l_state.vertical_mwin > 0)
+			ws->l_state.vertical_mwin--;
+		break;
+	default:
+		return;
+	}
+}
+
+void
+vertical_stack(struct workspace *ws, struct swm_geometry *g)
+{
+	DNPRINTF(SWM_D_STACK, "vertical_stack: workspace: %d\n", ws->idx);
+
+	stack_master(ws, g, 0, 0);
+}
+
+void
 horizontal_config(struct workspace *ws, int id)
 {
 	DNPRINTF(SWM_D_STACK, "horizontal_config: workspace: %d\n", ws->idx);
@@ -1240,74 +1399,11 @@ horizontal_config(struct workspace *ws, int id)
 }
 
 void
-horizontal_stack(struct workspace *ws, struct swm_geometry *g) {
-	XWindowChanges		wc;
-	struct swm_geometry	gg = *g;
-	struct ws_win		*win, *winfocus;
-	int			i, j, split, rowno, hrw, winno, main_height;
-	unsigned int		mask;
-
-	DNPRINTF(SWM_D_STACK, "horizontal_stack: workspace: %d\n", ws->idx);
-
-	if ((winno = count_win(ws, 0)) == 0)
-		return;
-
-	if (ws->focus == NULL)
-		ws->focus = TAILQ_FIRST(&ws->winlist);
-	winfocus = cur_focus ? cur_focus : ws->focus;
-
-	if (ws->l_state.horizontal_mwin &&
-	    winno > ws->l_state.horizontal_mwin) {
-		split = ws->l_state.horizontal_mwin;
-		rowno = split;
-		main_height = (g->h / SWM_V_SLICE) *
-		   ws->l_state.horizontal_msize;
-		gg.h = main_height;
-	} else {
-		rowno = winno;
-		split = 0;
-	}
-	hrw = g->w / rowno;
-	gg.w = hrw - 2;
-
-	i = j = 0;
-	TAILQ_FOREACH(win, &ws->winlist, entry) {
-		if (split && i == split) {
-			rowno = winno - split;
-			hrw = (g->w / rowno);
-			gg.y += main_height + 2;
-			gg.h = g->h - (main_height + 2);
-			gg.w = hrw - 2;
-			j = 0;
-		}
-		if (j == rowno - 1)
-			gg.w = (hrw + (g->w - (rowno * hrw)));
-
-		if (j == 0)
-			gg.x = g->x;
-		else
-			gg.x += hrw;
-
-		if (win->transient != 0 || win->floating != 0)
-			stack_floater(win, ws->r);
-		else {
-			bzero(&wc, sizeof wc);
-			wc.border_width = 1;
-			win->g.x = wc.x = gg.x;
-			win->g.y = wc.y = gg.y;
-			win->g.w = wc.width = gg.w;
-			win->g.h = wc.height = gg.h;
-			mask = CWX | CWY | CWWidth | CWHeight | CWBorderWidth;
-			XConfigureWindow(display, win->id, mask, &wc);
-		}
-
-		XMapRaised(display, win->id);
-		j++;
-		i++;
-	}
+horizontal_stack(struct workspace *ws, struct swm_geometry *g)
+{
+	DNPRINTF(SWM_D_STACK, "vertical_stack: workspace: %d\n", ws->idx);
 
-	if (winfocus)
-		focus_win(winfocus); /* this has to be done outside of the loop */
+	stack_master(ws, g, 1, 0);
 }
 
 /* fullscreen view */
@@ -1362,6 +1458,8 @@ send_to_ws(struct swm_region *r, union arg *args)
 	int			wsid = args->id;
 	struct ws_win		*win = cur_focus;
 	struct workspace	*ws, *nws;
+	Atom			ws_idx_atom = 0;
+	unsigned char		ws_idx_str[SWM_PROPLEN];
 
 	DNPRINTF(SWM_D_MOVE, "send_to_ws: win: %lu\n", win->id);
 
@@ -1382,6 +1480,16 @@ send_to_ws(struct swm_region *r, union arg *args)
 	TAILQ_INSERT_TAIL(&nws->winlist, win, entry);
 	win->ws = nws;
 
+	/* Try to update the window's workspace property */
+	ws_idx_atom = XInternAtom(display, "_SWM_WS", False);
+	if (ws_idx_atom &&
+	    snprintf(ws_idx_str, SWM_PROPLEN, "%d", nws->idx) < SWM_PROPLEN) {
+		DNPRINTF(SWM_D_PROP, "setting property _SWM_WS to %s\n",
+		    ws_idx_str);
+		XChangeProperty(display, win->id, ws_idx_atom, XA_STRING, 8,
+		    PropModeReplace, ws_idx_str, SWM_PROPLEN);
+	}
+
 	if (count_win(nws, 1) == 1)
 		nws->focus = win;
 	ws->restack = 1;
@@ -1398,6 +1506,30 @@ wkill(struct swm_region *r, union arg *args)
 		XKillClient(display, r->ws->focus->id);
 }
 
+void
+screenshot(struct swm_region *r, union arg *args)
+{
+	union arg			a;
+
+	DNPRINTF(SWM_D_MISC, "screenshot\n");
+
+	if (ss_enabled == 0)
+		return;
+
+	switch (args->id) {
+	case SWM_ARG_ID_SS_ALL:
+		spawn_screenshot[1] = "full";
+		break;
+	case SWM_ARG_ID_SS_WINDOW:
+		spawn_screenshot[1] = "window";
+		break;
+	default:
+		return;
+	}
+	a.argv = spawn_screenshot;
+	spawn(r, &a);
+}
+
 /* key definitions */
 struct key {
 	unsigned int		mod;
@@ -1434,6 +1566,8 @@ struct key {
 	{ MODKEY,		XK_0,		switchws,	{.id = 9} },
 	{ MODKEY,		XK_Right,	cyclews,	{.id = SWM_ARG_ID_CYCLEWS_UP} }, 
 	{ MODKEY,		XK_Left,	cyclews,	{.id = SWM_ARG_ID_CYCLEWS_DOWN} }, 
+	{ MODKEY | ShiftMask,	XK_Right,	cyclescr,	{.id = SWM_ARG_ID_CYCLESC_UP} }, 
+	{ MODKEY | ShiftMask,	XK_Left,	cyclescr,	{.id = SWM_ARG_ID_CYCLESC_DOWN} }, 
 	{ MODKEY | ShiftMask,	XK_1,		send_to_ws,	{.id = 0} },
 	{ MODKEY | ShiftMask,	XK_2,		send_to_ws,	{.id = 1} },
 	{ MODKEY | ShiftMask,	XK_3,		send_to_ws,	{.id = 2} },
@@ -1448,6 +1582,8 @@ struct key {
 	{ MODKEY,		XK_Tab,		focus,		{.id = SWM_ARG_ID_FOCUSNEXT} },
 	{ MODKEY | ShiftMask,	XK_Tab,		focus,		{.id = SWM_ARG_ID_FOCUSPREV} },
 	{ MODKEY | ShiftMask,	XK_x,		wkill,		{0} },
+	{ MODKEY,		XK_s,		screenshot,	{.id = SWM_ARG_ID_SS_ALL} },
+	{ MODKEY | ShiftMask,	XK_s,		screenshot,	{.id = SWM_ARG_ID_SS_WINDOW} },
 };
 
 void
@@ -1609,23 +1745,49 @@ set_win_state(struct ws_win *win, long state)
 }
 
 struct ws_win *
-manage_window(Window id, struct workspace *ws)
+manage_window(Window id)
 {
 	Window			trans;
+	struct workspace	*ws;
 	struct ws_win		*win;
 	XClassHint		ch;
+	int			format;
+	unsigned long		nitems, bytes;
+	Atom			ws_idx_atom = 0, type;
+	unsigned char		ws_idx_str[SWM_PROPLEN], *prop = NULL;
+	struct swm_region	*r;
+	long			mask;
 
-	TAILQ_FOREACH(win, &ws->winlist, entry) {
-		if (win->id == id)
+	if ((win = find_window(id)) != NULL)
 			return (win);	/* already being managed */
-	}
 
 	if ((win = calloc(1, sizeof(struct ws_win))) == NULL)
 		errx(1, "calloc: failed to allocate memory for new window");
 
+	ws_idx_atom = XInternAtom(display, "_SWM_WS", False);
+	if (ws_idx_atom)
+		XGetWindowProperty(display, id, ws_idx_atom, 0, SWM_PROPLEN,
+		    False, XA_STRING, &type, &format, &nitems, &bytes, &prop);
+
+	XGetWindowAttributes(display, id, &win->wa);
+	r = root_to_region(win->wa.root);
+	/* If the window was managed before, put it in the same workspace */
+	if (prop) {
+		int		ws_idx;
+		const char	*errstr;
+
+		DNPRINTF(SWM_D_PROP, "got property _SWM_WS=%s\n", prop);
+		ws_idx = strtonum(prop, 0, 9, &errstr);
+		if (errstr)
+			DNPRINTF(SWM_D_EVENT, "window idx is %s: %s",
+			    errstr, prop);
+		ws = &r->s->ws[ws_idx];
+	} else
+		ws = r->ws;
+
 	win->id = id;
 	win->ws = ws;
-	win->s = ws->r->s;	/* this never changes */
+	win->s = r->s;	/* this never changes */
 	TAILQ_INSERT_TAIL(&ws->winlist, win, entry);
 
 	/* make new win focused */
@@ -1637,12 +1799,26 @@ manage_window(Window id, struct workspace *ws)
 		DNPRINTF(SWM_D_MISC, "manage_window: win %u transient %u\n",
 		    (unsigned)win->id, win->transient);
 	}
-	XGetWindowAttributes(display, id, &win->wa);
 	win->g.w = win->wa.width;
 	win->g.h = win->wa.height;
 	win->g.x = win->wa.x;
 	win->g.y = win->wa.y;
 
+	XGetWMNormalHints(display, win->id, &win->sh, &mask);
+
+	if (ws_idx_atom && prop == NULL &&
+	    snprintf(ws_idx_str, SWM_PROPLEN, "%d", ws->idx) < SWM_PROPLEN) {
+		DNPRINTF(SWM_D_PROP, "setting property _SWM_WS to %s\n",
+		    ws_idx_str);
+		XChangeProperty(display, win->id, ws_idx_atom, XA_STRING, 8,
+		    PropModeReplace, ws_idx_str, SWM_PROPLEN);
+	}
+	XFree(prop);
+
+	/*
+	fprintf(stderr, "manage window: %d x %d y %d w %d h %d\n", win->id, win->g.x, win->g.y, win->g.w, win->g.h);
+	*/
+
 	/* XXX make this a table */
 	bzero(&ch, sizeof ch);
 	if (XGetClassHint(display, win->id, &ch)) {
@@ -1669,7 +1845,7 @@ configurerequest(XEvent *e)
 {
 	XConfigureRequestEvent	*ev = &e->xconfigurerequest;
 	struct ws_win		*win;
-	int			new = 1;
+	int			new = 0;
 	XWindowChanges		wc;
 
 	if ((win = find_window(ev->window)) == NULL)
@@ -1678,6 +1854,10 @@ configurerequest(XEvent *e)
 	if (new) {
 		DNPRINTF(SWM_D_EVENT, "configurerequest: new window: %lu\n",
 		    ev->window);
+		/*
+		fprintf(stderr, "configurerequest: new window: %lu x %d y %d w %d h %d bw %d s %d sm %d\n",
+		    ev->window, ev->x, ev->y, ev->width, ev->height, ev->border_width, ev->above, ev->detail);
+		*/
 		bzero(&wc, sizeof wc);
 		wc.x = ev->x;
 		wc.y = ev->y;
@@ -1688,6 +1868,10 @@ configurerequest(XEvent *e)
 		wc.stack_mode = ev->detail;
 		XConfigureWindow(display, ev->window, ev->value_mask, &wc);
 	} else {
+		/*
+		fprintf(stderr, "configurerequest: change window: %lu\n",
+		    ev->window);
+		*/
 		DNPRINTF(SWM_D_EVENT, "configurerequest: change window: %lu\n",
 		    ev->window);
 		if (win->floating) {
@@ -1745,16 +1929,16 @@ destroynotify(XEvent *e)
 			ws->focus = TAILQ_PREV(win, ws_win_list, entry);
 		if (ws->focus == NULL)
 			ws->focus = TAILQ_FIRST(&ws->winlist);
-		if (ws->focus == win)
+		if (ws->focus == NULL || ws->focus == win) {
 			ws->focus = NULL;
-		if (cur_focus == win)
+			unfocus_all();
+		} else
 			focus_win(ws->focus);
-
 		TAILQ_REMOVE(&ws->winlist, win, entry);
 		set_win_state(win, WithdrawnState);
 		free(win);
+		stack();
 	}
-	stack();
 }
 
 void
@@ -1798,7 +1982,6 @@ maprequest(XEvent *e)
 {
 	XMapRequestEvent	*ev = &e->xmaprequest;
 	XWindowAttributes	wa;
-	struct swm_region	*r;
 
 	DNPRINTF(SWM_D_EVENT, "maprequest: window: %lu\n",
 	    e->xmaprequest.window);
@@ -1807,8 +1990,7 @@ maprequest(XEvent *e)
 		return;
 	if (wa.override_redirect)
 		return;
-	r = root_to_region(wa.root);
-	manage_window(e->xmaprequest.window, r->ws);
+	manage_window(e->xmaprequest.window);
 	stack();
 }
 
@@ -1947,11 +2129,16 @@ setup_screens(void)
 	int			ncrtc = 0, w = 0;
         int			i, j, k;
 	struct workspace	*ws;
+	int			ws_idx_atom;
+
 
 	if ((screens = calloc(ScreenCount(display),
 	     sizeof(struct swm_screen))) == NULL)
 		errx(1, "calloc: screens");
 
+	ws_idx_atom = XInternAtom(display, "_SWM_WS", False);
+	
+
 	/* map physical screens */
 	for (i = 0; i < ScreenCount(display); i++) {
 		DNPRINTF(SWM_D_WS, "setup_screens: init screen %d\n", i);
@@ -2010,7 +2197,7 @@ setup_screens(void)
 		else 
 			ncrtc = sr->ncrtc;
 
-		for (c = 0; c < ncrtc; c++) {
+		for (c = 0, ci = NULL; c < ncrtc; c++) {
 			ci = XRRGetCrtcInfo(display, sr, sr->crtcs[c]);
 			if (ci->noutput == 0)
 				continue;
@@ -2024,7 +2211,8 @@ setup_screens(void)
 				    ci->x, ci->y, ci->width, ci->height);
 			w++;
 		}
-		XRRFreeCrtcInfo(ci);
+		if (ci)
+			XRRFreeCrtcInfo(ci);
 		XRRFreeScreenResources(sr);
 #else
 		new_region(&screens[i], &screens[i].ws[w], 0, 0,
@@ -2046,7 +2234,7 @@ setup_screens(void)
 
 			if (wa.map_state == IsViewable ||
 			    getstate(wins[i]) == NormalState)
-				manage_window(wins[i], r->ws);
+				manage_window(wins[i]);
 		}
 		/* transient windows */
 		for (i = 0; i < no; i++) {
@@ -2056,7 +2244,7 @@ setup_screens(void)
 			if (XGetTransientForHint(display, wins[i], &d1) &&
 			    (wa.map_state == IsViewable || getstate(wins[i]) ==
 			    NormalState))
-				manage_window(wins[i], r->ws);
+				manage_window(wins[i]);
                 }
                 if (wins) {
                         XFree(wins);