]> git.sur5r.net Git - i3/i3/blob - src/tree.c
Go down the tree when moving windows, add testcase for moving
[i3/i3] / src / tree.c
1 /*
2  * vim:ts=4:sw=4:expandtab
3  */
4
5 #include "all.h"
6
7 struct Con *croot;
8 struct Con *focused;
9
10 struct all_cons_head all_cons = TAILQ_HEAD_INITIALIZER(all_cons);
11
12 /*
13  * Loads tree from ~/.i3/_restart.json
14  *
15  */
16 bool tree_restore() {
17     char *globbed = glob_path("~/.i3/_restart.json");
18
19     if (!path_exists(globbed)) {
20         LOG("%s does not exist, not restoring tree\n", globbed);
21         free(globbed);
22         return false;
23     }
24
25     /* TODO: refactor the following */
26     croot = con_new(NULL);
27     focused = croot;
28
29     tree_append_json(globbed);
30     char *old_restart = glob_path("~/.i3/_restart.json.old");
31     unlink(old_restart);
32     rename(globbed, old_restart);
33     free(globbed);
34     free(old_restart);
35
36     printf("appended tree, using new root\n");
37     croot = TAILQ_FIRST(&(croot->nodes_head));
38     printf("new root = %p\n", croot);
39     Con *out = TAILQ_FIRST(&(croot->nodes_head));
40     printf("out = %p\n", out);
41     Con *ws = TAILQ_FIRST(&(out->nodes_head));
42     printf("ws = %p\n", ws);
43     con_focus(ws);
44
45     return true;
46 }
47
48 /*
49  * Initializes the tree by creating the root node, adding all RandR outputs
50  * to the tree (that means randr_init() has to be called before) and
51  * assigning a workspace to each RandR output.
52  *
53  */
54 void tree_init() {
55     Output *output;
56
57     croot = con_new(NULL);
58     croot->name = "root";
59     croot->type = CT_ROOT;
60
61     Con *ws;
62     /* add the outputs */
63     TAILQ_FOREACH(output, &outputs, outputs) {
64         if (!output->active)
65             continue;
66
67         Con *oc = con_new(croot);
68         oc->name = strdup(output->name);
69         oc->type = CT_OUTPUT;
70         oc->rect = output->rect;
71
72         /* add a workspace to this output */
73         ws = con_new(oc);
74         ws->type = CT_WORKSPACE;
75         ws->name = strdup("1");
76         ws->fullscreen_mode = CF_OUTPUT;
77         ws->orientation = HORIZ;
78     }
79
80     con_focus(ws);
81 }
82
83 /*
84  * Opens an empty container in the current container
85  *
86  */
87 Con *tree_open_con(Con *con) {
88     if (con == NULL) {
89         /* every focusable Con has a parent (outputs have parent root) */
90         con = focused->parent;
91         /* If the parent is an output, we are on a workspace. In this case,
92          * the new container needs to be opened as a leaf of the workspace. */
93         if (con->type == CT_OUTPUT)
94             con = focused;
95     }
96
97     assert(con != NULL);
98
99     /* 3: re-calculate child->percent for each child */
100     con_fix_percent(con, WINDOW_ADD);
101
102     /* 4: add a new container leaf to this con */
103     Con *new = con_new(con);
104     con_focus(new);
105
106     return new;
107 }
108
109 /*
110  * vanishing is the container that is about to be closed (so any floating
111  * client which has old_parent == vanishing needs to be "re-parented").
112  *
113  */
114 static void fix_floating_parent(Con *con, Con *vanishing) {
115     Con *child;
116
117     if (con->old_parent == vanishing) {
118         LOG("Fixing vanishing old_parent (%p) of container %p to be %p\n",
119                 vanishing, con, vanishing->parent);
120         con->old_parent = vanishing->parent;
121     }
122
123     TAILQ_FOREACH(child, &(con->floating_head), floating_windows)
124         fix_floating_parent(child, vanishing);
125
126     TAILQ_FOREACH(child, &(con->nodes_head), nodes)
127         fix_floating_parent(child, vanishing);
128 }
129
130 /*
131  * Closes the given container including all children
132  *
133  */
134 void tree_close(Con *con, bool kill_window) {
135     /* check floating clients and adjust old_parent if necessary */
136     fix_floating_parent(croot, con);
137
138     /* Get the container which is next focused */
139     Con *next;
140     if (con->type == CT_FLOATING_CON) {
141         next = TAILQ_NEXT(con, floating_windows);
142         if (next == TAILQ_END(&(con->parent->floating_head)))
143             next = con->parent;
144     } else {
145         next = TAILQ_NEXT(con, focused);
146         if (next == TAILQ_END(&(con->parent->nodes_head)))
147             next = con->parent;
148     }
149
150     DLOG("closing %p, kill_window = %d\n", con, kill_window);
151     Con *child;
152     /* We cannot use TAILQ_FOREACH because the children get deleted
153      * in their parent’s nodes_head */
154     while (!TAILQ_EMPTY(&(con->nodes_head))) {
155         child = TAILQ_FIRST(&(con->nodes_head));
156         tree_close(child, kill_window);
157     }
158
159     if (con->window != NULL) {
160         if (kill_window)
161             x_window_kill(con->window->id);
162         else {
163             /* un-parent the window */
164             xcb_reparent_window(conn, con->window->id, root, 0, 0);
165             /* TODO: client_unmap to set state to withdrawn */
166
167         }
168         free(con->window);
169     }
170
171     /* kill the X11 part of this container */
172     x_con_kill(con);
173
174     con_detach(con);
175     con_fix_percent(con->parent, WINDOW_REMOVE);
176
177     if (con_is_floating(con)) {
178         DLOG("Container was floating, killing floating container\n");
179
180         TAILQ_REMOVE(&(con->parent->parent->floating_head), con->parent, floating_windows);
181         TAILQ_REMOVE(&(con->parent->parent->focus_head), con->parent, focused);
182         tree_close(con->parent, false);
183     }
184
185     free(con->name);
186     TAILQ_REMOVE(&all_cons, con, all_cons);
187     free(con);
188
189     /* TODO: check if the container (or one of its children) was focused */
190     con_focus(next);
191 }
192
193 void tree_close_con() {
194     assert(focused != NULL);
195     if (focused->type == CT_WORKSPACE) {
196         LOG("Cannot close workspace\n");
197         return;
198     }
199
200     /* Kill con */
201     tree_close(focused, true);
202 }
203
204 /*
205  * Splits (horizontally or vertically) the given container by creating a new
206  * container which contains the old one and the future ones.
207  *
208  */
209 void tree_split(Con *con, orientation_t orientation) {
210     /* for a workspace, we just need to change orientation */
211     if (con->type == CT_WORKSPACE) {
212         DLOG("Workspace, simply changing orientation to %d\n", orientation);
213         con->orientation = orientation;
214         return;
215     }
216
217     Con *parent = con->parent;
218     /* if we are in a container whose parent contains only one
219      * child and has the same orientation like we are trying to
220      * set, this operation is a no-op to not confuse the user */
221     if (parent->orientation == orientation &&
222         TAILQ_NEXT(con, nodes) == TAILQ_END(&(parent->nodes_head))) {
223         DLOG("Not splitting the same way again\n");
224         return;
225     }
226
227     DLOG("Splitting in orientation %d\n", orientation);
228
229     /* 2: replace it with a new Con */
230     Con *new = con_new(NULL);
231     TAILQ_REPLACE(&(parent->nodes_head), con, new, nodes);
232     TAILQ_REPLACE(&(parent->focus_head), con, new, focused);
233     new->parent = parent;
234     new->orientation = orientation;
235
236     /* 3: add it as a child to the new Con */
237     con_attach(con, new);
238 }
239
240 void level_up() {
241     /* We can focus up to the workspace, but not any higher in the tree */
242     if (focused->parent->type != CT_CON &&
243         focused->parent->type != CT_WORKSPACE) {
244         printf("cannot go up\n");
245         return;
246     }
247     con_focus(focused->parent);
248 }
249
250 void level_down() {
251     /* Go down the focus stack of the current node */
252     Con *next = TAILQ_FIRST(&(focused->focus_head));
253     if (next == TAILQ_END(&(focused->focus_head))) {
254         printf("cannot go down\n");
255         return;
256     }
257     con_focus(next);
258 }
259
260 static void mark_unmapped(Con *con) {
261     Con *current;
262
263     con->mapped = false;
264     TAILQ_FOREACH(current, &(con->nodes_head), nodes)
265         mark_unmapped(current);
266 }
267
268 void tree_render() {
269     if (croot == NULL)
270         return;
271
272     printf("-- BEGIN RENDERING --\n");
273     /* Reset map state for all nodes in tree */
274     /* TODO: a nicer method to walk all nodes would be good, maybe? */
275     mark_unmapped(croot);
276     croot->mapped = true;
277
278     /* We start rendering at an output */
279     Con *output;
280     TAILQ_FOREACH(output, &(croot->nodes_head), nodes) {
281         printf("output %p / %s\n", output, output->name);
282         render_con(output);
283     }
284     x_push_changes(croot);
285     printf("-- END RENDERING --\n");
286 }
287
288 void tree_next(char way, orientation_t orientation) {
289     /* 1: get the first parent with the same orientation */
290     Con *parent = focused->parent;
291     while (parent->orientation != orientation) {
292         LOG("need to go one level further up\n");
293         /* if the current parent is an output, we are at a workspace
294          * and the orientation still does not match */
295         if (parent->type == CT_WORKSPACE)
296             return;
297         parent = parent->parent;
298     }
299     Con *current = TAILQ_FIRST(&(parent->focus_head));
300     assert(current != TAILQ_END(&(parent->focus_head)));
301
302     /* 2: chose next (or previous) */
303     Con *next;
304     if (way == 'n') {
305         next = TAILQ_NEXT(current, nodes);
306         /* if we are at the end of the list, we need to wrap */
307         if (next == TAILQ_END(&(parent->nodes_head)))
308             next = TAILQ_FIRST(&(parent->nodes_head));
309     } else {
310         next = TAILQ_PREV(current, nodes_head, nodes);
311         /* if we are at the end of the list, we need to wrap */
312         if (next == TAILQ_END(&(parent->nodes_head)))
313             next = TAILQ_LAST(&(parent->nodes_head), nodes_head);
314     }
315
316     /* 3: focus choice comes in here. at the moment we will go down
317      * until we find a window */
318     /* TODO: check for window, atm we only go down as far as possible */
319     while (!TAILQ_EMPTY(&(next->focus_head)))
320         next = TAILQ_FIRST(&(next->focus_head));
321
322     con_focus(next);
323 }
324
325 void tree_move(char way, orientation_t orientation) {
326     /* 1: get the first parent with the same orientation */
327     Con *parent = focused->parent;
328     if (focused->type == CT_WORKSPACE)
329         return;
330     bool level_changed = false;
331     while (parent->orientation != orientation) {
332         LOG("need to go one level further up\n");
333         /* if the current parent is an output, we are at a workspace
334          * and the orientation still does not match */
335         if (parent->type == CT_WORKSPACE)
336             return;
337         parent = parent->parent;
338         level_changed = true;
339     }
340     Con *current = TAILQ_FIRST(&(parent->focus_head));
341     assert(current != TAILQ_END(&(parent->focus_head)));
342
343     /* 2: chose next (or previous) */
344     Con *next = current;
345     if (way == 'n') {
346         LOG("i would insert it after %p / %s\n", next, next->name);
347         if (!level_changed) {
348             next = TAILQ_NEXT(next, nodes);
349             if (next == TAILQ_END(&(next->parent->nodes_head))) {
350                 LOG("cannot move further to the right\n");
351                 return;
352             }
353
354             /* if this is a split container, we need to go down */
355             while (!TAILQ_EMPTY(&(next->focus_head)))
356                 next = TAILQ_FIRST(&(next->focus_head));
357         }
358
359         con_detach(focused);
360         focused->parent = next->parent;
361
362         TAILQ_INSERT_AFTER(&(next->parent->nodes_head), next, focused, nodes);
363         TAILQ_INSERT_HEAD(&(next->parent->focus_head), focused, focused);
364         /* TODO: don’t influence focus handling? */
365     } else {
366         LOG("i would insert it before %p / %s\n", current, current->name);
367         bool gone_down = false;
368         if (!level_changed) {
369             next = TAILQ_PREV(next, nodes_head, nodes);
370             if (next == TAILQ_END(&(next->parent->nodes_head))) {
371                 LOG("cannot move further\n");
372                 return;
373             }
374
375             /* if this is a split container, we need to go down */
376             while (!TAILQ_EMPTY(&(next->focus_head))) {
377                 gone_down = true;
378                 next = TAILQ_FIRST(&(next->focus_head));
379             }
380         }
381
382         con_detach(focused);
383         focused->parent = next->parent;
384
385         /* After going down in the tree, we insert the container *after*
386          * the currently focused one even though the command used "before".
387          * This is to keep the user experience clear, since the before/after
388          * only signifies the direction of the movement on top-level */
389         if (gone_down)
390             TAILQ_INSERT_AFTER(&(next->parent->nodes_head), next, focused, nodes);
391         else TAILQ_INSERT_BEFORE(next, focused, nodes);
392         TAILQ_INSERT_HEAD(&(next->parent->focus_head), focused, focused);
393         /* TODO: don’t influence focus handling? */
394     }
395 }