module BLAS.Matrix.Immutable (
IMatrix(..)
) where
import BLAS.Access
import BLAS.Elem ( BLAS3 )
import BLAS.Internal ( checkMatVecMult, checkMatMatMult )
import BLAS.Matrix.ReadOnly
import Data.Vector.Dense
import Data.Matrix.Dense.Internal
import System.IO.Unsafe ( unsafePerformIO )
infixr 7 <*>, <**>
class (RMatrix a e) => IMatrix a e where
(<*>) :: a (m,n) e -> Vector n e -> Vector m e
(<*>) a x =
checkMatVecMult (shape a) (dim x) $ unsafeApply a x
(<**>) :: a (m,k) e -> Matrix (k,n) e -> Matrix (m,n) e
(<**>) a b =
checkMatMatMult (numRows a, numCols a)
(numRows b, numCols b) $ unsafeApplyMat a b
sapply :: e -> a (m,n) e -> Vector n e -> Vector m e
sapply k a x =
checkMatVecMult (numRows a, numCols a) (dim x) $ unsafeSApply k a x
sapplyMat :: e -> a (m,k) e -> Matrix (k,n) e -> Matrix (m,n) e
sapplyMat k a b =
checkMatMatMult (numRows a, numCols a)
(numRows b, numCols b) $ unsafeSApplyMat k a b
unsafeApply :: a (m,n) e -> Vector n e -> Vector m e
unsafeApply = unsafeSApply 1
unsafeApplyMat :: a (m,k) e -> Matrix (k,n) e -> Matrix (m,n) e
unsafeApplyMat = unsafeSApplyMat 1
unsafeSApply :: e -> a (m,n) e -> Vector n e -> Vector m e
unsafeSApply alpha a x = unsafePerformIO $ unsafeGetSApply alpha a x
unsafeSApplyMat :: e -> a (m,k) e -> Matrix (k,n) e -> Matrix (m,n) e
unsafeSApplyMat alpha a b = unsafePerformIO $ unsafeGetSApplyMat alpha a b
instance (BLAS3 e) => IMatrix (DMatrix Imm) e where