ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/deliantra/server/utils/cfutil.in
Revision: 1.20
Committed: Mon Apr 2 18:04:47 2007 UTC (17 years, 1 month ago) by root
Branch: MAIN
Changes since 1.19: +3 -0 lines
Log Message:
moved setup to perlspace to facilitate easier extensions and some cleanups. feels more secure, too

File Contents

# User Rev Content
1 root 1.1 #!@PERL@
2    
3 root 1.2 use strict;
4    
5     my $prefix = "@prefix@";
6     my $exec_prefix = "@exec_prefix@";
7     my $datarootdir = "@datarootdir@";
8     my $DATADIR = "@datadir@/@PACKAGE@";
9    
10     my $CONVERT = "@CONVERT@";
11     my $IDENTIFY = "@IDENTIFY@";
12 root 1.3 my $OPTIPNG = "@OPTIPNG@";
13 root 1.2 my $RSYNC = "@RSYNC@";
14    
15     use Getopt::Long;
16 root 1.3 use Coro::Event;
17     use AnyEvent;
18     use IO::AIO ();
19 root 1.2 use File::Temp;
20     use Crossfire;
21 root 1.3 use Coro;
22     use Coro::AIO;
23     use POSIX ();
24 root 1.6 use Digest::MD5;
25 root 1.17 use Storable; $Storable::canonical = 1;
26 root 1.2
27     sub usage {
28     warn <<EOF;
29 root 1.3 Usage: cfutil [-v] [-q] [--force] [--cache]
30 root 1.2 [--install-arch path]
31     [--install-maps maps]
32     [--print-statedir]
33     [--print-confdir]
34     [--print-datadir]
35     [--print-libdir]
36     [--print-bindir]
37     EOF
38     exit 1;
39     }
40    
41     my $VERBOSE = 1;
42 root 1.3 my $CACHE = 0;
43 root 1.2 my $FORCE;
44 root 1.3 my $TMPDIR = "/tmp/cfutil$$~";
45     my $TMPFILE = "aaaa0";
46 root 1.2
47     END { system "rm", "-rf", $TMPDIR }
48    
49 root 1.3 Event->signal (signal => "INT", cb => sub { exit 1 });
50     Event->signal (signal => "TERM", cb => sub { exit 1 });
51    
52 root 1.2 mkdir $TMPDIR, 0700
53     or die "$TMPDIR: $!";
54    
55 root 1.3 sub fork_sub(&) {
56     my ($cb) = @_;
57    
58     if (my $pid = fork) {
59     my $current = $Coro::current;
60     my $w = AnyEvent->child (pid => $pid, cb => sub { $current->ready });
61     Coro::schedule;
62     } else {
63     eval { $cb->() };
64     POSIX::_exit 0 unless $@;
65     warn $@;
66     POSIX::_exit 1;
67     }
68     }
69    
70 root 1.2 sub inst_maps($) {
71     my (undef, $path) = @_;
72    
73 root 1.13 print "installing '$path' to '$DATADIR/maps'\n";
74 root 1.2
75     if (!-f "$path/regions") {
76     warn "'$path' does not look like a maps directory ('regions' file is missing).\n";
77     exit 1 unless $FORCE;
78     }
79    
80 root 1.13 system $RSYNC, "-av", "$path/.", "$DATADIR/maps/.", "--delete", "--exclude", "CVS", "--delete-excluded"
81 root 1.16 and die "map installation failed.\n";
82 root 1.13
83     print "maps installed successfully.\n";
84 root 1.2 }
85    
86     {
87 root 1.6 our %FACEINFO;
88 root 1.3 our @ARC;
89 root 1.12 our $TRS;
90 root 1.3 our $NFILE;
91 root 1.6 our %ANIM;
92 root 1.3
93 root 1.19 our $QUANTIZE = "+dither -colorspace RGB -colors 256";
94    
95 root 1.3 our (@png, @trs, @arc); # files we are interested in
96 root 1.2
97 root 1.19 sub commit_png($$$) {
98     my ($name, $data, $T) = @_;
99 root 1.5
100 root 1.19 $FACEINFO{$name}{"data$T"} = $data;
101 root 1.2 }
102    
103 root 1.3 sub process_png {
104     while (@png) {
105 root 1.19 my ($path, $delete) = @{pop @png};
106 root 1.2
107 root 1.3 my $png;
108     aio_lstat $path;
109 root 1.2 my ($size, $mtime) = (stat _)[7,9];
110    
111 root 1.3 if (0 > aio_load $path, $png) {
112     warn "$path: $!, skipping.\n";
113 root 1.5 next;
114 root 1.3 }
115    
116 root 1.19 my $stem = $path;
117     my $T;
118    
119     if ($stem =~ s/\.32x32\.png~?$//) {
120     $T = 32;
121     } elsif ($stem =~ s/\.64x64\.png~?$//) {
122     $T = 64;
123     } else {
124     warn "$path: weird filename, skipping.\n";
125     next;
126     }
127    
128 root 1.6 # quickly extract width and height of the (necessarily PNG) image
129 root 1.3 unless ($png =~ /^\x89PNG\x0d\x0a\x1a\x0a....IHDR(........)/s) {
130 root 1.19 warn "$path: not a recognized png file, skipping.\n";
131 root 1.5 next;
132 root 1.3 }
133 root 1.2
134 root 1.3 my ($w, $h) = unpack "NN", $1;
135 root 1.2
136 root 1.3 if ($w < $T || $h < $T) {
137     warn "$path: too small ($w $h), skipping.\n";
138 root 1.5 next;
139 root 1.3 }
140    
141     if ($w % $T || $h % $T) {
142     warn "$path: weird png size ($w $h), skipping.\n";
143 root 1.5 next;
144 root 1.3 }
145 root 1.2
146 root 1.19 unless ($path =~ /~$/) {
147     # possibly enlarge
148     if (0 > aio_stat "$stem.64x64.png") {
149     my $other = "$stem.64x64.png~";
150    
151     if (0 > aio_lstat $other or (-M _) > (-M $path)) {
152     warn "rescale $other\n";#d#
153     my $wrap = 0; # for the time being
154     fork_sub {
155     system "convert png:\Q$path\E -depth 8 rgba:-"
156     . "| $exec_prefix/bin/cfhq2xa $w $h $wrap"
157     . "| convert -depth 8 -size ".($w * 2)."x".($h * 2)." rgba:- $QUANTIZE -quality 00 png32:\Q$other\E~"
158     and die "convert/hq2xa pipeline error: status $? ($!)";
159     system $OPTIPNG, "-o5", "-i0", "-q", "$other~";
160     die "$other~ has zero size, aborting." unless -s "$other~";
161     rename "$other~", $other;
162     };
163     }
164    
165     push @png, [$other, !$CACHE];
166     }
167    
168     # possibly scale down
169     if (0 > aio_stat "$stem.32x32.png") {
170     my $other = "$stem.32x32.png~";
171    
172     if (0 > aio_lstat $other or (-M _) > (-M $path)) {
173     fork_sub {
174     system "convert png:\Q$path\E -geometry 50% -filter lanczos $QUANTIZE -quality 00 png32:\Q$other\E~";
175     system $OPTIPNG, "-o5", "-i0", "-q", "$other~";
176     die "$other~ has zero size, aborting." unless -s "$other~";
177     rename "$other~", $other;
178     };
179     }
180    
181     warn "scaled down $path to $other\n";#d#
182    
183     push @png, [$other, !$CACHE];
184     }
185     }
186    
187     (my $face = $stem) =~ s/^.*\///;
188    
189 root 1.3 if (($w > $T || $h > $T) && $face !~ /_S\./) {
190     # split
191     my @tile;
192     for my $x (0 .. (int $w / $T) - 1) {
193     for my $y (0 .. (int $h / $T) - 1) {
194     my $file = "$path+$x+$y~";
195     aio_lstat $file;
196     push @tile, [$x, $y, $file, (stat _)[9]];
197     }
198 root 1.2 }
199    
200 root 1.3 my $mtime = (lstat $path)[9];
201     my @todo = grep { $_->[3] <= $mtime } @tile;
202     if (@todo) {
203     fork_sub {
204     open my $convert, "|-", $CONVERT,
205     "png:-",
206     (map {
207     (
208     "(",
209     "+clone",
210     -crop => (sprintf "%dx%d+%d+%d", $T, $T, $_->[0] * $T, $_->[1] * $T),
211 root 1.7 "+repage",
212 root 1.19 -quality => "00",
213     -write => "png32:$_->[2]~",
214 root 1.3 "+delete",
215     ")",
216     )
217     } @todo),
218     "null:";
219    
220     binmode $convert;
221     print $convert $png;
222     close $convert;
223    
224     # pass 2, optimise, and rename
225     for (@todo) {
226     system $OPTIPNG, "-o5", "-i0", "-q", "$_->[2]~";
227 root 1.19 die "$_->[2]~ has zero size, aborting." unless -s "$_->[2]~";
228 root 1.3 rename "$_->[2]~", $_->[2];
229     }
230     };
231 root 1.2 }
232    
233 root 1.3 for (@tile) {
234     my ($x, $y, $file) = @$_;
235     my $tile;
236    
237     if (0 > aio_load $file, $tile) {
238     die "$path: unable to read tile +$x+$y, aborting.\n";
239     }
240     IO::AIO::aio_unlink $file unless $CACHE;
241 root 1.19 commit_png $x|$y ? "$face+$x+$y" : $face, $tile, $T;
242 root 1.2 }
243 root 1.3 } else {
244     # use as-is (either small, use smooth)
245 root 1.19 commit_png $face, $png, $T;
246 root 1.3 }
247 root 1.19
248     aio_unlink $path if $delete;
249 root 1.3 }
250 root 1.2 }
251    
252 root 1.3 sub process_arc {
253     while (@arc) {
254     my ($dir, $file) = @{pop @arc};
255 root 1.2
256 root 1.3 my $arc;
257     aio_load "$dir/$file", $arc; # simply pre-cache, as read_arch wants a file :/
258 root 1.4
259 root 1.2 my $arc = read_arch "$dir/$file";
260 root 1.4 for my $o (values %$arc) {
261     push @ARC, $o;
262 root 1.6
263 root 1.10 $o->{editor_folder} = $dir;
264    
265 root 1.6 my $visibility = delete $o->{visibility};
266     my $magicmap = delete $o->{magicmap};
267    
268     # find upper left corner :/
269     # omg, this is sooo broken
270 root 1.4 my ($dx, $dy);
271     for (my $o = $o; $o; $o = $o->{more}) {
272     $dx = $o->{x} if $o->{x} < $dx;
273     $dy = $o->{y} if $o->{y} < $dy;
274     }
275 root 1.6
276 root 1.4 for (my $o = $o; $o; $o = $o->{more}) {
277     my $x = $o->{x} - $dx;
278     my $y = $o->{y} - $dy;
279 root 1.6
280     my $ext = $x|$y ? "+$x+$y" : "";
281    
282 root 1.9 $o->{face} .= $ext unless /^blank.x11$|^empty.x11$/;
283 root 1.6
284     my $visibility = delete $o->{visibility} if exists $o->{visibility};
285     my $magicmap = delete $o->{magicmap} if exists $o->{magicmap};
286    
287     my $anim = delete $o->{anim};
288    
289     if ($anim) {
290 root 1.8 $o->{animation} = "$o->{_name}";
291 root 1.6
292     for (@$anim) {
293 root 1.9 $_ .= $ext unless /^facings\s|^blank.x11$|^empty.x11$/;
294 root 1.6 }
295    
296     $ANIM{"$o->{_name}$ext"} =
297     join "", map "$_\n",
298 root 1.8 "anim $o->{_name}",
299 root 1.6 @$anim,
300     "mina";
301     }
302    
303     for my $face ($o->{face} || (), @{$anim || []}) {
304 root 1.9 next if $face =~ /^facings\s|^blank.x11$|^empty.x11$/;
305 root 1.6
306     my $info = $FACEINFO{$face} ||= {};
307    
308     $info->{visibility} = $visibility if defined $visibility;
309     $info->{magicmap} = $magicmap if defined $magicmap;
310 root 1.4 }
311 root 1.12
312     if (my $smooth = delete $o->{smoothface}) {
313 root 1.14 my ($face, $smooth) = split /\s+/, $smooth;
314     # skip empty_S.x11, it seems to server no purpose whatsoever
315     # but increases bandwidth demands and worse.
316     unless ($smooth eq "empty_S.x11") {
317     $FACEINFO{$face}{smooth} = $smooth;
318     }
319 root 1.12 }
320 root 1.4 }
321     }
322 root 1.3 }
323 root 1.2 }
324    
325 root 1.3 sub process_trs {
326     while (@trs) {
327     my ($dir, $file) = @{pop @trs};
328 root 1.12 my $path = "$dir/$file";
329    
330     my $trs;
331     if (0 > aio_load $path, $trs) {
332     warn "$path: $!, skipping.\n";
333     next;
334     }
335    
336     $TRS .= $trs;
337 root 1.3 }
338 root 1.2 }
339    
340 root 1.3 sub find_files;
341     sub find_files {
342 root 1.2 my ($path) = @_;
343    
344 root 1.3 IO::AIO::aioreq_pri 4;
345     IO::AIO::aio_scandir $path, 4, sub {
346 root 1.2 my ($dirs, $nondirs) = @_;
347    
348 root 1.3 find_files "$path/$_"
349 root 1.2 for grep $_ !~ /^(?:CVS|dev)$/, @$dirs;
350    
351     for my $file (@$nondirs) {
352     if ($file =~ /\.png$/) {
353 root 1.19 push @png, ["$path/$file", 0];
354 root 1.2 } elsif ($file =~ /\.trs$/) {
355 root 1.3 push @trs, [$path, $file];
356 root 1.2 } elsif ($file =~ /\.arc$/) {
357 root 1.3 push @arc, [$path, $file];
358 root 1.2 } else {
359     warn "ignoring $path/$file\n" if $VERBOSE >= 2;
360     }
361     }
362     };
363     }
364    
365     sub inst_arch($) {
366     my (undef, $path) = @_;
367    
368 root 1.19 print "installing '$path' to '$DATADIR'\n",
369     "(this can take a long time if you run this\n",
370     "for the first time or do not use --cache).\n";
371 root 1.2
372     if (!-d "$path/treasures") {
373     warn "'$path' does not look like an arch directory ('treasures' directory is missing).\n";
374     exit 1 unless $FORCE;
375     }
376    
377 root 1.3 find_files $path;
378 root 1.2 IO::AIO::flush;
379    
380 root 1.3 $_->join for (
381 root 1.19 # four png crunchers work fine for my 2x smp machine
382     (async \&process_png), (async \&process_png), (async \&process_png), (async \&process_png),
383 root 1.3 (async \&process_trs), (async \&process_trs),
384     (async \&process_arc), (async \&process_arc),
385     );
386    
387 root 1.5 {
388     open my $fh, ">:utf8", "$DATADIR/animations~"
389     or die "$DATADIR/animations~: $!";
390 root 1.6 print $fh join "", map $ANIM{$_}, sort keys %ANIM
391 root 1.5 }
392    
393     {
394     open my $fh, ">:utf8", "$DATADIR/archetypes~"
395     or die "$DATADIR/archetypes~: $!";
396 root 1.11 substr $_->{editor_folder}, 0, 1 + length $path, "" for @ARC;
397 root 1.5 print $fh Crossfire::archlist_to_string \@ARC;
398     }
399    
400 root 1.6 {
401 root 1.12 open my $fh, ">:utf8", "$DATADIR/treasures~"
402     or die "$DATADIR/treasures~: $!";
403     print $fh $TRS;
404     }
405    
406     {
407 root 1.6 while (my ($k, $v) = each %FACEINFO) {
408 root 1.12 length $v->{data32} or warn "$k: face has no png32. this will not work (shoddy gcfclient will crash of course).\n";
409 root 1.19 length $v->{data64} or warn "$k: face has no png64. this will not work very well.\n";
410 root 1.5
411 root 1.20 length $v->{data32} <= 10000 or warn "$k: face32 larger than 10000 bytes, will not work with crossfire client.\n";
412     #length $v->{data64} <= 10000 or warn "$k: face64 larger than 10000 bytes.\n";
413    
414 root 1.6 $v->{chksum32} = Digest::MD5::md5 $v->{data32};
415 root 1.19 $v->{chksum64} = Digest::MD5::md5 $v->{data64};
416 root 1.6 }
417    
418 root 1.19 open my $fh, ">:perlio", "$DATADIR/facedata~"
419     or die "$DATADIR/facedata~: $!";
420 root 1.6
421 root 1.15 $FACEINFO{""} = { version => 1};
422 root 1.6 print $fh Storable::nfreeze \%FACEINFO;
423 root 1.5 }
424    
425 root 1.19 for (qw(archetypes facedata animations treasures)) {
426 root 1.6 chmod 0644, "$DATADIR/$_~";
427 root 1.13 rename "$DATADIR/$_~", "$DATADIR/$_"
428     or die "$DATADIR/$_: $!";
429 root 1.6 }
430 root 1.13
431     print "archetype data installed successfully.\n";
432 root 1.2 }
433     }
434    
435     Getopt::Long::Configure ("bundling", "no_ignore_case");
436     GetOptions (
437     "verbose|v:+" => \$VERBOSE,
438 root 1.3 "cache" => \$CACHE,
439 root 1.2 "quiet|q" => sub { $VERBOSE = 0 },
440     "force" => sub { $FORCE = 1 },
441     "install-arch=s" => \&inst_arch,
442     "install-maps=s" => \&inst_maps,
443     "print-statedir" => sub { print "@pkgstatedir@\n" },
444     "print-datadir" => sub { print "$DATADIR\n" },
445     "print-confdir" => sub { print "@pkgconfdir@\n" },
446     "print-libdir" => sub { print "@libdir@/@PACKAGE@\n" },
447     "print-bindir" => sub { print "@bindir@/@PACKAGE@\n" },
448     ) or usage;
449