faf575387782bdc9ce0d41fa53335a5dc551eb57
[p5sagit/Devel-Size.git] / static / MemView.pl
1 #!/usr/bin/env perl
2
3 use strict;
4 use warnings;
5
6 use JSON::XS;
7 use Mojolicious::Lite;
8 use Getopt::Long;
9 use Storable qw(dclone);
10
11 GetOptions(
12     'db=s' => \(my $opt_db = '../memnodes.db'),
13     'debug!' => \my $opt_debug,
14 ) or exit 1;
15
16 #warn "Reading from $opt_db\n";
17
18 use ORLite {
19     file => '../memnodes.db',
20     package => "MemView",
21     #user_version => 1,
22     readonly => 1,
23     #unicode => 1,
24 };
25
26 my $j = JSON::XS->new;
27
28 # Documentation browser under "/perldoc"
29 plugin 'PODRenderer';
30
31 get '/' => sub {
32     my $self = shift;
33     $self->render('index');
34 };
35
36 get '/jit_tree/:id/:depth' => sub {
37     my $self = shift;
38     my $logarea = $self->param('logarea');
39
40     my $id = $self->stash('id');
41     my $depth = $self->stash('depth');
42     my $node_tree = _fetch_node_tree($id, $depth);
43     my $jit_tree = _transform_node_tree($node_tree, sub {
44         my ($node) = @_;
45         my $children = delete $node->{children}; # XXX edits the src tree
46         my $area = $node->{self_size}+$node->{kids_size};
47         $node->{'$area'} = ($logarea) ? log($area) : $area; # XXX move to jit js
48         my $jit_node = {
49             id   => $node->{id},
50             name => $node->{title} || $node->{name},
51             data => $node,
52         };
53         $jit_node->{children} = $children if $children;
54         return $jit_node;
55     });
56 if(1){
57     use Devel::Dwarn;
58     use Data::Dump qw(pp);
59     local $jit_tree->{children};
60     pp(dclone($jit_tree)); # dclone to avoid stringification
61 }
62     $self->render_json($jit_tree);
63 };
64
65 sub _fetch_node_tree {
66     my ($id, $depth) = @_;
67     my $node = MemView->selectrow_hashref("select * from node where id = ?", undef, $id)
68         or die "Node '$id' not found";
69     $node->{$_} += 0 for (qw(child_count kids_node_count kids_size self_size));
70     $node->{leaves} = $j->decode(delete $node->{leaves_json});
71     $node->{attr}   = $j->decode(delete $node->{attr_json});
72     $node->{name} .= "->" if $node->{type} == 2 && $node->{name};
73
74     if ($node->{child_ids}) {
75         my @child_ids = split /,/, $node->{child_ids};
76         my $children;
77         if (@child_ids == 1 && $node->{type} == 2) {
78             my $child = _fetch_node_tree($child_ids[0], $depth); # same depth
79             # merge node into child
80             # XXX id, depth, parent_id
81             warn "Merged $node->{name} #$node->{id} with only child $child->{name} #$child->{id}\n"
82                 if $opt_debug;
83             $child->{name} = "$node->{name} $child->{name}";
84             $child->{$_} += $node->{$_} for (qw(self_size));
85             $child->{$_}  = $node->{$_} for (qw(parent_id));
86
87             $child->{title} = join " ", grep { defined && length } $child->{title}, $node->{title};
88             #warn "Titled $child->{title}" if $child->{title};
89
90             for my $attr_type (keys %{ $node->{attr} }) {
91                 my $src = $node->{attr}{$attr_type};
92                 if (ref $src eq 'HASH') { # eg NPattr_NAME: {attr}{1}{$name} = $value
93                     my $dst = $child->{attr}{$attr_type} ||= {};
94                     for my $k (keys %$src) {
95                         warn "Node $child->{id} attr $attr_type:$k=$dst->{$k} overwritten by $src->{$k}\n"
96                             if defined $dst->{$k};
97                         $dst->{$k} = $src->{$k};
98                     }
99                 }
100                 elsif (ref $src eq 'ARRAY') { # eg NPattr_PADNAME: {attr}{2}[$val] = $name
101                     my $dst = $child->{attr}{$attr_type} ||= [];
102                     my $idx = @$src;
103                     while (--$idx >= 0) {
104                         warn "Node $child->{id} attr $attr_type:$idx=$dst->[$idx] overwritten by $src->[$idx]\n"
105                             if defined $dst->[$idx];
106                         $dst->[$idx] = $src->[$idx];
107                     }
108                 }
109                 else { # assume scalar
110                     warn "Node $child->{id} attr $attr_type=$child->{attr}{$attr_type} overwritten by $src\n"
111                         if exists $child->{attr}{$attr_type};
112                     $child->{attr}{$attr_type} = $src;
113                 }
114             }
115
116             $child->{leaves}{$_} += $node->{leaves}{$_}
117                 for keys %{ $node->{leaves} };
118
119             $child->{_ids_merged} .= ",$node->{id}";
120             my @child_ids = split /,/, $node->{child_ids};
121             $child->{child_count} = @child_ids;
122
123             $node = $child;
124         }
125         elsif ($depth) {
126             $children = [ map { _fetch_node_tree($_, $depth-1) } @child_ids ];
127             $node->{children} = $children;
128             $node->{child_count} = @$children;
129         }
130     }
131     return $node;
132 }
133
134 sub _transform_node_tree {  # recurse depth first
135     my ($node, $transform) = @_;
136     if (my $children = $node->{children}) {
137         $_ = _transform_node_tree($_, $transform) for @$children;
138     }
139     return $transform->($node);
140 }
141
142
143 app->start;
144 __DATA__
145 @@ index.html.ep
146 % layout 'default';
147 % title 'Welcome';
148 Welcome to the Mojolicious real-time web framework!
149
150 @@ layouts/default.html.ep
151 <!DOCTYPE html>
152 <head>
153 <meta http-equiv="Content-Type" content="text/html; charset=UTF-8" />
154 <title>Perl Memory Treemap</title>
155
156 <!-- CSS Files -->
157 <link type="text/css" href="css/base.css" rel="stylesheet" />
158 <link type="text/css" href="css/Treemap.css" rel="stylesheet" />
159
160 <!--[if IE]><script language="javascript" type="text/javascript" src="excanvas.js"></script><![endif]-->
161
162 <!-- JIT Library File -->
163 <script language="javascript" type="text/javascript" src="jit.js"></script>
164 <script language="javascript" type="text/javascript" src="jquery-1.8.1-min.js"></script>
165
166 <!-- Example File -->
167 <script language="javascript" type="text/javascript" src="sprintf.js"></script>
168 <script language="javascript" type="text/javascript" src="tm.js"></script>
169 </head>
170
171 <body onload="init();">
172 <div id="container">
173
174 <div id="left-container">
175
176 <div class="text">
177 <h4>
178 Perl Memory TreeMap
179 </h4> 
180     Click on a node to zoom in.<br /><br />            
181 </div>
182
183 <a id="back" href="#" class="theme button white">Go to Parent</a>
184
185 <br />
186 <form name=params>
187 <label for="logarea">&nbsp;Logarithmic scale
188 <input type=checkbox id="logarea" name="logarea">
189 </form>
190
191 </div>
192
193 <div id="center-container">
194     <div id="infovis"></div>    
195 </div>
196
197 <div id="right-container">
198
199 <div id="inner-details"></div>
200
201 </div>
202
203 <div id="log"></div>
204 </div>
205 </body>
206 </html>