#! /usr/bin/perl
use strict;
use warnings;
no warnings qw(once);
use CGI;
use CGI::Carp qw( fatalsToBrowser );
use DBI;
use POSIX;
use Devel::Peek;
use locale;
require '../config.pm';

my $dbh = DBI->connect($config::local_connstr, $config::local_username, $config::local_password)
	or die "connect: " . $DBI::errstr;
$dbh->{AutoCommit} = 0;
$dbh->{RaiseError} = 1;

my $trials = 25_000;

binmode STDOUT, ':utf8';

my %players = ();
my %ratings = ();
my @matches = ();

sub sanitize {
	$_ = shift;
	y/ \t\n<>&/_/;
	return $_;
}

sub color {
	my $x = shift;
	return int(255.0 * ($x ** (1.80)));
}

sub make_table {
	my $used_ratings = shift;

	print <<"EOF";

    <table>
      <tr>
        <th></th>
EOF

	POSIX::setlocale(&POSIX::LC_ALL, 'C');

	my $tmpnam = POSIX::tmpnam();
	open MCCALC, ">", $tmpnam
		or die "$tmpnam: $!";

	printf MCCALC "%d\n", scalar keys %players;

	for my $id (keys %players) {
		if (exists($used_ratings->{$id})) {
			printf MCCALC "%s %f\n", $id, $used_ratings->{$id};
		} else {
			printf MCCALC "%s %f\n", $id, 1500.0;
		}
	}

	for my $match (@matches) {
		printf MCCALC "%s %s %d %d\n", $match->[0], $match->[1], $match->[2], $match->[3];
	}
	close MCCALC;

	POSIX::setlocale(&POSIX::LC_ALL, 'nb_NO.UTF-8');

	my %prob = ();

	open MCCALC, "$config::base_dir/mcwordfeud $trials < $tmpnam |"
		or die "mccalc: $!";
	while (<MCCALC>) {
		chomp;
		my @x = split /\s+/;
		my $id = $x[0];
		my $player = $players{$id};
		$prob{$player} = [ @x[1..$#x] ];
	}
	close MCCALC;
	#unlink $tmpnam;

	my $num_games = scalar keys %prob;
	for my $i (1..$num_games) {
		print "        <th>$i.</th>\n";
	}
	print "        <th>NEDRYKK</th>\n";
	print "      </tr>\n";

	for my $player (sort { $a cmp $b } keys %prob) {
		print "      <tr>\n";
		print "        <th>$player</th>\n";

		for my $i (1..$num_games) {
			my $pn = $prob{$player}->[$i - 1] / $trials;

			my $r = color(1.0 - $pn / 3);
			my $g = color(1.0 - $pn / 3);
			my $b = color(1.0);

			if ($i == 1) {
				($g, $b) = ($b, $g);
			} elsif ($i >= $num_games - 1) {
				($r, $b) = ($b, $r);
			}

			printf "        <td style=\"background-color: rgb($r, $g, $b)\">%.1f%%</td>\n", $pn * 100.0;
		}

		{
			my $pn = ($prob{$player}->[$num_games - 1] + $prob{$player}->[$num_games - 2]) / $trials;

			my $r = color(1.0);
			my $g = color(1.0 - $pn / 3);
			my $b = color(1.0 - $pn / 3);
			printf "        <td style=\"background-color: rgb($r, $g, $b)\">%.1f%%</td>\n", $pn * 100.0;
		}
		print "      </tr>\n";
	}

	print << "EOF";
    </table>
EOF
}

# Get players and ratings
my $season = 18;
my $division = 1;
my $subdivision = 1;

my $q = $dbh->prepare('SELECT fotballdeltagere.id,fotballdeltagere.navn,rating FROM fotballdeltagere JOIN fotballserier ON fotballdeltagere.serie=fotballserier.nr JOIN ratings ON fotballdeltagere.id=ratings.id AND sesong=? AND divisjon=? AND avdeling=?');
$q->execute($season, $division, $subdivision);

while (my $ref = $q->fetchrow_hashref) {
	my $id = $ref->{'id'};
	$players{$id} = sanitize($ref->{'navn'});
	$ratings{$id} = $ref->{'rating'};
}
$q->finish;

$q = $dbh->prepare('
SELECT
  d1.id AS p1, d2.id AS p2, maalfor AS score1, maalmot AS score2
FROM fotballresultater r
  JOIN fotballserier s ON r.serie=s.nr
  JOIN fotballdeltagere d1 ON r.lagrecno=d1.nr AND r.serie=d1.serie
  JOIN fotballdeltagere d2 ON r.motstander=d2.nr AND r.serie=d2.serie
WHERE
  sesong=? AND divisjon=? AND avdeling=?
  AND lagrecno > motstander
');
$q->execute($season, $division, $subdivision);

while (my $ref = $q->fetchrow_hashref) {
	push @matches, [ $ref->{'p1'}, $ref->{'p2'}, $ref->{'score1'}, $ref->{'score2'} ];
}
$q->finish;

print CGI->header(-type=>'text/html; charset=utf-8', -expires=>'Thu, 01 Dec 1994 16:00:00 GMT');
print <<"EOF";
<html>
  <head>
    <title>WLoH-plasseringsannsynlighetsberegning</title>
    <style type="text/css">
body {
	color: black;
	background: white;
	font-family: sans-serif;
}
table {
	border-collapse: collapse;
	border: 1px solid black;
}
td, th {
	border: 1px solid black;
	padding: 5px;
}
td {
	text-align: right;
}
    </style>
  </head>
  <body>
    <h1>WLoH-plasseringsannsynlighetsberegning</h1>

    <p>Beregningen tar ikke hensyn til ujevn spillestyrke, ting som er sagt i forumet e.l.;
      den antar at samtlige uspilte kamper trekkes fra en normalfordeling med standardavvik
      82,9 poeng. Sannsynlighetene kan summere til andre tall enn 100% pga. avrunding.
      Tallene vil variere litt fra gang til gang fordi utregningen skjer ved randomisering.</p>

    <p>Andre divisjoner enn 1. divisjon kommer etter hvert.</p>

    <p>Spillerne er sortert etter nick.</p>
EOF

make_table({});

print <<"EOF";
    <p>Under er en variant som pr&oslash;ver &aring; ta relativ spillestyrke med i betraktningen.
      Disse er basert p&aring; WLoH-data opp til og med sesong 17 (takk til Lobotommy for tilgang
      til disse!), men modellen er forel&oslash;pig ikke fullstendig tunet.</p>
EOF

make_table(\%ratings);

print << "EOF";
    </table>
  </body>
</html>
EOF
