MFEM Non-linear support

framework/doc/content/source/mfem/functions/MFEMGradientGridFunction.md

@@ -0,0 +1,16 @@
+# MFEMGradientGridFunction
+
+!if! function=hasCapability('mfem')
+
+## Summary
+
+This is class which can be use as vector cofficients to MFEM problems.
+
+## Overview
+
+`MFEMGradientGridFunction` is intended to allow the specification of `mfem::VectorCoefficient` object to add to the MFEM problem in a manner consistent with the standard MOOSE Materials system.
+
+!if-end!
+
+!else
+!include mfem/mfem_warning.md

framework/doc/content/source/mfem/kernels/MFEMDomainLFGradKernel.md

@@ -0,0 +1,35 @@
+# MFEMDomainLFGradKernel
+
+!if! function=hasCapability('mfem')
+
+!syntax description /Kernels/MFEMDomainLFGradKernel
+
+## Overview
+
+Adds the domain integrator for integrating the linear form
+
+!equation
+(\vec{f}, \nabla v)_\Omega \,\,\, \forall v \in V
+
+where $v \in H^1$ is the test variable and $\vec{f}$ is a
+vector forcing coefficient.
+
+This term arises from the weak form of the forcing term
+
+!equation
+-\nabla \cdot \vec{f}
+
+## Example Input File Syntax
+
+!listing mfem/kernels/nldiffusion.i block=/Kernels
+
+!syntax parameters /Kernels/MFEMDomainLFGradKernel
+
+!syntax inputs /Kernels/MFEMDomainLFGradKernel
+
+!syntax children /Kernels/MFEMDomainLFGradKernel
+
+!if-end!
+
+!else
+!include mfem/mfem_warning.md

framework/doc/content/source/mfem/kernels/MFEMMixedScalarWeakDivergenceKernel.md

@@ -0,0 +1,35 @@
+# MFEMMixedScalarWeakDivergenceKernel
+
+!if! function=hasCapability('mfem')
+
+!syntax description /Kernels/MFEMMixedScalarWeakDivergenceKernel
+
+## Overview
+
+Adds the domain integrator for integrating the bilinear form
+
+!equation
+(-\vec{\lambda} u, \nabla v)_\Omega \,\,\, \forall v \in V
+
+where $v \in H^1$ is the test variable and $\vec{\lambda}$ is a
+vector forcing coefficient.
+
+This term arises from the weak form of the forcing term
+
+!equation
+\nabla \cdot(\vec{\lambda} u)
+
+## Example Input File Syntax
+
+!listing mfem/kernels/nldiffusion.i block=/Kernels
+
+!syntax parameters /Kernels/MFEMMixedScalarWeakDivergenceKernel
+
+!syntax inputs /Kernels/MFEMMixedScalarWeakDivergenceKernel
+
+!syntax children /Kernels/MFEMMixedScalarWeakDivergenceKernel
+
+!if-end!
+
+!else
+!include mfem/mfem_warning.md

framework/include/mfem/equation_systems/EquationSystem.h

@@ -54,13 +54,15 @@ class EquationSystem : public mfem::Operator
 
   /// Build linear forms and eliminate constrained DoFs
   virtual void BuildLinearForms();
+  virtual void BuildNonLinearActions();
   virtual void ApplyEssentialBCs();
   virtual void EliminateCoupledVariables();
 
   /// Build bilinear forms
   virtual void BuildBilinearForms();
   virtual void BuildMixedBilinearForms();
-  virtual void BuildEquationSystem();
+  virtual void BuildEquationSystem(Moose::MFEM::GridFunctions & gridfunctions,
+                                   mfem::Array<int> & btoffsets);
 
   /// Form linear system, with essential boundary conditions accounted for
   virtual void FormLinearSystem(mfem::OperatorHandle & op,
@@ -73,6 +75,8 @@ class EquationSystem : public mfem::Operator
                                 mfem::BlockVector & trueX,
                                 mfem::BlockVector & trueRHS);
 
+  void UpdateJacobian() const;
+
   /// Build linear system, with essential boundary conditions accounted for
   virtual void BuildJacobian(mfem::BlockVector & trueX, mfem::BlockVector & trueRHS);
 
@@ -131,7 +135,7 @@ class EquationSystem : public mfem::Operator
   // Components of weak form. // Named according to test variable
   Moose::MFEM::NamedFieldsMap<mfem::ParBilinearForm> _blfs;
   Moose::MFEM::NamedFieldsMap<mfem::ParLinearForm> _lfs;
