52 lines
1.3 KiB
Rust
Executable File
52 lines
1.3 KiB
Rust
Executable File
use num_complex::Complex;
|
|
use simba::simd::SimdValue;
|
|
use std::ops::Deref;
|
|
|
|
use crate::base::Unit;
|
|
use crate::geometry::UnitComplex;
|
|
use crate::SimdRealField;
|
|
|
|
impl<N: SimdRealField> SimdValue for UnitComplex<N>
|
|
where
|
|
N::Element: SimdRealField,
|
|
{
|
|
type Element = UnitComplex<N::Element>;
|
|
type SimdBool = N::SimdBool;
|
|
|
|
#[inline]
|
|
fn lanes() -> usize {
|
|
N::lanes()
|
|
}
|
|
|
|
#[inline]
|
|
fn splat(val: Self::Element) -> Self {
|
|
Unit::new_unchecked(Complex::splat(val.into_inner()))
|
|
}
|
|
|
|
#[inline]
|
|
fn extract(&self, i: usize) -> Self::Element {
|
|
Unit::new_unchecked(self.deref().extract(i))
|
|
}
|
|
|
|
#[inline]
|
|
unsafe fn extract_unchecked(&self, i: usize) -> Self::Element {
|
|
Unit::new_unchecked(self.deref().extract_unchecked(i))
|
|
}
|
|
|
|
#[inline]
|
|
fn replace(&mut self, i: usize, val: Self::Element) {
|
|
self.as_mut_unchecked().replace(i, val.into_inner())
|
|
}
|
|
|
|
#[inline]
|
|
unsafe fn replace_unchecked(&mut self, i: usize, val: Self::Element) {
|
|
self.as_mut_unchecked()
|
|
.replace_unchecked(i, val.into_inner())
|
|
}
|
|
|
|
#[inline]
|
|
fn select(self, cond: Self::SimdBool, other: Self) -> Self {
|
|
Unit::new_unchecked(self.into_inner().select(cond, other.into_inner()))
|
|
}
|
|
}
|