https://mooseframework.inl.gov
Loading...
Searching...
No Matches
Public Member Functions | Protected Member Functions | Protected Attributes | List of all members
Moose::MFEM::EquationSystem Class Reference

Owns the weak-form mathematics of a MOOSE MFEM problem. More...

#include <EquationSystem.h>

Inheritance diagram for Moose::MFEM::EquationSystem:
[legend]

Public Member Functions

 EquationSystem ()=default
 
 ~EquationSystem () override
 
virtual void AddKernel (std::shared_ptr< MFEMKernel > kernel)
 Add kernels.
 
virtual void AddIntegratedBC (std::shared_ptr< MFEMIntegratedBC > kernel)
 
virtual void AddEssentialBC (std::shared_ptr< MFEMEssentialBC > bc)
 Add BC associated with essentially constrained DoFs on boundaries.
 
virtual void Init (GridFunctions &gridfunctions, ComplexGridFunctions &cmplx_gridfunctions, mfem::AssemblyLevel assembly_level)
 Initialise.
 
void FormSystem (mfem::BlockVector &trueX, mfem::BlockVector &trueRHS)
 Build all weak-form components via BuildEquationSystem(), form the constrained linear part of the system, and populate the true-DoF vectors used by the solve.
 
void Mult (const mfem::Vector &u, mfem::Vector &residual) const override
 Compute residual y = Mu.
 
virtual void ComputeNonlinearResidual (const mfem::Vector &u, mfem::Vector &residual) const
 Compute the contribution to the residual from nonlinear forms only.
 
mfem::Operator & GetGradient (const mfem::Vector &u) const override
 Get Jacobian at the provided vector of true DoFs of trial variables.
 
mfem::OperatorHandle & GetLinearOperator () const
 Get operator handle for linear component of system operator.
 
virtual void SetTrialVariablesFromTrueVectors (const mfem::BlockVector &trueX) const
 Update variable from solution vector after solve.
 
void SetGradientRequired (bool requires_gradient)
 Set whether an external object (such as a nonlinear solver) requires Jacobian information for this EquationSystem.
 
void SetCoefficientManager (CoefficientManager &coefficients)
 Set the coefficient manager to notify when trial variables are updated, so that stored projections of solution-dependent coefficients are invalidated.
 
const std::vector< std::string > & GetTrialVarNames () const
 
const std::vector< std::string > & GetTestVarNames () const
 
const mfem::Array< int > & GetBlockOffsets () const
 Getter for block true offsets associated with the EquationSystem operator.
 
mfem::ParBilinearForm & GetBilinearForm (const std::string &test_var_name)
 
mfem::ParGridFunction & GetGridFunction (const std::string &trial_var_name)
 
const mfem::Vector & GetLinearizationPoint () const
 The true-DoF vector used for the most recent Jacobian linearization.
 
std::shared_ptr< mfem::ParBilinearForm > BuildBilinearFormForFESpace (const std::string &var_name, mfem::ParFiniteElementSpace &fespace, mfem::AssemblyLevel assembly_level)
 Build a fresh ParBilinearForm on the given FESpace using the same kernels as the main system's bilinear form for var_name.
 
std::shared_ptr< mfem::ParNonlinearForm > BuildNonlinearFormForFESpace (const std::string &var_name, mfem::ParFiniteElementSpace &fespace, mfem::AssemblyLevel assembly_level)
 Build a fresh ParNonlinearForm on the given FESpace using the same kernels as the main system's nonlinear form for var_name.
 
mfem::Array< int > & GetEssentialBoundaryMarkers (const std::string &var_name)
 Return the essential boundary attribute marker array for a given trial variable.
 
virtual bool IsComplex () const
 
virtual bool IsEigen () const
 
virtual bool IsTimeDependent () const
 
bool IsMultivariate () const
 
bool IsNonlinear () const
 
virtual void BuildEquationSystem ()
 Build all forms comprising this EquationSystem.
 

Protected Member Functions

virtual void AddCoupledVariableNameIfMissing (const std::string &coupled_var_name)
 Add coupled variable to EquationSystem.
 
virtual void AddEliminatedVariableNameIfMissing (const std::string &eliminated_var_name)
 Add eliminated variable to EquationSystem.
 
virtual void AddTestVariableNameIfMissing (const std::string &test_var_name)
 Add test variable to EquationSystem.
 
virtual void SetTrialVariableNames ()
 Set trial variable names from subset of coupled variables that have an associated test variable.
 
void DeleteHBlocks ()
 Deletes the HypreParMatrix associated with any pointer stored in _h_blocks, and then proceeds to delete all dynamically allocated memory for _h_blocks itself, resetting all dimensions to zero.
 
void DeleteJacobianBlocks ()
 Deletes the HypreParMatrix associated with any pointer stored in _jacobian_blocks, and then proceeds to delete all dynamically allocated memory for _jacobian_blocks itself, resetting all dimensions to zero.
 
bool VectorContainsName (const std::vector< std::string > &the_vector, const std::string &name) const
 
virtual void ApplyEssentialBC (const std::string &var_name, mfem::ParGridFunction &trial_gf, mfem::Array< int > &global_ess_markers)
 Apply essential BC(s) associated with var_name to set true DoFs of trial_gf and update markers of all essential boundaries.
 
virtual void ApplyEssentialBCs ()
 Update all essentially constrained true DoF markers and values on boundaries.
 
virtual void EliminateCoupledVariables ()
 Perform trivial eliminations of coupled variables lacking corresponding test variables.
 
virtual void BuildLinearForms ()
 Build linear forms and eliminate constrained DoFs.
 
virtual void BuildNonlinearForms ()
 Build non-linear action forms.
 
virtual void BuildBilinearForms ()
 Build bilinear forms (diagonal Jacobian contributions)
 
virtual void BuildMixedBilinearForms ()
 Build mixed bilinear forms (off-diagonal Jacobian contributions)
 
virtual void FormLinearSystem (mfem::OperatorHandle &op, mfem::BlockVector &trueX, mfem::BlockVector &trueRHS)
 Form linear components of system based on on- and off-diagonal bilinear form contributions, populate solution and RHS vectors of true DoFs, and apply constraints.
 
virtual void FormSystemOperator (mfem::OperatorHandle &op, mfem::BlockVector &trueX, mfem::BlockVector &trueRHS)
 Form matrix-free representation of linear components of system operator.
 
virtual void FormSystemMatrix (mfem::OperatorHandle &op, mfem::BlockVector &trueX, mfem::BlockVector &trueRHS)
 Form matrix representation of linear components of system operator as a HypreParMatrix.
 
void FormJacobianMatrix (const mfem::Vector &u)
 Compute Jacobian matrix at the provided vector of true DoFs of trial variables.
 
template<class FormType >
void ApplyDomainBLFIntegrators (const std::string &trial_var_name, const std::string &test_var_name, std::shared_ptr< FormType > form, NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMKernel > > > > &kernels_map, std::optional< mfem::real_t > scale_factor=std::nullopt)
 Template method for applying BilinearFormIntegrators on domains from kernels to a BilinearForm, or MixedBilinearForm.
 
void ApplyDomainLFIntegrators (const std::string &test_var_name, std::shared_ptr< mfem::ParLinearForm > form, NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMKernel > > > > &kernels_map)
 Apply domain LinearFormIntegrators from kernels to the linear form associated with the supplied test variable.
 
void ApplyDomainNLFIntegrators (const std::string &test_var_name, std::shared_ptr< mfem::ParNonlinearForm > form, NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMKernel > > > > &kernels_map, std::optional< mfem::real_t > scale_factor=std::nullopt)
 Apply domain NonlinearFormIntegrators from kernels to the nonlinear form associated with the supplied test variable.
 
template<class FormType >
void ApplyBoundaryBLFIntegrators (const std::string &trial_var_name, const std::string &test_var_name, std::shared_ptr< FormType > form, NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMIntegratedBC > > > > &integrated_bc_map, std::optional< mfem::real_t > scale_factor=std::nullopt)
 Template method for applying BilinearFormIntegrators on boundaries from integrated boundary conditions to a BilinearForm, or MixedBilinearForm.
 
void ApplyBoundaryLFIntegrators (const std::string &test_var_name, std::shared_ptr< mfem::ParLinearForm > form, NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMIntegratedBC > > > > &integrated_bc_map)
 Apply boundary LinearFormIntegrators from integrated boundary conditions to the linear form associated with the supplied test variable.
 
void ApplyBoundaryNLFIntegrators (const std::string &test_var_name, std::shared_ptr< mfem::ParNonlinearForm > form, NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMIntegratedBC > > > > &integrated_bc_map, std::optional< mfem::real_t > scale_factor=std::nullopt)
 Apply boundary NonlinearFormIntegrators from integrated boundary conditions to the nonlinear form associated with the supplied test variable.
 

Protected Attributes

std::vector< std::string > _coupled_var_names
 Names of all trial variables of kernels and boundary conditions added to this EquationSystem.
 