-  Moose::MFEM::NamedFieldsMap<mfem::ParNonlinearForm> _nlfs;
+  Moose::MFEM::NamedFieldsMap<mfem::ParLinearForm> _nlfs;
   Moose::MFEM::NamedFieldsMap<Moose::MFEM::NamedFieldsMap<mfem::ParMixedBilinearForm>>
       _mblfs; // named according to trial variable
 
@@ -153,6 +157,12 @@ class EquationSystem : public mfem::Operator
       Moose::MFEM::NamedFieldsMap<
           Moose::MFEM::NamedFieldsMap<std::vector<std::shared_ptr<MFEMKernel>>>> & kernels_map);
 
+  void ApplyDomainNLActionIntegrators(
+      const std::string & test_var_name,
+      std::shared_ptr<mfem::ParLinearForm> form,
+      Moose::MFEM::NamedFieldsMap<
+          Moose::MFEM::NamedFieldsMap<std::vector<std::shared_ptr<MFEMKernel>>>> & kernels_map);
+
   template <class FormType>
   void ApplyBoundaryBLFIntegrators(
       const std::string & trial_var_name,
@@ -188,8 +198,16 @@ class EquationSystem : public mfem::Operator
   Moose::MFEM::NamedFieldsMap<std::vector<std::shared_ptr<MFEMEssentialBC>>> _essential_bc_map;
 
   mutable mfem::OperatorHandle _jacobian;
+  mutable mfem::Vector _trueRHS;
+  mutable mfem::BlockVector _trueBlockRHS, _trueBlockSol, _BlockResidual;
+
+  Moose::MFEM::GridFunctions * _gfuncs;
+  mfem::Array<int> * _block_true_offsets;
+  mfem::Array<int> empty_tdof;
 
   mfem::AssemblyLevel _assembly_level;
+
+  bool _non_linear = false;
 };
 
 template <class FormType>
@@ -240,6 +258,30 @@ EquationSystem::ApplyDomainLFIntegrators(
   }
 }
 
+inline void
+EquationSystem::ApplyDomainNLActionIntegrators(
+    const std::string & test_var_name,
+    std::shared_ptr<mfem::ParLinearForm> form,
+    Moose::MFEM::NamedFieldsMap<
+        Moose::MFEM::NamedFieldsMap<std::vector<std::shared_ptr<MFEMKernel>>>> & kernels_map)
+{
+  if (kernels_map.Has(test_var_name) && kernels_map.Get(test_var_name)->Has(test_var_name))
+  {
+    auto kernels = kernels_map.GetRef(test_var_name).GetRef(test_var_name);
+    for (auto & kernel : kernels)
+    {
+      mfem::LinearFormIntegrator * integ = kernel->createNLActionIntegrator();
+      if (integ != nullptr)
+      {
+        _non_linear = true;
+        kernel->isSubdomainRestricted()
+            ? form->AddDomainIntegrator(std::move(integ), kernel->getSubdomainMarkers())
+            : form->AddDomainIntegrator(std::move(integ));
+      }
+    }
+  }
+}
+
 template <class FormType>
 void
 EquationSystem::ApplyBoundaryBLFIntegrators(
@@ -303,7 +345,8 @@ class TimeDependentEquationSystem : public EquationSystem
   void AddCoupledVariableNameIfMissing(const std::string & coupled_var_name) override;
 
   virtual void SetTimeStep(mfem::real_t dt);
-  virtual void UpdateEquationSystem();
+  virtual void UpdateEquationSystem(Moose::MFEM::GridFunctions & gridfunctions,
+                                    mfem::Array<int> & btoffsets);
 
   virtual void AddKernel(std::shared_ptr<MFEMKernel> kernel) override;
   virtual void BuildBilinearForms() override;
@@ -314,6 +357,8 @@ class TimeDependentEquationSystem : public EquationSystem
                           mfem::BlockVector & truedXdt,
                           mfem::BlockVector & trueRHS) override;
 
+  void UpdateEssDerivativeVals(const mfem::real_t & dt, const mfem::Vector & x_old);
+
 protected:
   /// Coefficient for timestep scaling
   mfem::ConstantCoefficient _dt_coef;

framework/include/mfem/executioners/MFEMProblemSolve.h

@@ -35,6 +35,10 @@ class MFEMProblemSolve : public SolveObject
 protected:
   MFEMProblem & _mfem_problem;
   std::vector<std::shared_ptr<Moose::MFEM::ProblemOperatorBase>> & _problem_operators;
+  unsigned int _nl_max_its;
+  mfem::real_t _nl_abs_tol;
+  mfem::real_t _nl_rel_tol;
+  unsigned int _print_level;
 };
 
 #endif

framework/include/mfem/functions/MFEMGradientGridFunction.h

