# The Scottish Local Government Elections Order 2007
use strict;
+use Carp;
+use Data::Dumper;
+use Math::BigRat;
use bigrat;
# Data structures:
our $stage=0;
our $quota;
our %tie;
+our @elected;
+
+our $DIGS = 5;
+our $F = (new Math::BigRat 10)->bpow($DIGS);
+
+open DEBUG, ">.compute.log" or die $!;
+DEBUG->autoflush(1);
+
+sub debug_dump () {
+ print DEBUG Dumper(\%tie,
+ \@non_transferable,
+ \%cands,
+ \@allvotes,
+ $stage, $quota);
+}
+
+$SIG{__WARN__} = sub {
+ $SIG{__DIE__} = undef;
+ debug_dump;
+ confess $_[0];
+};
+
+$SIG{__DIE__} = sub {
+ debug_dump;
+ die $_[0];
+};
+
+sub total_history_cmp ();
sub unkopt ($$) {
my ($what,$opt) = @_;
}
}
+my $display_cmp = \&total_history_cmp;
+my $for_compare = 0;
+
+while (@ARGV && $ARGV[0] =~ m/^\-/) {
+ $_ = shift @ARGV;
+ if (m/^--$/) {
+ last;
+ } elsif (m/^--sort=alpha$/) {
+ $display_cmp = sub { $b->{Cand} cmp $a->{Cand} };
+ } elsif (m/^--for-compare$/) {
+ $for_compare = 1;
+ unshift @ARGV, '--sort=alpha';
+ } else {
+ die;
+ }
+}
+
for (;;) {
$_ = <>;
if (m/^\| /) {
}
}
} elsif (m/^(\w+) = (\S*) \|/) {
- my ($cand,$desc) = @_;
+ my ($cand,$desc) = ($1,$2);
unkopt "candidate $cand", $_ foreach split / /, $';
$cands{$cand}{Desc} = $desc;
} elsif (m/^(\w*) :\s*(.*)\s*\|(.*)/) {
my ($voter,$prefs,$opts) = ($1,$2,$3);
- my $v = { Voter => $voter };
+ my $v = { Voter => $voter, Prefs => [ ] };
push @{ $v->{Prefs} }, [ $_ =~ m/\w+/g ]
foreach split /\s+/, $prefs;
foreach $_ (split / /, $opts) {
}
$cands{$_}{Cand} = $_ foreach keys %cands;
+$_->{Weight} //= 1/1 foreach @allvotes;
+$_->{TransferredSurplus} //= [ ] foreach @allvotes;
+$_->{OrigPrefs} //= [ @{ $_->{Prefs} } ] foreach @allvotes;
sub votelog ($$) {
my ($vote,$m) = @_;
push @{ $vote->{Log} }, "stage $stage: $m";
}
+sub sortballots (@);
sub sortballots (@) {
# Takes each argument, which should be a ballot, sorts
# it into $cand{CAND}{Votes} according to first preference.
# Strips that first preference from the ballot.
# If the first preference has been eliminated, strips it
# and looks for further preferences.
+ print DEBUG "sortballots ".(scalar @_)."...\n";
foreach my $v (@_) {
my $firstprefs = shift @{ $v->{Prefs} };
my $w = $v->{Weight};
my $noncont = $c->{NonCont};
if ($noncont) {
votelog $v, "dropping pref $fp, $noncont";
- sortvallots $v;
+ sortballots $v;
next;
}
votelog $v, "sorted into pile for candidate $fp weight $w";
}
}
+sub floor ($) {
+ my ($v) = @_;
+ $v = new Math::BigRat $v; # we need a copy
+ return $v->bfloor();
+}
+
+sub sv ($) {
+ my ($in) = @_;
+ my $v = new Math::BigRat $in; # just in case
+ my $intpart = floor($v);
+ my $frac = $v - $intpart;
+ my $good = floor($frac * $F);
+ my $bad = $frac * $F - $good;
+ my $s = sprintf "%7d", $intpart;
+ if ($frac) {
+ $s .= sprintf ".%0${DIGS}d", $good;
+ $s .= sprintf "%-4s", ($bad ? "+$bad" : "");
+ } else {
+ $s .= sprintf " %${DIGS}s%4s", '', '';
+ }
+#print STDERR "# $in => $s # (intpart=$intpart frac=$frac)\n";
+ return $s;
+}
+
sub prf {
my $fmt = shift;
- printf "stage %d: ".$_, $stage, @_;
+ printf " ".$fmt, @_;
+}
+sub prfm {
+ prf @_ unless $for_compare;
}
sub countballots () {
- foreach my $cand (sort keys %cands) {
- my $c = $cands{$cand};
+ my @pr;
+
+ foreach my $c (values %cands) {
next if $c->{NonCont};
- $c->{Total} = 0;
- $c->{Total} += $_->{Weight} foreach @{ $c->{Voters} };
- prf "cand %s: %s votes\n", $stage, $cand, $c->{Total};
+ $c->{Total} = 0/1;
+ $c->{Total} += $_->{Weight} foreach @{ $c->{Votes} };
+ print DEBUG "counted $c->{Cand} $c->{Total}\n";
$c->{History}[$stage-1] = $c->{Total};
+ push @pr, $c;
+ }
+
+ if ($for_compare) {
+ # for comparison with OpenSTV, which always prints
+ # the quota for every elected candidate
+ push @pr, { %$_, Total => $quota } foreach @elected;
+ }
+
+ foreach my $c (reverse sort $display_cmp @pr) {
+ prf "candidate %-10s: %s votes\n", $c->{Cand}, sv $c->{Total};
}
}
sub computequota () {
- my $totalvalid = 0;
- $totalvalid += $_->{Total} foreach keys %cands;
- $quota = floor($totalvalid / (1 + $seats));
+ my $totalvalid = 0/1;
+ $totalvalid += $_->{Total} foreach values %cands;
+ $quota = floor($totalvalid / (1 + $seats) + 1);
+ prfm "total valid %s\n", sv $totalvalid;
+ prf "quota %s\n", sv $quota;
}
sub total_history_cmp () {
my $ha = $a->{History};
my $hb = $b->{History};
- foreach my $s (reverse 1 .. $stage) {
+ my $m = "stage $stage history cmp $a->{Cand} $b->{Cand}";
+ print DEBUG "$m...\n";
+ foreach my $s (reverse 0 .. $stage-1) {
my $d = $ha->[$s] <=> $hb->[$s];
next unless $d;
- print DEBUG "history cmp $a->{Cand} $b->{Cand}".
- " => $d (#$s $ha->[$s] $hb->[$s])\n";
+ print DEBUG "$m => $d (#$s $ha->[$s] $hb->[$s])\n";
return $d;
}
+ print DEBUG "$m => 0\n";
return 0;
}
for (;;) {
my $nextc = $maybe[$nequal];
+ last unless $nextc;
# Only interested in those who compare equal according to the
# history (SLGEO 49(2)); NB our history includes the current
# round.
last if $signum*($a = $maybe[0], $b = $nextc, total_history_cmp) > 0;
- $nextc++;
+ $nequal++;
}
if ($nequal > 1 && !$wanttiebreak->($maybe[0]{Total})) {
if ($nequal > 1) {
my @all = map { $_->{Cand} } @maybe[0 .. $nequal-1];
my $tiekey = $signum > 0 ? 'Win' : 'Lose';
- my $win = $tie{"@all"}{$tiekey};
- die "need tie break, want $tiekey from @all" unless defined $win;
+ $selectcand = $tie{"@all"}{$tiekey};
+ die "need tie break, want $tiekey from @all"
+ unless defined $selectcand;
prf "$what %s due to tie break amongst %s\n",
$selectcand, "@all";
} else {
- $selectcand = $maybe[0];
- prf "$what %s\n";
+ $selectcand = $maybe[0]{Cand};
+ prfm "$what %s\n", $selectcand;
}
return $cands{$selectcand};
sub elect_core ($) {
my ($c) = @_;
- prf "*** ELECT %s \`%s' ***\n", $c->{Cand}, $c->{Desc};
+ prfm "*** ELECT %s \`%s' ***\n", $c->{Cand}, $c->{Desc};
$c->{NonCont} = 'Elected';
+ push @elected, $c;
}
-sortballots @allvotes;
+$stage = 0;
for (;;) {
$stage++;
+ printf "stage %3d:\n", $stage;
+
+ sortballots @allvotes if $stage == 1;
+
+ my $seats_remain = $seats - @elected;
+
+ prfm "seats remaining %d\n", $seats_remain;
+
+ last unless $seats_remain;
- my $seats_remain = $seats
- - grep { $_->{NonCont} eq 'Elected' } values %cands;
if (continuing() <= $seats_remain) {
foreach my $c (continuing()) {
prf "electing %s to fill remaining place(s)\n", $c->{Cand};
elect_core $c;
}
+ countballots() if $for_compare;
last;
}
countballots();
+ computequota if $stage == 1;
+
my $c = select_best_worst
sub { $_->{Total} >= $quota },
sub { $_ > $quota },
# SLGEO 48
my $surplus = $c->{Total} - $quota;
- if ($surplus <= 0) {
+ if ($surplus <= 0/1) {
prf "no surplus\n";
next;
}
- my $B = $c->{Weight};
+ last if $seats_remain == 1; # don't bother doing more transfers
+
+ my $B = $c->{Total};
+ my %tspr;
+
+ prf "surplus %s\n", sv $surplus;
foreach my $v (@{ $c->{Votes} }) {
+ my $previously = $v->{TransferredSurplus};
+ push @$previously, $c->{Cand};
+
my $A = $surplus * $v->{Weight};
- my $F = 100000;
my $xfervalue = floor(($A * $F) / $B) / $F;
# SLGEO 48(3): we do arithmetic to 5 d3ecimal places,
# but always rounding down
votelog $v, "transferring with value $xfervalue (A=$A B=$B)";
$v->{Weight} = $xfervalue;
- sortballots $v;
+
+ if (defined $tspr{"@$previously"}) {
+ die unless $tspr{"@$previously"} == $xfervalue;
+ } else {
+ $tspr{"@$previously"} = $xfervalue;
+ prfm "transfer value of ballots %20s: %s\n",
+ "@$previously", sv $xfervalue;
+ }
}
+ sortballots @{ $c->{Votes} };
+
$c->{Votes} = { }; # will crash if we access it again
next;
}
-1, 'eliminating';
if ($c) {
- prf "=== eliminating %s (%s) ===\n", $c->{Cand}, $c->{Desc};
+ prfm "=== eliminating %s \`%s' ===\n", $c->{Cand}, $c->{Desc};
$c->{NonCont} = 'Eliminated';
+
+ sortballots @{ $c->{Votes} };
next;
}
die;
}
-print "done.\n";
+print "Winners:\n";
+
+if ($for_compare) {
+ foreach my $c (sort { $a->{Cand} cmp $b->{Cand} } @elected) {
+ printf " %s\n", $c->{Cand};
+ }
+} else {
+ foreach my $i (0..$#elected) {
+ my $c = $elected[$i];
+ printf " %3d. %-10s %s\n", $i+1, $c->{Cand}, $c->{Desc};
+ }
+}
+
+print "done.\n";
~ian / appendix-a6.git / blobdiff