lambert_izzo 2.0.0

//! Batch throughput: sequential `lambert` over a slice vs (with `--features rayon`)
//! parallel `lambert_par`.
//!
//! Run sequential: `cargo bench --bench batch`.
//! Run parallel:   `cargo bench --bench batch --features rayon`.

use criterion::{Criterion, criterion_group, criterion_main};
use lambert_izzo::{RevolutionBudget, lambert};
use lambert_izzo_test_support::bodies::MU_EARTH;
use lambert_izzo_test_support::elements_u64;
use lambert_izzo_test_support::random_inputs::{Spec, WayStrategy, generate};
use std::hint::black_box;

fn spec() -> Spec {
    Spec {
        n: 10_000,
        seed: 0xCAFE_F00D,
        radius_range: (3500.0, 28_000.0),
        tof_range: (100.0, 50_000.0),
        mu: MU_EARTH,
        way: WayStrategy::Short,
        revolutions: RevolutionBudget::try_up_to(3).expect("3 is within BoundedRevs::MAX"),
    }
}

fn batch_sequential(c: &mut Criterion) {
    let inputs = generate(&spec());

    let mut group = c.benchmark_group("batch_sequential");
    group.throughput(criterion::Throughput::Elements(elements_u64(inputs.len())));
    group.sample_size(20);
    group.bench_function("lambert_seq_x10000", |b| {
        b.iter(|| {
            for input in &inputs {
                black_box(lambert(input).ok());
            }
        });
    });
    group.finish();
}

#[cfg(feature = "rayon")]
fn batch_parallel(c: &mut Criterion) {
    use lambert_izzo::lambert_par;
    use rayon::iter::ParallelIterator;

    let inputs = generate(&spec());

    let mut group = c.benchmark_group("batch_parallel");
    group.throughput(criterion::Throughput::Elements(elements_u64(inputs.len())));
    group.sample_size(20);
    group.bench_function("lambert_par_x10000", |b| {
        b.iter(|| {
            lambert_par(&inputs).for_each(|sol| {
                black_box(sol.ok());
            });
        });
    });
    group.finish();
}

#[cfg(feature = "rayon")]
criterion_group!(benches, batch_sequential, batch_parallel);
#[cfg(not(feature = "rayon"))]
criterion_group!(benches, batch_sequential);

criterion_main!(benches);