more fixes.

Author: MFraters

source/simulator/assemblers/newton_stokes.cc

@@ -464,13 +464,14 @@ namespace aspect
               if (enable_prescribed_dilation)
                 {
                   const unsigned int index_direction=fe.system_to_component_index(i).first;
-                  data.local_rhs(i) += (
-                                         - pressure_scaling
-                                         * (prescribed_dilation->dilation_rhs_term[index_direction][q] -
-                                            prescribed_dilation->dilation_lhs_term[q] *
-                                            scratch.material_model_inputs.pressure[q])
-                                         * scratch.phi_p[i]
-                                       ) * JxW;
+                  if (index_direction < dim)
+                    data.local_rhs(i) += (
+                                           - pressure_scaling
+                                           * (prescribed_dilation->dilation_rhs_term[index_direction][q] -
+                                              prescribed_dilation->dilation_lhs_term[q] *
+                                              scratch.material_model_inputs.pressure[q])
+                                           * scratch.phi_p[i]
+                                         ) * JxW;
                 }
             }
 

source/simulator/assemblers/stokes.cc

@@ -458,31 +458,17 @@ namespace aspect
                   if (introspection.is_stokes_component(index_direction+1))
                     {
                       const unsigned int index_direction=fe.system_to_component_index(i).first;
-                      data.local_rhs(i) += (
-                                             - pressure_scaling
-                                             * (prescribed_dilation->dilation_rhs_term[index_direction][q] -
-                                                prescribed_dilation->dilation_lhs_term[q] *
-                                                scratch.material_model_inputs.pressure[q])
-                                             * scratch.phi_p[i]
-                                           ) * JxW;
+                      if (index_direction < dim)
+                        data.local_rhs(i) += (
+                                               - pressure_scaling
+                                               * (prescribed_dilation->dilation_rhs_term[index_direction][q] -
+                                                  prescribed_dilation->dilation_lhs_term[q] *
+                                                  scratch.material_model_inputs.pressure[q])
+                                               * scratch.phi_p[i]
+                                             ) * JxW;
                     }
                 }
 
-              // Only assemble this term if we are running incompressible, otherwise this term
-              // is already included on the LHS of the equation.
-              if (prescribed_dilation != nullptr && !material_model_is_compressible)
-                {
-                  const unsigned int index_direction=fe.system_to_component_index(i).first;
-                  // Only want the velocity components and not the pressure one (which is the last one), so add 1
-                  if (introspection.is_stokes_component(index_direction+1))
-                    data.local_rhs(i) += (
-                                           // RHS of momentum eqn: - \int 2/3 eta R, div v
-                                           - 2.0 / 3.0 * eta
-                                           * prescribed_dilation->dilation_rhs_term[index_direction][q]
-                                           * scratch.div_phi_u[i]
-                                         ) * JxW;
-                }
-
               if (scratch.rebuild_stokes_matrix)
                 for (unsigned int j=0; j<stokes_dofs_per_cell; ++j)
                   {