switch to using separate VecBool8/16/32/64
[vector-math.git] / src / f16.rs
index 5bf2e69c213a401a12faf75b6dc8953a8cad9229..ee13d9992135320be52859df3d5c51f6482ff933 100644 (file)
@@ -8,11 +8,10 @@ use crate::traits::{ConvertTo, Float};
 use half::f16 as F16Impl;
 
 #[cfg(not(feature = "f16"))]
-#[derive(Clone, Copy, PartialEq, PartialOrd, Debug)]
-enum F16Impl {}
+type F16Impl = u16;
 
 #[derive(Clone, Copy, PartialEq, PartialOrd, Debug)]
-#[cfg_attr(feature = "f16", repr(transparent))]
+#[repr(transparent)]
 pub struct F16(F16Impl);
 
 #[cfg(feature = "f16")]
@@ -67,10 +66,7 @@ macro_rules! impl_from_f16 {
         $(
             impl From<F16> for $ty {
                 fn from(v: F16) -> Self {
-                    #[cfg(feature = "f16")]
-                    return v.0.into();
-                    #[cfg(not(feature = "f16"))]
-                    match v.0 {}
+                    f16_impl!(v.0.into(), [v])
                 }
             }
 
@@ -133,7 +129,7 @@ impl Neg for F16 {
     type Output = Self;
 
     fn neg(self) -> Self::Output {
-        Self::from_bits(self.to_bits() ^ 0x8000)
+        f16_impl!(Self::from_bits(self.to_bits() ^ 0x8000), [])
     }
 }
 
@@ -165,11 +161,13 @@ impl_bin_op_using_f32! {
     Rem, rem, RemAssign, rem_assign;
 }
 
-impl Float<u32> for F16 {
+impl Float for F16 {
+    type FloatEncoding = F16;
     type BitsType = u16;
+    type SignedBitsType = i16;
 
     fn abs(self) -> Self {
-        Self::from_bits(self.to_bits() & 0x7FFF)
+        f16_impl!(Self::from_bits(self.to_bits() & 0x7FFF), [])
     }
 
     fn trunc(self) -> Self {
@@ -206,11 +204,17 @@ impl Float<u32> for F16 {
     }
 
     fn from_bits(v: Self::BitsType) -> Self {
-        f16_impl!(F16(F16Impl::from_bits(v)), [v])
+        #[cfg(feature = "f16")]
+        return F16(F16Impl::from_bits(v));
+        #[cfg(not(feature = "f16"))]
+        return F16(v);
     }
 
     fn to_bits(self) -> Self::BitsType {
-        f16_impl!(self.0.to_bits(), [])
+        #[cfg(feature = "f16")]
+        return self.0.to_bits();
+        #[cfg(not(feature = "f16"))]
+        return self.0;
     }
 }