It works, or so it seems

src/history.rs

@@ -11,13 +11,16 @@ pub struct History {
     gamma: f64,
     p_draw: f64,
     time: bool,
+    pub epsilon: f64,
+    pub iterations: usize,
+    pub verbose: bool,
 }
 
 impl History {
     pub fn new<C: AsRef<str> + Clone>(
         composition: Vec<Vec<Vec<C>>>,
         results: Vec<Vec<u16>>,
-        times: Vec<u64>,
+        times: Vec<f64>,
         priors: HashMap<String, Player>,
         mu: f64,
         sigma: f64,
@@ -60,6 +63,9 @@ impl History {
             gamma,
             p_draw,
             time: !times.is_empty(),
+            epsilon: 1e-6,
+            iterations: 30,
+            verbose: true,
         };
 
         this.trueskill(composition, results, times);
@@ -71,10 +77,10 @@ impl History {
         &mut self,
         composition: Vec<Vec<Vec<C>>>,
         results: Vec<Vec<u16>>,
-        times: Vec<u64>,
+        times: Vec<f64>,
     ) {
         let o = if self.time {
-            utils::sortperm(&times, false)
+            utils::sort_time(&times, false)
         } else {
             (0..composition.len()).collect::<Vec<_>>()
         };
@@ -83,7 +89,11 @@ impl History {
 
         while i < self.size {
             let mut j = i + 1;
-            let t = if self.time { times[o[i]] } else { i as u64 + 1 };
+            let t = if self.time {
+                times[o[i]]
+            } else {
+                i as f64 + 1.0
+            };
 
             while self.time && j < self.size && times[o[j]] == t {
                 j += 1;
@@ -173,15 +183,11 @@ impl History {
     }
 
     pub fn convergence(&mut self) -> ((f64, f64), usize) {
-        let epsilon = 1e-6;
-        let iterations = 30;
-        let verbose = true;
-
         let mut step = (f64::INFINITY, f64::INFINITY);
         let mut i = 0;
 
-        while (step.0 > epsilon || step.1 > epsilon) && i < iterations {
-            if verbose {
+        while (step.0 > self.epsilon || step.1 > self.epsilon) && i < self.iterations {
+            if self.verbose {
                 print!("Iteration = {}", i);
             }
 
@@ -189,12 +195,12 @@ impl History {
 
             i += 1;
 
-            if verbose {
+            if self.verbose {
                 println!(", step = {:?}", step);
             }
         }
 
-        if verbose {
+        if self.verbose {
             println!("End");
         }
 
@@ -278,7 +284,7 @@ mod tests {
         let mut h = History::new(
             composition,
             results,
-            vec![1, 2, 3],
+            vec![1.0, 2.0, 3.0],
             priors,
             MU,
             BETA,
@@ -319,7 +325,7 @@ mod tests {
             vec![vec!["cj"], vec!["aj"]],
         ];
         let results = vec![vec![1, 0], vec![1, 0], vec![1, 0]];
-        let times = vec![1, 1, 1];
+        let times = vec![1.0, 1.0, 1.0];
 
         let mut priors = HashMap::new();
 
@@ -396,7 +402,7 @@ mod tests {
             vec![vec!["cj"], vec!["aj"]],
         ];
         let results = vec![vec![1, 0], vec![1, 0], vec![1, 0]];
-        let times = vec![1, 2, 3];
+        let times = vec![1.0, 2.0, 3.0];
 
         let mut priors = HashMap::new();
 
@@ -476,7 +482,7 @@ mod tests {
             vec![vec!["cj"], vec!["aj"]],
         ];
         let results = vec![vec![1, 0], vec![1, 0], vec![1, 0]];
-        let times = vec![5, 6, 7];
+        let times = vec![5.0, 6.0, 7.0];
 
         let mut priors = HashMap::new();