Add all possible AsPrimitive for NotNan

reem · Mar 21, 2023 · a7a6b4e · a7a6b4e
1 parent 1972608
commit a7a6b4e
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -470,12 +470,27 @@ impl<T: NumCast> NumCast for OrderedFloat<T> {
 }
 
 macro_rules! impl_as_primitive {
+    (@ (NotNan<$T: ty>) => $(#[$cfg:meta])* impl (NotNan<$U: ty>) ) => {
+        $(#[$cfg])*
+        impl AsPrimitive<NotNan<$U>> for NotNan<$T> {
+            #[inline] fn as_(self) -> NotNan<$U> {
+                // Safety: `NotNan` guarantees that the value is not NaN.
+                unsafe {NotNan::new_unchecked(self.0 as $U) }
+            }
+        }
+    };
     (@ ($T: ty) => $(#[$cfg:meta])* impl (NotNan<$U: ty>) ) => {
         $(#[$cfg])*
         impl AsPrimitive<NotNan<$U>> for $T {
             #[inline] fn as_(self) -> NotNan<$U> { NotNan(self as $U) }
         }
     };
+    (@ (NotNan<$T: ty>) => $(#[$cfg:meta])* impl ($U: ty) ) => {
+        $(#[$cfg])*
+        impl AsPrimitive<$U> for NotNan<$T> {
+            #[inline] fn as_(self) -> $U { self.0 as $U }
+        }
+    };
     (@ (OrderedFloat<$T: ty>) => $(#[$cfg:meta])* impl (OrderedFloat<$U: ty>) ) => {
         $(#[$cfg])*
         impl AsPrimitive<OrderedFloat<$U>> for OrderedFloat<$T> {
@@ -502,6 +517,9 @@ macro_rules! impl_as_primitive {
 impl_as_primitive!((OrderedFloat<f32>) => { (OrderedFloat<f32>), (OrderedFloat<f64>) });
 impl_as_primitive!((OrderedFloat<f64>) => { (OrderedFloat<f32>), (OrderedFloat<f64>) });
 
+impl_as_primitive!((NotNan<f32>) => { (NotNan<f32>), (NotNan<f64>) });
+impl_as_primitive!((NotNan<f64>) => { (NotNan<f32>), (NotNan<f64>) });
+
 impl_as_primitive!((u8) => { (OrderedFloat<f32>), (OrderedFloat<f64>) });
 impl_as_primitive!((i8) => { (OrderedFloat<f32>), (OrderedFloat<f64>) });
 impl_as_primitive!((u16) => { (OrderedFloat<f32>), (OrderedFloat<f64>) });
@@ -529,6 +547,10 @@ impl_as_primitive!((isize) => { (NotNan<f32>), (NotNan<f64>) });
 impl_as_primitive!((OrderedFloat<f32>) => { (u8), (u16), (u32), (u64), (usize), (i8), (i16), (i32), (i64), (isize), (f32), (f64) });
 impl_as_primitive!((OrderedFloat<f64>) => { (u8), (u16), (u32), (u64), (usize), (i8), (i16), (i32), (i64), (isize), (f32), (f64) });
 
+impl_as_primitive!((NotNan<f32>) => { (u8), (u16), (u32), (u64), (usize), (i8), (i16), (i32), (i64), (isize), (f32), (f64) });
+impl_as_primitive!((NotNan<f64>) => { (u8), (u16), (u32), (u64), (usize), (i8), (i16), (i32), (i64), (isize), (f32), (f64) });
+
+
 impl<T: FromPrimitive> FromPrimitive for OrderedFloat<T> {
     fn from_i64(n: i64) -> Option<Self> {
         T::from_i64(n).map(OrderedFloat)