Revert "remove applyRaw"

This reverts commit 89ace10b752b18cb4b9498d38344e4b0a716f575.

1 files changed, 15 insertions, 2 deletions

diff --git a/packages/base/src/Internal/Matrix.hs b/packages/base/src/Internal/Matrix.hs
index 12ef05a..5163421 100644
--- a/packages/base/src/Internal/Matrix.hs
+++ b/packages/base/src/Internal/Matrix.hs
@@ -108,6 +108,14 @@ fmat m
     | otherwise  = extractAll ColumnMajor m
 
 
+-- C-Haskell matrix adapters
+{-# INLINE amatr #-}
+amatr :: Storable a => (CInt -> CInt -> Ptr a -> b) -> Matrix a -> b
+amatr f x = inlinePerformIO (unsafeWith (xdat x) (return . f r c))
+  where
+    r  = fi (rows x)
+    c  = fi (cols x)
+
 {-# INLINE amat #-}
 amat :: Storable a => (CInt -> CInt -> CInt -> CInt -> Ptr a -> b) -> Matrix a -> b
 amat f x = inlinePerformIO (unsafeWith (xdat x) (return . f r c sr sc))
@@ -117,11 +125,16 @@ amat f x = inlinePerformIO (unsafeWith (xdat x) (return . f r c sr sc))
     sr = fi (xRow x)
     sc = fi (xCol x)
 
+
 instance Storable t => TransArray (Matrix t)
   where
+    type Elem (Matrix t)       = t
+    type TransRaw (Matrix t) b = CInt -> CInt -> Ptr t -> b
     type Trans (Matrix t) b    = CInt -> CInt -> CInt -> CInt -> Ptr t -> b
     apply = amat
     {-# INLINE apply #-}
+    applyRaw = amatr
+    {-# INLINE applyRaw #-}
 
 infixl 1 #
 a # b = apply a b
@@ -564,7 +577,7 @@ foreign import ccall unsafe "gemm_mod_int64_t" c_gemmML :: Z -> Tgemm Z
 --------------------------------------------------------------------------------
 
 foreign import ccall unsafe "saveMatrix" c_saveMatrix
-    :: CString -> CString -> Double ::> Ok
+    :: CString -> CString -> Double ..> Ok
 
 {- | save a matrix as a 2D ASCII table
 -}
@@ -576,7 +589,7 @@ saveMatrix
 saveMatrix name format m = do
     cname   <- newCString name
     cformat <- newCString format
-    c_saveMatrix cname cformat `apply` m #|"saveMatrix"
+    c_saveMatrix cname cformat `applyRaw` m #|"saveMatrix"
     free cname
     free cformat
     return ()