From d48a644d5c9f3247bb8b3edc5e1e673f0ab99b1d Mon Sep 17 00:00:00 2001 From: sebcrozet Date: Sun, 5 Apr 2020 16:35:26 +0200 Subject: [PATCH] Add missing implementatino of SimdValue for UnitQuaternion. --- src/geometry/quaternion_simba.rs | 45 +++++++++++++++++++++++++++++++- 1 file changed, 44 insertions(+), 1 deletion(-) diff --git a/src/geometry/quaternion_simba.rs b/src/geometry/quaternion_simba.rs index 5bd3e2b5..0120defe 100755 --- a/src/geometry/quaternion_simba.rs +++ b/src/geometry/quaternion_simba.rs @@ -1,7 +1,7 @@ use simba::simd::SimdValue; use crate::base::Vector4; -use crate::geometry::Quaternion; +use crate::geometry::{Quaternion, UnitQuaternion}; use crate::{RealField, Scalar}; impl SimdValue for Quaternion @@ -45,3 +45,46 @@ where N::Element: Scalar self.coords.select(cond, other.coords).into() } } + +impl SimdValue for UnitQuaternion +where N::Element: Scalar +{ + type Element = UnitQuaternion; + type SimdBool = N::SimdBool; + + #[inline] + fn lanes() -> usize { + N::lanes() + } + + #[inline] + fn splat(val: Self::Element) -> Self { + UnitQuaternion::new_unchecked(Quaternion::splat(val.into_inner())) + } + + #[inline] + fn extract(&self, i: usize) -> Self::Element { + UnitQuaternion::new_unchecked(self.as_ref().extract(i)) + } + + #[inline] + unsafe fn extract_unchecked(&self, i: usize) -> Self::Element { + UnitQuaternion::new_unchecked(self.as_ref().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 { + UnitQuaternion::new_unchecked(self.into_inner().select(cond, other.into_inner())) + } +}