use crate::nums::*;
use crate::{BVec2x4, UnitVec2x4, Vec3x4};
use crate::Vec2;
#[cfg(not(target_arch = "spirv"))]
use core::fmt;
use core::iter::{Product, Sum};
use core::ops::*;
use auto_ops_det::impl_op_ex;
use core::ops;
#[inline]
pub const fn vec2x4(x: f32x4, y: f32x4) -> Vec2x4 {
Vec2x4::new(x, y)
}
#[derive(Clone, Copy, PartialEq)]
#[cfg_attr(feature = "cuda", repr(align(8)))]
#[repr(C)]
pub struct Vec2x4 {
pub x: f32x4,
pub y: f32x4,
}
impl Vec2x4 {
pub const ZERO: Self = Self::splat(f32x4::ZERO);
pub const ONE: Self = Self::splat(f32x4::ONE);
pub const NEG_ONE: Self = Self::splat(f32x4::NEG_ONE);
pub const NAN: Self = Self::splat(f32x4::NAN);
pub const X: Self = Self::new(f32x4::ONE, f32x4::ZERO);
pub const Y: Self = Self::new(f32x4::ZERO, f32x4::ONE);
pub const NEG_X: Self = Self::new(f32x4::NEG_ONE, f32x4::ZERO);
pub const NEG_Y: Self = Self::new(f32x4::ZERO, f32x4::NEG_ONE);
pub const AXES: [Self; 2] = [Self::X, Self::Y];
#[inline]
pub const fn new(x: f32x4, y: f32x4) -> Self {
Self { x, y }
}
#[inline]
pub const fn splat(v: f32x4) -> Self {
Self { x: v, y: v }
}
#[inline]
pub fn select(mask: BVec2x4, if_true: Self, if_false: Self) -> Self {
Self {
x: if_true.x.select(mask.x, if_false.x),
y: if_true.y.select(mask.y, if_false.y),
}
}
#[inline]
pub const fn from_array(a: [f32x4; 2]) -> Self {
Self::new(a[0], a[1])
}
#[inline]
pub const fn to_array(&self) -> [f32x4; 2] {
[self.x, self.y]
}
#[inline]
pub const fn from_slice(slice: &[f32x4]) -> Self {
Self::new(slice[0], slice[1])
}
#[inline]
pub fn write_to_slice(self, slice: &mut [f32x4]) {
slice[0] = self.x;
slice[1] = self.y;
}
#[inline]
pub const fn extend(self, z: f32x4) -> Vec3x4 {
Vec3x4::new(self.x, self.y, z)
}
#[inline]
pub(crate) fn dot(self, rhs: Self) -> f32x4 {
(self.x * rhs.x) + (self.y * rhs.y)
}
#[inline]
pub fn min(self, rhs: Self) -> Self {
Self {
x: self.x.minf(rhs.x),
y: self.y.minf(rhs.y),
}
}
#[inline]
pub fn max(self, rhs: Self) -> Self {
Self {
x: self.x.maxf(rhs.x),
y: self.y.maxf(rhs.y),
}
}
#[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) -> f32x4 {
self.x.minf(self.y)
}
#[inline]
pub fn max_element(self) -> f32x4 {
self.x.maxf(self.y)
}
#[inline]
pub fn cmpeq(self, rhs: Self) -> BVec2x4 {
BVec2x4::new(self.x.eq(rhs.x), self.y.eq(rhs.y))
}
#[inline]
pub fn cmpne(self, rhs: Self) -> BVec2x4 {
BVec2x4::new(self.x.ne(rhs.x), self.y.ne(rhs.y))
}
#[inline]
pub fn cmpge(self, rhs: Self) -> BVec2x4 {
BVec2x4::new(self.x.ge(rhs.x), self.y.ge(rhs.y))
}
#[inline]
pub fn cmpgt(self, rhs: Self) -> BVec2x4 {
BVec2x4::new(self.x.gt(rhs.x), self.y.gt(rhs.y))
}
#[inline]
pub fn cmple(self, rhs: Self) -> BVec2x4 {
BVec2x4::new(self.x.le(rhs.x), self.y.le(rhs.y))
}
#[inline]
pub fn cmplt(self, rhs: Self) -> BVec2x4 {
BVec2x4::new(self.x.lt(rhs.x), self.y.lt(rhs.y))
}
#[inline]
pub fn abs(self) -> Self {
Self {
x: self.x.absf(),
y: self.y.absf(),
}
}
#[inline]
pub fn signum(self) -> Self {
Self {
x: self.x.signumf(),
y: self.y.signumf(),
}
}
#[inline]
pub fn is_finite(self) -> bool32x4 {
self.x.is_finite() & self.y.is_finite()
}
#[inline]
pub fn is_nan(self) -> bool32x4 {
self.x.is_nan() | self.y.is_nan()
}
#[inline]
pub fn is_nan_mask(self) -> BVec2x4 {
BVec2x4::new(self.x.is_nan(), self.y.is_nan())
}
#[inline]
pub fn round(self) -> Self {
Self {
x: self.x.roundf(),
y: self.y.roundf(),
}
}
#[inline]
pub fn floor(self) -> Self {
Self {
x: self.x.floorf(),
y: self.y.floorf(),
}
}
#[inline]
pub fn ceil(self) -> Self {
Self {
x: self.x.ceilf(),
y: self.y.ceilf(),
}
}
#[inline]
pub fn fract(self) -> Self {
self - self.floor()
}
#[inline]
pub fn exp(self) -> Self {
Self::new(self.x.expf(), self.y.expf())
}
#[inline]
pub fn powf(self, n: f32x4) -> Self {
Self::new(self.x.powff(n), self.y.powff(n))
}
#[inline]
pub fn recip(self) -> Self {
Self {
x: self.x.recip(),
y: self.y.recip(),
}
}
#[doc(alias = "magnitude")]
#[inline]
pub fn length(self) -> f32x4 {
self.dot(self).sqrtf()
}
#[doc(alias = "magnitude2")]
#[inline]
pub fn length_squared(self) -> f32x4 {
self.dot(self)
}
#[inline]
pub fn length_recip(self) -> f32x4 {
self.length().recip()
}
#[inline]
pub fn distance(self, rhs: Self) -> f32x4 {
(self - rhs).length()
}
#[inline]
pub fn distance_squared(self, rhs: Self) -> f32x4 {
(self - rhs).length_squared()
}
#[must_use]
#[inline]
pub fn normalize(self) -> Self {
let length = self.length();
glam_assert!(length.gt(f32x4::ZERO), "Trying to normalize {:?}", self);
glam_assert!(self.is_finite(), "Trying to normalize {:?}", self);
glam_assert!(length.is_finite(), "Trying to normalize {:?}", self);
#[allow(clippy::let_and_return)]
let normalized = self.mul(length.recip());
glam_assert!(normalized.is_finite(), "Trying to normalize {:?}", self);
normalized
}
#[must_use]
#[inline]
pub fn normalize_and_length(self) -> (Self, f32x4) {
#[allow(clippy::let_and_return)]
let length = self.length();
glam_assert!(length.gt(f32x4::ZERO), "Trying to normalize {:?}", self);
glam_assert!(self.is_finite(), "Trying to normalize {:?}", self);
glam_assert!(length.is_finite(), "Trying to normalize {:?}", self);
let normalized = self.mul(length.recip());
glam_assert!(normalized.is_finite(), "Trying to normalize {:?}", self);
(normalized, length)
}
#[must_use]
#[inline]
pub fn normalize_to_unit(self) -> UnitVec2x4 {
self.normalize().as_unit_vec2x4_unchecked()
}
#[must_use]
#[inline]
pub fn normalize_to_unit_and_length(self) -> (UnitVec2x4, f32x4) {
let res = self.normalize_and_length();
(res.0.as_unit_vec2x4_unchecked(), res.1)
}
#[must_use]
#[inline]
pub fn normalize_or_zero(self) -> Self {
let rcp = self.length_recip();
let cond = rcp.is_finite() & rcp.gt(f32x4::ZERO);
Self::lane_select(cond, self * rcp, Self::ZERO)
}
#[inline]
pub fn is_normalized(self) -> bool32x4 {
(self.length_squared() - f32x4::ONE)
.absf()
.le(f32x4::from(1e-4))
}
#[inline]
pub fn lane_select(cond: bool32x4, if_true: Self, if_false: Self) -> Self {
Self {
x: if_true.x.select(cond, if_false.x),
y: if_true.y.select(cond, if_false.y),
}
}
#[inline]
pub fn splat_soa(v: Vec2) -> Self {
Self {
x: f32x4::splat(v.x),
y: f32x4::splat(v.y),
}
}
#[inline]
pub fn extract_lane(&self, i: usize) -> Vec2 {
Vec2::new(self.x.extract(i), self.y.extract(i))
}
#[inline]
pub fn replace_lane(&mut self, i: usize, v: Vec2) {
self.x.replace(i, v.x);
self.y.replace(i, v.y);
}
#[inline]
pub fn split_soa(self) -> [Vec2; 4] {
[
self.extract_lane(0),
self.extract_lane(1),
self.extract_lane(2),
self.extract_lane(3),
]
}
#[inline]
pub fn compose_soa(a: &[Vec2; 4]) -> Vec2x4 {
let mut v = Self::default();
v.replace_lane(0, a[0]);
v.replace_lane(1, a[1]);
v.replace_lane(2, a[2]);
v.replace_lane(3, a[3]);
v
}
#[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: UnitVec2x4) -> Self {
rhs.as_vec2x4() * self.dot(rhs.as_vec2x4())
}
#[must_use]
#[inline]
pub fn reject_from_normalized(self, rhs: UnitVec2x4) -> Self {
self - self.project_onto_normalized(rhs)
}
#[doc(alias = "mix")]
#[inline]
pub fn lerp(self, rhs: Self, s: f32x4) -> Self {
self + ((rhs - self) * s)
}
#[inline]
pub fn abs_diff_eq(self, rhs: Self, max_abs_diff: f32x4) -> bool32x4 {
self.sub(rhs).abs().cmple(Self::splat(max_abs_diff)).all()
}
#[inline]
pub fn clamp_length(self, min: f32x4, max: f32x4) -> Self {
glam_assert!(min.le(max));
let length_sq = self.length_squared();
let cond1 = length_sq.ge(min * min);
let cond2 = length_sq.lt(max * max);
let res1 = self * (length_sq.sqrtf().recip() * min);
let res2 = self * (length_sq.sqrtf().recip() * max);
let res3 = self;
let x = res3.x.select(cond1, res1.x).select(cond2, res2.x);
let y = res3.y.select(cond1, res1.y).select(cond2, res2.y);
Self { x, y }
}
#[inline]
pub fn clamp_length_max(self, max: f32x4) -> Self {
let length_sq = self.length_squared();
let cond = length_sq.gt(max * max);
let res1 = self * (length_sq.sqrtf().recip() * max);
let res2 = self;
let x = res1.x.select(cond, res2.x);
let y = res1.y.select(cond, res2.y);
Self { x, y }
}
#[inline]
pub fn clamp_length_min(self, min: f32x4) -> Self {
let length_sq = self.length_squared();
let cond = length_sq.lt(min * min);
let res1 = self * (length_sq.sqrtf().recip() * min);
let res2 = self;
let x = res1.x.select(cond, res2.x);
let y = res1.y.select(cond, res2.y);
Self { x, y }
}
#[inline]
pub fn mul_add(self, a: Self, b: Self) -> Self {
Self::new(self.x.mul_addf(a.x, b.x), self.y.mul_addf(a.y, b.y))
}
#[inline]
pub fn from_angle(angle: f32x4) -> Self {
let (sin, cos) = angle.sin_cosf();
Self { x: cos, y: sin }
}
#[inline]
pub fn angle_between(self, rhs: Self) -> f32x4 {
use crate::FloatEx;
let angle =
(self.dot(rhs) / (self.length_squared() * rhs.length_squared()).sqrtf()).acos_approx();
angle * self.perp_dot(rhs).signumf()
}
#[inline]
pub fn perp(self) -> Self {
Self {
x: -self.y,
y: self.x,
}
}
#[doc(alias = "wedge")]
#[doc(alias = "cross")]
#[doc(alias = "determinant")]
#[inline]
pub fn perp_dot(self, rhs: Self) -> f32x4 {
(self.x * rhs.y) - (self.y * rhs.x)
}
#[must_use]
#[inline]
pub fn rotate(self, rhs: Self) -> Self {
Self {
x: self.x * rhs.x - self.y * rhs.y,
y: self.y * rhs.x + self.x * rhs.y,
}
}
}
impl Default for Vec2x4 {
#[inline]
fn default() -> Self {
Self::ZERO
}
}
impl_op_ex!(/ |a: &Vec2x4, b: &Vec2x4| -> Vec2x4 {
Vec2x4 {
x: a.x.div(b.x),
y: a.y.div(b.y),
}
});
impl_op_ex!(/= |a: &mut Vec2x4, b: &Vec2x4| {
a.x.div_assign(b.x);
a.y.div_assign(b.y);
});
impl_op_ex!(/ |a: &Vec2x4, b: &f32x4| -> Vec2x4 {
Vec2x4 {
x: a.x.div(b),
y: a.y.div(b),
}
});
impl_op_ex!(/= |a: &mut Vec2x4, b: &f32x4| {
a.x.div_assign(b);
a.y.div_assign(b);
});
impl_op_ex!(/ |a: &f32x4, b: &Vec2x4| -> Vec2x4 {
Vec2x4 {
x: a.div(b.x),
y: a.div(b.y),
}
});
impl_op_ex!(*|a: &Vec2x4, b: &Vec2x4| -> Vec2x4 {
Vec2x4 {
x: a.x.mul(b.x),
y: a.y.mul(b.y),
}
});
impl_op_ex!(*= |a: &mut Vec2x4, b: &Vec2x4| {
a.x.mul_assign(b.x);
a.y.mul_assign(b.y);
});
impl_op_ex!(*|a: &Vec2x4, b: &f32x4| -> Vec2x4 {
Vec2x4 {
x: a.x.mul(b),
y: a.y.mul(b),
}
});
impl_op_ex!(*= |a: &mut Vec2x4, b: &f32x4| {
a.x.mul_assign(b);
a.y.mul_assign(b);
});
impl_op_ex!(*|a: &f32x4, b: &Vec2x4| -> Vec2x4 {
Vec2x4 {
x: a.mul(b.x),
y: a.mul(b.y),
}
});
impl_op_ex!(+ |a: &Vec2x4, b: &Vec2x4| -> Vec2x4 {
Vec2x4 {
x: a.x.add(b.x),
y: a.y.add(b.y),
}
});
impl_op_ex!(+= |a: &mut Vec2x4, b: &Vec2x4| {
a.x.add_assign(b.x);
a.y.add_assign(b.y);
});
impl_op_ex!(+ |a: &Vec2x4, b: &f32x4| -> Vec2x4 {
Vec2x4 {
x: a.x.add(b),
y: a.y.add(b),
}
});
impl_op_ex!(+= |a: &mut Vec2x4, b: &f32x4| {
a.x.add_assign(b);
a.y.add_assign(b);
});
impl_op_ex!(+ |a: &f32x4, b: &Vec2x4| -> Vec2x4 {
Vec2x4 {
x: a.add(b.x),
y: a.add(b.y),
}
});
impl_op_ex!(-|a: &Vec2x4, b: &Vec2x4| -> Vec2x4 {
Vec2x4 {
x: a.x.sub(b.x),
y: a.y.sub(b.y),
}
});
impl_op_ex!(-= |a: &mut Vec2x4, b: &Vec2x4| {
a.x.sub_assign(b.x);
a.y.sub_assign(b.y);
});
impl_op_ex!(-|a: &Vec2x4, b: &f32x4| -> Vec2x4 {
Vec2x4 {
x: a.x.sub(b),
y: a.y.sub(b),
}
});
impl_op_ex!(-= |a: &mut Vec2x4, b: &f32x4| {
a.x.sub_assign(b);
a.y.sub_assign(b);
});
impl_op_ex!(-|a: &f32x4, b: &Vec2x4| -> Vec2x4 {
Vec2x4 {
x: a.sub(b.x),
y: a.sub(b.y),
}
});
impl_op_ex!(% |a: &Vec2x4, b: &Vec2x4| -> Vec2x4 {
Vec2x4 {
x: a.x.rem(b.x),
y: a.y.rem(b.y),
}
});
impl_op_ex!(%= |a: &mut Vec2x4, b: &Vec2x4| {
a.x.rem_assign(b.x);
a.y.rem_assign(b.y);
});
impl_op_ex!(% |a: &Vec2x4, b: &f32x4| -> Vec2x4 {
Vec2x4 {
x: a.x.rem(b),
y: a.y.rem(b),
}
});
impl_op_ex!(%= |a: &mut Vec2x4, b: &f32x4| {
a.x.rem_assign(b);
a.y.rem_assign(b);
});
impl_op_ex!(% |a: &f32x4, b: &Vec2x4| -> Vec2x4 {
Vec2x4 {
x: a.rem(b.x),
y: a.rem(b.y),
}
});
impl<'a> Sum<&'a Self> for Vec2x4 {
#[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 Vec2x4 {
#[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 Vec2x4 {
type Output = Self;
#[inline]
fn neg(self) -> Self {
Self {
x: self.x.neg(),
y: self.y.neg(),
}
}
}
impl Index<usize> for Vec2x4 {
type Output = f32x4;
#[inline]
fn index(&self, index: usize) -> &Self::Output {
match index {
0 => &self.x,
1 => &self.y,
_ => panic!("index out of bounds"),
}
}
}
impl IndexMut<usize> for Vec2x4 {
#[inline]
fn index_mut(&mut self, index: usize) -> &mut Self::Output {
match index {
0 => &mut self.x,
1 => &mut self.y,
_ => panic!("index out of bounds"),
}
}
}
#[cfg(not(target_arch = "spirv"))]
impl fmt::Display for Vec2x4 {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "[{}, {}]", self.x, self.y)
}
}
#[cfg(not(target_arch = "spirv"))]
impl fmt::Debug for Vec2x4 {
fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
fmt.debug_tuple(stringify!(Vec2x4))
.field(&self.x)
.field(&self.y)
.finish()
}
}
impl From<[f32x4; 2]> for Vec2x4 {
#[inline]
fn from(a: [f32x4; 2]) -> Self {
Self::new(a[0], a[1])
}
}
impl From<Vec2x4> for [f32x4; 2] {
#[inline]
fn from(v: Vec2x4) -> Self {
[v.x, v.y]
}
}
impl From<(f32x4, f32x4)> for Vec2x4 {
#[inline]
fn from(t: (f32x4, f32x4)) -> Self {
Self::new(t.0, t.1)
}
}
impl From<Vec2x4> for (f32x4, f32x4) {
#[inline]
fn from(v: Vec2x4) -> Self {
(v.x, v.y)
}
}
#[cfg(not(target_arch = "spirv"))]
impl AsRef<[f32x4; 2]> for Vec2x4 {
#[inline]
fn as_ref(&self) -> &[f32x4; 2] {
unsafe { &*(self as *const Vec2x4 as *const [f32x4; 2]) }
}
}
#[cfg(not(target_arch = "spirv"))]
impl AsMut<[f32x4; 2]> for Vec2x4 {
#[inline]
fn as_mut(&mut self) -> &mut [f32x4; 2] {
unsafe { &mut *(self as *mut Vec2x4 as *mut [f32x4; 2]) }
}
}