]> git.sur5r.net Git - i3/i3/blob - contrib/dump-asy.pl
travis: clean old docs before building with asciidoc git config
[i3/i3] / contrib / dump-asy.pl
1 #!/usr/bin/env perl
2 # vim:ts=4:sw=4:expandtab
3 # renders the layout tree using asymptote
4 #
5 # ./dump-asy.pl
6 #   will render the entire tree
7 # ./dump-asy.pl 'name'
8 #   will render the tree starting from the node with the specified name,
9 #   e.g. ./dump-asy.pl 2 will render workspace 2 and below
10
11 use strict;
12 use warnings;
13 use Data::Dumper;
14 use AnyEvent::I3;
15 use File::Temp;
16 use v5.10;
17
18 my $i3 = i3();
19
20 my $tree = $i3->get_tree->recv;
21
22 my $tmp = File::Temp->new(UNLINK => 0, SUFFIX => '.asy');
23
24 say $tmp "import drawtree;";
25
26 say $tmp "treeLevelStep = 2cm;";
27
28 sub dump_node {
29         my ($n, $parent) = @_;
30
31     my $o = ($n->{orientation} eq 'none' ? "u" : ($n->{orientation} eq 'horizontal' ? "h" : "v"));
32     my $w = (defined($n->{window}) ? $n->{window} : "N");
33     my $na = $n->{name};
34     $na =~ s/#/\\#/g;
35     $na =~ s/\$/\\\$/g;
36     $na =~ s/&/\\&/g;
37     $na =~ s/_/\\_/g;
38     $na =~ s/~/\\textasciitilde{}/g;
39     my $type = 'leaf';
40     if (!defined($n->{window})) {
41         $type = $n->{orientation} . '-split';
42     }
43     my $name = qq|``$na'' ($type)|;
44
45     print $tmp "TreeNode n" . $n->{id} . " = makeNode(";
46
47     print $tmp "n" . $parent->{id} . ", " if defined($parent);
48     print $tmp "\"" . $name . "\");\n";
49
50         dump_node($_, $n) for @{$n->{nodes}};
51 }
52
53 sub find_node_with_name {
54     my ($node, $name) = @_;
55
56     return $node if ($node->{name} eq $name);
57     for my $child (@{$node->{nodes}}) {
58         my $res = find_node_with_name($child, $name);
59         return $res if defined($res);
60     }
61     return undef;
62 }
63
64 my $start = shift;
65 my $root;
66 if ($start) {
67     # Find the specified node in the tree
68     $root = find_node_with_name($tree, $start);
69 } else {
70     $root = $tree;
71 }
72 dump_node($root);
73 say $tmp "draw(n" . $root->{id} . ", (0, 0));";
74
75 close($tmp);
76 my $rep = "$tmp";
77 $rep =~ s/asy$/eps/;
78 system("cd /tmp && asy $tmp && gv --scale=-1000 --noresize --widgetless $rep && rm $rep");