MFEMSteady.C

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html
 
#ifdef MOOSE_MFEM_ENABLED
 
#include "MFEMSteady.h"
#include "MFEMProblem.h"
#include "MFEMEigenproblem.h"
#include "EigenproblemEquationSystem.h"
#include "EquationSystemProblemOperator.h"
#include "EigenproblemESProblemOperator.h"
 
registerMooseObject("MooseApp", MFEMSteady);
 
InputParameters
MFEMSteady::validParams()
{
  InputParameters params = MFEMProblemSolve::validParams();
  params += Executioner::validParams();
  params.addClassDescription("Executioner for steady state MFEM problems.");
  params.addParam<Real>("time", 0.0, "System time");
  return params;
}
 
MFEMSteady::MFEMSteady(const InputParameters & params)
  : Executioner(params),
    _mfem_problem(dynamic_cast<MFEMProblem &>(feProblem())),
    _mfem_problem_data(_mfem_problem.getProblemData()),
    _mfem_problem_solve(*this, getProblemOperators()),
    _system_time(getParam<Real>("time")),
    _time_step(_mfem_problem.timeStep()),
    _time([this]() -> Real & { return this->_mfem_problem.time() = this->_system_time; }()),
    _last_solve_converged(false)
{
  // If no ProblemOperators have been added by the user, add a default
  if (!_mfem_problem.getProblemComposer())
  {
    std::string name = "__DefaultWeakFormProblemComposer";
    InputParameters params = _factory.getValidParams("MFEMWeakFormProblemComposer");
 
    if (dynamic_cast<MFEMEigenproblem *>(&_mfem_problem))
      _mfem_problem.addMFEMProblemComposer("MFEMEigenWeakFormProblemComposer", name, params);
    else if (_mfem_problem.getNumericType() == MFEMProblem::NumericType::COMPLEX)
      _mfem_problem.addMFEMProblemComposer("MFEMComplexWeakFormProblemComposer", name, params);
    else
      _mfem_problem.addMFEMProblemComposer("MFEMWeakFormProblemComposer", name, params);
  }
  addProblemOperator(_mfem_problem.getProblemComposer()->createProblemOperator(_mfem_problem));
}
 
void
MFEMSteady::init()
{
  _mfem_problem.execute(EXEC_PRE_MULTIAPP_SETUP);
  _mfem_problem.initialSetup();
 
  if (_mfem_problem_data.eqn_system)
  {
    if (_mfem_problem_data.nonlinear_solver)
      _mfem_problem_data.eqn_system->SetGradientRequired(
          _mfem_problem_data.nonlinear_solver->RequiresGradient());
 
    _mfem_problem_data.eqn_system->SetCoefficientManager(_mfem_problem_data.coefficients);
 
    // Set up initial conditions
    _mfem_problem_data.eqn_system->Init(
        _mfem_problem_data.gridfunctions,
        _mfem_problem_data.cmplx_gridfunctions,
        getParam<MooseEnum>("assembly_level").getEnum<mfem::AssemblyLevel>());
  }
 
  for (const auto & problem_operator : getProblemOperators())
  {
    problem_operator->SetGridFunctions();
    problem_operator->Init(_mfem_problem_data.true_solution);
  }
}
 
void
MFEMSteady::execute()
{
  if (_app.isRecovering())
  {
    _console << "\nCannot recover steady solves!\nExiting...\n" << std::endl;
    return;
  }
 
  _time_step = 0;
  _time = _time_step;
  _mfem_problem.outputStep(EXEC_INITIAL);
  _time = _system_time;
 
  preExecute();
 
  _mfem_problem.advanceState();
 
  // first step in any steady state solve is always 1 (preserving backwards compatibility)
  _time_step = 1;
  _mfem_problem.timestepSetup();
  if (!_mfem_problem.execMultiApps(EXEC_TIMESTEP_BEGIN, true))
  {
    _last_solve_converged = false;
    return;
  }
  _mfem_problem.execute(EXEC_TIMESTEP_BEGIN);
 
  _last_solve_converged = _mfem_problem_solve.solve();
 
  _mfem_problem.computeIndicators();
  _mfem_problem.computeMarkers();
 
  // need to keep _time in sync with _time_step to get correct output
  _time = _time_step;
  _mfem_problem.execute(EXEC_TIMESTEP_END);
  _mfem_problem.execMultiApps(EXEC_TIMESTEP_END, true);
  _mfem_problem.outputStep(EXEC_TIMESTEP_END);
  _time = _system_time;
 
  {
    TIME_SECTION("final", 1, "Executing Final Objects")
    _mfem_problem.execMultiApps(EXEC_FINAL);
    _mfem_problem.finalizeMultiApps();
    _mfem_problem.postExecute();
    _mfem_problem.execute(EXEC_FINAL);
    _time = _time_step;
    _mfem_problem.outputStep(EXEC_FINAL);
    _time = _system_time;
  }
 
  postExecute();
}
 
#endif