Add subscript in Matrix to return the row.

tombao2007 · Feb 26, 2016 · aff7fd3 · aff7fd3
1 parent 27da496
commit aff7fd3
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/Source/Matrix.swift b/Source/Matrix.swift
@@ -59,6 +59,22 @@ public struct Matrix<T where T: FloatingPointType, T: FloatLiteralConvertible> {
             grid[(row * columns) + column] = newValue
         }
     }
+
+    public subscript(row: Int) -> ArraySlice<Element> {
+        get {
+            assert(row < rows)
+            let startIndex = row * columns
+            let endIndex = row * columns + columns
+            return grid[startIndex..<endIndex]
+        }
+
+        set {
+            assert(row > rows)
+            let startIndex = row * columns
+            let endIndex = row * columns + columns
+            grid.replaceRange(startIndex..<endIndex, with: newValue)
+        }
+    }
 
     private func indexIsValidForRow(row: Int, column: Int) -> Bool {
         return row >= 0 && row < rows && column >= 0 && column < columns