From 20173c939486ed6e27b8170e94f666d8ae3df152 Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Thu, 13 Jun 2024 13:09:04 +0200
Subject: Rename *LengthSNat to *Rank

---
 src/Data/Array/Nested/Internal/Ranked.hs | 26 +++++++++++++-------------
 src/Data/Array/Nested/Internal/Shape.hs  | 16 ++++++++--------
 2 files changed, 21 insertions(+), 21 deletions(-)

(limited to 'src/Data/Array/Nested/Internal')

diff --git a/src/Data/Array/Nested/Internal/Ranked.hs b/src/Data/Array/Nested/Internal/Ranked.hs
index 1518791..6a4db8e 100644
--- a/src/Data/Array/Nested/Internal/Ranked.hs
+++ b/src/Data/Array/Nested/Internal/Ranked.hs
@@ -227,7 +227,7 @@ rshape :: forall n a. Elt a => Ranked n a -> IShR n
 rshape (Ranked arr) = shCvtXR' (mshape arr)
 
 rrank :: Elt a => Ranked n a -> SNat n
-rrank = shrLengthSNat . rshape
+rrank = shrRank . rshape
 
 rindex :: Elt a => Ranked n a -> IIxR n -> a
 rindex (Ranked arr) idx = mindex arr (ixCvtRX idx)
@@ -235,14 +235,14 @@ rindex (Ranked arr) idx = mindex arr (ixCvtRX idx)
 rindexPartial :: forall n m a. Elt a => Ranked (n + m) a -> IIxR n -> Ranked m a
 rindexPartial (Ranked arr) idx =
   Ranked (mindexPartial @a @(Replicate n Nothing) @(Replicate m Nothing)
-            (castWith (subst2 (lemReplicatePlusApp (ixrLengthSNat idx) (Proxy @m) (Proxy @Nothing))) arr)
+            (castWith (subst2 (lemReplicatePlusApp (ixrRank idx) (Proxy @m) (Proxy @Nothing))) arr)
             (ixCvtRX idx))
 
 -- | __WARNING__: All values returned from the function must have equal shape.
 -- See the documentation of 'mgenerate' for more details.
 rgenerate :: forall n a. KnownElt a => IShR n -> (IIxR n -> a) -> Ranked n a
 rgenerate sh f
-  | sn@SNat <- shrLengthSNat sh
+  | sn@SNat <- shrRank sh
   , Dict <- lemKnownReplicate sn
   , Refl <- lemRankReplicate sn
   = Ranked (mgenerate (shCvtRX sh) (f . ixCvtXR))
@@ -274,7 +274,7 @@ rsumOuter1 = rfromPrimitive . rsumOuter1P . rtoPrimitive
 
 rtranspose :: forall n a. Elt a => PermR -> Ranked n a -> Ranked n a
 rtranspose perm arr
-  | sn@SNat <- shrLengthSNat (rshape arr)
+  | sn@SNat <- rrank arr
   , Dict <- lemKnownReplicate sn
   , length perm <= fromIntegral (natVal (Proxy @n))
   = rlift sn
@@ -291,7 +291,7 @@ rconcat
 rappend :: forall n a. Elt a
         => Ranked (n + 1) a -> Ranked (n + 1) a -> Ranked (n + 1) a
 rappend arr1 arr2
-  | sn@SNat <- shrLengthSNat (rshape arr1)
+  | sn@SNat <- rrank arr1
   , Dict <- lemKnownReplicate sn
   , Refl <- lemReplicateSucc @(Nothing @Nat) @n
   = coerce (mappend @Nothing @Nothing @(Replicate n Nothing))
@@ -302,7 +302,7 @@ rscalar x = Ranked (mscalar x)
 
 rfromVectorP :: forall n a. Storable a => IShR n -> VS.Vector a -> Ranked n (Primitive a)
 rfromVectorP sh v
-  | Dict <- lemKnownReplicate (shrLengthSNat sh)
+  | Dict <- lemKnownReplicate (shrRank sh)
   = Ranked (mfromVectorP (shCvtRX sh) v)
 
 rfromVector :: forall n a. PrimElt a => IShR n -> VS.Vector a -> Ranked n a
@@ -352,7 +352,7 @@ rfromOrthotope sn arr
 
 rtoOrthotope :: PrimElt a => Ranked n a -> S.Array n a
 rtoOrthotope (rtoPrimitive -> Ranked (M_Primitive sh (XArray arr)))
