kiddo 6.0.0-alpha.2

A high-performance, flexible, ergonomic k-d tree library. Ideal for geo- and astro- nearest-neighbour and k-nearest-neighbor queries
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

//! Autovectorization-friendly SIMD pruning implementations.

/// Autovec prune block for f64 values.
#[inline(always)]
#[allow(unused)]
pub(crate) fn simd_prune_block_f64(rd_values: &[f64; 8], max_dist: f64, sibling_mask: u8) -> u8 {
    let mut mask = 0u8;
    for i in 0..8 {
        if rd_values[i] <= max_dist {
            mask |= 1 << i;
        }
    }
    mask & sibling_mask
}

/// Autovec prune block for f32 values.
#[inline(always)]
#[allow(unused)]
pub(crate) fn simd_prune_block_f32(rd_values: &[f32; 8], max_dist: f32, sibling_mask: u8) -> u8 {
    let mut mask = 0u8;
    for i in 0..8 {
        if rd_values[i] <= max_dist {
            mask |= 1 << i;
        }
    }
    mask & sibling_mask
}