cheesy-arena-lite/game/ranking_fields.go

// Copyright 2017 Team 254. All Rights Reserved.
// Author: pat@patfairbank.com (Patrick Fairbank)
//
// Game-specific fields by which teams are ranked and the logic for sorting rankings.

package game

import "math/rand"

type RankingFields struct {
	RankingPoints     int
	MatchPoints       int
	AutoPoints        int
	RotorPoints       int
	TakeoffPoints     int
	PressurePoints    int
	Random            float64
	Wins              int
	Losses            int
	Ties              int
	Disqualifications int
	Played            int
}

type Ranking struct {
	TeamId int
	Rank   int
	RankingFields
}

type Rankings []*Ranking

func (fields *RankingFields) AddScoreSummary(ownScore *ScoreSummary, opponentScore *ScoreSummary, disqualified bool) {
	fields.Played += 1

	if disqualified {
		// Don't award any points.
		fields.Disqualifications += 1
		return
	}

	// Assign ranking points and wins/losses/ties.
	if ownScore.Score > opponentScore.Score {
		fields.RankingPoints += 2
		fields.Wins += 1
	} else if ownScore.Score == opponentScore.Score {
		fields.RankingPoints += 1
		fields.Ties += 1
	} else {
		fields.Losses += 1
	}
	if ownScore.PressureGoalReached {
		fields.RankingPoints += 1
	}
	if ownScore.RotorGoalReached {
		fields.RankingPoints += 1
	}

	// Assign tiebreaker points.
	fields.MatchPoints += ownScore.Score
	fields.AutoPoints += ownScore.AutoPoints
	fields.RotorPoints += ownScore.RotorPoints
	fields.TakeoffPoints += ownScore.TakeoffPoints
	fields.PressurePoints += ownScore.PressurePoints

	// Store a random value to be used as the last tiebreaker if necessary.
	fields.Random = rand.Float64()
}

// Helper function to implement the required interface for Sort.
func (rankings Rankings) Len() int {
	return len(rankings)
}

// Helper function to implement the required interface for Sort.
func (rankings Rankings) Less(i, j int) bool {
	a := rankings[i]
	b := rankings[j]

	// Use cross-multiplication to keep it in integer math.
	if a.RankingPoints*b.Played == b.RankingPoints*a.Played {
		if a.MatchPoints*b.Played == b.MatchPoints*a.Played {
			if a.AutoPoints*b.Played == b.AutoPoints*a.Played {
				if a.RotorPoints*b.Played == b.RotorPoints*a.Played {
					if a.TakeoffPoints*b.Played == b.TakeoffPoints*a.Played {
						if a.PressurePoints*b.Played == b.PressurePoints*a.Played {
							return a.Random > b.Random
						}
						return a.PressurePoints*b.Played > b.PressurePoints*a.Played
					}
					return a.TakeoffPoints*b.Played > b.TakeoffPoints*a.Played
				}
				return a.RotorPoints*b.Played > b.RotorPoints*a.Played
			}
			return a.AutoPoints*b.Played > b.AutoPoints*a.Played
		}
		return a.MatchPoints*b.Played > b.MatchPoints*a.Played
	}
	return a.RankingPoints*b.Played > b.RankingPoints*a.Played
}

// Helper function to implement the required interface for Sort.
func (rankings Rankings) Swap(i, j int) {
	rankings[i], rankings[j] = rankings[j], rankings[i]
}