-  | Refl <- lemRankReplicate (shrLengthSNat $ shCvtXR' sh)
+  | Refl <- lemRankReplicate (shrRank $ shCvtXR' sh)
   = arr
 
 runScalar :: Elt a => Ranked 0 a -> a
@@ -412,12 +412,12 @@ rrerank sn sh2 f (rtoPrimitive -> arr) =
 rreplicate :: forall n m a. Elt a
            => IShR n -> Ranked m a -> Ranked (n + m) a
 rreplicate sh (Ranked arr)
-  | Refl <- lemReplicatePlusApp (shrLengthSNat sh) (Proxy @m) (Proxy @(Nothing @Nat))
+  | Refl <- lemReplicatePlusApp (shrRank sh) (Proxy @m) (Proxy @(Nothing @Nat))
   = Ranked (mreplicate (shCvtRX sh) arr)
 
 rreplicateScalP :: forall n a. Storable a => IShR n -> a -> Ranked n (Primitive a)
 rreplicateScalP sh x
-  | Dict <- lemKnownReplicate (shrLengthSNat sh)
+  | Dict <- lemKnownReplicate (shrRank sh)
   = Ranked (mreplicateScalP (shCvtRX sh) x)
 
 rreplicateScal :: forall n a. PrimElt a
@@ -427,13 +427,13 @@ rreplicateScal sh x = rfromPrimitive (rreplicateScalP sh x)
 rslice :: forall n a. Elt a => Int -> Int -> Ranked (n + 1) a -> Ranked (n + 1) a
 rslice i n arr
   | Refl <- lemReplicateSucc @(Nothing @Nat) @n
-  = rlift (shrLengthSNat (rshape arr))
+  = rlift (rrank arr)
           (\_ -> X.sliceU i n)
           arr
 
 rrev1 :: forall n a. Elt a => Ranked (n + 1) a -> Ranked (n + 1) a
 rrev1 arr =
-  rlift (shrLengthSNat (rshape arr))
+  rlift (rrank arr)
         (\(_ :: StaticShX sh') ->
           case lemReplicateSucc @(Nothing @Nat) @n of
             Refl -> X.rev1 @Nothing @(Replicate n Nothing ++ sh'))
@@ -442,8 +442,8 @@ rrev1 arr =
 rreshape :: forall n n' a. Elt a
          => IShR n' -> Ranked n a -> Ranked n' a
 rreshape sh' rarr@(Ranked arr)
-  | Dict <- lemKnownReplicate (shrLengthSNat (rshape rarr))
-  , Dict <- lemKnownReplicate (shrLengthSNat sh')
+  | Dict <- lemKnownReplicate (rrank rarr)
+  , Dict <- lemKnownReplicate (shrRank sh')
   = Ranked (mreshape (shCvtRX sh') arr)
 
 rflatten :: Elt a => Ranked n a -> Ranked 1 a
diff --git a/src/Data/Array/Nested/Internal/Shape.hs b/src/Data/Array/Nested/Internal/Shape.hs
index 7d95f61..4fa4284 100644
--- a/src/Data/Array/Nested/Internal/Shape.hs
+++ b/src/Data/Array/Nested/Internal/Shape.hs
@@ -91,14 +91,14 @@ listrIndex SZ (x ::: _) = x
 listrIndex (SS i) (_ ::: xs) | Refl <- lemLeqSuccSucc (Proxy @k) (Proxy @n) = listrIndex i xs
 listrIndex _ ZR = error "k + 1 <= 0"
 
-listrLengthSNat :: ListR n i -> SNat n
-listrLengthSNat ZR = SNat
-listrLengthSNat (_ ::: (sh :: ListR n i)) = snatSucc (listrLengthSNat sh)
+listrRank :: ListR n i -> SNat n
+listrRank ZR = SNat
+listrRank (_ ::: (sh :: ListR n i)) = snatSucc (listrRank sh)
 
 listrPermutePrefix :: forall i n. [Int] -> ListR n i -> ListR n i
 listrPermutePrefix = \perm sh ->
   listrFromList perm $ \sperm ->
-    case (listrLengthSNat sperm, listrLengthSNat sh) of
+    case (listrRank sperm, listrRank sh) of
       (permlen@SNat, shlen@SNat) -> case cmpNat permlen shlen of
         LTI -> let (pre, post) = listrSplitAt permlen sh in listrAppend (applyPermRFull permlen sperm pre) post
         EQI -> let (pre, post) = listrSplitAt permlen sh in listrAppend (applyPermRFull permlen sperm pre) post
@@ -168,8 +168,8 @@ ixrTail (IxR list) = IxR (listrTail list)
 ixrAppend :: forall n m i. IxR n i -> IxR m i -> IxR (n + m) i
 ixrAppend = coerce (listrAppend @_ @i)
 
-ixrLengthSNat :: IxR n i -> SNat n
-ixrLengthSNat (IxR sh) = listrLengthSNat sh
+ixrRank :: IxR n i -> SNat n
+ixrRank (IxR sh) = listrRank sh
 
 ixrPermutePrefix :: forall n i. [Int] -> IxR n i -> IxR n i
 ixrPermutePrefix = coerce (listrPermutePrefix @i)
@@ -237,8 +237,8 @@ shrTail (ShR list) = ShR (listrTail list)
 shrAppend :: forall n m i. ShR n i -> ShR m i -> ShR (n + m) i
 shrAppend = coerce (listrAppend @_ @i)
 
-shrLengthSNat :: ShR n i -> SNat n
-shrLengthSNat (ShR sh) = listrLengthSNat sh
+shrRank :: ShR n i -> SNat n
+shrRank (ShR sh) = listrRank sh
 
 shrPermutePrefix :: forall n i. [Int] -> ShR n i -> ShR n i
 shrPermutePrefix = coerce (listrPermutePrefix @i)
-- 
cgit v1.2.3-70-g09d2