molframe-spatial 0.6.3

Deterministic spatial indices and neighbour search for molframe
Documentation 
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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use super::*;

#[test]
fn nearest_neighbours_are_sorted_by_distance_then_atom() {
    let positions = [
        [0.0, 0.0, 0.0],
        [1.0, 0.0, 0.0],
        [-1.0, 0.0, 0.0],
        [3.0, 0.0, 0.0],
    ];
    let tree = match KdTree::build(&positions, &[0, 1, 2, 3], None) {
        Ok(tree) => tree,
        Err(error) => panic!("build failed: {error}"),
    };
    let nearest = match tree.k_nearest(0, 2) {
        Ok(nearest) => nearest,
        Err(error) => panic!("query failed: {error}"),
    };
    assert_eq!(
        nearest.iter().map(|item| item.0).collect::<Vec<_>>(),
        vec![1, 2]
    );
}

#[test]
fn radius_search_excludes_self_and_deduplicates_overlapping_groups() {
    let positions = [[0.0, 0.0, 0.0], [0.5, 0.0, 0.0]];
    let tree = match KdTree::build(&positions, &[0, 1], None) {
        Ok(tree) => tree,
        Err(error) => panic!("build failed: {error}"),
    };
    let pairs = match tree.pairs(&[0, 1], 1.0) {
        Ok(pairs) => pairs,
        Err(error) => panic!("query failed: {error}"),
    };
    assert_eq!(pairs.len(), 1);
    assert_eq!((pairs[0].first, pairs[0].second), (0, 1));
}

#[test]
fn periodic_nearest_search_uses_tree_images_across_a_triclinic_boundary() {
    let positions = [[0.1, 0.2, 0.3], [7.9, 0.2, 0.3], [4.0, 4.0, 4.0]];
    let periodic = PeriodicBox::from_cell(molframe_core::structure::UnitCell {
        lengths: [8.0, 9.0, 10.0],
        angles: [70.0, 80.0, 65.0],
    });
    let periodic = match periodic {
        Ok(periodic) => periodic,
        Err(error) => panic!("cell failed: {error}"),
    };
    let tree = match KdTree::build(&positions, &[0, 1, 2], Some(periodic)) {
        Ok(tree) => tree,
        Err(error) => panic!("build failed: {error}"),
    };
    let nearest = match tree.k_nearest(0, 2) {
        Ok(nearest) => nearest,
        Err(error) => panic!("query failed: {error}"),
    };

    assert_eq!(nearest[0].0, 1);
    let expected = periodic.distance_squared(positions[0], positions[1]);
    assert!((nearest[0].1 - expected).abs() <= f32::EPSILON);
}

#[test]
fn periodic_image_budget_is_enforced_instead_of_falling_back() {
    let positions = [[0.0, 0.0, 0.0], [0.5, 0.0, 0.0]];
    let periodic = PeriodicBox::from_cell(molframe_core::structure::UnitCell {
        lengths: [8.0; 3],
        angles: [90.0; 3],
    });
    let periodic = match periodic {
        Ok(periodic) => periodic,
        Err(error) => panic!("cell failed: {error}"),
    };
    let tree = KdTree::build_with_options(
        &positions,
        &[0, 1],
        Some(periodic),
        KdPeriodicOptions {
            maximum_image_count: 1,
        },
    );
    let tree = match tree {
        Ok(tree) => tree,
        Err(error) => panic!("build failed: {error}"),
    };

    assert!(matches!(
        tree.pairs(&[0], 1.0),
        Err(SpatialError::PeriodicImageLimitExceeded { .. })
    ));
}