create perform_generations method

HyperCodec · Oct 1, 2024 · 92bcce1 · 92bcce1
1 parent bb324bc
commit 92bcce1
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/genetic-rs-common/src/lib.rs b/genetic-rs-common/src/lib.rs
@@ -158,6 +158,13 @@ where
             self.next_gen.next_gen(rewards)
         });
     }
+
+    /// Calls [`next_generation`][GeneticSim::next_generation] `count` number of times.
+    pub fn perform_generations(&mut self, count: usize) {
+        for _ in 0..count {
+            self.next_generation();
+        }
+    }
 }
 
 #[cfg(feature = "rayon")]
@@ -191,6 +198,13 @@ where
             self.next_gen.next_gen(rewards)
         });
     }
+
+    /// Calls [`next_generation`][GeneticSim::next_generation] `count` number of times.
+    pub fn perform_generations(&mut self, count: usize) {
+        for _ in 0..count {
+            self.next_generation();
+        }
+    }
 }
 
 #[cfg(feature = "genrand")]