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

# Content
1 #!@PERL@
2
3 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 my $OPTIPNG = "@OPTIPNG@";
13 my $RSYNC = "@RSYNC@";
14
15 use Getopt::Long;
16 use Coro::Event;
17 use AnyEvent;
18 use IO::AIO ();
19 use File::Temp;
20 use Crossfire;
21 use Coro;
22 use Coro::AIO;
23 use POSIX ();
24 use Digest::MD5;
25 use Storable; $Storable::canonical = 1;
26
27 sub usage {
28 warn <<EOF;
29 Usage: cfutil [-v] [-q] [--force] [--cache]
30 [--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 my $CACHE = 0;
43 my $FORCE;
44 my $TMPDIR = "/tmp/cfutil$$~";
45 my $TMPFILE = "aaaa0";
46
47 END { system "rm", "-rf", $TMPDIR }
48
49 Event->signal (signal => "INT", cb => sub { exit 1 });
50 Event->signal (signal => "TERM", cb => sub { exit 1 });
51
52 mkdir $TMPDIR, 0700
53 or die "$TMPDIR: $!";
54
55 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 sub inst_maps($) {
71 my (undef, $path) = @_;
72
73 print "installing '$path' to '$DATADIR/maps'\n";
74
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 system $RSYNC, "-av", "$path/.", "$DATADIR/maps/.", "--delete", "--exclude", "CVS", "--delete-excluded"
81 and die "map installation failed.\n";
82
83 print "maps installed successfully.\n";
84 }
85
86 {
87 our %FACEINFO;
88 our @ARC;
89 our $TRS;
90 our $NFILE;
91 our %ANIM;
92
93 our $QUANTIZE = "+dither -colorspace RGB -colors 256";
94
95 our (@png, @trs, @arc); # files we are interested in
96
97 sub commit_png($$$) {
98 my ($name, $data, $T) = @_;
99
100 $FACEINFO{$name}{"data$T"} = $data;
101 }
102
103 sub process_png {
104 while (@png) {
105 my ($path, $delete) = @{pop @png};
106
107 my $png;
108 aio_lstat $path;
109 my ($size, $mtime) = (stat _)[7,9];
110
111 if (0 > aio_load $path, $png) {
112 warn "$path: $!, skipping.\n";
113 next;
114 }
115
116 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 # quickly extract width and height of the (necessarily PNG) image
129 unless ($png =~ /^\x89PNG\x0d\x0a\x1a\x0a....IHDR(........)/s) {
130 warn "$path: not a recognized png file, skipping.\n";
131 next;
132 }
133
134 my ($w, $h) = unpack "NN", $1;
135
136 if ($w < $T || $h < $T) {
137 warn "$path: too small ($w $h), skipping.\n";
138 next;
139 }
140
141 if ($w % $T || $h % $T) {
142 warn "$path: weird png size ($w $h), skipping.\n";
143 next;
144 }
145
146 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 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 }
199
200 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 "+repage",
212 -quality => "00",
213 -write => "png32:$_->[2]~",
214 "+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 die "$_->[2]~ has zero size, aborting." unless -s "$_->[2]~";
228 rename "$_->[2]~", $_->[2];
229 }
230 };
231 }
232
233 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 commit_png $x|$y ? "$face+$x+$y" : $face, $tile, $T;
242 }
243 } else {
244 # use as-is (either small, use smooth)
245 commit_png $face, $png, $T;
246 }
247
248 aio_unlink $path if $delete;
249 }
250 }
251
252 sub process_arc {
253 while (@arc) {
254 my ($dir, $file) = @{pop @arc};
255
256 my $arc;
257 aio_load "$dir/$file", $arc; # simply pre-cache, as read_arch wants a file :/
258
259 my $arc = read_arch "$dir/$file";
260 for my $o (values %$arc) {
261 push @ARC, $o;
262
263 $o->{editor_folder} = $dir;
264
265 my $visibility = delete $o->{visibility};
266 my $magicmap = delete $o->{magicmap};
267
268 # find upper left corner :/
269 # omg, this is sooo broken
270 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
276 for (my $o = $o; $o; $o = $o->{more}) {
277 my $x = $o->{x} - $dx;
278 my $y = $o->{y} - $dy;
279
280 my $ext = $x|$y ? "+$x+$y" : "";
281
282 $o->{face} .= $ext unless /^blank.x11$|^empty.x11$/;
283
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 $o->{animation} = "$o->{_name}";
291
292 for (@$anim) {
293 $_ .= $ext unless /^facings\s|^blank.x11$|^empty.x11$/;
294 }
295
296 $ANIM{"$o->{_name}$ext"} =
297 join "", map "$_\n",
298 "anim $o->{_name}",
299 @$anim,
300 "mina";
301 }
302
303 for my $face ($o->{face} || (), @{$anim || []}) {
304 next if $face =~ /^facings\s|^blank.x11$|^empty.x11$/;
305
306 my $info = $FACEINFO{$face} ||= {};
307
308 $info->{visibility} = $visibility if defined $visibility;
309 $info->{magicmap} = $magicmap if defined $magicmap;
310 }
311
312 if (my $smooth = delete $o->{smoothface}) {
313 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 }
320 }
321 }
322 }
323 }
324
325 sub process_trs {
326 while (@trs) {
327 my ($dir, $file) = @{pop @trs};
328 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 }
338 }
339
340 sub find_files;
341 sub find_files {
342 my ($path) = @_;
343
344 IO::AIO::aioreq_pri 4;
345 IO::AIO::aio_scandir $path, 4, sub {
346 my ($dirs, $nondirs) = @_;
347
348 find_files "$path/$_"
349 for grep $_ !~ /^(?:CVS|dev)$/, @$dirs;
350
351 for my $file (@$nondirs) {
352 if ($file =~ /\.png$/) {
353 push @png, ["$path/$file", 0];
354 } elsif ($file =~ /\.trs$/) {
355 push @trs, [$path, $file];
356 } elsif ($file =~ /\.arc$/) {
357 push @arc, [$path, $file];
358 } 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 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
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 find_files $path;
378 IO::AIO::flush;
379
380 $_->join for (
381 # four png crunchers work fine for my 2x smp machine
382 (async \&process_png), (async \&process_png), (async \&process_png), (async \&process_png),
383 (async \&process_trs), (async \&process_trs),
384 (async \&process_arc), (async \&process_arc),
385 );
386
387 {
388 open my $fh, ">:utf8", "$DATADIR/animations~"
389 or die "$DATADIR/animations~: $!";
390 print $fh join "", map $ANIM{$_}, sort keys %ANIM
391 }
392
393 {
394 open my $fh, ">:utf8", "$DATADIR/archetypes~"
395 or die "$DATADIR/archetypes~: $!";
396 substr $_->{editor_folder}, 0, 1 + length $path, "" for @ARC;
397 print $fh Crossfire::archlist_to_string \@ARC;
398 }
399
400 {
401 open my $fh, ">:utf8", "$DATADIR/treasures~"
402 or die "$DATADIR/treasures~: $!";
403 print $fh $TRS;
404 }
405
406 {
407 while (my ($k, $v) = each %FACEINFO) {
408 length $v->{data32} or warn "$k: face has no png32. this will not work (shoddy gcfclient will crash of course).\n";
409 length $v->{data64} or warn "$k: face has no png64. this will not work very well.\n";
410
411 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 $v->{chksum32} = Digest::MD5::md5 $v->{data32};
415 $v->{chksum64} = Digest::MD5::md5 $v->{data64};
416 }
417
418 open my $fh, ">:perlio", "$DATADIR/facedata~"
419 or die "$DATADIR/facedata~: $!";
420
421 $FACEINFO{""} = { version => 1};
422 print $fh Storable::nfreeze \%FACEINFO;
423 }
424
425 for (qw(archetypes facedata animations treasures)) {
426 chmod 0644, "$DATADIR/$_~";
427 rename "$DATADIR/$_~", "$DATADIR/$_"
428 or die "$DATADIR/$_: $!";
429 }
430
431 print "archetype data installed successfully.\n";
432 }
433 }
434
435 Getopt::Long::Configure ("bundling", "no_ignore_case");
436 GetOptions (
437 "verbose|v:+" => \$VERBOSE,
438 "cache" => \$CACHE,
439 "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