From 15a021b7ecb54c257916239a8126025edc5ffb3e Mon Sep 17 00:00:00 2001
From: Marius Gavrilescu <marius@ieval.ro>
Date: Sun, 3 Aug 2014 00:22:37 +0300
Subject: [PATCH] Add a method for computing standings

---
 lib/Gruntmaster/Data.pm | 55 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)

diff --git a/lib/Gruntmaster/Data.pm b/lib/Gruntmaster/Data.pm
index 5754491..ae13295 100644
--- a/lib/Gruntmaster/Data.pm
+++ b/lib/Gruntmaster/Data.pm
@@ -19,6 +19,7 @@ our $VERSION = '5999.000_007';
 
 use Lingua::EN::Inflect qw/PL_N/;
 use JSON qw/decode_json/;
+use List::Util qw/sum/;
 use Sub::Name qw/subname/;
 
 use constant PROBLEM_PUBLIC_COLUMNS => [qw/id author writer level name owner private statement timeout olimit value/];
@@ -40,6 +41,60 @@ BEGIN {
 	}
 }
 
+use constant LEVEL_VALUES => {
+	beginner => 100,
+	easy => 250,
+	medium => 500,
+	hard => 1000,
+};
+
+sub calc_score{
+	my ($mxscore, $time, $tries, $totaltime) = @_;
+	my $score = $mxscore;
+	$time = 0 if $time < 0;
+	$time = 300 if $time > $totaltime;
+	$score = ($totaltime - $time) / $totaltime * $score;
+	$score -= $tries / 10 * $mxscore;
+	$score = $mxscore * 3 / 10 if $score < $mxscore * 3 / 10;
+	int $score + 0.5
+}
+
+sub standings {
+	my ($self, $ct) = @_;
+	$ct &&= $self->contest($ct);
+
+	my @problems = map { $_->problem } $self->contest_problems->search({contest => $ct && $ct->id}, {qw/join problem order_by problem.level/});
+	my (%scores, %tries);
+	for my $job ($self->jobs->search({contest => $ct && $ct->id}, {order_by => 'id'})) {
+		if ($ct) {
+			my $open = $self->opens->find($ct->id, $job->problem->id, $job->owner->id);
+			my $time = $job->date - ($open ? $open->time : $ct->start);
+			next if $time < 0;
+			my $value = $job->problem->value // LEVEL_VALUES->{$job->problem->level};
+			my $factor = $job->result ? 0 : 1;
+			$factor = $1 / 100 if $job->result_text =~ /^(\d+ )/;
+			$scores{$job->owner->id}{$job->problem->id} = int ($factor * calc_score ($value, $time, $tries{$job->owner->id}{$job->problem->id}++, $ct->stop - $ct->start));
+		} else {
+			no warnings 'numeric';
+			$scores{$job->owner->id}{$job->problem->id} = 0 + $job->result_text || ($job->result ? 0 : 100)
+		}
+	}
+
+	my @st = sort { $b->{score} <=> $a->{score} or $a->{user}->id cmp $b->{user}->id} map {
+		my $user = $_;
+		+{
+			user => $self->user($user),
+			score => sum (values $scores{$user}),
+			scores => [map { $scores{$user}{$_->id} // '-'} @problems],
+			problems => $ct,
+		}
+	} keys %scores;
+
+	$st[0]->{rank} = 1;
+	$st[$_]->{rank} = $st[$_ - 1]->{rank} + ($st[$_]->{score} < $st[$_ - 1]->{score}) for 1 .. $#st;
+	@st
+}
+
 sub user_list {
 	my $rs = $_[0]->users->search(undef, {order_by => 'name', columns => USER_PUBLIC_COLUMNS});
 	[ map +{ $_->get_columns }, $rs->all ]
-- 
2.39.2