Line data    Source code

       1             : //* This file is part of the MOOSE framework
       2             : //* https://mooseframework.inl.gov
       3             : //*
       4             : //* All rights reserved, see COPYRIGHT for full restrictions
       5             : //* https://github.com/idaholab/moose/blob/master/COPYRIGHT
       6             : //*
       7             : //* Licensed under LGPL 2.1, please see LICENSE for details
       8             : //* https://www.gnu.org/licenses/lgpl-2.1.html
       9             : 
      10             : #ifdef MOOSE_MFEM_ENABLED
      11             : 
      12             : #include "Executioner.h"
      13             : #include "MFEMProblemSolve.h"
      14             : #include "MFEMProblem.h"
      15             : 
      16             : InputParameters
      17        7076 : MFEMProblemSolve::validParams()
      18             : {
      19        7076 :   InputParameters params = emptyInputParameters();
      20       14152 :   params.addClassDescription("Solve object for MFEM problems.");
      21       28304 :   params.addParam<std::string>("device", "Run app on the chosen device.");
      22       28304 :   MooseEnum assembly_levels("legacy full element partial none", "legacy", true);
      23       21228 :   params.addParam<MooseEnum>("assembly_level", assembly_levels, "Matrix assembly level.");
      24       14152 :   return params;
      25        7076 : }
      26             : 
      27        1440 : MFEMProblemSolve::MFEMProblemSolve(
      28             :     Executioner & ex,
      29        1440 :     std::vector<std::shared_ptr<Moose::MFEM::ProblemOperatorBase>> & problem_operators)
      30             :   : SolveObject(ex),
      31        1440 :     _mfem_problem(dynamic_cast<MFEMProblem &>(_problem)),
      32        1440 :     _problem_operators(problem_operators)
      33             : {
      34        1440 :   if (const auto compute_device = _app.getComputeDevice())
      35         903 :     _app.setMFEMDevice(*compute_device, Moose::PassKey<MFEMProblemSolve>());
      36             :   else
      37        2148 :     _app.setMFEMDevice(isParamValid("device")    ? getParam<std::string>("device")
      38         187 :                        : _app.isUltimateMaster() ? "cpu"
      39             :                                                  : "",
      40        1977 :                        Moose::PassKey<MFEMProblemSolve>());
      41        1440 : }
      42             : 
      43             : bool
      44        2691 : MFEMProblemSolve::solve()
      45             : {
      46        2691 :   if (_mfem_problem.shouldSolve())
      47             :   {
      48        4270 :     for (const auto & problem_operator : _problem_operators)
      49        2135 :       problem_operator->Solve();
      50             : 
      51             :     // Short-circuit evaluation guarantees we only do one of p- or h-refinement between solves
      52        2161 :     while (_mfem_problem.pRefine() || _mfem_problem.hRefine())
      53             :     {
      54             :       // Reset linear solver and its preconditioner now the problem size has changed
      55          26 :       if (auto & solver = _mfem_problem.getProblemData().jacobian_solver)
      56             :       {
      57          26 :         if (auto * prec = solver->GetPreconditioner())
      58          26 :           prec->ConstructSolver();
      59          26 :         solver->ConstructSolver();
      60             :       }
      61             : 
      62             :       // Reset gridfunctions
      63          52 :       for (const auto & problem_operator : _problem_operators)
      64          26 :         problem_operator->SetGridFunctions();
      65             : 
      66             :       // Solve again
      67          52 :       for (const auto & problem_operator : _problem_operators)
      68          26 :         problem_operator->Solve();
      69             :     }
      70             :   }
      71             : 
      72        2691 :   _mfem_problem.displaceMesh();
      73             : 
      74        2691 :   return true;
      75             : }
      76             : 
      77             : #endif