From 03ef563a681fce281c07d03efc693ecd343cf23d Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Mon, 20 May 2024 17:38:25 +0200
Subject: rfromOrthotope

---
 src/Data/Array/Nested.hs          | 1 +
 src/Data/Array/Nested/Internal.hs | 8 +++++++-
 2 files changed, 8 insertions(+), 1 deletion(-)

(limited to 'src/Data')

diff --git a/src/Data/Array/Nested.hs b/src/Data/Array/Nested.hs
index 2208349..7298918 100644
--- a/src/Data/Array/Nested.hs
+++ b/src/Data/Array/Nested.hs
@@ -16,6 +16,7 @@ module Data.Array.Nested (
   -- ** Conversions
   rasXArrayPrim, rfromXArrayPrim,
   rcastToShaped,
+  rfromOrthotope,
 
   -- * Shaped arrays
   Shaped(Shaped),
diff --git a/src/Data/Array/Nested/Internal.hs b/src/Data/Array/Nested/Internal.hs
index a05ff84..8f18724 100644
--- a/src/Data/Array/Nested/Internal.hs
+++ b/src/Data/Array/Nested/Internal.hs
@@ -28,7 +28,7 @@
 
 {-|
 TODO:
-* Write `rconst :: OR.Array n a -> Ranked n a`
+(empty list)
 
 -}
 
@@ -1403,6 +1403,12 @@ rtoList (Ranked arr)
 rtoList1 :: Elt a => Ranked 1 a -> [a]
 rtoList1 = map runScalar . rtoList
 
+rfromOrthotope :: PrimElt a => SNat n -> S.Array n a -> Ranked n a
+rfromOrthotope sn arr
+  | Refl <- lemRankReplicate sn
+  = let xarr = XArray arr
+    in Ranked (fromPrimitive (M_Primitive (X.shape (ssxFromSNat sn) xarr) xarr))
+
 runScalar :: Elt a => Ranked 0 a -> a
 runScalar arr = rindex arr ZIR
 
-- 
cgit v1.2.3-70-g09d2