use super::Vec3;
#[cfg(not(target_arch = "spirv"))]
use core::fmt;
use core::ops::*;
#[derive(Clone, Copy, Default, PartialEq, Eq, Ord, PartialOrd, Hash)]
#[repr(C)]
pub struct Vec3Mask(pub(crate) u32, pub(crate) u32, pub(crate) u32);
impl Vec3Mask {
#[inline]
pub fn new(x: bool, y: bool, z: bool) -> Self {
const MASK: [u32; 2] = [0, 0xff_ff_ff_ff];
Self(MASK[x as usize], MASK[y as usize], MASK[z as usize])
}
#[inline]
pub fn bitmask(&self) -> u32 {
(self.0 & 0x1) | (self.1 & 0x1) << 1 | (self.2 & 0x1) << 2
}
#[inline]
pub fn any(&self) -> bool {
((self.0 | self.1 | self.2) & 0x1) != 0
}
#[inline]
pub fn all(&self) -> bool {
((self.0 & self.1 & self.2) & 0x1) != 0
}
#[inline]
pub fn select(self, if_true: Vec3, if_false: Vec3) -> Vec3 {
Vec3 {
x: if self.0 != 0 { if_true.x } else { if_false.x },
y: if self.1 != 0 { if_true.y } else { if_false.y },
z: if self.2 != 0 { if_true.z } else { if_false.z },
}
}
}
impl BitAnd for Vec3Mask {
type Output = Self;
#[inline]
fn bitand(self, other: Self) -> Self {
Self(self.0 & other.0, self.1 & other.1, self.2 & other.2)
}
}
impl BitAndAssign for Vec3Mask {
#[inline]
fn bitand_assign(&mut self, other: Self) {
self.0 &= other.0;
self.1 &= other.1;
self.2 &= other.2;
}
}
impl BitOr for Vec3Mask {
type Output = Self;
#[inline]
fn bitor(self, other: Self) -> Self {
Self(self.0 | other.0, self.1 | other.1, self.2 | other.2)
}
}
impl BitOrAssign for Vec3Mask {
#[inline]
fn bitor_assign(&mut self, other: Self) {
self.0 |= other.0;
self.1 |= other.1;
self.2 |= other.2;
}
}
impl Not for Vec3Mask {
type Output = Self;
#[inline]
fn not(self) -> Self {
Self(!self.0, !self.1, !self.2)
}
}
#[cfg(not(target_arch = "spirv"))]
impl fmt::Debug for Vec3Mask {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "Vec3Mask({:#x}, {:#x}, {:#x})", self.0, self.1, self.2)
}
}
#[cfg(not(target_arch = "spirv"))]
impl fmt::Display for Vec3Mask {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
let arr = self.as_ref();
write!(f, "[{}, {}, {}]", arr[0] != 0, arr[1] != 0, arr[2] != 0,)
}
}
impl From<Vec3Mask> for [u32; 3] {
#[inline]
fn from(mask: Vec3Mask) -> Self {
*mask.as_ref()
}
}
impl AsRef<[u32; 3]> for Vec3Mask {
#[inline]
fn as_ref(&self) -> &[u32; 3] {
unsafe { &*(self as *const Self as *const [u32; 3]) }
}
}