Improved sampling for z values

Cargo.lock

@@ -576,6 +576,12 @@ dependencies = [
  "winapi",
 ]
 
+[[package]]
+name = "libm"
+version = "0.2.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "33a33a362ce288760ec6a508b94caaec573ae7d3bbbd91b87aa0bad4456839db"
+
 [[package]]
 name = "lock_api"
 version = "0.4.7"
@@ -771,6 +777,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "578ede34cf02f8924ab9447f50c28075b4d3e5b269972345e7e0372b38c6cdcd"
 dependencies = [
  "autocfg",
+ "libm",
 ]
 
 [[package]]
@@ -917,6 +924,7 @@ dependencies = [
  "log",
  "pixels",
  "rand",
+ "rand_distr",
  "timer",
  "winit",
  "winit_input_helper",
@@ -961,6 +969,16 @@ dependencies = [
  "getrandom",
 ]
 
+[[package]]
+name = "rand_distr"
+version = "0.4.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "32cb0b9bc82b0a0876c2dd994a7e7a2683d3e7390ca40e6886785ef0c7e3ee31"
+dependencies = [
+ "num-traits",
+ "rand",
+]
+
 [[package]]
 name = "range-alloc"
 version = "0.1.2"

Cargo.toml

@@ -11,6 +11,7 @@ env_logger = "0.9.0"
 log = "0.4.17"
 pixels = "0.9.0"
 rand = "0.8.5"
+rand_distr = "0.4.3"
 timer = "0.2.0"
 winit = "0.26.1"
 winit_input_helper = "0.12.0"

src/main.rs

@@ -38,11 +38,12 @@ fn main() {
 
     let mut drops = Vec::new();
     let mut rng = rand::thread_rng();
-    for _i in 0..200 {
+
+    for _i in 0..1000 {
         drops.push(Drop {
             x: (rng.gen_range(0..WIDTH as i32)),
             y: (rng.gen_range(0..HEIGHT as i32)),
-            z: (rng.gen_range(1..20)),
+            z: ((rng.sample(rand_distr::Geometric::new(0.2).unwrap())+1) as i32),
         })
     }