https://mooseframework.inl.gov
Public Member Functions | Protected Member Functions | Protected Attributes | Friends | List of all members
Moose::MFEM::TimeDependentEquationSystem Class Reference

Class to store weak form components for time dependent PDEs. More...

#include <TimeDependentEquationSystem.h>

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

Public Member Functions

 TimeDependentEquationSystem (const Moose::MFEM::TimeDerivativeMap &time_derivative_map)
 
virtual void SetTimeStep (mfem::real_t &dt)
 
virtual void AddKernel (std::shared_ptr< MFEMKernel > kernel) override
 Add kernels. More...
 
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. More...
 
virtual void Init (GridFunctions &gridfunctions, ComplexGridFunctions &cmplx_gridfunctions, mfem::AssemblyLevel assembly_level)
 Initialise. More...
 
void FormSystem (mfem::BlockVector &trueX, mfem::BlockVector &trueRHS)
 Assemble the linear part of the operator, assemble the right-hand side, apply essential and eliminated-variable constraints, and populate the true-DoF vectors used by the solve. More...
 
void Mult (const mfem::Vector &u, mfem::Vector &residual) const override
 Compute residual y = Mu. More...
 
virtual void ComputeNonlinearResidual (const mfem::Vector &u, mfem::Vector &residual) const
 Compute the contribution to the residual from nonlinear forms only. More...
 
mfem::Operator & GetGradient (const mfem::Vector &u) const override
 Get Jacobian at the provided vector of true DoFs of trial variables. More...
 
virtual void SetTrialVariablesFromTrueVectors (const mfem::BlockVector &trueX) const
 Update variable from solution vector after solve. More...
 
void SetSolverRequiresGradient (bool requires_gradient)
 Set whether the nonlinear solver driving this equation system requires Jacobian information. More...
 
const std::vector< std::string > & GetTrialVarNames () const
 
const std::vector< std::string > & GetTestVarNames () const
 
bool Nonlinear () const
 
void PrepareLinearSolver (LinearSolverBase &solver)
 Prepare the provided linear solver. More...
 

Protected Member Functions

virtual void BuildBilinearForms () override
 Build bilinear forms (diagonal Jacobian contributions) More...
 
virtual void BuildMixedBilinearForms () override
 Build mixed bilinear forms (off-diagonal Jacobian contributions) More...
 
virtual void BuildNonlinearForms () override
 Build non-linear action forms. More...
 
virtual void EliminateCoupledVariables () override
 Perform trivial eliminations of coupled variables lacking corresponding test variables. More...
 
virtual void AddCoupledVariableNameIfMissing (const std::string &coupled_var_name)
 Add coupled variable to EquationSystem. More...
 
virtual void AddEliminatedVariableNameIfMissing (const std::string &eliminated_var_name)
 Add eliminated variable to EquationSystem. More...
 
virtual void AddTestVariableNameIfMissing (const std::string &test_var_name)
 Add test variable to EquationSystem. More...
 
virtual void SetTrialVariableNames ()
 Set trial variable names from subset of coupled variables that have an associated test variable. More...
 
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. More...
 
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. More...
 
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. More...
 
virtual void ApplyEssentialBCs ()
 Update all essentially constrained true DoF markers and values on boundaries. More...
 
virtual void BuildLinearForms ()
 Build linear forms and eliminate constrained DoFs. More...
 
virtual void BuildEquationSystem ()
 Build all forms comprising this EquationSystem. More...
 
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. More...
 
virtual void FormSystemOperator (mfem::OperatorHandle &op, mfem::BlockVector &trueX, mfem::BlockVector &trueRHS)
 Form matrix-free representation of linear components of system operator. More...
 
virtual void FormSystemMatrix (mfem::OperatorHandle &op, mfem::BlockVector &trueX, mfem::BlockVector &trueRHS)
 Form matrix representation of linear components of system operator as a HypreParMatrix. More...
 
void FormJacobianMatrix (const mfem::Vector &u)
 Compute Jacobian matrix at the provided vector of true DoFs of trial variables. More...
 
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. More...
 
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. More...
 
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. More...
 
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. More...
 
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. More...
 
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. More...
 
virtual bool Complex () const
 Whether this a complex equation system. More...
 

Protected Attributes

mfem::real_t _dt
 Timestep size. More...
 
Moose::MFEM::NamedFieldsMap< Moose::MFEM::NamedFieldsMap< std::vector< std::shared_ptr< MFEMKernel > > > > _td_kernels_map
 
Moose::MFEM::NamedFieldsMap< mfem::ParBilinearForm > _td_blfs
 Containers to store contributions to weak form of the form (F du/dt, v) More...
 
Moose::MFEM::NamedFieldsMap< Moose::MFEM::NamedFieldsMap< mfem::ParMixedBilinearForm > > _td_mblfs
 
const Moose::MFEM::TimeDerivativeMap_time_derivative_map
 Map between variable names and their time derivatives. More...
 
std::vector< std::string > _coupled_var_names
 Names of all trial variables of kernels and boundary conditions added to this EquationSystem. More...
 
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. More...
 
std::vector< std::string > _eliminated_var_names
 Names of all coupled variables without a corresponding test variable. More...
 
Moose::MFEM::GridFunctions _eliminated_variables
 Pointers to coupled variables not part of the reduced EquationSystem. More...
 
std::vector< std::string > _test_var_names
 Names of all test variables corresponding to linear forms in this equation system. More...
 
std::vector< mfem::ParFiniteElementSpace * > _test_pfespaces
 Pointers to finite element spaces associated with test variables. More...
 
std::vector< mfem::ParFiniteElementSpace * > _coupled_pfespaces
 Pointers to finite element spaces associated with coupled variables. More...
 
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. More...
 
std::vector< mfem::Array< int > > _ess_tdof_lists
 
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. More...
 
NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMIntegratedBC > > > > _integrated_bc_map
 Arrays to store integrated BCs to act on each component of weak form. More...
 
NamedFieldsMap< std::vector< std::shared_ptr< MFEMEssentialBC > > > _essential_bc_map
 Arrays to store essential BCs to act on each component of weak form. More...
 
mfem::OperatorHandle _jacobian
 
mfem::OperatorHandle _linear_operator
 
mfem::AssemblyLevel _assembly_level
 
Moose::MFEM::GridFunctions_gfuncs
 
mfem::Array< int_block_true_offsets
 
bool _non_linear = false
 
bool _solver_requires_gradient = false
 

Friends

class TimeDependentEquationSystemProblemOperator
 

Detailed Description

Class to store weak form components for time dependent PDEs.

Definition at line 21 of file TimeDependentEquationSystem.h.

Constructor & Destructor Documentation

◆ TimeDependentEquationSystem()

Moose::MFEM::TimeDependentEquationSystem::TimeDependentEquationSystem ( const Moose::MFEM::TimeDerivativeMap time_derivative_map)

Definition at line 16 of file TimeDependentEquationSystem.C.

18  : _dt(1.0), _time_derivative_map(time_derivative_map)
19 {
20 }
const Moose::MFEM::TimeDerivativeMap & _time_derivative_map
Map between variable names and their time derivatives.

Member Function Documentation

◆ AddCoupledVariableNameIfMissing()

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

Add coupled variable to EquationSystem.

Definition at line 56 of file EquationSystem.C.

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

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

◆ AddEliminatedVariableNameIfMissing()

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

Add eliminated variable to EquationSystem.

Definition at line 63 of file EquationSystem.C.

Referenced by AddKernel().

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

