more pp fixes

prisms-center · Nov 17, 2024 · b4e7492 · b4e7492
1 parent 9a016f1
commit b4e7492
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/include/variableAttributeLoader.h b/include/variableAttributeLoader.h
@@ -21,9 +21,9 @@ struct variableAttributes
   bool        need_value_nucleation = false;
   bool        nucleating_variable   = false;
   bool        is_pp                 = false;
-  bool        output_integral       = false;
   bool        is_nonlinear          = false;
   bool        calc_integral         = false;
+  bool        output_integral       = false;
 
   std::set<std::string> dependencies_value_RHS;
   std::set<std::string> dependencies_gradient_RHS;

diff --git a/src/variableAttributeLoader/variableAttributeLoader.cc b/src/variableAttributeLoader/variableAttributeLoader.cc
@@ -53,6 +53,23 @@ variableAttributes::parse_residual_dependencies()
          {&dependencies_value_LHS, &dependencies_gradient_LHS}}}}
   };
 
+  if (is_pp)
+    {
+      EvalFlags             &residual_flags          = eval_flags_residual_postprocess;
+      std::set<std::string> &value_dependency_set    = dependencies_value_PP;
+      std::set<std::string> &gradient_dependency_set = dependencies_gradient_PP;
+
+      if (!value_dependency_set.empty())
+        {
+          residual_flags |= dealii::EvaluationFlags::values;
+        }
+      if (!gradient_dependency_set.empty())
+        {
+          residual_flags |= dealii::EvaluationFlags::gradients;
+        }
+      return;
+    }
+
   for (const std::pair<EvalFlags *,
                        std::pair<std::set<std::string> *, std::set<std::string> *>>
          &eval_dep_pair : dependencies_for.at(eq_type))
@@ -222,6 +239,7 @@ void
 variableAttributeLoader::set_output_integral(const unsigned int &index, const bool &flag)
 {
   (*relevant_attributes)[index].output_integral = flag;
+  (*relevant_attributes)[index].calc_integral   = flag;
 }
 
 void