fixed chem and vib source residual computations with axisymmetric

Signed-off-by: jtneedels <jneedels@stanford.edu>

Author: jtneedels

SU2_CFD/src/solvers/CNEMOEulerSolver.cpp

@@ -1146,7 +1146,7 @@ void CNEMOEulerSolver::Source_Residual(CGeometry *geometry, CSolver **solver_con
     numerics->SetVolume(geometry->nodes->GetVolume(iPoint));
     numerics->SetCoord(geometry->nodes->GetCoord(iPoint),
                        geometry->nodes->GetCoord(iPoint) );
-
+  }
     /*--- Compute axisymmetric source terms (if needed) ---*/
     if (config->GetAxisymmetric()) {
 
@@ -1249,8 +1249,36 @@ void CNEMOEulerSolver::Source_Residual(CGeometry *geometry, CSolver **solver_con
 
     if(!monoatomic){
       if(!frozen){
-        /*--- Compute the non-equilibrium chemistry ---*/
-        auto residual = numerics->ComputeChemistry(config);
+        for (iPoint = 0; iPoint < nPointDomain; iPoint++) {
+          /*--- Compute the non-equilibrium chemistry ---*/
+          auto residual = numerics->ComputeChemistry(config);
+
+          /*--- Check for errors before applying source to the linear system ---*/
+          err = false;
+          for (iVar = 0; iVar < nVar; iVar++)
+            if (residual[iVar] != residual[iVar]) err = true;
+          //if (implicit)
+          //  for (iVar = 0; iVar < nVar; iVar++)
+          //    for (jVar = 0; jVar < nVar; jVar++)
+          //      if (Jacobian_i[iVar][jVar] != Jacobian_i[iVar][jVar]) err = true;
+
+          /*--- Apply the chemical sources to the linear system ---*/
+          if (!err) {
+            LinSysRes.SubtractBlock(iPoint, residual);
+            //if (implicit)
+            //  Jacobian.SubtractBlock(iPoint, iPoint, Jacobian_i);
+          } else
+            eChm_local++;
+        }
+      }
+    }
+
+    /*--- Compute vibrational energy relaxation ---*/
+    /// NOTE: Jacobians don't account for relaxation time derivatives
+
+    if (!monoatomic){
+      for (iPoint = 0; iPoint < nPointDomain; iPoint++) {
+        auto residual = numerics->ComputeVibRelaxation(config);
 
         /*--- Check for errors before applying source to the linear system ---*/
         err = false;
@@ -1261,41 +1289,16 @@ void CNEMOEulerSolver::Source_Residual(CGeometry *geometry, CSolver **solver_con
         //    for (jVar = 0; jVar < nVar; jVar++)
         //      if (Jacobian_i[iVar][jVar] != Jacobian_i[iVar][jVar]) err = true;
 
-        /*--- Apply the chemical sources to the linear system ---*/
+        /*--- Apply the vibrational relaxation terms to the linear system ---*/
         if (!err) {
           LinSysRes.SubtractBlock(iPoint, residual);
           //if (implicit)
           //  Jacobian.SubtractBlock(iPoint, iPoint, Jacobian_i);
         } else
-          eChm_local++;
+          eVib_local++;
       }
     }
 
-    /*--- Compute vibrational energy relaxation ---*/
-    /// NOTE: Jacobians don't account for relaxation time derivatives
-
-    if (!monoatomic){
-      auto residual = numerics->ComputeVibRelaxation(config);
-
-      /*--- Check for errors before applying source to the linear system ---*/
-      err = false;
-      for (iVar = 0; iVar < nVar; iVar++)
-        if (residual[iVar] != residual[iVar]) err = true;
-      //if (implicit)
-      //  for (iVar = 0; iVar < nVar; iVar++)
-      //    for (jVar = 0; jVar < nVar; jVar++)
-      //      if (Jacobian_i[iVar][jVar] != Jacobian_i[iVar][jVar]) err = true;
-
-      /*--- Apply the vibrational relaxation terms to the linear system ---*/
-      if (!err) {
-        LinSysRes.SubtractBlock(iPoint, residual);
-        //if (implicit)
-        //  Jacobian.SubtractBlock(iPoint, iPoint, Jacobian_i);
-      } else
-        eVib_local++;
-    }
-  }
-
   /*--- Checking for NaN ---*/
   eAxi_global = eAxi_local;
   eChm_global = eChm_local;