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    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        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        self.as_mut_unchecked()
40            .replace_unchecked(i, val.into_inner())
41    }
42
43    #[inline]
44    fn select(self, cond: Self::SimdBool, other: Self) -> Self {
45        Unit::new_unchecked(self.into_inner().select(cond, other.into_inner()))
46    }
47}