From be6012d6589e4605fd477de893c43f18b6c2b6d6 Mon Sep 17 00:00:00 2001
From: Dzuchun <Infernodambik@gmail.com>
Date: Sun, 17 Nov 2024 06:15:48 +0200
Subject: [PATCH] Added [`typenum`]-facing side to [`Conv`] trait

---
 src/lib.rs | 23 +++++++++++++++++++----
 1 file changed, 19 insertions(+), 4 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index 4d3b454..20b0fe0 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -35,7 +35,10 @@ pub trait Conv {
     /// [`nalgebra`]-faced type (matrix dimension)
     type Nalg: Dim;
 
-    /// [`typenum`]/[`generic_array`]-faced type (generic array length)
+    /// [`typenum`]-faced type (unsigned int)
+    type TNum: Unsigned;
+
+    /// [`generic_array`]-faced type (generic array length)
     type ArrLen: ArrayLength;
 
     /// Constructor method used in [`nalgebra`] implementations
@@ -44,7 +47,8 @@ pub trait Conv {
 
 impl<const N: usize> Conv for nalgebra::Const<N>
 where
-    typenum::Const<N>: generic_array::IntoArrayLength,
+    typenum::Const<N>: generic_array::IntoArrayLength + typenum::ToUInt,
+    <typenum::Const<N> as typenum::ToUInt>::Output: Unsigned,
 {
     const SEAL: Seal = Seal(());
 
@@ -52,6 +56,8 @@ where
 
     type Nalg = nalgebra::Const<N>;
 
+    type TNum = typenum::U<N>;
+
     type ArrLen = <typenum::Const<N> as generic_array::IntoArrayLength>::ArrayLength;
 
     fn new_nalg() -> Self::Nalg {
@@ -61,7 +67,8 @@ where
 
 impl<const N: usize> Conv for typenum::Const<N>
 where
-    typenum::Const<N>: generic_array::IntoArrayLength,
+    typenum::Const<N>: generic_array::IntoArrayLength + typenum::ToUInt,
+    <typenum::Const<N> as typenum::ToUInt>::Output: Unsigned,
 {
     const SEAL: Seal = Seal(());
 
@@ -69,6 +76,8 @@ where
 
     type Nalg = nalgebra::Const<N>;
 
+    type TNum = typenum::U<N>;
+
     type ArrLen = <typenum::Const<N> as generic_array::IntoArrayLength>::ArrayLength;
 
     fn new_nalg() -> Self::Nalg {
@@ -83,6 +92,8 @@ impl Conv for typenum::UTerm {
 
     type Nalg = nalgebra::Const<0>;
 
+    type TNum = typenum::consts::U0;
+
     type ArrLen = Self;
 
     fn new_nalg() -> Self::Nalg {
@@ -103,6 +114,8 @@ where
 
     type Nalg = <U::Nalg as nalgebra::dimension::DimMul<nalgebra::U2>>::Output;
 
+    type TNum = typenum::operator_aliases::Prod<U::ArrLen, typenum::U2>;
+
     type ArrLen = typenum::operator_aliases::Prod<U::ArrLen, typenum::U2>;
 
     fn new_nalg() -> Self::Nalg {
@@ -124,6 +137,8 @@ where
     type Nalg =
         <<typenum::UInt<U, B0> as Conv>::Nalg as nalgebra::dimension::DimAdd<nalgebra::U1>>::Output;
 
+    type TNum = typenum::operator_aliases::Sum<<typenum::UInt<U, B0> as Conv>::ArrLen, typenum::U1>;
+
     type ArrLen =
         typenum::operator_aliases::Sum<<typenum::UInt<U, B0> as Conv>::ArrLen, typenum::U1>;
 
@@ -202,7 +217,7 @@ use nalgebra::{
     allocator::Allocator, ArrayStorage, Dim, IsContiguous, Matrix, Owned, RawStorage,
     RawStorageMut, Scalar, Storage,
 };
-use typenum::{B0, B1};
+use typenum::{Unsigned, B0, B1};
 
 /// A stack-allocated storage, of [`typenum`]-backed col-major two dimensional array
 ///