nalgebra/geometry/
unit_complex_simba.rs

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