@@ -0,0 +1,32 @@
+//* This file is part of the MOOSE framework
+//* https://mooseframework.inl.gov
+//*
+//* All rights reserved, see COPYRIGHT for full restrictions
+//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
+//*
+//* Licensed under LGPL 2.1, please see LICENSE for details
+//* https://www.gnu.org/licenses/lgpl-2.1.html
+
+#ifdef MOOSE_MFEM_ENABLED
+
+#pragma once
+
+#include "MFEMGeneralUserObject.h"
+
+/**
+ * Declares arbitrary parsed function of position, time, and any number of problem variables.
+ */
+class MFEMGradientGridFunction : public MFEMGeneralUserObject
+{
+public:
+  static InputParameters validParams();
+
+  MFEMGradientGridFunction(const InputParameters & parameters);
+  virtual ~MFEMGradientGridFunction();
+
+protected:
+  /// grid function variable variable
+  const VariableName & _var_name;
+};
+
+#endif

framework/include/mfem/kernels/MFEMDomainLFGradKernel.h

@@ -0,0 +1,34 @@
+//* This file is part of the MOOSE framework
+//* https://mooseframework.inl.gov
+//*
+//* All rights reserved, see COPYRIGHT for full restrictions
+//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
+//*
+//* Licensed under LGPL 2.1, please see LICENSE for details
+//* https://www.gnu.org/licenses/lgpl-2.1.html
+
+#ifdef MOOSE_MFEM_ENABLED
+
+#pragma once
+#include "MFEMKernel.h"
+
+/**
+ * \f[
+ * (f, v)
+ * \f]
+ */
+class MFEMDomainLFGradKernel : public MFEMKernel
+{
+public:
+  static InputParameters validParams();
+
+  MFEMDomainLFGradKernel(const InputParameters & parameters);
+
+  virtual mfem::LinearFormIntegrator * createNLActionIntegrator() override;
+
+protected:
+  mfem::Coefficient & _coef;
+  mfem::ScalarVectorProductCoefficient * _product_coeff;
+};
+
+#endif

framework/include/mfem/kernels/MFEMKernel.h

@@ -29,6 +29,7 @@ class MFEMKernel : public MFEMGeneralUserObject, public MFEMBlockRestrictable
 
   /// Create a new MFEM integrator to apply to the weak form. Ownership managed by the caller.
   virtual mfem::LinearFormIntegrator * createLFIntegrator() { return nullptr; }
+  virtual mfem::LinearFormIntegrator * createNLActionIntegrator() { return nullptr; }
   virtual mfem::BilinearFormIntegrator * createBFIntegrator() { return nullptr; }
 
   /// Get name of the test variable labelling the weak form this kernel is added to

framework/include/mfem/kernels/MFEMMixedScalarWeakDivergenceKernel.h

@@ -0,0 +1,34 @@
+//* This file is part of the MOOSE framework
+//* https://mooseframework.inl.gov
+//*
+//* All rights reserved, see COPYRIGHT for full restrictions
+//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
+//*
+//* Licensed under LGPL 2.1, please see LICENSE for details
+//* https://www.gnu.org/licenses/lgpl-2.1.html
+
+#ifdef MOOSE_MFEM_ENABLED
+
+#pragma once
+#include "MFEMKernel.h"
+
+/**
+ * \f[
+ * (k \vec \nabla u, \vec \nabla v)
+ * \f]
+ */
+class MFEMMixedScalarWeakDivergenceKernel : public MFEMKernel
+{
+public:
+  static InputParameters validParams();
+
+  MFEMMixedScalarWeakDivergenceKernel(const InputParameters & parameters);
+
+  virtual mfem::BilinearFormIntegrator * createBFIntegrator() override;
+
+protected:
+  mfem::Coefficient & _coef;
+  mfem::ScalarVectorProductCoefficient * _product_coef;
+};
+
+#endif

framework/include/mfem/problem/MFEMProblem.h

@@ -210,8 +210,20 @@ class MFEMProblem : public ExternalProblem
    */
   std::shared_ptr<mfem::ParGridFunction> getGridFunction(const std::string & name);
 
+  /**
+   * set newton solver parameters
+   */
+  void setNewtonParamaters(unsigned int nl_max_its,
+                           mfem::real_t nl_abs_tol,
+                           mfem::real_t nl_rel_tol,
+                           unsigned int print_level);
+
 protected:
   MFEMProblemData _problem_data;
+  unsigned int _nl_max_its;
+  mfem::real_t _nl_abs_tol;
+  mfem::real_t _nl_rel_tol;
+  unsigned int _print_level;
 };
 
 #endif