Function main

tools/traverse.rs:142–210

()

}

fn main() -> eyre::Result<()> {
    color_eyre::install()?;
    let args = CmdlineOptions::parse();

    let filter = tracing_subscriber::EnvFilter::builder()
        .with_default_directive(args.log_level.into())
        .from_env_lossy();
    tracing_subscriber::fmt()
        .with_env_filter(filter)
        .with_ansi(true)
        .with_writer(std::io::stderr)
        .init();

    let seed = generate_random_seed_if_not_specified(args.seed);
    tracing::info!("Seeding RNG with: {seed}");
    let mut rng = StdRng::seed_from_u64(seed);

    let reader = get_input_reader(&args.input)?;
    let mut graph: GeometryGraph<Undirected> = read_tgf_graph(reader);

    let mut num_traversals = if args.random_traversals {
        let n = args.traversals * 2; // changes mean
        let p = 0.5; // changes skew
        let dist = Binomial::new(n as u64, p)?;
        dist.sample(&mut rng)
    } else {
        args.traversals as u64
    };
    if num_traversals == 0 {
        num_traversals = 1;
    }
    tracing::debug!("Making {num_traversals} traversals");

    let traversals = std::iter::repeat_with(|| {
        let mut length = if args.random_length {
            let n = args.traversals * 2; // changes mean
            let p = 0.5; // changes skew
            let dist = Binomial::new(n as u64, p).unwrap();
            dist.sample(&mut rng)
        } else {
            args.length as u64
        };
        if length < 2 {
            length = 2;
        }
        tracing::debug!("Making random traversal with length {length}");
        random_traversal(
            length as usize,
            args.remove_after_traverse,
            &mut graph,
            &mut rng,
        )
    })
    .take(num_traversals as usize)
    .flatten()
    .map(Geometry::LineString);

    let mut writer = get_output_writer(&args.output)?;