◆ AddEssentialBC()

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

Add BC associated with essentially constrained DoFs on boundaries.

Definition at line 138 of file EquationSystem.C.

Referenced by MFEMProblem::addBoundaryCondition().

139 {
140  const auto & test_var_name = bc->getTestVariableName();
141  AddTestVariableNameIfMissing(test_var_name);
142  // Register new essential bc map if not present for the test variable
143  if (!_essential_bc_map.Has(test_var_name))
144  {
145  auto bcs = std::make_shared<std::vector<std::shared_ptr<MFEMEssentialBC>>>();
146  _essential_bc_map.Register(test_var_name, std::move(bcs));
147  }
148  _essential_bc_map.GetRef(test_var_name).push_back(std::move(bc));
149 }
virtual void AddTestVariableNameIfMissing(const std::string &test_var_name)
Add test variable to EquationSystem.
bool Has(const std::string &field_name) const
Predicate to check if a field is registered with name field_name.
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.
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)
virtualinherited

Definition at line 115 of file EquationSystem.C.

Referenced by MFEMProblem::addBoundaryCondition().

116 {
117  const auto & trial_var_name = bc->getTrialVariableName();
118  const auto & test_var_name = bc->getTestVariableName();
119  AddCoupledVariableNameIfMissing(trial_var_name);
120  AddTestVariableNameIfMissing(test_var_name);
121  // Register new integrated bc map if not present for the test variable
122  if (!_integrated_bc_map.Has(test_var_name))
123  {
124  auto integrated_bc_field_map = std::make_shared<
126  _integrated_bc_map.Register(test_var_name, std::move(integrated_bc_field_map));
127  }
128  // Register new integrated bc map if not present for the test/trial variable pair
129  if (!_integrated_bc_map.Get(test_var_name)->Has(trial_var_name))
130  {
131  auto bcs = std::make_shared<std::vector<std::shared_ptr<MFEMIntegratedBC>>>();
132  _integrated_bc_map.Get(test_var_name)->Register(trial_var_name, std::move(bcs));
133  }
134  _integrated_bc_map.GetRef(test_var_name).Get(trial_var_name)->push_back(std::move(bc));
135 }
virtual void AddTestVariableNameIfMissing(const std::string &test_var_name)
Add test 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.
virtual void AddCoupledVariableNameIfMissing(const std::string &coupled_var_name)
Add coupled variable to EquationSystem.

◆ AddKernel()

void Moose::MFEM::TimeDependentEquationSystem::AddKernel ( std::shared_ptr< MFEMKernel kernel)
overridevirtual

Add kernels.

Reimplemented from Moose::MFEM::EquationSystem.

Definition at line 23 of file TimeDependentEquationSystem.C.

24 {
25  if (!_time_derivative_map.isTimeDerivative(kernel->getTrialVariableName()))
26  {
28  return;
29  }
30 
31  const auto & trial_var_name =
32  _time_derivative_map.getTimeIntegralName(kernel->getTrialVariableName());
33  const auto & test_var_name = kernel->getTestVariableName();
34  AddEliminatedVariableNameIfMissing(trial_var_name);
35  AddTestVariableNameIfMissing(test_var_name);
36  // Register new td kernels map if not present for the test variable
37  if (!_td_kernels_map.Has(test_var_name))
38  {
39  auto kernel_field_map =
40  std::make_shared<Moose::MFEM::NamedFieldsMap<std::vector<std::shared_ptr<MFEMKernel>>>>();
41  _td_kernels_map.Register(test_var_name, std::move(kernel_field_map));
42  }
43  // Register new td kernels map if not present for the test/trial variable pair
44  if (!_td_kernels_map.Get(test_var_name)->Has(trial_var_name))
45  {
46  auto kernels = std::make_shared<std::vector<std::shared_ptr<MFEMKernel>>>();
47  _td_kernels_map.Get(test_var_name)->Register(trial_var_name, std::move(kernels));
48  }
49  _td_kernels_map.GetRef(test_var_name).Get(trial_var_name)->push_back(std::move(kernel));
50 }
virtual void AddTestVariableNameIfMissing(const std::string &test_var_name)
Add test variable to EquationSystem.
virtual void AddKernel(std::shared_ptr< MFEMKernel > kernel)
Add kernels.
bool Has(const std::string &field_name) const
Predicate to check if a field is registered with name field_name.
Moose::MFEM::NamedFieldsMap< Moose::MFEM::NamedFieldsMap< std::vector< std::shared_ptr< MFEMKernel > > > > _td_kernels_map
bool isTimeDerivative(const std::string &time_derivative_var_name) const
const std::string & getTimeIntegralName(const std::string &time_derivative_var_name) const
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...
void Register(const std::string &field_name, FieldArgs &&... args)
Construct new field with name field_name and register.
const Moose::MFEM::TimeDerivativeMap & _time_derivative_map
Map between variable names and their time derivatives.
virtual void AddEliminatedVariableNameIfMissing(const std::string &eliminated_var_name)
Add eliminated variable to EquationSystem.
T & GetRef(const std::string &field_name) const
Returns a reference to a field.

◆ AddTestVariableNameIfMissing()

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

Add test variable to EquationSystem.

Definition at line 70 of file EquationSystem.C.

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

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

◆ 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 
)
protectedinherited

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

Definition at line 311 of file EquationSystem.h.

318 {
319  if (integrated_bc_map.Has(test_var_name) &&
320  integrated_bc_map.Get(test_var_name)->Has(trial_var_name))
321  {
322  auto bcs = integrated_bc_map.GetRef(test_var_name).GetRef(trial_var_name);
323  for (auto & bc : bcs)
324  {
325  mfem::BilinearFormIntegrator * integ = bc->createBFIntegrator();
326 
327  if (integ)
328  {
329  if (scale_factor.has_value())
330  integ = new ScaleIntegrator(integ, scale_factor.value(), true);
331  bc->isBoundaryRestricted()
332  ? form->AddBoundaryIntegrator(std::move(integ), bc->getBoundaryMarkers())
333  : form->AddBoundaryIntegrator(std::move(integ));
334  }
335  }
336  }
337 }

◆ 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 
)
protectedinherited

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

Definition at line 650 of file EquationSystem.C.

Referenced by Moose::MFEM::EquationSystem::BuildLinearForms().

655 {
656  if (integrated_bc_map.Has(test_var_name) &&
657  integrated_bc_map.Get(test_var_name)->Has(test_var_name))
658  {
659  auto bcs = integrated_bc_map.GetRef(test_var_name).GetRef(test_var_name);
660  for (auto & bc : bcs)
661  {
662  mfem::LinearFormIntegrator * integ = bc->createLFIntegrator();
663 
664  if (integ)
665  {
666  bc->isBoundaryRestricted()
667  ? form->AddBoundaryIntegrator(std::move(integ), bc->getBoundaryMarkers())
668  : form->AddBoundaryIntegrator(std::move(integ));
669  }
670  }
671  }
672 }

◆ 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 
)
protectedinherited

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

Definition at line 675 of file EquationSystem.C.

Referenced by BuildNonlinearForms(), and Moose::MFEM::EquationSystem::BuildNonlinearForms().

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

◆ 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 
)
protectedinherited

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

Definition at line 283 of file EquationSystem.h.

