bin/mo-score

   1 #!/usr/bin/perl
   2
   3 $debug = 0;
   4 $detail = 0;
   5 $html = 0;
   6 $tex = 0;
   7 $extras = 0;
   8 $alt = 0;
   9 $split = 0;
  10 $usage = "Usage: mo-score [--detail] [--alt] [--extras] [--html] [--tex] [--split] <task1> <task2> ...";
  11 while (($arg = $ARGV[0]) =~ /^--([a-z]+)$/) {
  12         shift @ARGV;
  13         $var = "\$$1";
  14         if (!eval "defined $var") { die $usage; }
  15         eval "$var = 1;";
  16 }
  17 @ARGV || die $usage;
  18
  19 @tasks = @ARGV;
  20
  21 print STDERR "Scanning contestants... ";
  22 open (CT, "bin/mo-get-users --full |") || die "Cannot get list of contestants";
  23 while (<CT>) {
  24         chomp;
  25         ($u,$f) = split /\t/;
  26         ($u eq "somebody") && next;
  27         $users{$u}=$f;
  28 }
  29 close CT;
  30 print STDERR 0+keys %users, "\n";
  31
  32 print STDERR "Scanning exceptions... ";
  33 if ($extras && open (EX, "exceptions")) {
  34         while (<EX>) {
  35                 chomp;
  36                 (/^$/ || /^#/) && next;
  37                 @a = split /\s+/;
  38                 $u = shift @a;
  39                 defined $users{$u} || die "Unknown user $u";
  40                 while (@a) { $extra{$u} += shift @a; }
  41         }
  42         close EX;
  43         print STDERR "OK\n";
  44 } else { print STDERR "none\n"; }
  45
  46 print STDERR "Scanning task results... ";
  47 $need_tasks = join("|", @ARGV);
  48 %messages = ();
  49 %error_codes = ();
  50 foreach $u (keys %users) {
  51         opendir (D, "testing/$u") or next;
  52         foreach $t (readdir(D)) {
  53                 $t =~ /^\./ && next;
  54                 $t =~ /$need_tasks/ || next;
  55                 $known_tasks{$t} = 1;
  56                 $tt = "testing/$u/$t/points" . ($alt ? ".alt" : "");
  57                 -f $tt || next;
  58                 print STDERR "$u/$t ";
  59                 open (X, $tt) || die "Unable to open $tt";
  60                 while (<X>) {
  61                         chomp;
  62                         /^(\S+) (-?\d+)\s*(.*)/ || die "Parse error: $_";
  63                         $ttest = $1;
  64                         $tpts = $2;
  65                         $trem = $3;
  66                         $trem =~ s/\[.*//;
  67                         $ttest =~ s/[^0-9]//g unless $split;
  68                         $known_tests{$t}{$ttest} = 1;
  69                         $cmt = $tpts;
  70                         if ($tpts == 0 && $trem ne "OK") {
  71                                 if ($trem =~ /^Compile /) { $cmt = "CE"; }
  72                                 elsif ($trem =~ /^Time limit exceeded/) { $cmt = "TO"; }
  73                                 elsif ($trem =~ /^Exited with error /) { $cmt = "RE"; }
  74                                 elsif ($trem =~ /^Caught fatal signal /) { $cmt = "SG"; }
  75                                 elsif ($trem =~ /^([A-Za-z])\S*\s+([A-Za-z])/) {
  76                                         ($cmt = "$1$2") =~ tr/a-z/A-Z/;
  77                                 }
  78                                 if (!defined $messages{$trem}) {
  79                                         $messages{$trem} = $cmt;
  80                                         if (!defined $error_codes{$cmt}) {
  81                                                 $error_codes{$cmt} = $trem;
  82                                         } else {
  83                                                 $error_codes{$cmt} .= ", $trem";
  84                                         }
  85                                 }
  86                         }
  87                         if (!defined($results{$u}{$t}{$ttest}) || $results{$u}{$t}{$ttest} > $tpts) {
  88                                 $results{$u}{$t}{$ttest} = $tpts;
  89                                 $comment{$u}{$t}{$ttest} = $cmt;
  90                         }
  91                 }
  92                 close X;
  93         }
  94         foreach my $t (keys %known_tasks) {
  95                 $total{$u}{$t} = 0;
  96                 foreach my $pts (values %{$results{$u}{$t}}) { $total{$u}{$t} += $pts; }
  97         }
  98         closedir D;
  99 }
 100 print STDERR "OK\n";
 101
 102 print STDERR "Creating table template... ";
 103 @header = ("Rank","User","Name");
 104 @body = ('','$u','$users{$u}');
 105 @bodysums = ();
 106 @footer = ('"Total"','','');
 107 if (keys %extra) {
 108         push @header, "Extra";
 109         push @body, '$extra{$u}';
 110         $col = 0+@footer;
 111         push @bodysums, $col;
 112         push @footer, "sum($col)";
 113 }
 114 foreach $t (@ARGV) {
 115         defined $known_tasks{$t} || die "Unknown task $t";
 116         push @header, substr($t, 0, 4);
 117         push @body, "(\$xx = \$total{\$u}{'$t'}) > 0 ? \$xx : 0";
 118         $col = 0+@footer;
 119         push @footer, "sum($col)";
 120         push @bodysums, $col;
 121         if ($detail) {
 122                 foreach $s (sort { $a <=> $b } keys %{$known_tests{$t}}) {
 123                         push @header, "$s";
 124                         push @body, "\$comment{\$u}{'$t'}{'$s'}";
 125                         $col = 0+@footer;
 126                         push @footer, "sum($col)";
 127                 }
 128         }
 129 }
 130 push @header, "Total";
 131 push @body, join('+', map { $_ = "\$$_" } @bodysums);
 132 $col = 0+@footer;
 133 push @footer, "sum($col)";
 134 print STDERR "OK\n";
 135
 136 print STDERR "h: ", join(':',@header), "\n" if $debug;
 137 print STDERR "b: ", join(':',@body), "\n" if $debug;
 138 print STDERR "f: ", join(':',@footer), "\n" if $debug;
 139
 140 print STDERR "Filling in results... ";
 141 @table = ();
 142 foreach $u (keys %users) {
 143         $row = [];
 144         foreach my $c (@body) {
 145                 $c =~ s/\$(\d+)/\$\$row[$1]/g;
 146                 $x = eval $c;
 147                 push @$row, (defined $x ? $x : '-');
 148         }
 149         print STDERR "row: ", join(':',@$row), "\n" if $debug;
 150         push @table, $row;
 151 }
 152 print STDERR "OK\n";
 153
 154 print STDERR "Sorting... ";
 155 $sortcol = @{$table[0]} - 1;
 156 $namecol = 2;
 157 @table = sort {
 158         my $p, $an, $bn;
 159         $p = $$b[$sortcol] <=> $$a[$sortcol];
 160         ($an = $$a[$namecol]) =~ s/(\S+)\s+(\S+)/$2 $1/;
 161         ($bn = $$b[$namecol]) =~ s/(\S+)\s+(\S+)/$2 $1/;
 162         $p ? $p : ($an cmp $bn);
 163 } @table;
 164 $i=0;
 165 while ($i < @table) {
 166         $j = $i;
 167         while ($i < @table && ${$table[$i]}[$sortcol] == ${$table[$j]}[$sortcol]) {
 168                 $i++;
 169         }
 170         if ($i == $j+1) {
 171                 ${table[$j]}[0] = "$i.";
 172         } else {
 173                 ${table[$j]}[0] = $j+1 . ".-" . $i . ".";
 174                 $j++;
 175                 while ($j < $i) { ${table[$j++]}[0] = ""; };
 176         }
 177 }
 178 print STDERR "OK\n";
 179
 180 print STDERR "Attaching headers and footers... ";
 181 sub sum { my $col=shift @_; my $t=0; foreach my $z (0..@table-1) { $t += ${$table[$z]}[$col]; } $t; }
 182 map { $_ = eval $_; } @footer;
 183 push @table, \@footer;
 184 unshift @table, \@header;
 185 print STDERR "OK\n";
 186
 187 if ($debug) {
 188         foreach $r (@table) { print join(':',@$r), "\n"; }
 189 } elsif ($html) {
 190         print '<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01//EN" "http://www.w3.org/TR/html40/strict.dtd">', "\n";
 191         print "<HTML><HEAD><TITLE>Rank list</TITLE></HEAD><BODY>\n";
 192         print "<H1>Rank list</H1>\n";
 193
 194         my @perm;
 195         &printHtmlHeader(\@perm);
 196         print "<tbody>";
 197
 198         foreach $r (@table[1..($#table - 1)]) {
 199                 &printHtmlRow(@{$r}[@perm]);
 200         }
 201
 202         print "<tbody>";
 203         &printHtmlRow(@{$table[$#table]}[@perm]);
 204
 205         print "</TABLE>\n";
 206         if ($detail) {
 207                 print "<H2>Error codes</H2><UL>\n";
 208                 foreach $r (sort keys %error_codes) { print "<LI>$r: $error_codes{$r}\n"; }
 209                 print "</UL>\n";
 210         }
 211         print "</BODY></HTML>\n";
 212 } elsif ($tex) {
 213         print "\\error{TeX output not supported yet!}\n";
 214 } else {
 215         foreach $r (@table) { print join("\t",@$r), "\n"; }
 216         print "\n";
 217         foreach $r (sort keys %error_codes) { print "$r: $error_codes{$r}\n"; }
 218 }
 219
 220
 221 sub printHtmlRow {
 222         print "<TR>", join('',map {
 223                 if ($hdr) { $_ = "<TH>$_"; }
 224                 else { $_ = "<TD align=" . (/^[0-9A-Z-]+$/ ? "right" : "left") . (length($_) > 14 ? " width=150" : "") . ">$_"; }
 225         } @_), "\n";
 226 }
 227
 228
 229 sub printHtmlHeader {
 230
 231   my ($perm) = @_;
 232
 233    my $colspec = "<colgroup span=3>";
 234    my $hdr1;
 235    my $hdr2;
 236
 237    @$perm = (0, 1, 2);
 238    my $p = 3;
 239
 240    if ($detail) {
 241      $hdr1 = "<th rowspan=2>Rank<th rowspan=2>User<th rowspan=2>Name";
 242      for my $task (@tasks) {
 243
 244         my $nSub = scalar(keys %{$known_tests{$task}});
 245
 246         $p++;
 247         map { push @$perm, $p++ } (1..$nSub);
 248         push @$perm, $p - $nSub - 1;
 249
 250         $colspec .= "<colgroup span='" . $nSub . "'>\n";
 251         $colspec .= "<colgroup span='1'>\n";
 252         $hdr1 .= "<th colspan='" . ($nSub + 1) . "' style='border-bottom:1px solid black;'>$task";
 253         $hdr2 .= join("", map { "<th>$_" } sort {$a <=> $b} keys %{$known_tests{$task}});
 254         $hdr2 .= "<th>Total";
 255      }
 256
 257      $hdr1 .= "<th rowspan='2'>Total";
 258
 259    } else {  ## no detail
 260
 261      $hdr1 = "<th>Rank<th>User<th>Name";
 262
 263      for my $task (@tasks) {
 264         push @$perm, $p++;
 265         $hdr1 .= "<th>$task";
 266      }
 267      $hdr1 .= "<th>Total";
 268      $colspec .= "<colgroup span='" . scalar (@tasks) . "'>";
 269    }
 270
 271    push @$perm, $p++;
 272
 273    print "<TABLE rules=groups frame=all border='1' cellpadding='2'>\n";
 274    print "$colspec<colgroup span='1'>\n";
 275    print "<tr>$hdr1</tr>\n";
 276    print "<tr>$hdr2</tr>\n" if $detail;
 277
 278 }