std::vector< std::string > _trial_var_names
 Subset of _coupled_var_names of all variables corresponding to gridfunctions with degrees of freedom that comprise the state vector of this EquationSystem.
 
std::vector< std::string > _eliminated_var_names
 Names of all coupled variables without a corresponding test variable.
 
Moose::MFEM::GridFunctions _eliminated_variables
 Pointers to coupled variables not part of the reduced EquationSystem.
 
std::vector< std::string > _test_var_names
 Names of all test variables corresponding to linear forms in this equation system.
 
std::vector< mfem::ParFiniteElementSpace * > _test_pfespaces
 Pointers to finite element spaces associated with test variables.
 
std::vector< mfem::ParFiniteElementSpace * > _coupled_pfespaces
 Pointers to finite element spaces associated with coupled variables.
 
NamedFieldsMap< mfem::ParBilinearForm > _blfs
 
NamedFieldsMap< mfem::ParLinearForm > _lfs
 
NamedFieldsMap< mfem::ParNonlinearForm > _nlfs
 
NamedFieldsMap< NamedFieldsMap< mfem::ParMixedBilinearForm > > _mblfs
 
std::vector< std::unique_ptr< mfem::ParGridFunction > > _var_ess_constraints
 Gridfunctions holding essential constraints from Dirichlet BCs.
 
std::vector< mfem::Array< int > > _ess_tdof_lists
 
std::vector< mfem::Array< int > > _ess_markers
 
mfem::Array2D< const mfem::HypreParMatrix * > _h_blocks
 
mfem::Array2D< const mfem::HypreParMatrix * > _jacobian_blocks
 
NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMKernel > > > > _kernels_map
 Arrays to store kernels to act on each component of weak form.
 
NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMIntegratedBC > > > > _integrated_bc_map
 Arrays to store integrated BCs to act on each component of weak form.
 
NamedFieldsMap< std::vector< std::shared_ptr< MFEMEssentialBC > > > _essential_bc_map
 Arrays to store essential BCs to act on each component of weak form.
 
mfem::OperatorHandle _jacobian
 
mfem::OperatorHandle _linear_operator
 
mfem::AssemblyLevel _assembly_level
 
Moose::MFEM::GridFunctions * _gfuncs
 
mfem::Array< int > _block_true_offsets
 
const mfem::Vector * _linearization_point = nullptr
 
bool _non_linear = false
 
bool _gradient_required = false
 
CoefficientManager * _coefficient_manager = nullptr
 

Detailed Description

Owns the weak-form mathematics of a MOOSE MFEM problem.

An EquationSystem stores weak-form components (bilinear, linear, mixed-bilinear, and nonlinear forms) contributed by kernels and boundary conditions. It forms the constrained linear part, keeps nonlinear action forms for residual/Jacobian evaluation, and exposes the solve interface as an mfem::Operator. It is responsible for applying essential-DoF constraints and propagating either itself or the assembled linear operator (and bilinear form, for LOR) to the configured solver tree.

EquationSystem is not responsible for grid-function bookkeeping, time stepping, or solver selection - those belong to the ProblemOperator layer.

See also
ProblemOperatorBase for the conceptual split between the two layers.

Definition at line 44 of file EquationSystem.h.

Constructor & Destructor Documentation

◆ EquationSystem()

Moose::MFEM::EquationSystem::EquationSystem ( )
default

◆ ~EquationSystem()

Moose::MFEM::EquationSystem::~EquationSystem ( )
override

Definition at line 20 of file EquationSystem.C.

21{
24}
void DeleteHBlocks()
Deletes the HypreParMatrix associated with any pointer stored in _h_blocks, and then proceeds to dele...
void DeleteJacobianBlocks()
Deletes the HypreParMatrix associated with any pointer stored in _jacobian_blocks,...

Member Function Documentation

◆ AddCoupledVariableNameIfMissing()

void Moose::MFEM::EquationSystem::AddCoupledVariableNameIfMissing ( const std::string &  coupled_var_name)
protectedvirtual

Add coupled variable to EquationSystem.

Definition at line 57 of file EquationSystem.C.

58{
59 if (!VectorContainsName(_coupled_var_names, coupled_var_name))
60 _coupled_var_names.push_back(coupled_var_name);
61}
std::vector< std::string > _coupled_var_names
Names of all trial variables of kernels and boundary conditions added to this EquationSystem.
bool VectorContainsName(const std::vector< std::string > &the_vector, const std::string &name) const

Referenced by Moose::MFEM::ComplexEquationSystem::AddComplexIntegratedBC(), Moose::MFEM::ComplexEquationSystem::AddComplexKernel(), AddIntegratedBC(), and AddKernel().

◆ AddEliminatedVariableNameIfMissing()

void Moose::MFEM::EquationSystem::AddEliminatedVariableNameIfMissing ( const std::string &  eliminated_var_name)
protectedvirtual

Add eliminated variable to EquationSystem.

Definition at line 64 of file EquationSystem.C.

65{
66 if (!VectorContainsName(_eliminated_var_names, eliminated_var_name))
67 _eliminated_var_names.push_back(eliminated_var_name);
68}
std::vector< std::string > _eliminated_var_names
Names of all coupled variables without a corresponding test variable.

Referenced by Moose::MFEM::TimeDependentEquationSystem::AddKernel().

◆ AddEssentialBC()

void Moose::MFEM::EquationSystem::AddEssentialBC ( std::shared_ptr< MFEMEssentialBC >  bc)
virtual

Add BC associated with essentially constrained DoFs on boundaries.

Definition at line 139 of file EquationSystem.C.

140{
141 const auto & test_var_name = bc->getTestVariableName();
142 AddTestVariableNameIfMissing(test_var_name);
143 // Register new essential bc map if not present for the test variable
144 if (!_essential_bc_map.Has(test_var_name))
145 {
146 auto bcs = std::make_shared<std::vector<std::shared_ptr<MFEMEssentialBC>>>();
147 _essential_bc_map.Register(test_var_name, std::move(bcs));
148 }
149 _essential_bc_map.GetRef(test_var_name).push_back(std::move(bc));
150}
virtual void AddTestVariableNameIfMissing(const std::string &test_var_name)
Add test variable to EquationSystem.
NamedFieldsMap< std::vector< std::shared_ptr< MFEMEssentialBC > > > _essential_bc_map
Arrays to store essential BCs to act on each component of weak form.
void Register(const std::string &field_name, FieldArgs &&... args)
Construct new field with name field_name and register.
bool Has(const std::string &field_name) const
Predicate to check if a field is registered with name field_name.
T & GetRef(const std::string &field_name) const
Returns a reference to a field.

◆ AddIntegratedBC()

void Moose::MFEM::EquationSystem::AddIntegratedBC ( std::shared_ptr< MFEMIntegratedBC >  kernel)
virtual

Definition at line 116 of file EquationSystem.C.

117{
118 const auto & trial_var_name = bc->getTrialVariableName();
119 const auto & test_var_name = bc->getTestVariableName();
120 AddCoupledVariableNameIfMissing(trial_var_name);
121 AddTestVariableNameIfMissing(test_var_name);
122 // Register new integrated bc map if not present for the test variable
123 if (!_integrated_bc_map.Has(test_var_name))
124 {
125 auto integrated_bc_field_map = std::make_shared<
127 _integrated_bc_map.Register(test_var_name, std::move(integrated_bc_field_map));
128 }
129 // Register new integrated bc map if not present for the test/trial variable pair
130 if (!_integrated_bc_map.Get(test_var_name)->Has(trial_var_name))
131 {
132 auto bcs = std::make_shared<std::vector<std::shared_ptr<MFEMIntegratedBC>>>();
133 _integrated_bc_map.Get(test_var_name)->Register(trial_var_name, std::move(bcs));
134 }
135 _integrated_bc_map.GetRef(test_var_name).Get(trial_var_name)->push_back(std::move(bc));
136}
virtual void AddCoupledVariableNameIfMissing(const std::string &coupled_var_name)
Add coupled variable to EquationSystem.
NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMIntegratedBC > > > > _integrated_bc_map
Arrays to store integrated BCs to act on each component of weak form.
Lightweight adaptor over an std::map from strings to pointer to T.

◆ AddKernel()

void Moose::MFEM::EquationSystem::AddKernel ( std::shared_ptr< MFEMKernel >  kernel)
virtual

Add kernels.

Reimplemented in Moose::MFEM::TimeDependentEquationSystem.

Definition at line 93 of file EquationSystem.C.

