fromList

1 files changed, 13 insertions, 0 deletions

diff --git a/src/Data/Array/Mixed.hs b/src/Data/Array/Mixed.hs
index 17b0ab4..246f8fc 100644
--- a/src/Data/Array/Mixed.hs
+++ b/src/Data/Array/Mixed.hs
@@ -13,6 +13,8 @@
 module Data.Array.Mixed where
 
 import qualified Data.Array.RankedS as S
+import qualified Data.Array.Ranked as ORB
+import Data.Coerce
 import Data.Kind
 import Data.Proxy
 import Data.Type.Equality
@@ -347,3 +349,14 @@ sumOuter :: forall sh sh' a. (Storable a, Num a)
 sumOuter ssh ssh'
   | Refl <- lemAppNil @sh
   = sumInner ssh' ssh . transpose2 ssh ssh'
+
+fromList :: forall n sh a. Storable a
+         => StaticShapeX (n : sh) -> [XArray sh a] -> XArray (n : sh) a
+fromList ssh l
+  | Dict <- lemKnownINatRankSSX ssh
+  , Dict <- knownNatFromINat (Proxy @(Rank (n : sh)))
+  = case ssh of
+      m@GHC_SNat :$@ _ | natVal m /= fromIntegral (length l) ->
+        error $ "Data.Array.Mixed.fromList: length of list (" ++ show (length l) ++ ")" ++
+                "does not match the type (" ++ show (natVal m) ++ ")" 
+      _ -> XArray (S.ravel (ORB.fromList [length l] (coerce @[XArray sh a] @[S.Array (FromINat (Rank sh)) a] l)))