use crate::f32::ffi::{UnitVec3A, Vec4};
use crate::{BVec3A, Vec2, Vec3};
#[cfg(not(target_arch = "spirv"))]
use core::fmt;
use core::iter::{Product, Sum};
use core::ops::*;
use crate::nums::*;
use auto_ops_det::impl_op_ex;
use core::ops;
union UnionCastFfi {
simd: crate::f32::simd_alias::Vec3A,
ffi: Vec3A,
}
impl From<crate::f32::simd_alias::Vec3A> for Vec3A {
#[inline]
fn from(simd: crate::f32::simd_alias::Vec3A) -> Self {
unsafe { UnionCastFfi { simd }.ffi }
}
}
impl From<Vec3A> for crate::f32::simd_alias::Vec3A {
#[inline]
fn from(ffi: Vec3A) -> Self {
unsafe { UnionCastFfi { ffi }.simd }
}
}
impl From<super::C128> for crate::f32::simd_alias::Vec3A {
#[inline]
fn from(t: super::C128) -> Self {
Vec3A(t).into()
}
}
#[inline]
pub const fn vec3a(x: f32, y: f32, z: f32) -> Vec3A {
Vec3A::new(x, y, z)
}
#[derive(Clone, Copy)]
#[repr(transparent)]
pub struct Vec3A(pub(crate) super::C128);
impl Vec3A {
pub const ZERO: Self = Self::splat(0.0_f32);
pub const ONE: Self = Self::splat(1.0_f32);
pub const NEG_ONE: Self = Self::splat(-1.0_f32);
pub const NAN: Self = Self::splat(f32::NAN);
pub const X: Self = Self::new(1.0_f32, 0.0_f32, 0.0_f32);
pub const Y: Self = Self::new(0.0_f32, 1.0_f32, 0.0_f32);
pub const Z: Self = Self::new(0.0_f32, 0.0_f32, 1.0_f32);
pub const NEG_X: Self = Self::new(-1.0_f32, 0.0_f32, 0.0_f32);
pub const NEG_Y: Self = Self::new(0.0_f32, -1.0_f32, 0.0_f32);
pub const NEG_Z: Self = Self::new(0.0_f32, 0.0_f32, -1.0_f32);
pub const AXES: [Self; 3] = [Self::X, Self::Y, Self::Z];
#[inline]
pub const fn new(x: f32, y: f32, z: f32) -> Self {
Self(super::C128(x, y, z, z))
}
#[inline]
pub const fn splat(v: f32) -> Self {
Self(super::C128(v, v, v, v))
}
#[inline]
pub fn select(mask: BVec3A, if_true: Self, if_false: Self) -> Self {
crate::f32::simd_alias::Vec3A::select(mask.into(), if_true.into(), if_false.into()).into()
}
#[inline]
pub const fn from_array(a: [f32; 3]) -> Self {
Self::new(a[0], a[1], a[2])
}
#[inline]
pub const fn to_array(&self) -> [f32; 3] {
unsafe { *(self as *const Vec3A as *const [f32; 3]) }
}
#[inline]
pub const fn from_slice(slice: &[f32]) -> Self {
Self::new(slice[0], slice[1], slice[2])
}
#[inline]
pub fn write_to_slice(self, slice: &mut [f32]) {
slice[0] = self.x;
slice[1] = self.y;
slice[2] = self.z;
}
#[allow(dead_code)]
#[inline]
pub(crate) fn from_vec4(v: Vec4) -> Self {
Self(v.0)
}
#[inline]
pub fn extend(self, w: f32) -> Vec4 {
Vec4::new(self.x, self.y, self.z, w)
}
#[inline]
pub fn truncate(self) -> Vec2 {
use crate::swizzles::Vec3Swizzles;
self.xy()
}
#[inline]
pub(crate) fn dot(self, rhs: Self) -> f32 {
crate::f32::simd_alias::Vec3A::dot(self.into(), rhs.into())
}
#[inline]
pub(crate) fn cross(self, rhs: Self) -> Self {
crate::f32::simd_alias::Vec3A::cross(self.into(), rhs.into()).into()
}
#[inline]
pub fn min(self, rhs: Self) -> Self {
crate::f32::simd_alias::Vec3A::min(self.into(), rhs.into()).into()
}
#[inline]
pub fn max(self, rhs: Self) -> Self {
crate::f32::simd_alias::Vec3A::max(self.into(), rhs.into()).into()
}
#[inline]
pub fn clamp(self, min: Self, max: Self) -> Self {
glam_assert!(min.cmple(max).all(), "clamp: expected min <= max");
self.max(min).min(max)
}
#[inline]
pub fn min_element(self) -> f32 {
crate::f32::simd_alias::Vec3A::min_element(self.into())
}
#[inline]
pub fn max_element(self) -> f32 {
crate::f32::simd_alias::Vec3A::max_element(self.into())
}
#[inline]
pub fn cmpeq(self, rhs: Self) -> BVec3A {
crate::f32::simd_alias::Vec3A::cmpeq(self.into(), rhs.into()).into()
}
#[inline]
pub fn cmpne(self, rhs: Self) -> BVec3A {
crate::f32::simd_alias::Vec3A::cmpne(self.into(), rhs.into()).into()
}
#[inline]
pub fn cmpge(self, rhs: Self) -> BVec3A {
crate::f32::simd_alias::Vec3A::cmpge(self.into(), rhs.into()).into()
}
#[inline]
pub fn cmpgt(self, rhs: Self) -> BVec3A {
crate::f32::simd_alias::Vec3A::cmpgt(self.into(), rhs.into()).into()
}
#[inline]
pub fn cmple(self, rhs: Self) -> BVec3A {
crate::f32::simd_alias::Vec3A::cmple(self.into(), rhs.into()).into()
}
#[inline]
pub fn cmplt(self, rhs: Self) -> BVec3A {
crate::f32::simd_alias::Vec3A::cmplt(self.into(), rhs.into()).into()
}
#[inline]
pub fn abs(self) -> Self {
crate::f32::simd_alias::Vec3A::abs(self.into()).into()
}
#[inline]
pub fn signum(self) -> Self {
crate::f32::simd_alias::Vec3A::signum(self.into()).into()
}
#[inline]
pub fn is_finite(self) -> bool {
self.x.is_finite() && self.y.is_finite() && self.z.is_finite()
}
#[inline]
pub fn is_nan(self) -> bool {
self.is_nan_mask().any()
}
#[inline]
pub fn is_nan_mask(self) -> BVec3A {
crate::f32::simd_alias::Vec3A::is_nan_mask(self.into()).into()
}
#[inline]
pub fn round(self) -> Self {
crate::f32::simd_alias::Vec3A::round(self.into()).into()
}
#[inline]
pub fn floor(self) -> Self {
crate::f32::simd_alias::Vec3A::floor(self.into()).into()
}
#[inline]
pub fn ceil(self) -> Self {
crate::f32::simd_alias::Vec3A::ceil(self.into()).into()
}
#[inline]
pub fn fract(self) -> Self {
self - self.floor()
}
#[inline]
pub fn exp(self) -> Self {
Self::new(self.x.expf(), self.y.expf(), self.z.expf())
}
#[inline]
pub fn powf(self, n: f32) -> Self {
Self::new(self.x.powff(n), self.y.powff(n), self.z.powff(n))
}
#[inline]
pub fn recip(self) -> Self {
crate::f32::simd_alias::Vec3A::recip(self.into()).into()
}
#[doc(alias = "magnitude")]
#[inline]
pub fn length(self) -> f32 {
self.dot(self).sqrtf()
}
#[doc(alias = "magnitude2")]
#[inline]
pub fn length_squared(self) -> f32 {
self.dot(self)
}
#[inline]
pub fn length_recip(self) -> f32 {
self.length().recip()
}
#[inline]
pub fn distance(self, rhs: Self) -> f32 {
(self - rhs).length()
}
#[inline]
pub fn distance_squared(self, rhs: Self) -> f32 {
(self - rhs).length_squared()
}
#[must_use]
#[inline]
pub fn normalize(self) -> Self {
crate::f32::simd_alias::Vec3A::normalize(self.into()).into()
}
#[must_use]
#[inline]
pub fn normalize_and_length(self) -> (Self, f32) {
let (v, len) = crate::f32::simd_alias::Vec3A::normalize_and_length(self.into());
(v.into(), len)
}
#[must_use]
#[inline]
pub fn normalize_to_unit(self) -> UnitVec3A {
self.normalize().as_unit_vec3a_unchecked()
}
#[must_use]
#[inline]
pub fn normalize_to_unit_and_length(self) -> (UnitVec3A, f32) {
let res = self.normalize_and_length();
(res.0.as_unit_vec3a_unchecked(), res.1)
}
#[must_use]
#[inline]
pub fn try_normalize(self, min_len: f32) -> Option<Self> {
let length = self.length();
if length.is_finite() && length > min_len {
Some(self * length.recip())
} else {
None
}
}
#[must_use]
#[inline]
pub fn try_normalize_to_unit(self, min_len: f32) -> Option<UnitVec3A> {
let length = self.length();
if length.is_finite() && length > min_len {
Some((self * length.recip()).as_unit_vec3a_unchecked())
} else {
None
}
}
#[must_use]
#[inline]
pub fn try_normalize_to_unit_and_length(self, min_len: f32) -> Option<(UnitVec3A, f32)> {
let length = self.length();
if length.is_finite() && length > min_len {
Some(((self * length.recip()).as_unit_vec3a_unchecked(), length))
} else {
None
}
}
#[must_use]
#[inline]
pub fn normalize_or_zero(self) -> Self {
let rcp = self.length_recip();
if rcp.is_finite() && rcp > 0.0 {
self * rcp
} else {
Self::ZERO
}
}
#[inline]
pub fn is_normalized(self) -> bool {
(self.length_squared() - 1.0_f32).absf() <= 1e-4
}
#[must_use]
#[inline]
pub fn project_onto(self, rhs: Self) -> Self {
let other_len_sq_rcp = rhs.dot(rhs).recip();
glam_assert!(
other_len_sq_rcp.is_finite(),
"Trying to project onto infinite rhs = {:?}",
rhs
);
rhs * self.dot(rhs) * other_len_sq_rcp
}
#[must_use]
#[inline]
pub fn reject_from(self, rhs: Self) -> Self {
self - self.project_onto(rhs)
}
#[must_use]
#[inline]
pub fn project_onto_normalized(self, rhs: UnitVec3A) -> Self {
rhs.as_vec3a() * self.dot(rhs.as_vec3a())
}
#[must_use]
#[inline]
pub fn reject_from_normalized(self, rhs: UnitVec3A) -> Self {
self - self.project_onto_normalized(rhs)
}
#[doc(alias = "mix")]
#[inline]
pub fn lerp(self, rhs: Self, s: f32) -> Self {
self + ((rhs - self) * s)
}
#[inline]
pub fn abs_diff_eq(self, rhs: Self, max_abs_diff: f32) -> bool {
self.sub(rhs).abs().cmple(Self::splat(max_abs_diff)).all()
}
#[inline]
pub fn clamp_length(self, min: f32, max: f32) -> Self {
glam_assert!(min <= max);
let length_sq = self.length_squared();
if length_sq < min * min {
self * (length_sq.sqrtf().recip() * min)
} else if length_sq > max * max {
self * (length_sq.sqrtf().recip() * max)
} else {
self
}
}
#[inline]
pub fn clamp_length_max(self, max: f32) -> Self {
let length_sq = self.length_squared();
if length_sq > max * max {
self * (length_sq.sqrtf().recip() * max)
} else {
self
}
}
#[inline]
pub fn clamp_length_min(self, min: f32) -> Self {
let length_sq = self.length_squared();
if length_sq < min * min {
self * (length_sq.sqrtf().recip() * min)
} else {
self
}
}
#[inline]
pub fn mul_add(self, a: Self, b: Self) -> Self {
#[cfg(target_feature = "fma")]
{
Self::from(crate::f32::simd_alias::Vec3A::mul_add(
self.into(),
a.into(),
b.into(),
))
}
#[cfg(not(target_feature = "fma"))]
Self::new(
self.x.mul_addf(a.x, b.x),
self.y.mul_addf(a.y, b.y),
self.z.mul_addf(a.z, b.z),
)
}
#[inline]
pub fn angle_between(self, rhs: Self) -> f32 {
use crate::FloatEx;
self.dot(rhs)
.div(self.length_squared().mul(rhs.length_squared()).sqrtf())
.acos_approx()
}
#[inline]
pub fn any_orthogonal_vector(&self) -> Self {
if self.x.absf() > self.y.absf() {
Self::new(-self.z, 0.0, self.x) } else {
Self::new(0.0, self.z, -self.y) }
}
#[inline]
pub fn as_dvec3(&self) -> crate::DVec3 {
crate::DVec3::new(self.x as f64, self.y as f64, self.z as f64)
}
#[inline]
pub fn as_ivec3(&self) -> crate::IVec3 {
crate::IVec3::new(self.x as i32, self.y as i32, self.z as i32)
}
#[inline]
pub fn as_uvec3(&self) -> crate::UVec3 {
crate::UVec3::new(self.x as u32, self.y as u32, self.z as u32)
}
}
impl Default for Vec3A {
#[inline]
fn default() -> Self {
Self::ZERO
}
}
impl PartialEq for Vec3A {
#[inline]
fn eq(&self, rhs: &Self) -> bool {
self.cmpeq(*rhs).all()
}
}
impl_op_ex!(/ |a: &Vec3A, b: &Vec3A| -> Vec3A {
(crate::f32::simd_alias::Vec3A::from(a.0) / crate::f32::simd_alias::Vec3A::from(b.0)).into()
});
impl_op_ex!(/= |a: &mut Vec3A, b: &Vec3A| {
*a = Vec3A::from(crate::f32::simd_alias::Vec3A::from(a.0) / crate::f32::simd_alias::Vec3A::from(b.0))
});
impl_op_ex!(/ |a: &Vec3A, b: &f32| -> Vec3A {
(crate::f32::simd_alias::Vec3A::from(a.0) / b).into()
});
impl_op_ex!(/= |a: &mut Vec3A, b: &f32| {
*a = Vec3A::from(crate::f32::simd_alias::Vec3A::from(a.0) / b)
});
impl_op_ex!(/ |a: &f32, b: &Vec3A| -> Vec3A {
(a / crate::f32::simd_alias::Vec3A::from(b.0)).into()
});
impl_op_ex!(*|a: &Vec3A, b: &Vec3A| -> Vec3A {
(crate::f32::simd_alias::Vec3A::from(a.0) * crate::f32::simd_alias::Vec3A::from(b.0)).into()
});
impl_op_ex!(*= |a: &mut Vec3A, b: &Vec3A| {
*a = Vec3A::from(crate::f32::simd_alias::Vec3A::from(a.0) * crate::f32::simd_alias::Vec3A::from(b.0))
});
impl_op_ex!(*|a: &Vec3A, b: &f32| -> Vec3A {
(crate::f32::simd_alias::Vec3A::from(a.0) * b).into()
});
impl_op_ex!(*= |a: &mut Vec3A, b: &f32| {
*a = Vec3A::from(crate::f32::simd_alias::Vec3A::from(a.0) * b)
});
impl_op_ex!(*|a: &f32, b: &Vec3A| -> Vec3A {
(a * crate::f32::simd_alias::Vec3A::from(b.0)).into()
});
impl_op_ex!(+ |a: &Vec3A, b: &Vec3A| -> Vec3A {
(crate::f32::simd_alias::Vec3A::from(a.0) + crate::f32::simd_alias::Vec3A::from(b.0)).into()
});
impl_op_ex!(+= |a: &mut Vec3A, b: &Vec3A| {
*a = Vec3A::from(crate::f32::simd_alias::Vec3A::from(a.0) + crate::f32::simd_alias::Vec3A::from(b.0))
});
impl_op_ex!(+ |a: &Vec3A, b: &f32| -> Vec3A {
(crate::f32::simd_alias::Vec3A::from(a.0) + b).into()
});
impl_op_ex!(+= |a: &mut Vec3A, b: &f32| {
*a = Vec3A::from(crate::f32::simd_alias::Vec3A::from(a.0) + b)
});
impl_op_ex!(+ |a: &f32, b: &Vec3A| -> Vec3A {
(a + crate::f32::simd_alias::Vec3A::from(b.0)).into()
});
impl_op_ex!(-|a: &Vec3A, b: &Vec3A| -> Vec3A {
(crate::f32::simd_alias::Vec3A::from(a.0) - crate::f32::simd_alias::Vec3A::from(b.0)).into()
});
impl_op_ex!(-= |a: &mut Vec3A, b: &Vec3A| {
*a = Vec3A::from(crate::f32::simd_alias::Vec3A::from(a.0) - crate::f32::simd_alias::Vec3A::from(b.0))
});
impl_op_ex!(-|a: &Vec3A, b: &f32| -> Vec3A {
(crate::f32::simd_alias::Vec3A::from(a.0) - b).into()
});
impl_op_ex!(-= |a: &mut Vec3A, b: &f32| {
*a = Vec3A::from(crate::f32::simd_alias::Vec3A::from(a.0) - b)
});
impl_op_ex!(-|a: &f32, b: &Vec3A| -> Vec3A {
(a - crate::f32::simd_alias::Vec3A::from(b.0)).into()
});
impl_op_ex!(% |a: &Vec3A, b: &Vec3A| -> Vec3A {
(crate::f32::simd_alias::Vec3A::from(a.0) % crate::f32::simd_alias::Vec3A::from(b.0)).into()
});
impl_op_ex!(%= |a: &mut Vec3A, b: &Vec3A| {
*a = Vec3A::from(crate::f32::simd_alias::Vec3A::from(a.0) % crate::f32::simd_alias::Vec3A::from(b.0))
});
impl_op_ex!(% |a: &Vec3A, b: &f32| -> Vec3A {
(crate::f32::simd_alias::Vec3A::from(a.0) % b).into()
});
impl_op_ex!(%= |a: &mut Vec3A, b: &f32| {
*a = Vec3A::from(crate::f32::simd_alias::Vec3A::from(a.0) % b)
});
impl_op_ex!(% |a: &f32, b: &Vec3A| -> Vec3A {
(a % crate::f32::simd_alias::Vec3A::from(b.0)).into()
});
impl<'a> Sum<&'a Self> for Vec3A {
#[inline]
fn sum<I>(iter: I) -> Self
where
I: Iterator<Item = &'a Self>,
{
iter.fold(Self::ZERO, |a, &b| Self::add(a, b))
}
}
impl<'a> Product<&'a Self> for Vec3A {
#[inline]
fn product<I>(iter: I) -> Self
where
I: Iterator<Item = &'a Self>,
{
iter.fold(Self::ONE, |a, &b| Self::mul(a, b))
}
}
impl Neg for Vec3A {
type Output = Self;
#[inline]
fn neg(self) -> Self {
crate::f32::simd_alias::Vec3A::neg(self.into()).into()
}
}
impl Index<usize> for Vec3A {
type Output = f32;
#[inline]
fn index(&self, index: usize) -> &Self::Output {
match index {
0 => &self.x,
1 => &self.y,
2 => &self.z,
_ => panic!("index out of bounds"),
}
}
}
impl IndexMut<usize> for Vec3A {
#[inline]
fn index_mut(&mut self, index: usize) -> &mut Self::Output {
match index {
0 => &mut self.x,
1 => &mut self.y,
2 => &mut self.z,
_ => panic!("index out of bounds"),
}
}
}
#[cfg(not(target_arch = "spirv"))]
impl fmt::Display for Vec3A {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "[{}, {}, {}]", self.x, self.y, self.z)
}
}
#[cfg(not(target_arch = "spirv"))]
impl fmt::Debug for Vec3A {
fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
fmt.debug_tuple(stringify!(Vec3A))
.field(&self.x)
.field(&self.y)
.field(&self.z)
.finish()
}
}
impl From<[f32; 3]> for Vec3A {
#[inline]
fn from(a: [f32; 3]) -> Self {
Self::new(a[0], a[1], a[2])
}
}
impl From<Vec3A> for [f32; 3] {
#[inline]
fn from(v: Vec3A) -> Self {
crate::f32::simd_alias::Vec3A::from(v).into()
}
}
impl From<(f32, f32, f32)> for Vec3A {
#[inline]
fn from(t: (f32, f32, f32)) -> Self {
Self::new(t.0, t.1, t.2)
}
}
impl From<Vec3A> for (f32, f32, f32) {
#[inline]
fn from(v: Vec3A) -> Self {
crate::f32::simd_alias::Vec3A::from(v).into()
}
}
impl From<Vec3> for Vec3A {
#[inline]
fn from(v: Vec3) -> Self {
Self::new(v.x, v.y, v.z)
}
}
impl From<Vec4> for Vec3A {
#[inline]
fn from(v: Vec4) -> Self {
Self(v.0)
}
}
impl From<Vec3A> for Vec3 {
#[inline]
fn from(v: Vec3A) -> Self {
Vec3::from(crate::f32::simd_alias::Vec3A::from(v))
}
}
impl From<(Vec2, f32)> for Vec3A {
#[inline]
fn from((v, z): (Vec2, f32)) -> Self {
Self::new(v.x, v.y, z)
}
}
impl Deref for Vec3A {
type Target = crate::deref::Vec3<f32>;
#[inline]
fn deref(&self) -> &Self::Target {
unsafe { &*(self as *const Self).cast() }
}
}
impl DerefMut for Vec3A {
#[inline]
fn deref_mut(&mut self) -> &mut Self::Target {
unsafe { &mut *(self as *mut Self).cast() }
}
}
#[cfg(not(target_arch = "spirv"))]
impl AsRef<[f32; 3]> for Vec3A {
#[inline]
fn as_ref(&self) -> &[f32; 3] {
unsafe { &*(self as *const Vec3A as *const [f32; 3]) }
}
}
#[cfg(not(target_arch = "spirv"))]
impl AsMut<[f32; 3]> for Vec3A {
#[inline]
fn as_mut(&mut self) -> &mut [f32; 3] {
unsafe { &mut *(self as *mut Vec3A as *mut [f32; 3]) }
}
}