aboutsummaryrefslogtreecommitdiff
path: root/src/geometry/proximity_detector/proximity_detector.rs
blob: 7c8ad204ecea4f7cac1aa23ac50b144c67c5586a (plain)
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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
use crate::geometry::{
    Collider, ColliderSet, Proximity, ProximityDispatcher, ProximityEvent, ProximityPair, Shape,
};
use crate::math::Isometry;
#[cfg(feature = "simd-is-enabled")]
use crate::math::{SimdFloat, SIMD_WIDTH};
use crate::pipeline::EventHandler;
use std::any::Any;

#[derive(Copy, Clone)]
pub enum ProximityPhase {
    NearPhase(ProximityDetector),
    ExactPhase(PrimitiveProximityDetector),
}

impl ProximityPhase {
    #[inline]
    pub fn detect_proximity(
        self,
        mut context: ProximityDetectionContext,
        events: &dyn EventHandler,
    ) {
        let proximity = match self {
            Self::NearPhase(gen) => (gen.detect_proximity)(&mut context),
            Self::ExactPhase(gen) => {
                // Build the primitive context from the non-primitive context and dispatch.
                let collider1 = &context.colliders[context.pair.pair.collider1];
                let collider2 = &context.colliders[context.pair.pair.collider2];

                let mut context2 = PrimitiveProximityDetectionContext {
                    prediction_distance: context.prediction_distance,
                    collider1,
                    collider2,
                    shape1: collider1.shape(),
                    shape2: collider2.shape(),
                    position1: collider1.position(),
                    position2: collider2.position(),
                    workspace: context.pair.detector_workspace.as_mut().map(|w| &mut **w),
                };

                (gen.detect_proximity)(&mut context2)
            }
        };

        if context.pair.proximity != proximity {
            events.handle_proximity_event(ProximityEvent::new(
                context.pair.pair.collider1,
                context.pair.pair.collider2,
                context.pair.proximity,
                proximity,
            ))
        }

        context.pair.proximity = proximity;
    }

    #[cfg(feature = "simd-is-enabled")]
    #[inline]
    pub fn detect_proximity_simd(
        self,
        mut context: ProximityDetectionContextSimd,
        events: &dyn EventHandler,
    ) {
        let proximities = match self {
            Self::NearPhase(gen) => (gen.detect_proximity_simd)(&mut context),
            Self::ExactPhase(gen) => {
                // Build the primitive context from the non-primitive context and dispatch.
                use arrayvec::ArrayVec;
                let mut colliders_arr: ArrayVec<[(&Collider, &Collider); SIMD_WIDTH]> =
                    ArrayVec::new();
                let mut workspace_arr: ArrayVec<
                    [Option<&mut (dyn Any + Send + Sync)>; SIMD_WIDTH],
                > = ArrayVec::new();

                for pair in context.pairs.iter_mut() {
                    let collider1 = &context.colliders[pair.pair.collider1];
                    let collider2 = &context.colliders[pair.pair.collider2];
                    colliders_arr.push((collider1, collider2));
                    workspace_arr.push(pair.detector_workspace.as_mut().map(|w| &mut **w));
                }

                let max_index = colliders_arr.len() - 1;
                let colliders1 = array![|ii| colliders_arr[ii.min(max_index)].0; SIMD_WIDTH];
                let colliders2 = array![|ii| colliders_arr[ii.min(max_index)].1; SIMD_WIDTH];

                let mut context2 = PrimitiveProximityDetectionContextSimd {
                    prediction_distance: context.prediction_distance,
                    colliders1,
                    colliders2,
                    shapes1: array![|ii| colliders1[ii].shape(); SIMD_WIDTH],
                    shapes2: array![|ii| colliders2[ii].shape(); SIMD_WIDTH],
                    positions1: &Isometry::from(
                        array![|ii| *colliders1[ii].position(); SIMD_WIDTH],
                    ),
                    positions2: &Isometry::from(
                        array![|ii| *colliders2[ii].position(); SIMD_WIDTH],
                    ),
                    workspaces: workspace_arr.as_mut_slice(),
                };

                (gen.detect_proximity_simd)(&mut context2)
            }
        };

        for (i, pair) in context.pairs.iter_mut().enumerate() {
            if pair.proximity != proximities[i] {
                events.handle_proximity_event(ProximityEvent::new(
                    pair.pair.collider1,
                    pair.pair.collider2,
                    pair.proximity,
                    proximities[i],
                ))
            }
            pair.proximity = proximities[i];
        }
    }
}