289 {
290  if (kernels_map.Has(test_var_name) && kernels_map.Get(test_var_name)->Has(trial_var_name))
291  {
292  auto kernels = kernels_map.GetRef(test_var_name).GetRef(trial_var_name);
293  for (auto & kernel : kernels)
294  {
295  mfem::BilinearFormIntegrator * integ = kernel->createBFIntegrator();
296 
297  if (integ)
298  {
299  if (scale_factor.has_value())
300  integ = new ScaleIntegrator(integ, scale_factor.value(), true);
301  kernel->isSubdomainRestricted()
302  ? form->AddDomainIntegrator(std::move(integ), kernel->getSubdomainMarkers())
303  : form->AddDomainIntegrator(std::move(integ));
304  }
305  }
306  }
307 }

◆ 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 
)
protectedinherited

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

Definition at line 595 of file EquationSystem.C.

Referenced by Moose::MFEM::EquationSystem::BuildLinearForms().

599 {
600  if (kernels_map.Has(test_var_name) && kernels_map.Get(test_var_name)->Has(test_var_name))
601  {
602  auto kernels = kernels_map.GetRef(test_var_name).GetRef(test_var_name);
603  for (auto & kernel : kernels)
604  {
605  mfem::LinearFormIntegrator * integ = kernel->createLFIntegrator();
606 
607  if (integ)
608  {
609  kernel->isSubdomainRestricted()
610  ? form->AddDomainIntegrator(std::move(integ), kernel->getSubdomainMarkers())
611  : form->AddDomainIntegrator(std::move(integ));
612  }
613  }
614  }
615 }

◆ 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 
)
protectedinherited

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

Definition at line 618 of file EquationSystem.C.

Referenced by BuildNonlinearForms(), and Moose::MFEM::EquationSystem::BuildNonlinearForms().

623 {
624  if (kernels_map.Has(test_var_name))
625  for (const auto & [trial_var_name, kernels] : kernels_map.GetRef(test_var_name))
626  for (auto & kernel : *kernels)
627  if (auto * integ = kernel->createNLIntegrator())
628  {
629  if (_solver_requires_gradient && (trial_var_name != test_var_name))
630  mooseError("Support for off-diagonal MFEM nonlinear domain integrators in conjunction "
631  "with a nonlinear solver that requires a gradient is not currently "
632  "implemented. Kernel '",
633  kernel->name(),
634  "' contributes to test variable '",
635  test_var_name,
636  "' from trial variable '",
637  trial_var_name,
638  "'.");
639 
640  _non_linear = true;
641  if (scale_factor.has_value())
642  integ = new NLScaleIntegrator(integ, scale_factor.value(), true);
643  kernel->isSubdomainRestricted()
644  ? form->AddDomainIntegrator(std::move(integ), kernel->getSubdomainMarkers())
645  : form->AddDomainIntegrator(std::move(integ));
646  }
647 }
void mooseError(Args &&... args)
Emit an error message with the given stringified, concatenated args and terminate the application...
Definition: MooseError.h:311

◆ ApplyEssentialBC()

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

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 207 of file EquationSystem.C.

Referenced by Moose::MFEM::EquationSystem::ApplyEssentialBCs().

210 {
211  if (_essential_bc_map.Has(var_name))
212  {
213  auto & bcs = _essential_bc_map.GetRef(var_name);
214  for (auto & bc : bcs)
215  {
216  // Set constrained DoFs values on essential boundaries
217  bc->ApplyBC(trial_gf);
218  // Fetch marker array labelling essential boundaries of current BC
219  mfem::Array<int> ess_bdrs(bc->getBoundaryMarkers());
220  // Add these boundary markers to the set of markers labelling all essential boundaries
221  for (const auto i : make_range(trial_gf.ParFESpace()->GetParMesh()->bdr_attributes.Max()))
222  global_ess_markers[i] = std::max(global_ess_markers[i], ess_bdrs[i]);
223  }
224  }
225 }
bool Has(const std::string &field_name) const
Predicate to check if a field is registered with name field_name.
auto max(const L &left, const R &right)
NamedFieldsMap< std::vector< std::shared_ptr< MFEMEssentialBC > > > _essential_bc_map
Arrays to store essential BCs to act on each component of weak form.
IntRange< T > make_range(T beg, T end)
T & GetRef(const std::string &field_name) const
Returns a reference to a field.

◆ ApplyEssentialBCs()

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

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

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

Definition at line 228 of file EquationSystem.C.

Referenced by Moose::MFEM::EquationSystem::BuildLinearForms().

229 {
230  _ess_tdof_lists.resize(_trial_var_names.size());
231  for (const auto i : index_range(_trial_var_names))
232  {
233  const auto & trial_var_name = _trial_var_names.at(i);
234  mfem::ParGridFunction & trial_gf = *_var_ess_constraints.at(i);
235 
236  // Make sure we update the size, if this mesh has changed recently for instance
237  trial_gf.Update();
238 
239  // Initial guess for iterative solvers (initial condition or the previous time step solution)
240  trial_gf = _gfuncs->GetRef(trial_var_name);
241 
242  mfem::Array<int> global_ess_markers(trial_gf.ParFESpace()->GetParMesh()->bdr_attributes.Max());
243  global_ess_markers = 0;
244  // Set strongly constrained DoFs of trial_gf on essential boundaries and add markers for all
245  // essential boundaries to the global_ess_markers array
246  ApplyEssentialBC(trial_var_name, trial_gf, global_ess_markers);
247  trial_gf.ParFESpace()->GetEssentialTrueDofs(global_ess_markers, _ess_tdof_lists.at(i));
248  }
249 }
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.
T & GetRef(const std::string &field_name) const
Returns a reference to a field.
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...
auto index_range(const T &sizable)
Moose::MFEM::GridFunctions * _gfuncs

◆ BuildBilinearForms()

void Moose::MFEM::TimeDependentEquationSystem::BuildBilinearForms ( )
overrideprotectedvirtual

Build bilinear forms (diagonal Jacobian contributions)

Reimplemented from Moose::MFEM::EquationSystem.

Definition at line 53 of file TimeDependentEquationSystem.C.

54 {
55  // Register bilinear forms
56  for (const auto i : index_range(_test_var_names))
57  {
58  const auto & test_var_name = _test_var_names.at(i);
59 
60  // Apply kernels to blf
61  _blfs.Register(test_var_name, std::make_shared<mfem::ParBilinearForm>(_test_pfespaces.at(i)));
62  auto blf = _blfs.GetShared(test_var_name);
63  blf->SetAssemblyLevel(_assembly_level);
64  ApplyBoundaryBLFIntegrators<mfem::ParBilinearForm>(
65  test_var_name, test_var_name, blf, _integrated_bc_map, _dt);
66  ApplyDomainBLFIntegrators<mfem::ParBilinearForm>(
67  test_var_name, test_var_name, blf, _kernels_map, _dt);
68  // Apply dt*du/dt contributions from the operator on the trial variable
69  ApplyDomainBLFIntegrators<mfem::ParBilinearForm>(
70  test_var_name, test_var_name, blf, _td_kernels_map);
71  // Assemble
72  blf->Assemble();
73 
74  // Apply kernels to td_blf
75  _td_blfs.Register(test_var_name,
76  std::make_shared<mfem::ParBilinearForm>(_test_pfespaces.at(i)));
77  auto td_blf = _td_blfs.GetShared(test_var_name);
78  td_blf->SetAssemblyLevel(_assembly_level);
79  ApplyDomainBLFIntegrators<mfem::ParBilinearForm>(
80  test_var_name, test_var_name, td_blf, _td_kernels_map);
81  // Assemble
82  td_blf->Assemble();
83  }
84 }
NamedFieldsMap< mfem::ParBilinearForm > _blfs
Moose::MFEM::NamedFieldsMap< Moose::MFEM::NamedFieldsMap< std::vector< std::shared_ptr< MFEMKernel > > > > _td_kernels_map
NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMIntegratedBC > > > > _integrated_bc_map
Arrays to store integrated BCs to act on each component of weak form.
mfem::AssemblyLevel _assembly_level
std::vector< mfem::ParFiniteElementSpace * > _test_pfespaces
Pointers to finite element spaces associated with test variables.
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...
NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMKernel > > > > _kernels_map
Arrays to store kernels to act on each component of weak form.
std::vector< std::string > _test_var_names
Names of all test variables corresponding to linear forms in this equation system.
void Register(const std::string &field_name, FieldArgs &&... args)
Construct new field with name field_name and register.
Moose::MFEM::NamedFieldsMap< mfem::ParBilinearForm > _td_blfs
Containers to store contributions to weak form of the form (F du/dt, v)
auto index_range(const T &sizable)