94{
95 const auto & trial_var_name = kernel->getTrialVariableName();
96 const auto & test_var_name = kernel->getTestVariableName();
97 AddCoupledVariableNameIfMissing(trial_var_name);
98 AddTestVariableNameIfMissing(test_var_name);
99 // Register new kernels map if not present for the test variable
100 if (!_kernels_map.Has(test_var_name))
101 {
102 auto kernel_field_map =
103 std::make_shared<Moose::MFEM::NamedFieldsMap<std::vector<std::shared_ptr<MFEMKernel>>>>();
104 _kernels_map.Register(test_var_name, std::move(kernel_field_map));
105 }
106 // Register new kernels map if not present for the test/trial variable pair
107 if (!_kernels_map.Get(test_var_name)->Has(trial_var_name))
108 {
109 auto kernels = std::make_shared<std::vector<std::shared_ptr<MFEMKernel>>>();
110 _kernels_map.Get(test_var_name)->Register(trial_var_name, std::move(kernels));
111 }
112 _kernels_map.GetRef(test_var_name).Get(trial_var_name)->push_back(std::move(kernel));
113}
NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMKernel > > > > _kernels_map
Arrays to store kernels to act on each component of weak form.

Referenced by Moose::MFEM::TimeDependentEquationSystem::AddKernel().

◆ AddTestVariableNameIfMissing()

void Moose::MFEM::EquationSystem::AddTestVariableNameIfMissing ( const std::string &  test_var_name)
protectedvirtual

Add test variable to EquationSystem.

Definition at line 71 of file EquationSystem.C.

72{
73 if (!VectorContainsName(_test_var_names, test_var_name))
74 _test_var_names.push_back(test_var_name);
75}
std::vector< std::string > _test_var_names
Names of all test variables corresponding to linear forms in this equation system.

Referenced by Moose::MFEM::ComplexEquationSystem::AddComplexEssentialBCs(), Moose::MFEM::ComplexEquationSystem::AddComplexIntegratedBC(), Moose::MFEM::ComplexEquationSystem::AddComplexKernel(), AddEssentialBC(), AddIntegratedBC(), AddKernel(), and Moose::MFEM::TimeDependentEquationSystem::AddKernel().

◆ ApplyBoundaryBLFIntegrators()

template<class FormType >
void Moose::MFEM::EquationSystem::ApplyBoundaryBLFIntegrators ( const std::string &  trial_var_name,
const std::string &  test_var_name,
std::shared_ptr< FormType >  form,
NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMIntegratedBC > > > > &  integrated_bc_map,
std::optional< mfem::real_t >  scale_factor = std::nullopt 
)
protected

Template method for applying BilinearFormIntegrators on boundaries from integrated boundary conditions to a BilinearForm, or MixedBilinearForm.

Definition at line 381 of file EquationSystem.h.

388{
389 if (integrated_bc_map.Has(test_var_name) &&
390 integrated_bc_map.Get(test_var_name)->Has(trial_var_name))
391 {
392 auto bcs = integrated_bc_map.GetRef(test_var_name).GetRef(trial_var_name);
393 for (auto & bc : bcs)
394 {
395 mfem::BilinearFormIntegrator * integ = bc->createBFIntegrator();
396
397 if (integ)
398 {
399 if (scale_factor.has_value())
400 integ = new ScaleIntegrator(integ, scale_factor.value(), true);
401 bc->isDGBC() ? bc->isBoundaryRestricted()
402 ? form->AddBdrFaceIntegrator(std::move(integ), bc->getBoundaryMarkers())
403 : form->AddBdrFaceIntegrator(std::move(integ))
404 : bc->isBoundaryRestricted()
405 ? form->AddBoundaryIntegrator(std::move(integ), bc->getBoundaryMarkers())
406 : form->AddBoundaryIntegrator(std::move(integ));
407 }
408 }
409 }
410}

◆ ApplyBoundaryLFIntegrators()

void Moose::MFEM::EquationSystem::ApplyBoundaryLFIntegrators ( const std::string &  test_var_name,
std::shared_ptr< mfem::ParLinearForm >  form,
NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMIntegratedBC > > > > &  integrated_bc_map 
)
protected

Apply boundary LinearFormIntegrators from integrated boundary conditions to the linear form associated with the supplied test variable.

Definition at line 641 of file EquationSystem.C.

646{
647 if (integrated_bc_map.Has(test_var_name) &&
648 integrated_bc_map.Get(test_var_name)->Has(test_var_name))
649 {
650 auto bcs = integrated_bc_map.GetRef(test_var_name).GetRef(test_var_name);
651 for (auto & bc : bcs)
652 {
653 mfem::LinearFormIntegrator * integ = bc->createLFIntegrator();
654
655 if (integ)
656 {
657 bc->isDGBC() ? bc->isBoundaryRestricted()
658 ? form->AddBdrFaceIntegrator(std::move(integ), bc->getBoundaryMarkers())
659 : form->AddBdrFaceIntegrator(std::move(integ))
660 : bc->isBoundaryRestricted()
661 ? form->AddBoundaryIntegrator(std::move(integ), bc->getBoundaryMarkers())
662 : form->AddBoundaryIntegrator(std::move(integ));
663 }
664 }
665 }
666}

Referenced by BuildLinearForms().

◆ ApplyBoundaryNLFIntegrators()

void Moose::MFEM::EquationSystem::ApplyBoundaryNLFIntegrators ( const std::string &  test_var_name,
std::shared_ptr< mfem::ParNonlinearForm >  form,
NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMIntegratedBC > > > > &  integrated_bc_map,
std::optional< mfem::real_t >  scale_factor = std::nullopt 
)
protected

Apply boundary NonlinearFormIntegrators from integrated boundary conditions to the nonlinear form associated with the supplied test variable.

Definition at line 669 of file EquationSystem.C.

