9956168612fe29ce5ba3903b1eda95d3a109189e
[perl/schtarr.git] / graphplay
1 #!/usr/bin/env perl
2 use strict;
3 use warnings;
4
5 use Template;
6 use List::Util;
7 use Data::Dumper;
8
9 my %area = (
10         fieldxmin => 45,
11         fieldxmax => 1590,
12         fieldymin => 10,
13         fieldymax => 575,
14         xmax => 1600,
15         ymax => 600,
16 );
17 $area{fieldxsize} = $area{fieldxmax} - $area{fieldxmin};
18 $area{fieldysize} = $area{fieldymax} - $area{fieldymin};
19
20 my @lines;
21 my @max = (0) x 4;
22 my $maxtime = 0;
23 my $player = 0;
24
25 for my $input (@ARGV) {
26         open my $datafile, '<', $input or next;
27         my @data;
28         my $lasttime;
29         while (defined ($_ = readline $datafile)) {
30                 my ($frame, $time, @vars) = split /\t/, $_;
31                 $frame =~ /^\d+:$/ or next;
32                 if ($time !~ /^\d+$/) {
33                         defined $lasttime or next;
34                         $time = $lasttime + 1;
35                         $lasttime = undef;
36                 } else {
37                         $lasttime = $time;
38                 }
39                 /^\d+$/ or $_ = undef for @vars;
40                 push @{$data[$time]} => \@vars;
41                 defined $vars[$_] and $vars[$_] > $max[$_] and $max[$_] = $vars[$_]
42                         for 0 .. $#vars;
43         }
44         $maxtime = $#data if $#data > $maxtime;
45         close $datafile;
46
47         for my $time (0 .. $#data) {
48                 my $vals = $data[$time];
49                 for my $subtime (0 .. $#$vals) {
50                         defined $vals->[$subtime]->[$_] and push @{$lines[$_][$player]} => [
51                                 ($time + $subtime/@$vals), # x
52                                 $vals->[$subtime]->[$_] # y
53                         ] for 0 .. 3;
54                 }
55         }
56
57         for (0 .. 2) {
58                 my $line = $lines[$_][$player];
59                 my $start;
60                 my $lasty;
61                 for (my $i = 1; $i <= $#$line; $i++) {
62                         defined $line->[$i] or next;
63                         if ($line->[$i][1] == $line->[$i - 1][1]) {
64                                 $start = $i unless defined $start;
65                         } elsif (defined $start) {
66                                 delete @$line[$start .. $i-2];
67                                 undef $start;
68                         }
69                 }
70                 delete @$line[$start .. $#$line-2] if defined $start; # or -1
71         }
72
73         $player++;
74 }
75
76 my @norm = ((map {$area{fieldysize} / $_} @max), $area{fieldxsize} / $maxtime);
77
78 my $xsteps = 25;
79 my $xstep = $maxtime / $xsteps;
80 my @xaxis = map [$area{fieldxmin} + $_ * $norm[4], sprintf '%d:%02d', $_/3600, $_/60%60],
81         map $_*$xstep, 0 .. $xsteps-1;
82
83 my $ysteps = 10;
84 my @yaxis;
85 for my $i (0 .. 3) {
86         my $ystep = $max[$i] / $ysteps;
87         $ystep = 10 ** int (log($ystep * 4) / log(10));
88         $ystep /= 2 while $ystep * $ysteps > $max[$i];
89         $yaxis[$i] = [
90                 map [$area{fieldymax} - $_ * $norm[$i], $_],
91                         map $_*$ystep, 0 .. int $max[$i] / $ystep
92         ];
93 }
94
95 undef $/;
96 my $svg = Template->new({POST_CHOMP => 1, BLOCKS => {timeline => readline DATA}});
97
98 for ([0 => "min", "Minerals"], [1 => "gas", "Gas"], [2 => "unit", "Units"]) {
99         my ($line, $filename, $title) = @$_;
100         open my $output, '>', "$filename.svg";
101         $svg->process('timeline', {
102                 title => "$title timeline for /replay/20080201md4x4",
103                 paths => [
104                         map { "\n\tM " . join "\tL ", map sprintf("%s %s\n",
105                                 $area{fieldxmin} + $_->[0] * $norm[4],
106                                 $area{fieldymax} - $_->[1] * $norm[$line]
107                         ), grep defined, @$_ } @{$lines[$line]}
108                 ],
109                 xaxis => \@xaxis,
110                 yaxis => $yaxis[$line],
111                 area => \%area,
112         }, $output) or die $svg->error;
113 }
114
115 __DATA__
116 <?xml version="1.0"?>
117 <!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.0//EN"
118  "http://www.w3.org/TR/2001/REC-SVG-20010904/DTD/svg10.dtd">
119
120 <?xml-stylesheet href="http://sc.shiar.net/apm.css" type="text/css"?>
121
122 <svg width="100%" height="100%" viewBox="0 0 [% area.xmax %] [% area.ymax %]"
123  xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink">
124 <title>[% title %]</title>
125
126 <rect x="0" y="0" width="[% area.xmax %]" height="[% area.ymax %]" class="svgBackground" />
127 <rect x="[% area.fieldxmin %]" y="[% area.fieldymin %]" width="[% area.fieldxsize %]" height="[% area.fieldysize %]" class="graphBackground" />
128
129 <path d="M[% area.fieldxmin %] [% area.fieldymin %] v[% area.fieldysize %]" class="axis" id="xAxis"/>
130 [% FOREACH entry = xaxis %]
131 <text x="[% entry.0 %]" y="[% area.fieldymax + 15 %]" class="xAxisLabels"> [% entry.1 %]</text>
132 [% END %]
133
134 <path d="M[% area.fieldxmin %] [% area.fieldymax %] h[% area.fieldxsize %]" class="axis" id="yAxis"/>
135 [% FOREACH entry = yaxis %]
136 <path d="M[% area.fieldxmin %] [% entry.0 %] h[% area.fieldxsize %]" class="guideLines" />
137 <text x="[% area.fieldxmin - 5 %]" y="[% entry.0 %]" class="yAxisLabels">[% entry.1 %]</text>
138 [% END %]
139
140 <g class="data">
141 [% count = 0 %]
142 [% FOREACH path = paths %][% count = count + 1 %]
143 <path d="[% path %]" class="line[% count %]" />
144 [% END %]
145 </g>
146
147 </svg>
148