◆ BuildEquationSystem()

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

Build all forms comprising this EquationSystem.

Reimplemented in Moose::MFEM::ComplexEquationSystem.

Definition at line 586 of file EquationSystem.C.

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

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

◆ BuildLinearForms()

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

Build linear forms and eliminate constrained DoFs.

Reimplemented in Moose::MFEM::ComplexEquationSystem.

Definition at line 483 of file EquationSystem.C.

Referenced by Moose::MFEM::EquationSystem::BuildEquationSystem().

484 {
485  // Register linear forms
486  for (const auto i : index_range(_test_var_names))
487  {
488  auto test_var_name = _test_var_names.at(i);
489  _lfs.Register(test_var_name, std::make_shared<mfem::ParLinearForm>(_test_pfespaces.at(i)));
490  _lfs.GetRef(test_var_name) = 0.0;
491  }
492 
493  for (auto & test_var_name : _test_var_names)
494  {
495  // Apply kernels
496  auto lf = _lfs.GetShared(test_var_name);
497  ApplyDomainLFIntegrators(test_var_name, lf, _kernels_map);
499  lf->Assemble();
500  }
501 
502  // Apply essential boundary conditions
504 
505  // Eliminate trivially eliminated variables by subtracting contributions from linear forms
507 }
virtual void EliminateCoupledVariables()
Perform trivial eliminations of coupled variables lacking corresponding test variables.
NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMIntegratedBC > > > > _integrated_bc_map
Arrays to store integrated BCs to act on each component of weak form.
std::vector< mfem::ParFiniteElementSpace * > _test_pfespaces
Pointers to finite element spaces associated with test variables.
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...
NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMKernel > > > > _kernels_map
Arrays to store kernels to act on each component of weak form.
std::vector< std::string > _test_var_names
Names of all test variables corresponding to linear forms in this equation system.
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...
void Register(const std::string &field_name, FieldArgs &&... args)
Construct new field with name field_name and register.
NamedFieldsMap< mfem::ParLinearForm > _lfs
virtual void ApplyEssentialBCs()
Update all essentially constrained true DoF markers and values on boundaries.
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 ...
T & GetRef(const std::string &field_name) const
Returns a reference to a field.
auto index_range(const T &sizable)

◆ BuildMixedBilinearForms()

void Moose::MFEM::TimeDependentEquationSystem::BuildMixedBilinearForms ( )
overrideprotectedvirtual

Build mixed bilinear forms (off-diagonal Jacobian contributions)

Reimplemented from Moose::MFEM::EquationSystem.

Definition at line 87 of file TimeDependentEquationSystem.C.

88 {
89  // Register mixed bilinear forms. Note that not all combinations may
90  // have a kernel.
91 
92  // Create mblf for each test/coupled variable pair with an added kernel.
93  // Mixed bilinear forms with coupled variables that are not trial variables are
94  // associated with contributions from eliminated variables.
95  for (const auto i : index_range(_test_var_names))
96  {
97  const auto & test_var_name = _test_var_names.at(i);
98  auto test_mblfs = std::make_shared<Moose::MFEM::NamedFieldsMap<mfem::ParMixedBilinearForm>>();
99  for (const auto j : index_range(_coupled_var_names))
100  {
101  const auto & coupled_var_name = _coupled_var_names.at(j);
102  auto mblf = std::make_shared<mfem::ParMixedBilinearForm>(_coupled_pfespaces.at(j),
103  _test_pfespaces.at(i));
104  // Register MixedBilinearForm if kernels exist for it, and assemble kernels
105  if (test_var_name != coupled_var_name)
106  {
107  // Apply all mixed kernels with this test/trial pair
108  ApplyBoundaryBLFIntegrators<mfem::ParMixedBilinearForm>(
109  coupled_var_name, test_var_name, mblf, _integrated_bc_map, _dt);
110  ApplyDomainBLFIntegrators<mfem::ParMixedBilinearForm>(
111  coupled_var_name, test_var_name, mblf, _kernels_map, _dt);
112  // Apply dt*du/dt contributions from the operator on the trial variable
113  ApplyDomainBLFIntegrators<mfem::ParMixedBilinearForm>(
114  coupled_var_name, test_var_name, mblf, _td_kernels_map);
115  if (mblf->GetDBFI()->Size() || mblf->GetBBFI()->Size())
116  {
117  // Assemble mixed bilinear forms
118  mblf->SetAssemblyLevel(_assembly_level);
119  mblf->Assemble();
120  // Register mixed bilinear forms associated with a single trial variable
121  // for the current test variable
122  test_mblfs->Register(coupled_var_name, mblf);
123  }
124  }
125  }
126  // Register all mixed bilinear form sets associated with a single test variable
127  _mblfs.Register(test_var_name, test_mblfs);
128  }
129 
130  // Register mixed bilinear forms. Note that not all combinations may
131  // have a kernel.
132 
133  // Create mblf for each test/trial variable pair with an added kernel
134  for (const auto i : index_range(_test_var_names))
135  {
136  const auto & test_var_name = _test_var_names.at(i);
137  auto test_td_mblfs =
138  std::make_shared<Moose::MFEM::NamedFieldsMap<mfem::ParMixedBilinearForm>>();
139  for (const auto j : index_range(_trial_var_names))
140  {
141  const auto & trial_var_name = _trial_var_names.at(j);
142  auto td_mblf = std::make_shared<mfem::ParMixedBilinearForm>(_test_pfespaces.at(j),
143  _test_pfespaces.at(i));
144  // Register MixedBilinearForm if kernels exist for it, and assemble kernels
145  if (test_var_name != trial_var_name)
146  {
147  // Apply all mixed kernels with this test/trial pair
148  ApplyDomainBLFIntegrators<mfem::ParMixedBilinearForm>(
149  trial_var_name, test_var_name, td_mblf, _td_kernels_map);
150  // Assemble mixed bilinear form
151  if (td_mblf->GetDBFI()->Size() || td_mblf->GetBBFI()->Size())
152  {
153  td_mblf->SetAssemblyLevel(_assembly_level);
154  td_mblf->Assemble();
155  // Register mixed bilinear forms associated with a single trial variable
156  // for the current test variable
157  test_td_mblfs->Register(trial_var_name, td_mblf);
158  }
159  }
160  }
161  // Register all mixed bilinear forms associated with a single test variable
162  _td_mblfs.Register(test_var_name, test_td_mblfs);
163  }
164 }
NamedFieldsMap< NamedFieldsMap< mfem::ParMixedBilinearForm > > _mblfs
std::vector< mfem::ParFiniteElementSpace * > _coupled_pfespaces
Pointers to finite element spaces associated with coupled variables.
Moose::MFEM::NamedFieldsMap< Moose::MFEM::NamedFieldsMap< std::vector< std::shared_ptr< MFEMKernel > > > > _td_kernels_map
NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMIntegratedBC > > > > _integrated_bc_map
Arrays to store integrated BCs to act on each component of weak form.
Moose::MFEM::NamedFieldsMap< Moose::MFEM::NamedFieldsMap< mfem::ParMixedBilinearForm > > _td_mblfs
mfem::AssemblyLevel _assembly_level
std::vector< mfem::ParFiniteElementSpace * > _test_pfespaces
Pointers to finite element spaces associated with test variables.
std::vector< std::string > _trial_var_names
Subset of _coupled_var_names of all variables corresponding to gridfunctions with degrees of freedom ...
NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMKernel > > > > _kernels_map
Arrays to store kernels to act on each component of weak form.
std::vector< std::string > _test_var_names
Names of all test variables corresponding to linear forms in this equation system.
std::vector< std::string > _coupled_var_names
Names of all trial variables of kernels and boundary conditions added to this EquationSystem.
void Register(const std::string &field_name, FieldArgs &&... args)
Construct new field with name field_name and register.
auto index_range(const T &sizable)