675{
676 if (integrated_bc_map.Has(test_var_name))
677 for (const auto & [trial_var_name, bcs] : integrated_bc_map.GetRef(test_var_name))
678 for (auto & bc : *bcs)
679 if (auto * integ = bc->createNLIntegrator())
680 {
681 if (_gradient_required && (test_var_name != trial_var_name))
683 "Support for Off-diagonal MFEM nonlinear boundary integrators in conjunction with "
684 "a nonlinear solver that requires a gradient is not currently "
685 "implemented. Boundary condition '",
686 bc->name(),
687 "' contributes to test variable '",
688 test_var_name,
689 "' from trial variable '",
690 trial_var_name,
691 "'.");
692
693 _non_linear = true;
694 if (scale_factor.has_value())
695 integ = new NLScaleIntegrator(integ, scale_factor.value(), true);
696 bc->isBoundaryRestricted()
697 ? form->AddBoundaryIntegrator(std::move(integ), bc->getBoundaryMarkers())
698 : form->AddBoundaryIntegrator(std::move(integ));
699 }
700}
void mooseError(Args &&... args)
Emit an error message with the given stringified, concatenated args and terminate the application.
Definition MooseError.h:311
for(PetscInt i=0;i< nvars;++i)
if(!dmm->_nl) SETERRQ(PETSC_COMM_WORLD

Referenced by BuildNonlinearFormForFESpace(), BuildNonlinearForms(), and Moose::MFEM::TimeDependentEquationSystem::BuildNonlinearForms().

◆ ApplyDomainBLFIntegrators()

template<class FormType >
void Moose::MFEM::EquationSystem::ApplyDomainBLFIntegrators ( const std::string &  trial_var_name,
const std::string &  test_var_name,
std::shared_ptr< FormType >  form,
NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMKernel > > > > &  kernels_map,
std::optional< mfem::real_t >  scale_factor = std::nullopt 
)
protected

Template method for applying BilinearFormIntegrators on domains from kernels to a BilinearForm, or MixedBilinearForm.

Definition at line 352 of file EquationSystem.h.

358{
359 if (kernels_map.Has(test_var_name) && kernels_map.Get(test_var_name)->Has(trial_var_name))
360 {
361 auto kernels = kernels_map.GetRef(test_var_name).GetRef(trial_var_name);
362 for (auto & kernel : kernels)
363 {
364 mfem::BilinearFormIntegrator * integ = kernel->createBFIntegrator();
365
366 if (integ)
367 {
368 if (scale_factor.has_value())
369 integ = new ScaleIntegrator(integ, scale_factor.value(), true);
370 kernel->isDGKernel() ? form->AddInteriorFaceIntegrator(std::move(integ))
371 : kernel->isSubdomainRestricted()
372 ? form->AddDomainIntegrator(std::move(integ), kernel->getSubdomainMarkers())
373 : form->AddDomainIntegrator(std::move(integ));
374 }
375 }
376 }
377}

◆ ApplyDomainLFIntegrators()

void Moose::MFEM::EquationSystem::ApplyDomainLFIntegrators ( const std::string &  test_var_name,
std::shared_ptr< mfem::ParLinearForm >  form,
NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMKernel > > > > &  kernels_map 
)
protected

Apply domain LinearFormIntegrators from kernels to the linear form associated with the supplied test variable.

Definition at line 586 of file EquationSystem.C.

590{
591 if (kernels_map.Has(test_var_name) && kernels_map.Get(test_var_name)->Has(test_var_name))
592 {
593 auto kernels = kernels_map.GetRef(test_var_name).GetRef(test_var_name);
594 for (auto & kernel : kernels)
595 {
596 mfem::LinearFormIntegrator * integ = kernel->createLFIntegrator();
597
598 if (integ)
599 {
600 kernel->isSubdomainRestricted()
601 ? form->AddDomainIntegrator(std::move(integ), kernel->getSubdomainMarkers())
602 : form->AddDomainIntegrator(std::move(integ));
603 }
604 }
605 }
606}

Referenced by BuildLinearForms().

◆ ApplyDomainNLFIntegrators()

void Moose::MFEM::EquationSystem::ApplyDomainNLFIntegrators ( const std::string &  test_var_name,
std::shared_ptr< mfem::ParNonlinearForm >  form,
NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMKernel > > > > &  kernels_map,
std::optional< mfem::real_t >  scale_factor = std::nullopt 
)
protected

Apply domain NonlinearFormIntegrators from kernels to the nonlinear form associated with the supplied test variable.

Definition at line 609 of file EquationSystem.C.

614{
615 if (kernels_map.Has(test_var_name))
616 for (const auto & [trial_var_name, kernels] : kernels_map.GetRef(test_var_name))
617 for (auto & kernel : *kernels)
618 if (auto * integ = kernel->createNLIntegrator())
619 {
620 if (_gradient_required && (trial_var_name != test_var_name))
621 mooseError("Support for off-diagonal MFEM nonlinear domain integrators in conjunction "
622 "with a nonlinear solver that requires a gradient is not currently "
623 "implemented. Kernel '",
624 kernel->name(),
625 "' contributes to test variable '",
626 test_var_name,
627 "' from trial variable '",
628 trial_var_name,
629 "'.");
630
631 _non_linear = true;
632 if (scale_factor.has_value())
633 integ = new NLScaleIntegrator(integ, scale_factor.value(), true);
634 kernel->isSubdomainRestricted()
635 ? form->AddDomainIntegrator(std::move(integ), kernel->getSubdomainMarkers())
636 : form->AddDomainIntegrator(std::move(integ));
637 }
638}

Referenced by BuildNonlinearFormForFESpace(), BuildNonlinearForms(), and Moose::MFEM::TimeDependentEquationSystem::BuildNonlinearForms().

◆ ApplyEssentialBC()

void Moose::MFEM::EquationSystem::ApplyEssentialBC ( const std::string &  var_name,
mfem::ParGridFunction &  trial_gf,
mfem::Array< int > &  global_ess_markers 
)
protectedvirtual

Apply essential BC(s) associated with var_name to set true DoFs of trial_gf and update markers of all essential boundaries.

Definition at line 204 of file EquationSystem.C.

207{
208 if (_essential_bc_map.Has(var_name))
209 for (auto & bc : _essential_bc_map.GetRef(var_name))
210 {
211 // Set constrained DoFs values on essential boundaries
212 bc->ApplyBC(trial_gf);
213 // Fetch marker array labelling essential boundaries of current BC
214 mfem::Array<int> ess_bdrs(bc->getBoundaryMarkers());
215 // Add these boundary markers to the set of markers labelling all essential boundaries
216 for (const auto i : make_range(ess_bdrs.Size()))
217 global_ess_markers[i] |= ess_bdrs[i];
218 }
219}
IntRange< T > make_range(T beg, T end)

Referenced by ApplyEssentialBCs(), and Moose::MFEM::EigenproblemEquationSystem::ApplyEssentialBCs().

◆ ApplyEssentialBCs()

void Moose::MFEM::EquationSystem::ApplyEssentialBCs ( )
protectedvirtual

Update all essentially constrained true DoF markers and values on boundaries.

Reimplemented in Moose::MFEM::ComplexEquationSystem, and Moose::MFEM::EigenproblemEquationSystem.

Definition at line 222 of file EquationSystem.C.

223{
224 _ess_tdof_lists.resize(_trial_var_names.size());
225 _ess_markers.resize(_trial_var_names.size());
226 for (const auto i : index_range(_trial_var_names))
227 {
228 const auto & trial_var_name = _trial_var_names.at(i);
229 mfem::ParGridFunction & trial_gf = *_var_ess_constraints.at(i);
230
231 // Make sure we update the size, if this mesh has changed recently for instance
232 trial_gf.Update();
233
234 // Initial guess for iterative solvers (initial condition or the previous time step solution)
235 trial_gf = _gfuncs->GetRef(trial_var_name);
236
237 _ess_markers.at(i).SetSize(trial_gf.ParFESpace()->GetParMesh()->bdr_attributes.Max(), 0);
238 // Set strongly constrained DoFs of trial_gf on essential boundaries and add markers for all
239 // essential boundaries to the _ess_markers array
240 ApplyEssentialBC(trial_var_name, trial_gf, _ess_markers.at(i));
241 trial_gf.ParFESpace()->GetEssentialTrueDofs(_ess_markers.at(i), _ess_tdof_lists.at(i));
242 }
243}
virtual void ApplyEssentialBC(const std::string &var_name, mfem::ParGridFunction &trial_gf, mfem::Array< int > &global_ess_markers)
Apply essential BC(s) associated with var_name to set true DoFs of trial_gf and update markers of all...
std::vector< mfem::Array< int > > _ess_tdof_lists
std::vector< std::string > _trial_var_names
Subset of _coupled_var_names of all variables corresponding to gridfunctions with degrees of freedom ...
std::vector< std::unique_ptr< mfem::ParGridFunction > > _var_ess_constraints
Gridfunctions holding essential constraints from Dirichlet BCs.
std::vector< mfem::Array< int > > _ess_markers
Moose::MFEM::GridFunctions * _gfuncs
auto index_range(const T &sizable)

Referenced by BuildLinearForms().

◆ BuildBilinearFormForFESpace()

std::shared_ptr< mfem::ParBilinearForm > Moose::MFEM::EquationSystem::BuildBilinearFormForFESpace ( const std::string &  var_name,
mfem::ParFiniteElementSpace &  fespace,
mfem::AssemblyLevel  assembly_level 
)

Build a fresh ParBilinearForm on the given FESpace using the same kernels as the main system's bilinear form for var_name.

Caller owns the returned form.

Definition at line 711 of file EquationSystem.C.

714{
715 auto blf = std::make_shared<mfem::ParBilinearForm>(&fespace);
716 blf->SetAssemblyLevel(assembly_level);
717 ApplyBoundaryBLFIntegrators<mfem::ParBilinearForm>(var_name, var_name, blf, _integrated_bc_map);
718 ApplyDomainBLFIntegrators<mfem::ParBilinearForm>(var_name, var_name, blf, _kernels_map);
719 blf->Assemble();
720 blf->Finalize();
721 return blf;
722}

◆ BuildBilinearForms()

void Moose::MFEM::EquationSystem::BuildBilinearForms ( )
protectedvirtual

Build bilinear forms (diagonal Jacobian contributions)

Reimplemented in Moose::MFEM::ComplexEquationSystem, and Moose::MFEM::TimeDependentEquationSystem.

Definition at line 515 of file EquationSystem.C.

516{
517 // Register bilinear forms
518 for (const auto i : index_range(_test_var_names))
519 {
520 auto test_var_name = _test_var_names.at(i);
521 _blfs.Register(test_var_name, std::make_shared<mfem::ParBilinearForm>(_test_pfespaces.at(i)));
522
523 // Apply kernels
524 auto blf = _blfs.GetShared(test_var_name);
525 blf->SetAssemblyLevel(_assembly_level);
526 ApplyBoundaryBLFIntegrators<mfem::ParBilinearForm>(
527 test_var_name, test_var_name, blf, _integrated_bc_map);
528 ApplyDomainBLFIntegrators<mfem::ParBilinearForm>(
529 test_var_name, test_var_name, blf, _kernels_map);
530 // Assemble
531 blf->Assemble();
532 blf->Finalize();
533 }
534}
std::vector< mfem::ParFiniteElementSpace * > _test_pfespaces
Pointers to finite element spaces associated with test variables.
mfem::AssemblyLevel _assembly_level
NamedFieldsMap< mfem::ParBilinearForm > _blfs
std::shared_ptr< T > GetShared(const std::string &field_name) const
Returns a shared pointer to the field. This is guaranteed to return a non-null shared pointer.

Referenced by BuildEquationSystem().

◆ BuildEquationSystem()

void Moose::MFEM::EquationSystem::BuildEquationSystem ( )
virtual

Build all forms comprising this EquationSystem.

Definition at line 577 of file EquationSystem.C.

578{
583}
virtual void BuildBilinearForms()
Build bilinear forms (diagonal Jacobian contributions)
virtual void BuildNonlinearForms()
Build non-linear action forms.
virtual void BuildMixedBilinearForms()
Build mixed bilinear forms (off-diagonal Jacobian contributions)
virtual void BuildLinearForms()
Build linear forms and eliminate constrained DoFs.

Referenced by Moose::MFEM::EigenproblemESProblemOperator::FormEquationSystemOperator(), and FormSystem().

◆ BuildLinearForms()

void Moose::MFEM::EquationSystem::BuildLinearForms ( )
protectedvirtual

Build linear forms and eliminate constrained DoFs.

Reimplemented in Moose::MFEM::ComplexEquationSystem.

Definition at line 472 of file EquationSystem.C.

473{
474 // Register linear forms
475 for (const auto i : index_range(_test_var_names))
476 {
477 auto test_var_name = _test_var_names.at(i);
478 _lfs.Register(test_var_name, std::make_shared<mfem::ParLinearForm>(_test_pfespaces.at(i)));
479 _lfs.GetRef(test_var_name) = 0.0;
480 }
481
482 for (auto & test_var_name : _test_var_names)
483 {
484 // Apply kernels
485 auto lf = _lfs.GetShared(test_var_name);
486 ApplyDomainLFIntegrators(test_var_name, lf, _kernels_map);
488 lf->Assemble();
489 }
490
491 // Apply essential boundary conditions
493
494 // Eliminate trivially eliminated variables by subtracting contributions from linear forms
496}
virtual void ApplyEssentialBCs()
Update all essentially constrained true DoF markers and values on boundaries.
virtual void EliminateCoupledVariables()
Perform trivial eliminations of coupled variables lacking corresponding test variables.
void ApplyBoundaryLFIntegrators(const std::string &test_var_name, std::shared_ptr< mfem::ParLinearForm > form, NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMIntegratedBC > > > > &integrated_bc_map)
Apply boundary LinearFormIntegrators from integrated boundary conditions to the linear form associate...
NamedFieldsMap< mfem::ParLinearForm > _lfs
void ApplyDomainLFIntegrators(const std::string &test_var_name, std::shared_ptr< mfem::ParLinearForm > form, NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMKernel > > > > &kernels_map)
Apply domain LinearFormIntegrators from kernels to the linear form associated with the supplied test ...

Referenced by BuildEquationSystem().

◆ BuildMixedBilinearForms()

void Moose::MFEM::EquationSystem::BuildMixedBilinearForms ( )
protectedvirtual

Build mixed bilinear forms (off-diagonal Jacobian contributions)

Reimplemented in Moose::MFEM::ComplexEquationSystem, and Moose::MFEM::TimeDependentEquationSystem.

Definition at line 537 of file EquationSystem.C.

538{
539 // Register mixed bilinear forms. Note that not all combinations may
540 // have a kernel.
541
542 // Create mblf for each test/coupled variable pair with an added kernel.
543 // Mixed bilinear forms with coupled variables that are not trial variables are
544 // associated with contributions from eliminated variables.
545 for (const auto i : index_range(_test_var_names))
546 {
547 auto test_var_name = _test_var_names.at(i);
548 auto test_mblfs = std::make_shared<Moose::MFEM::NamedFieldsMap<mfem::ParMixedBilinearForm>>();
549 for (const auto j : index_range(_coupled_var_names))
550 {
551 const auto & coupled_var_name = _coupled_var_names.at(j);
552 auto mblf = std::make_shared<mfem::ParMixedBilinearForm>(_coupled_pfespaces.at(j),
553 _test_pfespaces.at(i));
554 // Register MixedBilinearForm if kernels exist for it, and assemble kernels
555 if (_kernels_map.Has(test_var_name) &&
556 _kernels_map.Get(test_var_name)->Has(coupled_var_name) &&
557 test_var_name != coupled_var_name)
558 {
559 mblf->SetAssemblyLevel(_assembly_level);
560 // Apply all mixed kernels with this test/trial pair
561 ApplyDomainBLFIntegrators<mfem::ParMixedBilinearForm>(
562 coupled_var_name, test_var_name, mblf, _kernels_map);
563 // Assemble mixed bilinear forms
564 mblf->Assemble();
565 mblf->Finalize();
566 // Register mixed bilinear forms associated with a single trial variable
567 // for the current test variable
568 test_mblfs->Register(coupled_var_name, mblf);
569 }
570 }
571 // Register all mixed bilinear form sets associated with a single test variable
572 _mblfs.Register(test_var_name, test_mblfs);
573 }
574}
std::vector< mfem::ParFiniteElementSpace * > _coupled_pfespaces
Pointers to finite element spaces associated with coupled variables.
NamedFieldsMap< NamedFieldsMap< mfem::ParMixedBilinearForm > > _mblfs

Referenced by BuildEquationSystem().

◆ BuildNonlinearFormForFESpace()

std::shared_ptr< mfem::ParNonlinearForm > Moose::MFEM::EquationSystem::BuildNonlinearFormForFESpace ( const std::string &  var_name,
mfem::ParFiniteElementSpace &  fespace,
mfem::AssemblyLevel  assembly_level 
)

Build a fresh ParNonlinearForm on the given FESpace using the same kernels as the main system's nonlinear form for var_name.

Caller owns the returned form.

Definition at line 725 of file EquationSystem.C.

728{
729 auto nlf = std::make_shared<mfem::ParNonlinearForm>(&fespace);
730 ApplyDomainNLFIntegrators(var_name, nlf, _kernels_map, std::nullopt);
731 ApplyBoundaryNLFIntegrators(var_name, nlf, _integrated_bc_map, std::nullopt);
732 return nlf;
733}
void ApplyDomainNLFIntegrators(const std::string &test_var_name, std::shared_ptr< mfem::ParNonlinearForm > form, NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMKernel > > > > &kernels_map, std::optional< mfem::real_t > scale_factor=std::nullopt)
Apply domain NonlinearFormIntegrators from kernels to the nonlinear form associated with the supplied...
void ApplyBoundaryNLFIntegrators(const std::string &test_var_name, std::shared_ptr< mfem::ParNonlinearForm > form, NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMIntegratedBC > > > > &integrated_bc_map, std::optional< mfem::real_t > scale_factor=std::nullopt)
Apply boundary NonlinearFormIntegrators from integrated boundary conditions to the nonlinear form ass...

◆ BuildNonlinearForms()

void Moose::MFEM::EquationSystem::BuildNonlinearForms ( )
protectedvirtual

Build non-linear action forms.

Reimplemented in Moose::MFEM::TimeDependentEquationSystem.

Definition at line 499 of file EquationSystem.C.

500{
501 // Register non-linear Action forms
502 for (const auto i : index_range(_test_var_names))
503 {
504 auto test_var_name = _test_var_names.at(i);
505 _nlfs.Register(test_var_name, std::make_shared<mfem::ParNonlinearForm>(_test_pfespaces.at(i)));
506 // Apply kernels
507 auto nlf = _nlfs.GetShared(test_var_name);
508 nlf->SetEssentialTrueDofs(_ess_tdof_lists.at(i));
509 ApplyDomainNLFIntegrators(test_var_name, nlf, _kernels_map, std::nullopt);
510 ApplyBoundaryNLFIntegrators(test_var_name, nlf, _integrated_bc_map, std::nullopt);
511 }
512}
NamedFieldsMap< mfem::ParNonlinearForm > _nlfs

Referenced by BuildEquationSystem().

◆ ComputeNonlinearResidual()

void Moose::MFEM::EquationSystem::ComputeNonlinearResidual ( const mfem::Vector &  u,
mfem::Vector &  residual 
) const
virtual

Compute the contribution to the residual from nonlinear forms only.

Definition at line 391 of file EquationSystem.C.

392{
393 mooseAssert(IsNonlinear(), "Should not be calling this method if our forms are not nonlinear");
394
395 const mfem::BlockVector block_solution(const_cast<mfem::Vector &>(sol), _block_true_offsets);
396 SetTrialVariablesFromTrueVectors(block_solution);
397
398 mfem::BlockVector block_residual(residual, _block_true_offsets);
399 for (unsigned int i = 0; i < _test_var_names.size(); i++)
400 {
401 auto & test_var_name = _test_var_names.at(i);
402 auto nlf = _nlfs.GetShared(test_var_name);
403 nlf->AddMult(block_solution.GetBlock(i), block_residual.GetBlock(i));
404 block_residual.GetBlock(i).SyncAliasMemory(block_residual);
405 }
406}
mfem::Array< int > _block_true_offsets
virtual void SetTrialVariablesFromTrueVectors(const mfem::BlockVector &trueX) const
Update variable from solution vector after solve.

Referenced by Mult().

◆ DeleteHBlocks()

void Moose::MFEM::EquationSystem::DeleteHBlocks ( )
protected

Deletes the HypreParMatrix associated with any pointer stored in _h_blocks, and then proceeds to delete all dynamically allocated memory for _h_blocks itself, resetting all dimensions to zero.

Definition at line 27 of file EquationSystem.C.

28{
29 for (const auto i : make_range(_h_blocks.NumRows()))
30 for (const auto j : make_range(_h_blocks.NumCols()))
31 {
32 if (_jacobian_blocks.NumRows() && _jacobian_blocks(i, j) == _h_blocks(i, j))
33 _jacobian_blocks(i, j) = nullptr;
34 delete _h_blocks(i, j);
35 }
36 _h_blocks.DeleteAll();
37}
mfem::Array2D< const mfem::HypreParMatrix * > _h_blocks
mfem::Array2D< const mfem::HypreParMatrix * > _jacobian_blocks

Referenced by FormSystemMatrix(), Moose::MFEM::ComplexEquationSystem::FormSystemMatrix(), and ~EquationSystem().

◆ DeleteJacobianBlocks()

void Moose::MFEM::EquationSystem::DeleteJacobianBlocks ( )
protected

Deletes the HypreParMatrix associated with any pointer stored in _jacobian_blocks, and then proceeds to delete all dynamically allocated memory for _jacobian_blocks itself, resetting all dimensions to zero.

Definition at line 40 of file EquationSystem.C.

41{
42 for (const auto i : make_range(_jacobian_blocks.NumRows()))
43 for (const auto j : make_range(_jacobian_blocks.NumCols()))
44 if (!_h_blocks.NumRows() || _jacobian_blocks(i, j) != _h_blocks(i, j))
45 delete _jacobian_blocks(i, j);
46 _jacobian_blocks.DeleteAll();
47}

Referenced by FormJacobianMatrix(), and ~EquationSystem().

◆ EliminateCoupledVariables()

void Moose::MFEM::EquationSystem::EliminateCoupledVariables ( )
protectedvirtual

Perform trivial eliminations of coupled variables lacking corresponding test variables.

Reimplemented in Moose::MFEM::ComplexEquationSystem, and Moose::MFEM::TimeDependentEquationSystem.

Definition at line 246 of file EquationSystem.C.

247{
248 for (const auto & test_var_name : _test_var_names)
249 for (const auto & eliminated_var_name : _eliminated_var_names)
250 if (_mblfs.Has(test_var_name) && _mblfs.Get(test_var_name)->Has(eliminated_var_name) &&
251 !VectorContainsName(_test_var_names, eliminated_var_name))
252 {
253 auto & mblf = *_mblfs.Get(test_var_name)->Get(eliminated_var_name);
254 mblf.AddMult(*_eliminated_variables.Get(eliminated_var_name), *_lfs.Get(test_var_name), -1);
255 }
256}
Moose::MFEM::GridFunctions _eliminated_variables
Pointers to coupled variables not part of the reduced EquationSystem.
T * Get(const std::string &field_name) const
Returns a non-owning pointer to the field. This is guaranteed to return a non-null pointer.

Referenced by BuildLinearForms(), and Moose::MFEM::TimeDependentEquationSystem::EliminateCoupledVariables().

◆ FormJacobianMatrix()

void Moose::MFEM::EquationSystem::FormJacobianMatrix ( const mfem::Vector &  u)
protected

Compute Jacobian matrix at the provided vector of true DoFs of trial variables.

Definition at line 409 of file EquationSystem.C.

410{
412 _jacobian_blocks.SetSize(_test_var_names.size(), _trial_var_names.size());
413 _jacobian_blocks = nullptr;
414
415 const mfem::BlockVector update_vector(const_cast<mfem::Vector &>(u), _block_true_offsets);
416 for (const auto i : index_range(_test_var_names))
417 {
418 auto test_var_name = _test_var_names.at(i);
419 if (_nlfs.Has(test_var_name))
420 {
421 auto nlf = _nlfs.Get(test_var_name);
422 mfem::HypreParMatrix * nlf_jac =
423 dynamic_cast<mfem::HypreParMatrix *>(&nlf->GetGradient(update_vector.GetBlock(i)));
424 mooseAssert(nlf_jac,
425 "Jacobian contribution of nonlinear form associated with " + test_var_name +
426 " is not castable into a HypreParMatrix");
427 _jacobian_blocks(i, i) = mfem::ParAdd(_h_blocks(i, i), nlf_jac);
428 }
429 else
430 _jacobian_blocks(i, i) = _h_blocks(i, i);
431 for (const auto j : index_range(_trial_var_names))
432 if (i != j) // nlf->GetGradient only contributes to on-diagonal blocks
433 _jacobian_blocks(i, j) = _h_blocks(i, j);
434 }
435 // Create monolithic matrix
436 _jacobian.Reset(mfem::HypreParMatrixFromBlocks(_jacobian_blocks));
437}
mfem::OperatorHandle _jacobian

Referenced by GetGradient().

◆ FormLinearSystem()

void Moose::MFEM::EquationSystem::FormLinearSystem ( mfem::OperatorHandle &  op,
mfem::BlockVector &  trueX,
mfem::BlockVector &  trueRHS 
)
protectedvirtual

Form linear components of system based on on- and off-diagonal bilinear form contributions, populate solution and RHS vectors of true DoFs, and apply constraints.

Definition at line 259 of file EquationSystem.C.

262{
263 mooseAssert(_test_var_names.size() == _trial_var_names.size(),
264 "Number of test and trial variables must be the same for block matrix assembly.");
265
266 if (_assembly_level == mfem::AssemblyLevel::LEGACY)
267 FormSystemMatrix(op, trueX, trueRHS);
268 else
269 FormSystemOperator(op, trueX, trueRHS);
270}
virtual void FormSystemOperator(mfem::OperatorHandle &op, mfem::BlockVector &trueX, mfem::BlockVector &trueRHS)
Form matrix-free representation of linear components of system operator.
virtual void FormSystemMatrix(mfem::OperatorHandle &op, mfem::BlockVector &trueX, mfem::BlockVector &trueRHS)
Form matrix representation of linear components of system operator as a HypreParMatrix.

Referenced by FormSystem().

◆ FormSystem()

void Moose::MFEM::EquationSystem::FormSystem ( mfem::BlockVector &  trueX,
mfem::BlockVector &  trueRHS 
)

Build all weak-form components via BuildEquationSystem(), form the constrained linear part of the system, and populate the true-DoF vectors used by the solve.

For nonlinear problems, nonlinear forms are registered here but are not folded into the assembled linear operator. Their residual action is evaluated by Mult(), and any Jacobian contribution is formed at the current nonlinear iterate by GetGradient().

This is the single public entry point for callers that want the equation system prepared for a solve. Subclasses customize the form-building step by overriding BuildEquationSystem().

Definition at line 367 of file EquationSystem.C.

368{
370 height = trueX.Size();
371 width = trueRHS.Size();
372 // Store block offsets
373 _block_true_offsets.SetSize(trueX.NumBlocks() + 1);
374 _block_true_offsets[0] = 0;
375 for (unsigned i = 0; i < _trial_var_names.size(); i++)
376 _block_true_offsets[i + 1] = trueX.BlockSize(i);
377 _block_true_offsets.PartialSum();
378 FormLinearSystem(_linear_operator, trueX, trueRHS);
379}
virtual void BuildEquationSystem()
Build all forms comprising this EquationSystem.
mfem::OperatorHandle _linear_operator
virtual void FormLinearSystem(mfem::OperatorHandle &op, mfem::BlockVector &trueX, mfem::BlockVector &trueRHS)
Form linear components of system based on on- and off-diagonal bilinear form contributions,...

Referenced by Moose::MFEM::EquationSystemProblemOperator::FormEquationSystemOperator(), and Moose::MFEM::TimeDependentEquationSystemProblemOperator::FormEquationSystemOperator().

◆ FormSystemMatrix()

void Moose::MFEM::EquationSystem::FormSystemMatrix ( mfem::OperatorHandle &  op,
mfem::BlockVector &  trueX,
mfem::BlockVector &  trueRHS 
)
protectedvirtual

Form matrix representation of linear components of system operator as a HypreParMatrix.

Used when EquationSystem assembly level is set to 'LEGACY'.

Reimplemented in Moose::MFEM::ComplexEquationSystem.

Definition at line 303 of file EquationSystem.C.

306{
307 // Allocate block operator
309 _h_blocks.SetSize(_test_var_names.size(), _trial_var_names.size());
310 _h_blocks = nullptr;
311 // Zero out RHS and sync memory
312 trueRHS = 0.0;
313 trueRHS.SyncToBlocks();
314
315 for (const auto i : index_range(_test_var_names))
316 {
317 auto test_var_name = _test_var_names.at(i);
318
319 for (const auto j : index_range(_trial_var_names))
320 {
321 auto trial_var_name = _trial_var_names.at(j);
322
323 mfem::Vector aux_x, aux_rhs;
324 mfem::ParLinearForm aux_lf(_test_pfespaces.at(i));
325 mfem::HypreParMatrix * aux_a = new mfem::HypreParMatrix;
326
327 if (test_var_name == trial_var_name)
328 {
329 mooseAssert(i == j, "Trial and test variables must have the same ordering.");
330 auto blf = _blfs.Get(test_var_name);
331 blf->FormLinearSystem(_ess_tdof_lists.at(j),
333 *_lfs.Get(test_var_name),
334 *aux_a,
335 aux_x,
336 aux_rhs,
337 /*copy_interior=*/true);
338 trueX.GetBlock(j) = aux_x;
339 }
340 else if (_mblfs.Has(test_var_name) && _mblfs.Get(test_var_name)->Has(trial_var_name))
341 {
342 auto mblf = _mblfs.Get(test_var_name)->Get(trial_var_name);
343 mblf->FormRectangularLinearSystem(_ess_tdof_lists.at(j),
344 _ess_tdof_lists.at(i),
346 aux_lf = 0.,
347 *aux_a,
348 aux_x,
349 aux_rhs);
350 }
351 else
352 continue;
353
354 trueRHS.GetBlock(i) += aux_rhs;
355 _h_blocks(i, j) = aux_a;
356 }
357 }
358 // Sync memory
359 trueX.SyncFromBlocks();
360 trueRHS.SyncFromBlocks();
361
362 // Create monolithic matrix
363 op.Reset(mfem::HypreParMatrixFromBlocks(_h_blocks));
364}

Referenced by FormLinearSystem().

◆ FormSystemOperator()

void Moose::MFEM::EquationSystem::FormSystemOperator ( mfem::OperatorHandle &  op,
mfem::BlockVector &  trueX,
mfem::BlockVector &  trueRHS 
)
protectedvirtual

Form matrix-free representation of linear components of system operator.

Used when EquationSystem assembly level is set to 'FULL', 'ELEMENT', 'PARTIAL', or 'NONE'.

Reimplemented in Moose::MFEM::ComplexEquationSystem.

Definition at line 273 of file EquationSystem.C.

276{
277 mooseAssert(_test_var_names.size() == 1 && _test_var_names.size() == _trial_var_names.size(),
278 "Non-legacy assembly is only supported for single test and trial variable systems");
279
280 auto & test_var_name = _test_var_names.at(0);
281 mfem::Vector aux_x, aux_rhs;
282 mfem::OperatorPtr aux_a;
283
284 auto blf = _blfs.Get(test_var_name);
285 blf->FormLinearSystem(_ess_tdof_lists.at(0),
287 *_lfs.Get(test_var_name),
288 aux_a,
289 aux_x,
290 aux_rhs,
291 /*copy_interior=*/true);
292
293 trueX.GetBlock(0) = aux_x;
294 trueRHS.GetBlock(0) = aux_rhs;
295 trueX.SyncFromBlocks();
296 trueRHS.SyncFromBlocks();
297
298 op.Reset(aux_a.Ptr());
299 aux_a.SetOperatorOwner(false);
300}

Referenced by FormLinearSystem().

◆ GetBilinearForm()

mfem::ParBilinearForm & Moose::MFEM::EquationSystem::GetBilinearForm ( const std::string &  test_var_name)
inline
Returns
a reference to the MFEM ParBilinearForm corresponding to test_var_name

Definition at line 106 of file EquationSystem.h.

107 {
108 return _blfs.GetRef(test_var_name);
109 }

◆ GetBlockOffsets()

const mfem::Array< int > & Moose::MFEM::EquationSystem::GetBlockOffsets ( ) const
inline

Getter for block true offsets associated with the EquationSystem operator.

Definition at line 101 of file EquationSystem.h.

101{ return _block_true_offsets; }

◆ GetEssentialBoundaryMarkers()

mfem::Array< int > & Moose::MFEM::EquationSystem::GetEssentialBoundaryMarkers ( const std::string &  var_name)

Return the essential boundary attribute marker array for a given trial variable.

The returned array has size == pmesh.bdr_attributes.Max() with 1 at essential boundaries.

Definition at line 736 of file EquationSystem.C.

737{
738 for (const auto i : index_range(_trial_var_names))
739 if (_trial_var_names.at(i) == var_name)
740 return _ess_markers.at(i);
741
742 mooseError("No essential boundary markers found for variable '", var_name, "'.");
743}

◆ GetGradient()

mfem::Operator & Moose::MFEM::EquationSystem::GetGradient ( const mfem::Vector &  u) const
override

Get Jacobian at the provided vector of true DoFs of trial variables.

Definition at line 440 of file EquationSystem.C.

441{
443
444 if (IsNonlinear())
445 {
446 if (_assembly_level != mfem::AssemblyLevel::LEGACY)
447 mooseError("MFEM nonlinear solvers that require GetGradient() currently require legacy "
448 "assembly in EquationSystem.");
449 const_cast<EquationSystem *>(this)->FormJacobianMatrix(u);
450 }
451 else
453
454 return *_jacobian;
455}
void FormJacobianMatrix(const mfem::Vector &u)
Compute Jacobian matrix at the provided vector of true DoFs of trial variables.
const mfem::Vector * _linearization_point

◆ GetGridFunction()

mfem::ParGridFunction & Moose::MFEM::EquationSystem::GetGridFunction ( const std::string &  trial_var_name)
inline
Returns
a reference to the MFEM ParGridFunction corresponding to trial_var_name

Definition at line 114 of file EquationSystem.h.

115 {
116 return _gfuncs->GetRef(trial_var_name);
117 }

◆ GetLinearizationPoint()

const mfem::Vector & Moose::MFEM::EquationSystem::GetLinearizationPoint ( ) const

The true-DoF vector used for the most recent Jacobian linearization.

Definition at line 703 of file EquationSystem.C.

704{
706 mooseError("EquationSystem::GetLinearizationPoint() called before GetGradient().");
707 return *_linearization_point;
708}

◆ GetLinearOperator()

mfem::OperatorHandle & Moose::MFEM::EquationSystem::GetLinearOperator ( ) const
inline

Get operator handle for linear component of system operator.

Definition at line 80 of file EquationSystem.h.

80{ return _linear_operator; };

◆ GetTestVarNames()

const std::vector< std::string > & Moose::MFEM::EquationSystem::GetTestVarNames ( ) const
inline

◆ GetTrialVarNames()

const std::vector< std::string > & Moose::MFEM::EquationSystem::GetTrialVarNames ( ) const
inline

◆ Init()

void Moose::MFEM::EquationSystem::Init ( GridFunctions &  gridfunctions,
ComplexGridFunctions &  cmplx_gridfunctions,
mfem::AssemblyLevel  assembly_level 
)
virtual

Initialise.

Reimplemented in Moose::MFEM::ComplexEquationSystem.

Definition at line 153 of file EquationSystem.C.

156{
157 _assembly_level = assembly_level;
158
159 // Extract which coupled variables are to be trivially eliminated and which are trial variables
161
162 for (auto & test_var_name : _test_var_names)
163 {
164 if (!gridfunctions.Has(test_var_name))
165 {
166 mooseError("MFEM variable ",
167 test_var_name,
168 " requested by equation system during initialization was "
169 "not found in gridfunctions");
170 }
171 // Store pointers to test FESpaces
172 _test_pfespaces.push_back(gridfunctions.Get(test_var_name)->ParFESpace());
173 }
174
175 for (auto & trial_var_name : _trial_var_names)
176 {
177 if (!gridfunctions.Has(trial_var_name))
178 {
179 mooseError("MFEM variable ",
180 trial_var_name,
181 " requested by equation system during initialization was "
182 "not found in gridfunctions");
183 }
184 // Create auxiliary gridfunctions for storing essential constraints from Dirichlet conditions
185 _var_ess_constraints.emplace_back(
186 std::make_unique<mfem::ParGridFunction>(gridfunctions.Get(trial_var_name)->ParFESpace()));
187 }
188
189 // Store pointers to FESpaces of all coupled variables
190 for (auto & coupled_var_name : _coupled_var_names)
191 _coupled_pfespaces.push_back(gridfunctions.Get(coupled_var_name)->ParFESpace());
192
193 // Store pointers to coupled variable GridFunctions that are to be eliminated prior to forming the
194 // jacobian
195 for (auto & eliminated_var_name : _eliminated_var_names)
196 _eliminated_variables.Register(eliminated_var_name,
197 gridfunctions.GetShared(eliminated_var_name));
198
199 // Get a reference to the GridFunctions
200 _gfuncs = &gridfunctions;
201}
virtual void SetTrialVariableNames()
Set trial variable names from subset of coupled variables that have an associated test variable.

◆ IsComplex()

virtual bool Moose::MFEM::EquationSystem::IsComplex ( ) const
inlinevirtual
Returns
Whether this EquationSystem includes complex components

Reimplemented in Moose::MFEM::ComplexEquationSystem.

Definition at line 147 of file EquationSystem.h.

147{ return false; }

◆ IsEigen()

virtual bool Moose::MFEM::EquationSystem::IsEigen ( ) const
inlinevirtual
Returns
Whether this EquationSystem represents an eigenproblem

Reimplemented in Moose::MFEM::EigenproblemEquationSystem.

Definition at line 149 of file EquationSystem.h.

149{ return false; }

◆ IsMultivariate()

bool Moose::MFEM::EquationSystem::IsMultivariate ( ) const
inline
Returns
Whether this is a multivariate (maybe mixed) equation system

Definition at line 153 of file EquationSystem.h.

153{ return _test_var_names.size() > 1; }

◆ IsNonlinear()

bool Moose::MFEM::EquationSystem::IsNonlinear ( ) const
inline
Returns
Whether nonlinear integrators are present in the equation system

Definition at line 155 of file EquationSystem.h.

155{ return _non_linear; }

Referenced by ComputeNonlinearResidual(), GetGradient(), and Mult().

◆ IsTimeDependent()

virtual bool Moose::MFEM::EquationSystem::IsTimeDependent ( ) const
inlinevirtual
Returns
Whether this EquationSystem has time-dependent components

Reimplemented in Moose::MFEM::TimeDependentEquationSystem.

Definition at line 151 of file EquationSystem.h.

151{ return false; }

◆ Mult()

void Moose::MFEM::EquationSystem::Mult ( const mfem::Vector &  u,
mfem::Vector &  residual 
) const
override

Compute residual y = Mu.

Definition at line 382 of file EquationSystem.C.

383{
384 _linear_operator->Mult(sol, residual);
385
386 if (IsNonlinear())
387 ComputeNonlinearResidual(sol, residual);
388}
virtual void ComputeNonlinearResidual(const mfem::Vector &u, mfem::Vector &residual) const
Compute the contribution to the residual from nonlinear forms only.

◆ SetCoefficientManager()

void Moose::MFEM::EquationSystem::SetCoefficientManager ( CoefficientManager &  coefficients)
inline

Set the coefficient manager to notify when trial variables are updated, so that stored projections of solution-dependent coefficients are invalidated.

Definition at line 90 of file EquationSystem.h.

91 {
92 _coefficient_manager = &coefficients;
93 }
CoefficientManager * _coefficient_manager

◆ SetGradientRequired()

void Moose::MFEM::EquationSystem::SetGradientRequired ( bool  requires_gradient)
inline

Set whether an external object (such as a nonlinear solver) requires Jacobian information for this EquationSystem.

Definition at line 86 of file EquationSystem.h.

86{ _gradient_required = requires_gradient; }

◆ SetTrialVariableNames()

void Moose::MFEM::EquationSystem::SetTrialVariableNames ( )
protectedvirtual

Set trial variable names from subset of coupled variables that have an associated test variable.

Definition at line 78 of file EquationSystem.C.

79{
80 // If a coupled variable has an equation associated with it,
81 // add it to the set of trial variables.
82 for (const auto & test_var_name : _test_var_names)
84 _trial_var_names.push_back(test_var_name);
85
86 // Otherwise, add it to the set of eliminated variables.
87 for (const auto & coupled_var_name : _coupled_var_names)
88 if (!VectorContainsName(_test_var_names, coupled_var_name))
89 _eliminated_var_names.push_back(coupled_var_name);
90}

Referenced by Init(), and Moose::MFEM::ComplexEquationSystem::Init().

◆ SetTrialVariablesFromTrueVectors()

void Moose::MFEM::EquationSystem::SetTrialVariablesFromTrueVectors ( const mfem::BlockVector &  trueX) const
virtual

Update variable from solution vector after solve.

Reimplemented in Moose::MFEM::ComplexEquationSystem.

Definition at line 458 of file EquationSystem.C.

459{
460 for (const auto i : index_range(_trial_var_names))
461 {
462 auto & trial_var_name = _trial_var_names.at(i);
463 trueX.GetBlock(i).SyncMemory(trueX);
464 _gfuncs->Get(trial_var_name)->Distribute(&(trueX.GetBlock(i)));
465 }
466 // Solution variables changed: stored projections of solution-dependent coefficients are stale.
469}
void markSolutionChanged()
Notify quadrature function coefficients that solution variables have changed, marking the stored valu...

Referenced by ComputeNonlinearResidual().

◆ VectorContainsName()

bool Moose::MFEM::EquationSystem::VectorContainsName ( const std::vector< std::string > &  the_vector,
const std::string &  name 
) const
protected

Definition at line 50 of file EquationSystem.C.

52{
53 return std::find(the_vector.begin(), the_vector.end(), name) != the_vector.end();
54}

Referenced by AddCoupledVariableNameIfMissing(), AddEliminatedVariableNameIfMissing(), AddTestVariableNameIfMissing(), EliminateCoupledVariables(), Moose::MFEM::ComplexEquationSystem::EliminateCoupledVariables(), and SetTrialVariableNames().

Member Data Documentation

◆ _assembly_level

mfem::AssemblyLevel Moose::MFEM::EquationSystem::_assembly_level
protected

◆ _blfs

NamedFieldsMap<mfem::ParBilinearForm> Moose::MFEM::EquationSystem::_blfs
protected

◆ _block_true_offsets

mfem::Array<int> Moose::MFEM::EquationSystem::_block_true_offsets
protected

◆ _coefficient_manager

CoefficientManager* Moose::MFEM::EquationSystem::_coefficient_manager = nullptr
protected

◆ _coupled_pfespaces

std::vector<mfem::ParFiniteElementSpace *> Moose::MFEM::EquationSystem::_coupled_pfespaces
protected

◆ _coupled_var_names

std::vector<std::string> Moose::MFEM::EquationSystem::_coupled_var_names
protected

◆ _eliminated_var_names

std::vector<std::string> Moose::MFEM::EquationSystem::_eliminated_var_names
protected

◆ _eliminated_variables

Moose::MFEM::GridFunctions Moose::MFEM::EquationSystem::_eliminated_variables
protected

Pointers to coupled variables not part of the reduced EquationSystem.

Definition at line 295 of file EquationSystem.h.

Referenced by EliminateCoupledVariables(), Moose::MFEM::TimeDependentEquationSystem::EliminateCoupledVariables(), and Init().

◆ _ess_markers

std::vector<mfem::Array<int> > Moose::MFEM::EquationSystem::_ess_markers
protected

◆ _ess_tdof_lists

std::vector<mfem::Array<int> > Moose::MFEM::EquationSystem::_ess_tdof_lists
protected

◆ _essential_bc_map

NamedFieldsMap<std::vector<std::shared_ptr<MFEMEssentialBC> > > Moose::MFEM::EquationSystem::_essential_bc_map
protected

Arrays to store essential BCs to act on each component of weak form.

Named according to test variable.

Definition at line 323 of file EquationSystem.h.

Referenced by AddEssentialBC(), and ApplyEssentialBC().

◆ _gfuncs

Moose::MFEM::GridFunctions* Moose::MFEM::EquationSystem::_gfuncs
protected

◆ _gradient_required

bool Moose::MFEM::EquationSystem::_gradient_required = false
protected

◆ _h_blocks

mfem::Array2D<const mfem::HypreParMatrix *> Moose::MFEM::EquationSystem::_h_blocks
protected

◆ _integrated_bc_map

NamedFieldsMap<NamedFieldsMap<std::vector<std::shared_ptr<MFEMIntegratedBC> > > > Moose::MFEM::EquationSystem::_integrated_bc_map
protected

◆ _jacobian

mfem::OperatorHandle Moose::MFEM::EquationSystem::_jacobian
mutableprotected

◆ _jacobian_blocks

mfem::Array2D<const mfem::HypreParMatrix *> Moose::MFEM::EquationSystem::_jacobian_blocks
protected

Definition at line 314 of file EquationSystem.h.

Referenced by DeleteHBlocks(), DeleteJacobianBlocks(), and FormJacobianMatrix().

◆ _kernels_map

NamedFieldsMap<NamedFieldsMap<std::vector<std::shared_ptr<MFEMKernel> > > > Moose::MFEM::EquationSystem::_kernels_map
protected

◆ _lfs

NamedFieldsMap<mfem::ParLinearForm> Moose::MFEM::EquationSystem::_lfs
protected

◆ _linear_operator

mfem::OperatorHandle Moose::MFEM::EquationSystem::_linear_operator
mutableprotected

◆ _linearization_point

const mfem::Vector* Moose::MFEM::EquationSystem::_linearization_point = nullptr
mutableprotected

Definition at line 336 of file EquationSystem.h.

Referenced by GetGradient(), and GetLinearizationPoint().

◆ _mblfs

NamedFieldsMap<NamedFieldsMap<mfem::ParMixedBilinearForm> > Moose::MFEM::EquationSystem::_mblfs
protected

◆ _nlfs

NamedFieldsMap<mfem::ParNonlinearForm> Moose::MFEM::EquationSystem::_nlfs
protected

◆ _non_linear

bool Moose::MFEM::EquationSystem::_non_linear = false
protected

◆ _test_pfespaces

std::vector<mfem::ParFiniteElementSpace *> Moose::MFEM::EquationSystem::_test_pfespaces
protected

◆ _test_var_names

std::vector<std::string> Moose::MFEM::EquationSystem::_test_var_names
protected

◆ _trial_var_names

std::vector<std::string> Moose::MFEM::EquationSystem::_trial_var_names
protected

◆ _var_ess_constraints

std::vector<std::unique_ptr<mfem::ParGridFunction> > Moose::MFEM::EquationSystem::_var_ess_constraints
protected

The documentation for this class was generated from the following files: