Function main

tools/attractor.rs:216–277

()

}

fn main() -> eyre::Result<()> {
    color_eyre::install()?;
    let mut 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 dynamical_system = build_dynamical_system_function(&args)?;

    let dist = Uniform::new(-1.0, 1.0).unwrap();
    if args.num_points > 1 {
        args.initial_x = None;
        args.initial_y = None;
    }

    let mut initial_values = Vec::with_capacity(args.num_points as usize);
    for _ in 0..args.num_points {
        let initial_x = args.initial_x.unwrap_or_else(|| dist.sample(&mut rng));
        let initial_y = args.initial_y.unwrap_or_else(|| dist.sample(&mut rng));
        initial_values.push((initial_x, initial_y));
    }

    let expected_coords = (args.iterations * args.num_points) as usize;
    let mut formatter = AttractorFormatter::new(
        args.output_format,
        args.output,
        expected_coords,
        args.width,
        args.height,
    )?;

    let start = std::time::Instant::now();
    for (i, (mut x, mut y)) in initial_values.into_iter().enumerate() {
        tracing::trace!("i={i}: Starting at: ({x}, {y})");
        for j in 0..args.iterations {
            let (x_new, y_new) = dynamical_system(x, y);
            if !x_new.is_finite() || !y_new.is_finite() || x_new.abs() > 1e6 || y_new.abs() > 1e6 {
                tracing::warn!(
                    "Dynamical system produced non-finite value at iteration {j} for point {i}: ({x}, {y})"
                );
                break;
            }
            (x, y) = (x_new, y_new);
            formatter.handle_point(x, y)?;
        }
    }
    formatter.flush()?;
    let elapsed = start.elapsed();