factor out function for the gauss jordan vector in original order

arnemertz · Jul 10, 2016 · 76288c9 · 76288c9
1 parent a8248d5
commit 76288c9
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/2016-07-refactoring-session-02/main.cpp b/2016-07-refactoring-session-02/main.cpp
@@ -72,6 +72,14 @@ struct GaussJordanMatrix {
     }
     y[rowIndex] /= diagonalElement;
   }
+
+  Vector getVectorInOriginalOrder() {
+    Vector v = y;
+    for (int i=0;i<rowCount();++i) {
+      std::swap(v[i], v[rowIndices[i]]);
+    }
+    return v;
+  }
 };
 
 // Solve y=m*x for x
@@ -97,10 +105,7 @@ Vector gaussJordanElimination(Matrix m, Vector y) {
       }
     }
   }
-  for (int i=0;i<rowCount;++i) {
-    std::swap(gaussJordan.y[i], gaussJordan.y[gaussJordan.rowIndices[i]]);
-  }
-  return gaussJordan.y;
+  return gaussJordan.getVectorInOriginalOrder();
 }