]> gitweb.michael.orlitzky.com - numerical-analysis.git/commitdiff
Remove the "row" function which returned a vector instead of a matrix.
authorMichael Orlitzky <michael@orlitzky.com>
Sat, 8 Feb 2014 00:10:02 +0000 (19:10 -0500)
committerMichael Orlitzky <michael@orlitzky.com>
Sat, 8 Feb 2014 00:10:02 +0000 (19:10 -0500)
src/Integration/Gaussian.hs
src/Linear/Matrix.hs

index 38214a0a5be7468b0b061e484ca0bb16d01556ff..be92e5c8bdfd85e232cee5db3b1d57b7ce7675b1 100644 (file)
@@ -29,7 +29,7 @@ import Linear.Matrix (
   fromList,
   identity_matrix,
   map2,
-  row',
+  row,
   transpose )
 import Linear.QR ( eigenvectors_symmetric )
 import Normed ( Normed(..) )
@@ -119,7 +119,7 @@ nodes_and_weights iterations =
     nodes = shifted_nodes + ones -- unshift the nodes
 
     -- Get the first component of each column.
-    first_components = row' vecs 0
+    first_components = row vecs 0
 
     -- Square it and multiply by 2; see the Golub-Welsch paper for
     -- this magic.
index 6d13a693b71e37631e8600f87652f6c9eb73b883..a5fbb8403b0832becf04b0d48a68be4f0725345c 100644 (file)
@@ -152,49 +152,37 @@ fromList vs = Mat (V.fromList $ map V.fromList vs)
 
 -- | Unsafe indexing.
 (!!!) :: (Arity m, Arity n) => Mat m n a -> (Int, Int) -> a
-(!!!) m (i, j) = (row m i) ! j
+(!!!) (Mat rows) (i, j) = (rows ! i) ! j
+
 
 -- | Safe indexing.
-(!!?) :: Mat m n a -> (Int, Int) -> Maybe a
-(!!?) m@(Mat rows) (i, j)
+(!!?) :: (Arity m, Arity n) => Mat m n a -> (Int, Int) -> Maybe a
+(!!?) matrix (i, j)
   | i < 0 || j < 0 = Nothing
-  | i > V.length rows = Nothing
-  | otherwise = if j > V.length (row m j)
-                then Nothing
-                else Just $ (row m j) ! j
+  | i > (nrows matrix) - 1 = Nothing
+  | j > (ncols matrix) - 1 = Nothing
+  | otherwise = Just $ matrix !!! (i,j)
 
 
 -- | The number of rows in the matrix.
 nrows :: forall m n a. (Arity m) => Mat m n a -> Int
 nrows _ = arity (undefined :: m)
 
+
 -- | The number of columns in the first row of the
 --   matrix. Implementation stolen from Data.Vector.Fixed.length.
 ncols :: forall m n a. (Arity n) => Mat m n a -> Int
 ncols _ = arity (undefined :: n)
 
 
--- | Return the @i@th row of @m@. Unsafe.
-row :: Mat m n a -> Int -> (Vec n a)
-row (Mat rows) i = rows ! i
-
-
 -- | Return the @i@th row of @m@ as a matrix. Unsafe.
-row' :: (Arity m, Arity n) => Mat m n a -> Int -> Row n a
-row' m i =
+row :: (Arity m, Arity n) => Mat m n a -> Int -> Row n a
+row m i =
   construct lambda
   where
     lambda _ j = m !!! (i, j)
 
 
--- | Return the @j@th column of @m@. Unsafe.
---column :: Mat m n a -> Int -> (Vec m a)
---column (Mat rows) j =
---  V.map (element j) rows
---  where
---    element = flip (!)
-
-
 -- | Return the @j@th column of @m@ as a matrix. Unsafe.
 column :: (Arity m, Arity n) => Mat m n a -> Int -> Col m a
 column m j =
@@ -270,6 +258,7 @@ identity_matrix :: (Arity m, Ring.C a) => Mat m m a
 identity_matrix =
   construct (\i j -> if i == j then (fromInteger 1) else (fromInteger 0))
 
+
 -- | Given a positive-definite matrix @m@, computes the
 --   upper-triangular matrix @r@ with (transpose r)*r == m and all
 --   values on the diagonal of @r@ positive.