From 651c339cedb7a6e09cf7209a538ab5aca7046711 Mon Sep 17 00:00:00 2001
From: Kiana Sheibani <kiana.a.sheibani@gmail.com>
Date: Sat, 23 Sep 2023 02:36:28 -0400
Subject: [PATCH] Write score calculation code

---
 cli/src/datasets.rs | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/cli/src/datasets.rs b/cli/src/datasets.rs
index b0098e6..373c172 100644
--- a/cli/src/datasets.rs
+++ b/cli/src/datasets.rs
@@ -24,3 +24,32 @@ pub fn new_dataset(dataset: &Path) -> sqlite::Result<Connection> {
     connection.execute(query)?;
     Ok(connection)
 }
+
+// Score calculation
+
+/// Calculate the collective expected score for each team.
+fn expected_scores(ratings: &Teams<&mut f64>) -> Vec<f64> {
+    let qs: Vec<f64> = ratings
+        .into_iter()
+        .map(|es| 10_f64.powf(es.iter().map(|x| **x).sum::<f64>() / es.len() as f64 / 400.0))
+        .collect();
+    let sumq: f64 = qs.iter().sum();
+    qs.into_iter().map(|q| q / sumq).collect()
+}
+
+/// Adjust the ratings of each player based on who won.
+fn adjust_ratings(ratings: Teams<&mut f64>, winner: usize) {
+    let exp_scores = expected_scores(&ratings);
+
+    ratings
+        .into_iter()
+        .zip(exp_scores.into_iter())
+        .enumerate()
+        .for_each(|(i, (es, exp_sc))| {
+            let len = es.len() as f64;
+            let score = f64::from(winner == i);
+            es.into_iter()
+                .for_each(|e| *e += 40.0 * (score - exp_sc) / len);
+        })
+}
+