flowmaid 0.18.0

//! Simple benchmark: `cargo run --release --example bench`
//! Emit mode:         `cargo run --release --example bench -- emit > big.mmd`

use std::time::Instant;

/// Deterministic LCG to stay pure-std with reproducible results.
struct Lcg(u64);
impl Lcg {
    fn next(&mut self, n: usize) -> usize {
        self.0 = self
            .0
            .wrapping_mul(6364136223846793005)
            .wrapping_add(1442695040888963407);
        ((self.0 >> 33) as usize) % n
    }
}

/// Build a layered diagram: `layers` layers x `width` nodes per
/// layer, each node gets 2 edges to the next layer, plus a few
/// back-edges from the last layer to the first (stresses the
/// heaviest routing path).
fn synth(layers: usize, width: usize) -> String {
    let mut s = String::from("flowchart TD\n");
    let mut rng = Lcg(42);
    for l in 0..layers.saturating_sub(1) {
        for w in 0..width {
            let from = l * width + w;
            for _ in 0..2 {
                let to = (l + 1) * width + rng.next(width);
                s.push_str(&format!(
                    "N{}[Node {}] --> N{}[Node {}]\n",
                    from, from, to, to
                ));
            }
        }
    }
    for _ in 0..layers {
        let a = (layers - 1) * width + rng.next(width);
        let b = rng.next(width);
        s.push_str(&format!("N{} --> N{}\n", a, b));
    }
    s
}

/// Run `f` three times, keep the best time (ms) to dampen noise.
fn best<T>(mut f: impl FnMut() -> T) -> (T, f64) {
    let mut t_best = f64::INFINITY;
    let mut out = None;
    for _ in 0..3 {
        let t = Instant::now();
        let v = f();
        t_best = t_best.min(t.elapsed().as_secs_f64() * 1e3);
        out = Some(v);
    }
    (out.unwrap(), t_best)
}

fn main() {
    if std::env::args().nth(1).as_deref() == Some("emit") {
        print!("{}", synth(200, 25));
        return;
    }
    println!(
        "{:>6} {:>6} {:>6} | {:>9} {:>9} {:>9} | {:>8}",
        "nodes", "edges", "layers", "parse", "layout", "render*", "SVG"
    );
    let cases: &[(usize, usize)] = &[(10, 5), (20, 10), (50, 20), (100, 25), (200, 25), (2, 2500)];
    for &(l, w) in cases {
        let src = synth(l, w);
        let (g, tp) = best(|| flowmaid::parser::parse(&src).unwrap());
        let (_lay, tl) = best(|| flowmaid::layout::layout(&g));
        let (svg, tr) = best(|| flowmaid::render::render(&g));
        println!(
            "{:>6} {:>6} {:>6} | {:>7.2}ms {:>7.2}ms {:>7.2}ms | {:>6}KB",
            g.nodes.len(),
            g.edges.len(),
            l,
            tp,
            tl,
            tr,
            svg.len() / 1024
        );
    }
    println!("* render includes running layout internally");
}