◆ BuildNonlinearForms()

void Moose::MFEM::TimeDependentEquationSystem::BuildNonlinearForms ( )
overrideprotectedvirtual

Build non-linear action forms.

Reimplemented from Moose::MFEM::EquationSystem.

Definition at line 167 of file TimeDependentEquationSystem.C.

168 {
169  // Register non-linear Action forms
170  for (const auto i : index_range(_test_var_names))
171  {
172  auto test_var_name = _test_var_names.at(i);
173  _nlfs.Register(test_var_name, std::make_shared<mfem::ParNonlinearForm>(_test_pfespaces.at(i)));
174  // Apply kernels
175  auto nlf = _nlfs.GetShared(test_var_name);
176  nlf->SetEssentialTrueDofs(_ess_tdof_lists.at(i));
177  ApplyDomainNLFIntegrators(test_var_name, nlf, _kernels_map, _dt);
178  ApplyBoundaryNLFIntegrators(test_var_name, nlf, _integrated_bc_map, _dt);
179  }
180 }
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...
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< mfem::ParNonlinearForm > _nlfs
std::vector< mfem::Array< int > > _ess_tdof_lists
std::vector< mfem::ParFiniteElementSpace * > _test_pfespaces
Pointers to finite element spaces associated with test variables.
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...
NamedFieldsMap< NamedFieldsMap< std::vector< std::shared_ptr< MFEMKernel > > > > _kernels_map
Arrays to store kernels to act on each component of weak form.
std::vector< std::string > _test_var_names
Names of all test variables corresponding to linear forms in this equation system.
void Register(const std::string &field_name, FieldArgs &&... args)
Construct new field with name field_name and register.
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...
auto index_range(const T &sizable)

◆ Complex()

virtual bool Moose::MFEM::EquationSystem::Complex ( ) const
inlineprotectedvirtualinherited

Whether this a complex equation system.

Reimplemented in Moose::MFEM::ComplexEquationSystem.

Definition at line 218 of file EquationSystem.h.

Referenced by Moose::MFEM::EquationSystem::PrepareLinearSolver().

218 { return false; }

◆ ComputeNonlinearResidual()

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

Compute the contribution to the residual from nonlinear forms only.

Definition at line 406 of file EquationSystem.C.

Referenced by Moose::MFEM::EquationSystem::Mult().

407 {
408  mooseAssert(_non_linear, "Should not be calling this method if our forms are not nonlinear");
409  residual = 0.0;
410 
411  const mfem::BlockVector block_solution(const_cast<mfem::Vector &>(sol), _block_true_offsets);
412  SetTrialVariablesFromTrueVectors(block_solution);
413 
414  mfem::BlockVector block_residual(residual, _block_true_offsets);
415  for (unsigned int i = 0; i < _test_var_names.size(); i++)
416  {
417  auto & test_var_name = _test_var_names.at(i);
418  auto nlf = _nlfs.GetShared(test_var_name);
419  nlf->Mult(block_solution.GetBlock(i), block_residual.GetBlock(i));
420  block_residual.GetBlock(i).SyncAliasMemory(block_residual);
421  }
422 }
virtual void SetTrialVariablesFromTrueVectors(const mfem::BlockVector &trueX) const
Update variable from solution vector after solve.
NamedFieldsMap< mfem::ParNonlinearForm > _nlfs
mfem::Array< int > _block_true_offsets
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...
std::vector< std::string > _test_var_names
Names of all test variables corresponding to linear forms in this equation system.

◆ DeleteHBlocks()

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

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 26 of file EquationSystem.C.

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

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

◆ DeleteJacobianBlocks()

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

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 39 of file EquationSystem.C.

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

40 {
41  for (const auto i : make_range(_jacobian_blocks.NumRows()))
42  for (const auto j : make_range(_jacobian_blocks.NumCols()))
43  if (!_h_blocks.NumRows() || _jacobian_blocks(i, j) != _h_blocks(i, j))
44  delete _jacobian_blocks(i, j);
45  _jacobian_blocks.DeleteAll();
46 }
mfem::Array2D< const mfem::HypreParMatrix * > _h_blocks
IntRange< T > make_range(T beg, T end)
mfem::Array2D< const mfem::HypreParMatrix * > _jacobian_blocks

◆ EliminateCoupledVariables()

void Moose::MFEM::TimeDependentEquationSystem::EliminateCoupledVariables ( )
overrideprotectedvirtual

Perform trivial eliminations of coupled variables lacking corresponding test variables.

Reimplemented from Moose::MFEM::EquationSystem.

Definition at line 183 of file TimeDependentEquationSystem.C.

184 {
185  for (const auto & test_var_name : _test_var_names)
186  {
187  auto & lf = *_lfs.Get(test_var_name) *= _dt;
188  for (const auto & eliminated_var_name : _eliminated_var_names)
189  if (eliminated_var_name == test_var_name)
190  {
191  // if implicit, add contribution to linear form from terms involving state
192  // The AddMult method in mfem::BilinearForm is not defined for non-legacy assembly
193  mfem::Vector lf_prev(lf.Size());
194  auto & td_blf = *_td_blfs.Get(test_var_name);
195  td_blf.Mult(*_eliminated_variables.Get(test_var_name), lf_prev);
196  lf += lf_prev;
197  }
198  else if (_td_mblfs.Has(test_var_name) &&
199  _td_mblfs.Get(test_var_name)->Has(eliminated_var_name))
200  {
201  auto & td_mblf = *_td_mblfs.Get(test_var_name)->Get(eliminated_var_name);
202  td_mblf.AddMult(*_eliminated_variables.Get(eliminated_var_name), lf);
203  }
204  }
205  // Eliminate contributions from other coupled variables.
207 }
virtual void EliminateCoupledVariables()
Perform trivial eliminations of coupled variables lacking corresponding test variables.
bool Has(const std::string &field_name) const
Predicate to check if a field is registered with name field_name.
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.
Moose::MFEM::NamedFieldsMap< Moose::MFEM::NamedFieldsMap< mfem::ParMixedBilinearForm > > _td_mblfs
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...
std::vector< std::string > _test_var_names
Names of all test variables corresponding to linear forms in this equation system.
Moose::MFEM::NamedFieldsMap< mfem::ParBilinearForm > _td_blfs
Containers to store contributions to weak form of the form (F du/dt, v)
NamedFieldsMap< mfem::ParLinearForm > _lfs

