document sample usage in scmap
[perl/schtarr.git] / scmap
1 #!/usr/bin/perl
2 use strict;
3 use warnings;
4 use Data::Dumper;
5
6 my $SHOWWARN = 0;
7 my $SHOWMAP = "head"; # ascii, num, ppm
8 my $SHOWCOL = 0;
9
10 use Getopt::Long;
11 GetOptions(
12         "verbose|v!" => \$SHOWWARN,
13         "map|m=s" => \$SHOWMAP,
14         "color|c" => \$SHOWCOL,
15 );
16
17 use Data::StarCraft::Map;
18 my $map = Data::StarCraft::Map->new->open(\*STDIN);
19 $map->{DEBUG} = 1 if $SHOWWARN;
20
21 if ($SHOWMAP ne "ppm") {
22         printf("%s size %dx%d, %d player\n",
23                 $map->version,
24                 $map->info->{x}, $map->info->{y},
25                 scalar grep {$_->{id} == 214} $map->units,
26         );
27         print "\n";
28 }
29
30 if ($SHOWMAP eq "head") {
31         if ($map->{STR}) {
32                 my @str = split /\0/, substr $map->{STR}, 2051;
33                 $SHOWCOL ? (
34                         s/([\001-\007])/sprintf '\e[0;%dm', 30+ord($1)/eg
35                         and $_ .= "\e[0;37m"
36                 ) : s/[\001-\017]//g, print "* $_\n" for @str;
37                 print "\n";
38         }
39         printf "%-4s %d\n", $_, defined $map->{$_} ? length $map->{$_} : 0
40                 for sort keys %$map;
41         print "\n";
42         printf "%s: %s\n", $_, join ",", unpack "C*", $map->{$_}
43                 for sort grep { defined $map->{$_} and length $map->{$_} < 32 }
44                         keys %$map;
45         print "\n";
46 }
47
48 sub world {
49         my $self = shift;
50         # ERA:          0        1        2       3        4      5      6   7
51         my @worlds = qw(badlands platform install ashworld jungle desert ice twilight);
52         return $worlds[$self->era] || "?";
53 }
54
55 #        MTXM TILE
56 # ?      ?    yes
57 # v205 = yes  no
58
59 my %mapsep = (
60         num => ',',
61         ppm => '  ',
62         ascii => '',
63 );
64
65 if (defined $mapsep{$SHOWMAP}) {
66         my $MAPCHARSEP = $mapsep{$SHOWMAP};
67
68         my $tiles = $SHOWMAP eq "num" ? [ map sprintf('%5d', $_), @{$map->tiles} ]
69                 : $map->tiles;#_parsed;
70
71         if ($SHOWMAP eq "ppm") {
72                 use Data::StarCraft::Tileset;
73                 my $era = world($map);
74                 if ($era eq '?') {
75                         warn 'Tileset '.$map->era.' not recognized; fallback to jungle';
76                         $era = 'jungle';
77                 }
78                 my $tileset = Data::StarCraft::Tileset->open("/home/shiar/sc/tileset_$era")
79                         or die "No tileset for world $era";
80
81                 printf "P3\n%d %d\n255\n", $map->info->{x}, $map->info->{y};
82                 for (@$tiles) {
83                         my $tile = $tileset->tileavg($_);
84                         my $div = 1;#$tile->{walk} > 1 ? 1 : 1.8;
85                         my @rgb = map {int($_ / $div)} @{ $tile->{col} };
86                         if (($tile->{walk} & 1) == 0 and $tile->{walk} < 10) {
87 #                               $rgb[2] += 128;
88                                 $rgb[0] *= 3;
89                                 $rgb[1] *= 3;
90                                 $rgb[2] *= 3;
91                         } elsif ($tile->{build}) {
92                                 $rgb[0] += 48;
93 #                               $rgb[$_] = int($rgb[$_] / 1.5) for 0..2;
94                         }
95                         $_ = join ' ', @rgb;
96                 }
97         }
98
99         while (my @line = splice @$tiles, 0, $map->width) {
100                 printf "%s\n", join $MAPCHARSEP, @line;
101         }
102 }
103
104 =head1 scmap
105
106 From replay:
107
108         old/screptomap somereplay.rep | ./scmap -m=ppm > map.ppm
109
110 From map:
111
112         ./scmtomap starcraft/maps/ladder/'(4)Lost Temple.scm'
113         ./scmap -m=ppm < file000001.xxx > map.ppm
114
115 =cut