nalgebra/geometry/
unit_complex_simba.rs

1use num_complex::Complex;
2use simba::simd::SimdValue;
3use std::ops::Deref;
4
5use crate::base::Unit;
6use crate::geometry::UnitComplex;
7use crate::SimdRealField;
8
9impl<T: SimdRealField> SimdValue for UnitComplex<T>
10where
11    T::Element: SimdRealField,
12{
13    type Element = UnitComplex<T::Element>;
14    type SimdBool = T::SimdBool;
15
16    #[inline]
17    fn lanes() -> usize {
18        T::lanes()
19    }
20
21    #[inline]
22    fn splat(val: Self::Element) -> Self {
23        Unit::new_unchecked(Complex::splat(val.into_inner()))
24    }
25
26    #[inline]
27    fn extract(&self, i: usize) -> Self::Element {
28        Unit::new_unchecked(self.deref().extract(i))
29    }
30
31    #[inline]
32    unsafe fn extract_unchecked(&self, i: usize) -> Self::Element {
33        Unit::new_unchecked(self.deref().extract_unchecked(i))
34    }
35
36    #[inline]
37    fn replace(&mut self, i: usize, val: Self::Element) {
38        self.as_mut_unchecked().replace(i, val.into_inner())
39    }
40
41    #[inline]
42    unsafe fn replace_unchecked(&mut self, i: usize, val: Self::Element) {
43        self.as_mut_unchecked()
44            .replace_unchecked(i, val.into_inner())
45    }
46
47    #[inline]
48    fn select(self, cond: Self::SimdBool, other: Self) -> Self {
49        Unit::new_unchecked(self.into_inner().select(cond, other.into_inner()))
50    }
51}