pub struct PrimitiveProximityDetectionContext<'a> {
    pub prediction_distance: f32,
    pub collider1: &'a Collider,
    pub collider2: &'a Collider,
    pub shape1: &'a dyn Shape,
    pub shape2: &'a dyn Shape,
    pub position1: &'a Isometry<f32>,
    pub position2: &'a Isometry<f32>,
    pub workspace: Option<&'a mut (dyn Any + Send + Sync)>,
}

#[cfg(feature = "simd-is-enabled")]
pub struct PrimitiveProximityDetectionContextSimd<'a, 'b> {
    pub prediction_distance: f32,
    pub colliders1: [&'a Collider; SIMD_WIDTH],
    pub colliders2: [&'a Collider; SIMD_WIDTH],
    pub shapes1: [&'a dyn Shape; SIMD_WIDTH],
    pub shapes2: [&'a dyn Shape; SIMD_WIDTH],
    pub positions1: &'a Isometry<SimdFloat>,
    pub positions2: &'a Isometry<SimdFloat>,
    pub workspaces: &'a mut [Option<&'b mut (dyn Any + Send + Sync)>],
}

#[derive(Copy, Clone)]
pub struct PrimitiveProximityDetector {
    pub detect_proximity: fn(&mut PrimitiveProximityDetectionContext) -> Proximity,
    #[cfg(feature = "simd-is-enabled")]
    pub detect_proximity_simd:
        fn(&mut PrimitiveProximityDetectionContextSimd) -> [Proximity; SIMD_WIDTH],
}

impl PrimitiveProximityDetector {
    fn unimplemented_fn(_ctxt: &mut PrimitiveProximityDetectionContext) -> Proximity {
        Proximity::Disjoint
    }
    #[cfg(feature = "simd-is-enabled")]
    fn unimplemented_simd_fn(
        _ctxt: &mut PrimitiveProximityDetectionContextSimd,
    ) -> [Proximity; SIMD_WIDTH] {
        [Proximity::Disjoint; SIMD_WIDTH]
    }
}

impl Default for PrimitiveProximityDetector {
    fn default() -> Self {
        Self {
            detect_proximity: Self::unimplemented_fn,
            #[cfg(feature = "simd-is-enabled")]
            detect_proximity_simd: Self::unimplemented_simd_fn,
        }
    }
}

pub struct ProximityDetectionContext<'a> {
    pub dispatcher: &'a dyn ProximityDispatcher,
    pub prediction_distance: f32,
    pub colliders: &'a ColliderSet,
    pub pair: &'a mut ProximityPair,
}

#[cfg(feature = "simd-is-enabled")]
pub struct ProximityDetectionContextSimd<'a, 'b> {
    pub dispatcher: &'a dyn ProximityDispatcher,
    pub prediction_distance: f32,
    pub colliders: &'a ColliderSet,
    pub pairs: &'a mut [&'b mut ProximityPair],
}

#[derive(Copy, Clone)]
pub struct ProximityDetector {
    pub detect_proximity: fn(&mut ProximityDetectionContext) -> Proximity,
    #[cfg(feature = "simd-is-enabled")]
    pub detect_proximity_simd: fn(&mut ProximityDetectionContextSimd) -> [Proximity; SIMD_WIDTH],
}

impl ProximityDetector {
    fn unimplemented_fn(_ctxt: &mut ProximityDetectionContext) -> Proximity {
        Proximity::Disjoint
    }
    #[cfg(feature = "simd-is-enabled")]
    fn unimplemented_simd_fn(_ctxt: &mut ProximityDetectionContextSimd) -> [Proximity; SIMD_WIDTH] {
        [Proximity::Disjoint; SIMD_WIDTH]
    }
}

impl Default for ProximityDetector {
    fn default() -> Self {
        Self {
            detect_proximity: Self::unimplemented_fn,
            #[cfg(feature = "simd-is-enabled")]
            detect_proximity_simd: Self::unimplemented_simd_fn,
        }
    }
}