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 "MFEMCGSolver.h"
      13             : #include "MFEMProblem.h"
      14             : 
      15             : registerMooseObject("MooseApp", MFEMCGSolver);
      16             : 
      17             : InputParameters
      18        2190 : MFEMCGSolver::validParams()
      19             : {
      20        2190 :   InputParameters params = Moose::MFEM::LinearSolverBase::validParams();
      21        4380 :   params.addClassDescription("MFEM native solver for the iterative solution of MFEM equation "
      22             :                              "systems using the conjugate gradient method.");
      23        4380 :   params.set<bool>("use_initial_guess", /*quiet_mode=*/true) = true;
      24        8760 :   params.addParam<mfem::real_t>("l_tol", 1e-5, "Set the relative tolerance.");
      25        8760 :   params.addParam<mfem::real_t>("l_abs_tol", 1e-50, "Set the absolute tolerance.");
      26        8760 :   params.addParam<int>("l_max_its", 10000, "Set the maximum number of iterations.");
      27        8760 :   params.addParam<int>("print_level", 2, "Set the solver verbosity.");
      28        6570 :   params.addParam<MFEMSolverName>("preconditioner", "Optional choice of preconditioner to use.");
      29             : 
      30        2190 :   return params;
      31           0 : }
      32             : 
      33          46 : MFEMCGSolver::MFEMCGSolver(const InputParameters & parameters)
      34          46 :   : Moose::MFEM::LinearSolverBase(parameters)
      35             : {
      36          46 :   ConstructSolver();
      37          46 : }
      38             : 
      39             : void
      40          46 : MFEMCGSolver::ConstructSolver()
      41             : {
      42          46 :   auto solver = std::make_unique<mfem::CGSolver>(getMFEMProblem().getComm());
      43          92 :   solver->iterative_mode = getParam<bool>("use_initial_guess");
      44          92 :   solver->SetRelTol(getParam<mfem::real_t>("l_tol"));
      45          92 :   solver->SetAbsTol(getParam<mfem::real_t>("l_abs_tol"));
      46          92 :   solver->SetMaxIter(getParam<int>("l_max_its"));
      47          92 :   solver->SetPrintLevel(getParam<int>("print_level"));
      48          46 :   SetPreconditioner(*solver);
      49          46 :   _solver = std::move(solver);
      50          46 : }
      51             : 
      52             : void
      53          42 : MFEMCGSolver::SetupLOR(mfem::ParBilinearForm & a, mfem::Array<int> & tdofs)
      54             : {
      55          42 :   if (_lor && _preconditioner)
      56           0 :     mooseError("LOR solver cannot take a preconditioner");
      57             : 
      58          42 :   if (_preconditioner)
      59             :   {
      60          38 :     _preconditioner->SetupLOR(a, tdofs);
      61          38 :     SetPreconditioner(static_cast<mfem::CGSolver &>(*_solver));
      62             :   }
      63           4 :   else if (_lor)
      64             :   {
      65           2 :     CheckSpectralEquivalence(a);
      66           2 :     auto lor_solver = new mfem::LORSolver<mfem::CGSolver>(a, tdofs);
      67           4 :     lor_solver->GetSolver().SetRelTol(getParam<mfem::real_t>("l_tol"));
      68           4 :     lor_solver->GetSolver().SetAbsTol(getParam<mfem::real_t>("l_abs_tol"));
      69           4 :     lor_solver->GetSolver().SetMaxIter(getParam<int>("l_max_its"));
      70           4 :     lor_solver->GetSolver().SetPrintLevel(getParam<int>("print_level"));
      71             : 
      72           2 :     _solver.reset(lor_solver);
      73             :   }
      74          42 : }
      75             : 
      76             : #endif