From aeb62e1d59b4cdc426f6ca0ef8df6a0737bb0f86 Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Tue, 18 Nov 2025 11:39:39 +0100
Subject: Add shsEnum and shrEnum

---
 src/Data/Array/Nested/Shaped/Shape.hs | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'src/Data/Array/Nested/Shaped/Shape.hs')

diff --git a/src/Data/Array/Nested/Shaped/Shape.hs b/src/Data/Array/Nested/Shaped/Shape.hs
index 10e1a9d..6eaec10 100644
--- a/src/Data/Array/Nested/Shaped/Shape.hs
+++ b/src/Data/Array/Nested/Shaped/Shape.hs
@@ -389,6 +389,13 @@ shsFromListS (_ ::$ l) = SNat :$$ shsFromListS l
 shsFromIxS :: IxS sh i -> ShS sh
 shsFromIxS (IxS l) = shsFromListS l
 
+shsEnum :: ShS sh -> [IIxS sh]
+shsEnum = \sh -> go sh id []
+  where
+    go :: ShS sh -> (IIxS sh -> a) -> [a] -> [a]
+    go ZSS f = (f ZIS :)
+    go (n :$$ sh) f = foldr (.) id [go sh (f . (i :.$)) | i <- [0 .. fromSNat' n - 1]]
+
 
 -- | Untyped: length is checked at runtime.
 instance KnownShS sh => IsList (ListS sh (Const i)) where
-- 
cgit v1.2.3-70-g09d2