Fix asserts in Equation methods gUx() & gUt()

andrei-punko · Oct 24, 2024 · b812034 · b812034
1 parent f56db3a
commit b812034
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/main/java/by/andd3dfx/math/pde/equation/Equation.java b/src/main/java/by/andd3dfx/math/pde/equation/Equation.java
@@ -171,9 +171,10 @@ public void sUx(String fileName, double[] x) {
      * @return U(x) slice
      */
     protected Matrix gUt(int it) {
-        int N = solution.getN();
-        assert (0 <= it && it < N);
+        int M = solution.getM();
+        assert (0 <= it && it < M);
 
+        int N = solution.getN();
         var matrix = new Matrix(2, N);
         for (int i = 0; i < N; i++) {
             matrix.setX(i, area.x().x(i));
@@ -199,9 +200,10 @@ protected Matrix gUt(double t) {
      * @return U(t) slice
      */
     protected Matrix gUx(int ix) {
-        int M = solution.getM();
-        assert (0 <= ix && ix < M);
+        int N = solution.getN();
+        assert (0 <= ix && ix < N);
 
+        int M = solution.getM();
         var matrix = new Matrix(2, M);
         for (int i = 0; i <= M; i++) {
             matrix.setX(i, area.t().x(i));