124 lines
3.1 KiB
Rust
124 lines
3.1 KiB
Rust
use spacedog::*;
|
|
|
|
use rand::{Rng, SeedableRng};
|
|
use rand_hc::Hc128Rng;
|
|
|
|
use criterion::Criterion;
|
|
use criterion::{criterion_group, criterion_main};
|
|
|
|
const SEED_1: &[u8; 32] = b"Gv0aHMtHkBGsUXNspGU9fLRuCWkZWHZx";
|
|
|
|
const DEFAULT_NUM_POINTS: usize = 100_000;
|
|
|
|
fn new_index(c: &mut Criterion) {
|
|
c.bench_function("new index", move |b| {
|
|
let points: Vec<_> = create_random_points(DEFAULT_NUM_POINTS, SEED_1);
|
|
|
|
b.iter(|| {
|
|
Spindex2d::new(&points);
|
|
});
|
|
});
|
|
}
|
|
|
|
fn update_positions(c: &mut Criterion) {
|
|
c.bench_function("update positions", move |b| {
|
|
let points: Vec<_> = create_random_points(DEFAULT_NUM_POINTS, SEED_1);
|
|
|
|
let mut rng = Hc128Rng::from_seed(*SEED_1);
|
|
|
|
let jitters: Vec<_> = points
|
|
.iter()
|
|
.map(|p| {
|
|
let x = p.x + rng.random_range(-0.5..=0.5);
|
|
let y = p.y + rng.random_range(-0.5..=0.5);
|
|
Point2d::new(x, y)
|
|
})
|
|
.collect();
|
|
|
|
let index = Spindex2d::new(&points);
|
|
let mut round = 0;
|
|
b.iter(|| {
|
|
if round % 2 == 0 {
|
|
index.update(&jitters);
|
|
} else {
|
|
index.update(&points);
|
|
}
|
|
round += 1;
|
|
});
|
|
});
|
|
}
|
|
|
|
fn nearest_default_horizon(c: &mut Criterion) {
|
|
c.bench_function("nearest default", move |b| {
|
|
let points: Vec<_> = create_random_points(DEFAULT_NUM_POINTS, SEED_1);
|
|
|
|
let index = Spindex2d::new(&points);
|
|
|
|
let mut i = 0;
|
|
b.iter(|| {
|
|
i += 1;
|
|
index.nearest(&points[i - 1]).is_some()
|
|
});
|
|
});
|
|
}
|
|
|
|
fn nearest_short_horizon(c: &mut Criterion) {
|
|
c.bench_function("nearest short", move |b| {
|
|
let points: Vec<_> = create_random_points(DEFAULT_NUM_POINTS, SEED_1);
|
|
|
|
let index = Spindex2d::new(&points).with_horizon(20);
|
|
let len = points.len();
|
|
|
|
let mut i = 0;
|
|
b.iter(|| {
|
|
i = (i + 1) % len;
|
|
index.nearest(&points[i]).is_some()
|
|
});
|
|
});
|
|
}
|
|
|
|
fn aabb_search(c: &mut Criterion) {
|
|
c.bench_function("aabb", move |b| {
|
|
let points: Vec<_> = create_random_points(DEFAULT_NUM_POINTS, SEED_1);
|
|
|
|
let index = Spindex2d::new(&points);
|
|
|
|
let mut i = 0;
|
|
b.iter(|| {
|
|
i += 1;
|
|
!index.aabb(&points[i - 1], 0.05).is_empty()
|
|
});
|
|
});
|
|
}
|
|
|
|
fn radius_search(c: &mut Criterion) {
|
|
c.bench_function("radius 0.1", move |b| {
|
|
let points: Vec<_> = create_random_points(DEFAULT_NUM_POINTS, SEED_1);
|
|
|
|
let index = Spindex2d::new(&points);
|
|
|
|
let mut i = 0;
|
|
b.iter(|| {
|
|
i += 1;
|
|
!index.radius(&points[i - 1], 0.1).is_empty()
|
|
});
|
|
});
|
|
}
|
|
|
|
criterion_group!(
|
|
benches,
|
|
new_index,
|
|
update_positions,
|
|
nearest_default_horizon,
|
|
nearest_short_horizon,
|
|
aabb_search,
|
|
radius_search
|
|
);
|
|
criterion_main!(benches);
|
|
|
|
fn create_random_points(num_points: usize, seed: &[u8; 32]) -> Vec<Point2d> {
|
|
let mut rng = Hc128Rng::from_seed(*seed);
|
|
(0..num_points)
|
|
.map(|_| rng.random::<[f64; 2]>().into())
|
|
.collect()
|
|
}
|