Src. Implement option for distribution

diff --git a/src/main.rs b/src/main.rs
index 7a194d9591dafbeb9f1dd4c696a45d22480c7f91..56721ab2a770a840bdaeab5c33323f91c657590e 100644 (file)
--- a/src/main.rs
+++ b/src/main.rs
@@ -101,7 +101,11 @@ fn lognormal(n: i32, std: f64) -> Vec<f64> {
 
 struct Args {
 
-    /// Ranking distribution skewness
+    /// Ranking distribution
+    #[arg(short, default_value_t = String::from("lognormal"))]
+    distribution: String,
+
+    /// Skewness of the ranking
     #[arg(short)]
     skewness: Option<String>,
 }
@@ -129,9 +133,14 @@ fn main() {
 
     let ranked: Vec<String> = input();
     let num = ranked.len() as i32;
-    let _prio1 = delta(num);
-    let _prio2 = exp(num);
-    let prio3 = lognormal(num, skew);
+
+    let prio = if args.distribution == String::from("delta") {
+        delta(num)
+    } else if  args.distribution == String::from("exponential") {
+        exp(num)
+    } else { // Default
+        lognormal(num, skew)
+    };
 
     if DEBUG {
         for l in ranked.iter() {
@@ -140,6 +149,6 @@ fn main() {
         eprintln!("\x08\x08.");
     }
 
-    output(&prio3, &ranked);
+    output(&prio, &ranked);
 
 }