rstared 0.14.0

Simple decorator that adds rstar::RTree to arbitrary collections.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

// SPDX-FileCopyrightText: 2026 rstared contributors
//
// SPDX-License-Identifier: MIT OR Apache-2.0

extern crate alloc;

use maplike::{
    containers::Container,
    ops::{Get, Push, Remove},
};
use rand::Rng;
use rstar::{AABB, primitives::Rectangle};
use rstared::RTreed;

#[test]
fn test_push_and_remove_random_aars_in_stable_vec() {
    use stable_vec::StableVec;
    test_push_and_remove_random_aars::<usize, StableVec<Rectangle<(i32, i32)>>>(StableVec::<
        Rectangle<(i32, i32)>,
    >::new());
}

#[test]
fn test_push_and_remove_random_aars_in_thunderdome() {
    use thunderdome::Arena;
    test_push_and_remove_random_aars::<thunderdome::Index, Arena<Rectangle<(i32, i32)>>>(Arena::<
        Rectangle<(i32, i32)>,
    >::new(
    ));
}

/// "AAR" stands for "axis-aligned rectangle".
fn test_push_and_remove_random_aars<
    K: Clone + PartialEq,
    C: Container<Key = K, Value = Rectangle<(i32, i32)>>
        + Get<K>
        + Push<K>
        + Remove<K, Output = Option<Rectangle<(i32, i32)>>>,
>(
    collection: C,
) {
    let mut rtreed: RTreed<C> = RTreed::new(collection);
    let mut rng = rand::rng();
    let mut keys = alloc::vec![];

    for _ in 0..100 {
        let x = rng.random_range(0..=90);
        let y = rng.random_range(0..=90);
        let width = rng.random_range(0..=10);
        let height = rng.random_range(0..=10);

        keys.push(rtreed.push(Rectangle::from_corners((x, y), (x + width, y + height))));
    }

    assert_eq!(
        rtreed
            .rtree()
            .locate_in_envelope(AABB::from_corners((0, 0), (100, 100)))
            .count(),
        100
    );

    for i in 0..10 {
        rtreed.remove(&keys[i]);
    }

    assert_eq!(
        rtreed
            .rtree()
            .locate_in_envelope(AABB::from_corners((0, 0), (100, 100)))
            .count(),
        90
    );
}