use const for probability

wooster0 · Aug 21, 2021 · 085c055 · 085c055
1 parent dbe0edf
commit 085c055
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/grid/random.rs b/src/grid/random.rs
@@ -5,10 +5,14 @@ use rand::SeedableRng;
 use super::{Cell, Grid};
 use terminal::util::Size;
 
+const PROBABILITY: f64 = 0.75;
+
 fn random_cells(size: u32) -> Vec<Cell> {
     let mut cells = Vec::<Cell>::with_capacity(size as usize);
     let mut small_rng = SmallRng::from_entropy();
-    let distribution = Bernoulli::new(0.75).unwrap();
+
+    let distribution = Bernoulli::new(PROBABILITY);
+    let distribution = distribution.unwrap(); // `PROBABILITY` is in range 0 to 1
 
     for _ in 0..size {
         let filled = distribution.sample(&mut small_rng);