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
use crate::bounding_volume::SimdAABB;
use crate::math::{Point, Real, SimdReal, SIMD_WIDTH};
use crate::partitioning::{SimdVisitStatus, SimdVisitor};
use simba::simd::{SimdBool as _, SimdValue};
use std::marker::PhantomData;
pub struct PointIntersectionsVisitor<'a, T, F> {
simd_point: Point<SimdReal>,
callback: &'a mut F,
_phantom: PhantomData<T>,
}
impl<'a, T, F> PointIntersectionsVisitor<'a, T, F>
where
F: FnMut(&T) -> bool,
{
#[inline]
pub fn new(point: &'a Point<Real>, callback: &'a mut F) -> PointIntersectionsVisitor<'a, T, F> {
PointIntersectionsVisitor {
simd_point: Point::splat(*point),
callback,
_phantom: PhantomData,
}
}
}
impl<'a, T, F> SimdVisitor<T, SimdAABB> for PointIntersectionsVisitor<'a, T, F>
where
F: FnMut(&T) -> bool,
{
#[inline]
fn visit(&mut self, bv: &SimdAABB, b: Option<[Option<&T>; SIMD_WIDTH]>) -> SimdVisitStatus {
let mask = bv.contains_local_point(&self.simd_point);
if let Some(data) = b {
let bitmask = mask.bitmask();
for ii in 0..SIMD_WIDTH {
if (bitmask & (1 << ii)) != 0 && data[ii].is_some() {
if !(self.callback)(data[ii].unwrap()) {
return SimdVisitStatus::ExitEarly;
}
}
}
}
SimdVisitStatus::MaybeContinue(mask)
}
}