◆ FormJacobianMatrix()

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

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

Definition at line 425 of file EquationSystem.C.

Referenced by Moose::MFEM::EquationSystem::GetGradient().

426 {
428  _jacobian_blocks.SetSize(_test_var_names.size(), _trial_var_names.size());
429  _jacobian_blocks = nullptr;
430 
431  const mfem::BlockVector update_vector(const_cast<mfem::Vector &>(u), _block_true_offsets);
432  for (const auto i : index_range(_test_var_names))
433  {
434  auto test_var_name = _test_var_names.at(i);
435  if (_nlfs.Has(test_var_name))
436  {
437  auto nlf = _nlfs.Get(test_var_name);
438  mfem::HypreParMatrix * nlf_jac =
439  dynamic_cast<mfem::HypreParMatrix *>(&nlf->GetGradient(update_vector.GetBlock(i)));
440  mooseAssert(nlf_jac,
441  "Jacobian contribution of nonlinear form associated with " + test_var_name +
442  " is not castable into a HypreParMatrix");
443  _jacobian_blocks(i, i) = mfem::ParAdd(_h_blocks(i, i), nlf_jac);
444  }
445  else
446  _jacobian_blocks(i, i) = _h_blocks(i, i);
447  for (const auto j : index_range(_trial_var_names))
448  if (i != j) // nlf->GetGradient only contributes to on-diagonal blocks
449  _jacobian_blocks(i, j) = _h_blocks(i, j);
450  }
451  // Create monolithic matrix
452  _jacobian.Reset(mfem::HypreParMatrixFromBlocks(_jacobian_blocks));
453 }
bool Has(const std::string &field_name) const
Predicate to check if a field is registered with name field_name.
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.
NamedFieldsMap< mfem::ParNonlinearForm > _nlfs
mfem::Array< int > _block_true_offsets
std::vector< std::string > _trial_var_names
Subset of _coupled_var_names of all variables corresponding to gridfunctions with degrees of freedom ...
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...
mfem::OperatorHandle _jacobian
std::vector< std::string > _test_var_names
Names of all test variables corresponding to linear forms in this equation system.
mfem::Array2D< const mfem::HypreParMatrix * > _h_blocks
auto index_range(const T &sizable)
mfem::Array2D< const mfem::HypreParMatrix * > _jacobian_blocks

◆ FormLinearSystem()

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

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 265 of file EquationSystem.C.

Referenced by Moose::MFEM::EquationSystem::FormSystem().

268 {
269  mooseAssert(_test_var_names.size() == _trial_var_names.size(),
270  "Number of test and trial variables must be the same for block matrix assembly.");
271 
272  if (_assembly_level == mfem::AssemblyLevel::LEGACY)
273  FormSystemMatrix(op, trueX, trueRHS);
274  else
275  {
276  mooseAssert(_test_var_names.size() == 1 && _test_var_names.size() == _trial_var_names.size(),
277  "Non-legacy assembly is only supported for single test and trial variable systems");
278  FormSystemOperator(op, trueX, trueRHS);
279  }
280 }
mfem::AssemblyLevel _assembly_level
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::string > _test_var_names
Names of all test variables corresponding to linear forms in this equation system.
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.

◆ FormSystem()

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

Assemble the linear part of the operator, assemble the right-hand side, apply essential and eliminated-variable constraints, and populate the true-DoF vectors used by the solve.

Definition at line 374 of file EquationSystem.C.

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

375 {
376  height = trueX.Size();
377  width = trueRHS.Size();
378  // Store block offsets
379  _block_true_offsets.SetSize(trueX.NumBlocks() + 1);
380  _block_true_offsets[0] = 0;
381  for (unsigned i = 0; i < _trial_var_names.size(); i++)
382  _block_true_offsets[i + 1] = trueX.BlockSize(i);
383  _block_true_offsets.PartialSum();
384  FormLinearSystem(_linear_operator, trueX, trueRHS);
385 }
mfem::Array< int > _block_true_offsets
std::vector< std::string > _trial_var_names
Subset of _coupled_var_names of all variables corresponding to gridfunctions with degrees of freedom ...
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.
mfem::OperatorHandle _linear_operator

◆ FormSystemMatrix()

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

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 310 of file EquationSystem.C.

Referenced by Moose::MFEM::EquationSystem::FormLinearSystem().

313 {
314  // Allocate block operator
315  DeleteHBlocks();
316  _h_blocks.SetSize(_test_var_names.size(), _trial_var_names.size());
317  _h_blocks = nullptr;
318  // Zero out RHS and sync memory
319  trueRHS = 0.0;
320  trueRHS.SyncToBlocks();
321 
322  for (const auto i : index_range(_test_var_names))
323  {
324  auto test_var_name = _test_var_names.at(i);
325 
326  for (const auto j : index_range(_trial_var_names))
327  {
328  auto trial_var_name = _trial_var_names.at(j);
329 
330  mfem::Vector aux_x, aux_rhs;
331  mfem::ParLinearForm aux_lf(_test_pfespaces.at(i));
332  mfem::HypreParMatrix * aux_a = new mfem::HypreParMatrix;
333 
334  if (test_var_name == trial_var_name)
335  {
336  mooseAssert(i == j, "Trial and test variables must have the same ordering.");
337  auto blf = _blfs.Get(test_var_name);
338  blf->FormLinearSystem(_ess_tdof_lists.at(j),
339  *_var_ess_constraints.at(j),
340  *_lfs.Get(test_var_name),
341  *aux_a,
342  aux_x,
343  aux_rhs,
344  /*copy_interior=*/true);
345  trueX.GetBlock(j) = aux_x;
346  }
347  else if (_mblfs.Has(test_var_name) && _mblfs.Get(test_var_name)->Has(trial_var_name))
348  {
349  auto mblf = _mblfs.Get(test_var_name)->Get(trial_var_name);
350  mblf->FormRectangularLinearSystem(_ess_tdof_lists.at(j),
351  _ess_tdof_lists.at(i),
352  *_var_ess_constraints.at(j),
353  aux_lf = 0,
354  *aux_a,
355  aux_x,
356  aux_rhs);
357  }
358  else
359  continue;
360 
361  trueRHS.GetBlock(i) += aux_rhs;
362  _h_blocks(i, j) = aux_a;
363  }
364  }
365  // Sync memory
366  trueX.SyncFromBlocks();
367  trueRHS.SyncFromBlocks();
368 
369  // Create monolithic matrix
370  op.Reset(mfem::HypreParMatrixFromBlocks(_h_blocks));
371 }
NamedFieldsMap< mfem::ParBilinearForm > _blfs
NamedFieldsMap< NamedFieldsMap< mfem::ParMixedBilinearForm > > _mblfs
std::vector< mfem::Array< int > > _ess_tdof_lists
std::vector< mfem::ParFiniteElementSpace * > _test_pfespaces
Pointers to finite element spaces associated with test variables.
std::vector< std::string > _trial_var_names
Subset of _coupled_var_names of all variables corresponding to gridfunctions with degrees of freedom ...
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...
std::vector< std::string > _test_var_names
Names of all test variables corresponding to linear forms in this equation system.
std::vector< std::unique_ptr< mfem::ParGridFunction > > _var_ess_constraints
Gridfunctions holding essential constraints from Dirichlet BCs.
mfem::Array2D< const mfem::HypreParMatrix * > _h_blocks
void DeleteHBlocks()
Deletes the HypreParMatrix associated with any pointer stored in _h_blocks, and then proceeds to dele...
NamedFieldsMap< mfem::ParLinearForm > _lfs
auto index_range(const T &sizable)

