initial commit

src/main.rs

@@ -0,0 +1,203 @@
+use mcps::{schedule::Schedule, simulation::run_multiple_simulations};
+
+use clap::{Arg, Command};
+
+use std::time::Duration;
+
+const AFTER_HELP_TEXT: &str = "\
+Examples of supported file formats:
+
+    YAML Format:
+
+        num_workers: 5
+        estimate_confidence: 0.80
+        tasks:
+        - id: DesignPhase
+            min_time: 1.5  # 1.5 days
+            max_time: 3.5  # 3.5 days
+            dependencies: []
+        - id: ImplementationPhase
+            min_time: 2.25  # 2.25 days
+            max_time: 4.75  # 4.75 days
+            dependencies: [DesignPhase]
+
+    JSON Format:
+
+        {
+            \"num_workers\": 5,
+            \"estimate_confidence\": 0.80,
+            \"tasks\": [
+                {
+                    \"id\": \"DesignPhase\",
+                    \"min_time\": 1.5,
+                    \"max_time\": 3.5,
+                    \"dependencies\": []
+                },
+                {
+                    \"id\": \"ImplementationPhase\",
+                    \"min_time\": 2.25,
+                    \"max_time\": 4.75,
+                    \"dependencies\": [\"DesignPhase\"]
+                }
+            ]
+        }
+
+    Note: `estimate_confidence` is the confidence (0.0,1.0) that the
+          actual values fall in the task estimate ranges. For example,
+          0.8 means you believe that the actual time it will take to
+          complete a task falls in your [min_time,max_time] estimate
+          range for that task 80% of the time.";
+
+fn main() -> Result<(), Box<dyn std::error::Error>> {
+    let matches = Command::new("Monte Carlo Project Scheduler")
+        .version("1.0")
+        .author("Stephen Waits <steve@waits.net>")
+        .about("Runs Monte Carlo simulations on project schedules")
+        .arg(
+            Arg::new("filename")
+                .help("Path to the schedule file (.yaml or .json).")
+                .required(true)
+                .index(1),
+        )
+        .arg(
+            Arg::new("iterations")
+                .short('i')
+                .long("iterations")
+                .help("Number of iterations to run")
+                .default_value("50000"),
+        )
+        .arg(
+            Arg::new("workers")
+                .short('w')
+                .long("workers")
+                .help("Override number of workers specified in schedule file")
+                .value_name("num_workers"),
+        )
+        .after_help(AFTER_HELP_TEXT)
+        .get_matches();
+
+    let schedule_path = matches.get_one::<String>("filename").unwrap();
+    let num_simulations: usize = matches.get_one::<String>("iterations").unwrap().parse()?;
+
+    // Ensure iterations is >= 100
+    if num_simulations < 100 {
+        return Err("Iterations must be at least 100.".into());
+    }
+
+    // Load the schedule
+    let mut schedule = Schedule::from_file(schedule_path)?;
+
+    // Check if workers are overridden by command-line argument
+    if let Some(workers_str) = matches.get_one::<String>("workers") {
+        let workers: usize = workers_str.parse()?;
+        if workers < 1 {
+            return Err("Invalid number of workers, must be 1 or more".into());
+        }
+        schedule.num_workers = workers;
+    }
+
+    // Monte Carlo simulation
+    let (project_durations, effort_times) = run_multiple_simulations(&schedule, num_simulations);
+
+    // Results output
+    print_ascii_cdf(
+        &project_durations,
+        "Project Completion Time (by probability)",
+        "Duration",
+    );
+    println!();
+    print_ascii_cdf(
+        &effort_times,
+        "Total Work Effort (by probability)",
+        "Effort",
+    );
+
+    Ok(())
+}
+
+fn print_ascii_cdf(data: &[Duration], title: &str, units: &str) {
+    let mut sorted_data = data.to_vec();
+    sorted_data.sort_unstable();
+    let min = sorted_data[0];
+    let max = *sorted_data.last().unwrap();
+
+    let width = 60; // Total width of the field
+
+    // Calculate padding
+    let padding = (width - title.len()) / 2;
+
+    // Create the centered title with padding on both sides
+    let centered_title = format!(
+        "{:padding_left$}{}{:padding_right$}",
+        "",
+        title,
+        "",
+        padding_left = padding,
+        padding_right = width - padding - title.len()
+    );
+
+    println!("────┬────────────────────────────────────────────────────────────┬──────────");
+    println!("%ile│{}│{:>10}", centered_title, units);
+    println!("────┼────────────────────────────────────────────────────────────┼──────────");
+
+    let bar_width = 60;
+
+    for i in 0..=20 {
+        let percentile = 100 - i * 5;
+        let lower_percentile = if i == 20 {
+            0.0000000001
+        } else {
+            (95 - i * 5) as f64 / 100.0
+        };
+        let upper_percentile = if i == 0 {
+            0.9999999999
+        } else {
+            (100 - i * 5) as f64 / 100.0
+        };
+        let lower_index = (lower_percentile * (sorted_data.len() - 1) as f64).round() as usize;
+        let upper_index = (upper_percentile * (sorted_data.len() - 1) as f64).round() as usize;
+        let mid_duration = (sorted_data[lower_index] + sorted_data[upper_index]) / 2;
+        let days = mid_duration.as_secs_f64() / 86400.0;
+        let normalized_position = (days - min.as_secs_f64() / 86400.0)
+            / (max.as_secs_f64() / 86400.0 - min.as_secs_f64() / 86400.0);
+        let bar_position = (normalized_position * bar_width as f64).round() as usize;
+
+        let (fg, bg) = if i % 2 == 0 {
+            ('░', '▓')
+        } else {
+            ('▒', '█')
+        };
+
+        let color_code = match percentile {
+            0..=50 => "\x1b[31m",    // Red
+            55..=75 => "\x1b[33m",   // Orange
+            80..=90 => "\x1b[32m",   // Green
+            95..=95 => "\x1b[33m",   // Orange
+            100..=100 => "\x1b[31m", // Red
+            _ => "\x1b[0m",          // Default (shouldn't happen)
+        };
+        let reset_code = "\x1b[0m";
+
+        let bar_with_divider: String = (0..bar_width)
+            .map(|j| match j {
+                _ if j == bar_position => '▮',
+                _ if j < bar_position => fg,
+                _ => bg,
+            })
+            .collect();
+
+        println!(
+            "{}{:>4}{}│{}{}{}│{}{:5.0} days{}",
+            color_code,
+            format!("p{}", percentile),
+            reset_code,
+            color_code,
+            bar_with_divider,
+            reset_code,
+            color_code,
+            days,
+            reset_code
+        );
+    }
+    println!("────┴────────────────────────────────────────────────────────────┴──────────");
+}