chiark / gitweb /
wip nz
[pandemic-rising-tide.git] / generate-board
1 #!/usr/bin/perl -w
2
3 use strict;
4 use Carp;
5 use Data::Dumper;
6 use Math::GSL::Vector qw/:all/;
7 use Math::GSL::Matrix qw/:all/;
8 use Math::GSL::Const;
9 use Math::GSL::BLAS qw/:all/;
10 use Math::GSL::CBLAS qw/:all/;
11 use Math::GSL::Machine qw/:all/;
12
13 use POSIX qw(M_PI);
14
15 BEGIN { unshift @INC, qw(.); }
16
17 our %c;
18 require 'misc-data.pl';
19
20 use Parse;
21
22 our $facesf;
23 our %vxname2pos; # $vxname2pos{VXNAME} = Math::GSL::Vector
24
25 sub TAU { M_PI * 2.0; }
26 sub MM2PT { 72.0 / 25.4; }
27
28 # ----- region names from plag, incl. reverse mapping -----
29
30 our %prs2region;
31
32 sub prep_region_rmap () {
33   foreach my $rn (keys %region) {
34     my $prs = plag_prs($rn);
35     die if $prs2region{$prs};
36     $prs2region{$prs} = $rn;
37   }
38
39   # bodgery for the sea and land
40   $prs2region{'NZ | L'} = 'NZ';
41   $prs2region{'L | NZ'} = 'NZ';
42 }
43
44 our $sre = qr{(?:(\")|(?=\w))(.*)\1$}; # 2 captures, $2 is the name
45
46 sub prs2r ($) { $prs2region{$_[0]} // confess "@_ ?" }
47 sub prs2rr ($) { $region{prs2r($_[0])} }
48
49 #----- file reader for plag output -----
50
51 sub read_faces () {
52   # Sets
53   #  $region{NAME}{Adj}[]{Ends}[]{VxName}
54   #  $region{NAME}{Adj}[]{Ends}[]{Pos}
55   #  $region{NAME}{Polygon}{Pos}
56   #  $region{NAME}{Special}
57   #
58   # also incidentally
59   #  $region{NAME}{Adj}[]{Ends}[]{Adjoins}
60   #  $region{NAME}{Polygon}{Adjoins}
61   # which should be ignored
62
63   open P, "<", $facesf or die $!;
64   while (<P>) { last if m/^\&faces$/; }
65   my $rr;
66   my @edges;
67   my $process_rr = sub {
68     my $last_ai;
69     if ($rr->{Name} eq 'NZ') {
70       # We combined L and NZ; now we must split them again
71       # in fact, we just throw away L entirely.
72       # Here, when processing NZ, we keep only vertices that
73       # are part of NZ.
74       @edges = grep {
75         my $evxname = $_->{VxName};
76         my @eregions = split / \| ?/, $evxname;
77         grep { $_ eq 'NZ' } @eregions
78       } @edges;
79     }
80     $rr->{Polygon} = [ @edges ];
81     if ($rr->{Name} ne 'NZ') {
82       for my $ei (0..$#edges) {
83         my $ej = $ei % @edges;
84         if (!defined $last_ai) {
85           my $ai;
86           for my $ai (0..$#{ $rr->{Adj} }) {
87             next unless $rr->{Adj}[$ai]{Name} eq $edges[$ei]{Adjoins};
88             $last_ai = $ai+1;
89           }
90           confess $edges[$ei]{Adjoins}.' ?' unless defined $last_ai;
91         }
92         my $ai = ($last_ai-1+@edges) % @edges;
93         $last_ai = $ai;
94         my $adj = $rr->{Adj}[$ai];
95         confess Dumper($rr, \@edges, $ei, $ej, $adj, $last_ai, $ai)." ?"
96             unless $adj->{Name} eq $edges[$ei]{Adjoins};
97       
98         for my $endi (0..1) {
99           $adj->{Ends}[$endi] = $edges[ ($ei + $endi) % @edges ];
100         }
101       }
102     }
103     @edges = ();
104     $rr = undef;
105   };
106   my $vxname;
107   for (;;) {
108     $!=0; $_=<P> // confess $!;
109     last if m/^\&$/;
110     if (m/^$sre$/) {
111       my $new_face= $2;
112       $process_rr->() if $rr;
113       $rr= prs2rr($new_face);
114     } elsif (m/^\s+$sre$/) {
115       confess unless $rr;
116       $vxname = $2;
117       push @edges, { VxName => $vxname };
118     } elsif (m/^\s+\^adjoins\s+$sre$/) {
119       $edges[-1]{Adjoins} = prs2r($2);
120     } elsif (m/^\s+\^\@([-e.0-9]+)\,([-e.0-9]+)$/) {
121       my $pos = Math::GSL::Vector->new([$1,$2]);
122       confess unless defined $vxname;
123       $edges[-1]{Pos} = $pos;
124       $vxname2pos{$vxname} = $pos;
125     }
126   }
127   $process_rr->();
128
129   $region{$_}{Special} = 1 foreach qw(NZ L);
130 }
131
132 #----- geometry mangling -----
133
134 sub calculate_centres () {
135   # Sets
136   #  $region{NAME}{Centre}
137   foreach my $rr (values %region) {
138     next if $rr->{Special};
139     my $poly = $rr->{Polygon};
140     my $sum = Math::GSL::Vector->new(2);
141     $sum += $_->{Pos} foreach @$poly;
142     $rr->{Centre} = $sum * (1.0 / @$poly);
143   }
144 }
145
146 sub for_each_pos ($) {
147   my ($f) = @_;
148   my $call = sub {
149     my ($pr,$rr,$why) = @_;
150     return unless defined $$pr;
151     $f->($pr,$rr,$why);
152   };
153   foreach my $rr (values %region) {
154     $call->( \ $rr->{Centre}, $rr, $rr->{Name}." C" );
155     foreach my $vertex (@{ $rr->{Polygon} }) {
156       $call->( \ $vertex->{Pos}, $rr, $rr->{Name}." | ".$vertex->{Adjoins} );
157     }
158   }
159 }
160
161 sub prvec ($) {
162   my ($v) = @_;
163   confess unless $v;
164   sprintf "%g,%g", $v->get(0), $v->get(1);
165 }
166
167 sub transform_coordinates () {
168   # Adjusts coordinates in graph to be [0,0] .. top right (scaled)
169   # until it's all in PostScript points
170   my @or = map { $region{$_}{Centre} } @{ $c{OrientRegions} };
171   my $dir = $or[1] - $or[0];
172   my $theta = atan2 $dir->get(1), $dir->get(0);
173   my $rotateby = (90. - $c{OrientBearing}) * ((TAU)/360.) - $theta;
174   my $s = sin($rotateby);
175   my $c = cos($rotateby);
176   my $transform = Math::GSL::Matrix->new(2,2);
177   $transform->set_row(0, [  $c, -$s ]);
178   $transform->set_row(1, [  $s,  $c ]);
179   print STDERR "rotate was=",prvec($dir)," theta=$theta",
180       " rotateby=$rotateby s=$s c=$c\n";
181   my @lims;
182   foreach my $topend (qw(0 1)) {
183     my $v = $topend ? -$GSL_DBL_MAX : $GSL_DBL_MAX;
184     $lims[$topend] = Math::GSL::Vector->new([$v,$v]);
185   }
186   for_each_pos(sub {
187     my ($pr, $rr, $why) = @_;
188     my $y = Math::GSL::Vector->new(2);
189     gsl_blas_dgemv($CblasNoTrans,
190                    1.0, $transform->raw,
191                    $$pr->raw,
192                    0, $y->raw)
193         and confess;
194     #print STDERR "gsl_blas_dgemv ",prvec($$pr)," => ",prvec($y),"\n";
195     gsl_blas_dcopy($y->raw, $$pr->raw)
196         and confess;
197     foreach my $topend (qw(0 1)) {
198       foreach my $xy (qw(0 1)) {
199         my $now = $y->get($xy);
200         my $lim = $lims[$topend]->get($xy);
201         #print STDERR "?set $topend $xy $now $lim\n";
202         next if $topend ? ($now <= $lim) : ($now >= $lim);
203         $lims[$topend]->set([$xy], [$now]);
204         #print STDERR "set $topend $xy $now\n";
205       }
206     }
207   });
208   print STDERR "lims ",prvec($lims[0])," .. ",prvec($lims[1]),"\n";
209   my $translate = -$lims[0];
210   print STDERR "translate ",prvec($translate),"\n";
211   my $scale = $c{GraphScale} * MM2PT;
212   for_each_pos(sub {
213     my ($pr) = @_;
214     gsl_vector_add($$pr->raw, $translate->raw) and confess;
215     gsl_vector_scale($$pr->raw, $scale) and confess;
216   });
217 }
218
219 sub adjust_nz() {
220   my $poly = $region{NZ}{Polygon};
221   my %occurs;
222   foreach my $v (@$poly) {
223     $occurs{$v->{VxName}}++
224   }
225   foreach my $v (@$poly) {
226     $v->{NZ_Occurs} = $occurs{$v->{VxName}};
227   }
228   my $ei;
229   my $occurs_m = sub {
230     my ($delta) = @_;
231     $poly->[($ei+$delta+@$poly) % @$poly]{NZ_Occurs} > 1
232   };
233   for $ei (0..$#$poly) {
234     next if $occurs_m->(0);
235     next unless $occurs_m->(-1);
236     next unless $occurs_m->(+1);
237     # adjust coord outwards
238   }
239   for $ei (0..$#$poly) {
240     next unless $occurs_m->(-1);
241     next unless $occurs_m->(0);
242     my $pos = \ $poly->{$ei}{Pos};
243   }
244 }
245
246 sub write_face_edges() {
247   o("5 setlinewidth 1 setlinejoin\n");
248   foreach my $rr (values %region) {
249     next if $rr->{Name} eq 'L';
250     o("% $rr->{Name}\n");
251     o("  newpath\n");
252     my $xto = 'moveto';
253     foreach my $vertex (@{ $rr->{Polygon} }) {
254       my $pos = $vertex->{Pos};
255       # xxx break this out
256       o(sprintf "    %20.6f %20.6f %s\n", $pos->get(0),$pos->get(1), $xto);
257       $xto = 'lineto';
258     }
259     o("    closepath\n");
260     my $chex = $c{Colours}{$rr->{Colour}};
261     if ($chex) {
262       o("    gsave ");
263       o(sprintf " %.6f ", hex($_) / 15.0) foreach split //, $chex;
264       o(" setrgbcolor eofill grestore\n");
265     }
266     if ($rr->{Name} ne 'NZ') {
267       o("    stroke\n");
268     }
269   }
270 }
271
272 #----- main program -----
273
274 ($facesf, @ARGV) = @ARGV or die;
275 parse_input_graph();
276 prep_region_rmap();
277 read_faces();
278 calculate_centres();
279 transform_coordinates();
280
281 o("%!\n");
282 o(" 0.52 dup scale\n"); # xxx bodge
283 # xxx boundingbox
284 write_face_edges();
285
286 print STDERR Dumper(\%region) if $ENV{'GENERATE_BOARD_DUP'};
287
288 # Local variables:
289 # cperl-indent-level: 2
290 # End.