initial support for static dimension checking

3 files changed, 95 insertions, 5 deletions

diff --git a/packages/base/src/Numeric/LinearAlgebra/Data.hs b/packages/base/src/Numeric/LinearAlgebra/Data.hs
index 89bebbe..3128a24 100644
--- a/packages/base/src/Numeric/LinearAlgebra/Data.hs
+++ b/packages/base/src/Numeric/LinearAlgebra/Data.hs
@@ -48,16 +48,20 @@ module Numeric.LinearAlgebra.Data(
     -- * Find elements
     find, maxIndex, minIndex, maxElement, minElement, atIndex,
 
+    -- * Sparse
+    SMatrix, AssocMatrix, mkCSR, toDense,
+    mkDiag,
+
+    -- * Static dimensions
+    
+    Static, ddata, R, vect0, sScalar, vect2, vect3, (&),
+
     -- * IO
     disp,
     loadMatrix, saveMatrix,
     latexFormat,
     dispf, disps, dispcf, format,
 
-    -- * Sparse
-    SMatrix, AssocMatrix, mkCSR, toDense,
-    mkDiag,
-
 -- * Conversion
     Convert(..),
     
@@ -75,7 +79,9 @@ module Numeric.LinearAlgebra.Data(
 import Data.Packed.Vector
 import Data.Packed.Matrix
 import Data.Packed.Numeric
-import Numeric.LinearAlgebra.Util
+import Numeric.LinearAlgebra.Util hiding ((&))
 import Data.Complex
 import Numeric.Sparse
+import Numeric.LinearAlgebra.Util.Static
+
 
diff --git a/packages/base/src/Numeric/LinearAlgebra/Util.hs b/packages/base/src/Numeric/LinearAlgebra/Util.hs
index a7d6946..a319785 100644
--- a/packages/base/src/Numeric/LinearAlgebra/Util.hs
+++ b/packages/base/src/Numeric/LinearAlgebra/Util.hs
@@ -28,6 +28,7 @@ module Numeric.LinearAlgebra.Util(
     (&), (¦), (——), (#),
     (?), (¿),
     Indexable(..), size,
+    Numeric,
     rand, randn,
     cross,
     norm,
@@ -101,6 +102,19 @@ mat
   -> Matrix ℝ
 mat c = reshape c . fromList
 
+
+
+class ( Container Vector t
+      , Container Matrix t
+      , Konst t Int Vector
+      , Konst t (Int,Int) Matrix
+      ) => Numeric t
+
+instance Numeric Double
+instance Numeric (Complex Double)
+
+
+
 {- | print a real matrix with given number of digits after the decimal point
 
 >>> disp 5 $ ident 2 / 3
diff --git a/packages/base/src/Numeric/LinearAlgebra/Util/Static.hs b/packages/base/src/Numeric/LinearAlgebra/Util/Static.hs
new file mode 100644
index 0000000..a3f8eb0
--- /dev/null
+++ b/packages/base/src/Numeric/LinearAlgebra/Util/Static.hs
@@ -0,0 +1,70 @@
+{-# LANGUAGE DataKinds #-}
+{-# LANGUAGE KindSignatures #-}
+{-# LANGUAGE GeneralizedNewtypeDeriving #-}
+{-# LANGUAGE MultiParamTypeClasses #-}
+{-# LANGUAGE FlexibleContexts #-}
+{-# LANGUAGE ScopedTypeVariables #-}
+{-# LANGUAGE EmptyDataDecls #-}
+{-# LANGUAGE Rank2Types #-}
+{-# LANGUAGE FlexibleInstances #-}
+{-# LANGUAGE TypeOperators #-}
+
+module Numeric.LinearAlgebra.Util.Static(
+    Static (ddata),
+    R,
+    vect0, sScalar, vect2, vect3, (&)
+) where
+
+
+import GHC.TypeLits
+import Data.Packed.Numeric
+import Numeric.Vector()
+import Numeric.LinearAlgebra.Util(Numeric,ℝ)
+
+lift1F :: (Vector t -> Vector t) -> Static n (Vector t) -> Static n (Vector t)
+lift1F f (Static v) = Static (f v)
+
+lift2F :: (Vector t -> Vector t -> Vector t) -> Static n (Vector t) -> Static n (Vector t) -> Static n (Vector t)
+lift2F f (Static u) (Static v) = Static (f u v)
+
+newtype Static (n :: Nat) t = Static { ddata :: t } deriving Show
+
+type R n = Static n (Vector ℝ)
+
+
+infixl 4 &
+(&) :: R n -> ℝ -> R (n+1)
+Static v & x = Static (vjoin [v, scalar x])
+
+vect0 :: R 0
+vect0 = Static (fromList[])
+
+sScalar :: ℝ -> R 1
+sScalar = Static . scalar
+
+
+vect2 :: ℝ -> ℝ -> R 2
+vect2 x1 x2 = Static (fromList [x1,x2])
+
+vect3 :: ℝ -> ℝ -> ℝ -> R 3
+vect3 x1 x2 x3 = Static (fromList [x1,x2,x3])
+
+
+
+
+
+
+instance forall n t . (KnownNat n, Num (Vector t), Numeric t )=> Num (Static n (Vector t))
+  where
+    (+) = lift2F add
+    (*) = lift2F mul
+    (-) = lift2F sub
+    abs = lift1F abs
+    signum = lift1F signum
+    negate = lift1F (scale (-1))
+    fromInteger x = Static (konst (fromInteger x) d)
+      where
+        d = fromIntegral . natVal $ (undefined :: Proxy n)
+
+data Proxy :: Nat -> *
+