Get rid of most ListX operations

1 files changed, 24 insertions, 57 deletions

diff --git a/src/Data/Array/Nested/Mixed/Shape.hs b/src/Data/Array/Nested/Mixed/Shape.hs
index 9869d03..a01e0f3 100644
--- a/src/Data/Array/Nested/Mixed/Shape.hs
+++ b/src/Data/Array/Nested/Mixed/Shape.hs
@@ -50,43 +50,6 @@ import Data.Array.Nested.Mixed.ListX
 import Data.Array.Nested.Types
 
 
--- * Mixed lists
-
-{-# INLINE listxFromList #-}
-listxFromList :: StaticShX sh -> [i] -> ListX sh i
-listxFromList sh l = assert (ssxLength sh == length l) $ IsList.fromList l
-
-listxRank :: ListX sh i -> SNat (Rank sh)
-listxRank ZX = SNat
-listxRank (_ ::% l) | SNat <- listxRank l = SNat
-
-listxHead :: ListX (mn ': sh) i -> i
-listxHead (i ::% _) = i
-
-listxTail :: ListX (n : sh) i -> ListX sh i
-listxTail (_ ::% sh) = sh
-
-listxAppend :: forall sh sh' i. ListX sh i -> ListX sh' i -> ListX (sh ++ sh') i
-listxAppend = lazilyConcat (++)
-
-listxDrop :: forall i j sh sh'. ListX sh j -> ListX (sh ++ sh') i -> ListX sh' i
-listxDrop ZX long = long
-listxDrop (_ ::% short) long = case long of _ ::% long' -> listxDrop short long'
-
-listxInit :: forall i n sh. ListX (n : sh) i -> ListX (Init (n : sh)) i
-listxInit (i ::% sh@(_ ::% _)) = i ::% listxInit sh
-listxInit (_ ::% ZX) = ZX
-
-listxLast :: forall i n sh. ListX (n : sh) i -> i
-listxLast (_ ::% sh@(_ ::% _)) = listxLast sh
-listxLast (x ::% ZX) = x
-
-{-# INLINE listxZipWith #-}
-listxZipWith :: (i -> j -> k) -> ListX sh i -> ListX sh j -> ListX sh k
-listxZipWith _ ZX ZX = ZX
-listxZipWith f (i ::% is) (j ::% js) = f i j ::% listxZipWith f is js
-
-
 -- * Mixed indices
 
 -- | An index into a mixed-typed array.
@@ -119,8 +82,13 @@ instance Show i => Show (IxX sh i) where
   showsPrec _ (IxX l) = listxShow shows l
 #endif
 
+{-# INLINE ixxFromList #-}
+ixxFromList :: StaticShX sh -> [i] -> IxX sh i
+ixxFromList sh l = assert (ssxLength sh == length l) $ IsList.fromList l
+
 ixxRank :: IxX sh i -> SNat (Rank sh)
-ixxRank (IxX l) = listxRank l
+ixxRank ZIX = SNat
+ixxRank (_ :.% l) | SNat <- ixxRank l = SNat
 
 ixxZero :: StaticShX sh -> IIxX sh
 ixxZero ZKX = ZIX
@@ -130,32 +98,26 @@ ixxZero' :: IShX sh -> IIxX sh
 ixxZero' ZSX = ZIX
 ixxZero' (_ :$% sh) = 0 :.% ixxZero' sh
 
-{-# INLINEABLE ixxFromList #-}
-ixxFromList :: forall sh i. StaticShX sh -> [i] -> IxX sh i
-ixxFromList = coerce (listxFromList @_ @i)
-
-ixxHead :: IxX (n : sh) i -> i
-ixxHead (IxX list) = listxHead list
+ixxHead :: IxX (mn ': sh) i -> i
+ixxHead (i :.% _) = i
 
 ixxTail :: IxX (n : sh) i -> IxX sh i
-ixxTail (IxX list) = IxX (listxTail list)
+ixxTail (_ :.% sh) = sh
 
 ixxAppend :: forall sh sh' i. IxX sh i -> IxX sh' i -> IxX (sh ++ sh') i
-ixxAppend = coerce (listxAppend @_ @_ @i)
-
-ixxDrop :: forall sh sh' i. IxX sh i -> IxX (sh ++ sh') i -> IxX sh' i
-ixxDrop = coerce (listxDrop @i @i)
+ixxAppend (IxX l1) (IxX l2) = IxX $ lazilyConcat (++) l1 l2
 
-ixxInit :: forall n sh i. IxX (n : sh) i -> IxX (Init (n : sh)) i
-ixxInit = coerce (listxInit @i)
+ixxDrop :: forall i j sh sh'. IxX sh j -> IxX (sh ++ sh') i -> IxX sh' i
+ixxDrop ZIX long = long
+ixxDrop (_ :.% short) long = case long of _ :.% long' -> ixxDrop short long'
 
-ixxLast :: forall n sh i. IxX (n : sh) i -> i
-ixxLast = coerce (listxLast @i)
+ixxInit :: forall i n sh. IxX (n : sh) i -> IxX (Init (n : sh)) i
+ixxInit (i :.% sh@(_ :.% _)) = i :.% ixxInit sh
+ixxInit (_ :.% ZIX) = ZIX
 
-ixxCast :: StaticShX sh' -> IxX sh i -> IxX sh' i
-ixxCast ZKX ZIX = ZIX
-ixxCast (_ :!% sh) (i :.% idx) = i :.% ixxCast sh idx
-ixxCast _ _ = error "ixxCast: ranks don't match"
+ixxLast :: forall i n sh. IxX (n : sh) i -> i
+ixxLast (_ :.% sh@(_ :.% _)) = ixxLast sh
+ixxLast (x :.% ZIX) = x
 
 ixxZip :: IxX sh i -> IxX sh j -> IxX sh (i, j)
 ixxZip ZIX ZIX = ZIX
@@ -166,6 +128,11 @@ ixxZipWith :: (i -> j -> k) -> IxX sh i -> IxX sh j -> IxX sh k
 ixxZipWith _ ZIX ZIX = ZIX
 ixxZipWith f (i :.% is) (j :.% js) = f i j :.% ixxZipWith f is js
 
+ixxCast :: StaticShX sh' -> IxX sh i -> IxX sh' i
+ixxCast ZKX ZIX = ZIX
+ixxCast (_ :!% sh) (i :.% idx) = i :.% ixxCast sh idx
+ixxCast _ _ = error "ixxCast: ranks don't match"
+
 -- | Given a multidimensional index, get the corresponding linear
 -- index into the buffer.
 {-# INLINEABLE ixxToLinear #-}