◆ FormSystemOperator()

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

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 283 of file EquationSystem.C.

Referenced by Moose::MFEM::EquationSystem::FormLinearSystem().

286 {
287  auto & test_var_name = _test_var_names.at(0);
288  mfem::Vector aux_x, aux_rhs;
289  mfem::OperatorPtr aux_a;
290 
291  auto blf = _blfs.Get(test_var_name);
292  blf->FormLinearSystem(_ess_tdof_lists.at(0),
293  *_var_ess_constraints.at(0),
294  *_lfs.Get(test_var_name),
295  aux_a,
296  aux_x,
297  aux_rhs,
298  /*copy_interior=*/true);
299 
300  trueX.GetBlock(0) = aux_x;
301  trueRHS.GetBlock(0) = aux_rhs;
302  trueX.SyncFromBlocks();
303  trueRHS.SyncFromBlocks();
304 
305  op.Reset(aux_a.Ptr());
306  aux_a.SetOperatorOwner(false);
307 }
NamedFieldsMap< mfem::ParBilinearForm > _blfs
std::vector< mfem::Array< int > > _ess_tdof_lists
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...
std::vector< std::string > _test_var_names
Names of all test variables corresponding to linear forms in this equation system.
std::vector< std::unique_ptr< mfem::ParGridFunction > > _var_ess_constraints
Gridfunctions holding essential constraints from Dirichlet BCs.
NamedFieldsMap< mfem::ParLinearForm > _lfs

◆ GetGradient()

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

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

Definition at line 456 of file EquationSystem.C.

457 {
458  if (_non_linear)
459  {
460  if (_assembly_level != mfem::AssemblyLevel::LEGACY)
461  mooseError("MFEM nonlinear solvers that require GetGradient() currently require legacy "
462  "assembly in EquationSystem.");
463  const_cast<EquationSystem *>(this)->FormJacobianMatrix(u);
464  }
465  else
467 
468  return *_jacobian;
469 }
void mooseError(Args &&... args)
Emit an error message with the given stringified, concatenated args and terminate the application...
Definition: MooseError.h:311
void FormJacobianMatrix(const mfem::Vector &u)
Compute Jacobian matrix at the provided vector of true DoFs of trial variables.
mfem::AssemblyLevel _assembly_level
mfem::OperatorHandle _jacobian
mfem::OperatorHandle _linear_operator

◆ GetTestVarNames()

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

Definition at line 74 of file EquationSystem.h.

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

74 { return _test_var_names; }
std::vector< std::string > _test_var_names
Names of all test variables corresponding to linear forms in this equation system.

◆ GetTrialVarNames()

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

Definition at line 73 of file EquationSystem.h.

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

73 { return _trial_var_names; }
std::vector< std::string > _trial_var_names
Subset of _coupled_var_names of all variables corresponding to gridfunctions with degrees of freedom ...

◆ Init()

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

Initialise.

Reimplemented in Moose::MFEM::ComplexEquationSystem.

Definition at line 152 of file EquationSystem.C.

155 {
156  _assembly_level = assembly_level;
157 
158  if (cmplx_gridfunctions.size())
159  mooseError("Complex variables have been created but the executioner numeric type has not been "
160  "set to complex. Please set Executioner/numeric_type = complex.");
161 
162  // Extract which coupled variables are to be trivially eliminated and which are trial variables
164 
165  for (auto & test_var_name : _test_var_names)
166  {
167  if (!gridfunctions.Has(test_var_name))
168  {
169  mooseError("MFEM variable ",
170  test_var_name,
171  " requested by equation system during initialization was "
172  "not found in gridfunctions");
173  }
174  // Store pointers to test FESpaces
175  _test_pfespaces.push_back(gridfunctions.Get(test_var_name)->ParFESpace());
176  }
177 
178  for (auto & trial_var_name : _trial_var_names)
179  {
180  if (!gridfunctions.Has(trial_var_name))
181  {
182  mooseError("MFEM variable ",
183  trial_var_name,
184  " requested by equation system during initialization was "
185  "not found in gridfunctions");
186  }
187  // Create auxiliary gridfunctions for storing essential constraints from Dirichlet conditions
188  _var_ess_constraints.emplace_back(
189  std::make_unique<mfem::ParGridFunction>(gridfunctions.Get(trial_var_name)->ParFESpace()));
190  }
191 
192  // Store pointers to FESpaces of all coupled variables
193  for (auto & coupled_var_name : _coupled_var_names)
194  _coupled_pfespaces.push_back(gridfunctions.Get(coupled_var_name)->ParFESpace());
195 
196  // Store pointers to coupled variable GridFunctions that are to be eliminated prior to forming the
197  // jacobian
198  for (auto & eliminated_var_name : _eliminated_var_names)
199  _eliminated_variables.Register(eliminated_var_name,
200  gridfunctions.GetShared(eliminated_var_name));
201 
202  // Get a reference to the GridFunctions
203  _gfuncs = &gridfunctions;
204 }
std::vector< mfem::ParFiniteElementSpace * > _coupled_pfespaces
Pointers to finite element spaces associated with coupled variables.
void mooseError(Args &&... args)
Emit an error message with the given stringified, concatenated args and terminate the application...
Definition: MooseError.h:311
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.
mfem::AssemblyLevel _assembly_level
std::vector< mfem::ParFiniteElementSpace * > _test_pfespaces
Pointers to finite element spaces associated with test variables.
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::string > _test_var_names
Names of all test variables corresponding to linear forms in this equation system.
std::vector< std::unique_ptr< mfem::ParGridFunction > > _var_ess_constraints
Gridfunctions holding essential constraints from Dirichlet BCs.
virtual void SetTrialVariableNames()
Set trial variable names from subset of coupled variables that have an associated test variable...
std::vector< std::string > _coupled_var_names
Names of all trial variables of kernels and boundary conditions added to this EquationSystem.
void Register(const std::string &field_name, FieldArgs &&... args)
Construct new field with name field_name and register.
Moose::MFEM::GridFunctions * _gfuncs

◆ Mult()

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

Compute residual y = Mu.

Definition at line 388 of file EquationSystem.C.

389 {
390  if (_non_linear)
391  {
392  ComputeNonlinearResidual(sol, residual);
393  _linear_operator->AddMult(sol, residual);
394  }
395  else
396  {
397  residual = 0.0;
398  _linear_operator->Mult(sol, residual);
399  }
400 
401  sol.HostRead();
402  residual.HostRead();
403 }
mfem::OperatorHandle _linear_operator
virtual void ComputeNonlinearResidual(const mfem::Vector &u, mfem::Vector &residual) const
Compute the contribution to the residual from nonlinear forms only.

◆ Nonlinear()

bool Moose::MFEM::EquationSystem::Nonlinear ( ) const
inlineinherited
Returns
Whether nonlinear integrators are present

Definition at line 79 of file EquationSystem.h.

Referenced by Moose::MFEM::ProblemOperatorBase::SolveWithOperator().

◆ PrepareLinearSolver()

void Moose::MFEM::EquationSystem::PrepareLinearSolver ( LinearSolverBase solver)
inherited

Prepare the provided linear solver.

