chore: Add helper to turn typenums into const values

Cargo.toml

@@ -51,6 +51,7 @@ allocator-api2 = "0.2.14"
 allocator-api2-tests = "0.2.14"
 memsec = "0.6.3"
 rand = "0.8.5"
+typenum = "1.17.0"
 log = { version = "0.4.20" }
 clap = { version = "4.4.10", features = ["derive"] }
 serde = { version = "1.0.193", features = ["derive"] }

util/Cargo.toml

@@ -14,3 +14,5 @@ readme = "readme.md"
 [dependencies]
 base64 = { workspace = true }
 anyhow = { workspace = true }
+typenum = { workspace = true }
+static_assertions = { workspace = true }

util/src/lib.rs

@@ -1,3 +1,5 @@
+#![recursion_limit = "256"]
+
 pub mod b64;
 pub mod file;
 pub mod functional;
@@ -5,3 +7,4 @@ pub mod mem;
 pub mod ord;
 pub mod result;
 pub mod time;
+pub mod typenum;

util/src/typenum.rs

@@ -0,0 +1,326 @@
+use typenum::bit::{B1, B0};
+use typenum::int::{Z0, NInt, PInt};
+use typenum::uint::{UInt, UTerm};
+use typenum::marker_traits as markers;
+
+/// Convenience macro to convert type numbers to constant integers
+#[macro_export]
+macro_rules! typenum2const {
+    ($val:ty) => { typenum2const!($val as _) };
+    ($val:ty as $type:ty) => { < $val as $crate::typenum::IntoConst<$type> >::VALUE };
+}
+
+/// Trait implemented by constant integers to facilitate conversion to constant integers
+pub trait IntoConst<T> {
+    const VALUE: T;
+}
+
+struct ConstApplyNegSign<T: AssociatedUnsigned, Param: IntoConst<<T as AssociatedUnsigned>::Type>>(*const T, *const Param);
+struct ConstApplyPosSign<T: AssociatedUnsigned, Param: IntoConst<<T as AssociatedUnsigned>::Type>>(*const T, *const Param);
+struct ConstLshift<T, Param: IntoConst<T>, const SHIFT : i32>(*const T, *const Param); // impl IntoConst<T>
+struct ConstAdd<T, Lhs: IntoConst<T>, Rhs: IntoConst<T>>(*const T, *const Lhs, *const Rhs); // impl IntoConst<T>
+
+/// Assigns an unsigned type to a signed type
+trait AssociatedUnsigned {
+    type Type;
+}
+
+macro_rules! impl_into_const {
+    ( $from:ty as $to:ty := $impl:expr) =>  {
+        impl IntoConst<$to> for $from {
+            const VALUE : $to = $impl;
+        }
+    };
+}
+
+macro_rules! impl_numeric_into_const_common {
+    ($type:ty) =>  {
+        impl_into_const! { Z0 as $type := 0 }
+        impl_into_const! { B0 as $type := 0 }
+        impl_into_const! { B1 as $type := 1 }
+        impl_into_const! { UTerm as $type := 0 }
+
+        impl<Param: IntoConst<$type>, const SHIFT : i32> IntoConst<$type> for ConstLshift<$type, Param, SHIFT> {
+            const VALUE : $type = Param::VALUE << SHIFT;
+        }
+
+        impl<Lhs: IntoConst<$type>, Rhs: IntoConst<$type>> IntoConst<$type> for ConstAdd<$type, Lhs, Rhs> {
+            const VALUE: $type = <Lhs as IntoConst<$type>>::VALUE + <Rhs as IntoConst<$type>>::VALUE;
+        }
+    }
+}
+
+macro_rules! impl_numeric_into_const_unsigned {
+    ($($to_list:ty),*) =>  {
+        $( impl_numeric_into_const_unsigned! { @impl $to_list } )*
+    };
+
+    (@impl $type:ty) =>  {
+        impl_numeric_into_const_common!{ $type }
+
+        impl AssociatedUnsigned for $type {
+            type Type = $type;
+        }
+
+        impl<Param: IntoConst<$type>> IntoConst<$type> for ConstApplyPosSign<$type, Param> {
+            const VALUE : $type = Param::VALUE;
+        }
+    };
+}
+
+macro_rules! impl_numeric_into_const_signed {
+    ($($to_list:ty : $unsigned_list:ty),*) =>  {
+        $( impl_numeric_into_const_signed! { @impl $to_list : $unsigned_list} )*
+    };
+
+    (@impl $type:ty : $unsigned:ty) =>  {
+        impl_numeric_into_const_common!{ $type }
+
+        impl AssociatedUnsigned for $type {
+            type Type = $unsigned;
+        }
+
+        impl<Param: IntoConst<$unsigned>> IntoConst<$type> for ConstApplyPosSign<$type, Param> {
+            const VALUE : $type = Param::VALUE as $type;
+        }
+
+        impl<Param: IntoConst<$unsigned>> IntoConst<$type> for ConstApplyNegSign<$type, Param> {
+            const VALUE : $type =
+                if Param::VALUE == (1 as $unsigned).rotate_right(1) {
+                    // Handle the negative value without an associated positive value (e.g. -128
+                    // for i8)
+                    < $type >::MIN
+                } else {
+                    -(Param::VALUE as $type)
+                };
+        }
+    };
+}
+
+impl_into_const! { B0 as bool := false }
+impl_into_const! { B1 as bool := true }
+impl_numeric_into_const_unsigned!{ usize, u8, u16, u32, u64, u128 }
+impl_numeric_into_const_signed!{ isize : usize, i8 : u8, i16 : u16, i32 : u32, i64 : u64, i128 : u128 }
+
+// Unsigned type numbers to const integers
+impl<Ret, Rest, Bit> IntoConst<Ret> for UInt<Rest, Bit>
+    where 
+        Rest: IntoConst<Ret>,
+        Bit: IntoConst<Ret>,
+        ConstLshift<Ret, Rest, 1>: IntoConst<Ret>,
+        ConstAdd<Ret, ConstLshift<Ret, Rest, 1>, Bit>: IntoConst<Ret>,
+{
+    const VALUE: Ret = <
+            ConstAdd<Ret, ConstLshift<Ret, Rest, 1>, Bit> as IntoConst<Ret>
+        >::VALUE;
+}
+
+// Signed type numbers with positive sign to const integers
+impl<Ret, Unsigned> IntoConst<Ret> for PInt<Unsigned>
+    where 
+        Ret: AssociatedUnsigned,
+        Unsigned: markers::Unsigned + markers::NonZero + IntoConst<<Ret as AssociatedUnsigned>::Type>,
+        ConstApplyPosSign<Ret, Unsigned>: IntoConst<Ret> {
+    const VALUE: Ret = <ConstApplyPosSign::<Ret, Unsigned> as IntoConst<Ret>>::VALUE;
+}
+
+// Signed type numbers with negative sign to const integers
+impl<Ret, Unsigned> IntoConst<Ret> for NInt<Unsigned>
+    where 
+        Ret: AssociatedUnsigned,
+        Unsigned: markers::Unsigned + markers::NonZero + IntoConst<<Ret as AssociatedUnsigned>::Type>,
+        ConstApplyNegSign<Ret, Unsigned>: IntoConst<Ret> {
+    const VALUE: Ret = <ConstApplyNegSign::<Ret, Unsigned> as IntoConst<Ret>>::VALUE;
+}
+
+mod test {
+    use static_assertions::const_assert_eq;
+    use typenum::consts::*;
+    use typenum::op;
+
+    macro_rules! test_const_conversion {
+        // Type groups
+
+        (($($typenum:ty),*) >= u7 = $const:expr $(; $($rest:tt)*)?) => {
+            $( test_const_conversion! { ($typenum) as (u8, u16, u32, u64, u128) = $const } )*
+            $( test_const_conversion! { ($typenum) as (i8, i16, i32, i64, i128) = $const } )*
+            $( test_const_conversion! { $($rest)* } )?
+        };
+
+        (($($typenum:ty),*) >= u8 = $const:expr $(; $($rest:tt)*)?) => {
+            $( test_const_conversion! { ($typenum) as (u8, u16, u32, u64, u128) = $const } )*
+            $( test_const_conversion! { ($typenum) as (    i16, i32, i64, i128) = $const } )*
+            $( test_const_conversion! { $($rest)* } )?
+        };
+
+        (($($typenum:ty),*) >= u15 = $const:expr $(; $($rest:tt)*)?) => {
+            $( test_const_conversion! { ($typenum) as (    u16, u32, u64, u128) = $const } )*
+            $( test_const_conversion! { ($typenum) as (    i16, i32, i64, i128) = $const } )*
+            $( test_const_conversion! { $($rest)* } )?
+        };
+
+        (($($typenum:ty),*) >= u16 = $const:expr $(; $($rest:tt)*)?) => {
+            $( test_const_conversion! { ($typenum) as (    u16, u32, u64, u128) = $const } )*
+            $( test_const_conversion! { ($typenum) as (         i32, i64, i128) = $const } )*
+            $( test_const_conversion! { $($rest)* } )?
+        };
+
+        (($($typenum:ty),*) >= u31 = $const:expr $(; $($rest:tt)*)?) => {
+            $( test_const_conversion! { ($typenum) as (         u32, u64, u128) = $const } )*
+            $( test_const_conversion! { ($typenum) as (         i32, i64, i128) = $const } )*
+            $( test_const_conversion! { $($rest)* } )?
+        };
+
+        (($($typenum:ty),*) >= u32 = $const:expr $(; $($rest:tt)*)?) => {
+            $( test_const_conversion! { ($typenum) as (         u32, u64, u128) = $const } )*
+            $( test_const_conversion! { ($typenum) as (              i64, i128) = $const } )*
+            $( test_const_conversion! { $($rest)* } )?
+        };
+
+        (($($typenum:ty),*) >= u63 = $const:expr $(; $($rest:tt)*)?) => {
+            $( test_const_conversion! { ($typenum) as (              u64, u128) = $const } )*
+            $( test_const_conversion! { ($typenum) as (              i64, i128) = $const } )*
+            $( test_const_conversion! { $($rest)* } )?
+        };
+
+        (($($typenum:ty),*) >= u64 = $const:expr $(; $($rest:tt)*)?) => {
+            $( test_const_conversion! { ($typenum) as (              u64, u128) = $const } )*
+            $( test_const_conversion! { ($typenum) as (                   i128) = $const } )*
+            $( test_const_conversion! { $($rest)* } )?
+        };
+
+        (($($typenum:ty),*) >= u127 = $const:expr $(; $($rest:tt)*)?) => {
+            $( test_const_conversion! { ($typenum) as (                   u128) = $const } )*
+            $( test_const_conversion! { ($typenum) as (                   i128) = $const } )*
+            $( test_const_conversion! { $($rest)* } )?
+        };
+
+        (($($typenum:ty),*) >= u128 = $const:expr $(; $($rest:tt)*)?) => {
+            $( test_const_conversion! { ($typenum) as (                   u128) = $const } )*
+            $( test_const_conversion! { ($typenum) as (                       ) = $const } )*
+            $( test_const_conversion! { $($rest)* } )?
+        };
+
+        (($($typenum:ty),*) >= i8 = $const:expr $(; $($rest:tt)*)?) => {
+            $( test_const_conversion! { ($typenum) as (i8, i16, i32, i64, i128) = $const } )*
+            $( test_const_conversion! { $($rest)* } )?
+        };
+
+        (($($typenum:ty),*) >= i16 = $const:expr $(; $($rest:tt)*)?) => {
+            $( test_const_conversion! { ($typenum) as (    i16, i32, i64, i128) = $const } )*
+            $( test_const_conversion! { $($rest)* } )?
+        };
+
+        (($($typenum:ty),*) >= i32 = $const:expr $(; $($rest:tt)*)?) => {
+            $( test_const_conversion! { ($typenum) as (         i32, i64, i128) = $const } )*
+            $( test_const_conversion! { $($rest)* } )?
+        };
+
+        (($($typenum:ty),*) >= i64 = $const:expr $(; $($rest:tt)*)?) => {
+            $( test_const_conversion! { ($typenum) as (              i64, i128) = $const } )*
+            $( test_const_conversion! { $($rest)* } )?
+        };
+
+        (($($typenum:ty),*) >= i128 = $const:expr $(; $($rest:tt)*)?) => {
+            $( test_const_conversion! { ($typenum) as (                   i128) = $const } )*
+            $( test_const_conversion! { $($rest)* } )?
+        };
+
+        // Basic operation
+
+        () => {};
+
+        (($($typenum:ty),*) as () = $const:expr $(; $($rest:tt)*)?) => {
+            $( test_const_conversion! { $($rest)* } )?
+        };
+
+        (($($typenum:ty),*) as ($type:ty) = $const:expr $(; $($rest:tt)*)?) => {
+            $( test_const_conversion! { @impl ($typenum) as ($type) = $const } )*
+            $( test_const_conversion! { $($rest)* } )?
+        };
+
+        (($($typenum:ty),*) as ($type_head:ty, $($type_tail:ty),*) = $const:expr $(; $($rest:tt)*)?) => {
+            $( test_const_conversion! { @impl ($typenum) as ($type_head) = $const } )*
+            test_const_conversion! { ($($typenum),*) as ($($type_tail),*) = $const }
+            $( test_const_conversion! { $($rest)* } )?
+        };
+
+        (@impl ($typenum:ty) as ($type:ty) = $const:expr $(; $($rest:tt)*)?) => {
+            const_assert_eq!(typenum2const!($typenum as $type), $const);
+            $( test_const_conversion!($($rest)*); )?
+        };
+    }
+
+    test_const_conversion! {
+        (B0, False) as (bool, bool) = false;
+
+        (B0, U0, Z0) >= u7 = 0;
+        (B1, U1, P1) >= u7 = 1;
+
+        (U2, P2) >= u7 = 2;
+        (B1, True) as (bool) = true;
+        (U3, P3) >= u7 = 3;
+        (U8, P8) >= u7 = 8;
+        (U127, P127) >= u7 = 127;
+        (U220, P220) >= u8 = 220;
+        (U255, P255) >= u8 = 255;
+        (U1000, P1000) >= u15 = 1000;
+        (U10000, P10000) >= u15 = 10000;
+        (U16384, P16384) >= u15 = 16384;
+        (U32768, P32768) >= u16 = 32768;
+        (U65536, P65536) >= u31 = 65536;
+        (U100000, P100000) >= u31 = 100000;
+        (U1000000000, P1000000000) >= u31 = 1000000000;
+        (U2147483648, P2147483648) >= u32 = 2147483648;
+        (U1000000000000000000, P1000000000000000000) >= u63 = 1000000000000000000;
+        (U1000000000000000000, P1000000000000000000) >= u63 = 1000000000000000000;
+
+        (U9223372036854775808) >= u64 = 9223372036854775808;
+        (U10000000000000000000) >= u64 = 10000000000000000000;
+
+        (N10000) >= i16 = -10000;
+        (N1000000) >= i32 = -1000000;
+        (N1000000000) >= i32 = -1000000000;
+        (N1000000000000) >= i64 = -1000000000000;
+    }
+
+    const_assert_eq!(127, (!(1u8.rotate_right(1)) - 0) as _);
+    const_assert_eq!(126, (!(1u8.rotate_right(1)) - 1) as _);
+    const_assert_eq!(255, (!(0u8.rotate_right(1)) - 0) as _);
+    const_assert_eq!(254, (!(0u8.rotate_right(1)) - 1) as _);
+
+    test_const_conversion! {
+        (op!(pow(U2, U7) - U1))   >= u7   = (!(1u8.rotate_right(1)) - 0) as _;
+        (op!(pow(U2, U7) - U2))   >= u7   = (!(1u8.rotate_right(1)) - 1) as _;
+        (op!(pow(U2, U15) - U1))  >= u15  = (!(1u16.rotate_right(1)) - 0) as _;
+        (op!(pow(U2, U15) - U2))  >= u15  = (!(1u16.rotate_right(1)) - 1) as _;
+        (op!(pow(U2, U31) - U1))  >= u31  = (!(1u32.rotate_right(1)) - 0) as _;
+        (op!(pow(U2, U31) - U2))  >= u31  = (!(1u32.rotate_right(1)) - 1) as _;
+        (op!(pow(U2, U63) - U1))  >= u63  = (!(1u64.rotate_right(1)) - 0) as _;
+        (op!(pow(U2, U63) - U2))  >= u63  = (!(1u64.rotate_right(1)) - 1) as _;
+        (op!(pow(U2, U127) - U1)) >= u127 = (!(1u128.rotate_right(1)) - 0) as _;
+        (op!(pow(U2, U127) - U2)) >= u127 = (!(1u128.rotate_right(1)) - 1) as _;
+
+        (op!(pow(U2, U8) - U1))   >= u8   = (u8::MAX - 0) as _;
+        (op!(pow(U2, U8) - U2))   >= u8   = (u8::MAX - 1) as _;
+        (op!(pow(U2, U16) - U1))  >= u16  = (u16::MAX - 0) as _;
+        (op!(pow(U2, U16) - U2))  >= u16  = (u16::MAX - 1) as _;
+        (op!(pow(U2, U32) - U1))  >= u32  = (u32::MAX - 0) as _;
+        (op!(pow(U2, U32) - U2))  >= u32  = (u32::MAX - 1) as _;
+        (op!(pow(U2, U64) - U1))  >= u64  = (u64::MAX - 0) as _;
+        (op!(pow(U2, U64) - U2))  >= u64  = (u64::MAX - 1) as _;
+        (op!(pow(U2, U128) - U1)) >= u128 = (u128::MAX - 0) as _;
+        (op!(pow(U2, U128) - U2)) >= u128 = (u128::MAX - 1) as _;
+
+        (op!(Z0 - pow(P2, P7) + Z0)) >= i8 = (i8::MIN + 0) as _;
+        (op!(Z0 - pow(P2, P7) + P1)) >= i8 = (i8::MIN + 1) as _;
+        (op!(Z0 - pow(P2, P15) + Z0)) >= i16 = (i16::MIN + 0) as _;
+        (op!(Z0 - pow(P2, P15) + P1)) >= i16 = (i16::MIN + 1) as _;
+        (op!(Z0 - pow(P2, P31) + Z0)) >= i32 = (i32::MIN + 0) as _;
+        (op!(Z0 - pow(P2, P31) + P1)) >= i32 = (i32::MIN + 1) as _;
+        (op!(Z0 - pow(P2, P63) + Z0)) >= i64 = (i64::MIN + 0) as _;
+        (op!(Z0 - pow(P2, P63) + P1)) >= i64 = (i64::MIN + 1) as _;
+        (op!(Z0 - pow(P2, P127) + Z0)) >= i128 = (i128::MIN + 0) as _;
+        (op!(Z0 - pow(P2, P127) + P1)) >= i128 = (i128::MIN + 1) as _;
+    }
+}