Only initialize shapes for Jacobians (idaholab#5913)

dschwen · Nov 3, 2015 · 9314dbc · 9314dbc
1 parent a7b61c6
commit 9314dbc
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 9 deletions.
diff --git a/framework/include/userobject/ShapeElementUserObject.h b/framework/include/userobject/ShapeElementUserObject.h
@@ -25,8 +25,11 @@ InputParameters validParams<ShapeElementUserObject>();
 
 /**
  * ElementUserObject class in which the _phi and _grad_phi shape function data
- * is available and correctly initialized. This enables the calculation of
- * Jacobian matrix contributions inside a UO
+ * is available and correctly initialized on EXEC_NONLINEAR (the Jacobian calculation).
+ * This enables the calculation of Jacobian matrix contributions inside a UO.
+ *
+ * \warning It is up to the user to ensure _fe_problem.currentlyComputingJacobian()
+ *          returns true before utilizing the shape functions.
  */
 class ShapeElementUserObject : public ElementUserObject
 {

diff --git a/framework/src/base/Assembly.C b/framework/src/base/Assembly.C
@@ -1526,14 +1526,14 @@ Assembly::addCachedJacobianContributions(SparseMatrix<Number> & jacobian)
 void
 Assembly::registerUserObjectShapeVariable(unsigned int var)
 {
-  std::vector<unsigned int> & vec = _user_object_shape_variables;
+  std::vector<unsigned int> & reg = _user_object_shape_variables;
 
-  // add variable to the vector
-  vec.push_back(var);
+  // add variable number to the list of registered variables
+  reg.push_back(var);
 
   // sort and keep elements unique
-  std::sort(vec.begin(), vec.end());
-  vec.erase(std::unique(vec.begin(), vec.end()), vec.end());
+  std::sort(reg.begin(), reg.end());
+  reg.erase(std::unique(reg.begin(), reg.end()), reg.end());
 }
 
 void

diff --git a/framework/src/base/ComputeUserObjectsThread.C b/framework/src/base/ComputeUserObjectsThread.C
@@ -170,8 +170,9 @@ ComputeUserObjectsThread::onElement(const Elem * elem)
   _fe_problem.reinitMaterials(_subdomain, _tid);
 
   // Prepare shape functions for ShapeElementUserObjects
-  for (unsigned int i = 0; i < _user_object_shape_variables.size(); ++i)
-    _fe_problem.prepareShapes(_user_object_shape_variables[i], _tid);
+  if (_fe_problem.currentlyComputingJacobian())
+    for (unsigned int i = 0; i < _user_object_shape_variables.size(); ++i)
+      _fe_problem.prepareShapes(_user_object_shape_variables[i], _tid);
 
   //Global UserObjects
   for (std::vector<ElementUserObject *>::const_iterator UserObject_it = _user_objects[_tid].elementUserObjects(Moose::ANY_BLOCK_ID, _group).begin();

diff --git a/framework/src/base/FEProblem.C b/framework/src/base/FEProblem.C
@@ -2247,6 +2247,9 @@ FEProblem::execute(const ExecFlagType & exec_type)
 {
   // Set the current flag
   _current_execute_on_flag = exec_type;
+  if (exec_type == EXEC_NONLINEAR)
+    _currently_computing_jacobian = true;
+
 
   // Pre-aux UserObjects
   Moose::perf_log.push("computeUserObjects()", "Setup");
@@ -2265,6 +2268,7 @@ FEProblem::execute(const ExecFlagType & exec_type)
 
   // Return the current flag to None
   _current_execute_on_flag = EXEC_NONE;
+  _currently_computing_jacobian = false;
 }
 
 void