First calls SetupLOR on the solver if it's using a Low Order Refined methodology and then calls SetOperator on the solver with the assembled linear operator

Definition at line 709 of file EquationSystem.C.

Referenced by Moose::MFEM::ProblemOperatorBase::SolveWithOperator().

710 {
711  if (solver.IsLOR())
712  {
713  if (Complex())
714  mooseError("LOR solve is not supported for complex equation systems.");
715  if (_test_var_names.size() > 1)
716  mooseError("LOR solve is only supported for single-variable systems");
717  solver.SetupLOR(*_blfs.Get(_test_var_names.at(0)), _ess_tdof_lists.at(0));
718  }
719 
720  mooseAssert(_linear_operator.Ptr(),
721  "If we are preparing a linear solver, we better have a linear operator");
722  solver.SetOperator(_linear_operator);
723 }
NamedFieldsMap< mfem::ParBilinearForm > _blfs
void mooseError(Args &&... args)
Emit an error message with the given stringified, concatenated args and terminate the application...
Definition: MooseError.h:311
std::vector< mfem::Array< int > > _ess_tdof_lists
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...
std::vector< std::string > _test_var_names
Names of all test variables corresponding to linear forms in this equation system.
mfem::OperatorHandle _linear_operator
virtual bool Complex() const
Whether this a complex equation system.

◆ SetSolverRequiresGradient()

void Moose::MFEM::EquationSystem::SetSolverRequiresGradient ( bool  requires_gradient)
inlineinherited

Set whether the nonlinear solver driving this equation system requires Jacobian information.

Definition at line 66 of file EquationSystem.h.

67  {
68  _solver_requires_gradient = requires_gradient;
69  }

◆ SetTimeStep()

virtual void Moose::MFEM::TimeDependentEquationSystem::SetTimeStep ( mfem::real_t &  dt)
inlinevirtual

◆ SetTrialVariableNames()

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

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

Definition at line 77 of file EquationSystem.C.

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

78 {
79  // If a coupled variable has an equation associated with it,
80  // add it to the set of trial variables.
81  for (const auto & test_var_name : _test_var_names)
82  if (VectorContainsName(_coupled_var_names, test_var_name))
83  _trial_var_names.push_back(test_var_name);
84 
85  // Otherwise, add it to the set of eliminated variables.
86  for (const auto & coupled_var_name : _coupled_var_names)
87  if (!VectorContainsName(_test_var_names, coupled_var_name))
88  _eliminated_var_names.push_back(coupled_var_name);
89 }
bool VectorContainsName(const std::vector< std::string > &the_vector, const std::string &name) const
std::vector< std::string > _eliminated_var_names
Names of all coupled variables without a corresponding test variable.
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::string > _test_var_names
Names of all test variables corresponding to linear forms in this equation system.
std::vector< std::string > _coupled_var_names
Names of all trial variables of kernels and boundary conditions added to this EquationSystem.

◆ SetTrialVariablesFromTrueVectors()

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

Update variable from solution vector after solve.

Reimplemented in Moose::MFEM::ComplexEquationSystem.

Definition at line 472 of file EquationSystem.C.

Referenced by Moose::MFEM::EquationSystem::ComputeNonlinearResidual().

473 {
474  for (const auto i : index_range(_trial_var_names))
475  {
476  auto & trial_var_name = _trial_var_names.at(i);
477  trueX.GetBlock(i).SyncMemory(trueX);
478  _gfuncs->Get(trial_var_name)->Distribute(&(trueX.GetBlock(i)));
479  }
480 }
std::vector< std::string > _trial_var_names
Subset of _coupled_var_names of all variables corresponding to gridfunctions with degrees of freedom ...
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...
auto index_range(const T &sizable)
Moose::MFEM::GridFunctions * _gfuncs

◆ VectorContainsName()

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

Definition at line 49 of file EquationSystem.C.

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

51 {
52  return std::find(the_vector.begin(), the_vector.end(), name) != the_vector.end();
53 }
std::string name(const ElemQuality q)
KOKKOS_INLINE_FUNCTION const T * find(const T &target, const T *const begin, const T *const end)
Find a value in an array.
Definition: KokkosUtils.h:40

Friends And Related Function Documentation

◆ TimeDependentEquationSystemProblemOperator

Definition at line 49 of file TimeDependentEquationSystem.h.

Member Data Documentation

◆ _assembly_level

mfem::AssemblyLevel Moose::MFEM::EquationSystem::_assembly_level
protectedinherited

◆ _blfs

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

◆ _block_true_offsets

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

◆ _coupled_pfespaces

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

Pointers to finite element spaces associated with coupled variables.

Definition at line 236 of file EquationSystem.h.

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

◆ _coupled_var_names

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

◆ _dt

mfem::real_t Moose::MFEM::TimeDependentEquationSystem::_dt
protected

◆ _eliminated_var_names

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

◆ _eliminated_variables

Moose::MFEM::GridFunctions Moose::MFEM::EquationSystem::_eliminated_variables
protectedinherited

Pointers to coupled variables not part of the reduced EquationSystem.

Definition at line 230 of file EquationSystem.h.

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

◆ _ess_tdof_lists

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

◆ _essential_bc_map

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

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

Named according to test variable.

Definition at line 257 of file EquationSystem.h.

Referenced by Moose::MFEM::EquationSystem::AddEssentialBC(), and Moose::MFEM::EquationSystem::ApplyEssentialBC().

◆ _gfuncs

Moose::MFEM::GridFunctions* Moose::MFEM::EquationSystem::_gfuncs
protectedinherited

◆ _h_blocks

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

◆ _integrated_bc_map

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

◆ _jacobian

mfem::OperatorHandle Moose::MFEM::EquationSystem::_jacobian
mutableprotectedinherited

◆ _jacobian_blocks

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

◆ _kernels_map

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

◆ _lfs

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

◆ _linear_operator

mfem::OperatorHandle Moose::MFEM::EquationSystem::_linear_operator
mutableprotectedinherited

◆ _mblfs

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

◆ _nlfs

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

◆ _non_linear

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

◆ _solver_requires_gradient

bool Moose::MFEM::EquationSystem::_solver_requires_gradient = false
protectedinherited

◆ _td_blfs

Moose::MFEM::NamedFieldsMap<mfem::ParBilinearForm> Moose::MFEM::TimeDependentEquationSystem::_td_blfs
protected

Containers to store contributions to weak form of the form (F du/dt, v)

Definition at line 41 of file TimeDependentEquationSystem.h.

Referenced by BuildBilinearForms(), and EliminateCoupledVariables().

◆ _td_kernels_map

Moose::MFEM::NamedFieldsMap<Moose::MFEM::NamedFieldsMap<std::vector<std::shared_ptr<MFEMKernel> > > > Moose::MFEM::TimeDependentEquationSystem::_td_kernels_map
protected

◆ _td_mblfs

Moose::MFEM::NamedFieldsMap<Moose::MFEM::NamedFieldsMap<mfem::ParMixedBilinearForm> > Moose::MFEM::TimeDependentEquationSystem::_td_mblfs
protected

◆ _test_pfespaces

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

◆ _test_var_names

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

◆ _time_derivative_map

const Moose::MFEM::TimeDerivativeMap& Moose::MFEM::TimeDependentEquationSystem::_time_derivative_map
protected

Map between variable names and their time derivatives.

Definition at line 46 of file TimeDependentEquationSystem.h.

Referenced by AddKernel().

◆ _trial_var_names

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

◆ _var_ess_constraints

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

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